agno 2.0.0a1__py3-none-any.whl → 2.0.0rc1__py3-none-any.whl

agno/db/dynamo/dynamo.py

@@ -506,8 +506,6 @@ class DynamoDb(BaseDb):
             item = serialize_to_dynamo_item(serialized_session)
             self.client.put_item(TableName=table_name, Item=item)
 
-            log_debug(f"Upserted session with id '{session.session_id}'")
-
             return deserialize_session_result(serialized_session, session, deserialize)
 
         except Exception as e:
@@ -852,8 +850,6 @@ class DynamoDb(BaseDb):
 
             self.client.put_item(TableName=table_name, Item=item)
 
-            log_debug(f"Upserted user memory with id '{memory.memory_id}'")
-
             if not deserialize:
                 return memory_dict
 
@@ -1530,8 +1526,6 @@ class DynamoDb(BaseDb):
 
             self.client.put_item(TableName=table_name, Item=item)
 
-            log_debug(f"Upserted knowledge content with id '{knowledge_row.id}'")
-
             return knowledge_row
 
         except Exception as e:

agno/db/firestore/firestore.py

@@ -532,8 +532,6 @@ class FirestoreDb(BaseDb):
                 return None
             deserialized_session = deserialize_session_json_fields(result)
 
-            log_debug(f"Upserted session with id '{session.session_id}'")
-
             if not deserialize:
                 return deserialized_session
 
@@ -844,8 +842,6 @@ class FirestoreDb(BaseDb):
 
             doc_ref.set(update_doc, merge=True)
 
-            log_debug(f"Upserted user memory with id '{memory.memory_id}'")
-
             if not deserialize:
                 return update_doc
 
@@ -1170,8 +1166,6 @@ class FirestoreDb(BaseDb):
 
             doc_ref.set(update_doc, merge=True)
 
-            log_debug(f"Upserted knowledge source with id '{knowledge_row.id}'")
-
             return knowledge_row
 
         except Exception as e:

agno/db/in_memory/in_memory_db.py

@@ -280,8 +280,6 @@ class InMemoryDb(BaseDb):
                 session_dict["updated_at"] = session_dict.get("created_at")
                 self._sessions.append(session_dict)
 
-            log_debug(f"Upserted session with id '{session.session_id}'")
-
             if not deserialize:
                 return session_dict
 
@@ -464,8 +462,6 @@ class InMemoryDb(BaseDb):
             if not memory_updated:
                 self._memories.append(memory_dict)
 
-            log_debug(f"Upserted user memory with id '{memory.memory_id}'")
-
             if not deserialize:
                 return memory_dict
             return UserMemory.from_dict(memory_dict)
@@ -739,8 +735,6 @@ class InMemoryDb(BaseDb):
             if not item_updated:
                 self._knowledge.append(knowledge_dict)
 
-            log_debug(f"Upserted knowledge source with id '{knowledge_row.id}'")
-
             return knowledge_row
 
         except Exception as e:

agno/db/json/json_db.py

@@ -375,8 +375,6 @@ class JsonDb(BaseDb):
 
             self._write_json_file(self.session_table_name, sessions)
 
-            log_debug(f"Upserted session with id '{session.session_id}'")
-
             if not deserialize:
                 return session_dict
 
@@ -580,8 +578,6 @@ class JsonDb(BaseDb):
 
             self._write_json_file(self.memory_table_name, memories)
 
-            log_debug(f"Upserted user memory with id '{memory.memory_id}'")
-
             if not deserialize:
                 return memory_dict
             return UserMemory.from_dict(memory_dict)
@@ -874,8 +870,6 @@ class JsonDb(BaseDb):
 
             self._write_json_file(self.knowledge_table_name, knowledge_items)
 
-            log_debug(f"Upserted knowledge source with id '{knowledge_row.id}'")
-
             return knowledge_row
 
         except Exception as e:

agno/db/mongo/mongo.py

@@ -504,8 +504,6 @@ class MongoDb(BaseDb):
 
                 session = deserialize_session_json_fields(result)  # type: ignore
 
-                log_debug("Upserted session'")
-
                 if not deserialize:
                     return session
 
