fix llm_key_override (#2552)

This commit is contained in:
Shuchang Zheng 2025-05-31 11:11:25 -07:00 committed by GitHub
parent 07bf256779
commit 48f5f0913e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 45 additions and 18 deletions

View file

@ -94,14 +94,14 @@ class Workflow(BaseModel):
mapping = settings.get_model_name_to_llm_key()
if block:
model_name = (block.model or {}).get("model")
model_name = (block.model or {}).get("name")
if model_name:
llm_key = mapping.get(model_name)
if llm_key:
return llm_key
workflow_model_name = (self.model or {}).get("model")
workflow_model_name = (self.model or {}).get("name")
if workflow_model_name:
llm_key = mapping.get(workflow_model_name)