agno 1.7.6__py3-none-any.whl → 1.7.8__py3-none-any.whl

agno/tools/postgres.py

@@ -2,14 +2,12 @@ import csv
 from typing import Any, Dict, List, Optional
 
 try:
-    import psycopg2
-    from psycopg2 import sql
-    from psycopg2.extensions import connection as PgConnection
-    from psycopg2.extras import DictCursor
+    import psycopg
+    from psycopg import sql
+    from psycopg.connection import Connection as PgConnection
+    from psycopg.rows import DictRow, dict_row
 except ImportError:
-    raise ImportError(
-        "`psycopg2` not installed. Please install using `pip install psycopg2`. If you face issues, try `pip install psycopg2-binary`."
-    )
+    raise ImportError("`psycopg` not installed. Please install using `pip install 'psycopg-binary'`.")
 
 from agno.tools import Toolkit
 from agno.utils.log import log_debug, log_error
@@ -18,7 +16,7 @@ from agno.utils.log import log_debug, log_error
 class PostgresTools(Toolkit):
     def __init__(
         self,
-        connection: Optional[PgConnection] = None,
+        connection: Optional[PgConnection[DictRow]] = None,
         db_name: Optional[str] = None,
         user: Optional[str] = None,
         password: Optional[str] = None,
@@ -31,7 +29,7 @@ class PostgresTools(Toolkit):
         table_schema: str = "public",
         **kwargs,
     ):
-        self._connection: Optional[PgConnection] = connection
+        self._connection: Optional[PgConnection[DictRow]] = connection
         self.db_name: Optional[str] = db_name
         self.user: Optional[str] = user
         self.password: Optional[str] = password
@@ -55,16 +53,16 @@ class PostgresTools(Toolkit):
         super().__init__(name="postgres_tools", tools=tools, **kwargs)
 
     @property
-    def connection(self) -> PgConnection:
+    def connection(self) -> PgConnection[DictRow]:
         """
-        Returns the Postgres psycopg2 connection.
-        :return psycopg2.extensions.connection: psycopg2 connection
+        Returns the Postgres psycopg connection.
+        :return psycopg.connection.Connection: psycopg connection
         """
         if self._connection is None or self._connection.closed:
             log_debug("Establishing new PostgreSQL connection.")
-            connection_kwargs: Dict[str, Any] = {"cursor_factory": DictCursor}
+            connection_kwargs: Dict[str, Any] = {"row_factory": dict_row}
             if self.db_name:
-                connection_kwargs["database"] = self.db_name
+                connection_kwargs["dbname"] = self.db_name
             if self.user:
                 connection_kwargs["user"] = self.user
             if self.password:
@@ -76,8 +74,8 @@ class PostgresTools(Toolkit):
 
             connection_kwargs["options"] = f"-c search_path={self.table_schema}"
 
-            self._connection = psycopg2.connect(**connection_kwargs)
-            self._connection.set_session(readonly=True)
+            self._connection = psycopg.connect(**connection_kwargs)
+            self._connection.read_only = True
 
         return self._connection
 
@@ -110,10 +108,10 @@ class PostgresTools(Toolkit):
                     return f"Query returned no results.\nColumns: {', '.join(columns)}"
 
                 header = ",".join(columns)
-                data_rows = [",".join(map(str, row)) for row in rows]
+                data_rows = [",".join(map(str, row.values())) for row in rows]
                 return f"{header}\n" + "\n".join(data_rows)
 
-        except psycopg2.Error as e:
+        except psycopg.Error as e:
             log_error(f"Database error: {e}")
             if self.connection and not self.connection.closed:
                 self.connection.rollback()
@@ -203,15 +201,18 @@ class PostgresTools(Toolkit):
                         cursor.execute(query)
                         stats = cursor.fetchone()
                         summary_parts.append(f"\n--- Column: {col_name} (Type: {data_type}) ---")
-                        for key, value in stats.items():
-                            val_str = (
-                                f"{value:.2f}" if isinstance(value, (float, int)) and value is not None else str(value)
-                            )
-                            summary_parts.append(f"  {key}: {val_str}")
+                        if stats is not None:
+                            for key, value in stats.items():
+                                val_str = (
+                                    f"{value:.2f}" if isinstance(value, float) and value is not None else str(value)
+                                )
+                                summary_parts.append(f"  {key}: {val_str}")
+                        else:
+                            summary_parts.append("  No statistics available")
 
                 return "\n".join(summary_parts)
 