@@ -537,8 +535,6 @@ class MongoDb(BaseDb):
 
                 session = deserialize_session_json_fields(result)  # type: ignore
 
-                log_debug(f"Upserted session with id '{session.session_id}'")
-
                 if not deserialize:
                     return session
 
@@ -570,8 +566,6 @@ class MongoDb(BaseDb):
 
                 session = deserialize_session_json_fields(result)  # type: ignore
 
-                log_debug(f"Upserted session with id '{session.session_id}'")
-
                 if not deserialize:
                     return session
 

agno/db/mysql/utils.py

@@ -111,7 +111,6 @@ def is_valid_table(db_engine: Engine, table_name: str, table_type: str, db_schem
             log_warning(f"Missing columns {missing_columns} in table {db_schema}.{table_name}")
             return False
 
-        log_debug(f"Table {db_schema}.{table_name} has all expected columns")
         return True
     except Exception as e:
         log_error(f"Error validating table schema for {db_schema}.{table_name}: {e}")

agno/db/postgres/postgres.py

@@ -610,8 +610,6 @@ class PostgresDb(BaseDb):
                     row = result.fetchone()
                     session_dict = dict(row._mapping)
 
-                    log_debug(f"Upserted agent session with id '{session_dict.get('session_id')}'")
-
                     if session_dict is None or not deserialize:
                         return session_dict
                     return AgentSession.from_dict(session_dict)
@@ -648,8 +646,6 @@ class PostgresDb(BaseDb):
                     row = result.fetchone()
                     session_dict = dict(row._mapping)
 
-                    log_debug(f"Upserted team session with id '{session_dict.get('session_id')}'")
-
                     if session_dict is None or not deserialize:
                         return session_dict
                     return TeamSession.from_dict(session_dict)
@@ -686,8 +682,6 @@ class PostgresDb(BaseDb):
                     row = result.fetchone()
                     session_dict = dict(row._mapping)
 
-                    log_debug(f"Upserted workflow session with id '{session_dict.get('session_id')}'")
-
                     if session_dict is None or not deserialize:
                         return session_dict
                     return WorkflowSession.from_dict(session_dict)
@@ -1033,8 +1027,6 @@ class PostgresDb(BaseDb):
 
             memory_raw = dict(row._mapping)
 
-            log_debug(f"Upserted user memory with id '{memory.memory_id}'")
-
             if not memory_raw or not deserialize:
                 return memory_raw
 
@@ -1405,8 +1397,6 @@ class PostgresDb(BaseDb):
                     )
                     sess.execute(stmt)
 
-            log_debug(f"Upserted knowledge row with id '{knowledge_row.id}'")
-
             return knowledge_row
 
         except Exception as e:

agno/db/postgres/utils.py

