g#\UddlmZddlmZmZddlmZddlmZm Z m Z m Z m Z ddl Z ddlZddlmZmZddlmZeGd d ZeGd d Ze j.e e j0d Zded<eGddZe j.eej:e j0d ZeGddZee eeeefe j@dfZ! eGddZ"eGddZ#eGddZ$eGddZ%eGdd Z&ee e#e&fe j@dfZ' eGd!d"Z(e e"e(fZ) e j.eee)e j@d#fe j0d Z*y)$) annotations) dataclassfield)datetime) AnnotatedAnyLiteralUnioncastN)Self assert_never)now_utcc<eZdZUdZded< dZded< dZded <y) SystemPromptPartzA system prompt, generally written by the application developer. This gives the model context and guidance on how to respond. strcontentN str | None dynamic_refz system-promptzLiteral['system-prompt'] part_kind)__name__ __module__ __qualname____doc____annotations__rrI/var/www/openai/venv/lib/python3.12/site-packages/pydantic_ai/messages.pyrrs0 L$"K" +:I'9RrrcJeZdZUdZded< eeZded< dZded <y ) UserPromptPartzA user prompt, generally written by the end user. Content comes from the `user_prompt` parameter of [`Agent.run`][pydantic_ai.Agent.run], [`Agent.run_sync`][pydantic_ai.Agent.run_sync], and [`Agent.run_stream`][pydantic_ai.Agent.run_stream]. rrdefault_factoryr timestampz user-promptzLiteral['user-prompt']rN) rrrrrr_now_utcr#rrrrr r "s1 L$9Ix9&(5I%5Rrr T) defer_build)configzpydantic.TypeAdapter[Any]tool_return_tacxeZdZUdZded< ded< dZded< ee Zd ed < d Z d ed< ddZ ddZ y)ToolReturnPartzAA tool return message, this encodes the result of running a tool.r tool_namerrNr tool_call_idr!rr#z tool-returnzLiteral['tool-return']rct|jtr |jStj |jj SN) isinstancerrr' dump_jsondecodeselfs rmodel_response_strz!ToolReturnPart.model_response_strJs8 dllC (<< !++DLL9@@B Brct|jtr!tj |jdSdtj |jdiS)Njson)mode return_value)r.rdictr' dump_pythonr1s rmodel_response_objectz$ToolReturnPart.model_response_objectPsJ dllD )!--dll-H H"N$>$>t||RX$>$YZ Zrreturnrr<dict[str, Any]) rrrrrr+rr$r#rr3r:rrrr)r)7sNKN, L#L*#V9Ix90(5I%5RC [rr)cteZdZUdZded< dZded< dZded< eeZ d ed < d Z d ed < ddZ y)RetryPromptPartaA message back to a model asking it to try again. This can be sent for a number of reasons: * Pydantic validation of tool arguments failed, here content is derived from a Pydantic [`ValidationError`][pydantic_core.ValidationError] * a tool raised a [`ModelRetry`][pydantic_ai.exceptions.ModelRetry] exception * no tool was found for the tool name * the model returned plain text when a structured response was expected * Pydantic validation of a structured response failed, here content is derived from a Pydantic [`ValidationError`][pydantic_core.ValidationError] * a result validator raised a [`ModelRetry`][pydantic_ai.exceptions.ModelRetry] exception z&list[pydantic_core.ErrorDetails] | strrNrr*r+r!rr#z retry-promptzLiteral['retry-prompt']rct|jtr |j}nMtj |jddhid}t |jd|j }|dS)N__all__ctx)excludeindentz validation errors: z Fix the errors and try again.)r.rrerror_details_tar/lenr0)r2 description json_errorss rmodel_responsezRetryPromptPart.model_response~sq dllC (,,K*44T\\IX]W^K_hi4jK .//CKDVDVDXCYZK?@@rr;) rrrrrr*r+rr$r#rrKrrrr@r@[sV 43 !Iz 7#L*#V9Ix96)7I&7RArr@rc,eZdZUdZded< dZded<y) ModelRequestzkA request generated by PydanticAI and sent to a model, e.g. a message from the PydanticAI app to the model.zlist[ModelRequestPart]partsrequestzLiteral['request']kindN)rrrrrrPrrrrMrMsu !!((D (UrrMc,eZdZUdZded< dZded<y)TextPartz#A plain text response from a model.rrtextzLiteral['text']rN)rrrrrrrrrrRrRs- L+!'I'RrrRceZdZUdZded<y)ArgsJsonz Tool arguments as a JSON string.r args_jsonNrrrrrrrrrUrUs *N%rrUceZdZUdZded<y)ArgsDictz&Tool arguments as a Python dictionary.r> args_dictNrWrrrrYrYs0+rrYcveZdZUdZded< ded< dZded< d Zd ed < eddd Zdd Z ddZ ddZ y) ToolCallPartzA tool call from a model.rr*zArgsJson | ArgsDictargsNrr+z tool-callzLiteral['tool-call']rct|tr||t||St|tr||t ||St |y)z+Create a `ToolCallPart` from raw arguments.N)r.rrUr8rYr )clsr*r]r+s r from_raw_argszToolCallPart.from_raw_argssF dC y(4.,? ? d #y(4.,? ?  rct|jtr|jjSt j |jj }t|tsJdttttf|S)zReturn the arguments as a Python dictionary. This is just for convenience with models that require dicts as input. zargs should be a dict) r.r]rYrZ pydantic_core from_jsonrVr8r rr)r2r]s r args_as_dictzToolCallPart.args_as_dictsg dii *99&& &&&tyy':':;$%>'>>%DcND))rct|jtr|jjSt j |jj jS)zReturn the arguments as a JSON string. This is just for convenience with models that require JSON strings as input. )r.r]rUrVrbto_jsonrZr0r1s rargs_as_json_strzToolCallPart.args_as_json_strsF dii *99&& &$$TYY%8%89@@BBrct|jtr-t|jjj St |jjSr-)r.r]rYanyrZvaluesboolrVr1s r has_contentzToolCallPart.has_contentsC dii *tyy**1134 4 ++, ,rr-)r*rr]zstr | dict[str, Any]r+rr<r r=r;)r<rk) rrrrrr+r classmethodr`rdrgrlrrrr\r\sW#N'  $L*#V&1I#1R *C-rr\creZdZUdZded< eeZded< dZded < e d dd Z e dd Z y ) ModelResponsezMA response from a model, e.g. a message from the model to the PydanticAI app.zlist[ModelResponsePart]rNr!rr#responsezLiteral['response']rPNcB|t|g|xs tS)N)r#)rRr$)r_rr#s r from_textzModelResponse.from_textsHW%&)2IxzJJrc||gSr-r)r_ tool_calls rfrom_tool_callzModelResponse.from_tool_callsI;rr-)rrr#zdatetime | Noner<r )rtr\r<r ) rrrrrrr$r#rPrmrrrurrrrorosZW "")9Ix9 !+D *UKK  rrorP)+ __future__r _annotations dataclassesrrrtypingrrr r r pydanticrbtyping_extensionsr r _utilsrr$rr TypeAdapter ConfigDictr'rr)list ErrorDetailsrGr@ DiscriminatorModelRequestPartrMrRrUrYr\ModelResponsePartro ModelMessageModelMessagesTypeAdapterrrrrsQ2(770' SS S& SS S"-AH,@,@M`XM`M`mqMr,s)s [[ [@(8'']-G-G(HQdQYQdQdquQvw (A(A (AV NNO KLNdhNdNdepNqq4 VV V SS S && & ,, , 4-4- 4-neHl$:;=SX=S=ST_=``a)     0\=01 1/8//