gD ddlmZddlmZmZddlmZmZm Z ddl m Z ddl m Z mZddlmZddlmZmZmZddlZdd lmZd d lmZmZmZmZmZd d lmZm Z d d l!m"Z"m#Z#dZ$ede%Z& ee e ee&ge&fe e ee&gee&fe e&ge&fe e&gee&ffZ' ejPdZ)e Gddeee&Z*e Gdde*e&Z+e Gdde*e&eee&fZ, ddZ-y)) annotations)ABCabstractmethod) AsyncIterator AwaitableCallable)deepcopy) dataclassfield)datetime)GenericUnioncastN)TypeVar)_result_utils exceptionsmessagesmodels) AgentDeps RunContext)Usage UsageLimits) ResultDataResultValidatorFunc RunResultStreamedRunResultr)defaultz pydantic-ai) otel_scopecpeZdZUdZded<ded<ddd dZdddd Zddd d Zdddd Ze dd Z y)_BaseRunResultzBase type for results. You should not import or use this type directly, instead use its subclasses `RunResult` and `StreamedRunResult`. list[_messages.ModelMessage] _all_messagesint_new_message_indexNresult_tool_return_contentc4| td|jS)Return the history of _messages. Args: result_tool_return_content: The return content of the tool call to set in the last message. This provides a convenient way to modify the content of the result tool call if you want to continue the conversation and want to set the response to the result tool call. If `None`, the last message will not be modified. Returns: List of messages. zISetting result tool return content is not supported for this result type.)NotImplementedErrorr$selfr(s G/var/www/openai/venv/lib/python3.12/site-packages/pydantic_ai/result.py all_messagesz_BaseRunResult.all_messages3s" & 1%&qr r!!!c`tjj|j|S)a:Return all messages from [`all_messages`][pydantic_ai.result._BaseRunResult.all_messages] as JSON bytes. Args: result_tool_return_content: The return content of the tool call to set in the last message. This provides a convenient way to modify the content of the result tool call if you want to continue the conversation and want to set the response to the result tool call. If `None`, the last message will not be modified. Returns: JSON bytes representing the messages. r') _messagesModelMessagesTypeAdapter dump_jsonr/r,s r.all_messages_jsonz _BaseRunResult.all_messages_jsonD/11;;   9S  T  r0c@|j||jdS)aReturn new messages associated with this run. Messages from older runs are excluded. Args: result_tool_return_content: The return content of the tool call to set in the last message. This provides a convenient way to modify the content of the result tool call if you want to continue the conversation and want to set the response to the result tool call. If `None`, the last message will not be modified. Returns: List of new messages. r'N)r/r&r,s r. new_messagesz_BaseRunResult.new_messagesTs(  Return new messages from [`new_messages`][pydantic_ai.result._BaseRunResult.new_messages] as JSON bytes. Args: result_tool_return_content: The return content of the tool call to set in the last message. This provides a convenient way to modify the content of the result tool call if you want to continue the conversation and want to set the response to the result tool call. If `None`, the last message will not be modified. Returns: JSON bytes representing the new messages. r')r2r3r4r8r,s r.new_messages_jsonz _BaseRunResult.new_messages_jsondr6r0ctN)r+r-s r.usagez_BaseRunResult.usagets !##r0r( str | Nonereturnr#)r(r@rAbytesrAr) __name__ __module__ __qualname____doc____annotations__r/r5r8r:rr>r0r.r"r")sV 0/GK""MQ HLs MQ $$r0r"cPeZdZUdZded< ded<ded<d dZd d dd Zdd Zy )rzResult of a non-streamed run.rdatar@_result_tool_namer_usagec|jS)z"Return the usage of the whole run.)rMr=s r.r>zRunResult.usages {{r0Nr'c@||j|S|jS)r*)_set_result_tool_returnr$r,s r.r/zRunResult.all_messagess( & 1//0JK K%% %r0c6|js tdt|j}|d}|jD]A}t |t js|j|jk(s8||_ |cStd|jd)zSet return content for the result tool. Useful if you want to continue the conversation and want to set the response to the result tool call. zDCannot set result tool return content when the return type is `str`.z"No tool call found with tool name .) rL ValueErrorr r$parts isinstancer2ToolReturnPart tool_namecontent LookupError)r-return_contentr last_messageparts r.rPz!RunResult._set_result_tool_returns %%cd dD../|  &&D$ 8 89dnnPTPfPf>f- '>t?U?U>XXYZ[[r0rCr?)r[strrAr#)rDrErFrGrHr>r/rPrIr0r.rrys0' 2!! MHL&" \r0rceZdZUdZded<ded<ded<ded <d ed <d ed <ded<eddZded< ddd#dZdddd$dZdd d%dZ d&dZ e d'dZ d(dZ d)dZdd d*dZd+d Zd,d!Zy")-rzFResult of a streamed run that returns structured data via a tool call.UsageLimits | None _usage_limitszmodels.EitherStreamedResponse_stream_responsez'_result.ResultSchema[ResultData] | None_result_schemazRunContext[AgentDeps]_run_ctxz4list[_result.ResultValidator[AgentDeps, ResultData]]_result_validatorsr@rLzCallable[[], Awaitable[None]] _on_completeF)rinitbool is_completeg? debounce_byc@Kt|jtjr/|j |23d{}t t ||j|23d{\}}|j|| d{,7V6y7+76yw)amStream the response as an async iterable. The pydantic validator for structured data will be called in [partial mode](https://docs.pydantic.dev/dev/concepts/experimental/#partial-validation) on each iteration. Args: debounce_by: by how much (if at all) to debounce/group the response chunks by. `None` means no debouncing. Debouncing is particularly important for long structured responses to reduce the overhead of performing validation as each token is received. Returns: An async iterable of the response data. rjN allow_partial) rVrbrStreamTextResponse stream_textrrstream_structuredvalidate_structured_result)r-rktextstructured_messageis_lasts r.streamzStreamedRunResult.streams d++V-F-F G"..;.G -d:t,,595K5KXc5K5d k1(' ;;_logfirespanrVrStreamStructuredResponser UserErrorrgroup_by_temporaljoingetappend_validate_text_result set_attribute_marked_completedr2 ModelResponse from_text) r-rwrkusage_checking_streamlf_span group_iter_ final_deltachunkscombinednewchunks r.rpzStreamedRunResult.stream_textsF"!D  ! !4#5#5tzz! ]]1 2g$//1P1PQ **+_``!334I;WW[e#-Ca ggd&;&;&?&?&ABB%'!334I;WW[e#-+a#%)%:%:%>%>%@E"MM%0"&C&A-1-G-GPV-X'XH"*N'XC:XWWWW!ggd&;&;&?&?d&?&KL %% X+ (Y $.XWWWW!2266T6BEMM%(CC%)%?%?%PPPH"N%%ox@,,Y-D-D-N-Nx-XYYYC3 2 2sAJ:AJ.E)J.F E-$E+ %E-(.FJ.4G5J.8G!;G?G G7G!;#G!G  G!)J.+E--F. J.9E<:J.F F F ;J. G G!G! J.GJ.!G3 'G*(G3 /A"J.IA J.J" J.% J:.J73J:cKt|j|j|j}tj d5}t |jtjrtjd|jj}|jD]6}t |tjs|js/|dfnt!j"||4d{}|23d{}|jj}|jD]6}t |tjs|js/|dfgi7o7f6dddd{7n#1d{7swYnxYw|jjd}|df|j%d||j'|d{7 dddy#1swYyxYww)aStream the response as an async iterable of Structured LLM Messages. !!! note This method will fail if the response is text, e.g. if [`is_structured`][pydantic_ai.result.StreamedRunResult.is_structured] returns `False`. Args: debounce_by: by how much (if at all) to debounce/group the response chunks by. `None` means no debouncing. Debouncing is particularly important for long structured responses to reduce the overhead of performing validation as each token is received. Returns: An async iterable of the structured response message and whether that is the last message. zresponse stream structuredz>stream_structured() can only be used with structured responsesFNTrzstructured_response)r}rbrar>r~rrVrrorrrrUr2 ToolCallPart has_contentrrrr)r-rkrrmsgitemrrs r.rqz#StreamedRunResult.stream_structured s"!D  ! !4#5#5tzz! ]]7 8G$//1J1JK **+kll++//1IID!$ (>(>?DDTDTDV!5j(&"334I;WW[e#-&a"33779$'IID)$ 0F0FGDL\L\L^&)5j 0 %%.X&:XWWWW++//d/;4i%%&;SA,,S111)9 8 8sAG7A=G+G+$G+6E&7G+:E==E*E( E*AE= E= E=&G+(E**E=+ G+6E97G+=F FF AG+GG+" G7+G40G7c~Kt|j|j|j}|23d{} 76t |jt j rdj|jjd}|j|d{7}|jtjj|d{7tt|S|jjd}|j|d{7|j!|d{7Sw)z2Stream the whole response, validate and return it.NryTrz)r}rbrar>rVrrorrrrr2rrrrrr)r-rrrsmessages r.get_datazStreamedRunResult.get_data5s C  ! !4#5#5tzz! - !  , d++V-F-F G774004444@AD33D999D(()@)@)J)J4)PQ Q Q D) )++//d/;G((1 1 188AAA AsQ.D=<:<D=<A$D= B#!6D=CAD=D D=7D:8D=cJt|jtjS)zQReturn whether the stream response contains structured data (as opposed to text).)rVrbrrr=s r. is_structuredzStreamedRunResult.is_structuredHs$//1P1PQQr0cd|jj|jjzS)zReturn the usage of the whole run. !!! note This won't return the full usage until the stream is finished. )rdr>rbr=s r.r>zStreamedRunResult.usageMs( }}""T%:%:%@%@%BBBr0c6|jjS)z"Get the timestamp of the response.)rb timestampr=s r.rzStreamedRunResult.timestampUs$$..00r0rmcK|jJd|jJd|jj|j|j}|0t j d|jj |\}}|j||d}|jD]'}|j|||jd{})|S7w)z%Validate a structured result message.Nz&Expected _result_schema to not be Nonez)Expected _result_tool_name to not be Nonez&Invalid message, unable to find tool: F)rnwrap_validation_errors) rcrLfind_named_toolrUrUnexpectedModelBehavior tool_namesvalidatererd)r-rrnmatchcall result_tool result_data validators r.rrz,StreamedRunResult.validate_structured_resultYs"".X0XX.%%1^3^^1##33GMM4CYCYZ =4489L9L9W9W9Y8Z[ "k!**4}ej*k 00I ) 2 2;dmm TTK1UsCCC Cc~K|jD]'}|j|d|jd{})|S7wr<)rerrd)r-rsrs r.rz'StreamedRunResult._validate_text_resultlsD00I"++ D1  s 0=; =cKd|_|jj||jd{y7w)NT)rir$rrf)r-rs r.rz#StreamedRunResult._marked_completedus4 !!'*!!!s6A>AN)rk float | NonerAAsyncIterator[ResultData])rwrhrkrrAzAsyncIterator[str])rkrrAz3AsyncIterator[tuple[_messages.ModelResponse, bool]])rAr)rArhrC)rAr )r_messages.ModelResponsernrhrAr)rsr^rAr^)rrrANone)rDrErFrGrHr rirvrprqrpropertyrr>rrrrrrIr0r.rrsP%%33;;##LL!!//e%8K8;>k,27TW6Zr.1)2*)2 <)2VB&RRC1 JO.BF &"r0rcLjrfd}|SS)NcbK23d{}j|%7 6ywr<) check_tokens)r get_usagelimitsstream_responses r._usage_checking_iteratorzE_get_usage_checking_stream_response.._usage_checking_iterators.- d##IK0  os/-+-/-/)has_token_limits)rrrrs``` r.r}r}{s,f557  ())r0)rrrr`rzCallable[[], Usage]rAr). __future__r _annotationsabcrrcollections.abcrrrcopyr dataclassesr r r typingr rr logfire_apityping_extensionsrryrrrrr2rtoolsrrr>rr__all__r^rrLogfirer~r"rrr}rIr0r.rsf2#>>(''%HH(% O\3 / 1 j#Z 0* <= j#Z 0)J2G GH j\: %& j\9Z0 013  ;  - 8 L$S'*-L$ L$^ *\z**\ *\Z P"z2GIz