Skip to content

Commit

Permalink
fix: updating agent's llm model (#823)
Browse files Browse the repository at this point in the history
  • Loading branch information
elisalimli authored Feb 14, 2024
1 parent a76fa8e commit b257e74
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion libs/superagent/app/api/agents.py
Original file line number Diff line number Diff line change
Expand Up @@ -366,7 +366,7 @@ async def update(
if old_llm_model and new_llm_model and old_llm_model != new_llm_model:
from app.utils.llm import get_llm_provider

new_provider = get_llm_provider(new_llm_model)
new_provider = get_llm_provider(LLM_MAPPING.get(new_llm_model))
new_llm = await prisma.llm.find_first_or_raise(
where={"provider": new_provider, "apiUserId": api_user.id}
)
Expand Down

0 comments on commit b257e74

Please sign in to comment.