ModelSelector: add 'Skyvern Optimized' as hard choice (#2558)

2025-05-31 18:42:57 -07:00
parent 800a26d323
commit b4d5837196
6 changed files with 59 additions and 21 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)