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/schemas/task_v2.py
+++ b/skyvern/forge/sdk/schemas/task_v2.py
@@ -57,10 +57,10 @@ class TaskV2(BaseModel):
        """

        if self.model:
-            model_name = self.model.get("name")
+            model_name = self.model.get("model_name")
            if model_name:
                mapping = settings.get_model_name_to_llm_key()
-                llm_key = mapping.get(model_name)
+                llm_key = mapping.get(model_name, {}).get("llm_key")
                if llm_key:
                    return llm_key