o i-@sdZddlmZddlmZmZddlmZm Z ddl m Z m Z ddl mZmZmZmZmZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd lm Z dd l!m"Z"m#Z#ddl$m%Z%m&Z&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-m.Z.m/Z/m0Z0erddl1m2Z2m3Z3dZ4eddGdddZ5   d?d@d'd(Z6e%d)Z7e'd*Z8eee+ee7fe8fZ9ee7e8fZ:e9e7e8fe:e7e8fBZ;dAd.d/ZZ>dS)DzUsed to build pydantic validators and JSON schemas from functions. This module has to use numerous internal Pydantic APIs and is therefore brittle to changes in Pydantic. ) annotations) AwaitableCallable) dataclassfield) Parameter signature) TYPE_CHECKINGAny Concatenatecast get_origin) ConfigDict) _decorators_generate_schema _typing_extra) ConfigWrapper) FieldInfo)GenerateJsonSchema)create_schema_validator)SchemaValidator core_schema) ParamSpecTypeIsTypeVar)doc_descriptions) RunContext)check_object_json_schemais_async_callable is_model_likerun_in_executor)DocstringFormatObjectJsonSchema)function_schemaT)kw_onlyc@seZdZUdZded<ded<ded<ded <d ed <d ed <d Zded<eedZded<d Z ded<dddZ dddZ d S) FunctionSchemaz-Internal information about a function schema.Callable[..., Any]functionz str | None descriptionr validatorr# json_schemabool takes_ctxis_asyncNsingle_arg_name)default_factoryz list[str]positional_fieldsvar_positional_field args_dictdict[str, Any]ctxRunContext[Any]returnr csx|||\}}|jr#tttgttf|j}||i|IdHStttgtf|j}t|g|Ri|IdHSN) _call_argsr.r rr rstrr(r!)selfr3r5argskwargsr(r>b/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/_function_schema.pycall0szFunctionSchema.call tuple[list[Any], dict[str, Any]]cCs\|jr|j|i}|jr|gng}|jD] }|||q|jr*|||j||fSr8)r/r-r1appendpopr2extend)r;r3r5r<positional_fieldr>r>r?r99s  zFunctionSchema._call_args)r3r4r5r6r7r )r3r4r5r6r7rA) __name__ __module__ __qualname____doc____annotations__r/rlistr1r2r@r9r>r>r>r?r&!s     r&NautoFr(r'schema_generatortype[GenerateJsonSchema]r- bool | Nonedocstring_formatr"require_parameter_descriptionsr,r7c# s\|durt|}t|jdd}t|}t|}g}zt|Wnty?} z|t | tddWYd} ~ nd} ~ wwt |} d} i} g} d}t }t||d\}}|r|rqtfddjD}|t|}n tjt|}|r|d d |tjD]\}\}}|jjur|r|d krqt}n.| |}|d kr|rt|s|d q|st|r|d q|d krt|r|dq|j}|jtjkr||} q|jtjkrt|}|jtju}t t!t|}|rt"#|}nt"$||j}|j%dur|&||_%|j'||||d| |<}t(||)did<|jtj*kr8| |q|jtjkrA|}q|r[ddl+m,}d|}|d|j-d||.d}| rednd|d<t/| | ||\}} |0|}t1|||j2|j-d||j3}!t t4|!}!|5|}"| dur|s|"6dd}t7||!t8|"| | ||t9||d S)aBuild a Pydantic validator and JSON schema from a tool function. Args: function: The function to build a validator and JSON schema for. takes_ctx: Whether the function takes a `RunContext` first argument. docstring_format: The docstring format to use. require_parameter_descriptions: Whether to require descriptions for all tool function parameters. schema_generator: The JSON schema generator class to use. Returns: A `FunctionSchema` instance. NT)titleuse_attribute_docstringscSsdSr8r>r>r>r>r?isz!function_schema..)rPc3s$|] }tj|js|VqdSr8) _is_call_ctx parameters annotation).0namesigr>r? ws  z"function_schema..z#Missing parameter descriptions for z, rzQFirst parameter of tools that take context must be annotated with RunContext[...]zDRunContext annotations can only be used with tools that take contextz=RunContext annotations can only be used as the first argument)requiredmetadatar r) UserErrorz zError generating schema for z: allowforbidextra_fields_behavior validate_callr)) r)r*r+r/r1r2r-r.r(): _takes_ctxrrFrrGenerateSchemar ValueErrorrBr:rget_function_type_hintsrDecoratorInfosrsetrVjoin enumerateitemsrWemptyr rUrYkindr VAR_KEYWORDgenerate_schemaVAR_POSITIONALrKdefaultr typerfrom_annotationfrom_annotated_attributer)get_generate_td_field_schemar setdefaultPOSITIONAL_ONLY exceptionsr_rH core_config _build_schema clean_schemarrGplugin_settingsrgeneraterCr&rr)#r(rMr-rPrQconfigconfig_wrapper gen_schemaerrorse type_hintsvar_kwargs_schemafieldsr1r2 decoratorsr)field_descriptionsparameters_without_ctxmissing_paramsindexrYprW field_namer] field_info td_schemar_ error_detailsr{schemar/schema_validatorr+r>rZr?r$Js                         r$PR callable_objTargetCallable[P, R]TypeIs[WithCtx[P, R]]cCszt|}Wn tyYdSwz tt|j}Wn ty&YdSwt|tj s>t t |dd}|dur<|}ndSt t|}||}|durQdSd|juoYt|S)zCheck if a callable takes a `RunContext` first argument. Args: callable_obj: The callable to check. Returns: `True` if the callable takes a `RunContext` as first argument, `False` otherwise. F__call__NT)rrfnextiterrVkeys StopIteration isinstancer_function_likegetattrrsrrgunwrap_wrapped_functionrvrmrU)rr[first_param_name call_funcrrWr>r>r?rds(     rdr%dict[str, core_schema.TypedDictField]rcore_schema.CoreSchema | Noner_generate_schema.GenerateSchemar{core_schema.CoreConfig)tuple[core_schema.CoreSchema, str | None]cCsft|dkr |dur tt|}||}|ddr |d|fStj|||r+||ndd}|dfS)apGenerate a typed dict schema for function parameters. Args: fields: The fields to generate a typed dict schema for. var_kwargs_schema: The variable keyword arguments schema. gen_schema: The `GenerateSchema` instance. core_config: The core configuration. Returns: tuple of (generated core schema, single arg name). rNr^r r)r extras_schema)lenrrrtyped_dict_schemarp)rrrr{rYtd_fieldrr>r>r?r|s   r|rWr cCs|tup t|tuS)zNReturn whether the annotation is the `RunContext` class, parameterized or not.)rr )rWr>r>r?rU-srU)NrLF) r(r'rMrNr-rOrPr"rQr,r7r&)rrr7r) rrrrrrr{rr7r)rWr r7r,)?rI __future__r _annotationscollections.abcrr dataclassesrrinspectrrtypingr r r r r pydanticrpydantic._internalrrrpydantic._internal._configrpydantic.fieldsrpydantic.json_schemar!pydantic.plugin._schema_validatorr pydantic_corerrtyping_extensionsrrr_griffer _run_contextr_utilsrrr r!toolsr"r#__all__r&r$rrWithCtx WithoutCtxTargetCallablerdr|rUr>r>r>r?sF        +    !