fix llm_key_override (#2552)

2025-05-31 11:11:25 -07:00
parent 07bf256779
commit 48f5f0913e
10 changed files with 45 additions and 18 deletions
--- a/skyvern/forge/sdk/schemas/task_v2.py
+++ b/skyvern/forge/sdk/schemas/task_v2.py
@@ -57,7 +57,7 @@ class TaskV2(BaseModel):
        """

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