=i7DSrSSKJr SSKrSSKJrJr SSKJr SSK J r J r J r J r JrJrJrJr SSKJrJr SSKJrJr SS KJrJr \ (aSS KJrJr SS KJr \"S 5r \ S r!"SS\"5r#"SS\#5r$"SS\#5r%S#Sjr&\"SS9"SS\ \ 55r'\"SS9"SS\ \ 55r(\"SS9"SS\ \ 55r)\"SS\ \ 55r*\"SS \ \ 55r+"S!S"\ \ 5r,\(\ \)\ -\,\ -r-g)$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_schemac$\rSrSr%SrS\S'Srg)StructuredOutputError"z(Base class for structured output errors.r ai_messageN)__name__ __module__ __qualname____firstlineno____doc____annotations____static_attributes__r r/home/dmtnaga/Documents/work/airagagent/rag_env/lib/python3.13/site-packages/langchain/agents/structured_output.pyrr"s 2r(rc0^\rSrSrSrSU4SjjrSrU=r$)MultipleStructuredOutputsError(zZRaised when model returns multiple structured output tool calls when only one is expected.cb>XlX l[TU] SSR U5S35 g)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)r0'MultipleStructuredOutputsError.__init__+s9%$   *%&&B D r()rr.)r.z list[str]rrreturnNoner!r"r#r$r%r0r' __classcell__r3s@r)r+r+(sd   r(r+c0^\rSrSrSrSU4SjjrSrU=r$)StructuredOutputValidationError;zXRaised when structured output tool call arguments fail to parse according to the schema.cV>XlX lX0l[TU]SUSUS35 g)zInitialize `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/r0)r2r?r@rr3s r)r0(StructuredOutputValidationError.__init__>s5# $ G {RUV\U]]^_`r()rr@r?)r?strr@ Exceptionrrr5r6r7r9s@r)r;r;;sb a ar(r;cUS:XaU$[U5nURU5$![a/n[US[ U55nSUSU3n[ U5UeSnAff=f)aeParse 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 rr!zFailed to parse data to z: N)rvalidate_pythonrCgetattrrB ValueError)schema schema_kinddataadaptere schema_namemsgs r)_parse_with_schemarOLsq"m# %(3F(;&&t,, %fj#f+> ( Rs;o1$%s& A*AAF)initc\rSrSr%SrS\S'S\S'S\S'S\S 'S \S 'S rS \S'SSS S.SSjjrSrg) _SchemaSpechz%Describes a structured output schema. type[SchemaT]rHrBname description SchemaKindrIdict[str, Any]rFboolstrictN)rUrVrZc XlU(aX lO[U[5(aC[ UR SS[ [ R"55SS355UlO=[ [USS[ [ R"55SS355UlU=(d= [U[5(aUR SS5O[USS5=(d SUl X@l [U[5(aS Ul Xl g[U[5(a2[U[5(aS Ul UR!5Ul g[#U5(a&S Ul [%U5R5Ul g['U5(a&S Ul [%U5R5Ul gS [U5S3n[)U5e)z:Initialize SchemaSpec with schema and optional parameters.titleresponse_format_Nr!rVr%rrrrzUnsupported schema type: zS. Supported types: Pydantic models, dataclasses, TypedDicts, and JSON schema dicts.)rHrU isinstancedictrBgetuuiduuid4rFrVrZrIrtype issubclassrmodel_json_schemarrrrG)r2rHrUrVrZrNs r)r0_SchemaSpec.__init__s I  % %FJJw2B3tzz|CTUWVWCXBY0Z[\DIGFJ:J3tzz|K\]_^_K`Ja8bcdDI& &$'' JJ}b )D17R   fd # #,D %   % %*VY*G*G)D %779D  & ! !*D *62>>@D  & ! !*D *62>>@D ,DL>:de S/ !r()rVrrUrHrIrZ) rHrTrU str | NonerVrirZrYr5r6) r!r"r#r$r%r&rZr0r'r r(r)rRrRhs/  I  1FD=  "& +"+" +"  +"  +" +"+"r(rRcl\rSrSr%SrS\S'S\S'S\S'S \S 'S S S .SSjjrSrg ) ToolStrategyz0Use a tool calling strategy for model responses.rTrHzlist[_SchemaSpec[SchemaT]] schema_specsritool_message_contentWbool | str | type[Exception] | tuple[type[Exception], ...] | Callable[[Exception], str] handle_errorsNT)rnrpc^XlX lX0lSU4SjjmT"U5Vs/sHn[U5PM snUlgs snf)zwInitialize `ToolStrategy`. Initialize `ToolStrategy` with schemas, tool message content, and error handling strategy. c3 ># [U5[[4;a#[U5HnT"U5ShvN M g[ U[ 5(a0SU;a*UR S/5HnT"U5ShvN M gUv gNSN7f)z5Yield leaf variants from Union and JSON Schema oneOf.NoneOf)rrr r r`rarb)rHargsub_iter_variantss r)rv-ToolStrategy.__init__.._iter_variantss&!i%77#F+C-c222,&$''Gv,=!::gr2C-c2223L3 3s"3BB AB;B <B BN)rHrr5z Iterable[Any])rHrnrprRrm)r2rHrnrpsrvs @r)r0ToolStrategy.__init__sC  $8!* 6DF5KL5K[^5KLLsA)rprHrmrn)rHrTrnrirpror5r6r!r"r#r$r%r&r0r'r r(r)rkrksr: $,,*$$- ` ,0 (,"M"M) "M % "M "M"Mr(rkcN\rSrSr%SrS\S'S\S'S SjrS SjrS rg ) ProviderStrategyz9Use the model provider's native structured output method.rTrH_SchemaSpec[SchemaT] schema_specc0Xl[U5Ulg)z(Initialize ProviderStrategy with schema.N)rHrRrr2rHs r)r0ProviderStrategy.__init__s  &v.r(cjSURRURRS.S.nSU0$)z@Convert to kwargs to bind to a model to force structured output.r)rUrH)rerresponse_format)rrUr)r2rs r)to_model_kwargs ProviderStrategy.to_model_kwargss> "((--**66 "?33r()rHrNrHrTr5r6)r5rX) r!r"r#r$r%r&r0rr'r r(r)r|r|s3C !%%&// / 4r(r|c\\rSrSr%SrS\S'S\S'S\S'\S S j5rSS jrS r g )OutputToolBindingiaInformation 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. rTrHrWrIrtoolc U"URUR[URURUR S9S9$)zCreate an `OutputToolBinding` instance from a `SchemaSpec`. Args: schema_spec: The `SchemaSpec` to convert Returns: An `OutputToolBinding` instance with the appropriate tool created ) args_schemarUrV)rHrIr)rHrIrrrUrVclsrs r)from_schema_spec"OutputToolBinding.from_schema_spec(sF%%#//'33 %%'33  r(cD[URURU5$)zParse 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 )rOrHrI)r2 tool_argss r)parseOutputToolBinding.parse<s"$++t/?/?KKr(r Nrr~r5r)rrXr5r) r!r"r#r$r%r& classmethodrrr'r r(r)rrs@ DM NL  & Lr(rcZ\rSrSr%SrS\S'S\S'\S Sj5rS SjrSS jr S r g )ProviderStrategyBindingiKaInformation 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. rTrHrWrIc8U"URURS9$)zCreate a `ProviderStrategyBinding` instance from a `SchemaSpec`. Args: schema_spec: The `SchemaSpec` to convert Returns: A `ProviderStrategyBinding` instance for parsing native structured output rHrIrrs r)r(ProviderStrategyBinding.from_schema_spec[s#%%#//  r(cURU5nSSKnURU5n[UR URU5$![a1n[ UR SS5nSUSUS3n[ U5UeSnAff=f)aParse `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 rNr!rz1Native structured output expected valid JSON for z, but parsing failed: r>) "_extract_text_content_from_messagejsonloadsrCrFrHrGrOrI)r2responseraw_textrrJrLrMrNs r)rProviderStrategyBinding.parsejs::8D )::h'D"$++t/?/?FF )!$++z;LMKCK=Q''(c, S/q (  )sA B,A??BcURn[U[5(aU$[U[5(a/nUHn[U[5(arUR S5S:Xa%SU;aUR [US55 MRSU;a0[US[5(aUR US5 MMMUR [U55 M SRU5$[U5$)zExtract text content from an AIMessage. Args: message: The AI message to extract text from Returns: The extracted text content retextcontentr_)rr`rBlistrarbappendr1)r2messagerpartscs r)r:ProviderStrategyBinding._extract_text_content_from_messages// gs # #N gt $ $!Ea&&uuV}.6Q; S6^4"aJq|S,I,I Qy\2-JLLQ(775> !7|r(r Nr)rrr5r)rrr5rB) r!r"r#r$r%r&rrrrr'r r(r)rrKs= HM    G<r(rc8\rSrSr%SrS\S'SSjrSrg) AutoStrategyiz=Automatically select the best strategy for structured output.rTrHcXlg)z$Initialize AutoStrategy with schema.NrHrs r)r0AutoStrategy.__init__s  r(rNrrzr r(r)rrs%G $ r(r)rHztype[SchemaT] | dictrIrWrJrXr5r).r% __future__rrc dataclassesrrtypesrtypingrrr r r r r rlangchain_core.toolsrrrrrtyping_extensionsrrcollections.abcrrlangchain_core.messagesrrrWrCrr+r;rOrRrkr|rrrResponseFormatr 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(