Update agents.py

This commit is contained in:
Sai Soundararaj 2025-07-01 15:19:12 -07:00
parent 62cd8b1a61
commit c5fd9886ae
4 changed files with 935 additions and 75 deletions

View file

@ -6562,14 +6562,16 @@
"type": "object",
"properties": {
"agent_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the created agent"
}
},
"additionalProperties": false,
"required": [
"agent_id"
],
"title": "AgentCreateResponse"
"title": "AgentCreateResponse",
"description": "Response returned when creating a new agent."
},
"CreateAgentSessionRequest": {
"type": "object",
@ -6589,14 +6591,16 @@
"type": "object",
"properties": {
"session_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the created session"
}
},
"additionalProperties": false,
"required": [
"session_id"
],
"title": "AgentSessionCreateResponse"
"title": "AgentSessionCreateResponse",
"description": "Response returned when creating a new agent session."
},
"CreateAgentTurnRequest": {
"type": "object",
@ -6994,10 +6998,12 @@
"type": "object",
"properties": {
"turn_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the turn within a session"
},
"session_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the conversation session"
},
"input_messages": {
"type": "array",
@ -7010,7 +7016,8 @@
"$ref": "#/components/schemas/ToolResponseMessage"
}
]
}
},
"description": "List of messages that initiated this turn"
},
"steps": {
"type": "array",
@ -7038,10 +7045,12 @@
"memory_retrieval": "#/components/schemas/MemoryRetrievalStep"
}
}
}
},
"description": "Ordered list of processing steps executed during this turn"
},
"output_message": {
"$ref": "#/components/schemas/CompletionMessage"
"$ref": "#/components/schemas/CompletionMessage",
"description": "The model's generated response containing content and metadata"
},
"output_attachments": {
"type": "array",
@ -7080,15 +7089,18 @@
],
"title": "Attachment",
"description": "An attachment to an agent turn."
}
},
"description": "(Optional) Files or media attached to the agent's response"
},
"started_at": {
"type": "string",
"format": "date-time"
"format": "date-time",
"description": "Timestamp when the turn began"
},
"completed_at": {
"type": "string",
"format": "date-time"
"format": "date-time",
"description": "(Optional) Timestamp when the turn finished, if completed"
}
},
"additionalProperties": false,
@ -7116,14 +7128,16 @@
"type": "object",
"properties": {
"payload": {
"$ref": "#/components/schemas/AgentTurnResponseEventPayload"
"$ref": "#/components/schemas/AgentTurnResponseEventPayload",
"description": "Event-specific payload containing event data"
}
},
"additionalProperties": false,
"required": [
"payload"
],
"title": "AgentTurnResponseEvent"
"title": "AgentTurnResponseEvent",
"description": "An event in an agent turn response stream."
},
"AgentTurnResponseEventPayload": {
"oneOf": [
@ -7171,9 +7185,9 @@
"turn_complete",
"turn_awaiting_input"
],
"title": "AgentTurnResponseEventType",
"const": "step_complete",
"default": "step_complete"
"default": "step_complete",
"description": "Type of event being reported"
},
"step_type": {
"type": "string",
@ -7183,11 +7197,11 @@
"shield_call",
"memory_retrieval"
],
"title": "StepType",
"description": "Type of the step in an agent turn."
"description": "Type of step being executed"
},
"step_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the step within a turn"
},
"step_details": {
"oneOf": [
@ -7212,7 +7226,8 @@
"shield_call": "#/components/schemas/ShieldCallStep",
"memory_retrieval": "#/components/schemas/MemoryRetrievalStep"
}
}
},
"description": "Complete details of the executed step"
}
},
"additionalProperties": false,
@ -7222,7 +7237,8 @@
"step_id",
"step_details"
],
"title": "AgentTurnResponseStepCompletePayload"
"title": "AgentTurnResponseStepCompletePayload",
"description": "Payload for step completion events in agent turn responses."
},
"AgentTurnResponseStepProgressPayload": {
"type": "object",
@ -7237,9 +7253,9 @@
"turn_complete",
"turn_awaiting_input"
],
"title": "AgentTurnResponseEventType",
"const": "step_progress",
"default": "step_progress"
"default": "step_progress",
"description": "Type of event being reported"
},
"step_type": {
"type": "string",
@ -7249,14 +7265,15 @@
"shield_call",
"memory_retrieval"
],
"title": "StepType",
"description": "Type of the step in an agent turn."
"description": "Type of step being executed"
},
"step_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the step within a turn"
},
"delta": {
"$ref": "#/components/schemas/ContentDelta"
"$ref": "#/components/schemas/ContentDelta",
"description": "Incremental content changes during step execution"
}
},
"additionalProperties": false,
@ -7266,7 +7283,8 @@
"step_id",
"delta"
],
"title": "AgentTurnResponseStepProgressPayload"
"title": "AgentTurnResponseStepProgressPayload",
"description": "Payload for step progress events in agent turn responses."
},
"AgentTurnResponseStepStartPayload": {
"type": "object",
@ -7281,9 +7299,9 @@
"turn_complete",
"turn_awaiting_input"
],
"title": "AgentTurnResponseEventType",
"const": "step_start",
"default": "step_start"
"default": "step_start",
"description": "Type of event being reported"
},
"step_type": {
"type": "string",
@ -7293,11 +7311,11 @@
"shield_call",
"memory_retrieval"
],
"title": "StepType",
"description": "Type of the step in an agent turn."
"description": "Type of step being executed"
},
"step_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the step within a turn"
},
"metadata": {
"type": "object",
@ -7322,7 +7340,8 @@
"type": "object"
}
]
}
},
"description": "(Optional) Additional metadata for the step"
}
},
"additionalProperties": false,
@ -7331,13 +7350,15 @@
"step_type",
"step_id"
],
"title": "AgentTurnResponseStepStartPayload"
"title": "AgentTurnResponseStepStartPayload",
"description": "Payload for step start events in agent turn responses."
},
"AgentTurnResponseStreamChunk": {
"type": "object",
"properties": {
"event": {
"$ref": "#/components/schemas/AgentTurnResponseEvent"
"$ref": "#/components/schemas/AgentTurnResponseEvent",
"description": "Individual event in the agent turn response stream"
}
},
"additionalProperties": false,
@ -7345,7 +7366,7 @@
"event"
],
"title": "AgentTurnResponseStreamChunk",
"description": "streamed agent turn completion response."
"description": "Streamed agent turn completion response."
},
"AgentTurnResponseTurnAwaitingInputPayload": {
"type": "object",
@ -7360,12 +7381,13 @@
"turn_complete",
"turn_awaiting_input"
],
"title": "AgentTurnResponseEventType",
"const": "turn_awaiting_input",
"default": "turn_awaiting_input"
"default": "turn_awaiting_input",
"description": "Type of event being reported"
},
"turn": {
"$ref": "#/components/schemas/Turn"
"$ref": "#/components/schemas/Turn",
"description": "Turn data when waiting for external tool responses"
}
},
"additionalProperties": false,
@ -7373,7 +7395,8 @@
"event_type",
"turn"
],
"title": "AgentTurnResponseTurnAwaitingInputPayload"
"title": "AgentTurnResponseTurnAwaitingInputPayload",
"description": "Payload for turn awaiting input events in agent turn responses."
},
"AgentTurnResponseTurnCompletePayload": {
"type": "object",
@ -7388,12 +7411,13 @@
"turn_complete",
"turn_awaiting_input"
],
"title": "AgentTurnResponseEventType",
"const": "turn_complete",
"default": "turn_complete"
"default": "turn_complete",
"description": "Type of event being reported"
},
"turn": {
"$ref": "#/components/schemas/Turn"
"$ref": "#/components/schemas/Turn",
"description": "Complete turn data including all steps and results"
}
},
"additionalProperties": false,
@ -7401,7 +7425,8 @@
"event_type",
"turn"
],
"title": "AgentTurnResponseTurnCompletePayload"
"title": "AgentTurnResponseTurnCompletePayload",
"description": "Payload for turn completion events in agent turn responses."
},
"AgentTurnResponseTurnStartPayload": {
"type": "object",
@ -7416,12 +7441,13 @@
"turn_complete",
"turn_awaiting_input"
],
"title": "AgentTurnResponseEventType",
"const": "turn_start",
"default": "turn_start"
"default": "turn_start",
"description": "Type of event being reported"
},
"turn_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the turn within a session"
}
},
"additionalProperties": false,
@ -7429,7 +7455,8 @@
"event_type",
"turn_id"
],
"title": "AgentTurnResponseTurnStartPayload"
"title": "AgentTurnResponseTurnStartPayload",
"description": "Payload for turn start events in agent turn responses."
},
"OpenAIResponseAnnotationCitation": {
"type": "object",
@ -9596,14 +9623,17 @@
"type": "object",
"properties": {
"agent_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the agent"
},
"agent_config": {
"$ref": "#/components/schemas/AgentConfig"
"$ref": "#/components/schemas/AgentConfig",
"description": "Configuration settings for the agent"
},
"created_at": {
"type": "string",
"format": "date-time"
"format": "date-time",
"description": "Timestamp when the agent was created"
}
},
"additionalProperties": false,
@ -9612,26 +9642,31 @@
"agent_config",
"created_at"
],
"title": "Agent"
"title": "Agent",
"description": "An agent instance with configuration and metadata."
},
"Session": {
"type": "object",
"properties": {
"session_id": {
"type": "string"
"type": "string",
"description": "Unique identifier for the conversation session"
},
"session_name": {
"type": "string"
"type": "string",
"description": "Human-readable name for the session"
},
"turns": {
"type": "array",
"items": {
"$ref": "#/components/schemas/Turn"
}
},
"description": "List of all turns that have occurred in this session"
},
"started_at": {
"type": "string",
"format": "date-time"
"format": "date-time",
"description": "Timestamp when the session was created"
}
},
"additionalProperties": false,
@ -9670,14 +9705,16 @@
"shield_call": "#/components/schemas/ShieldCallStep",
"memory_retrieval": "#/components/schemas/MemoryRetrievalStep"
}
}
},
"description": "The complete step data and execution details"
}
},
"additionalProperties": false,
"required": [
"step"
],
"title": "AgentStepResponse"
"title": "AgentStepResponse",
"description": "Response containing details of a specific agent step."
},
"Benchmark": {
"type": "object",