agno 2.1.2__tar.gz → 2.1.3__tar.gz

{agno-2.1.2 → agno-2.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agno
-Version: 2.1.2
+Version: 2.1.3
 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.2 → agno-2.1.3}/agno/db/firestore/firestore.py

@@ -682,9 +682,7 @@ class FirestoreDb(BaseDb):
             log_error(f"Error deleting memories: {e}")
             raise e
 
-    def get_all_memory_topics(
-        self, create_collection_if_not_found: Optional[bool] = True
-    ) -> List[str]:
+    def get_all_memory_topics(self, create_collection_if_not_found: Optional[bool] = True) -> List[str]:
         """Get all memory topics from the database.
 
         Returns:

{agno-2.1.2 → agno-2.1.3}/agno/db/gcs_json/gcs_json_db.py

@@ -472,7 +472,8 @@ class GcsJsonDb(BaseDb):
 
             # Filter out the memory, with optional user_id verification
             memories = [
-                m for m in memories
+                m
+                for m in memories
                 if not (m.get("memory_id") == memory_id and (user_id is None or m.get("user_id") == user_id))
             ]
 
@@ -499,7 +500,8 @@ class GcsJsonDb(BaseDb):
 
             # Filter out memories, with optional user_id verification
             memories = [
-                m for m in memories
+                m
+                for m in memories
                 if not (m.get("memory_id") in memory_ids and (user_id is None or m.get("user_id") == user_id))
             ]
 
@@ -642,7 +644,7 @@ class GcsJsonDb(BaseDb):
                         user_stats[memory_user_id] = {
                             "user_id": memory_user_id,
                             "total_memories": 0,
-                            "last_memory_updated_at": 0
+                            "last_memory_updated_at": 0,
                         }
                     user_stats[memory_user_id]["total_memories"] += 1
                     updated_at = memory.get("updated_at", 0)

{agno-2.1.2 → agno-2.1.3}/agno/db/in_memory/in_memory_db.py

@@ -359,8 +359,7 @@ class InMemoryDb(BaseDb):
             # If user_id is provided, verify ownership before deleting
             if user_id is not None:
                 self._memories = [
-                    m for m in self._memories
-                    if not (m.get("memory_id") == memory_id and m.get("user_id") == user_id)
+                    m for m in self._memories if not (m.get("memory_id") == memory_id and m.get("user_id") == user_id)
                 ]
             else:
                 self._memories = [m for m in self._memories if m.get("memory_id") != memory_id]
@@ -388,8 +387,7 @@ class InMemoryDb(BaseDb):
             # If user_id is provided, verify ownership before deleting
             if user_id is not None:
                 self._memories = [
-                    m for m in self._memories
-                    if not (m.get("memory_id") in memory_ids and m.get("user_id") == user_id)
+                    m for m in self._memories if not (m.get("memory_id") in memory_ids and m.get("user_id") == user_id)
                 ]
             else:
                 self._memories = [m for m in self._memories if m.get("memory_id") not in memory_ids]
@@ -532,7 +530,11 @@ class InMemoryDb(BaseDb):
 
                 if memory_user_id:
                     if memory_user_id not in user_stats:
-                        user_stats[memory_user_id] = {"user_id": memory_user_id, "total_memories": 0, "last_memory_updated_at": 0}
+                        user_stats[memory_user_id] = {
+                            "user_id": memory_user_id,
+                            "total_memories": 0,
+                            "last_memory_updated_at": 0,
+                        }
                     user_stats[memory_user_id]["total_memories"] += 1
                     updated_at = memory.get("updated_at", 0)
                     if updated_at > user_stats[memory_user_id]["last_memory_updated_at"]:

{agno-2.1.2 → agno-2.1.3}/agno/models/aws/bedrock.py

@@ -360,7 +360,7 @@ class AwsBedrock(Model):
             formatted_messages, system_message = self._format_messages(messages)
 
             tool_config = None
-            if tools: 
+            if tools:
                 tool_config = {"tools": self._format_tools_for_request(tools)}
 
             body = {

agno-2.1.3/agno/models/openrouter/openrouter.py

@@ -0,0 +1,66 @@
+from dataclasses import dataclass, field
+from os import getenv
+from typing import Any, Dict, List, Optional, Type, Union
+
+from pydantic import BaseModel
+
+from agno.models.openai.like import OpenAILike
+from agno.run.agent import RunOutput
+
+
+@dataclass
+class OpenRouter(OpenAILike):
+    """
+    A class for using models hosted on OpenRouter.
+
+    Attributes:
+        id (str): The model id. Defaults to "gpt-4o".
+        name (str): The model name. Defaults to "OpenRouter".
+        provider (str): The provider name. Defaults to "OpenRouter".
+        api_key (Optional[str]): The API key.
+        base_url (str): The base URL. Defaults to "https://openrouter.ai/api/v1".
+        max_tokens (int): The maximum number of tokens. Defaults to 1024.
+        fallback_models (Optional[List[str]]): List of fallback model IDs to use if the primary model
+            fails due to rate limits, timeouts, or unavailability. OpenRouter will automatically try
+            these models in order. Example: ["anthropic/claude-sonnet-4", "deepseek/deepseek-r1"]
+    """
+
+    id: str = "gpt-4o"
+    name: str = "OpenRouter"
+    provider: str = "OpenRouter"
+
+    api_key: Optional[str] = field(default_factory=lambda: getenv("OPENROUTER_API_KEY"))
+    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
+
+    def get_request_params(
+        self,
+        response_format: Optional[Union[Dict, Type[BaseModel]]] = None,
+        tools: Optional[List[Dict[str, Any]]] = None,
+        tool_choice: Optional[Union[str, Dict[str, Any]]] = None,
+        run_response: Optional[RunOutput] = None,
+    ) -> Dict[str, Any]:
+        """
+        Returns keyword arguments for API requests, including fallback models configuration.
+
+        Returns:
+            Dict[str, Any]: A dictionary of keyword arguments for API requests.
+        """
+        # Get base request params from parent class
+        request_params = super().get_request_params(
+            response_format=response_format, tools=tools, tool_choice=tool_choice, run_response=run_response
+        )
+
+        # Add fallback models to extra_body if specified
+        if self.models:
+            # Get existing extra_body or create new dict
+            extra_body = request_params.get("extra_body") or {}
+
+            # Merge fallback models into extra_body
+            extra_body["models"] = self.models
+
+            # Update request params
+            request_params["extra_body"] = extra_body
+
+        return request_params

agno-2.1.3/agno/models/vertexai/claude.py

@@ -0,0 +1,74 @@
+from dataclasses import dataclass
+from os import getenv
+from typing import Any, Dict, Optional
+
+from agno.models.anthropic import Claude as AnthropicClaude
+
+try:
+    from anthropic import AnthropicVertex as AnthropicClient
+    from anthropic import (
+        AsyncAnthropicVertex as AsyncAnthropicClient,
+    )
+except ImportError as e:
+    raise ImportError("`anthropic` not installed. Please install it with `pip install anthropic`") from e
+
+
+@dataclass
+class Claude(AnthropicClaude):
+    """
+    A class representing Anthropic Claude model.
+
+    For more information, see: https://docs.anthropic.com/en/api/messages
+    """
+
+    id: str = "claude-sonnet-4@20250514"
+    name: str = "Claude"
+    provider: str = "VertexAI"
+
+    # Client parameters
+    region: Optional[str] = None
+    project_id: Optional[str] = None
+    base_url: Optional[str] = None
+
+    # Anthropic clients
+    client: Optional[AnthropicClient] = None
+    async_client: Optional[AsyncAnthropicClient] = None
+
+    def _get_client_params(self) -> Dict[str, Any]:
+        client_params: Dict[str, Any] = {}
+
+        # Add API key to client parameters
+        client_params["region"] = self.region or getenv("CLOUD_ML_REGION")
+        client_params["project_id"] = self.project_id or getenv("ANTHROPIC_VERTEX_PROJECT_ID")
+        client_params["base_url"] = self.base_url or getenv("ANTHROPIC_VERTEX_BASE_URL")
+        if self.timeout is not None:
+            client_params["timeout"] = self.timeout
+
+        # Add additional client parameters
+        if self.client_params is not None:
+            client_params.update(self.client_params)
+        if self.default_headers is not None:
+            client_params["default_headers"] = self.default_headers
+        return client_params
+
+    def get_client(self) -> AnthropicClient:
+        """
+        Returns an instance of the Anthropic client.
+        """
+        if self.client and not self.client.is_closed():
+            return self.client
+
+        _client_params = self._get_client_params()
+        self.client = AnthropicClient(**_client_params)
+        return self.client
+
+    def get_async_client(self) -> AsyncAnthropicClient:
+        """
+        Returns an instance of the async Anthropic client.
+        """
+        if self.async_client:
+            return self.async_client
+
+        _client_params = self._get_client_params()
+        self.async_client = AsyncAnthropicClient(**_client_params)
+        return self.async_client

{agno-2.1.2 → agno-2.1.3}/agno/os/app.py

@@ -64,6 +64,30 @@ async def mcp_lifespan(_, mcp_tools):
         await tool.close()
 
 
+def _combine_app_lifespans(lifespans: list) -> Any:
+    """Combine multiple FastAPI app lifespan context managers into one."""
+    if len(lifespans) == 1:
+        return lifespans[0]
+
+    from contextlib import asynccontextmanager
+
+    @asynccontextmanager
+    async def combined_lifespan(app):
+        async def _run_nested(index: int):
+            if index >= len(lifespans):
+                yield
+                return
+
+            async with lifespans[index](app):
+                async for _ in _run_nested(index + 1):
+                    yield
+
+        async for _ in _run_nested(0):
+            yield
+
+    return combined_lifespan
+
+
 class AgentOS:
     def __init__(
         self,
@@ -220,7 +244,7 @@ class AgentOS:
                         async with mcp_tools_lifespan(app):  # type: ignore
                             yield
 
-                app_lifespan = combined_lifespan  # type: ignore
+                app_lifespan = combined_lifespan
             else:
                 app_lifespan = mcp_tools_lifespan
 
@@ -237,6 +261,32 @@ class AgentOS:
     def get_app(self) -> FastAPI:
         if self.base_app:
             fastapi_app = self.base_app
+
+            # Initialize MCP server if enabled
+            if self.enable_mcp_server:
+                from agno.os.mcp import get_mcp_server
+
+                self._mcp_app = get_mcp_server(self)
+
+            # Collect all lifespans that need to be combined
+            lifespans = []
+
+            if fastapi_app.router.lifespan_context:
+                lifespans.append(fastapi_app.router.lifespan_context)
+
+            if self.mcp_tools:
+                lifespans.append(partial(mcp_lifespan, mcp_tools=self.mcp_tools))
+
+            if self.enable_mcp_server and self._mcp_app:
+                lifespans.append(self._mcp_app.lifespan)
+
+            if self.lifespan:
+                lifespans.append(self.lifespan)
+
+            # Combine lifespans and set them in the app
+            if lifespans:
+                fastapi_app.router.lifespan_context = _combine_app_lifespans(lifespans)
+
         else:
             if self.enable_mcp_server:
                 from contextlib import asynccontextmanager
@@ -255,7 +305,7 @@ class AgentOS:
                             async with self._mcp_app.lifespan(app):  # type: ignore
                                 yield
 
-                    final_lifespan = combined_lifespan  # type: ignore
+                    final_lifespan = combined_lifespan
 
                 fastapi_app = self._make_app(lifespan=final_lifespan)
             else:

{agno-2.1.2 → agno-2.1.3}/agno/os/mcp.py

@@ -78,21 +78,21 @@ def get_mcp_server(
         agent = get_agent_by_id(agent_id, os.agents)
         if agent is None:
             raise Exception(f"Agent {agent_id} not found")
-        return agent.run(message)
+        return await agent.arun(message)
 
     @mcp.tool(name="run_team", description="Run a team", tags={"core"})  # type: ignore
     async def run_team(team_id: str, message: str) -> TeamRunOutput:
         team = get_team_by_id(team_id, os.teams)
         if team is None:
             raise Exception(f"Team {team_id} not found")
-        return team.run(message)
+        return await team.arun(message)
 
     @mcp.tool(name="run_workflow", description="Run a workflow", tags={"core"})  # type: ignore
     async def run_workflow(workflow_id: str, message: str) -> WorkflowRunOutput:
         workflow = get_workflow_by_id(workflow_id, os.workflows)
         if workflow is None:
             raise Exception(f"Workflow {workflow_id} not found")
-        return workflow.run(message)
+        return await workflow.arun(message)
 
     # Session Management Tools
     @mcp.tool(name="get_sessions_for_agent", description="Get list of sessions for an agent", tags={"session"})  # type: ignore

{agno-2.1.2 → agno-2.1.3}/agno/os/utils.py

@@ -96,7 +96,7 @@ def get_session_name(session: Dict[str, Any]) -> str:
             run = None
             for r in runs:
                 # If agent_id is not present, it's a team run
-                if not r.get("agent_id"):  
+                if not r.get("agent_id"):
                     run = r
                     break
             # Fallback to first run if no team run found
@@ -112,6 +112,7 @@ def get_session_name(session: Dict[str, Any]) -> str:
                 elif isinstance(workflow_input, dict):
                     try:
                         import json
+
                         return json.dumps(workflow_input)
                     except (TypeError, ValueError):
                         pass

{agno-2.1.2 → agno-2.1.3}/agno/tools/file.py

@@ -75,7 +75,9 @@ class FileTools(Toolkit):
         """
         try:
             log_info(f"Reading files in : {self.base_dir}")
