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

agno/api/evals.py

@@ -8,7 +8,7 @@ def create_eval_run_telemetry(eval_run: EvalRunCreate) -> None:
     """Telemetry recording for Eval runs"""
     with api.Client() as api_client:
         try:
-            api_client.post(ApiRoutes.EVAL_RUN_CREATE, json={"eval_run": eval_run.model_dump(exclude_none=True)})
+            api_client.post(ApiRoutes.EVAL_RUN_CREATE, json=eval_run.model_dump(exclude_none=True))
         except Exception as e:
             log_debug(f"Could not create evaluation run: {e}")
 
@@ -17,6 +17,6 @@ async def async_create_eval_run_telemetry(eval_run: EvalRunCreate) -> None:
     """Telemetry recording for async Eval runs"""
     async with api.AsyncClient() as api_client:
         try:
-            await api_client.post(ApiRoutes.EVAL_RUN_CREATE, json={"eval_run": eval_run.model_dump(exclude_none=True)})
+            await api_client.post(ApiRoutes.EVAL_RUN_CREATE, json=eval_run.model_dump(exclude_none=True))
         except Exception as e:
             log_debug(f"Could not create evaluation run: {e}")

agno/api/os.py

@@ -10,7 +10,7 @@ def log_os_telemetry(launch: OSLaunch) -> None:
         try:
             response = api_client.post(
                 ApiRoutes.AGENT_OS_LAUNCH,
-                json={"launch": launch.model_dump(exclude_none=True)},
+                json=launch.model_dump(exclude_none=True),
             )
             response.raise_for_status()
         except Exception as e:

agno/api/settings.py

@@ -16,7 +16,7 @@ class AgnoAPISettings(BaseSettings):
     api_runtime: str = "prd"
     alpha_features: bool = False
 
-    api_url: str = "https://api.agno.com"
+    api_url: str = "https://os-api.agno.com"
 
     model_config = SettingsConfigDict(env_prefix="AGNO_")
 
@@ -42,7 +42,7 @@ class AgnoAPISettings(BaseSettings):
         elif api_runtime == "stg":
             return "https://api-stg.agno.com"
         else:
-            return "https://api.agno.com"
+            return "https://os-api.agno.com"
 
     def gate_alpha_feature(self):
         if not self.alpha_features:

agno/api/team.py

@@ -10,7 +10,7 @@ def create_team_run(run: TeamRunCreate) -> None:
         try:
             response = api_client.post(
                 ApiRoutes.RUN_CREATE,
-                json={"run": run.model_dump(exclude_none=True)},
+                json=run.model_dump(exclude_none=True),
             )
             response.raise_for_status()
         except Exception as e:
@@ -23,7 +23,7 @@ async def acreate_team_run(run: TeamRunCreate) -> None:
         try:
             response = await api_client.post(
                 ApiRoutes.RUN_CREATE,
-                json={"run": run.model_dump(exclude_none=True)},
+                json=run.model_dump(exclude_none=True),
             )
             response.raise_for_status()
         except Exception as e:

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
 
@@ -678,7 +672,9 @@ class MongoDb(BaseDb):
             if result is None or not deserialize:
                 return result
 
-            return UserMemory.from_dict(result)
+            # Remove MongoDB's _id field before creating UserMemory object
+            result_filtered = {k: v for k, v in result.items() if k != "_id"}
+            return UserMemory.from_dict(result_filtered)
 
         except Exception as e:
             log_error(f"Exception reading from collection: {e}")
@@ -756,7 +752,8 @@ class MongoDb(BaseDb):
             if not deserialize:
                 return records, total_count
 
-            return [UserMemory.from_dict(record) for record in records]
+            # Remove MongoDB's _id field before creating UserMemory objects
+            return [UserMemory.from_dict({k: v for k, v in record.items() if k != "_id"}) for record in records]
 
         except Exception as e:
             log_error(f"Exception reading from collection: {e}")
@@ -867,7 +864,9 @@ class MongoDb(BaseDb):
             if not deserialize:
                 return update_doc
 
-            return UserMemory.from_dict(update_doc)
+            # Remove MongoDB's _id field before creating UserMemory object
+            update_doc_filtered = {k: v for k, v in update_doc.items() if k != "_id"}
+            return UserMemory.from_dict(update_doc_filtered)
 
         except Exception as e:
             log_error(f"Exception upserting user memory: {e}")

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/eval/accuracy.py

