agno 2.0.1__py3-none-any.whl → 2.3.0__py3-none-any.whl

agno/os/schema.py

@@ -3,99 +3,106 @@ from enum import Enum
 from typing import Any, Dict, Generic, List, Optional, TypeVar, Union
 from uuid import uuid4
 
-from pydantic import BaseModel
+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, KnowledgeConfig, MemoryConfig, MetricsConfig, SessionConfig
 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.workflow.agent import WorkflowAgent
 from agno.workflow.workflow import Workflow
 
 
 class BadRequestResponse(BaseModel):
-    detail: str
-    error_code: Optional[str] = None
+    model_config = ConfigDict(json_schema_extra={"example": {"detail": "Bad request", "error_code": "BAD_REQUEST"}})
 
-    class Config:
-        json_schema_extra = {"example": {"detail": "Bad request", "error_code": "BAD_REQUEST"}}
+    detail: str = Field(..., description="Error detail message")
+    error_code: Optional[str] = Field(None, description="Error code for categorization")
 
 
 class NotFoundResponse(BaseModel):
-    detail: str
-    error_code: Optional[str] = None
+    model_config = ConfigDict(json_schema_extra={"example": {"detail": "Not found", "error_code": "NOT_FOUND"}})
 
-    class Config:
-        json_schema_extra = {"example": {"detail": "Not found", "error_code": "NOT_FOUND"}}
+    detail: str = Field(..., description="Error detail message")
+    error_code: Optional[str] = Field(None, description="Error code for categorization")
 
 
 class UnauthorizedResponse(BaseModel):
-    detail: str
-    error_code: Optional[str] = None
+    model_config = ConfigDict(
+        json_schema_extra={"example": {"detail": "Unauthorized access", "error_code": "UNAUTHORIZED"}}
+    )
 
-    class Config:
-        json_schema_extra = {"example": {"detail": "Unauthorized access", "error_code": "UNAUTHORIZED"}}
+    detail: str = Field(..., description="Error detail message")
+    error_code: Optional[str] = Field(None, description="Error code for categorization")
 
 
 class UnauthenticatedResponse(BaseModel):
-    detail: str
-    error_code: Optional[str] = None
+    model_config = ConfigDict(
+        json_schema_extra={"example": {"detail": "Unauthenticated access", "error_code": "UNAUTHENTICATED"}}
+    )
 
-    class Config:
-        json_schema_extra = {"example": {"detail": "Unauthenticated access", "error_code": "UNAUTHENTICATED"}}
+    detail: str = Field(..., description="Error detail message")
+    error_code: Optional[str] = Field(None, description="Error code for categorization")
 
 
 class ValidationErrorResponse(BaseModel):
-    detail: str
-    error_code: Optional[str] = None
+    model_config = ConfigDict(
+        json_schema_extra={"example": {"detail": "Validation error", "error_code": "VALIDATION_ERROR"}}
+    )
 
-    class Config:
-        json_schema_extra = {"example": {"detail": "Validation error", "error_code": "VALIDATION_ERROR"}}
+    detail: str = Field(..., description="Error detail message")
+    error_code: Optional[str] = Field(None, description="Error code for categorization")
 
 
 class InternalServerErrorResponse(BaseModel):
-    detail: str
-    error_code: Optional[str] = None
+    model_config = ConfigDict(
+        json_schema_extra={"example": {"detail": "Internal server error", "error_code": "INTERNAL_SERVER_ERROR"}}
+    )
 
-    class Config:
-        json_schema_extra = {"example": {"detail": "Internal server error", "error_code": "INTERNAL_SERVER_ERROR"}}
+    detail: str = Field(..., description="Error detail message")
+    error_code: Optional[str] = Field(None, description="Error code for categorization")
 
 
 class HealthResponse(BaseModel):
-    status: str
+    model_config = ConfigDict(json_schema_extra={"example": {"status": "ok", "instantiated_at": "1760169236.778903"}})
 
-    class Config:
-        json_schema_extra = {"example": {"status": "ok"}}
+    status: str = Field(..., description="Health status of the service")
+    instantiated_at: str = Field(..., description="Unix timestamp when service was instantiated")
 
 
 class InterfaceResponse(BaseModel):
-    type: str
-    version: str
-    route: str
+    type: str = Field(..., description="Type of the interface")
+    version: str = Field(..., description="Version of the interface")
+    route: str = Field(..., description="API route path")
 
 
 class ManagerResponse(BaseModel):
