agno 2.3.5__py3-none-any.whl → 2.3.6__py3-none-any.whl

agno/agent/agent.py

@@ -1645,7 +1645,10 @@ class Agent:
 
         # Initialize session state
         session_state = self._initialize_session_state(
-            session_state=session_state or {}, user_id=user_id, session_id=session_id, run_id=run_id
+            session_state=session_state if session_state is not None else {},
+            user_id=user_id,
+            session_id=session_id,
+            run_id=run_id,
         )
         # Update session state from DB
         session_state = self._load_session_state(session=agent_session, session_state=session_state)
@@ -1866,7 +1869,7 @@ class Agent:
         self._update_metadata(session=agent_session)
         # Initialize session state
         run_context.session_state = self._initialize_session_state(
-            session_state=run_context.session_state or {},
+            session_state=run_context.session_state if run_context.session_state is not None else {},
             user_id=user_id,
             session_id=session_id,
             run_id=run_response.run_id,
@@ -2144,7 +2147,7 @@ class Agent:
         self._update_metadata(session=agent_session)
         # Initialize session state
         run_context.session_state = self._initialize_session_state(
-            session_state=run_context.session_state or {},
+            session_state=run_context.session_state if run_context.session_state is not None else {},
             user_id=user_id,
             session_id=session_id,
             run_id=run_response.run_id,
@@ -6988,7 +6991,7 @@ class Agent:
 
         # Should already be resolved and passed from run() method
         format_variables = ChainMap(
-            session_state or {},
+            session_state if session_state is not None else {},
             dependencies or {},
             metadata or {},
             {"user_id": user_id} if user_id is not None else {},

agno/db/postgres/async_postgres.py

@@ -32,6 +32,7 @@ from agno.utils.log import log_debug, log_error, log_info, log_warning
 try:
     from sqlalchemy import Index, String, Table, UniqueConstraint, func, update
     from sqlalchemy.dialects import postgresql
+    from sqlalchemy.exc import ProgrammingError
     from sqlalchemy.ext.asyncio import AsyncEngine, async_sessionmaker, create_async_engine
     from sqlalchemy.schema import Column, MetaData
     from sqlalchemy.sql.expression import select, text
@@ -897,10 +898,19 @@ class AsyncPostgresDb(AsyncBaseDb):
             table = await self._get_table(table_type="memories")
 
             async with self.async_session_factory() as sess, sess.begin():
-                stmt = select(func.json_array_elements_text(table.c.topics))
-                if user_id is not None:
-                    stmt = stmt.where(table.c.user_id == user_id)
-                result = await sess.execute(stmt)
+                try:
+                    stmt = select(func.jsonb_array_elements_text(table.c.topics))
+                    if user_id is not None:
+                        stmt = stmt.where(table.c.user_id == user_id)
+                    result = await sess.execute(stmt)
+                except ProgrammingError:
+                    # Retrying with json_array_elements_text. This works in older versions,
+                    # where the topics column was of type JSON instead of JSONB
+                    stmt = select(func.json_array_elements_text(table.c.topics))
+                    if user_id is not None:
+                        stmt = stmt.where(table.c.user_id == user_id)
+                    result = await sess.execute(stmt)
+
                 records = result.fetchall()
 
                 return list(set([record[0] for record in records]))

agno/db/postgres/postgres.py

@@ -33,6 +33,7 @@ try:
     from sqlalchemy import ForeignKey, Index, String, UniqueConstraint, func, select, update
     from sqlalchemy.dialects import postgresql
     from sqlalchemy.engine import Engine, create_engine
+    from sqlalchemy.exc import ProgrammingError
     from sqlalchemy.orm import scoped_session, sessionmaker
     from sqlalchemy.schema import Column, MetaData, Table
     from sqlalchemy.sql.expression import text
@@ -1082,9 +1083,14 @@ class PostgresDb(BaseDb):
                 return []
 
             with self.Session() as sess, sess.begin():
-                stmt = select(func.json_array_elements_text(table.c.topics))
-
-                result = sess.execute(stmt).fetchall()
+                try:
+                    stmt = select(func.jsonb_array_elements_text(table.c.topics))
+                    result = sess.execute(stmt).fetchall()
+                except ProgrammingError:
+                    # Retrying with json_array_elements_text. This works in older versions,
+                    # where the topics column was of type JSON instead of JSONB
+                    stmt = select(func.json_array_elements_text(table.c.topics))
+                    result = sess.execute(stmt).fetchall()
 
                 return list(set([record[0] for record in result]))
 

agno/db/sqlite/async_sqlite.py

@@ -1114,7 +1114,7 @@ class AsyncSqliteDb(AsyncBaseDb):
 
             async with self.async_session_factory() as sess, sess.begin():
                 # Select topics from all results
-                stmt = select(func.json_array_elements_text(table.c.topics)).select_from(table)
+                stmt = select(table.c.topics)
                 result = (await sess.execute(stmt)).fetchall()
 
                 return list(set([record[0] for record in result]))

agno/db/sqlite/sqlite.py

@@ -245,9 +245,9 @@ class SqliteDb(BaseDb):
             return table
 
         except Exception as e:
-            from traceback import format_exc
+            from traceback import print_exc
 
-            print(format_exc())
+            print_exc()
             log_error(f"Could not create table '{table_name}': {e}")
             raise e
 
@@ -1109,9 +1109,8 @@ class SqliteDb(BaseDb):
 
             with self.Session() as sess, sess.begin():
                 # Select topics from all results
-                stmt = select(func.json_array_elements_text(table.c.topics)).select_from(table)
+                stmt = select(table.c.topics)
                 result = sess.execute(stmt).fetchall()
-
                 return list(set([record[0] for record in result]))
 
         except Exception as e:

agno/eval/accuracy.py

@@ -359,10 +359,12 @@ Remember: You must only compare the agent_output to the expected_output. The exp
                 status = Status(f"Running evaluation {i + 1}...", spinner="dots", speed=1.0, refresh_per_second=10)
                 live_log.update(status)
 
+                agent_session_id = f"eval_{self.eval_id}_{i + 1}"
+
                 if self.agent is not None:
-                    output = self.agent.run(input=eval_input).content
+                    output = self.agent.run(input=eval_input, session_id=agent_session_id).content
                 elif self.team is not None:
-                    output = self.team.run(input=eval_input).content
+                    output = self.team.run(input=eval_input, session_id=agent_session_id).content
 
                 if not output:
                     logger.error(f"Failed to generate a valid answer on iteration {i + 1}: {output}")
@@ -500,11 +502,13 @@ Remember: You must only compare the agent_output to the expected_output. The exp
                 status = Status(f"Running evaluation {i + 1}...", spinner="dots", speed=1.0, refresh_per_second=10)
                 live_log.update(status)
 
+                agent_session_id = f"eval_{self.eval_id}_{i + 1}"
+
                 if self.agent is not None:
-                    response = await self.agent.arun(input=eval_input)
+                    response = await self.agent.arun(input=eval_input, session_id=agent_session_id)
                     output = response.content
                 elif self.team is not None:
-                    response = await self.team.arun(input=eval_input)  # type: ignore
+                    response = await self.team.arun(input=eval_input, session_id=agent_session_id)  # type: ignore
                     output = response.content
 
                 if not output:

agno/integrations/discord/client.py

@@ -14,7 +14,7 @@ try:
     import discord
 
 except (ImportError, ModuleNotFoundError):
-    print("`discord.py` not installed. Please install using `pip install discord.py`")
+    raise ImportError("`discord.py` not installed. Please install using `pip install discord.py`")
 
 
 class RequiresConfirmationView(discord.ui.View):

agno/models/cerebras/cerebras.py

@@ -512,14 +512,16 @@ class Cerebras(Model):
 
             # Extend the list if needed
             while len(tool_calls) <= index:
-                tool_calls.append({
-                    "id": None,
-                    "type": None,
-                    "function": {
-                        "name": "",
-                        "arguments": "",
-                    },
-                })
+                tool_calls.append(
+                    {
+                        "id": None,
+                        "type": None,
+                        "function": {
+                            "name": "",
+                            "arguments": "",
+                        },
+                    }
+                )
 
             tool_call_entry = tool_calls[index]
 
@@ -540,10 +542,7 @@ class Cerebras(Model):
                     tool_call_entry["function"]["arguments"] += func_delta["arguments"]
 
         # Filter out any incomplete tool calls (missing id or function name)
-        complete_tool_calls = [
-            tc for tc in tool_calls
-            if tc.get("id") and tc.get("function", {}).get("name")
-        ]
+        complete_tool_calls = [tc for tc in tool_calls if tc.get("id") and tc.get("function", {}).get("name")]
 
         return complete_tool_calls
 

agno/os/interfaces/whatsapp/security.py

@@ -3,6 +3,8 @@ import hmac
 import os
 from typing import Optional
 
+from agno.utils.log import log_warning
+
 
 def is_development_mode() -> bool:
     """Check if the application is running in development mode."""
@@ -36,7 +38,7 @@ def validate_webhook_signature(payload: bytes, signature_header: Optional[str])
     """
     # In development mode, we can bypass signature validation
     if is_development_mode():
-        print("WARNING: Bypassing signature validation in development mode")
+        log_warning("Bypassing signature validation in development mode")
         return True
 
     if not signature_header or not signature_header.startswith("sha256="):

agno/os/schema.py

@@ -742,10 +742,11 @@ class SessionSchema(BaseModel):
     @classmethod
     def from_dict(cls, session: Dict[str, Any]) -> "SessionSchema":
         session_name = get_session_name(session)
+        session_data = session.get("session_data", {}) or {}
         return cls(
             session_id=session.get("session_id", ""),
             session_name=session_name,
-            session_state=session.get("session_data", {}).get("session_state", None),
+            session_state=session_data.get("session_state", None),
             created_at=datetime.fromtimestamp(session.get("created_at", 0), tz=timezone.utc)
             if session.get("created_at")
             else None,

agno/session/team.py

@@ -46,7 +46,6 @@ class TeamSession:
         session_dict = asdict(self)
 
         session_dict["runs"] = [run.to_dict() for run in self.runs] if self.runs else None
-        print(session_dict["runs"])
         session_dict["summary"] = self.summary.to_dict() if self.summary else None
 
         return session_dict

agno/table.py

@@ -7,4 +7,4 @@ conn = sqlite3.connect(db_path)
 cur = conn.cursor()
 cur.execute(f"DROP TABLE IF EXISTS {table_name}")
 conn.commit()
-conn.close()
+conn.close()

agno/team/team.py

@@ -2060,7 +2060,10 @@ class Team:
 
         # Initialize session state
         session_state = self._initialize_session_state(
-            session_state=session_state or {}, user_id=user_id, session_id=session_id, run_id=run_id
+            session_state=session_state if session_state is not None else {},
+            user_id=user_id,
+            session_id=session_id,
+            run_id=run_id,
         )
         # Update session state from DB
         session_state = self._load_session_state(session=team_session, session_state=session_state)
@@ -2293,7 +2296,7 @@ class Team:
         self._update_metadata(session=team_session)
         # Initialize session state
         run_context.session_state = self._initialize_session_state(
-            session_state=run_context.session_state or {},
+            session_state=run_context.session_state if run_context.session_state is not None else {},
             user_id=user_id,
             session_id=session_id,
             run_id=run_response.run_id,
@@ -2533,7 +2536,7 @@ class Team:
         self._update_metadata(session=team_session)
         # Initialize session state
         run_context.session_state = self._initialize_session_state(
-            session_state=run_context.session_state or {},
+            session_state=run_context.session_state if run_context.session_state is not None else {},
             user_id=user_id,
             session_id=session_id,
             run_id=run_response.run_id,
@@ -6877,7 +6880,7 @@ class Team:
             return message
         # Should already be resolved and passed from run() method
         format_variables = ChainMap(
-            session_state or {},
+            session_state if session_state is not None else {},
             dependencies or {},
             metadata or {},
             {"user_id": user_id} if user_id is not None else {},

agno/tools/google_drive.py

@@ -69,6 +69,7 @@ from pathlib import Path
 from typing import Any, List, Optional, Union
 
 from agno.tools import Toolkit
+from agno.utils.log import log_error
 
 try:
     from google.auth.transport.requests import Request
@@ -202,7 +203,7 @@ class GoogleDriveTools(Toolkit):
             items = results.get("files", [])
             return items
         except Exception as error:
-            print(f"Could not list files: {error}")
+            log_error(f"Could not list files: {error}")
             return []
 
     @authenticate
@@ -238,7 +239,7 @@ class GoogleDriveTools(Toolkit):
             )
             return uploaded_file
         except Exception as error:
-            print(f"Could not upload file '{file_path}': {error}")
+            log_error(f"Could not upload file '{file_path}': {error}")
             return None
 
     @authenticate
@@ -266,5 +267,5 @@ class GoogleDriveTools(Toolkit):
                     print(f"Download progress: {int(status.progress() * 100)}%.")
             return dest_path
         except Exception as error:
-            print(f"Could not download file '{file_id}': {error}")
+            log_error(f"Could not download file '{file_id}': {error}")
             return None