reasoning-fix-ag-2692

agno-agi · Feb 12, 2025 · 8ff755b · 8ff755b
1 parent 551aadb
commit 8ff755b
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/libs/agno/agno/agent/agent.py b/libs/agno/agno/agent/agent.py
@@ -541,8 +541,7 @@ def _run(
                         self.run_response.content = model_response_chunk.content
                         self.run_response.created_at = model_response_chunk.created_at
                         yield self.create_run_response(
-                            content=model_response_chunk.content,
-                            created_at=model_response_chunk.created_at
+                            content=model_response_chunk.content, created_at=model_response_chunk.created_at
                         )
                 # If the model response is a tool_call_started, add the tool call to the run_response
                 elif model_response_chunk.event == ModelResponseEvent.tool_call_started.value:
@@ -2740,7 +2739,7 @@ def reason(self, run_messages: RunMessages) -> Iterator[RunResponse]:
                 reasoning_agent_messages=[ds_reasoning_message],
             )
         # Use Groq for reasoning
-        if reasoning_model.__class__.__name__ == "Groq" and "deepseek" in reasoning_model.id:
+        elif reasoning_model.__class__.__name__ == "Groq" and "deepseek" in reasoning_model.id:
             from agno.reasoning.groq import get_groq_reasoning, get_groq_reasoning_agent
 
             groq_reasoning_agent = self.reasoning_agent or get_groq_reasoning_agent(
@@ -2893,7 +2892,7 @@ async def areason(self, run_messages: RunMessages) -> Any:
                 reasoning_agent_messages=[ds_reasoning_message],
             )
         # Use Groq for reasoning
-        if reasoning_model.__class__.__name__ == "Groq" and "deepseek" in reasoning_model.id:
+        elif reasoning_model.__class__.__name__ == "Groq" and "deepseek" in reasoning_model.id:
             from agno.reasoning.groq import aget_groq_reasoning, get_groq_reasoning_agent
 
             groq_reasoning_agent = self.reasoning_agent or get_groq_reasoning_agent(