-    type: str
-    name: str
-    version: str
-    route: str
+    type: str = Field(..., description="Type of the manager")
+    name: str = Field(..., description="Name of the manager")
+    version: str = Field(..., description="Version of the manager")
+    route: str = Field(..., description="API route path")
 
 
 class AgentSummaryResponse(BaseModel):
-    id: Optional[str] = None
-    name: Optional[str] = None
-    description: Optional[str] = None
-    db_id: Optional[str] = None
+    id: Optional[str] = Field(None, description="Unique identifier for the agent")
+    name: Optional[str] = Field(None, description="Name of the agent")
+    description: Optional[str] = Field(None, description="Description of the agent")
+    db_id: Optional[str] = Field(None, description="Database identifier")
 
     @classmethod
     def from_agent(cls, agent: Agent) -> "AgentSummaryResponse":
@@ -103,10 +110,10 @@ class AgentSummaryResponse(BaseModel):
 
 
 class TeamSummaryResponse(BaseModel):
-    id: Optional[str] = None
-    name: Optional[str] = None
-    description: Optional[str] = None
-    db_id: Optional[str] = None
+    id: Optional[str] = Field(None, description="Unique identifier for the team")
+    name: Optional[str] = Field(None, description="Name of the team")
+    description: Optional[str] = Field(None, description="Description of the team")
+    db_id: Optional[str] = Field(None, description="Database identifier")
 
     @classmethod
     def from_team(cls, team: Team) -> "TeamSummaryResponse":
@@ -114,10 +121,10 @@ class TeamSummaryResponse(BaseModel):
 
 
 class WorkflowSummaryResponse(BaseModel):
-    id: Optional[str] = None
-    name: Optional[str] = None
-    description: Optional[str] = None
-    db_id: Optional[str] = None
+    id: Optional[str] = Field(None, description="Unique identifier for the workflow")
+    name: Optional[str] = Field(None, description="Name of the workflow")
+    description: Optional[str] = Field(None, description="Description of the workflow")
+    db_id: Optional[str] = Field(None, description="Database identifier")
 
     @classmethod
     def from_workflow(cls, workflow: Workflow) -> "WorkflowSummaryResponse":
@@ -132,34 +139,34 @@ class WorkflowSummaryResponse(BaseModel):
 class ConfigResponse(BaseModel):
     """Response schema for the general config endpoint"""
 
-    os_id: str
-    name: Optional[str] = None
-    description: Optional[str] = None
-    available_models: Optional[List[str]] = None
-    databases: List[str]
-    chat: Optional[ChatConfig] = None
+    os_id: str = Field(..., description="Unique identifier for the OS instance")
+    name: Optional[str] = Field(None, description="Name of the OS instance")
+    description: Optional[str] = Field(None, description="Description of the OS instance")
+    available_models: Optional[List[str]] = Field(None, description="List of available models")
+    databases: List[str] = Field(..., description="List of database IDs")
+    chat: Optional[ChatConfig] = Field(None, description="Chat configuration")
 
-    session: Optional[SessionConfig] = None
-    metrics: Optional[MetricsConfig] = None
-    memory: Optional[MemoryConfig] = None
-    knowledge: Optional[KnowledgeConfig] = None
-    evals: Optional[EvalsConfig] = None
+    session: Optional[SessionConfig] = Field(None, description="Session configuration")
+    metrics: Optional[MetricsConfig] = Field(None, description="Metrics configuration")
+    memory: Optional[MemoryConfig] = Field(None, description="Memory configuration")
+    knowledge: Optional[KnowledgeConfig] = Field(None, description="Knowledge configuration")
+    evals: Optional[EvalsConfig] = Field(None, description="Evaluations configuration")
 
-    agents: List[AgentSummaryResponse]
-    teams: List[TeamSummaryResponse]
-    workflows: List[WorkflowSummaryResponse]
-    interfaces: List[InterfaceResponse]
+    agents: List[AgentSummaryResponse] = Field(..., description="List of registered agents")
+    teams: List[TeamSummaryResponse] = Field(..., description="List of registered teams")
+    workflows: List[WorkflowSummaryResponse] = Field(..., description="List of registered workflows")
+    interfaces: List[InterfaceResponse] = Field(..., description="List of available interfaces")
 
 
 class Model(BaseModel):
-    id: Optional[str] = None
-    provider: Optional[str] = None
+    id: Optional[str] = Field(None, description="Model identifier")
+    provider: Optional[str] = Field(None, description="Model provider name")
 
 
 class ModelResponse(BaseModel):
