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/tasks.py
+++ b/skyvern/forge/sdk/schemas/tasks.py
@@ -248,7 +248,7 @@ class Task(TaskBase):
        Otherwise return `None`.
        """
        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()
                return mapping.get(model_name)