>'hh- d dl Z d dlmZmZ d dlmZmZ d dlmZ d dlm Z
d dl m Z d dl
mc mZ dZd d lmZ e j, e Zer! G d d
e Z G d d
e j4 Zyy# e$ r dZ G d d
ZY Ow xY w) N)datetimetimezone)Optional TypedDict)uuid4) 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 h/home/kushmeetdev/Regenta/Chatbot/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 F e Zd ZU eed< eed< eed< eed< ee ed< y)RunDataidtrace_id
start_time dotted_order
parent_run_idN)r r r str__annotations__r r r r r r r S s"
}$r r c
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 | _ i | _ i | _ y N)
rtget_cached_clientr _metadata_tags
_project_name_name_first_response_inputs_last_response_outputs_runs)r r r$ r% r& r' s r r z%OpenAIAgentsTracingProcessor.__init__ sM !:B$8$8$:DK %DN DJ!-D DJ02D '02D '-/DJr tracereturnc
| j r
| j }n|j r
|j }nd}t t }t j
t j }t j || }t ||||d | j |j < d| j xs i i}|j xs i }|j d |d |d d< t! |i d||||d || j" | j$ } | j& j( d
i | y # t* $ r"} t, j/ d | Y d } ~ y d } ~ ww xY w) NzAgent workflow)r run_idr r r r r r$ group_id thread_idchain) r' inputsrun_typer r r r revision_idextrar% r& zError creating trace run: r )r/ r' r! r r nowr utc agent_utilsensure_dotted_orderr r2 r r, exportgetdictr- r. r
create_run Exceptionlogger exception)
r r3 run_name trace_run_idr r run_extra
trace_dictrun_dataes
r on_trace_startz+OpenAIAgentsTracingProcessor.on_trace_start sI zz :: ::+ uw zError updating trace run: )
r2 popr rC rD r, r
update_runr0 r1 rG rH rI )r r3 runrM r$ rO s r on_trace_endz)OpenAIAgentsTracingProcessor.on_trace_end s **.. 6C -2JW:>>*5;W AUSUWH
GKK** "4y!$Z %( %8 # : : > >u~~r R $ ; ; ? ?PR S )8 4
+ ! G$$'A!%EFFGs %A4C D$DDspanc |j r%| j j |j n$| j j |j }|#t j d|j y |d }t t }|j r t j |j n"t j t j }t j |||r|d nd }t# |||||d | j |j < t j$ | }t j& | }t j( | } t+ |||||d || j di }
|j r"t j |j |
d < | j, j. d i |
y # t0 $ r"} t j3 d
| Y d } ~ y d } ~ ww xY w) Nz(No trace info found for span, skipping: r r )r r6 parent_dotted_orderr r7 r; )r' r< r r r r r; r zError creating span run: r ) parent_idr2 rD r rH warningspan_idr! r
started_atr
fromisoformatr? r r@ rA rB r get_run_name get_run_typeextract_span_datarE r rF rG rI ) r rW
parent_runr span_run_idspan_start_timer rJ r< extractedrN rO s r
on_span_startz*OpenAIAgentsTracingProcessor.on_span_start s >>
t~~.ZZ^^DMM2
!>t||nM !*-H eg,K ?? &&t7 \\(,,/
'::*"BLJ~$>RV L
(/ !* ) ( . (DJJt|| $ #//5H"//5H#55d;I
B!%! %" %",T"2!-$==26" ??-5-C-CDOO-TH\*& &&22
B # end_timer )r2 rS r\ rA ra rD rZ r rE rk r! ended_atr r^
isinstance span_datar ResponseSpanDatar0 r1 r rT )r rW rU re r$ rR r; rN s r on_span_endz(OpenAIAgentsTracingProcessor.on_span_end ss **.. t4C ' 9 9$ ? $==R8/3~~+,.2mm*+-1\\)*(0 *%#-- 26" x4!% t9 _!$^!4"%o"6-1ZZ#djj/T #!# " ==+3+A+A$--+PHZ( dnng.F.FG 3377
FP& //
> BID//
>& &&229 r c 8 | j j y r) r flushr s r shutdownz%OpenAIAgentsTracingProcessor.shutdown> KK r c 8 | j j y r) rs ru s r force_flushz(OpenAIAgentsTracingProcessor.force_flushA rw r r) )r4 N)r r r r r ls_clientClientrE listr! r r TracerP rV Spanrf rq rv ry r r r r r Z s 6 t 26 0 (,(,*."& 0 Y - -. 0 tn 0
49% 0 #3-
0 3- 0&. C
. C$ . C` Ggmm G G$4 Bgll 4 Bt 4 Bl 3GLL 3T 3@ r )loggingr r typingr r uuidr langsmithr r* agentsr langsmith.wrappers._agent_utilswrappers _agent_utilsrA HAVE_AGENTSr
r r rz getLoggerr rH r TracingProcessorr r r r s ' & %B 99K@ * 8 $%) %h w'?'? h G <K9 9<s A/ /BB