Ng JUdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddlm Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-m.Z/ddl-m0Z0m1Z1m2Z2dd l3m4Z5dd l6m7Z7er dd l8m9Z9dd l:m;Z;e j<e=Z>ej?ee0j@d dZAej?eeBddZCej?eeeBddZDej?eeeBefddZEej?ee#eFedfddZGej?ee/jHddZIeAeCeDeEeGeIdZJdeKd<ddZL dddZMejNddddddddd/ZOeLZPdd2ZQddddddddd3dd>ZRddAZSGdBdCe!dDEZTe"dFdGHZUe)dIZVe&GdJdKeeeVeUfZWe%ddLZXe% dddddddddddN ddSZXddVZXGdWdXZYddYZZdd\Z[hd]Z\Gd^d_e!dDEZ]Gd`dae!dDEZ^ dddhZ_ddlZ` dddoZa dddrZb dddtZcdduZdddxZeddyZfejgdz{dd}ZhddZiddZjddZkddZle"dZmGddeemZnGddeneemZoGddeneemZpddZqdS)z1Decorator for creating a run tree from functions.) annotationsN) copy_context) TYPE_CHECKINGAnyAsyncGenerator AsyncIterator AwaitableCallableDict GeneratorGenericIteratorListLiteralMappingOptionalProtocolSequenceSetTupleType TypedDictTypeVarUnioncastoverloadruntime_checkable) Annotated ParamSpec TypeGuardget_args get_originclient) run_treesschemasutils)_aiter) _runtime_env) TracebackType)Runnable_PARENT_RUN_TREEdefault _PROJECT_NAME_TAGS _METADATAlocal_TRACING_ENABLED_CLIENTparent project_nametagsmetadataenabledr$z!Dict[str, contextvars.ContextVar] _CONTEXT_KEYSreturnOptional[run_trees.RunTree]c4tS)zGet the current run tree.)r,getQ/var/www/html/ai-engine/env/lib/python3.11/site-packages/langsmith/run_helpers.pyget_current_run_treerCNs    ! !!rAcontextOptional[contextvars.Context]Dict[str, Any]cxttttt t dSfdtDS)z Get the current tracing context.Nr5cBi|]\}}||Sr@r?).0kvrDs rB z'get_tracing_context..`s) @ @ @$!QAw{{1~~ @ @ @rA) r,r?r/r0r1r3r4r;itemsrDs`rBget_tracing_contextrPSs&**,,)--//IIKK! '++--kkmm     A @ @ @-*=*=*?*? @ @ @@rA)r7r8r9r6r:r$r7 Optional[str]r8Optional[List[str]]r9Optional[Dict[str, Any]]r60Optional[Union[run_trees.RunTree, Mapping, str]]r:'Optional[Union[bool, Literal['local']]]r$Optional[ls_client.Client]kwargsrGenerator[None, None, None]c +K|rtjd|dtt}t d|p|di}|Bt t|pgt|jpgz}i|j |pi}||n|d}t||||||d dVt|dS#t|wxYw)aSet the tracing context for a block of code. Args: project_name: The name of the project to log the run to. Defaults to None. tags: The tags to add to the run. Defaults to None. metadata: The metadata to add to the run. Defaults to None. parent: The parent run to use for the context. Can be a Run/RunTree object, request headers (for distributed tracing), or the dotted order string. Defaults to None. client: The client to use for logging the run to LangSmith. Defaults to None, enabled: Whether tracing is enabled. Defaults to None, meaning it will use the current context value or environment variables. z Unrecognized keyword arguments: .r6 parent_runNr:r5) warningswarnDeprecationWarningrP_get_parent_runr?sortedsetr8r9_set_tracing_context) r7r8r9r6r:r$rWcurrent_contextr[s rBtracing_contextrdcs-4  8v 8 8 8    *++O (F,Nfjj6N6N!OPPJc$*"ooJO,Ar(B(BBCC>j)>hn"> ,gg/2E2Ei2P2PG (       . _-----_----s C C1func'TypeGuard[SupportsLangsmithExtra[P, R]]ct|pQt|tjot|jp#t |dot|jS)z,Check if a function is @traceable decorated.__call__)_is_traceable_function isinstance functoolspartialrehasattrrhres rBis_traceable_functionros] t$$ Q tY. / / U4J494U4U Q D* % % O*@*O*OrAnamer9r8r$ reduce_fnr7process_inputsprocess_outputsCallable[P, R]rqOptional[Mapping[str, Any]]rr$Optional[Callable[[Sequence], dict]]rs Optional[Callable[[dict], dict]]rtOptional[Callable[..., dict]]SupportsLangsmithExtra[P, R]c dt|r|St|||||||||S)z$Ensure that a function is traceable.rp)ro traceable) rerqr9r8r$rrr7rsrts rBensure_traceabler}s[T""  9  !%'      rAr boolc|tj|p(t|dotj|jS)z;Inspect function or wrapped function to see if it is async. __wrapped__)inspectiscoroutinefunctionrmrrns rBis_asyncrs:  &t , , m$$V)DTEU)V)VrAceZdZUdZded< ded< ded< ded < d ed < ded < d ed< ded< ded< ded< ded<dS)LangSmithExtraz float: return x + y my_function(5, 6) @traceable async def my_async_function(query_params: dict) -> dict: async with httpx.AsyncClient() as http_client: response = await http_client.get( "https://api.example.com/data", params=query_params, ) return response.json() asyncio.run(my_async_function({"param": "value"})) Streaming data with a generator: .. code-block:: python @traceable def my_generator(n: int) -> Iterable: for i in range(n): yield i for item in my_generator(5): print(item) Async streaming data: .. code-block:: python @traceable async def my_async_generator(query_params: dict) -> Iterable: async with httpx.AsyncClient() as http_client: response = await http_client.get( "https://api.example.com/data", params=query_params, ) for item in response.json(): yield item async def async_code(): async for item in my_async_generator({"param": "value"}): print(item) asyncio.run(async_code()) Specifying a run type and name: .. code-block:: python @traceable(name="CustomName", run_type="tool") def another_function(a: float, b: float) -> float: return a * b another_function(5, 6) Logging with custom metadata and tags: .. code-block:: python @traceable( metadata={"version": "1.0", "author": "John Doe"}, tags=["beta", "test"] ) def tagged_function(x): return x**2 tagged_function(5) Specifying a custom client and project name: .. code-block:: python custom_client = Client(api_key="your_api_key") @traceable(client=custom_client, project_name="My Special Project") def project_specific_function(data): return data project_specific_function({"data": "to process"}) Manually passing langsmith_extra: .. code-block:: python @traceable def manual_extra_function(x): return x**2 manual_extra_function(5, langsmith_extra={"metadata": {"version": "1.0"}}) rrNrzUnrecognized run_type: z. Must be one of: z . Did you mean @traceable(name='z')?zThe `traceable()` decorator only accepts one positional argument, which should be the run_type. All other arguments should be passed as keyword arguments.extrazJThe `extra` keyword argument is deprecated. Please use `metadata` instead.rrrqr9r8r$r7rsr) extra_outerrqr9r8r$r7rrsinvocation_params_fnrt)outputs_processorzHThe following keyword arguments are not recognized and will be ignored: rZrer c  tj}|jdddu |jdddu t jddd fd }t jddd fd }t jddd fd }t jddd fd}t jddd fd}t jddd fd}tjr|}n5tjr|}ntrr|}n |}nr|}n|}t|ddtj|} | jds| gd| j Dtj dtj j dd| j D} | |_|S)NrconfigrrrrrrWr<cbKtjt |||d{V} tj} r |d|d< s|dd |i|}|r#t j||dd{V}n?td it|d5|d{V}dddn #1swxYwYnC#t$r6}t j tj||d{V|d}~wwxYwtj|| d{V|S) z"Async version of wrapper function.container_inputrrrWNnew_runrrrDrOerroroutputsr@) aitertools aio_to_thread _setup_runasyncio_accepts_contextpopasyncio create_taskrdrP BaseExceptionshield) rrrW run_containeraccepts_contextfr_corofunction_resulte _on_run_endrrefunc_accepts_configfunc_accepts_parent_runs rB async_wrapperz3traceable..decorator..async_wrappersA#-": / / ###M ","D"F"F*B)6y)AF:&*/JJx...$///" 8,3,?y)A---''''''OO )-mI.FGG8818------888888888888888!   n,[-qQQQ   *]O       # "s<BC. C7 CCC C C D1D  Drc >Ktjt |||d{V}g} r |d|d< s|dd |i|}tj}t j|rd|r#tj||dd{V}n?td it|d5|d{V}dddn #1swxYwYt|||dr|dj dknd|| 23d{V}|WV 6nR#t$rE}tjtj ||t| d{V|d}~wwxYwtj |t| d{VdS) NrrrrrDrOllmF generatorr is_llm_runrresultsrrrr@)rrrrrr iscoroutinerrrdrP_process_async_iteratorrrr_get_function_result)rrrWrrasync_gen_resultritemrrrrerrrrs rBasync_generator_wrapperz=traceable..decorator..async_generator_wrappers!#-": / / ###M"$G, *B)6y)AF:&+/JJx...#'4#8#8#8 ","D"F"F&'788 F& F181D,mI6N222,,,,,,(( -1- 2JKKFF6F/E/E/E/E/E/E,FFFFFFFFFFFFFFF #:."/)3# i09UBB"$3# # # #       $JJJJJ # #!   n,#% 4Wi H H  *,Wi@@         sIBD  C D CD C!0D DD E/*AE**E/cft |||}tj jdddu} |r |d|d< s|dd|dj g|Ri|}n!#t$r}|||d}~wwxYw|||S) >Create a new run or create_child() if run is passed in kwargs.rrNrrrDrr)rr signature parametersr?rrunr) rrrWrrrrrrrers rBwrapperz-traceable..decorator..wrapperPs' / / M!$''266z4HHPTT $ *B)6y)AF:&*/JJx...">- ":">t"Ud"U"U"Uf"U"U     M3333  K  ? ? ? ?" "s.decorator..generator_wrapperms' / / M!$''266z4HHPTT $"$G#'O *B)6y)AF:&*/JJx...#?=#;#?#Vt#V#V#Vv#V#V ->$!'50# ...#######.NN?333     !0)DD    K /CGY/W/W X X X X" "s A0B== C*C%%C*ct|||} r |d|d< s|dd|djg|Ri|}n #t$r}||d}~wwxYwt |drt || St |drt || S|| |S) NrrrrrDr__iter__ __aiter__r)rrr Exceptionrm _TracedStream_TracedAsyncStream rrrWtrace_containerstreamrrrrerrrrs rBstream_wrapperz4traceable..decorator..stream_wrappers)) / / O *D)8)CF:&*/JJx...737NtNNNvNN    O15555 vz** N$V_iHHH-- N)&/9MMM K 8 8 8 8Ms.decorator..async_stream_wrappers|%/$< / / %%%O *D)8)CF:&*/JJx...#tT4V44444444    .{OSTUUUUUUUUUU v{++ I)&/9MMM,, I$V_iHHH*;QWXXX X X X X X X XMs3A B (BB __langsmith_traceable__Tc3NK|] }|jtjjk|V!dSrkindr Parameter VAR_KEYWORDrJparams rB z/traceable..decorator..A! :):)FFFFFFFrAr-c3NK|] }|jtjjk|V!dSrrrs rBrz/traceable..decorator..rrA)r)rrrrrWrr<r)rrrrrWrr<r)rrrr?rkwrapsisasyncgenfunctionisgeneratorfunctionrsetattrreplacevaluesr KEYWORD_ONLY __signature__)refunc_sigrrrrrrselected_wrappersigrrrrrrrs` @@rB decoratorztraceable..decorators$T**"*"5"9"9*d"K"KSW"W&155hEETQ   9=) #) #) #) #) #) #) #) #) #) #) #  ) #V   DH= = = = = = = = = = = =   = ~   9= # # # # # # # # # #   #8   DH) #) #) #) #) #) #) #) #) #) #) #) #  ) #\   DH               >   DH               @  %d + + +)@    ( . . +0   d^^ + 1#7  #0   +#1  #*  ";TBBB 011~!!(++ 1++%(^%:%:%<%< % '"3"@$ %(^%:%:%<%<C".1  *rA)rer )r ls_client RUN_TYPE_Trjstrr_VALID_RUN_TYPESr\r]lenr^_ContainerInputrkrl_handle_container_endr`keyscallable)rrWrr rrrrrs @@@@rBr|r| sZ ;"47C00 ;DGG**Z..9' H'''  >> with trace("My Operation", run_type="tool", tags=["important"]) as run: ... result = "foo" # Perform operation ... run.metadata["some-key"] = "some-value" ... run.end(outputs={"result": result}) Asynchronous usage: .. code-block:: python >>> async def main(): ... async with trace("Async Operation", run_type="tool", tags=["async"]) as run: ... result = "foo" # Await async operation ... run.metadata["some-key"] = "some-value" ... # "end" just adds the outputs and sets error to None ... # The actual patching of the run happens when the context exits ... run.end(outputs={"result": result}) >>> asyncio.run(main()) Handling specific exceptions: .. code-block:: python >>> import pytest >>> import sys >>> with trace("Test", exceptions_to_handle=(pytest.skip.Exception,)): ... if sys.platform == "win32": # Just an example ... pytest.skip("Skipping test for windows") ... result = "foo" # Perform test operation rN) inputsrr7r6r8r9r$rrexceptions_to_handle attachmentsrqr rrrrrr7rQr6rr8rRr9rvr$rVrrrr)Optional[Tuple[Type[BaseException], ...]]rOptional[schemas.Attachments]rWrc  |r=tjdt|dt||_||_||_| |_||_ ||_ ||_ | d|_ ||_||_| |_| |_| |_| |_d|_d|_dS)z_Initialize the trace context manager. Warns if unsupported kwargs are passed. zEThe `trace` context manager no longer supports the following kwargs: rZrN)r\r]r`rr^rqrrrrr7r6r?rr8r9r$rrrrold_ctx)rrqrrrr7r6r8r9r$rrrrrWs rB__init__ztrace.__init__@s,   M,&++--((,,,"        & (  :..      $8!$8!48 '+ rAr<run_trees.RunTreec t|_tj|j}t}t }|jp|jd}t|j |j |d}tt|j pg|pgz}i|jpi|piddi}|jpi}||d<t!|j} |=|r;||j|j|j||j||j|_nmt3j|jt7j|jt7j|jd |j|| pd |jpi|||jpi |_|dur|j|rt |t |t> |jt@ | tB ||jS) a]Set up the tracing context and create a new run. This method initializes the tracing context, merges tags and metadata, creates a new run (either as a child of an existing run or as a new root run), and sets up the necessary context variables. Returns: run_trees.RunTree: The newly created run. r$)r6rr$ ls_methodrr9N)rqrrrrr8rT accept_nullr.) rqidrrrr7rr8r$r)"rPrr'tracing_is_enabledr0r?r1r$r_r6rr`rar8r9r_get_project_namer7 create_childrqrrrrrr%RunTreer  _ensure_uuidrpostr,r/r4) rr: outer_tagsouter_metadataclient_ parent_run_tags_r9r project_name_s rB_setupz trace._setuposT+,, *4<88YY[[ "+;!1!1(!;!;%+ M!    sDIO 0@bABBCC }" #     j&B "* J)$*;<<  "w "&33Y{!{ ,4DLL%,Y)$+66%.%;-4&&&!*7i{(b ,2   DL d?? L       ! IIe    MM( # # #   . . .   m , , , KK |rAexc_typeOptional[Type[BaseException]] exc_valueOptional[BaseException] tracebackOptional[TracebackType]Nonec|jdS|\|jrt||jrd}n"tj}|jd|d|}|j||jLtj|j}|dur|j t|jdStj dtdS)aClean up the tracing context and finalize the run. This method handles exceptions, ends the run if necessary, patches the run if it's not disabled, and resets the tracing context. Args: exc_type: The type of the exception that occurred, if any. exc_value: The exception instance that occurred, if any. traceback: The traceback object associated with the exception, if any. N:  rTz(Tracing context was not set up properly.)rr issubclassr' _format_excrendrr$patchrbr\r]RuntimeWarning)rr1r3r5tbr:s rB _teardownztrace._teardowns <  F  ( AZ$3.. A&(( )@@Y@@B@@ L  2  & & & < #.t|<>> @traceable ... def my_function(input_data): ... # Function implementation ... pass >>> runnable = as_runnable(my_function) r)RunnableConfigRunnableLambda)InputOutputzjas_runnable requires langchain-core to be installed. You can install it with `pip install langchain-core`.NzzRas_runnable expects a function wrapped by the LangSmith @traceable decorator. Got z defined as: cneZdZdZ ddfd Zedfd Zedfd ZxZS)&as_runnable..RunnableTraceablezConverts a @traceable decorated function to a Runnable. This helps maintain a consistent LangSmith tracing context. Nrer afunc*Optional[Callable[..., Awaitable[Output]]]r<r7c Bd}||}t|rC|tdttgf||}n@t |r1ttgf||}|t|jj d|t |tttgtf|dS)NzFunc was provided as a coroutine function, but afunc was also provided. If providing both, func should be a regular function to avoid ambiguity.zG expects a function wrapped by the LangSmith @traceable decorator. Got ) _wrap_asyncr TypeErrorrr ro _wrap_sync ValueError __class__rsuperrrr )rrer]wrappedawrappedrYrZrds rBrz/as_runnable..RunnableTraceable.__init__Ts7 <@G''..H~~ Q'#7 x8$:J:J4:P:PQQ&t,, Qx8$//$:O:OPP ~.992699 GG  Xugy/@&@AB     rACallable[..., Output])Callable[[Input, RunnableConfig], Output]cPdfd }ttgf|S) z4Wrap a synchronous function to make it asynchronous.rdictrrWr<rctjtt|}di|dd|iiSNrrr@r%r'from_runnable_configrrk)rrrres rBwrap_traceablezIas_runnable..RunnableTraceable._wrap_sync..wrap_traceablezsI$,AA$tVBTBTUUtMMfMMz86LMMMMrArrkrrWr<r)rr )rerprYrZrWs` rBrbz1as_runnable..RunnableTraceable._wrap_synctsI  N N N N N N%!8&!@A>RR RrA>Optional[Callable[[Input, RunnableConfig], Awaitable[Output]]]c|dSt|std|ttdtf|d fd }ttgtf|S) z.Wrap an async function to make it synchronous.NzXRunnableTraceable expects a function wrapped by the LangSmith @traceable decorator. Got .rrkrrWr<rcKtjtt|}di|dd|iid{VSrmrn)rrrafunc_s rBawrap_traceablezKas_runnable..RunnableTraceable._wrap_async..awrap_traceables_$,AA$tVBTBTUU#VUUfUUz8>TUUUUUUUUUUrArq)rorcrr r )r]rvrurYrZrWs @rBr`z2as_runnable..RunnableTraceable._wrap_asyncs }t(//  :27::(3 &(9#9:EBBF V V V V V V%0)F2CCDo rAr)rer r]r^r<r7)rerhr<ri)r]r^r<rr) rrrrr staticmethodrbr` __classcell__)rdrYrZrWs@rBRunnableTraceabler\Ns  AE         @  S S S S S S S  S              rAry) langchain_core.runnablesrWrXlangchain_core.runnables.utilsrYrZ ImportErrorror getsourcerrc)rUrXrfn_srcryrYrZrWs @@@rB as_runnabler'sZ*KKKKKKKK@@@@@@@@@  D    ! . .  ,&|44FF , , ,+FFF , O*6 O OFL O O   GGGGGGGGGNGGGR  \ * **s" 505A A,+A,>rtoolrparserprompt embedding retrievercZeZdZUdZded<ded<ded<ded<d ed <d ed <d ed<dS)_TraceableContainer3Typed response when initializing a run a traceable.r=rrQr7 outer_projectrSr+rRr*rrzcontextvars.ContextrDNrr@rArBrrsl==((((    ,,,,####8888      rArcxeZdZUdZded<ded<ded<ded <d ed <d ed <ded<ded<ded<ded<dS)rrrrrQrqrSr9rRr8rVr$Optional[Callable]rrr7rrrxrsrNrr@rArBrrs==&&&&&&&&!!!!""""4444::::::rAr containerr Optional[Any]rr4r7c|d}|dSt|tr|nd|i}d}|r'tj}t |d|}|||tjdur||d}|Rt|rE ||dS#t$r(}t d|Yd}~dSd}~wwxYwdSdS) z End the run.rNoutputr:rrTrzFailed to run on_end function: ) r?rjrkr'r<reprr=r$r>rrLOGGERwarning) rrrroutputs_error_ stacktracerrs rB_container_endrsI }}Y''H$Wd33Lww(G9LH F 2&(( KK11Z11 LLL000 !!T)) ]]8 $ $F hv.. B F8      B B B NN@Q@@ A A A A A A A A A Bs C DDDrrkrcH|dd}|ri||}n|}|S)NrrI)rrr extra_inners rB_collect_extrars;##K66I"22 2 ! rArOptional[dict]c J|d}t|tjr|St|trUtj||dt |dSt|trWtj||dt |d}|S|d}|r|St}tj atj ||dx}r+|r'||ds|j |j kr|S|S)Nr6r$r7)r$r7rr# callbacks)r?rjr%r'rk from_headersr%r from_dotted_orderrCr)get_langchain_core_versionro dotted_order)rrr6dortrcrtrts rBr_r_s  * *F&)+,, &$  -- "&&x00*?+>+>~+N+NOO .   &# 22 "&&x00*?+>+>~+N+NOO 3    "":..H  C.00<"77 ?..x888   2  &6::k+B+B& ?S%555 JrArrc x |dpi}|d}|d}|d}|dpd} t} |p t}|dp|d} |d|pt} t i|d| i|d} t}|p:| r| jnd p/|d p|d pt j}|d }|d }| sgt jsTt j tj d td || d d |dtS|pttj}t#j|}| pt j|}|j}t+||}t,}t.}t}i|dpi|pi}|t,j|||pid|d<||d<t7|g|Ri|\}}|d}|r d||D}|r*t;|t<r||n7#t>$r*}t@!d|d|Yd }~nd }~wwxYw|d}|rD ||}n7#t>$r*}t@!d|d|Yd }~nd }~wwxYw|dpg|pgz} |t.j| | |pgz } | /| "|| | t||d|| |||}!nVtGj$tKj&||| t||d|| tKj&|d||| | | }!t jdurR |!'n<#t>$r/}t@!d|!j(d|Yd }~nd }~wwxYwt|!|| |||d|}"|tj|"d |tRj|"d|"S)rrr9r8r$rrrqrNr7rrz>LangSmith tracing is not enabled, returning original function.r)rr7rr+r*rrDr|r rci|] \}}||| Srr@)rJrKrLs rBrMz_setup_run..Xs&!!!A1rAz&Failed to infer invocation params for r9rszFailed to filter inputs for )rqrdoc)rqr serializedrr8rrrTr!) r#rqrrrrr7rr8r$rzFailed to post run r)*r?r/rr4r_rRr'rSr$log_onceloggingDEBUGrrr uuiduuid4rr_get_function_namerrr1r0rraupdate _get_inputs_and_attachments_saferNrjrkrrrr&r%r'r r(r)r#r,)#rerrrrWrr9r8r$rrrqr,r- project_cvselected_projectrid_rname_ docstringrr+r*rD metadata_rrrinvocation_paramsrrsr.rresponse_containers# rBrrs"%%m44:K"":..H   v & &D   * *F"":..9'H!%%''M%9)9)9O   v & & E/*=*=f*E*ED!!(F33Dw{{}}G!.?.Hg.. 80D0DK""$$J &(3 =K $ $ &   ~ . .  & > *  &  # % %+../EFF   h ' 'C   u799    M L   #)'"&&x00 NN      "TZ\\""C!$''I  2E,T22E I o>>K]]__NJnnG   z * * 0b  RI KK y))) X^$$$(Ik'K :9VtVVVvVVFK*../EFFP P!!!5!5f!=!=!C!C!E!E!!! ! 4Z0A4%H%H 4  !2333 P P P LLN%NN1NN O O O O O O O O P$(()9::NF F#^F++FF F F F LLDDDDD E E E E E E E E F   ( ( .B:3C DE KK 5!!! TZRE** ^^  #+   #%c** ^^  !*!7$$"""*##   & !!T)) B LLNNNN B B B LL@wz@@Q@@ A A A A A A A A B,%#%""8,, KK !#5n#EFFF KK $&8&CDDD sI.AN N6 N11N6 O P' P  P4T U%T==Urc | ||}t|||dS#t$r5}tdt |Yd}~dSd}~wwxYw)zHandle the end of run.Nrz!Unable to process trace outputs: )rrrrr)rrrrrs rBrrsF  (''00Gy'?????? FFFD477DDEEEEEEEEEFs# A"*AA"c$t|ddS)NrF)getattrrns rBriris 42E : ::rArinspect.Signaturec|j|i|}|t|j}|dd|dd|jD]S\}}|jtj j kr4||vr0| ||||T|S)z:Return a dictionary of inputs from the function signature.rNcls) bind_partialapply_defaultsrk argumentsrrrNrrrrr)rrrWboundr param_namers rB _get_inputsrs #I "D 3F 3 3E U_%%I MM&$ MM%&17799** E :*6 6 6Y&&  :!6777 j))) rAc t|g|Ri|S#t$r/}td|d|||dcYd}~Sd}~wwxYwNzFailed to get inputs for r9)rrW)rrrdebug)rrrWrs rB_get_inputs_safers096t666v666 000 AAAaAABBB////////0s A $AA A i)maxsizeSet[str]cVddfd|jDS)Nrinspect.Parameterr<r~c|jtjksHt|jtkr-t dt |jDrdSdS)Nc36K|]}|tjkVdSr)r& Attachment)rJargs rBrz;_attachment_args.._is_attachment..s*TT#C7--TTTTTTrATF) annotationr&rr"ranyr!)rs rB_is_attachmentz(_attachment_args.._is_attachments_  w1 1 1 u' ( (I 5 5TT%BR9S9STTTTT 64urAc0h|]\}}||Sr@r@)rJrqrrs rB z#_attachment_args..s<   uu@U@U     rA)rrr<r~)rrN)rrs @rB_attachment_argsrsQ     ) 4 : : < <   rA Tuple[dict, schemas.Attachments]c, t|g|Ri|}t|}|r1ii}}|D]\}}||vr|||<|||<||fS|ifS#t$r1} td|d| ||difcYd} ~ Sd} ~ wwxYwr)rrrNrrr) rrrWinferredattachment_argsrrrKrLrs rBrrs4y:4:::6::*955  '"$bKF (( " "1''%&KNN !F1II;& &| 444 AAAaAABBB//33333334s$AAA B"&BBBc~|D]'\}}t|}||(dS)zSet the tracing context.N)rNr;ra)rDrKrLvars rBrbrbsB 1A  rAr Iterator[T]rrr List[Any]Generator[T, None, Any]c#K |dt|}|rg|dr_|ddtjtjjd|id|||V#t$r}|j cYd}~Sd}~wwxYw)NTrDr new_tokentokenrqtimerW) rnext add_eventdatetimenowtimezoneutc isoformatr StopIterationvalue)rrrrrrs rBrrs #I.224CCD mI6 i(22 + ( 1 5 5$-1!!#)++#*D/  NN4 JJJ  wsB%B)) C3B?9C?CAsyncIterator[T]rAsyncGenerator[T, None]cFK |r5tjtj||dd{V}nQt dit |d5tj|d{V}dddn #1swxYwY|rg|dr_|ddtjtj j  d|id| ||WV #t$rYdSwxYw) NTrDrOrrrrr@)rrrpy_anextrdrPrrrrrrrStopAsyncIteration)rrrrrrs rBrrs   @$0' 22))4 %UU':=;S'T'TUU@@!+!4Y!?!???????D@@@@@@@@@@@@@@@ mI6 i(22 + ( 1 5 5$-1!!#)++#*D/  NN4 JJJJ+ ,      s7ADB6 DBD B BD D D TcHeZdZdZ ddd Zdd Zd ZdZdZdZ dddZ dS)_TracedStreamBasez%Base class for traced stream objects.Nr$Union[Iterator[T], AsyncIterator[T]]rrrrrc||_||_d|_||_g|_|dr|djdknd|_dS)NFrr) __ls_stream____ls_trace_container____ls_completed____ls_reduce_fn____ls_accumulated_output__r__is_llm_run__)rrrrrs rBrz_TracedStreamBase.__init__<s^ $&5# % )24&y) OI & /5 8 8 rArqr c,t|j|Sr)rr)rrqs rB __getattr__z_TracedStreamBase.__getattr__Mst)4000rActtt|jt|jzSr)listradirrdrrCs rB__dir__z_TracedStreamBase.__dir__Ps2CDN++c$2D.E.EEFFGGGrAcd|jdS)Nz Traceable())rrCs rB__repr__z_TracedStreamBase.__repr__Ss3D.3333rAc*t|jSr)r rrCs rB__str__z_TracedStreamBase.__str__Vs4%&&&rAc |js|n#t$rYnwxYw |jdS#t$rYdSwxYwr)r _end_tracerr__del__rCs rBr z_TracedStreamBase.__del__Ys ( "!!!    D     & & ( ( ( ( (    DD s ++A AArr4c|jrdS |jr||j}n|j}t|j||d|_dS#d|_wxYw)NrT)rrrrr)rrreduced_outputs rBr z_TracedStreamBase._end_traceds    F )$ @!%!6!6t7U!V!V!%!? +^5    %)D ! ! !DD ! ( ( ( (s AA Ar)rrrrrrr)rqr rr4) rrrrrrrrrr r r@rArBrr9s// )-      "1111HHH444'''    ) ) ) ) ) ) )rArcDeZdZdZ ddfd Zdd Zdd ZdZdZxZ S)rz>A wrapper for synchronous stream objects that handles tracing.Nrrrrrrrct|||||_t|j|j|j|j|_dS)Nrrrrr)rerrrrrr __ls__gen__rrrrrrds rBrz_TracedStream.__init__vsl ?i    $,    '*2    rAr<rcr t|jS#t$r|wxYwr)rrrr rCs rB__next__z_TracedStream.__next__sF ()) )    OO     s 6c#K |jEd{V|dS#t$r}||d}~wwxYwNr)rr r)rrs rBrz_TracedStream.__iter__sv ' ' ' ' ' ' ' ' OO          OO!O $ $ $  s ' A AA c4|jSr)rrDrCs rBrDz_TracedStream.__enter__s!++---rAc |j|||||r|ndS#||r|ndwxYwr)rrFr rr1exc_valexc_tbs rBrFz_TracedStream.__exit__sc A%..x&II OOX"?''4O @ @ @ @DOOX"?''4O @ @ @ @s 8Ar)rrrrrrrr<r)r<r) rrrrrrrrDrFrxrds@rBrrssHH )-        "...AAAAAAArArcNeZdZdZ ddfd Zddd ZddZddZdZdZ xZ S)rz?A wrapper for asynchronous stream objects that handles tracing.Nrrrrrrrct|||||_t|j|j|jt j|j|_ dS)Nrr) rerrrrrrrr_TracedAsyncStream__ls_genrs rBrz_TracedAsyncStream.__init__su ?i    $/(5*&>@@2     rArr4cKt}tjtj|j||d{Vt t|dS)NrH)rrrrrr rbrP)rrrJs rB _aend_tracez_TracedAsyncStream._aend_traceswnnn  $T_e3 G G G          05566666rAr<rcK tttj|jd{VS#t $r|d{VwxYwr)rrrrr rr"rCs rB __anext__z_TracedAsyncStream.__anext__s| !4T]!C!CCCCCCCDD D!   ""$$ $ $ $ $ $ $ $  s 16&AcK |j23d{V}|WV 6 |d{VdS#t$r|d{VwxYwr)r r"r)rrs rBrz_TracedAsyncStream.__aiter__s %"m       d ,m ""$$ $ $ $ $ $ $ $ $ $    ""$$ $ $ $ $ $ $ $  s66&AcDK|jd{VSr)rrLrCs rBrLz_TracedAsyncStream.__aenter__s-'22444444444rAcK |j|||d{V |d{VS#|d{VwxYwr)rrNr"rs rBrNz_TracedAsyncStream.__aexit__s %+55hPPPPPPPP P""$$ $ $ $ $ $ $ $ $$""$$ $ $ $ $ $ $ $ $s !AAr)rrrrrrrr r)r<r) rrrrrr"r$rrLrNrxrs@rBrrsII )-        $77777%%%%555%%%%%%%rArrc|rC|? ||S#t$r&}t||cYd}~Sd}~wwxYw|SdSr)rrr)rrrrs rBrrss    y)))     Q Ns  A<AA)r<r=r)rDrEr<rF)r7rQr8rRr9rSr6rTr:rUr$rVrWrr<rX)rerr<rf)rerurqrQr9rvr8rRr$rVrrrwr7rQrsrxrtryr<rz)rer r<r~)rerur<rzrO)rrrqrQr9rvr8rRr$rVrrrwr7rQrsrxrtryrrxr<r)rrrWrr<r)r7rQr<rQ)rUr r<r+)NN)rrrrrr4r<r7)rrkrrr<rk)rrrrr<r=rP) rer rrrrrrrWrr<r) rrrrrr4rryr<r7)rerr<r~)rrrrrWrr<rF)rrr<r)rrrrrWrr<r)rDrF) rrrrrr~rrr<r) rrrrrr~rr~rrr<r)rrrrr r<r)rr __future__rr contextlib contextvarsrrkrrrr\rtypingrrrrr r r r r rrrrrrrrrrrrrrrrtyping_extensionsrrr r!r" langsmithr$r r%r&r'langsmith._internalr(r langsmith.envr)typesr*rzr+ getLoggerrr ContextVarr'r,r r/r0r1r~r3Clientr4r;rrCrPcontextmanagerrdget_run_tree_contextror}rrrrrr|rr%rrrrrrr_rrrirr lru_cacherrrbrrrrrrrr@rArBr8s777"""""" $$$$$$8TSSSSSSSSSSSSS))))))//////////444444&&&&&&2######111111  8 $ $F;)(93D*EF6 &x}5otTTT 3 xS 23GTJJJ B C C C C C  CggggTM7M7M7M7M7M7M7M7`    p+p+p+p+h ! ! ! ! !)5 ! ! ! ! ; ; ; ; ;iu ; ; ; ;""%)BBBBB4"/////j15 EEEEET"%)7; F F F F F;;;;(0000T"""   #" 4444(4     F GCLL7)7)7)7)7) 7)7)7)t+A+A+A+A+A%wqz+A+A+A\4%4%4%4%4%*GAJ4%4%4%n      rA