Fix amp v1beta1 routing and gemini retry config

internal/api/modules/amp/routes.go

@@ -10,6 +10,7 @@ import (
 	"github.com/router-for-me/CLIProxyAPI/v6/sdk/api/handlers/claude"
 	"github.com/router-for-me/CLIProxyAPI/v6/sdk/api/handlers/gemini"
 	"github.com/router-for-me/CLIProxyAPI/v6/sdk/api/handlers/openai"
+	"github.com/router-for-me/CLIProxyAPI/v6/internal/util"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -105,7 +106,31 @@ func (m *AmpModule) registerManagementRoutes(engine *gin.Engine, baseHandler *ha
 	geminiV1Beta1Fallback := NewFallbackHandler(func() *httputil.ReverseProxy {
 		return m.proxy
 	})
-	ampAPI.POST("/provider/google/v1beta1/*path", geminiV1Beta1Fallback.WrapHandler(geminiBridge))
+	geminiV1Beta1Handler := geminiV1Beta1Fallback.WrapHandler(geminiBridge)
+
+	// Route POST model calls through Gemini bridge when a local provider exists, otherwise proxy.
+	// All other methods (e.g., GET model listing) always proxy to upstream to preserve Amp CLI behavior.
+	ampAPI.Any("/provider/google/v1beta1/*path", func(c *gin.Context) {
+		if c.Request.Method == "POST" {
+			// Attempt to extract the model name from the AMP-style path
+			if path := c.Param("path"); strings.Contains(path, "/models/") {
+				modelPart := path[strings.Index(path, "/models/")+len("/models/"):]
+				if colonIdx := strings.Index(modelPart, ":"); colonIdx > 0 {
+					modelPart = modelPart[:colonIdx]
+				}
+				if modelPart != "" {
+					normalized, _ := util.NormalizeGeminiThinkingModel(modelPart)
+					// Only handle locally when we have a provider; otherwise fall back to proxy
+					if providers := util.GetProviderName(normalized); len(providers) > 0 {
+						geminiV1Beta1Handler(c)
+						return
+					}
+				}
+			}
+		}
+		// Non-POST or no local provider available -> proxy upstream
+		proxyHandler(c)
+	})
 }
 
 // registerProviderAliases registers /api/provider/{provider}/... routes

internal/api/modules/amp/routes_test.go

@@ -21,34 +21,40 @@ func TestRegisterManagementRoutes(t *testing.T) {
 	}
 
 	m := &AmpModule{}
-	m.registerManagementRoutes(r, proxyHandler, false) // false = don't restrict to localhost in tests
+	base := &handlers.BaseAPIHandler{}
+	m.registerManagementRoutes(r, base, proxyHandler, false) // false = don't restrict to localhost in tests
 
-	managementPaths := []string{
-		"/api/internal",
-		"/api/internal/some/path",
-		"/api/user",
-		"/api/user/profile",
-		"/api/auth",
-		"/api/auth/login",
-		"/api/meta",
-		"/api/telemetry",
-		"/api/threads",
-		"/api/otel",
-		"/api/provider/google/v1beta1/models",
+	managementPaths := []struct {
+		path   string
+		method string
+	}{
+		{"/api/internal", http.MethodGet},
+		{"/api/internal/some/path", http.MethodGet},
+		{"/api/user", http.MethodGet},
+		{"/api/user/profile", http.MethodGet},
+		{"/api/auth", http.MethodGet},
+		{"/api/auth/login", http.MethodGet},
+		{"/api/meta", http.MethodGet},
+		{"/api/telemetry", http.MethodGet},
+		{"/api/threads", http.MethodGet},
+		{"/api/otel", http.MethodGet},
+		// Google v1beta1 bridge should still proxy non-model requests (GET) and allow POST
+		{"/api/provider/google/v1beta1/models", http.MethodGet},
+		{"/api/provider/google/v1beta1/models", http.MethodPost},
 	}
 
 	for _, path := range managementPaths {
-		t.Run(path, func(t *testing.T) {
+		t.Run(path.path, func(t *testing.T) {
 			proxyCalled = false
-			req := httptest.NewRequest(http.MethodGet, path, nil)
+			req := httptest.NewRequest(path.method, path.path, nil)
 			w := httptest.NewRecorder()
 			r.ServeHTTP(w, req)
 
 			if w.Code == http.StatusNotFound {
-				t.Fatalf("route %s not registered", path)
+				t.Fatalf("route %s not registered", path.path)
 			}
 			if !proxyCalled {
-				t.Fatalf("proxy handler not called for %s", path)
+				t.Fatalf("proxy handler not called for %s", path.path)
 			}
 		})
 	}