-    name: Optional[str] = None
-    model: Optional[str] = None
-    provider: Optional[str] = None
+    name: Optional[str] = Field(None, description="Name of the model")
+    model: Optional[str] = Field(None, description="Model identifier")
+    provider: Optional[str] = Field(None, description="Model provider name")
 
 
 class AgentResponse(BaseModel):
@@ -178,12 +185,10 @@ class AgentResponse(BaseModel):
     response_settings: Optional[Dict[str, Any]] = None
     streaming: Optional[Dict[str, Any]] = None
     metadata: Optional[Dict[str, Any]] = None
-
-    class Config:
-        exclude_none = True
+    input_schema: Optional[Dict[str, Any]] = None
 
     @classmethod
-    def from_agent(cls, agent: Agent) -> "AgentResponse":
+    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 = {}
@@ -239,13 +244,17 @@ class AgentResponse(BaseModel):
             "parse_response": True,
             "use_json_mode": False,
             # Streaming defaults
+            "stream_events": False,
             "stream_intermediate_steps": False,
         }
 
-        agent_tools = agent.get_tools(
-            session=AgentSession(session_id=str(uuid4()), session_data={}),
-            run_response=RunOutput(run_id=str(uuid4())),
-            async_mode=True,
+        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
 
@@ -371,8 +380,10 @@ class AgentResponse(BaseModel):
 
         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,
@@ -389,6 +400,7 @@ class AgentResponse(BaseModel):
             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),
         )
 
 
@@ -409,9 +421,10 @@ class TeamResponse(BaseModel):
     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
-    def from_team(cls, team: Team) -> "TeamResponse":
+    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 = {}
@@ -445,7 +458,7 @@ class TeamResponse(BaseModel):
             "reasoning_max_steps": 10,
             # Default tools defaults
             "search_knowledge": True,
-            "read_team_history": False,
+            "read_chat_history": False,
             "get_member_information_tool": False,
             # System message defaults
             "system_message_role": "system",
@@ -457,22 +470,23 @@ class TeamResponse(BaseModel):
             "parse_response": True,
             "use_json_mode": False,
             # Streaming defaults
+            "stream_events": False,
             "stream_intermediate_steps": False,
             "stream_member_events": False,
         }
 
-        if team.model is None:
-            raise ValueError("Team model is required")
-
-        team.determine_tools_for_model(
-            model=team.model,
-            session=TeamSession(session_id=str(uuid4()), session_data={}),
-            run_response=TeamRunOutput(run_id=str(uuid4())),
+        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,
-            session_state={},
             team_run_context={},
+            check_mcp_tools=False,
         )
-        team_tools = list(team._functions_for_model.values()) if team._functions_for_model else []
+        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
@@ -542,7 +556,7 @@ class TeamResponse(BaseModel):
 
         default_tools_info = {
             "search_knowledge": team.search_knowledge,
-            "read_team_history": team.read_team_history,
+            "read_chat_history": team.read_chat_history,
             "get_member_information_tool": team.get_member_information_tool,
         }
 
@@ -579,6 +593,7 @@ class TeamResponse(BaseModel):
 
         streaming_info = {
             "stream": team.stream,
+            "stream_events": team.stream_events,
             "stream_intermediate_steps": team.stream_intermediate_steps,
             "stream_member_events": team.stream_member_events,
         }
@@ -592,6 +607,15 @@ class TeamResponse(BaseModel):
         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,
@@ -606,34 +630,26 @@ class TeamResponse(BaseModel):
             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=[  # type: ignore
-                AgentResponse.from_agent(member)
-                if isinstance(member, Agent)
-                else TeamResponse.from_team(member)
-                if isinstance(member, Team)
-                else None
-                for member in team.members
-            ],
+            members=members if members else None,
             metadata=team.metadata,
+            input_schema=get_team_input_schema_dict(team),
         )
 
 
 class WorkflowResponse(BaseModel):
