agno 2.1.6__tar.gz → 2.1.7__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agno
-Version: 2.1.6
+Version: 2.1.7
 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.7/agno/db/postgres/__init__.py

@@ -0,0 +1,4 @@
+from agno.db.async_postgres import AsyncPostgresDb
+from agno.db.postgres.postgres import PostgresDb
+
+__all__ = ["PostgresDb", "AsyncPostgresDb"]

{agno-2.1.6 → agno-2.1.7}/agno/memory/manager.py

@@ -116,11 +116,17 @@ class MemoryManager:
 
     async def aread_from_db(self, user_id: Optional[str] = None):
         if self.db:
-            # If no user_id is provided, read all memories
-            if user_id is None:
-                all_memories: List[UserMemory] = await self.db.get_user_memories()  # type: ignore
+            if isinstance(self.db, AsyncBaseDb):
+                # If no user_id is provided, read all memories
+                if user_id is None:
+                    all_memories: List[UserMemory] = await self.db.get_user_memories()  # type: ignore
+                else:
+                    all_memories = await self.db.get_user_memories(user_id=user_id)  # type: ignore
             else:
-                all_memories = await self.db.get_user_memories(user_id=user_id)  # type: ignore
+                if user_id is None:
+                    all_memories = self.db.get_user_memories()  # type: ignore
+                else:
+                    all_memories = self.db.get_user_memories(user_id=user_id)  # type: ignore
 
             memories: Dict[str, List[UserMemory]] = {}
             for memory in all_memories:
@@ -434,7 +440,11 @@ class MemoryManager:
         if user_id is None:
             user_id = "default"
 
-        memories = self.read_from_db(user_id=user_id)
+        if isinstance(self.db, AsyncBaseDb):
+            memories = await self.aread_from_db(user_id=user_id)
+        else:
+            memories = self.read_from_db(user_id=user_id)
+
         if memories is None:
             memories = {}
 
@@ -453,7 +463,10 @@ class MemoryManager:
         )
 
         # We refresh from the DB
-        self.read_from_db(user_id=user_id)
+        if isinstance(self.db, AsyncBaseDb):
+            await self.aread_from_db(user_id=user_id)
+        else:
+            self.read_from_db(user_id=user_id)
 
         return response
 

{agno-2.1.6 → agno-2.1.7}/agno/team/team.py

@@ -3759,7 +3759,7 @@ class Team:
 
             if not member.store_media or not member.store_tool_messages or not member.store_history_messages:
                 member._scrub_run_output_for_storage(member_response)  # type: ignore
-            
+
             # If this is a nested team, recursively scrub its member responses
             if isinstance(member_response, TeamRunOutput) and member_response.member_responses:
                 member._scrub_member_responses(member_response.member_responses)  # type: ignore

{agno-2.1.6 → agno-2.1.7}/agno.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agno
-Version: 2.1.6
+Version: 2.1.7
 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.6 → agno-2.1.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agno"
-version = "2.1.6"
+version = "2.1.7"
 description = "Agno: a lightweight library for building Multi-Agent Systems"
 requires-python = ">=3.7,<4"
 readme = "README.md"

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

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