o
Zhh- @ s d dl Z d dlmZmZ d dlmZmZ d dlmZ d dlm Z
zd dl m Z d dl
m mZ dZW n eyE dZG d d
d
ZY nw d d lmZ e eZerfG d d
d
eZG dd
d
e jZdS dS ) N)datetimetimezone)Optional TypedDict)uuid4) run_trees)tracingTFc @ s e Zd ZdZdd ZdS ) OpenAIAgentsTracingProcessora Tracing processor for the `OpenAI Agents SDK `_.
Traces all intermediate steps of your OpenAI Agent to LangSmith.
Requirements: Make sure to install ``pip install -U langsmith[openai-agents]``.
Args:
client: An instance of langsmith.client.Client. If not provided,
a default client is created.
Example:
.. code-block:: python
from agents import (
Agent,
FileSearchTool,
Runner,
WebSearchTool,
function_tool,
set_trace_processors,
)
from langsmith.wrappers import OpenAIAgentsTracingProcessor
set_trace_processors([OpenAIAgentsTracingProcessor()])
@function_tool
def get_weather(city: str) -> str:
return f"The weather in {city} is sunny"
haiku_agent = Agent(
name="Haiku agent",
instructions="Always respond in haiku form",
model="o3-mini",
tools=[get_weather],
)
agent = Agent(
name="Assistant",
tools=[WebSearchTool()],
instructions="speak in spanish. use Haiku agent if they ask for a haiku or for the weather",
handoffs=[haiku_agent],
)
result = await Runner.run(
agent,
"write a haiku about the weather today and tell me a recent news story about new york",
)
print(result.final_output)
c O s t d)NzeThe `agents` package is not installed. Please install it with `pip install langsmith[openai-agents]`.) ImportError)selfargskwargs r X/var/www/html/lang_env/lib/python3.10/site-packages/langsmith/wrappers/_openai_agents.py__init__F s %OpenAIAgentsTracingProcessor.__init__N)__name__
__module__ __qualname____doc__r r r r r r s 4r )clientc @ s: e Zd ZU eed< eed< eed< eed< ee ed< dS )RunDataidtrace_id
start_time dotted_order
parent_run_idN)r r r str__annotations__r r r r r r r S s
r c
@ s e Zd ZdZ dddddddeej dee deee dee dee f
d d
Z
d e j d dfd
dZ
d e j d dfddZde jd dfddZde jd dfddZdddZdddZdS ) r a Tracing processor for the `OpenAI Agents SDK `_.
Traces all intermediate steps of your OpenAI Agent to LangSmith.
Requirements: Make sure to install ``pip install -U langsmith[openai-agents]``.
Args:
client: An instance of langsmith.client.Client. If not provided,
a default client is created.
metadata: Metadata to associate with all traces.
tags: Tags to associate with all traces.
project_name: LangSmith project to trace to.
name: Name of the root trace.
Example:
.. code-block:: python
from agents import (
Agent,
FileSearchTool,
Runner,
WebSearchTool,
function_tool,
set_trace_processors,
)
from langsmith.wrappers import OpenAIAgentsTracingProcessor
set_trace_processors([OpenAIAgentsTracingProcessor()])
@function_tool
def get_weather(city: str) -> str:
return f"The weather in {city} is sunny"
haiku_agent = Agent(
name="Haiku agent",
instructions="Always respond in haiku form",
model="o3-mini",
tools=[get_weather],
)
agent = Agent(
name="Assistant",
tools=[WebSearchTool()],
instructions="speak in spanish. use Haiku agent if they ask for a haiku or for the weather",
handoffs=[haiku_agent],
)
result = await Runner.run(
agent,
"write a haiku about the weather today and tell me a recent news story about new york",
)
print(result.final_output)
N)metadatatags project_namenamer r r r! r" c C s< |pt | _|| _|| _|| _|| _i | _i | _i | _ d S N)
rtZget_cached_clientr _metadata_tags
_project_name_name_first_response_inputs_last_response_outputs_runs)r r r r r! r" r r r r s
r tracereturnc
C s | j r| j }n |jr|j}nd}tt }ttj}tj ||d}t
||||d d| j |j < d| j
p4i i}| p;i }|dd urK|d |d d< z t|i d||||d || j| jd }| jjd
i | W d S ty } ztd | W Y d } ~ d S d } ~ ww ) NzAgent workflow)r run_idr r r r r r Zgroup_idZ thread_idchain) r" inputsrun_typer r r r Z revision_idextrar r! zError creating trace run: r )r( r" r r r nowr utc agent_utilsensure_dotted_orderr r+ r r% exportgetdictr& r' r
create_run Exceptionlogger exception)
r r, run_nameZ trace_run_idr r Z run_extra
trace_dictrun_dataer r r on_trace_start sR
z+OpenAIAgentsTracingProcessor.on_trace_startc
C s | j |jd }| p
i }i |dpi | jpi }|r]z#| jj|d |d |d | j|ji | j |ji d|id W d S t
y\ } zt d| W Y d }~d S d }~ww d S )Nr r r r )r. r r r1 outputsr3 zError updating trace run: )
r+ popr r8 r9 r% r
update_runr) r* r< r= r> )r r, runr@ r rB r r r on_trace_end s$ z)OpenAIAgentsTracingProcessor.on_trace_endspanc C sP |j r
| j|j n| j|j}|d u r td|j d S |d }tt }|j r2t
|j nt
t
j}tj|||rB|d nd d}t|||||d d| j|j< t|}t|}t|} z(t|||||d || di d}
|j rt
|j |
d < | jjd i |
W d S ty } ztd
| W Y d } ~ d S d } ~ ww ) Nz(No trace info found for span, skipping: r r )r r. Zparent_dotted_orderr r/ r1 )r" r2 r r r r r1 r zError creating span run: r ) parent_idr+ r9 r r= warningspan_idr r Z
started_atr
fromisoformatr4 r r5 r6 r7 r Z get_run_nameZ get_run_typeextract_span_datar: r r; r< r> ) r rI Z
parent_runr Z span_run_idZspan_start_timer r? r2 extractedrA rB r r r
on_span_start s^
z*OpenAIAgentsTracingProcessor.on_span_startc
C s | j |jd }|rt|}|di }|j|d< |j|d< |j|d< ||d< |di }|di }t|d |d |d |d
|j rIt
|j nd |||d }|j r[t
|j |d < t|jtjru| j|jpj|| j|j< || j|j< | jjd
i | d S d S )Nr Zopenai_parent_idZopenai_trace_idZopenai_span_idrD r1 r r r r )r. r r r errorrD r1 r3 Zend_timer )r+ rE rL r6 rN r9 rJ r r: rQ r Zended_atr rM
isinstanceZ span_datar ZResponseSpanDatar) r* r rF )r rI rG rO r rD r1 rA r r r on_span_end s8
z(OpenAIAgentsTracingProcessor.on_span_endc C | j d S r# r flushr r r r shutdown> z%OpenAIAgentsTracingProcessor.shutdownc C rT r# rU rW r r r force_flushA rY z(OpenAIAgentsTracingProcessor.force_flushr# )r- N)r r r r r ls_clientZClientr: listr r r TracerC rH SpanrP rS rX rZ r r r r r Z s2 :
06
)loggingr r typingr r uuidr Z langsmithr r$ Zagentsr Z langsmith.wrappers._agent_utilsZwrappersZ _agent_utilsr6 Z HAVE_AGENTSr
r r r[ getLoggerr r= r ZTracingProcessorr r r r s& ?