-    id: Optional[str] = None
-    name: Optional[str] = None
-    db_id: Optional[str] = None
-    description: Optional[str] = None
-    input_schema: Optional[Dict[str, Any]] = None
-    steps: Optional[List[Dict[str, Any]]] = None
-    agent: Optional[AgentResponse] = None
-    team: Optional[TeamResponse] = None
-    metadata: Optional[Dict[str, Any]] = None
-
-    class Config:
-        exclude_none = True
+    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
-    def _resolve_agents_and_teams_recursively(cls, steps: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
+    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."""
@@ -651,13 +667,15 @@ class WorkflowResponse(BaseModel):
         for idx, step in enumerate(steps):
             if step.get("agent"):
                 # Convert to dict and exclude fields that are None
-                step["agent"] = AgentResponse.from_agent(step["agent"]).model_dump(exclude_none=True)
+                agent_response = await AgentResponse.from_agent(step.get("agent"))  # type: ignore
+                step["agent"] = agent_response.model_dump(exclude_none=True)
 
             if step.get("team"):
-                step["team"] = TeamResponse.from_team(step["team"]).model_dump(exclude_none=True)
+                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"] = cls._resolve_agents_and_teams_recursively(step["steps"])
+                step["steps"] = await cls._resolve_agents_and_teams_recursively(step["steps"])
 
             # Prune None values in the entire step
             steps[idx] = _prune_none(step)
@@ -665,12 +683,12 @@ class WorkflowResponse(BaseModel):
         return steps
 
     @classmethod
-    def from_workflow(cls, workflow: Workflow) -> "WorkflowResponse":
+    async def from_workflow(cls, workflow: Workflow) -> "WorkflowResponse":
         workflow_dict = workflow.to_dict()
         steps = workflow_dict.get("steps")
 
         if steps:
-            steps = cls._resolve_agents_and_teams_recursively(steps)
+            steps = await cls._resolve_agents_and_teams_recursively(steps)
 
         return cls(
             id=workflow.id,
@@ -680,21 +698,22 @@ class WorkflowResponse(BaseModel):
             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]
-    user_id: Optional[str] = None
-    session_id: Optional[str] = None
+    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")
+    session_id: Optional[str] = Field(None, description="Session identifier for context persistence")
 
 
 class SessionSchema(BaseModel):
-    session_id: str
-    session_name: str
-    session_state: Optional[dict]
-    created_at: Optional[datetime]
-    updated_at: Optional[datetime]
+    session_id: str = Field(..., description="Unique identifier for the session")
+    session_name: str = Field(..., description="Human-readable name for the session")
+    session_state: Optional[dict] = Field(None, description="Current state data of the session")
+    created_at: Optional[datetime] = Field(None, description="Timestamp when session was created")
+    updated_at: Optional[datetime] = Field(None, description="Timestamp when session was last updated")
 
     @classmethod
     def from_dict(cls, session: Dict[str, Any]) -> "SessionSchema":
@@ -713,24 +732,43 @@ class SessionSchema(BaseModel):
 
 
 class DeleteSessionRequest(BaseModel):
-    session_ids: List[str]
-    session_types: List[SessionType]
+    session_ids: List[str] = Field(..., description="List of session IDs to delete", min_length=1)
+    session_types: List[SessionType] = Field(..., description="Types of sessions to delete", min_length=1)
+
+
+class CreateSessionRequest(BaseModel):
+    session_id: Optional[str] = Field(None, description="Optional session ID (generated if not provided)")
+    session_name: Optional[str] = Field(None, description="Name for the session")
+    session_state: Optional[Dict[str, Any]] = Field(None, description="Initial session state")
+    metadata: Optional[Dict[str, Any]] = Field(None, description="Additional metadata")
+    user_id: Optional[str] = Field(None, description="User ID associated with the session")
+    agent_id: Optional[str] = Field(None, description="Agent ID if this is an agent session")
+    team_id: Optional[str] = Field(None, description="Team ID if this is a team session")
+    workflow_id: Optional[str] = Field(None, description="Workflow ID if this is a workflow session")
+
+
+class UpdateSessionRequest(BaseModel):
+    session_name: Optional[str] = Field(None, description="Updated session name")
+    session_state: Optional[Dict[str, Any]] = Field(None, description="Updated session state")
+    metadata: Optional[Dict[str, Any]] = Field(None, description="Updated metadata")
+    summary: Optional[Dict[str, Any]] = Field(None, description="Session summary")
 
 
 class AgentSessionDetailSchema(BaseModel):
-    user_id: Optional[str]
-    agent_session_id: str
-    session_id: str
-    session_name: str
-    session_summary: Optional[dict]
-    session_state: Optional[dict]
-    agent_id: Optional[str]
-    total_tokens: Optional[int]
-    agent_data: Optional[dict]
-    metrics: Optional[dict]
-    chat_history: Optional[List[dict]]
-    created_at: Optional[datetime]
-    updated_at: Optional[datetime]
+    user_id: Optional[str] = Field(None, description="User ID associated with the session")
+    agent_session_id: str = Field(..., description="Unique agent session identifier")
+    session_id: str = Field(..., description="Session identifier")
+    session_name: str = Field(..., description="Human-readable session name")
+    session_summary: Optional[dict] = Field(None, description="Summary of session interactions")
+    session_state: Optional[dict] = Field(None, description="Current state of the session")
+    agent_id: Optional[str] = Field(None, description="Agent ID used in this session")
+    total_tokens: Optional[int] = Field(None, description="Total tokens used in this session")
+    agent_data: Optional[dict] = Field(None, description="Agent-specific data")
+    metrics: Optional[dict] = Field(None, description="Session metrics")
+    metadata: Optional[dict] = Field(None, description="Additional metadata")
+    chat_history: Optional[List[dict]] = Field(None, description="Complete chat history")
+    created_at: Optional[datetime] = Field(None, description="Session creation timestamp")
+    updated_at: Optional[datetime] = Field(None, description="Last update timestamp")
 
     @classmethod
     def from_session(cls, session: AgentSession) -> "AgentSessionDetailSchema":
@@ -748,6 +786,7 @@ class AgentSessionDetailSchema(BaseModel):
             if session.session_data
             else None,
             metrics=session.session_data.get("session_metrics", {}) if session.session_data else None,  # type: ignore
+            metadata=session.metadata,
             chat_history=[message.to_dict() for message in session.get_chat_history()],
             created_at=datetime.fromtimestamp(session.created_at, tz=timezone.utc) if session.created_at else None,
             updated_at=datetime.fromtimestamp(session.updated_at, tz=timezone.utc) if session.updated_at else None,
@@ -755,16 +794,19 @@ class AgentSessionDetailSchema(BaseModel):
 
 
 class TeamSessionDetailSchema(BaseModel):
-    session_id: str
-    session_name: str
-    user_id: Optional[str]
-    team_id: Optional[str]
-    session_summary: Optional[dict]
-    session_state: Optional[dict]
-    metrics: Optional[dict]
-    team_data: Optional[dict]
-    created_at: Optional[datetime]
-    updated_at: Optional[datetime]
+    session_id: str = Field(..., description="Unique session identifier")
+    session_name: str = Field(..., description="Human-readable session name")
+    user_id: Optional[str] = Field(None, description="User ID associated with the session")
+    team_id: Optional[str] = Field(None, description="Team ID used in this session")
+    session_summary: Optional[dict] = Field(None, description="Summary of team interactions")
+    session_state: Optional[dict] = Field(None, description="Current state of the session")
+    metrics: Optional[dict] = Field(None, description="Session metrics")
+    team_data: Optional[dict] = Field(None, description="Team-specific data")
+    metadata: Optional[dict] = Field(None, description="Additional metadata")
+    chat_history: Optional[List[dict]] = Field(None, description="Complete chat history")
+    created_at: Optional[datetime] = Field(None, description="Session creation timestamp")
+    updated_at: Optional[datetime] = Field(None, description="Last update timestamp")
+    total_tokens: Optional[int] = Field(None, description="Total tokens used in this session")
 
     @classmethod
     def from_session(cls, session: TeamSession) -> "TeamSessionDetailSchema":
@@ -783,26 +825,27 @@ class TeamSessionDetailSchema(BaseModel):
             if session.session_data
             else None,
             metrics=session.session_data.get("session_metrics", {}) if session.session_data else None,
+            metadata=session.metadata,
+            chat_history=[message.to_dict() for message in session.get_chat_history()],
             created_at=datetime.fromtimestamp(session.created_at, tz=timezone.utc) if session.created_at else None,
             updated_at=datetime.fromtimestamp(session.updated_at, tz=timezone.utc) if session.updated_at else None,
         )
 
 
 class WorkflowSessionDetailSchema(BaseModel):
-    user_id: Optional[str]
-    workflow_id: Optional[str]
-    workflow_name: Optional[str]
+    user_id: Optional[str] = Field(None, description="User ID associated with the session")
+    workflow_id: Optional[str] = Field(None, description="Workflow ID used in this session")
+    workflow_name: Optional[str] = Field(None, description="Name of the workflow")
+    session_id: str = Field(..., description="Unique session identifier")
+    session_name: str = Field(..., description="Human-readable session name")
 
-    session_id: str
-    session_name: str
+    session_data: Optional[dict] = Field(None, description="Complete session data")
+    session_state: Optional[dict] = Field(None, description="Current workflow state")
+    workflow_data: Optional[dict] = Field(None, description="Workflow-specific data")
+    metadata: Optional[dict] = Field(None, description="Additional metadata")
 
-    session_data: Optional[dict]
-    session_state: Optional[dict]
-    workflow_data: Optional[dict]
-    metadata: Optional[dict]
-
-    created_at: Optional[int]
-    updated_at: Optional[int]
+    created_at: Optional[int] = Field(None, description="Unix timestamp of session creation")
+    updated_at: Optional[int] = Field(None, description="Unix timestamp of last update")
 
     @classmethod
     def from_session(cls, session: WorkflowSession) -> "WorkflowSessionDetailSchema":
@@ -825,18 +868,29 @@ class WorkflowSessionDetailSchema(BaseModel):
 
 
 class RunSchema(BaseModel):
-    run_id: str
-    agent_session_id: Optional[str]
-    user_id: Optional[str]
-    run_input: Optional[str]
-    content: Optional[Union[str, dict]]
-    run_response_format: Optional[str]
-    reasoning_content: Optional[str]
-    metrics: Optional[dict]
-    messages: Optional[List[dict]]
-    tools: Optional[List[dict]]
-    events: Optional[List[dict]]
-    created_at: Optional[datetime]
+    run_id: str = Field(..., description="Unique identifier for the run")
+    parent_run_id: Optional[str] = Field(None, description="Parent run ID if this is a nested run")
+    agent_id: Optional[str] = Field(None, description="Agent ID that executed this run")
+    user_id: Optional[str] = Field(None, description="User ID associated with the run")
+    run_input: Optional[str] = Field(None, description="Input provided to the run")
+    content: Optional[Union[str, dict]] = Field(None, description="Output content from the run")
+    run_response_format: Optional[str] = Field(None, description="Format of the response (text/json)")
+    reasoning_content: Optional[str] = Field(None, description="Reasoning content if reasoning was enabled")
+    reasoning_steps: Optional[List[dict]] = Field(None, description="List of reasoning steps")
+    metrics: Optional[dict] = Field(None, description="Performance and usage metrics")
+    messages: Optional[List[dict]] = Field(None, description="Message history for the run")
+    tools: Optional[List[dict]] = Field(None, description="Tools used in the run")
+    events: Optional[List[dict]] = Field(None, description="Events generated during the run")
+    created_at: Optional[datetime] = Field(None, description="Run creation timestamp")
+    references: Optional[List[dict]] = Field(None, description="References cited in the run")
+    reasoning_messages: Optional[List[dict]] = Field(None, description="Reasoning process messages")
+    session_state: Optional[dict] = Field(None, description="Session state at the end of the run")
+    images: Optional[List[dict]] = Field(None, description="Images included in the run")
+    videos: Optional[List[dict]] = Field(None, description="Videos included in the run")
+    audio: Optional[List[dict]] = Field(None, description="Audio files included in the run")
+    files: Optional[List[dict]] = Field(None, description="Files included in the run")
+    response_audio: Optional[dict] = Field(None, description="Audio response if generated")
+    input_media: Optional[Dict[str, Any]] = Field(None, description="Input media attachments")
 
     @classmethod
     def from_dict(cls, run_dict: Dict[str, Any]) -> "RunSchema":
@@ -844,16 +898,27 @@ class RunSchema(BaseModel):
         run_response_format = "text" if run_dict.get("content_type", "str") == "str" else "json"
         return cls(
             run_id=run_dict.get("run_id", ""),
-            agent_session_id=run_dict.get("session_id", ""),
+            parent_run_id=run_dict.get("parent_run_id", ""),
+            agent_id=run_dict.get("agent_id", ""),
             user_id=run_dict.get("user_id", ""),
             run_input=run_input,
             content=run_dict.get("content", ""),
             run_response_format=run_response_format,
             reasoning_content=run_dict.get("reasoning_content", ""),
+            reasoning_steps=run_dict.get("reasoning_steps", []),
             metrics=run_dict.get("metrics", {}),
             messages=[message for message in run_dict.get("messages", [])] if run_dict.get("messages") else None,
             tools=[tool for tool in run_dict.get("tools", [])] if run_dict.get("tools") else None,
             events=[event for event in run_dict["events"]] if run_dict.get("events") else None,
+            references=run_dict.get("references", []),
+            reasoning_messages=run_dict.get("reasoning_messages", []),
+            session_state=run_dict.get("session_state"),
+            images=run_dict.get("images", []),
+            videos=run_dict.get("videos", []),
+            audio=run_dict.get("audio", []),
+            files=run_dict.get("files", []),
+            response_audio=run_dict.get("response_audio", None),
+            input_media=extract_input_media(run_dict),
             created_at=datetime.fromtimestamp(run_dict.get("created_at", 0), tz=timezone.utc)
             if run_dict.get("created_at") is not None
             else None,
@@ -861,17 +926,28 @@ class RunSchema(BaseModel):
 
 
 class TeamRunSchema(BaseModel):
-    run_id: str
-    parent_run_id: Optional[str]
-    content: Optional[Union[str, dict]]
-    reasoning_content: Optional[str]
-    run_input: Optional[str]
-    run_response_format: Optional[str]
-    metrics: Optional[dict]
-    tools: Optional[List[dict]]
-    messages: Optional[List[dict]]
-    events: Optional[List[dict]]
-    created_at: Optional[datetime]
+    run_id: str = Field(..., description="Unique identifier for the team run")
+    parent_run_id: Optional[str] = Field(None, description="Parent run ID if this is a nested run")
+    team_id: Optional[str] = Field(None, description="Team ID that executed this run")
+    content: Optional[Union[str, dict]] = Field(None, description="Output content from the team run")
+    reasoning_content: Optional[str] = Field(None, description="Reasoning content if reasoning was enabled")
+    reasoning_steps: Optional[List[dict]] = Field(None, description="List of reasoning steps")
+    run_input: Optional[str] = Field(None, description="Input provided to the run")
+    run_response_format: Optional[str] = Field(None, description="Format of the response (text/json)")
+    metrics: Optional[dict] = Field(None, description="Performance and usage metrics")
+    tools: Optional[List[dict]] = Field(None, description="Tools used in the run")
+    messages: Optional[List[dict]] = Field(None, description="Message history for the run")
+    events: Optional[List[dict]] = Field(None, description="Events generated during the run")
+    created_at: Optional[datetime] = Field(None, description="Run creation timestamp")
+    references: Optional[List[dict]] = Field(None, description="References cited in the run")
+    reasoning_messages: Optional[List[dict]] = Field(None, description="Reasoning process messages")
+    session_state: Optional[dict] = Field(None, description="Session state at the end of the run")
+    input_media: Optional[Dict[str, Any]] = Field(None, description="Input media attachments")
+    images: Optional[List[dict]] = Field(None, description="Images included in the run")
+    videos: Optional[List[dict]] = Field(None, description="Videos included in the run")
+    audio: Optional[List[dict]] = Field(None, description="Audio files included in the run")
+    files: Optional[List[dict]] = Field(None, description="Files included in the run")
+    response_audio: Optional[dict] = Field(None, description="Audio response if generated")
 
     @classmethod
     def from_dict(cls, run_dict: Dict[str, Any]) -> "TeamRunSchema":
@@ -880,10 +956,12 @@ class TeamRunSchema(BaseModel):
         return cls(
             run_id=run_dict.get("run_id", ""),
             parent_run_id=run_dict.get("parent_run_id", ""),
+            team_id=run_dict.get("team_id", ""),
             run_input=run_input,
             content=run_dict.get("content", ""),
             run_response_format=run_response_format,
             reasoning_content=run_dict.get("reasoning_content", ""),
+            reasoning_steps=run_dict.get("reasoning_steps", []),
             metrics=run_dict.get("metrics", {}),
             messages=[message for message in run_dict.get("messages", [])] if run_dict.get("messages") else None,
             tools=[tool for tool in run_dict.get("tools", [])] if run_dict.get("tools") else None,
@@ -891,20 +969,40 @@ class TeamRunSchema(BaseModel):
             created_at=datetime.fromtimestamp(run_dict.get("created_at", 0), tz=timezone.utc)
             if run_dict.get("created_at") is not None
             else None,
+            references=run_dict.get("references", []),
+            reasoning_messages=run_dict.get("reasoning_messages", []),
+            session_state=run_dict.get("session_state"),
+            images=run_dict.get("images", []),
+            videos=run_dict.get("videos", []),
+            audio=run_dict.get("audio", []),
+            files=run_dict.get("files", []),
+            response_audio=run_dict.get("response_audio", None),
+            input_media=extract_input_media(run_dict),
         )
 
 
 class WorkflowRunSchema(BaseModel):
-    run_id: str
-    run_input: Optional[str]
-    user_id: Optional[str]
-    content: Optional[Union[str, dict]]
-    content_type: Optional[str]
-    status: Optional[str]
-    step_results: Optional[list[dict]]
-    step_executor_runs: Optional[list[dict]]
-    metrics: Optional[dict]
-    created_at: Optional[int]
+    run_id: str = Field(..., description="Unique identifier for the workflow run")
+    run_input: Optional[str] = Field(None, description="Input provided to the workflow")
+    events: Optional[List[dict]] = Field(None, description="Events generated during the workflow")
+    workflow_id: Optional[str] = Field(None, description="Workflow ID that was executed")
+    user_id: Optional[str] = Field(None, description="User ID associated with the run")
+    content: Optional[Union[str, dict]] = Field(None, description="Output content from the workflow")
+    content_type: Optional[str] = Field(None, description="Type of content returned")
+    status: Optional[str] = Field(None, description="Status of the workflow run")
+    step_results: Optional[list[dict]] = Field(None, description="Results from each workflow step")
+    step_executor_runs: Optional[list[dict]] = Field(None, description="Executor runs for each step")
+    metrics: Optional[dict] = Field(None, description="Performance and usage metrics")
+    created_at: Optional[int] = Field(None, description="Unix timestamp of run creation")
+    reasoning_content: Optional[str] = Field(None, description="Reasoning content if reasoning was enabled")
+    reasoning_steps: Optional[List[dict]] = Field(None, description="List of reasoning steps")
+    references: Optional[List[dict]] = Field(None, description="References cited in the workflow")
+    reasoning_messages: Optional[List[dict]] = Field(None, description="Reasoning process messages")
+    images: Optional[List[dict]] = Field(None, description="Images included in the workflow")
+    videos: Optional[List[dict]] = Field(None, description="Videos included in the workflow")
+    audio: Optional[List[dict]] = Field(None, description="Audio files included in the workflow")
+    files: Optional[List[dict]] = Field(None, description="Files included in the workflow")
+    response_audio: Optional[dict] = Field(None, description="Audio response if generated")
 
     @classmethod
     def from_dict(cls, run_response: Dict[str, Any]) -> "WorkflowRunSchema":
@@ -912,6 +1010,8 @@ class WorkflowRunSchema(BaseModel):
         return cls(
             run_id=run_response.get("run_id", ""),
             run_input=run_input,
+            events=run_response.get("events", []),
+            workflow_id=run_response.get("workflow_id", ""),
             user_id=run_response.get("user_id", ""),
             content=run_response.get("content", ""),
             content_type=run_response.get("content_type", ""),
@@ -920,6 +1020,15 @@ class WorkflowRunSchema(BaseModel):
             step_results=run_response.get("step_results", []),
             step_executor_runs=run_response.get("step_executor_runs", []),
             created_at=run_response["created_at"],
+            reasoning_content=run_response.get("reasoning_content", ""),
+            reasoning_steps=run_response.get("reasoning_steps", []),
+            references=run_response.get("references", []),
+            reasoning_messages=run_response.get("reasoning_messages", []),
+            images=run_response.get("images", []),
+            videos=run_response.get("videos", []),
+            audio=run_response.get("audio", []),
+            files=run_response.get("files", []),
+            response_audio=run_response.get("response_audio", None),
         )
 
 
@@ -932,14 +1041,15 @@ class SortOrder(str, Enum):
 
 
 class PaginationInfo(BaseModel):
-    page: Optional[int] = 0
-    limit: Optional[int] = 20
-    total_pages: Optional[int] = 0
-    total_count: Optional[int] = 0
+    page: int = Field(0, description="Current page number (0-indexed)", ge=0)
+    limit: int = Field(20, description="Number of items per page", ge=1, le=100)
+    total_pages: int = Field(0, description="Total number of pages", ge=0)
+    total_count: int = Field(0, description="Total count of items", ge=0)
+    search_time_ms: float = Field(0, description="Search execution time in milliseconds", ge=0)
 
 
 class PaginatedResponse(BaseModel, Generic[T]):
     """Wrapper to add pagination info to classes used as response models"""
 
-    data: List[T]
-    meta: PaginationInfo
+    data: List[T] = Field(..., description="List of items for the current page")
+    meta: PaginationInfo = Field(..., description="Pagination metadata")