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

agno/models/openrouter/openrouter.py

@@ -1,10 +1,10 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from os import getenv
 from typing import Any, Dict, List, Optional, Type, Union
 
 from pydantic import BaseModel
 
-from agno.exceptions import ModelProviderError
+from agno.exceptions import ModelAuthenticationError
 from agno.models.openai.like import OpenAILike
 from agno.run.agent import RunOutput
 
@@ -30,7 +30,7 @@ class OpenRouter(OpenAILike):
     name: str = "OpenRouter"
     provider: str = "OpenRouter"
 
-    api_key: Optional[str] = field(default_factory=lambda: getenv("OPENROUTER_API_KEY"))
+    api_key: Optional[str] = None
     base_url: str = "https://openrouter.ai/api/v1"
     max_tokens: int = 1024
     models: Optional[List[str]] = None  # Dynamic model routing https://openrouter.ai/docs/features/model-routing
@@ -46,10 +46,9 @@ class OpenRouter(OpenAILike):
         if not self.api_key:
             self.api_key = getenv("OPENROUTER_API_KEY")
             if not self.api_key:
-                raise ModelProviderError(
+                raise ModelAuthenticationError(
                     message="OPENROUTER_API_KEY not set. Please set the OPENROUTER_API_KEY environment variable.",
                     model_name=self.name,
-                    model_id=self.id,
                 )
 
         return super()._get_client_params()

agno/models/perplexity/perplexity.py

@@ -4,7 +4,7 @@ from typing import Any, Dict, Optional, Type, Union
 
 from pydantic import BaseModel
 
-from agno.exceptions import ModelProviderError
+from agno.exceptions import ModelAuthenticationError, ModelProviderError
 from agno.models.message import Citations, UrlCitation
 from agno.models.metrics import Metrics
 from agno.models.response import ModelResponse
@@ -60,10 +60,9 @@ class Perplexity(OpenAILike):
         if not self.api_key:
             self.api_key = getenv("PERPLEXITY_API_KEY")
             if not self.api_key:
-                raise ModelProviderError(
+                raise ModelAuthenticationError(
                     message="PERPLEXITY_API_KEY not set. Please set the PERPLEXITY_API_KEY environment variable.",
                     model_name=self.name,
-                    model_id=self.id,
                 )
         return super()._get_client_params()
 

agno/models/portkey/portkey.py

@@ -1,8 +1,8 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from os import getenv
 from typing import Any, Dict, Optional, cast
 
-from agno.exceptions import ModelProviderError
+from agno.exceptions import ModelAuthenticationError
 from agno.models.openai.like import OpenAILike
 
 try:
@@ -30,20 +30,21 @@ class Portkey(OpenAILike):
     name: str = "Portkey"
     provider: str = "Portkey"
 
-    portkey_api_key: Optional[str] = field(default_factory=lambda: getenv("PORTKEY_API_KEY"))
-    virtual_key: Optional[str] = field(default_factory=lambda: getenv("PORTKEY_VIRTUAL_KEY"))
+    portkey_api_key: Optional[str] = None
+    virtual_key: Optional[str] = None
     config: Optional[Dict[str, Any]] = None
     base_url: str = PORTKEY_GATEWAY_URL
 
     def _get_client_params(self) -> Dict[str, Any]:
         # Check for required keys
         if not self.portkey_api_key:
-            raise ModelProviderError(
+            raise ModelAuthenticationError(
                 message="PORTKEY_API_KEY not set. Please set the PORTKEY_API_KEY environment variable.",
                 model_name=self.name,
-                model_id=self.id,
             )
 
+        self.virtual_key = self.virtual_key or getenv("PORTKEY_VIRTUAL_KEY")
+
         # Create headers using Portkey's createHeaders function
         header_params: Dict[str, Any] = {
             "api_key": self.portkey_api_key,

agno/models/requesty/requesty.py

@@ -1,10 +1,10 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from os import getenv
 from typing import Any, Dict, List, Optional, Type, Union
 
 from pydantic import BaseModel
 
-from agno.exceptions import ModelProviderError
+from agno.exceptions import ModelAuthenticationError
 from agno.models.openai.like import OpenAILike
 from agno.run.agent import RunOutput
 from agno.run.team import TeamRunOutput
@@ -27,7 +27,7 @@ class Requesty(OpenAILike):
     name: str = "Requesty"
     provider: str = "Requesty"
 
-    api_key: Optional[str] = field(default_factory=lambda: getenv("REQUESTY_API_KEY"))
+    api_key: Optional[str] = None
     base_url: str = "https://router.requesty.ai/v1"
     max_tokens: int = 1024
 
@@ -41,10 +41,9 @@ class Requesty(OpenAILike):
         if not self.api_key:
             self.api_key = getenv("REQUESTY_API_KEY")
             if not self.api_key:
-                raise ModelProviderError(
+                raise ModelAuthenticationError(
                     message="REQUESTY_API_KEY not set. Please set the REQUESTY_API_KEY environment variable.",
                     model_name=self.name,
-                    model_id=self.id,
                 )
         return super()._get_client_params()
 

agno/models/response.py

@@ -35,7 +35,7 @@ class ToolExecution:
     # If True, the agent will stop executing after this tool call.
     stop_after_tool_call: bool = False
 
-    created_at: int = int(time())
+    created_at: int = field(default_factory=lambda: int(time()))
 
     # User control flow requirements
     requires_confirmation: Optional[bool] = None
@@ -81,6 +81,7 @@ class ToolExecution:
             else None,
             external_execution_required=data.get("external_execution_required"),
             metrics=Metrics(**(data.get("metrics", {}) or {})),
+            **{"created_at": data["created_at"]} if "created_at" in data else {},
         )
 
 

agno/models/sambanova/sambanova.py

@@ -1,8 +1,8 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from os import getenv
 from typing import Any, Dict, Optional
 
-from agno.exceptions import ModelProviderError
+from agno.exceptions import ModelAuthenticationError
 from agno.models.openai.like import OpenAILike
 
 
@@ -23,7 +23,7 @@ class Sambanova(OpenAILike):
     name: str = "Sambanova"
     provider: str = "Sambanova"
 
-    api_key: Optional[str] = field(default_factory=lambda: getenv("SAMBANOVA_API_KEY"))
+    api_key: Optional[str] = None
     base_url: str = "https://api.sambanova.ai/v1"
 
     supports_native_structured_outputs: bool = False
@@ -38,9 +38,8 @@ class Sambanova(OpenAILike):
         if not self.api_key:
             self.api_key = getenv("SAMBANOVA_API_KEY")
             if not self.api_key:
-                raise ModelProviderError(
+                raise ModelAuthenticationError(
                     message="SAMBANOVA_API_KEY not set. Please set the SAMBANOVA_API_KEY environment variable.",
                     model_name=self.name,
-                    model_id=self.id,
                 )
         return super()._get_client_params()

agno/models/siliconflow/siliconflow.py

@@ -2,7 +2,7 @@ from dataclasses import dataclass
 from os import getenv
 from typing import Any, Dict, Optional
 
-from agno.exceptions import ModelProviderError
+from agno.exceptions import ModelAuthenticationError
 from agno.models.openai.like import OpenAILike
 
 
@@ -22,7 +22,7 @@ class Siliconflow(OpenAILike):
     id: str = "Qwen/QwQ-32B"
     name: str = "Siliconflow"
     provider: str = "Siliconflow"
-    api_key: Optional[str] = getenv("SILICONFLOW_API_KEY")
+    api_key: Optional[str] = None
     base_url: str = "https://api.siliconflow.com/v1"
 
     def _get_client_params(self) -> Dict[str, Any]:
@@ -35,9 +35,8 @@ class Siliconflow(OpenAILike):
         if not self.api_key:
             self.api_key = getenv("SILICONFLOW_API_KEY")
             if not self.api_key:
-                raise ModelProviderError(
+                raise ModelAuthenticationError(
                     message="SILICONFLOW_API_KEY not set. Please set the SILICONFLOW_API_KEY environment variable.",
                     model_name=self.name,
-                    model_id=self.id,
                 )
         return super()._get_client_params()

agno/models/together/together.py

@@ -1,8 +1,8 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from os import getenv
 from typing import Any, Dict, Optional
 
-from agno.exceptions import ModelProviderError
+from agno.exceptions import ModelAuthenticationError
 from agno.models.openai.like import OpenAILike
 
 
@@ -22,7 +22,7 @@ class Together(OpenAILike):
     id: str = "mistralai/Mixtral-8x7B-Instruct-v0.1"
     name: str = "Together"
     provider: str = "Together"
-    api_key: Optional[str] = field(default_factory=lambda: getenv("TOGETHER_API_KEY"))
+    api_key: Optional[str] = None
     base_url: str = "https://api.together.xyz/v1"
 
     def _get_client_params(self) -> Dict[str, Any]:
@@ -35,9 +35,8 @@ class Together(OpenAILike):
         if not self.api_key:
             self.api_key = getenv("TOGETHER_API_KEY")
             if not self.api_key:
-                raise ModelProviderError(
+                raise ModelAuthenticationError(
                     message="TOGETHER_API_KEY not set. Please set the TOGETHER_API_KEY environment variable.",
                     model_name=self.name,
-                    model_id=self.id,
                 )
         return super()._get_client_params()

agno/models/vercel/v0.py

@@ -1,8 +1,8 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from os import getenv
 from typing import Any, Dict, Optional
 
-from agno.exceptions import ModelProviderError
+from agno.exceptions import ModelAuthenticationError
 from agno.models.openai.like import OpenAILike
 
 
@@ -23,7 +23,7 @@ class V0(OpenAILike):
     name: str = "v0"
     provider: str = "Vercel"
 
-    api_key: Optional[str] = field(default_factory=lambda: getenv("V0_API_KEY"))
+    api_key: Optional[str] = None
     base_url: str = "https://api.v0.dev/v1/"
 
     def _get_client_params(self) -> Dict[str, Any]:
@@ -36,9 +36,8 @@ class V0(OpenAILike):
         if not self.api_key:
             self.api_key = getenv("V0_API_KEY")
             if not self.api_key:
-                raise ModelProviderError(
+                raise ModelAuthenticationError(
                     message="V0_API_KEY not set. Please set the V0_API_KEY environment variable.",
                     model_name=self.name,
-                    model_id=self.id,
                 )
         return super()._get_client_params()

agno/models/vllm/vllm.py

@@ -4,6 +4,7 @@ from typing import Any, Dict, List, Optional, Type, Union
 
 from pydantic import BaseModel
 
+from agno.exceptions import ModelAuthenticationError
 from agno.models.openai.like import OpenAILike
 from agno.utils.log import log_debug
 
@@ -29,8 +30,8 @@ class VLLM(OpenAILike):
     name: str = "VLLM"
     provider: str = "VLLM"
 
-    api_key: Optional[str] = getenv("VLLM_API_KEY") or "EMPTY"
-    base_url: Optional[str] = getenv("VLLM_BASE_URL", "http://localhost:8000/v1/")
+    api_key: Optional[str] = None
+    base_url: Optional[str] = None
 
     temperature: float = 0.7
     top_p: float = 0.8
@@ -38,19 +39,23 @@ class VLLM(OpenAILike):
     top_k: Optional[int] = None
     enable_thinking: Optional[bool] = None
 
-    def __post_init__(self):
-        """Validate required configuration"""
-        if not self.base_url:
-            raise ValueError("VLLM_BASE_URL must be set via environment variable or explicit initialization")
-        if self.id == "not-set":
-            raise ValueError("Model ID must be set via environment variable or explicit initialization")
+    def _get_client_params(self) -> Dict[str, Any]:
+        """
+        Returns client parameters for API requests, checking for VLLM_API_KEY.
 
-        body: Dict[str, Any] = {}
-        if self.top_k is not None:
-            body["top_k"] = self.top_k
-        if self.enable_thinking is not None:
-            body["chat_template_kwargs"] = {"enable_thinking": self.enable_thinking}
-        self.extra_body = body or None
+        Returns:
+            Dict[str, Any]: A dictionary of client parameters for API requests.
+        """
+        if not self.api_key:
+            self.api_key = getenv("VLLM_API_KEY")
+            if not self.api_key:
+                raise ModelAuthenticationError(
+                    message="VLLM_API_KEY not set. Please set the VLLM_API_KEY environment variable.",
+                    model_name=self.name,
+                )
+        if not self.base_url:
+            self.base_url = getenv("VLLM_BASE_URL", "http://localhost:8000/v1/")
+        return super()._get_client_params()
 
     def get_request_params(
         self,

agno/models/xai/xai.py

@@ -1,10 +1,10 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from os import getenv
 from typing import Any, Dict, List, Optional, Type, Union
 
 from pydantic import BaseModel
 
-from agno.exceptions import ModelProviderError
+from agno.exceptions import ModelAuthenticationError
 from agno.models.message import Citations, UrlCitation
 from agno.models.openai.like import OpenAILike
 from agno.models.response import ModelResponse
@@ -35,7 +35,7 @@ class xAI(OpenAILike):
     name: str = "xAI"
     provider: str = "xAI"
 
-    api_key: Optional[str] = field(default_factory=lambda: getenv("XAI_API_KEY"))
+    api_key: Optional[str] = None
     base_url: str = "https://api.x.ai/v1"
 
     search_parameters: Optional[Dict[str, Any]] = None
@@ -50,10 +50,9 @@ class xAI(OpenAILike):
         if not self.api_key:
             self.api_key = getenv("XAI_API_KEY")
             if not self.api_key:
-                raise ModelProviderError(
+                raise ModelAuthenticationError(
                     message="XAI_API_KEY not set. Please set the XAI_API_KEY environment variable.",
                     model_name=self.name,
-                    model_id=self.id,
                 )
         return super()._get_client_params()
 

agno/os/app.py

@@ -27,6 +27,8 @@ from agno.os.config import (
     MetricsDomainConfig,
     SessionConfig,
     SessionDomainConfig,
+    TracesConfig,
+    TracesDomainConfig,
 )
 from agno.os.interfaces.base import BaseInterface
 from agno.os.router import get_base_router, get_websocket_router
@@ -37,12 +39,14 @@ from agno.os.routers.knowledge import get_knowledge_router
 from agno.os.routers.memory import get_memory_router
 from agno.os.routers.metrics import get_metrics_router
 from agno.os.routers.session import get_session_router
+from agno.os.routers.traces import get_traces_router
 from agno.os.settings import AgnoAPISettings
 from agno.os.utils import (
     collect_mcp_tools_from_team,
     collect_mcp_tools_from_workflow,
     find_conflicting_routes,
     load_yaml_config,
+    setup_tracing_for_os,
     update_cors_middleware,
 )
 from agno.team.team import Team
@@ -109,7 +113,10 @@ class AgentOS:
         base_app: Optional[FastAPI] = None,
         on_route_conflict: Literal["preserve_agentos", "preserve_base_app", "error"] = "preserve_agentos",
         telemetry: bool = True,
+        tracing: bool = False,
+        tracing_db: Optional[Union[BaseDb, AsyncBaseDb]] = None,
         auto_provision_dbs: bool = True,
+        run_hooks_in_background: bool = False,
     ):
         """Initialize AgentOS.
 
@@ -131,6 +138,10 @@ class AgentOS:
             base_app: Optional base FastAPI app to use for the AgentOS. All routes and middleware will be added to this app.
             on_route_conflict: What to do when a route conflict is detected in case a custom base_app is provided.
             telemetry: Whether to enable telemetry
+            tracing: If True, enables OpenTelemetry tracing for all agents and teams in the OS
+            tracing_db: Dedicated database for storing and reading traces. Recommended for multi-db setups.
+                       If not provided and tracing=True, the first available db from agents/teams/workflows is used.
+            run_hooks_in_background: If True, run agent/team pre/post hooks as FastAPI background tasks (non-blocking)
 
         """
         if not agents and not workflows and not teams and not knowledge:
@@ -169,10 +180,15 @@ class AgentOS:
         self.description = description
 
         self.telemetry = telemetry
+        self.tracing = tracing
+        self.tracing_db = tracing_db
 
         self.enable_mcp_server = enable_mcp_server
         self.lifespan = lifespan
 
+        # If True, run agent/team hooks as FastAPI background tasks
+        self.run_hooks_in_background = run_hooks_in_background
+
         # List of all MCP tools used inside the AgentOS
         self.mcp_tools: List[Any] = []
         self._mcp_app: Optional[Any] = None
@@ -181,6 +197,9 @@ class AgentOS:
         self._initialize_teams()
         self._initialize_workflows()
 
+        if self.tracing:
+            self._setup_tracing()
+
         if self.telemetry:
             from agno.api.os import OSLaunch, log_os_telemetry
 
@@ -235,6 +254,7 @@ class AgentOS:
             get_session_router(dbs=self.dbs),
             get_metrics_router(dbs=self.dbs),
             get_knowledge_router(knowledge_instances=self.knowledge_instances),
+            get_traces_router(dbs=self.dbs),
             get_memory_router(dbs=self.dbs),
             get_eval_router(dbs=self.dbs, agents=self.agents, teams=self.teams),
         ]
@@ -332,6 +352,9 @@ class AgentOS:
             # Required for the built-in routes to work
             agent.store_events = True
 
+            # Propagate run_hooks_in_background setting from AgentOS to agents
+            agent._run_hooks_in_background = self.run_hooks_in_background
+
     def _initialize_teams(self) -> None:
         """Initialize and configure all teams for AgentOS usage."""
         if not self.teams:
@@ -353,6 +376,9 @@ class AgentOS:
             # Required for the built-in routes to work
             team.store_events = True
 
+            # Propagate run_hooks_in_background setting to team and all nested members
+            team.propagate_run_hooks_in_background(self.run_hooks_in_background)
+
     def _initialize_workflows(self) -> None:
         """Initialize and configure all workflows for AgentOS usage."""
         if not self.workflows:
@@ -369,6 +395,49 @@ class AgentOS:
                 # Required for the built-in routes to work
                 workflow.store_events = True
 
+                # Propagate run_hooks_in_background setting to workflow and all its step agents/teams
+                workflow.propagate_run_hooks_in_background(self.run_hooks_in_background)
+
+    def _setup_tracing(self) -> None:
+        """Set up OpenTelemetry tracing for this AgentOS.
+
+        Uses tracing_db if provided, otherwise falls back to the first available
+        database from agents/teams/workflows.
+        """
+        # Use tracing_db if explicitly provided
+        if self.tracing_db is not None:
+            setup_tracing_for_os(db=self.tracing_db)
+            return
+
+        # Fall back to finding the first available database
+        db: Optional[Union[BaseDb, AsyncBaseDb]] = None
+
+        for agent in self.agents or []:
+            if agent.db:
+                db = agent.db
+                break
+
+        if db is None:
+            for team in self.teams or []:
+                if team.db:
+                    db = team.db
+                    break
+
+        if db is None:
+            for workflow in self.workflows or []:
+                if workflow.db:
+                    db = workflow.db
+                    break
+
+        if db is None:
+            log_warning(
+                "tracing=True but no database found. "
+                "Provide 'tracing_db' parameter or 'db' parameter to at least one agent/team/workflow."
+            )
+            return
+
+        setup_tracing_for_os(db=db)
+
     def get_app(self) -> FastAPI:
         if self.base_app:
             fastapi_app = self.base_app
@@ -447,6 +516,7 @@ class AgentOS:
             get_eval_router(dbs=self.dbs, agents=self.agents, teams=self.teams),
             get_metrics_router(dbs=self.dbs),
             get_knowledge_router(knowledge_instances=self.knowledge_instances),
+            get_traces_router(dbs=self.dbs),
         ]
 
         for router in routers:
@@ -591,6 +661,10 @@ class AgentOS:
             elif interface.team and interface.team.db:
                 self._register_db_with_validation(dbs, interface.team.db)
 
+        # Register tracing_db if provided (for traces reading)
+        if self.tracing_db is not None:
+            self._register_db_with_validation(dbs, self.tracing_db)
+
         self.dbs = dbs
         self.knowledge_dbs = knowledge_dbs
 
@@ -821,6 +895,36 @@ class AgentOS:
 
         return evals_config
 
+    def _get_traces_config(self) -> TracesConfig:
+        traces_config = self.config.traces if self.config and self.config.traces else TracesConfig()
+
+        if traces_config.dbs is None:
+            traces_config.dbs = []
+
+        dbs_with_specific_config = [db.db_id for db in traces_config.dbs]
+
+        # If tracing_db is explicitly set, only use that database for traces
+        if self.tracing_db is not None:
+            if self.tracing_db.id not in dbs_with_specific_config:
+                traces_config.dbs.append(
+                    DatabaseConfig(
+                        db_id=self.tracing_db.id,
+                        domain_config=TracesDomainConfig(display_name=self.tracing_db.id),
+                    )
+                )
+        else:
+            # Fall back to all discovered databases
+            for db_id in self.dbs.keys():
+                if db_id not in dbs_with_specific_config:
+                    traces_config.dbs.append(
+                        DatabaseConfig(
+                            db_id=db_id,
+                            domain_config=TracesDomainConfig(display_name=db_id),
+                        )
+                    )
+
+        return traces_config
+
     def serve(
         self,
         app: Union[str, FastAPI],

agno/os/config.py

@@ -36,6 +36,12 @@ class MemoryDomainConfig(BaseModel):
     display_name: Optional[str] = None
 
 
+class TracesDomainConfig(BaseModel):
+    """Configuration for the Traces domain of the AgentOS"""
+
+    display_name: Optional[str] = None
+
+
 DomainConfigType = TypeVar("DomainConfigType")
 
 
@@ -77,6 +83,12 @@ class MetricsConfig(MetricsDomainConfig):
     dbs: Optional[List[DatabaseConfig[MetricsDomainConfig]]] = None
 
 
+class TracesConfig(TracesDomainConfig):
+    """Configuration for the Traces domain of the AgentOS"""
+
+    dbs: Optional[List[DatabaseConfig[TracesDomainConfig]]] = None
+
+
 class ChatConfig(BaseModel):
     """Configuration for the Chat page of the AgentOS"""
 
@@ -102,3 +114,4 @@ class AgentOSConfig(BaseModel):
     memory: Optional[MemoryConfig] = None
     session: Optional[SessionConfig] = None
     metrics: Optional[MetricsConfig] = None
+    traces: Optional[TracesConfig] = None

agno/os/interfaces/whatsapp/router.py

@@ -171,7 +171,6 @@ def attach_routes(router: APIRouter, agent: Optional[Agent] = None, team: Option
                             f"Could not process image content for user {phone_number}. Type: {type(image_content)}"
                         )
                         await _send_whatsapp_message(phone_number, response.content)  # type: ignore
-                await _send_whatsapp_message(phone_number, response.content)  # type: ignore
             else:
                 await _send_whatsapp_message(phone_number, response.content)  # type: ignore
 

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/mcp.py

@@ -64,6 +64,7 @@ def get_mcp_server(
             knowledge=os._get_knowledge_config(),
             evals=os._get_evals_config(),
             metrics=os._get_metrics_config(),
+            traces=os._get_traces_config(),
             agents=[AgentSummaryResponse.from_agent(agent) for agent in os.agents] if os.agents else [],
             teams=[TeamSummaryResponse.from_team(team) for team in os.teams] if os.teams else [],
             workflows=[WorkflowSummaryResponse.from_workflow(w) for w in os.workflows] if os.workflows else [],