-        except psycopg2.Error as e:
+        except psycopg.Error as e:
             return f"Error summarizing table: {e}"
 
     def inspect_query(self, query: str) -> str:
@@ -239,15 +240,19 @@ class PostgresTools(Toolkit):
         try:
             with self.connection.cursor() as cursor:
                 cursor.execute(stmt)
+
+                if cursor.description is None:
+                    return f"Error: Query returned no description for table '{table}'."
+
                 columns = [desc[0] for desc in cursor.description]
 
                 with open(path, "w", newline="", encoding="utf-8") as f:
                     writer = csv.writer(f)
                     writer.writerow(columns)
-                    writer.writerows(cursor)
+                    writer.writerows(row.values() for row in cursor)
 
             return f"Successfully exported table '{table}' to '{path}'."
-        except (psycopg2.Error, IOError) as e:
+        except (psycopg.Error, IOError) as e:
             return f"Error exporting table: {e}"
 
     def run_query(self, query: str) -> str:

agno/tools/zep.py

@@ -118,7 +118,7 @@ class ZepTools(Toolkit):
 
             # Create session associated with the user
             try:
-                self.zep_client.memory.add_session(session_id=self.session_id, user_id=self.user_id)
+                self.zep_client.thread.create(thread_id=self.session_id, user_id=self.user_id)
                 log_debug(f"Created session {self.session_id} for user {self.user_id}")
             except Exception as e:
                 log_debug(f"Session may already exist: {e}")
@@ -148,7 +148,7 @@ class ZepTools(Toolkit):
 
         try:
             zep_message = ZepMessage(
-                role_type=role,
+                role=role,
                 content=content,
             )
 
@@ -156,8 +156,8 @@ class ZepTools(Toolkit):
             ignore_roles_list = ["assistant"] if self.ignore_assistant_messages else None
 
             # Add message to Zep memory
