tj;dZddlmZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZmZmZddlmZmZddlmZmZdd lmZmZe rdd lmZmZdd lm Z ed Z!ed Z"Gdde#Z$Gdde$Z%Gdde$Z&d+dZ'edGdd e e!Z(edGd!d"e e!Z)edGd#d$e e!Z*eGd%d&e e!Z+eGd'd(e e!Z,Gd)d*e e!Z-e)e!e*e!ze-e!zZ.dS),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<dS)StructuredOutputErrorz(Base class for structured output errors.r ai_messageN)__name__ __module__ __qualname____doc____annotations__`/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain/agents/structured_output.pyrr#s$22r%rc$eZdZdZd fd ZxZS) MultipleStructuredOutputsErrorzZRaised when model returns multiple structured output tool calls when only one is expected. tool_names list[str]rrreturnNonec||_||_tdd|ddS)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)r)rsuper__init__join)selfr)r __class__s r&r/z'MultipleStructuredOutputsError.__init__,s`%$  D *%% D D D     r%)r)r*rrr+r,rr r!r"r/ __classcell__r2s@r&r(r()sCdd           r%r(c$eZdZdZd fd ZxZS) StructuredOutputValidationErrorzXRaised when structured output tool call arguments fail to parse according to the schema. tool_namestrsource Exceptionrrr+r,c||_||_||_td|d|ddS)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)r8r:rr.r/)r1r8r:rr2s r&r/z(StructuredOutputValidationError.__init__?sM# $ _ __V\___`````r%)r8r9r:r;rrr+r,r3r5s@r&r7r7<sMbb a a a a a a a a a ar%r7schematype[SchemaT] | dict[str, Any] schema_kind SchemaKinddatadict[str, Any]r+SchemaT | dict[str, Any]c|dkr|S tt|}||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)rrvalidate_pythonr;getattrr9 ValueError)r>r@rBadaptere schema_namemsgs r&_parse_with_schemarMMs"m## %g&v..&&t,,, %%%fj#f++>> ;;;;;oo1$%s.9 A>6A99A>F)initcreZdZUdZded< ded< ded< ded< d ed < d Zd ed < d d d dddZd S) _SchemaSpecz%Describes a structured output schema.r?r>r9name descriptionrAr@rCrN bool | Nonestrict)rQrRrT str | Noner+r,c @||_|r||_nt|trRt |ddt t jdd|_nLt t|ddt t jdd|_|p=t|tr|ddnt|ddpd|_ ||_ t|trd |_ ||_ dSt|tr7t|tr"d |_ ||_ dSt#|r/d |_ t%| |_ dSt'|r/d |_ t%| |_ dSd t|d}t)|)a~Initialize `SchemaSpec` with schema and optional parameters. Args: schema: Schema to describe. name: Optional name for the schema. description: Optional description for the schema. strict: Whether to enforce strict validation of the schema. Raises: ValueError: If the schema type is unsupported. titleresponse_format_NrrRr"rrrrzUnsupported schema type: zS. Supported types: Pydantic models, dataclasses, TypedDicts, and JSON schema dicts.)r>rQ isinstancedictr9getuuiduuid4rGrRrTr@rtype issubclassrmodel_json_schemarrrrH)r1r>rQrRrTrLs r&r/z_SchemaSpec.__init__s&  eDII  % % eFJJw0Z3tz||CTCTUWVWUWCX0Z0Z[[\\DIIGFJ8b3tz||K\K\]_^_]_K`8b8bccddDI& &$'' 8FJJ}b ) ) )D117R   fd # # ",D %D     % % "*VY*G*G ")D %7799D    & ! ! "*D *622>>@@D    & ! ! "*D *622>>@@D   eDLLeee S// !r%) r>r?rQrUrRrUrTrSr+r,)rr r!r"r#rTr/r$r%r&rPrPjs//****III 1F=  "&" 5"5"5"5"5"5"5"5"r%rPcTeZdZUdZded< ded< ded< ded < d d d ddZd S) ToolStrategyz0Use a tool calling strategy for model responses.*type[SchemaT] | UnionType | dict[str, Any]r>zlist[_SchemaSpec[Any]] schema_specsrUtool_message_contentWbool | str | type[Exception] | tuple[type[Exception], ...] | Callable[[Exception], str] handle_errorsNT)rgrir+r,cr||_||_||_dfd d|D|_dS) zInitialize `ToolStrategy`. Initialize `ToolStrategy` with schemas, tool message content, and error handling strategy. r>rr+ Iterable[Any]c3"Kt|tthvr%t|D]}|Ed{VdSt |t r0d|vr,|dgD]}|Ed{VdS|VdS)z5Yield leaf variants from Union and JSON Schema oneOf.NoneOf)rrr r r[r\r])r>argsub_iter_variantss r&rpz-ToolStrategy.__init__.._iter_variantss&!!i%777#F++33C-~c2222222222&$'' Gv,=,=!::gr2233C-~c2222222222LLLLLr%c,g|]}t|Sr$)rP).0ss r& z)ToolStrategy.__init__..sLLL[^^LLLr%N)r>rr+rk)r>rgrirf)r1r>rgrirps @r&r/zToolStrategy.__init__sc  $8!*      ML^^F5K5KLLLr%)r>rergrUrirhr+r,rr r!r"r#r/r$r%r&rdrds::6666$((((*$$$$ ,0 (,"M"M"M"M"M"M"M"Mr%rdcBeZdZUdZded< ded< dddd ZddZdS)ProviderStrategyz9Use the model provider's native structured output method.r?r>_SchemaSpec[SchemaT] schema_specNrTrTrSr+r,c@||_t|||_dS)zInitialize `ProviderStrategy` with schema. Args: schema: Schema to enforce via the provider's native structured output. strict: Whether to request strict provider-side schema enforcement. rzN)r>rPry)r1r>rTs r&r/zProviderStrategy.__init__s& &vf===r%rCch|jj|jjd}|jjrd|d<d|d}d|iS)zConvert to kwargs to bind to a model to force structured output. Returns: The kwargs to bind to a model. )rQr>TrTr)r`rresponse_format)ryrQrrT)r1rr}s r&to_model_kwargsz ProviderStrategy.to_model_kwargss]$)&2' '    " )$(K !"&+ + "?33r%)r>r?rTrSr+r,)r+rC)rr r!r"r#r/r~r$r%r&rwrwsqCC****!%%%%& # > > > > > >444444r%rwcXeZdZUdZded< ded< ded< edd ZddZdS)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. r?r>rAr@rtoolryrxr+rc p||j|jt|j|j|jS)zCreate an `OutputToolBinding` instance from a `SchemaSpec`. Args: schema_spec: The `SchemaSpec` to convert Returns: An `OutputToolBinding` instance with the appropriate tool created ) args_schemarQrR)r>r@r)r>r@rrrQrRclsrys r&from_schema_specz"OutputToolBinding.from_schema_specHsMs%#/'3 %'3    r% tool_argsrCrDc8t|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 )rMr>r@)r1rs r&parsezOutputToolBinding.parse\s"$+t/?KKKr%Nryrxr+r)rrCr+rD)rr r!r"r# classmethodrrr$r%r&rr4s+***MNNNL   [ & L L L L L Lr%rcdeZdZUdZded< ded< edd ZddZeddZ dS)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. r?r>rAr@ryrxr+rc0||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 r>r@rrs r&rz(ProviderStrategyBinding.from_schema_spec|s)s%#/    r%responserrDc||} tj|}nA#t$r4}t |jdd}d|d|d}t ||d}~wwxYwt|j|j|S)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 rr}z1Native structured output expected valid JSON for z, but parsing failed: r=N) "_extract_text_content_from_messagejsonloadsr;rGr>rHrMr@)r1rraw_textrBrJrKrLs r&rzProviderStrategyBinding.parses::8DD ):h''DD ) ) )!$+z;LMMK,K,,'(,,, S//q (  )"$+t/?FFFs, A*/A%%A*messager9c|j}t|tr|Sg}|D]}t|tr|ddkr-d|vr)|t|d]d|vr6t|dtr||d|t|d|S)zExtract text content from an `AIMessage`. Args: message: The AI message to extract text from Returns: The extracted text content r`textcontentrZ)rr[r9r\r]appendr0)rrpartscs r&rz:ProviderStrategyBinding._extract_text_content_from_messages/ gs # # N % %A!T"" %55==F**v{{LLQvY0000!^^ 1Y<(E(E^LL9... SVV$$$$wwu~~r%Nr)rrr+rD)rrr+r9) rr r!r"r#rrr staticmethodrr$r%r&rrks+***M    [  GGGG8\r%rc(eZdZUdZded< ddZdS) AutoStrategyz=Automatically select the best strategy for structured output.r?r>r+r,c||_dS)z&Initialize `AutoStrategy` with schema.N)r>)r1r>s r&r/zAutoStrategy.__init__s  r%N)r>r?r+r,rur$r%r&rrs?GG****$r%r)r>r?r@rArBrCr+rD)/r" __future__rrr^ dataclassesrrtypesrtypingrrr r r r r rlangchain_core.toolsrrrrrtyping_extensionsrrcollections.abcrrlangchain_core.messagesrrrAr;rr(r7rMrPrdrwrrrResponseFormatr$r%r&rs//"""""" ////////                    :9999999++++++++00000000222222222111111 ')   H I I      %:   &aaaaa&;aaa"%%%%: U"U"U"U"U"''"U"U"U"p >M>M>M>M>M77#>M>M>MB +4+4+4+4+4ww'+4+4+4\ 3L3L3L3L3L(3L3L 3Ll QQQQQgg.QQ Qh     77#   g&)9')BB\RYEZZ>>r%