Jon/model name massage (#2559)

2025-05-31 19:34:30 -07:00
parent b4d5837196
commit 2167d88c20
11 changed files with 67 additions and 60 deletions
--- a/skyvern/forge/sdk/routes/agent_protocol.py
+++ b/skyvern/forge/sdk/routes/agent_protocol.py
@@ -810,9 +810,9 @@ async def models() -> ModelsResponse:
    Get a list of available models.
    """
    mapping = settings.get_model_name_to_llm_key()
-    models = list(mapping.keys())
+    just_labels = {k: v["label"] for k, v in mapping.items() if "anthropic" not in k.lower()}

-    return ModelsResponse(models=models)
+    return ModelsResponse(models=just_labels)


@legacy_base_router.post(