Fix agent bug (#658)

CHange
cohere-ai · Aug 14, 2024 · e8d0ceb · e8d0ceb
1 parent 3b89f69
commit e8d0ceb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/backend/routers/conversation.py b/src/backend/routers/conversation.py
@@ -599,7 +599,7 @@ async def generate_title(
     agent_id = conversation.agent_id if conversation.agent_id else None
 
     if agent_id:
-        agent = agent_crud.get_agent_by_id(session, agent_id)
+        agent = agent_crud.get_agent_by_id(session, agent_id, user_id)
         agent_schema = Agent.model_validate(agent)
         ctx.with_agent(agent_schema)
         ctx.with_metrics_agent(agent_to_metrics_agent(agent))