agno 2.3.8__py3-none-any.whl → 2.3.10__py3-none-any.whl

agno/os/schema.py

@@ -1,13 +1,11 @@
 from datetime import datetime, timezone
 from enum import Enum
 from typing import Any, Dict, Generic, List, Optional, TypeVar, Union
-from uuid import uuid4
 
 from pydantic import BaseModel, ConfigDict, Field
 
 from agno.agent import Agent
 from agno.db.base import SessionType
-from agno.models.message import Message
 from agno.os.config import (
     ChatConfig,
     EvalsConfig,
@@ -19,21 +17,11 @@ from agno.os.config import (
 )
 from agno.os.utils import (
     extract_input_media,
-    format_team_tools,
-    format_tools,
-    get_agent_input_schema_dict,
     get_run_input,
     get_session_name,
-    get_team_input_schema_dict,
-    get_workflow_input_schema_dict,
 )
-from agno.run import RunContext
-from agno.run.agent import RunOutput
-from agno.run.team import TeamRunOutput
 from agno.session import AgentSession, TeamSession, WorkflowSession
 from agno.team.team import Team
-from agno.utils.agent import aexecute_instructions, aexecute_system_message
-from agno.workflow.agent import WorkflowAgent
 from agno.workflow.workflow import Workflow
 
 
@@ -179,553 +167,6 @@ class ModelResponse(BaseModel):
     provider: Optional[str] = Field(None, description="Model provider name")
 
 
-class AgentResponse(BaseModel):
-    id: Optional[str] = None
-    name: Optional[str] = None
-    db_id: Optional[str] = None
-    model: Optional[ModelResponse] = None
-    tools: Optional[Dict[str, Any]] = None
-    sessions: Optional[Dict[str, Any]] = None
-    knowledge: Optional[Dict[str, Any]] = None
-    memory: Optional[Dict[str, Any]] = None
-    reasoning: Optional[Dict[str, Any]] = None
-    default_tools: Optional[Dict[str, Any]] = None
-    system_message: Optional[Dict[str, Any]] = None
-    extra_messages: Optional[Dict[str, Any]] = None
-    response_settings: Optional[Dict[str, Any]] = None
-    streaming: Optional[Dict[str, Any]] = None
-    metadata: Optional[Dict[str, Any]] = None
-    input_schema: Optional[Dict[str, Any]] = None
-
-    @classmethod
-    async def from_agent(cls, agent: Agent) -> "AgentResponse":
-        def filter_meaningful_config(d: Dict[str, Any], defaults: Dict[str, Any]) -> Optional[Dict[str, Any]]:
-            """Filter out fields that match their default values, keeping only meaningful user configurations"""
-            filtered = {}
-            for key, value in d.items():
-                if value is None:
-                    continue
-                # Skip if value matches the default exactly
-                if key in defaults and value == defaults[key]:
-                    continue
-                # Keep non-default values
-                filtered[key] = value
-            return filtered if filtered else None
-
-        # Define default values for filtering
-        agent_defaults = {
-            # Sessions defaults
-            "add_history_to_context": False,
-            "num_history_runs": 3,
-            "enable_session_summaries": False,
-            "search_session_history": False,
-            "cache_session": False,
-            # Knowledge defaults
-            "add_references": False,
-            "references_format": "json",
-            "enable_agentic_knowledge_filters": False,
-            # Memory defaults
-            "enable_agentic_memory": False,
-            "enable_user_memories": False,
-            # Reasoning defaults
-            "reasoning": False,
-            "reasoning_min_steps": 1,
-            "reasoning_max_steps": 10,
-            # Default tools defaults
-            "read_chat_history": False,
-            "search_knowledge": True,
-            "update_knowledge": False,
-            "read_tool_call_history": False,
-            # System message defaults
-            "system_message_role": "system",
-            "build_context": True,
-            "markdown": False,
-            "add_name_to_context": False,
-            "add_datetime_to_context": False,
-            "add_location_to_context": False,
-            "resolve_in_context": True,
-            # Extra messages defaults
-            "user_message_role": "user",
-            "build_user_context": True,
-            # Response settings defaults
-            "retries": 0,
-            "delay_between_retries": 1,
-            "exponential_backoff": False,
-            "parse_response": True,
-            "use_json_mode": False,
-            # Streaming defaults
-            "stream_events": False,
-            "stream_intermediate_steps": False,
-        }
-
-        session_id = str(uuid4())
-        run_id = str(uuid4())
-        agent_tools = await agent.aget_tools(
-            session=AgentSession(session_id=session_id, session_data={}),
-            run_response=RunOutput(run_id=run_id, session_id=session_id),
-            run_context=RunContext(run_id=run_id, session_id=session_id, user_id=agent.user_id),
-            check_mcp_tools=False,
-        )
-        formatted_tools = format_tools(agent_tools) if agent_tools else None
-
-        additional_input = agent.additional_input
-        if additional_input and isinstance(additional_input[0], Message):
-            additional_input = [message.to_dict() for message in additional_input]  # type: ignore
-
-        # Build model only if it has at least one non-null field
-        model_name = agent.model.name if (agent.model and agent.model.name) else None
-        model_provider = agent.model.provider if (agent.model and agent.model.provider) else None
-        model_id = agent.model.id if (agent.model and agent.model.id) else None
-        _agent_model_data: Dict[str, Any] = {}
-        if model_name is not None:
-            _agent_model_data["name"] = model_name
-        if model_id is not None:
-            _agent_model_data["model"] = model_id
-        if model_provider is not None:
-            _agent_model_data["provider"] = model_provider
-
-        session_table = agent.db.session_table_name if agent.db else None
-        knowledge_table = agent.db.knowledge_table_name if agent.db and agent.knowledge else None
-
-        tools_info = {
-            "tools": formatted_tools,
-            "tool_call_limit": agent.tool_call_limit,
-            "tool_choice": agent.tool_choice,
-        }
-
-        sessions_info = {
-            "session_table": session_table,
-            "add_history_to_context": agent.add_history_to_context,
-            "enable_session_summaries": agent.enable_session_summaries,
-            "num_history_runs": agent.num_history_runs,
-            "search_session_history": agent.search_session_history,
-            "num_history_sessions": agent.num_history_sessions,
-            "cache_session": agent.cache_session,
-        }
-
-        knowledge_info = {
-            "knowledge_table": knowledge_table,
-            "enable_agentic_knowledge_filters": agent.enable_agentic_knowledge_filters,
-            "knowledge_filters": agent.knowledge_filters,
-            "references_format": agent.references_format,
-        }
-
-        memory_info: Optional[Dict[str, Any]] = None
-        if agent.memory_manager is not None:
-            memory_info = {
-                "enable_agentic_memory": agent.enable_agentic_memory,
-                "enable_user_memories": agent.enable_user_memories,
-                "metadata": agent.metadata,
-                "memory_table": agent.db.memory_table_name if agent.db and agent.enable_user_memories else None,
-            }
-
-            if agent.memory_manager.model is not None:
-                memory_info["model"] = ModelResponse(
-                    name=agent.memory_manager.model.name,
-                    model=agent.memory_manager.model.id,
-                    provider=agent.memory_manager.model.provider,
-                ).model_dump()
-
-        reasoning_info: Dict[str, Any] = {
-            "reasoning": agent.reasoning,
-            "reasoning_agent_id": agent.reasoning_agent.id if agent.reasoning_agent else None,
-            "reasoning_min_steps": agent.reasoning_min_steps,
-            "reasoning_max_steps": agent.reasoning_max_steps,
-        }
-
-        if agent.reasoning_model:
-            reasoning_info["reasoning_model"] = ModelResponse(
-                name=agent.reasoning_model.name,
-                model=agent.reasoning_model.id,
-                provider=agent.reasoning_model.provider,
-            ).model_dump()
-
-        default_tools_info = {
-            "read_chat_history": agent.read_chat_history,
-            "search_knowledge": agent.search_knowledge,
-            "update_knowledge": agent.update_knowledge,
-            "read_tool_call_history": agent.read_tool_call_history,
-        }
-
-        instructions = agent.instructions if agent.instructions else None
-        if instructions and callable(instructions):
-            instructions = await aexecute_instructions(instructions=instructions, agent=agent)
-
-        system_message = agent.system_message if agent.system_message else None
-        if system_message and callable(system_message):
-            system_message = await aexecute_system_message(system_message=system_message, agent=agent)
-
-        system_message_info = {
-            "system_message": str(system_message) if system_message else None,
-            "system_message_role": agent.system_message_role,
-            "build_context": agent.build_context,
-            "description": agent.description,
-            "instructions": instructions,
-            "expected_output": agent.expected_output,
-            "additional_context": agent.additional_context,
-            "markdown": agent.markdown,
-            "add_name_to_context": agent.add_name_to_context,
-            "add_datetime_to_context": agent.add_datetime_to_context,
-            "add_location_to_context": agent.add_location_to_context,
-            "timezone_identifier": agent.timezone_identifier,
-            "resolve_in_context": agent.resolve_in_context,
-        }
-
-        extra_messages_info = {
-            "additional_input": additional_input,  # type: ignore
-            "user_message_role": agent.user_message_role,
-            "build_user_context": agent.build_user_context,
-        }
-
-        response_settings_info: Dict[str, Any] = {
-            "retries": agent.retries,
-            "delay_between_retries": agent.delay_between_retries,
-            "exponential_backoff": agent.exponential_backoff,
-            "output_schema_name": agent.output_schema.__name__ if agent.output_schema else None,
-            "parser_model_prompt": agent.parser_model_prompt,
-            "parse_response": agent.parse_response,
-            "structured_outputs": agent.structured_outputs,
-            "use_json_mode": agent.use_json_mode,
-            "save_response_to_file": agent.save_response_to_file,
-        }
-
-        if agent.parser_model:
-            response_settings_info["parser_model"] = ModelResponse(
-                name=agent.parser_model.name,
-                model=agent.parser_model.id,
-                provider=agent.parser_model.provider,
-            ).model_dump()
-
-        streaming_info = {
-            "stream": agent.stream,
-            "stream_events": agent.stream_events,
-            "stream_intermediate_steps": agent.stream_intermediate_steps,
-        }
-
-        return AgentResponse(
-            id=agent.id,
-            name=agent.name,
-            db_id=agent.db.id if agent.db else None,
-            model=ModelResponse(**_agent_model_data) if _agent_model_data else None,
-            tools=filter_meaningful_config(tools_info, {}),
-            sessions=filter_meaningful_config(sessions_info, agent_defaults),
-            knowledge=filter_meaningful_config(knowledge_info, agent_defaults),
-            memory=filter_meaningful_config(memory_info, agent_defaults) if memory_info else None,
-            reasoning=filter_meaningful_config(reasoning_info, agent_defaults),
-            default_tools=filter_meaningful_config(default_tools_info, agent_defaults),
-            system_message=filter_meaningful_config(system_message_info, agent_defaults),
-            extra_messages=filter_meaningful_config(extra_messages_info, agent_defaults),
-            response_settings=filter_meaningful_config(response_settings_info, agent_defaults),
-            streaming=filter_meaningful_config(streaming_info, agent_defaults),
-            metadata=agent.metadata,
-            input_schema=get_agent_input_schema_dict(agent),
-        )
-
-
-class TeamResponse(BaseModel):
-    id: Optional[str] = None
-    name: Optional[str] = None
-    db_id: Optional[str] = None
-    description: Optional[str] = None
-    model: Optional[ModelResponse] = None
-    tools: Optional[Dict[str, Any]] = None
-    sessions: Optional[Dict[str, Any]] = None
-    knowledge: Optional[Dict[str, Any]] = None
-    memory: Optional[Dict[str, Any]] = None
-    reasoning: Optional[Dict[str, Any]] = None
-    default_tools: Optional[Dict[str, Any]] = None
-    system_message: Optional[Dict[str, Any]] = None
-    response_settings: Optional[Dict[str, Any]] = None
-    streaming: Optional[Dict[str, Any]] = None
-    members: Optional[List[Union[AgentResponse, "TeamResponse"]]] = None
-    metadata: Optional[Dict[str, Any]] = None
-    input_schema: Optional[Dict[str, Any]] = None
-
-    @classmethod
-    async def from_team(cls, team: Team) -> "TeamResponse":
-        def filter_meaningful_config(d: Dict[str, Any], defaults: Dict[str, Any]) -> Optional[Dict[str, Any]]:
-            """Filter out fields that match their default values, keeping only meaningful user configurations"""
-            filtered = {}
-            for key, value in d.items():
-                if value is None:
-                    continue
-                # Skip if value matches the default exactly
-                if key in defaults and value == defaults[key]:
-                    continue
-                # Keep non-default values
-                filtered[key] = value
-            return filtered if filtered else None
-
-        # Define default values for filtering (similar to agent defaults)
-        team_defaults = {
-            # Sessions defaults
-            "add_history_to_context": False,
-            "num_history_runs": 3,
-            "enable_session_summaries": False,
-            "cache_session": False,
-            # Knowledge defaults
-            "add_references": False,
-            "references_format": "json",
-            "enable_agentic_knowledge_filters": False,
-            # Memory defaults
-            "enable_agentic_memory": False,
-            "enable_user_memories": False,
-            # Reasoning defaults
-            "reasoning": False,
-            "reasoning_min_steps": 1,
-            "reasoning_max_steps": 10,
-            # Default tools defaults
-            "search_knowledge": True,
-            "read_chat_history": False,
-            "get_member_information_tool": False,
-            # System message defaults
-            "system_message_role": "system",
-            "markdown": False,
-            "add_datetime_to_context": False,
-            "add_location_to_context": False,
-            "resolve_in_context": True,
-            # Response settings defaults
-            "parse_response": True,
-            "use_json_mode": False,
-            # Streaming defaults
-            "stream_events": False,
-            "stream_intermediate_steps": False,
-            "stream_member_events": False,
-        }
-
-        run_id = str(uuid4())
-        session_id = str(uuid4())
-        _tools = team._determine_tools_for_model(
-            model=team.model,  # type: ignore
-            session=TeamSession(session_id=session_id, session_data={}),
-            run_response=TeamRunOutput(run_id=run_id),
-            run_context=RunContext(run_id=run_id, session_id=session_id, session_state={}),
-            async_mode=True,
-            team_run_context={},
-            check_mcp_tools=False,
-        )
-        team_tools = _tools
-        formatted_tools = format_team_tools(team_tools) if team_tools else None
-
-        model_name = team.model.name or team.model.__class__.__name__ if team.model else None
-        model_provider = team.model.provider or team.model.__class__.__name__ if team.model else ""
-        model_id = team.model.id if team.model else None
-
-        if model_provider and model_id:
-            model_provider = f"{model_provider} {model_id}"
-        elif model_name and model_id:
-            model_provider = f"{model_name} {model_id}"
-        elif model_id:
-            model_provider = model_id
-
-        session_table = team.db.session_table_name if team.db else None
-        knowledge_table = team.db.knowledge_table_name if team.db and team.knowledge else None
-
-        tools_info = {
-            "tools": formatted_tools,
-            "tool_call_limit": team.tool_call_limit,
-            "tool_choice": team.tool_choice,
-        }
-
-        sessions_info = {
-            "session_table": session_table,
-            "add_history_to_context": team.add_history_to_context,
-            "enable_session_summaries": team.enable_session_summaries,
-            "num_history_runs": team.num_history_runs,
-            "cache_session": team.cache_session,
-        }
-
-        knowledge_info = {
-            "knowledge_table": knowledge_table,
-            "enable_agentic_knowledge_filters": team.enable_agentic_knowledge_filters,
-            "knowledge_filters": team.knowledge_filters,
-            "references_format": team.references_format,
-        }
-
-        memory_info: Optional[Dict[str, Any]] = None
-        if team.memory_manager is not None:
-            memory_info = {
-                "enable_agentic_memory": team.enable_agentic_memory,
-                "enable_user_memories": team.enable_user_memories,
-                "metadata": team.metadata,
-                "memory_table": team.db.memory_table_name if team.db and team.enable_user_memories else None,
-            }
-
-            if team.memory_manager.model is not None:
-                memory_info["model"] = ModelResponse(
-                    name=team.memory_manager.model.name,
-                    model=team.memory_manager.model.id,
-                    provider=team.memory_manager.model.provider,
-                ).model_dump()
-
-        reasoning_info: Dict[str, Any] = {
-            "reasoning": team.reasoning,
-            "reasoning_agent_id": team.reasoning_agent.id if team.reasoning_agent else None,
-            "reasoning_min_steps": team.reasoning_min_steps,
-            "reasoning_max_steps": team.reasoning_max_steps,
-        }
-
-        if team.reasoning_model:
-            reasoning_info["reasoning_model"] = ModelResponse(
-                name=team.reasoning_model.name,
-                model=team.reasoning_model.id,
-                provider=team.reasoning_model.provider,
-            ).model_dump()
-
-        default_tools_info = {
-            "search_knowledge": team.search_knowledge,
-            "read_chat_history": team.read_chat_history,
-            "get_member_information_tool": team.get_member_information_tool,
-        }
-
-        team_instructions = team.instructions if team.instructions else None
-        if team_instructions and callable(team_instructions):
-            team_instructions = await aexecute_instructions(instructions=team_instructions, agent=team, team=team)
-
-        team_system_message = team.system_message if team.system_message else None
-        if team_system_message and callable(team_system_message):
-            team_system_message = await aexecute_system_message(
-                system_message=team_system_message, agent=team, team=team
-            )
-
-        system_message_info = {
-            "system_message": team_system_message,
-            "system_message_role": team.system_message_role,
-            "description": team.description,
-            "instructions": team_instructions,
-            "expected_output": team.expected_output,
-            "additional_context": team.additional_context,
-            "markdown": team.markdown,
-            "add_datetime_to_context": team.add_datetime_to_context,
-            "add_location_to_context": team.add_location_to_context,
-            "resolve_in_context": team.resolve_in_context,
-        }
-
-        response_settings_info: Dict[str, Any] = {
-            "output_schema_name": team.output_schema.__name__ if team.output_schema else None,
-            "parser_model_prompt": team.parser_model_prompt,
-            "parse_response": team.parse_response,
-            "use_json_mode": team.use_json_mode,
-        }
-
-        if team.parser_model:
-            response_settings_info["parser_model"] = ModelResponse(
-                name=team.parser_model.name,
-                model=team.parser_model.id,
-                provider=team.parser_model.provider,
-            ).model_dump()
-
-        streaming_info = {
-            "stream": team.stream,
-            "stream_events": team.stream_events,
-            "stream_intermediate_steps": team.stream_intermediate_steps,
-            "stream_member_events": team.stream_member_events,
-        }
-
-        # Build team model only if it has at least one non-null field
-        _team_model_data: Dict[str, Any] = {}
-        if team.model and team.model.name is not None:
-            _team_model_data["name"] = team.model.name
-        if team.model and team.model.id is not None:
-            _team_model_data["model"] = team.model.id
-        if team.model and team.model.provider is not None:
-            _team_model_data["provider"] = team.model.provider
-
-        members: List[Union[AgentResponse, TeamResponse]] = []
-        for member in team.members:
-            if isinstance(member, Agent):
-                agent_response = await AgentResponse.from_agent(member)
-                members.append(agent_response)
-            if isinstance(member, Team):
-                team_response = await TeamResponse.from_team(member)
-                members.append(team_response)
-
-        return TeamResponse(
-            id=team.id,
-            name=team.name,
-            db_id=team.db.id if team.db else None,
-            model=ModelResponse(**_team_model_data) if _team_model_data else None,
-            tools=filter_meaningful_config(tools_info, {}),
-            sessions=filter_meaningful_config(sessions_info, team_defaults),
-            knowledge=filter_meaningful_config(knowledge_info, team_defaults),
-            memory=filter_meaningful_config(memory_info, team_defaults) if memory_info else None,
-            reasoning=filter_meaningful_config(reasoning_info, team_defaults),
-            default_tools=filter_meaningful_config(default_tools_info, team_defaults),
-            system_message=filter_meaningful_config(system_message_info, team_defaults),
-            response_settings=filter_meaningful_config(response_settings_info, team_defaults),
-            streaming=filter_meaningful_config(streaming_info, team_defaults),
-            members=members if members else None,
-            metadata=team.metadata,
-            input_schema=get_team_input_schema_dict(team),
-        )
-
-
-class WorkflowResponse(BaseModel):
-    id: Optional[str] = Field(None, description="Unique identifier for the workflow")
-    name: Optional[str] = Field(None, description="Name of the workflow")
-    db_id: Optional[str] = Field(None, description="Database identifier")
-    description: Optional[str] = Field(None, description="Description of the workflow")
-    input_schema: Optional[Dict[str, Any]] = Field(None, description="Input schema for the workflow")
-    steps: Optional[List[Dict[str, Any]]] = Field(None, description="List of workflow steps")
-    agent: Optional[AgentResponse] = Field(None, description="Agent configuration if used")
-    team: Optional[TeamResponse] = Field(None, description="Team configuration if used")
-    metadata: Optional[Dict[str, Any]] = Field(None, description="Additional metadata")
-    workflow_agent: bool = Field(False, description="Whether this workflow uses a WorkflowAgent")
-
-    @classmethod
-    async def _resolve_agents_and_teams_recursively(cls, steps: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
-        """Parse Agents and Teams into AgentResponse and TeamResponse objects.
-
-        If the given steps have nested steps, recursively work on those."""
-        if not steps:
-            return steps
-
-        def _prune_none(value: Any) -> Any:
-            # Recursively remove None values from dicts and lists
-            if isinstance(value, dict):
-                return {k: _prune_none(v) for k, v in value.items() if v is not None}
-            if isinstance(value, list):
-                return [_prune_none(v) for v in value]
-            return value
-
-        for idx, step in enumerate(steps):
-            if step.get("agent"):
-                # Convert to dict and exclude fields that are None
-                agent_response = await AgentResponse.from_agent(step.get("agent"))  # type: ignore
-                step["agent"] = agent_response.model_dump(exclude_none=True)
-
-            if step.get("team"):
-                team_response = await TeamResponse.from_team(step.get("team"))  # type: ignore
-                step["team"] = team_response.model_dump(exclude_none=True)
-
-            if step.get("steps"):
-                step["steps"] = await cls._resolve_agents_and_teams_recursively(step["steps"])
-
-            # Prune None values in the entire step
-            steps[idx] = _prune_none(step)
-
-        return steps
-
-    @classmethod
-    async def from_workflow(cls, workflow: Workflow) -> "WorkflowResponse":
-        workflow_dict = workflow.to_dict()
-        steps = workflow_dict.get("steps")
-
-        if steps:
-            steps = await cls._resolve_agents_and_teams_recursively(steps)
-
-        return cls(
-            id=workflow.id,
-            name=workflow.name,
-            db_id=workflow.db.id if workflow.db else None,
-            description=workflow.description,
-            steps=steps,
-            input_schema=get_workflow_input_schema_dict(workflow),
-            metadata=workflow.metadata,
-            workflow_agent=isinstance(workflow.agent, WorkflowAgent) if workflow.agent else False,
-        )
-
-
 class WorkflowRunRequest(BaseModel):
     input: Dict[str, Any] = Field(..., description="Input parameters for the workflow run")
     user_id: Optional[str] = Field(None, description="User identifier for the workflow run")
@@ -924,6 +365,7 @@ class RunSchema(BaseModel):
     def from_dict(cls, run_dict: Dict[str, Any]) -> "RunSchema":
         run_input = get_run_input(run_dict)
         run_response_format = "text" if run_dict.get("content_type", "str") == "str" else "json"
+
         return cls(
             run_id=run_dict.get("run_id", ""),
             parent_run_id=run_dict.get("parent_run_id", ""),