-            return json.dumps([str(file_path) for file_path in self.base_dir.iterdir()], indent=4)
+            return json.dumps(
+                [str(file_path.relative_to(self.base_dir)) for file_path in self.base_dir.iterdir()], indent=4
+            )
         except Exception as e:
             log_error(f"Error reading files: {e}")
             return f"Error reading files: {e}"
@@ -93,7 +95,7 @@ class FileTools(Toolkit):
             log_debug(f"Searching files in {self.base_dir} with pattern {pattern}")
             matching_files = list(self.base_dir.glob(pattern))
 
-            file_paths = [str(file_path) for file_path in matching_files]
+            file_paths = [str(file_path.relative_to(self.base_dir)) for file_path in matching_files]
 
             result = {
                 "pattern": pattern,

{agno-2.1.2 → agno-2.1.3}/agno/utils/merge_dict.py

@@ -27,7 +27,7 @@ def merge_parallel_session_states(original_state: Dict[str, Any], modified_state
     """
     if not original_state or not modified_states:
         return
-    
+
     # Collect all actual changes (keys where value differs from original)
     all_changes = {}
     for modified_state in modified_states:
@@ -35,7 +35,7 @@ def merge_parallel_session_states(original_state: Dict[str, Any], modified_state
             for key, value in modified_state.items():
                 if key not in original_state or original_state[key] != value:
                     all_changes[key] = value
-    
+
     # Apply all collected changes to the original state
     for key, value in all_changes.items():
-        original_state[key] = value
+        original_state[key] = value

{agno-2.1.2 → agno-2.1.3}/agno/workflow/parallel.py

@@ -14,8 +14,8 @@ from agno.run.workflow import (
     WorkflowRunOutput,
     WorkflowRunOutputEvent,
 )
-from agno.utils.merge_dict import merge_parallel_session_states
 from agno.utils.log import log_debug, logger
+from agno.utils.merge_dict import merge_parallel_session_states
 from agno.workflow.condition import Condition
 from agno.workflow.step import Step
 from agno.workflow.types import StepInput, StepOutput, StepType
@@ -220,7 +220,7 @@ class Parallel:
             idx, step = step_with_index
             # Use the individual session_state copy for this step
             step_session_state = session_state_copies[idx]
-            
+
             try:
                 step_result = step.execute(
                     step_input,
@@ -350,7 +350,7 @@ class Parallel:
             idx, step = step_with_index
             # Use the individual session_state copy for this step
             step_session_state = session_state_copies[idx]
-            
+
             try:
                 step_events = []
 
@@ -506,7 +506,7 @@ class Parallel:
             idx, step = step_with_index
             # Use the individual session_state copy for this step
             step_session_state = session_state_copies[idx]
-            
+
             try:
                 inner_step_result = await step.aexecute(
                     step_input,
@@ -637,7 +637,7 @@ class Parallel:
             idx, step = step_with_index
             # Use the individual session_state copy for this step
             step_session_state = session_state_copies[idx]
-            
+
             try:
                 step_events = []
 

{agno-2.1.2 → agno-2.1.3}/agno/workflow/workflow.py

@@ -875,9 +875,7 @@ class Workflow:
             return func(**call_kwargs)
         except TypeError as e:
             # If signature inspection fails, fall back to original method
-            logger.error(
-                f"Function signature inspection failed: {e}. Falling back to original calling convention."
-            )
+            logger.error(f"Function signature inspection failed: {e}. Falling back to original calling convention.")
             return func(**kwargs)
 
     def _execute(
@@ -892,7 +890,8 @@ class Workflow:
         from inspect import isasyncgenfunction, iscoroutinefunction, isgeneratorfunction
 
         workflow_run_response.status = RunStatus.running
-        register_run(workflow_run_response.run_id)  # type: ignore
+        if workflow_run_response.run_id:
+            register_run(workflow_run_response.run_id)  # type: ignore
 
         if callable(self.steps):
             if iscoroutinefunction(self.steps) or isasyncgenfunction(self.steps):
@@ -1346,7 +1345,8 @@ class Workflow:
         workflow_run_response.status = RunStatus.running
 
         # Register run for cancellation tracking
-        register_run(workflow_run_response.run_id)  # type: ignore
+        if workflow_run_response.run_id:
+            register_run(workflow_run_response.run_id)  # type: ignore
 
         if callable(self.steps):
             # Execute the workflow with the custom executor
@@ -1507,6 +1507,10 @@ class Workflow:
 
         workflow_run_response.status = RunStatus.running
 
+        # Register run for cancellation tracking
+        if workflow_run_response.run_id:
+            register_run(workflow_run_response.run_id)
+
         workflow_started_event = WorkflowStartedEvent(
             run_id=workflow_run_response.run_id or "",
             workflow_name=workflow_run_response.workflow_name,

{agno-2.1.2 → agno-2.1.3}/agno.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agno
-Version: 2.1.2
+Version: 2.1.3
 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.2 → agno-2.1.3}/agno.egg-info/SOURCES.txt

@@ -243,6 +243,8 @@ agno/models/together/__init__.py
 agno/models/together/together.py
 agno/models/vercel/__init__.py
 agno/models/vercel/v0.py
+agno/models/vertexai/__init__.py
+agno/models/vertexai/claude.py
 agno/models/vllm/__init__.py
 agno/models/vllm/vllm.py
 agno/models/xai/__init__.py

{agno-2.1.2 → agno-2.1.3}/pyproject.toml

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

agno-2.1.2/agno/models/openrouter/openrouter.py

@@ -1,28 +0,0 @@
-from dataclasses import dataclass, field
-from os import getenv
-from typing import Optional
-
-from agno.models.openai.like import OpenAILike
-
-
-@dataclass
-class OpenRouter(OpenAILike):
-    """
-    A class for using models hosted on OpenRouter.
-
-    Attributes:
-        id (str): The model id. Defaults to "gpt-4o".
-        name (str): The model name. Defaults to "OpenRouter".
-        provider (str): The provider name. Defaults to "OpenRouter".
-        api_key (Optional[str]): The API key.
-        base_url (str): The base URL. Defaults to "https://openrouter.ai/api/v1".
-        max_tokens (int): The maximum number of tokens. Defaults to 1024.
-    """
-
-    id: str = "gpt-4o"
-    name: str = "OpenRouter"
-    provider: str = "OpenRouter"
-
-    api_key: Optional[str] = field(default_factory=lambda: getenv("OPENROUTER_API_KEY"))
-    base_url: str = "https://openrouter.ai/api/v1"
-    max_tokens: int = 1024