agno 2.1.5__tar.gz → 2.1.6__tar.gz

{agno-2.1.5 → agno-2.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agno
-Version: 2.1.5
+Version: 2.1.6
 Summary: Agno: a lightweight library for building Multi-Agent Systems
 Author-email: Ashpreet Bedi <ashpreet@agno.com>
 Project-URL: homepage, https://agno.com

{agno-2.1.5 → agno-2.1.6}/agno/agent/agent.py

@@ -248,7 +248,7 @@ class Agent:
     # If True, store media in run output
     store_media: bool = True
     # If True, store tool results in run output
-    store_tool_results: bool = True
+    store_tool_messages: bool = True
     # If True, store history messages in run output
     store_history_messages: bool = True
 
@@ -388,7 +388,7 @@ class Agent:
         add_history_to_context: bool = False,
         num_history_runs: int = 3,
         store_media: bool = True,
-        store_tool_results: bool = True,
+        store_tool_messages: bool = True,
         store_history_messages: bool = True,
         knowledge: Optional[Knowledge] = None,
         knowledge_filters: Optional[Dict[str, Any]] = None,
@@ -490,7 +490,7 @@ class Agent:
             )
 
         self.store_media = store_media
-        self.store_tool_results = store_tool_results
+        self.store_tool_messages = store_tool_messages
         self.store_history_messages = store_history_messages
 
         self.knowledge = knowledge
