Fix model_response.extra

agno-agi · Feb 12, 2025 · b5d3477 · b5d3477
1 parent c8dafa9
commit b5d3477
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 10 deletions.
diff --git a/libs/agno/agno/models/aws/claude.py b/libs/agno/agno/models/aws/claude.py
@@ -1,4 +1,3 @@
-
 from dataclasses import dataclass
 from os import getenv
 from typing import Any, Dict, Optional
@@ -115,4 +114,3 @@ def request_kwargs(self) -> Dict[str, Any]:
         if self.request_params:
             _request_params.update(self.request_params)
         return _request_params
-
diff --git a/libs/agno/agno/models/base.py b/libs/agno/agno/models/base.py
@@ -187,7 +187,7 @@ def response(self, messages: List[Message]) -> ModelResponse:
 
                 # Format and add results to messages
                 self.format_function_call_results(
-                    messages=messages, function_call_results=function_call_results, **model_response.extra
+                    messages=messages, function_call_results=function_call_results, **model_response.extra or {}
                 )
 
                 logger.debug(f"---------- {self.get_provider()} Response ----------")
@@ -249,7 +249,7 @@ async def aresponse(self, messages: List[Message]) -> ModelResponse:
 
                 # Format and add results to messages
                 self.format_function_call_results(
-                    messages=messages, function_call_results=function_call_results, **model_response.extra
+                    messages=messages, function_call_results=function_call_results, **model_response.extra or {}
                 )
 
                 # Check if we should stop after tool calls
@@ -288,7 +288,7 @@ def _process_model_response(
         assistant_message.metrics.stop_timer()
 
         # Parse provider response
-        provider_response = self.parse_provider_response(response)
+        provider_response: ModelResponse = self.parse_provider_response(response)
 
         # Add parsed data to model response
         if provider_response.parsed is not None:
@@ -338,7 +338,7 @@ async def _aprocess_model_response(
         assistant_message.metrics.stop_timer()
 
         # Parse provider response
-        provider_response = self.parse_provider_response(response)
+        provider_response: ModelResponse = self.parse_provider_response(response)
 
         # Add parsed data to model response
         if provider_response.parsed is not None:
@@ -1084,4 +1084,4 @@ def __deepcopy__(self, memo):
 
         # Clear the new model to remove any references to the old model
         new_model.clear()
-        return new_model
+        return new_model
diff --git a/libs/agno/agno/models/cohere/chat.py b/libs/agno/agno/models/cohere/chat.py
@@ -297,7 +297,6 @@ def _process_stream_response(
             assistant_message.tool_calls.append(tool_use)
             tool_use = {}
 
-
         elif (
             response.type == "message-end"
             and response.delta is not None

diff --git a/libs/agno/agno/models/groq/groq.py b/libs/agno/agno/models/groq/groq.py
@@ -338,7 +338,6 @@ def parse_tool_calls(tool_calls_data: List[ChoiceDeltaToolCall]) -> List[Dict[st
                     tool_call_entry["type"] = _tool_call_type
         return tool_calls
 
-
     def parse_provider_response(self, response: ChatCompletion) -> ModelResponse:
         """
         Parse the Groq response into a ModelResponse.

diff --git a/libs/agno/agno/models/response.py b/libs/agno/agno/models/response.py
@@ -35,7 +35,6 @@ class ModelResponse:
     extra: Optional[Dict[str, Any]] = None
 
 
-
 class FileType(str, Enum):
     MP4 = "mp4"
     GIF = "gif"