@@ -107,7 +107,6 @@ def is_valid_table(db_engine: Engine, table_name: str, table_type: str, db_schem
             log_warning(f"Missing columns {missing_columns} in table {db_schema}.{table_name}")
             return False
 
-        log_debug(f"Table {db_schema}.{table_name} has all expected columns")
         return True
     except Exception as e:
         log_error(f"Error validating table schema for {db_schema}.{table_name}: {e}")

agno/db/redis/redis.py

@@ -572,8 +572,6 @@ class RedisDb(BaseDb):
                 if not success:
                     return None
 
-                log_debug(f"Upserted session with id '{session.session_id}'")
-
                 if not deserialize:
                     return data
 
@@ -1140,8 +1138,6 @@ class RedisDb(BaseDb):
             data = knowledge_row.model_dump()
             success = self._store_record("knowledge", knowledge_row.id, data)  # type: ignore
 
-            log_debug(f"Upserted knowledge content with id '{knowledge_row.id}'")
-
             return knowledge_row if success else None
 
         except Exception as e:

agno/db/singlestore/singlestore.py

@@ -697,8 +697,6 @@ class SingleStoreDb(BaseDb):
                     if row is None:
                         return None
 
-                    log_debug(f"Upserted session with id '{session_dict.get('session_id')}'")
-
                     if not deserialize:
                         return row._mapping
 
@@ -740,8 +738,6 @@ class SingleStoreDb(BaseDb):
                     if row is None:
                         return None
 
-                    log_debug(f"Upserted session with id '{session_dict.get('session_id')}'")
-
                     if not deserialize:
                         return row._mapping
 
@@ -783,8 +779,6 @@ class SingleStoreDb(BaseDb):
                     if row is None:
                         return None
 
-                    log_debug(f"Upserted session with id '{session_dict.get('session_id')}'")
-
                     if not deserialize:
                         return row._mapping
 
@@ -1117,8 +1111,6 @@ class SingleStoreDb(BaseDb):
                 if row is None:
                     return None
 
-            log_debug(f"Upserted user memory with id '{memory.memory_id}'")
-
             memory_raw = row._mapping
             if not memory_raw or not deserialize:
                 return memory_raw
@@ -1461,8 +1453,6 @@ class SingleStoreDb(BaseDb):
                 stmt = stmt.on_duplicate_key_update(**update_fields)
                 sess.execute(stmt)
 
-            log_debug(f"Upserted knowledge source with id '{knowledge_row.id}'")
-
             return knowledge_row
 
         except Exception as e:

agno/db/singlestore/utils.py

@@ -133,7 +133,6 @@ def is_valid_table(db_engine: Engine, table_name: str, table_type: str, db_schem
             log_warning(f"Missing columns {missing_columns} in table {table_ref}")
             return False
 
-        log_debug(f"Table {table_ref} has all expected columns")
         return True
 
     except Exception as e:

agno/db/sqlite/sqlite.py

@@ -995,8 +995,6 @@ class SqliteDb(BaseDb):
                 if row is None:
                     return None
 
-            log_debug(f"Upserted user memory with id '{memory.memory_id}'")
-
             memory_raw = row._mapping
             if not memory_raw or not deserialize:
                 return memory_raw
@@ -1358,8 +1356,6 @@ class SqliteDb(BaseDb):
                 )
                 sess.execute(stmt)
 
-            log_debug(f"Upserted knowledge content with id '{knowledge_row.id}'")
-
             return knowledge_row
 
         except Exception as e:

agno/db/sqlite/utils.py

@@ -91,7 +91,6 @@ def is_valid_table(db_engine: Engine, table_name: str, table_type: str, db_schem
             log_warning(f"Missing columns {missing_columns} in table {table_name}")
             return False
 
-        log_debug(f"Table {table_name} has all expected columns")
         return True
     except Exception as e:
         log_error(f"Error validating table schema for {table_name}: {e}")

agno/integrations/discord/client.py

@@ -8,6 +8,7 @@ from agno.agent.agent import Agent, RunOutput
 from agno.media import Audio, File, Image, Video
 from agno.team.team import Team, TeamRunOutput
 from agno.utils.log import log_info, log_warning
+from agno.utils.message import get_text_from_message
 
 try:
     import discord
@@ -167,7 +168,10 @@ class DiscordClient:
                 thread=thread, message=f"Reasoning: \n{response.reasoning_content}", italics=True
             )
 
-        await self._send_discord_messages(thread=thread, message=str(response.content))
+        # Handle structured outputs properly
+        content_message = get_text_from_message(response.content) if response.content is not None else ""
+
+        await self._send_discord_messages(thread=thread, message=content_message)
 
     async def _send_discord_messages(self, thread: discord.channel, message: str, italics: bool = False):  # type: ignore
         if len(message) < 1500:

agno/knowledge/embedder/aws_bedrock.py

@@ -272,7 +272,7 @@ class AwsBedrockEmbedder(Embedder):
                     contentType="application/json",
                     accept="application/json",
                 )
-                response_body = json.loads(response["body"].read().decode("utf-8"))
+                response_body = json.loads((await response["body"].read()).decode("utf-8"))
 
                 # Extract embeddings using the same logic as get_embedding
                 if "embeddings" in response_body:
@@ -308,7 +308,7 @@ class AwsBedrockEmbedder(Embedder):
                     contentType="application/json",
                     accept="application/json",
                 )
