111213141516171819202122232425

classSpanData(abc.ABC):"""    Represents span data in the trace.    """@abc.abstractmethoddefexport(self)->dict[str,Any]:"""Export the span data as a dictionary."""pass@property@abc.abstractmethoddeftype(self)->str:"""Return the type of the span."""pass

2829303132333435363738394041424344454647484950515253545556575859

classAgentSpanData(SpanData):"""    Represents an Agent Span in the trace.    Includes name, handoffs, tools, and output type.    """__slots__=("name","handoffs","tools","output_type")def__init__(self,name:str,handoffs:list[str]|None=None,tools:list[str]|None=None,output_type:str|None=None,):self.name=nameself.handoffs:list[str]|None=handoffsself.tools:list[str]|None=toolsself.output_type:str|None=output_type@propertydeftype(self)->str:return"agent"defexport(self)->dict[str,Any]:return{"type":self.type,"name":self.name,"handoffs":self.handoffs,"tools":self.tools,"output_type":self.output_type,}

6263646566676869707172737475767778798081828384858687888990919293

classFunctionSpanData(SpanData):"""    Represents a Function Span in the trace.    Includes input, output and MCP data (if applicable).    """__slots__=("name","input","output","mcp_data")def__init__(self,name:str,input:str|None,output:Any|None,mcp_data:dict[str,Any]|None=None,):self.name=nameself.input=inputself.output=outputself.mcp_data=mcp_data@propertydeftype(self)->str:return"function"defexport(self)->dict[str,Any]:return{"type":self.type,"name":self.name,"input":self.input,"output":str(self.output)ifself.outputelseNone,"mcp_data":self.mcp_data,}

 96 97 98 99100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136

classGenerationSpanData(SpanData):"""    Represents a Generation Span in the trace.    Includes input, output, model, model configuration, and usage.    """__slots__=("input","output","model","model_config","usage",)def__init__(self,input:Sequence[Mapping[str,Any]]|None=None,output:Sequence[Mapping[str,Any]]|None=None,model:str|None=None,model_config:Mapping[str,Any]|None=None,usage:dict[str,Any]|None=None,):self.input=inputself.output=outputself.model=modelself.model_config=model_configself.usage=usage@propertydeftype(self)->str:return"generation"defexport(self)->dict[str,Any]:return{"type":self.type,"input":self.input,"output":self.output,"model":self.model,"model_config":self.model_config,"usage":self.usage,}

139140141142143144145146147148149150151152153154155156157158159160161162163164165

classResponseSpanData(SpanData):"""    Represents a Response Span in the trace.    Includes response and input.    """__slots__=("response","input")def__init__(self,response:Response|None=None,input:str|list[ResponseInputItemParam]|None=None,)->None:self.response=response# This is not used by the OpenAI trace processors, but is useful for other tracing# processor implementationsself.input=input@propertydeftype(self)->str:return"response"defexport(self)->dict[str,Any]:return{"type":self.type,"response_id":self.response.idifself.responseelseNone,}

168169170171172173174175176177178179180181182183184185186187188189

classHandoffSpanData(SpanData):"""    Represents a Handoff Span in the trace.    Includes source and destination agents.    """__slots__=("from_agent","to_agent")def__init__(self,from_agent:str|None,to_agent:str|None):self.from_agent=from_agentself.to_agent=to_agent@propertydeftype(self)->str:return"handoff"defexport(self)->dict[str,Any]:return{"type":self.type,"from_agent":self.from_agent,"to_agent":self.to_agent,}

192193194195196197198199200201202203204205206207208209210211212213

classCustomSpanData(SpanData):"""    Represents a Custom Span in the trace.    Includes name and data property bag.    """__slots__=("name","data")def__init__(self,name:str,data:dict[str,Any]):self.name=nameself.data=data@propertydeftype(self)->str:return"custom"defexport(self)->dict[str,Any]:return{"type":self.type,"name":self.name,"data":self.data,}

216217218219220221222223224225226227228229230231232233234235236237

classGuardrailSpanData(SpanData):"""    Represents a Guardrail Span in the trace.    Includes name and triggered status.    """__slots__=("name","triggered")def__init__(self,name:str,triggered:bool=False):self.name=nameself.triggered=triggered@propertydeftype(self)->str:return"guardrail"defexport(self)->dict[str,Any]:return{"type":self.type,"name":self.name,"triggered":self.triggered,}

240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281

classTranscriptionSpanData(SpanData):"""    Represents a Transcription Span in the trace.    Includes input, output, model, and model configuration.    """__slots__=("input","output","model","model_config",)def__init__(self,input:str|None=None,input_format:str|None="pcm",output:str|None=None,model:str|None=None,model_config:Mapping[str,Any]|None=None,):self.input=inputself.input_format=input_formatself.output=outputself.model=modelself.model_config=model_config@propertydeftype(self)->str:return"transcription"defexport(self)->dict[str,Any]:return{"type":self.type,"input":{"data":self.inputor"","format":self.input_format,},"output":self.output,"model":self.model,"model_config":self.model_config,}

284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323

classSpeechSpanData(SpanData):"""    Represents a Speech Span in the trace.    Includes input, output, model, model configuration, and first content timestamp.    """__slots__=("input","output","model","model_config","first_content_at")def__init__(self,input:str|None=None,output:str|None=None,output_format:str|None="pcm",model:str|None=None,model_config:Mapping[str,Any]|None=None,first_content_at:str|None=None,):self.input=inputself.output=outputself.output_format=output_formatself.model=modelself.model_config=model_configself.first_content_at=first_content_at@propertydeftype(self)->str:return"speech"defexport(self)->dict[str,Any]:return{"type":self.type,"input":self.input,"output":{"data":self.outputor"","format":self.output_format,},"model":self.model,"model_config":self.model_config,"first_content_at":self.first_content_at,}

326327328329330331332333334335336337338339340341342343344345346347

classSpeechGroupSpanData(SpanData):"""    Represents a Speech Group Span in the trace.    """__slots__="input"def__init__(self,input:str|None=None,):self.input=input@propertydeftype(self)->str:return"speech_group"defexport(self)->dict[str,Any]:return{"type":self.type,"input":self.input,}

350351352353354355356357358359360361362363364365366367368369370371372373374

classMCPListToolsSpanData(SpanData):"""    Represents an MCP List Tools Span in the trace.    Includes server and result.    """__slots__=("server","result",)def__init__(self,server:str|None=None,result:list[str]|None=None):self.server=serverself.result=result@propertydeftype(self)->str:return"mcp_tools"defexport(self)->dict[str,Any]:return{"type":self.type,"server":self.server,"result":self.result,}