@@ -97,11 +97,18 @@ class AccuracyResult:
             title_justify="center",
         )
         summary_table.add_row("Number of Runs", f"{len(self.results)}")
-        summary_table.add_row("Average Score", f"{self.avg_score:.2f}")
-        summary_table.add_row("Mean Score", f"{self.mean_score:.2f}")
-        summary_table.add_row("Minimum Score", f"{self.min_score:.2f}")
-        summary_table.add_row("Maximum Score", f"{self.max_score:.2f}")
-        summary_table.add_row("Standard Deviation", f"{self.std_dev_score:.2f}")
+
+        if self.avg_score is not None:
+            summary_table.add_row("Average Score", f"{self.avg_score:.2f}")
+        if self.mean_score is not None:
+            summary_table.add_row("Mean Score", f"{self.mean_score:.2f}")
+        if self.min_score is not None:
+            summary_table.add_row("Minimum Score", f"{self.min_score:.2f}")
+        if self.max_score is not None:
+            summary_table.add_row("Maximum Score", f"{self.max_score:.2f}")
+        if self.std_dev_score is not None:
+            summary_table.add_row("Standard Deviation", f"{self.std_dev_score:.2f}")
+
         console.print(summary_table)
 
     def print_results(self, console: Optional["Console"] = None):

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/chunking/strategy.py

@@ -35,13 +35,13 @@ class ChunkingStrategy(ABC):
 class ChunkingStrategyType(str, Enum):
     """Enumeration of available chunking strategies."""
 
-    AGENTIC_CHUNKING = "AgenticChunking"
-    DOCUMENT_CHUNKING = "DocumentChunking"
-    RECURSIVE_CHUNKING = "RecursiveChunking"
-    SEMANTIC_CHUNKING = "SemanticChunking"
-    FIXED_SIZE_CHUNKING = "FixedSizeChunking"
-    ROW_CHUNKING = "RowChunking"
-    MARKDOWN_CHUNKING = "MarkdownChunking"
+    AGENTIC_CHUNKER = "AgenticChunker"
+    DOCUMENT_CHUNKER = "DocumentChunker"
+    RECURSIVE_CHUNKER = "RecursiveChunker"
+    SEMANTIC_CHUNKER = "SemanticChunker"
+    FIXED_SIZE_CHUNKER = "FixedSizeChunker"
+    ROW_CHUNKER = "RowChunker"
+    MARKDOWN_CHUNKER = "MarkdownChunker"
 
     @classmethod
     def from_string(cls, strategy_name: str) -> "ChunkingStrategyType":
@@ -63,13 +63,13 @@ class ChunkingStrategyFactory:
     def create_strategy(cls, strategy_type: ChunkingStrategyType, **kwargs) -> ChunkingStrategy:
         """Create an instance of the chunking strategy with the given parameters."""
         strategy_map = {
-            ChunkingStrategyType.AGENTIC_CHUNKING: cls._create_agentic_chunking,
-            ChunkingStrategyType.DOCUMENT_CHUNKING: cls._create_document_chunking,
-            ChunkingStrategyType.RECURSIVE_CHUNKING: cls._create_recursive_chunking,
-            ChunkingStrategyType.SEMANTIC_CHUNKING: cls._create_semantic_chunking,
-            ChunkingStrategyType.FIXED_SIZE_CHUNKING: cls._create_fixed_chunking,
-            ChunkingStrategyType.ROW_CHUNKING: cls._create_row_chunking,
-            ChunkingStrategyType.MARKDOWN_CHUNKING: cls._create_markdown_chunking,
+            ChunkingStrategyType.AGENTIC_CHUNKER: cls._create_agentic_chunking,
+            ChunkingStrategyType.DOCUMENT_CHUNKER: cls._create_document_chunking,
+            ChunkingStrategyType.RECURSIVE_CHUNKER: cls._create_recursive_chunking,
+            ChunkingStrategyType.SEMANTIC_CHUNKER: cls._create_semantic_chunking,
+            ChunkingStrategyType.FIXED_SIZE_CHUNKER: cls._create_fixed_chunking,
+            ChunkingStrategyType.ROW_CHUNKER: cls._create_row_chunking,
+            ChunkingStrategyType.MARKDOWN_CHUNKER: cls._create_markdown_chunking,
         }
         return strategy_map[strategy_type](**kwargs)
 

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