diff --git a/internal/runtime/executor/antigravity_executor.go b/internal/runtime/executor/antigravity_executor.go index 55cc1626..f475289f 100644 --- a/internal/runtime/executor/antigravity_executor.go +++ b/internal/runtime/executor/antigravity_executor.go @@ -994,7 +994,7 @@ func FetchAntigravityModels(ctx context.Context, auth *cliproxyauth.Auth, cfg *c now := time.Now().Unix() modelConfig := registry.GetAntigravityModelConfig() models := make([]*registry.ModelInfo, 0, len(result.Map())) - for originalName := range result.Map() { + for originalName, modelData := range result.Map() { modelID := strings.TrimSpace(originalName) if modelID == "" { continue @@ -1004,12 +1004,18 @@ func FetchAntigravityModels(ctx context.Context, auth *cliproxyauth.Auth, cfg *c continue } modelCfg := modelConfig[modelID] - modelName := modelID + + // Extract displayName from upstream response, fallback to modelID + displayName := modelData.Get("displayName").String() + if displayName == "" { + displayName = modelID + } + modelInfo := ®istry.ModelInfo{ ID: modelID, - Name: modelName, - Description: modelID, - DisplayName: modelID, + Name: modelID, + Description: displayName, + DisplayName: displayName, Version: modelID, Object: "model", Created: now, diff --git a/sdk/api/handlers/gemini/gemini_handlers.go b/sdk/api/handlers/gemini/gemini_handlers.go index 27d8d1f5..71c485ad 100644 --- a/sdk/api/handlers/gemini/gemini_handlers.go +++ b/sdk/api/handlers/gemini/gemini_handlers.go @@ -60,8 +60,12 @@ func (h *GeminiAPIHandler) GeminiModels(c *gin.Context) { if !strings.HasPrefix(name, "models/") { normalizedModel["name"] = "models/" + name } - normalizedModel["displayName"] = name - normalizedModel["description"] = name + if displayName, _ := normalizedModel["displayName"].(string); displayName == "" { + normalizedModel["displayName"] = name + } + if description, _ := normalizedModel["description"].(string); description == "" { + normalizedModel["description"] = name + } } if _, ok := normalizedModel["supportedGenerationMethods"]; !ok { normalizedModel["supportedGenerationMethods"] = defaultMethods