-            self.zep_client.memory.add(  # type: ignore
-                session_id=self.session_id,
+            self.zep_client.thread.add_messages(  # type: ignore
+                thread_id=self.session_id,
                 messages=[zep_message],
                 ignore_roles=ignore_roles_list,
             )
@@ -171,7 +171,7 @@ class ZepTools(Toolkit):
         """
         Retrieves the memory for the current Zep session.
         Args:
-            memory_type: The type of memory to retrieve ('context', 'messages', 'relevant_facts').
+            memory_type: The type of memory to retrieve ('context', 'messages').
         Returns:
             The requested memory content as a string, or an error string.
         """
@@ -181,21 +181,16 @@ class ZepTools(Toolkit):
 
         try:
             log_debug(f"Getting Zep memory for session {self.session_id}")
-            memory_data = self.zep_client.memory.get(session_id=self.session_id)  # type: ignore
 
             if memory_type == "context":
                 # Ensure context is a string
-                return memory_data.context or "No context available."
+                user_context = self.zep_client.thread.get_user_context(thread_id=self.session_id, mode="basic")  # type: ignore
+                log_debug(f"Memory data: {user_context}")
+                return user_context.context or "No context available."
             elif memory_type == "messages":
+                messages_list = self.zep_client.thread.get(thread_id=self.session_id)  # type: ignore
                 # Ensure messages string representation is returned
-                return str(memory_data.messages) if memory_data.messages else "No messages available."
-            elif memory_type == "relevant_facts":
-                # Return all relevant facts from memory
-                if memory_data.relevant_facts:
-                    facts_str = "\n".join([f"- {fact.fact}" for fact in memory_data.relevant_facts])
-                    return f"Relevant facts:\n{facts_str}"
-                else:
-                    return "No relevant facts available."
+                return str(messages_list.messages) if messages_list.messages else "No messages available."
             else:
                 warning_msg = f"Unsupported memory_type requested: {memory_type}. Returning empty string."
                 log_warning(warning_msg)
@@ -328,7 +323,7 @@ class ZepAsyncTools(Toolkit):
 
             # Create session associated with the user
             try:
-                await self.zep_client.memory.add_session(session_id=self.session_id, user_id=self.user_id)  # type: ignore
+                await self.zep_client.thread.create(thread_id=self.session_id, user_id=self.user_id)  # type: ignore
                 log_debug(f"Created session {self.session_id} for user {self.user_id}")
             except Exception as e:
                 log_debug(f"Session may already exist: {e}")
@@ -361,7 +356,7 @@ class ZepAsyncTools(Toolkit):
 
         try:
             zep_message = ZepMessage(
-                role_type=role,
+                role=role,
                 content=content,
             )
 
@@ -369,8 +364,8 @@ class ZepAsyncTools(Toolkit):
             ignore_roles_list = ["assistant"] if self.ignore_assistant_messages else None
 
             # Add message to Zep memory
-            await self.zep_client.memory.add(  # type: ignore
-                session_id=self.session_id,
+            await self.zep_client.thread.add_messages(  # type: ignore
+                thread_id=self.session_id,
                 messages=[zep_message],
                 ignore_roles=ignore_roles_list,
             )
@@ -384,7 +379,7 @@ class ZepAsyncTools(Toolkit):
         """
         Retrieves the memory for the current Zep session.
         Args:
-            memory_type: The type of memory to retrieve ('context', 'messages', 'relevant_facts').
+            memory_type: The type of memory to retrieve ('context', 'messages').
         Returns:
             The requested memory content as a string, or an error string.
         """
@@ -396,25 +391,19 @@ class ZepAsyncTools(Toolkit):
             return "Error: Zep client/session not initialized."
 
         try:
-            memory_data = await self.zep_client.memory.get(session_id=self.session_id)  # type: ignore
-
             if memory_type == "context":
                 # Ensure context is a string
-                return memory_data.context or "No context available."
+                user_context = await self.zep_client.thread.get_user_context(thread_id=self.session_id, mode="basic")  # type: ignore
+                log_debug(f"Memory data: {user_context}")
+                return user_context.context or "No context available."
             elif memory_type == "messages":
                 # Ensure messages string representation is returned
-                return str(memory_data.messages) if memory_data.messages else "No messages available."
-            elif memory_type == "relevant_facts":
-                # Return relevant facts from memory
-                if memory_data.relevant_facts:
-                    facts_str = "\n".join([f"- {fact.fact}" for fact in memory_data.relevant_facts])
-                    return f"Relevant facts:\n{facts_str}"
-                else:
-                    return "No relevant facts available."
+                messages_list = await self.zep_client.thread.get(thread_id=self.session_id)  # type: ignore
+                return str(messages_list.messages) if messages_list.messages else "No messages available."
             else:
                 warning_msg = f"Unsupported memory_type requested: {memory_type}. Returning context."
                 log_warning(warning_msg)
-                return memory_data.context or "No context available."
+                return "No context available."
 
         except Exception as e:
             error_msg = f"Failed to get Zep memory for session {self.session_id}: {e}"

agno/utils/events.py

@@ -7,6 +7,8 @@ from agno.reasoning.step import ReasoningStep
 from agno.run.response import (
     MemoryUpdateCompletedEvent,
     MemoryUpdateStartedEvent,
+    OutputModelResponseCompletedEvent,
+    OutputModelResponseStartedEvent,
     ParserModelResponseCompletedEvent,
     ParserModelResponseStartedEvent,
     ReasoningCompletedEvent,
@@ -25,6 +27,8 @@ from agno.run.response import (
 )
 from agno.run.team import MemoryUpdateCompletedEvent as TeamMemoryUpdateCompletedEvent
 from agno.run.team import MemoryUpdateStartedEvent as TeamMemoryUpdateStartedEvent
+from agno.run.team import OutputModelResponseCompletedEvent as TeamOutputModelResponseCompletedEvent
+from agno.run.team import OutputModelResponseStartedEvent as TeamOutputModelResponseStartedEvent
 from agno.run.team import ParserModelResponseCompletedEvent as TeamParserModelResponseCompletedEvent
 from agno.run.team import ParserModelResponseStartedEvent as TeamParserModelResponseStartedEvent
 from agno.run.team import ReasoningCompletedEvent as TeamReasoningCompletedEvent
@@ -356,6 +360,7 @@ def create_run_response_content_event(
     content: Optional[Any] = None,
     content_type: Optional[str] = None,
     thinking: Optional[str] = None,
+    reasoning_content: Optional[str] = None,
     redacted_thinking: Optional[str] = None,
     citations: Optional[Citations] = None,
     response_audio: Optional[AudioResponse] = None,
@@ -372,6 +377,7 @@ def create_run_response_content_event(
         content=content,
         content_type=content_type or "str",
         thinking=thinking_combined,
+        reasoning_content=reasoning_content,
         citations=citations,
         response_audio=response_audio,
         image=image,
@@ -452,3 +458,47 @@ def create_team_parser_model_response_completed_event(
         team_session_id=from_run_response.team_session_id,  # type: ignore
         run_id=from_run_response.run_id,
     )
+
+
+def create_output_model_response_started_event(from_run_response: RunResponse) -> OutputModelResponseStartedEvent:
+    return OutputModelResponseStartedEvent(
+        session_id=from_run_response.session_id,
+        agent_id=from_run_response.agent_id,  # type: ignore
+        agent_name=from_run_response.agent_name,  # type: ignore
+        team_session_id=from_run_response.team_session_id,  # type: ignore
+        run_id=from_run_response.run_id,
+    )
+
+
+def create_output_model_response_completed_event(from_run_response: RunResponse) -> OutputModelResponseCompletedEvent:
+    return OutputModelResponseCompletedEvent(
+        session_id=from_run_response.session_id,
+        agent_id=from_run_response.agent_id,  # type: ignore
+        agent_name=from_run_response.agent_name,  # type: ignore
+        team_session_id=from_run_response.team_session_id,  # type: ignore
+        run_id=from_run_response.run_id,
+    )
+
+
+def create_team_output_model_response_started_event(
+    from_run_response: TeamRunResponse,
+) -> TeamOutputModelResponseStartedEvent:
+    return TeamOutputModelResponseStartedEvent(
+        session_id=from_run_response.session_id,
+        team_id=from_run_response.team_id,  # type: ignore
+        team_name=from_run_response.team_name,  # type: ignore
+        team_session_id=from_run_response.team_session_id,  # type: ignore
+        run_id=from_run_response.run_id,
+    )
+
+
+def create_team_output_model_response_completed_event(
+    from_run_response: TeamRunResponse,
+) -> TeamOutputModelResponseCompletedEvent:
+    return TeamOutputModelResponseCompletedEvent(
+        session_id=from_run_response.session_id,
+        team_id=from_run_response.team_id,  # type: ignore
+        team_name=from_run_response.team_name,  # type: ignore
+        team_session_id=from_run_response.team_session_id,  # type: ignore
+        run_id=from_run_response.run_id,
+    )

agno/utils/models/claude.py

@@ -277,6 +277,7 @@ def format_messages(messages: List[Message]) -> Tuple[List[Dict[str, str]], str]
                             type="tool_use",
                         )
                     )
+
         # Skip empty assistant responses
         if message.role == "assistant" and not content:
             continue

agno/utils/response.py

@@ -65,13 +65,15 @@ def format_tool_calls(tool_calls: List[ToolExecution]) -> List[str]:
         List[str]: List of formatted tool call strings
     """
     formatted_tool_calls = []
+
     for tool_call in tool_calls:
-        if tool_call.tool_name and tool_call.tool_args:
+        if tool_call.tool_name is not None:
             tool_name = tool_call.tool_name
             args_str = ""
             if tool_call.tool_args is not None:
                 args_str = ", ".join(f"{k}={v}" for k, v in tool_call.tool_args.items())
             formatted_tool_calls.append(f"{tool_name}({args_str})")
+
     return formatted_tool_calls
 
 

agno/workflow/v2/workflow.py

@@ -1631,7 +1631,7 @@ class Workflow:
         videos: Optional[List[Video]] = None,
         stream: bool = False,
         stream_intermediate_steps: bool = False,
-        markdown: bool = True,
+        markdown: bool = False,
         show_time: bool = True,
         show_step_details: bool = True,
         console: Optional[Any] = None,
@@ -1699,7 +1699,7 @@ class Workflow:
         audio: Optional[List[Audio]] = None,
         images: Optional[List[Image]] = None,
         videos: Optional[List[Video]] = None,
-        markdown: bool = True,
+        markdown: bool = False,
         show_time: bool = True,
         show_step_details: bool = True,
         console: Optional[Any] = None,
@@ -1856,7 +1856,7 @@ class Workflow:
         images: Optional[List[Image]] = None,
         videos: Optional[List[Video]] = None,
         stream_intermediate_steps: bool = False,
-        markdown: bool = True,
+        markdown: bool = False,
         show_time: bool = True,
         show_step_details: bool = True,
         console: Optional[Any] = None,
@@ -2407,7 +2407,7 @@ class Workflow:
         videos: Optional[List[Video]] = None,
         stream: bool = False,
         stream_intermediate_steps: bool = False,
-        markdown: bool = True,
+        markdown: bool = False,
         show_time: bool = True,
         show_step_details: bool = True,
         console: Optional[Any] = None,
@@ -2471,7 +2471,7 @@ class Workflow:
         audio: Optional[List[Audio]] = None,
         images: Optional[List[Image]] = None,
         videos: Optional[List[Video]] = None,
-        markdown: bool = True,
+        markdown: bool = False,
         show_time: bool = True,
         show_step_details: bool = True,
         console: Optional[Any] = None,
@@ -2629,7 +2629,7 @@ class Workflow:
         images: Optional[List[Image]] = None,
         videos: Optional[List[Video]] = None,
         stream_intermediate_steps: bool = False,
-        markdown: bool = True,
+        markdown: bool = False,
         show_time: bool = True,
         show_step_details: bool = True,
         console: Optional[Any] = None,

{agno-1.7.6.dist-info → agno-1.7.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agno
-Version: 1.7.6
+Version: 1.7.8
 Summary: Agno: a lightweight library for building Multi-Agent Systems
 Author-email: Ashpreet Bedi <ashpreet@agno.com>
 License: Copyright (c) Agno, Inc.
@@ -421,7 +421,6 @@ Requires-Dist: types-pyyaml; extra == "dev"
 Requires-Dist: types-aiofiles; extra == "dev"
 Requires-Dist: fastapi; extra == "dev"
 Requires-Dist: uvicorn; extra == "dev"
-Requires-Dist: arxiv; extra == "dev"
 Provides-Extra: integration-tests
 Requires-Dist: exa_py; extra == "integration-tests"
 Requires-Dist: duckduckgo-search; extra == "integration-tests"
@@ -481,6 +480,8 @@ Provides-Extra: agentql
 Requires-Dist: agentql; extra == "agentql"
 Provides-Extra: apify
 Requires-Dist: apify-client; extra == "apify"
+Provides-Extra: arxiv
+Requires-Dist: arxiv; extra == "arxiv"
 Provides-Extra: brave
 Requires-Dist: brave-search; extra == "brave"
 Provides-Extra: browserbase
@@ -528,8 +529,9 @@ Requires-Dist: newspaper4k; extra == "newspaper"
 Requires-Dist: lxml_html_clean; extra == "newspaper"
 Provides-Extra: opencv
 Requires-Dist: opencv-python; extra == "opencv"
-Provides-Extra: psycopg2
-Requires-Dist: psycopg2-binary; extra == "psycopg2"
+Provides-Extra: psycopg
+Requires-Dist: psycopg-binary; extra == "psycopg"
+Requires-Dist: psycopg; extra == "psycopg"
 Provides-Extra: todoist
 Requires-Dist: todoist-api-python; extra == "todoist"
 Provides-Extra: valyu
@@ -550,7 +552,6 @@ Provides-Extra: sql
 Requires-Dist: sqlalchemy; extra == "sql"
 Provides-Extra: postgres
 Requires-Dist: psycopg-binary; extra == "postgres"
-Requires-Dist: psycopg; extra == "postgres"
 Provides-Extra: sqlite
 Requires-Dist: sqlalchemy; extra == "sqlite"
 Provides-Extra: gcs
@@ -632,6 +633,7 @@ Requires-Dist: agno[openai]; extra == "models"
 Requires-Dist: agno[portkey]; extra == "models"
 Provides-Extra: tools
 Requires-Dist: agno[apify]; extra == "tools"
+Requires-Dist: agno[arxiv]; extra == "tools"
 Requires-Dist: agno[brave]; extra == "tools"
 Requires-Dist: agno[exa]; extra == "tools"
 Requires-Dist: agno[cartesia]; extra == "tools"
@@ -660,7 +662,7 @@ Requires-Dist: agno[oxylabs]; extra == "tools"
 Requires-Dist: agno[zep]; extra == "tools"
 Requires-Dist: agno[mem0]; extra == "tools"
 Requires-Dist: agno[google_bigquery]; extra == "tools"
-Requires-Dist: agno[psycopg2]; extra == "tools"
+Requires-Dist: agno[psycopg]; extra == "tools"
 Provides-Extra: storage
 Requires-Dist: agno[sql]; extra == "storage"
 Requires-Dist: agno[postgres]; extra == "storage"