From edc654edf9a0fd6886b666fdc04a96bd65784719 Mon Sep 17 00:00:00 2001 From: "huynguyen03.dev" Date: Sat, 6 Dec 2025 22:07:40 +0700 Subject: [PATCH] refactor: simplify provider check logic in amp routes Amp-Thread-ID: https://ampcode.com/threads/T-a18fd71c-32ce-4c29-93d7-09f082740e51 --- internal/api/modules/amp/routes.go | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/internal/api/modules/amp/routes.go b/internal/api/modules/amp/routes.go index 1f61d5ac..33525e61 100644 --- a/internal/api/modules/amp/routes.go +++ b/internal/api/modules/amp/routes.go @@ -188,14 +188,10 @@ func (m *AmpModule) registerManagementRoutes(engine *gin.Engine, baseHandler *ha if modelPart != "" { normalized, _ := util.NormalizeGeminiThinkingModel(modelPart) // Only handle locally when we have a provider or a valid mapping; otherwise fall back to proxy - hasProvider := false - if providers := util.GetProviderName(normalized); len(providers) > 0 { - hasProvider = true - } else if m.modelMapper != nil { + hasProvider := len(util.GetProviderName(normalized)) > 0 + if !hasProvider && m.modelMapper != nil { // Check if mapped model has provider (MapModel returns target only if it has providers) - if mapped := m.modelMapper.MapModel(normalized); mapped != "" { - hasProvider = true - } + hasProvider = m.modelMapper.MapModel(normalized) != "" } if hasProvider {