9i7JdZddlmZddlZddlmZmZddlmZddl m Z m Z m Z m Z mZmZmZmZddlmZmZddlmZmZdd lmZmZe rdd lmZmZdd lmZed Z e d Z!Gdde"Z#Gdde#Z$Gdde#Z% d#dZ&edGdde e Z'edGdde e Z(edGdde e Z)eGdde e Z*eGdd e e Z+Gd!d"e e Z,e(e e)e ze,e zZ-y)$z)Types for setting agent response formats.) annotationsN) dataclass is_dataclass) UnionType) TYPE_CHECKINGAnyGenericLiteralTypeVarUnionget_args get_origin)BaseToolStructuredTool) BaseModel TypeAdapter)Self is_typeddict)CallableIterable) AIMessageSchemaT)pydanticr typeddict json_schemaceZdZUdZded<y)StructuredOutputErrorz(Base class for structured output errors.r ai_messageN)__name__ __module__ __qualname____doc____annotations__l/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/langchain/agents/structured_output.pyrr"s 2r%rc$eZdZdZdfd ZxZS)MultipleStructuredOutputsErrorzZRaised when model returns multiple structured output tool calls when only one is expected.cf||_||_t| ddj |dy)zInitialize `MultipleStructuredOutputsError`. Args: tool_names: The names of the tools called for structured output. ai_message: The AI message that contained the invalid multiple tool calls. z:Model incorrectly returned multiple structured responses (z, z) when only one is expected.N) tool_namesrsuper__init__join)selfr*r __class__s r&r,z'MultipleStructuredOutputsError.__init__+s=%$   *%&&B D r%)r*z list[str]rrreturnNonerr r!r"r, __classcell__r/s@r&r(r((sd   r%r(c$eZdZdZdfd ZxZS)StructuredOutputValidationErrorzXRaised when structured output tool call arguments fail to parse according to the schema.c\||_||_||_t|d|d|dy)aInitialize `StructuredOutputValidationError`. Args: tool_name: The name of the tool that failed. source: The exception that occurred. ai_message: The AI message that contained the invalid structured output. z,Failed to parse structured output for tool 'z': .N) tool_namesourcerr+r,)r.r9r:rr/s r&r,z(StructuredOutputValidationError.__init__>s;# $ G {RUV\U]]^_`r%)r9strr: Exceptionrrr0r1r2r4s@r&r6r6;sb a ar%r6c|dk(r|S t|}|j|S#t$r/}t|dt |}d|d|}t ||d}~wwxYw)aParse data using for any supported schema type. Args: schema: The schema type (Pydantic model, `dataclass`, or `TypedDict`) schema_kind: One of `"pydantic"`, `"dataclass"`, `"typeddict"`, or `"json_schema"` data: The data to parse Returns: The parsed instance according to the schema type Raises: ValueError: If parsing fails rrzFailed to parse data to z: N)rvalidate_pythonr<getattrr; ValueError)schema schema_kinddataadaptere schema_namemsgs r&_parse_with_schemarHLsq"m# %(3F(;&&t,, %fj#f+> ( Rs;o1$%s% A*AAF)initceZdZUdZded< ded< ded< ded< d ed < d Zd ed < ddd d ddZy) _SchemaSpecz%Describes a structured output schema. type[SchemaT]rAr;name description SchemaKindrBdict[str, Any]rFboolstrictN)rMrNrRc v||_|r||_nt|trBt |j ddt t jdd|_n>@D  & !*D *62>>@D ,DL>:de S/ !r%) rArLrM str | NonerNr`rRrQr0r1)rr r!r"r#rRr,r$r%r&rKrKhs/  I  1FD=  "& +"+" +"  +"  +" +"r%rKc`eZdZUdZded< ded< ded< ded < d d d  dd Zy ) ToolStrategyz0Use a tool calling strategy for model responses.rLrAzlist[_SchemaSpec[SchemaT]] schema_specsr`tool_message_contentWbool | str | type[Exception] | tuple[type[Exception], ...] | Callable[[Exception], str] handle_errorsNT)rdrfc||_||_||_dfd |Dcgc] }t|c}|_ycc}w)zInitialize `ToolStrategy`. Initialize `ToolStrategy` with schemas, tool message content, and error handling strategy. c3Kt|ttfvr!t|D]}|Ed{yt |t r,d|vr(|j dgD]}|Ed{y|y7H7w)z5Yield leaf variants from Union and JSON Schema oneOf.NoneOf)rrr r rXrYrZ)rAargsub_iter_variantss r&rlz-ToolStrategy.__init__.._iter_variantss&!i%77#F+3C-c2223&$'Gv,=!::gr23C-c2223L3 3s!1BA=;B0A?1 B?BN)rArr0z Iterable[Any])rArdrfrKrc)r.rArdrfsrls @r&r,zToolStrategy.__init__sC  $8!* 6DF5KL[^LLsA)rArLrdr`rfrer0r1rr r!r"r#r,r$r%r&rbrbsl: $,,*$$- ` ,0 (,"M"M) "M % "M "Mr%rbcBeZdZUdZded< ded< d dZd dZy) ProviderStrategyz9Use the model provider's native structured output method.rLrA_SchemaSpec[SchemaT] schema_specc2||_t||_y)z(Initialize ProviderStrategy with schema.N)rArKrrr.rAs r&r,zProviderStrategy.__init__s  &v.r%cjd|jj|jjdd}d|iS)z@Convert to kwargs to bind to a model to force structured output.r)rMrA)r]rresponse_format)rrrMr)r.rvs r&to_model_kwargsz ProviderStrategy.to_model_kwargss> "((--**66 "?33r%NrArLr0r1)r0rP)rr r!r"r#r,rwr$r%r&rprps3C !%%&// / 4r%rpcPeZdZUdZded< ded< ded< ed dZd d Zy ) OutputToolBindinga*Information for tracking structured output tool metadata. This contains all necessary information to handle structured responses generated via tool calls, including the original schema, its type classification, and the corresponding tool implementation used by the tools strategy. rLrArOrBrtoolc ||j|jt|j|j|j S)zCreate an `OutputToolBinding` instance from a `SchemaSpec`. Args: schema_spec: The `SchemaSpec` to convert Returns: An `OutputToolBinding` instance with the appropriate tool created ) args_schemarMrN)rArBr{)rArBrrrMrNclsrrs r&from_schema_specz"OutputToolBinding.from_schema_spec(sF%%#//'33 %%'33  r%cDt|j|j|S)aParse tool arguments according to the schema. Args: tool_args: The arguments from the tool call Returns: The parsed response according to the schema type Raises: ValueError: If parsing fails )rHrArB)r. tool_argss r&parsezOutputToolBinding.parse<s"$++t/?/?KKr%Nrrrqr0r)rrPr0r)rr r!r"r# classmethodrrr$r%r&rzrzs@ DM NL  & Lr%rzcLeZdZUdZded< ded< ed dZd dZd dZy ) ProviderStrategyBindinga Information for tracking native structured output metadata. This contains all necessary information to handle structured responses generated via native provider output, including the original schema, its type classification, and parsing logic for provider-enforced JSON. rLrArOrBc>||j|jS)zCreate a `ProviderStrategyBinding` instance from a `SchemaSpec`. Args: schema_spec: The `SchemaSpec` to convert Returns: A `ProviderStrategyBinding` instance for parsing native structured output rArBrr~s r&rz(ProviderStrategyBinding.from_schema_spec[s#%%#//  r%c |j|}ddl} |j|}t|j |j|S#t$r1}t |j dd}d|d|d}t ||d}~wwxYw)a@Parse `AIMessage` content according to the schema. Args: response: The `AIMessage` containing the structured output Returns: The parsed response according to the schema Raises: ValueError: If text extraction, JSON parsing or schema validation fails rNrrvz1Native structured output expected valid JSON for z, but parsing failed: r8) "_extract_text_content_from_messagejsonloadsr<r?rAr@rHrB)r.responseraw_textrrCrErFrGs r&rzProviderStrategyBinding.parsejs::8D )::h'D"$++t/?/?FF )!$++z;LMKCK=Q''(c, S/q (  )sA B,A>>Bc|j}t|tr|St|trg}|D]}t|trd|j ddk(r"d|vr|j t|dId|vsNt|dtsb|j |dw|j t|dj|St|S)zExtract text content from an AIMessage. Args: message: The AI message to extract text from Returns: The extracted text content r]textcontentrW)rrXr;listrYrZappendr-)r.messagerpartscs r&rz:ProviderStrategyBinding._extract_text_content_from_messages// gs #N gt $!E )a&uuV}.6Q; S6^4"aJq|S,I Qy\2LLQ( )775> !7|r%Nr)rrr0r)rrr0r;) rr r!r"r#rrrrr$r%r&rrKs= HM    G<r%rc.eZdZUdZded< ddZy) AutoStrategyz=Automatically select the best strategy for structured output.rLrAc||_y)z$Initialize AutoStrategy with schema.N)rArts r&r,zAutoStrategy.__init__s  r%Nrxrnr$r%r&rrs%G $ r%r)rAztype[SchemaT] | dictrBrOrCrPr0r).r" __future__rr[ dataclassesrrtypesrtypingrrr r r r r rlangchain_core.toolsrrrrrtyping_extensionsrrcollections.abcrrlangchain_core.messagesrrrOr<rr(r6rHrKrbrprzrrResponseFormatr$r%r&rs/" /   :+021 )  H I I  %: &a&;a"% %/9%AO%%8 H"''"H"H"V =M77#=M=M@ 4ww'44> 2L(2L 2Lj Sgg.S Sl 77# g&)9')BB\RYEZZr%