-                response_body = json.loads(response["body"].read().decode("utf-8"))
+                response_body = json.loads((await response["body"].read()).decode("utf-8"))
 
                 embedding: List[float] = []
                 # Extract embeddings using the same logic as get_embedding_and_usage

agno/models/anthropic/claude.py

@@ -13,7 +13,7 @@ from agno.models.metrics import Metrics
 from agno.models.response import ModelResponse
 from agno.run.agent import RunOutput
 from agno.utils.log import log_debug, log_error, log_warning
-from agno.utils.models.claude import MCPServerConfiguration, format_messages
+from agno.utils.models.claude import MCPServerConfiguration, format_messages, format_tools_for_model
 
 try:
     from anthropic import Anthropic as AnthropicClient
@@ -174,59 +174,12 @@ class Claude(Model):
                 request_kwargs["system"] = [{"text": system_message, "type": "text"}]
 
         if tools:
-            request_kwargs["tools"] = self._format_tools_for_model(tools)
+            request_kwargs["tools"] = format_tools_for_model(tools)
 
         if request_kwargs:
             log_debug(f"Calling {self.provider} with request parameters: {request_kwargs}", log_level=2)
         return request_kwargs
 
-    def _format_tools_for_model(self, tools: Optional[List[Dict[str, Any]]] = None) -> Optional[List[Dict[str, Any]]]:
-        """
-        Transforms function definitions into a format accepted by the Anthropic API.
-        """
-        if not tools:
-            return None
-
-        parsed_tools: List[Dict[str, Any]] = []
-        for tool_def in tools:
-            if tool_def.get("type", "") != "function":
-                parsed_tools.append(tool_def)
-                continue
-
-            func_def = tool_def.get("function", {})
-            parameters: Dict[str, Any] = func_def.get("parameters", {})
-            properties: Dict[str, Any] = parameters.get("properties", {})
-            required_params: List[str] = []
-
-            for param_name, param_info in properties.items():
-                param_type = param_info.get("type", "")
-                param_type_list: List[str] = [param_type] if isinstance(param_type, str) else param_type or []
-
-                if "null" not in param_type_list:
-                    required_params.append(param_name)
-
-            input_properties: Dict[str, Dict[str, Union[str, List[str]]]] = {}
-            for param_name, param_info in properties.items():
-                input_properties[param_name] = {
-                    "description": param_info.get("description", ""),
-                }
-                if "type" not in param_info and "anyOf" in param_info:
-                    input_properties[param_name]["anyOf"] = param_info["anyOf"]
-                else:
-                    input_properties[param_name]["type"] = param_info.get("type", "")
-
-            tool = {
-                "name": func_def.get("name") or "",
-                "description": func_def.get("description") or "",
-                "input_schema": {
-                    "type": parameters.get("type", "object"),
-                    "properties": input_properties,
-                    "required": required_params,
-                },
-            }
-            parsed_tools.append(tool)
-        return parsed_tools
-
     def invoke(
         self,
         messages: List[Message],

agno/models/message.py

@@ -228,12 +228,13 @@ class Message(BaseModel):
                             if isinstance(tool_call_arguments, dict)
                             else json.loads(tool_call_arguments)
                         )
-                        # Ensure tool_call_args is a dictionary before calling .items()
-                        if isinstance(tool_call_args, dict):
-                            arguments = ", ".join(f"{k}: {v}" for k, v in tool_call_args.items())
-                            tool_calls_list.append(f"    Arguments: '{arguments}'")
-                        else:
-                            tool_calls_list.append(f"    Arguments: '{tool_call_args}'")
+                        if tool_call_args:
+                            # Ensure tool_call_args is a dictionary before calling .items()
+                            if isinstance(tool_call_args, dict):
+                                arguments = ", ".join(f"{k}: {v}" for k, v in tool_call_args.items())
+                                tool_calls_list.append(f"    Arguments: '{arguments}'")
+                            else:
+                                tool_calls_list.append(f"    Arguments: '{tool_call_args}'")
                     except json.JSONDecodeError:
                         tool_calls_list.append("    Arguments: 'Invalid JSON format'")
             tool_calls_str = "\n".join(tool_calls_list)