o i@sZddlmZddlZddlmZddlmZmZddl m Z dd d Z GdddeZ dS)) annotationsN) UserError) JsonSchemaJsonSchemaTransformer) ModelProfile model_namestrreturnModelProfile | NonecCs d|v}tt|| | | dS)z)Get the model profile for a Google model.image)json_schema_transformersupports_image_outputsupports_json_schema_outputsupports_json_object_outputsupports_tools)rGoogleJsonSchemaTransformer)r is_image_modelra/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/profiles/google.pygoogle_model_profile srcs2eZdZdZddd fdd Zdd d ZZS)ruwTransforms the JSON Schema from Pydantic to be suitable for Gemini. Gemini which [supports](https://ai.google.dev/gemini-api/docs/function-calling#function_declarations) a subset of OpenAPI v3.0.3. Specifically: * gemini doesn't allow the `title` keyword to be set * gemini doesn't allow `$defs` — we need to inline the definitions where possible N)strictschemarr bool | Nonecstj||ddddS)NT)rprefer_inlined_defssimplify_nullable_unions)super__init__)selfrr __class__rrr"sz$GoogleJsonSchemaTransformer.__init__r c Cs|dd}|ri|d|i}td|jd|dt|dd|dd|dd}dur9|g|d<|d d|d d|d d|d d|d}red |d<dd|D|d<|d}d|vryd|vry|d|d<|d kr|dd}r|d}|r|d|d|d<nd||d<d|vrtd|dd|vr|d} |d} | dur| gng} | D] } | | vr| | qt| dkrd| i|d<n t| dkr| d|d<| dt| | dur| dt| |S) NadditionalPropertieszn`additionalProperties` is not supported by Gemini; it will be removed from the tool JSON schema. Full schema: z9 Source of additionalProperties within the full schema: z If this came from a field with a type like `dict[str, MyType]`, that field will always be empty. If Google's APIs are updated to support this properly, please create an issue on the Pydantic AI GitHub and we will fix this behavior.titlez$schemaconstenum discriminatorexamplesexclusiveMaximumexclusiveMinimumstringtypecSsg|]}t|qSr)r ).0valrrr Gsz9GoogleJsonSchemaTransformer.transform..oneOfanyOfformat descriptionz (format: )zFormat: z$refz>Recursive `$ref`s in JSON Schema are not supported by Gemini: prefixItemsitemsrrminItemsmaxItems) popwarningswarnr UserWarninggetrappendlen setdefault) rradditional_propertiesoriginal_schemar$r%type_fmtr2 prefix_itemsr5 unique_itemsitemrrr transform%sd              z%GoogleJsonSchemaTransformer.transform)rrrr)rrr r)__name__ __module__ __qualname____doc__rrG __classcell__rrr rrs r)r r r r ) __future__r _annotationsr9pydantic_ai.exceptionsr _json_schemarrrrrrrrrs