use model_name instead of name (#2557)

2025-05-31 18:40:36 -07:00
parent deb38af17d
commit 800a26d323
5 changed files with 17 additions and 45 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("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)