@@ -1473,7 +1473,6 @@ class Agent:
 
     async def _arun(
         self,
-        input: Union[str, List, Dict, Message, BaseModel, List[Message]],
         run_response: RunOutput,
         session_id: str,
         session_state: Optional[Dict[str, Any]] = None,
@@ -1696,7 +1695,6 @@ class Agent:
 
     async def _arun_stream(
         self,
-        input: Union[str, List, Dict, Message, BaseModel, List[Message]],
         run_response: RunOutput,
         session_id: str,
         session_state: Optional[Dict[str, Any]] = None,
@@ -2133,7 +2131,6 @@ class Agent:
                 # Pass the new run_response to _arun
                 if stream:
                     return self._arun_stream(  # type: ignore
-                        input=validated_input,
                         run_response=run_response,
                         user_id=user_id,
                         response_format=response_format,
@@ -2152,7 +2149,6 @@ class Agent:
                     )  # type: ignore[assignment]
                 else:
                     return self._arun(  # type: ignore
-                        input=validated_input,
                         run_response=run_response,
                         user_id=user_id,
                         response_format=response_format,
@@ -4421,7 +4417,9 @@ class Agent:
 
             if len(parsed_messages) > 0:
                 tasks.append(
-                    self.memory_manager.acreate_user_memories(messages=parsed_messages, user_id=user_id, agent_id=self.id)
+                    self.memory_manager.acreate_user_memories(
+                        messages=parsed_messages, user_id=user_id, agent_id=self.id
+                    )
                 )
             else:
                 log_warning("Unable to add messages to memory")
@@ -5879,9 +5877,7 @@ class Agent:
             user_memories = self.memory_manager.get_user_memories(user_id=user_id)  # type: ignore
 
             if user_memories and len(user_memories) > 0:
-                system_message_content += (
-                    "You have access to memories from previous interactions with the user that you can use:\n\n"
-                )
+                system_message_content += "You have access to user info and preferences from previous interactions that you can use to personalize your response:\n\n"
                 system_message_content += "<memories_from_previous_interactions>"
                 for _memory in user_memories:  # type: ignore
                     system_message_content += f"\n- {_memory.memory}"
@@ -6155,9 +6151,7 @@ class Agent:
                 user_memories = self.memory_manager.get_user_memories(user_id=user_id)  # type: ignore
 
             if user_memories and len(user_memories) > 0:
-                system_message_content += (
-                    "You have access to memories from previous interactions with the user that you can use:\n\n"
-                )
+                system_message_content += "You have access to user info and preferences from previous interactions that you can use to personalize your response:\n\n"
                 system_message_content += "<memories_from_previous_interactions>"
                 for _memory in user_memories:  # type: ignore
                     system_message_content += f"\n- {_memory.memory}"
@@ -8837,8 +8831,6 @@ class Agent:
             run_response.input.audios = []
             run_response.input.files = []
 
-        # 2. RunOutput artifact media are skipped since we don't store them when store_media=False
-
         # 3. Scrub media from all messages
         if run_response.messages:
             for message in run_response.messages:
@@ -8869,23 +8861,36 @@ class Agent:
 
     def _scrub_tool_results_from_run_output(self, run_response: RunOutput) -> None:
         """
-        Remove all tool-related data from RunOutput when store_tool_results=False.
-        This includes tool calls, tool results, and tool-related message fields.
+        Remove all tool-related data from RunOutput when store_tool_messages=False.
+        This removes both the tool call and its corresponding result to maintain API consistency.
         """
-        # Remove tool results (messages with role="tool")
-        if run_response.messages:
-            run_response.messages = [msg for msg in run_response.messages if msg.role != "tool"]
-            # Also scrub tool-related fields from remaining messages
-            for message in run_response.messages:
-                self._scrub_tool_data_from_message(message)
+        if not run_response.messages:
+            return
+
+        # Step 1: Collect all tool_call_ids from tool result messages
+        tool_call_ids_to_remove = set()
+        for message in run_response.messages:
+            if message.role == "tool" and message.tool_call_id:
+                tool_call_ids_to_remove.add(message.tool_call_id)
+
+        # Step 2: Remove tool result messages (role="tool")
+        run_response.messages = [msg for msg in run_response.messages if msg.role != "tool"]
+
+        # Step 3: Remove the assistant messages related to the scrubbed tool calls
+        filtered_messages = []
+        for message in run_response.messages:
+            # Check if this assistant message made any of the tool calls we're removing
+            should_remove = False
+            if message.role == "assistant" and message.tool_calls:
+                for tool_call in message.tool_calls:
+                    if tool_call.get("id") in tool_call_ids_to_remove:
+                        should_remove = True
+                        break
+
+            if not should_remove:
+                filtered_messages.append(message)
 
-    def _scrub_tool_data_from_message(self, message: Message) -> None:
-        """Remove all tool-related data from a Message object."""
-        message.tool_calls = None
-        message.tool_call_id = None
-        message.tool_name = None
-        message.tool_args = None
-        message.tool_call_error = None
+        run_response.messages = filtered_messages
 
     def _scrub_history_messages_from_run_output(self, run_response: RunOutput) -> None:
         """
@@ -8907,7 +8912,7 @@ class Agent:
             self._scrub_media_from_run_output(run_response)
             scrubbed = True
 
-        if not self.store_tool_results:
+        if not self.store_tool_messages:
             self._scrub_tool_results_from_run_output(run_response)
             scrubbed = True
 

{agno-2.1.5 → agno-2.1.6}/agno/db/migrations/v1_to_v2.py

@@ -1,5 +1,6 @@
 """Migration utility to migrate your Agno tables from v1 to v2"""
 
+import gc
 import json
 from typing import Any, Dict, List, Optional, Union, cast
 
@@ -423,6 +424,18 @@ def migrate_table_in_batches(
 
         log_info(f"Completed batch {batch_count}: migrated {batch_size_actual} records")
 
+        # Explicit cleanup to free memory before next batch
+        del batch_content
+        if v1_table_type in ["agent_sessions", "team_sessions", "workflow_sessions"]:
+            del sessions
+        elif v1_table_type == "memories":
+            del memories
+
+        # Force garbage collection to return memory to OS
+        # This is necessary because Python's memory allocator retains memory after large operations
+        # See: https://github.com/sqlalchemy/sqlalchemy/issues/4616
+        gc.collect()
+
     log_info(f"✅ Migration completed for table {v1_table_name}: {total_migrated} total records migrated")
 
 

{agno-2.1.5 → agno-2.1.6}/agno/db/mongo/mongo.py

@@ -631,11 +631,7 @@ class MongoDb(BaseDb):
                 session_dict = session.to_dict()
 
                 # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                updated_at = (
-                    session_dict.get("updated_at")
-                    if preserve_updated_at and session_dict.get("updated_at")
-                    else int(time.time())
-                )
+                updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
 
                 if isinstance(session, AgentSession):
                     record = {
@@ -1096,7 +1092,7 @@ class MongoDb(BaseDb):
                     memory.memory_id = str(uuid4())
 
                 # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                updated_at = memory.updated_at if preserve_updated_at and memory.updated_at else current_time
+                updated_at = memory.updated_at if preserve_updated_at else current_time
 
                 record = {
                     "user_id": memory.user_id,

{agno-2.1.5 → agno-2.1.6}/agno/db/mongo/utils.py

@@ -16,8 +16,6 @@ except ImportError:
 
 
 # -- DB util methods --
-
-
 def create_collection_indexes(collection: Collection, collection_type: str) -> None:
     """Create all required indexes for a collection"""
     try:
@@ -58,8 +56,6 @@ def apply_pagination(
 
 
 # -- Metrics util methods --
-
-
 def calculate_date_metrics(date_to_process: date, sessions_data: dict) -> dict:
     """Calculate metrics for the given single date."""
     metrics = {

{agno-2.1.5 → agno-2.1.6}/agno/db/mysql/mysql.py

@@ -760,11 +760,7 @@ class MySQLDb(BaseDb):
                     for session in agent_sessions:
                         session_dict = session.to_dict()
                         # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                        updated_at = (
-                            session_dict.get("updated_at")
-                            if preserve_updated_at and session_dict.get("updated_at")
-                            else int(time.time())
-                        )
+                        updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
                         agent_data.append(
                             {
                                 "session_id": session_dict.get("session_id"),
@@ -816,11 +812,7 @@ class MySQLDb(BaseDb):
                     for session in team_sessions:
                         session_dict = session.to_dict()
                         # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                        updated_at = (
-                            session_dict.get("updated_at")
-                            if preserve_updated_at and session_dict.get("updated_at")
-                            else int(time.time())
-                        )
+                        updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
                         team_data.append(
                             {
                                 "session_id": session_dict.get("session_id"),
@@ -872,11 +864,7 @@ class MySQLDb(BaseDb):
                     for session in workflow_sessions:
                         session_dict = session.to_dict()
                         # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                        updated_at = (
-                            session_dict.get("updated_at")
-                            if preserve_updated_at and session_dict.get("updated_at")
-                            else int(time.time())
-                        )
+                        updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
                         workflow_data.append(
                             {
                                 "session_id": session_dict.get("session_id"),
@@ -1338,7 +1326,7 @@ class MySQLDb(BaseDb):
                     memory.memory_id = str(uuid4())
 
                 # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                updated_at = memory.updated_at if preserve_updated_at and memory.updated_at else current_time
+                updated_at = memory.updated_at if preserve_updated_at else current_time
                 bulk_data.append(
                     {
                         "memory_id": memory.memory_id,

{agno-2.1.5 → agno-2.1.6}/agno/db/postgres/postgres.py

@@ -740,12 +740,8 @@ class PostgresDb(BaseDb):
                 session_records = []
                 for agent_session in agent_sessions:
                     session_dict = agent_session.to_dict()
-                    # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                    updated_at = (
-                        session_dict.get("updated_at")
-                        if preserve_updated_at and session_dict.get("updated_at")
-                        else int(time.time())
-                    )
+                    # Use preserved updated_at if flag is set (even if None), otherwise use current time
+                    updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
                     session_records.append(
                         {
                             "session_id": session_dict.get("session_id"),
@@ -789,12 +785,8 @@ class PostgresDb(BaseDb):
                 session_records = []
                 for team_session in team_sessions:
                     session_dict = team_session.to_dict()
-                    # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                    updated_at = (
-                        session_dict.get("updated_at")
-                        if preserve_updated_at and session_dict.get("updated_at")
-                        else int(time.time())
-                    )
+                    # Use preserved updated_at if flag is set (even if None), otherwise use current time
+                    updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
                     session_records.append(
                         {
                             "session_id": session_dict.get("session_id"),
@@ -838,12 +830,8 @@ class PostgresDb(BaseDb):
                 session_records = []
                 for workflow_session in workflow_sessions:
                     session_dict = workflow_session.to_dict()
-                    # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                    updated_at = (
-                        session_dict.get("updated_at")
-                        if preserve_updated_at and session_dict.get("updated_at")
-                        else int(time.time())
-                    )
+                    # Use preserved updated_at if flag is set (even if None), otherwise use current time
+                    updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
                     session_records.append(
                         {
                             "session_id": session_dict.get("session_id"),
@@ -1286,8 +1274,8 @@ class PostgresDb(BaseDb):
                 if memory.memory_id is None:
                     memory.memory_id = str(uuid4())
 
-                # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                updated_at = memory.updated_at if preserve_updated_at and memory.updated_at else current_time
+                # Use preserved updated_at if flag is set (even if None), otherwise use current time
+                updated_at = memory.updated_at if preserve_updated_at else current_time
                 memory_records.append(
                     {
                         "memory_id": memory.memory_id,

{agno-2.1.5 → agno-2.1.6}/agno/db/singlestore/singlestore.py

@@ -841,6 +841,8 @@ class SingleStoreDb(BaseDb):
                     agent_data = []
                     for session in agent_sessions:
                         session_dict = session.to_dict()
+                        # Use preserved updated_at if flag is set, otherwise use current time
+                        updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
                         agent_data.append(
                             {
                                 "session_id": session_dict.get("session_id"),
@@ -853,7 +855,7 @@ class SingleStoreDb(BaseDb):
                                 "summary": session_dict.get("summary"),
                                 "metadata": session_dict.get("metadata"),
                                 "created_at": session_dict.get("created_at"),
-                                "updated_at": session_dict.get("created_at"),
+                                "updated_at": updated_at,
                             }
                         )
 
@@ -867,7 +869,7 @@ class SingleStoreDb(BaseDb):
                             summary=stmt.inserted.summary,
                             metadata=stmt.inserted.metadata,
                             runs=stmt.inserted.runs,
-                            updated_at=int(time.time()),
+                            updated_at=stmt.inserted.updated_at,
                         )
                         sess.execute(stmt, agent_data)
 
@@ -890,6 +892,8 @@ class SingleStoreDb(BaseDb):
                     team_data = []
                     for session in team_sessions:
                         session_dict = session.to_dict()
+                        # Use preserved updated_at if flag is set, otherwise use current time
+                        updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
                         team_data.append(
                             {
                                 "session_id": session_dict.get("session_id"),
@@ -902,7 +906,7 @@ class SingleStoreDb(BaseDb):
                                 "summary": session_dict.get("summary"),
                                 "metadata": session_dict.get("metadata"),
                                 "created_at": session_dict.get("created_at"),
-                                "updated_at": session_dict.get("created_at"),
+                                "updated_at": updated_at,
                             }
                         )
 
@@ -916,7 +920,7 @@ class SingleStoreDb(BaseDb):
                             summary=stmt.inserted.summary,
                             metadata=stmt.inserted.metadata,
                             runs=stmt.inserted.runs,
-                            updated_at=int(time.time()),
+                            updated_at=stmt.inserted.updated_at,
                         )
                         sess.execute(stmt, team_data)
 
@@ -939,6 +943,8 @@ class SingleStoreDb(BaseDb):
                     workflow_data = []
                     for session in workflow_sessions:
                         session_dict = session.to_dict()
+                        # Use preserved updated_at if flag is set, otherwise use current time
+                        updated_at = session_dict.get("updated_at") if preserve_updated_at else int(time.time())
                         workflow_data.append(
                             {
                                 "session_id": session_dict.get("session_id"),
@@ -951,7 +957,7 @@ class SingleStoreDb(BaseDb):
                                 "summary": session_dict.get("summary"),
                                 "metadata": session_dict.get("metadata"),
                                 "created_at": session_dict.get("created_at"),
-                                "updated_at": session_dict.get("created_at"),
+                                "updated_at": updated_at,
                             }
                         )
 
@@ -965,7 +971,7 @@ class SingleStoreDb(BaseDb):
                             summary=stmt.inserted.summary,
                             metadata=stmt.inserted.metadata,
                             runs=stmt.inserted.runs,
-                            updated_at=int(time.time()),
+                            updated_at=stmt.inserted.updated_at,
                         )
                         sess.execute(stmt, workflow_data)
 
@@ -1361,9 +1367,12 @@ class SingleStoreDb(BaseDb):
 
             # Prepare data for bulk insert
             memory_data = []
+            current_time = int(time.time())
             for memory in memories:
                 if memory.memory_id is None:
                     memory.memory_id = str(uuid4())
+                # Use preserved updated_at if flag is set, otherwise use current time
+                updated_at = memory.updated_at if preserve_updated_at else current_time
                 memory_data.append(
                     {
                         "memory_id": memory.memory_id,
@@ -1373,7 +1382,7 @@ class SingleStoreDb(BaseDb):
                         "agent_id": memory.agent_id,
                         "team_id": memory.team_id,
                         "topics": memory.topics,
-                        "updated_at": int(time.time()),
+                        "updated_at": updated_at,
                     }
                 )
 
@@ -1389,7 +1398,7 @@ class SingleStoreDb(BaseDb):
                         user_id=stmt.inserted.user_id,
                         agent_id=stmt.inserted.agent_id,
                         team_id=stmt.inserted.team_id,
-                        updated_at=int(time.time()),
+                        updated_at=stmt.inserted.updated_at,
                     )
                     sess.execute(stmt, memory_data)
 

{agno-2.1.5 → agno-2.1.6}/agno/db/sqlite/sqlite.py

@@ -717,11 +717,7 @@ class SqliteDb(BaseDb):
                     for session in agent_sessions:
                         serialized_session = serialize_session_json_fields(session.to_dict())
                         # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                        updated_at = (
-                            serialized_session.get("updated_at")
-                            if preserve_updated_at and serialized_session.get("updated_at")
-                            else int(time.time())
-                        )
+                        updated_at = serialized_session.get("updated_at") if preserve_updated_at else int(time.time())
                         agent_data.append(
                             {
                                 "session_id": serialized_session.get("session_id"),
@@ -776,11 +772,7 @@ class SqliteDb(BaseDb):
                     for session in team_sessions:
                         serialized_session = serialize_session_json_fields(session.to_dict())
                         # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                        updated_at = (
-                            serialized_session.get("updated_at")
-                            if preserve_updated_at and serialized_session.get("updated_at")
-                            else int(time.time())
-                        )
+                        updated_at = serialized_session.get("updated_at") if preserve_updated_at else int(time.time())
                         team_data.append(
                             {
                                 "session_id": serialized_session.get("session_id"),
@@ -835,11 +827,7 @@ class SqliteDb(BaseDb):
                     for session in workflow_sessions:
                         serialized_session = serialize_session_json_fields(session.to_dict())
                         # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                        updated_at = (
-                            serialized_session.get("updated_at")
-                            if preserve_updated_at and serialized_session.get("updated_at")
-                            else int(time.time())
-                        )
+                        updated_at = serialized_session.get("updated_at") if preserve_updated_at else int(time.time())
                         workflow_data.append(
                             {
                                 "session_id": serialized_session.get("session_id"),
@@ -1280,7 +1268,7 @@ class SqliteDb(BaseDb):
                     memory.memory_id = str(uuid4())
 
                 # Use preserved updated_at if flag is set and value exists, otherwise use current time
-                updated_at = memory.updated_at if preserve_updated_at and memory.updated_at else current_time
+                updated_at = memory.updated_at if preserve_updated_at else current_time
                 bulk_data.append(
                     {
                         "user_id": memory.user_id,

agno-2.1.6/agno/db/surrealdb/__init__.py

@@ -0,0 +1,3 @@
+from agno.db.surrealdb.surrealdb import SurrealDb
+
+__all__ = ["SurrealDb"]