Fix models endpoint

Author: dvcrn

internal/server/logging_middleware.go

@@ -0,0 +1,32 @@
+package server
+
+import (
+	"net/http"
+	"time"
+
+	"github.com/dvcrn/gemini-code-assist-proxy/internal/logger"
+)
+
+// loggingMiddleware logs all incoming requests
+func loggingMiddleware(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		start := time.Now()
+
+		// Log the request
+		logger.Get().Info().
+			Str("method", r.Method).
+			Str("url", r.URL.String()).
+			Str("remote_addr", r.RemoteAddr).
+			Msg("Incoming request")
+
+		// Call the next handler
+		next.ServeHTTP(w, r)
+
+		// Log the response
+		logger.Get().Info().
+			Str("method", r.Method).
+			Str("url", r.URL.String()).
+			Dur("duration", time.Since(start)).
+			Msg("Finished request")
+	})
+}

internal/server/server.go

@@ -48,7 +48,7 @@ func (s *Server) Start(addr string) error {
 	s.startTokenRefreshLoop()
 
 	logger.Get().Info().Msgf("Starting proxy server on %s", addr)
-	return http.ListenAndServe(addr, s.mux)
+	return http.ListenAndServe(addr, loggingMiddleware(s.mux))
 }
 
 // LoadCredentials loads OAuth credentials using the configured provider
@@ -123,6 +123,7 @@ func (s *Server) setupRoutes() {
 	s.mux.HandleFunc("/admin/credentials/status", s.adminMiddleware(s.credentialsStatusHandler))
 	s.mux.HandleFunc("/v1beta/models/", s.adminMiddleware(s.streamGenerateContentHandler))
 	s.mux.HandleFunc("/v1/models/", s.modelsHandler)
+	s.mux.HandleFunc("/v1/models", s.modelsHandler)
 	s.mux.HandleFunc("/v1/chat/completions", s.adminMiddleware(s.openAIChatCompletionsHandler))
 }