%Gg\" d dl Z d dlZd dlZd dlmZmZ d dlmZ d dl m
Z
d dl m c m
Z dZd d lmZ ej( e Zer G d
de
j. Zyy# e$ r dZ G d d ZY Dw xY w) N)DictOptional) run_trees)tracingTFc e Zd ZdZd Zy) 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 t d )NzeThe `agents` package is not installed. Please install it with `pip install langsmith[openai-agents]`.) ImportError)selfargskwargss ~/home/kushmeetdev/apache_webroot/langgraph_flaskproject/venv/lib/python3.12/site-packages/langsmith/wrappers/_openai_agents.py__init__z%OpenAIAgentsTracingProcessor.__init__F s Q
N)__name__
__module__ __qualname____doc__r r r r r s
2 h r r )clientc
e Zd ZdZ dddddddeej dee deee dee dee f
d Z
d
e j d dfd Z
d
e j d dfd
Zde j d dfdZde j d dfdZddZddZy)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 |xs t j | _ || _ || _ || _ || _ i | _ y N)rtget_cached_clientr _metadata_tags
_project_name_name_runs)r r r r r r s r r z%OpenAIAgentsTracingProcessor.__init__ s= !:B$8$8$:DK %DN DJ!-D DJ)+DJr tracereturnc
| j r
| j }n|j r
|j }nd}t t j }|| j
|j < | j rd| j ini } t |i d|d || j | j } | j j di | y # t $ r"}t j d| Y d }~y d }~ww xY w)NzAgent workflowr chain)r inputsrun_typeid revision_idextrar r zError creating trace run: r )r# r struuiduuid4r$ trace_idr dictr! r" r
create_run Exceptionlogger exception)r r% run_name trace_run_id run_extrarun_dataes r on_trace_startz+OpenAIAgentsTracingProcessor.on_trace_start s zz :: ::+ tzz|,L)5DJJu~~ &8<T^^4BI
C!%! $# $# !%!3!3 " ' &&22
C #=aS!ABB
Cs AC C7C22C7c h | j j |j d }|j xs i }i |j d xs i | j
xs i }|r! | j j |d|i y y # t $ r"}t j d| Y d }~y d }~ww xY w)Nr )run_idr- zError updating trace run: ) r$ popr1 exportgetr r
update_runr4 r5 r6 )r r% r>
trace_dictr r; s r on_trace_endz)OpenAIAgentsTracingProcessor.on_trace_end s ZZ^^ENND9F -2JW:>>*5;W AUSUWHGKK**&X@V*W ! G$$'A!%EFFGs % B B1B,,B1spanc
| j j |j xs |j }t t j }|| j |j < t j | }t j | }t j | } t |||||j di }|j r,t j j |j |d< | j j" di | y # t$ $ r"}t& j) d| Y d }~y d }~ww xY w)Nr) )r r* r+
parent_run_idr)
start_timezError creating span run: r )r$ rA parent_idr1 r. r/ r0 span_id agent_utils get_run_name get_run_typeextract_span_datar2
started_atdatetime
fromisoformatr r3 r4 r5 r6 ) r rE rG span_run_idr7 r* extractedr: r; s r
on_span_startz*OpenAIAgentsTracingProcessor.on_span_start s JJNN4>>+JT]] KM djjl+K'2DJJt|| $"//5H"//5H#55d;I
B!%! %""/$==26 " ??-5->->-L-L .H\* ' &&22
B # errorrY r) r- end_timer )r$ r? rJ rK rN rA rI r1 r2 rZ r. ended_atrP rQ r rB )r rE r> rS r r: s r on_span_endz(OpenAIAgentsTracingProcessor.on_span_end s ZZ^^DLL$7F ' 9 9$ ? $==R8/3~~+,.2mm*+-1\\)*(0 *%!%!-1ZZ#djj/T %MM)R 8$==26# " ==+3+<+<+J+J
,HZ( ' &&22% r c 8 | j j y r r flushr s r shutdownz%OpenAIAgentsTracingProcessor.shutdown KK r c 8 | j j y r r_ ra s r force_flushz(OpenAIAgentsTracingProcessor.force_flush rc r r )r& N)r r r r r ls_clientClientr2 listr. r r Tracer< rD SpanrT r] rb re r r r r r S s 6 t 26 , (,(,*."& , Y - -. , tn ,
49% , #3-
, 3- ," C
C$ C4 Ggmm G G Bgll Bt B2 3GLL 3T 3, r )rP loggingr/ typingr r langsmithr r agentsr langsmith.wrappers._agent_utilswrappers _agent_utilsrK HAVE_AGENTSr
r r rf getLoggerr r5 TracingProcessorr r r ru s} ! %B 99K@ * 8 $a w'?'? a G <K9 9<s A A21A2