vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-15 17:59:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

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

Browse source
This commit is contained in:
Shuchang Zheng 2025-05-31 18:42:57 -07:00 • committed by GitHub
parent 800a26d323
commit b4d5837196
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 59 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

2
skyvern/forge/sdk/schemas/tasks.py
View file

@ -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)