o
i% @ s d dl mZ d dlmZmZmZ d dlmZm Z m
Z
m Z m Z d dl
mZmZmZ d dlmZ d dlmZmZ d dlmZ d dlmZ d Zd d
lm Z m Z
d$d%dd Z e G d d d Z d&d"d#Z d S )' ) annotations)IterableIteratorMapping)asdict dataclassfieldfields is_dataclass)datetime timedelta)Enum)AnyLiteral) ElementTree) BaseModel)
format_as_xml)ComputedFieldInfo FieldInfoNitemnull Fobjr root_tag
str | Noneitem_tagstrnone_strindentinclude_field_infoLiteral['once'] | boolreturnc C sl t | |||d|}|du r$|jdu r$|du rdnd}|t||S |dur/tj||d tj|ddS )a= Format a Python object as XML.
This is useful since LLMs often find it easier to read semi-structured data (e.g. examples) as XML,
rather than JSON etc.
Supports: `str`, `bytes`, `bytearray`, `bool`, `int`, `float`, `date`, `datetime`, `time`, `timedelta`, `Enum`,
`Mapping`, `Iterable`, `dataclass`, and `BaseModel`.
Args:
obj: Python Object to serialize to XML.
root_tag: Outer tag to wrap the XML in, use `None` to omit the outer tag.
item_tag: Tag to use for each item in an iterable (e.g. list), this is overridden by the class name
for dataclasses and Pydantic models.
none_str: String to use for `None` values.
indent: Indentation string to use for pretty printing.
include_field_info: Whether to include attributes like Pydantic `Field` attributes and dataclasses `field()`
`metadata` as XML attributes. In both cases the allowed `Field` attributes and `field()` metadata keys are
`title` and `description`. If a field is repeated in the data (e.g. in a list) by setting `once`
the attributes are included only in the first occurrence of an XML element relative to the same field.
Returns:
XML representation of the object.
Example:
```python {title="format_as_xml_example.py" lint="skip"}
from pydantic_ai import format_as_xml
print(format_as_xml({'name': 'John', 'height': 6, 'weight': 200}, root_tag='user'))
'''
John
6
200
'''
```
)datar r r N
spaceunicodeencoding)_ToXmlto_xmltextjoin_rootless_xml_elementsr r tostring)r r r r r r elr. r2 _/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/format_prompt.pyr s -r c @ s e Zd ZU ded< ded< ded< ded< eedZd ed
< eedZd ed < eedZ d
ed< dZ
ded< dZ d/d0ddZ d/d1d d Z
d2d d Zd d! Z "d3d4d'd(Z "d3d5d)d*Zed6d-d.ZdS )7r+ r r# r r r r! r )default_factoryz4dict[str, tuple[str, FieldInfo | ComputedFieldInfo]] _fields_infozset[str]_included_fieldsdict[str, str]_element_namesFbool_is_info_extracted)title descriptionNtagr r" ElementTree.Elementc C s | j | jd|dS )Nr$ valuepathr= )_to_xmlr# )selfr= r2 r2 r3 r, ^ s z
_ToXml.to_xmlr@ rA c C s | |d u r | jn||}|d u r| j|_|S t|tr!||_|S t|ttB r1|jdd|_|S t|t t
B t B t B rCt||_|S t|t
tB rQ| |_|S t|tr]t||_|S t|trz|d u rq|| jv rq| j| |_| ||| |S t|rt|ts| |d u r|jj|_| |t|| |S t|tr| |d u r|jj|_| || | |S t|t rt |D ] \}}| | j ||r| d| dnd| dd q|S t!dt| )Nignore)errors.[][)r@ rA z%Unsupported type for XML formatting: )"_create_elementr r r-
isinstancer bytes bytearraydecoder9 intfloatr r r isoformatr
r r8 r= _mapping_to_xmlr
type_init_structure_info __class____name__r r
model_dumpr enumerateappendrB TypeError)rC r@ rA r= elementnr r2 r2 r3 rB a sR
2z_ToXml._to_xmlc C st t |}|| jv r8| j| \}}| jr| jdks || jvr8| |}| D ]
\}}||| q'| j| |S )Nonce) r Elementr5 r r6 _extract_attributesitemssetadd) rC r= rA rZ
field_repr
field_infofield_attributeskvr2 r2 r3 rI s
z_ToXml._create_elementc C s | j s| | j d| _ dS dS )z\Create maps with all data information (fields info and class names), if not already created.TN)r: _parse_data_structuresr# rC r2 r2 r3 rS s
z_ToXml._init_structure_infor$ rZ mappingMapping[Any, Any]Nonec C sp | D ]1\}}t|trt|}nt|ts!tdt| d|| j||r/| d| n||d qd S )Nz)Unsupported key type for XML formatting: z , only str and int are allowed.r? )r_ rJ rN r rY rR rX rB )rC rZ ri rA keyr@ r2 r2 r3 rQ s
*z_ToXml._mapping_to_xmlc
s |du st |ttB tB tB tB tB tB tB t B rdS t |t
r;| D ]\}} ||r4| d| n| q$dS t
|rt |ts|jj j|< t|D ]C}|r[| d|j n|j} jr|jr fdd|j D }|r|jj d|j }|tdi |f j|< t||j| qOdS t |tr|jj j|< |jj|jjfD ]:} | D ]3\}}
|r| d| n|} jrt |
t s|
j s|jj d| }||
f j|< t||| qqdS t |t r t |D ] \} } |r| d| dnd| d} | | qdS dS ) z`Parse data structures as dataclasses or Pydantic models to extract element names and attributes.Nrl c s i | ] \}}| j v r||qS r2 _FIELD_ATTRIBUTES).0re rf rh r2 r3
s z1_ToXml._parse_data_structures..rF rG rH r2 ) rJ r rN rO r r r
rL rK r9 r r_ rg r
rR rT rU r8 r namer metadatar r5 getattrr model_fieldsmodel_computed_fieldsr excluder rW )
rC r@ rA re rf r new_path
attributesrb ru infor[ r r2 rh r3 rg sD 2
"
z _ToXml._parse_data_structuresrz FieldInfo | ComputedFieldInfoc s fdd| j D S )Nc s* i | ]}t |d d ur|tqS N)rt r )rp attrrz r@ r2 r3 rq s * z._ToXml._extract_attributes..rn )clsrz r2 r~ r3 r^ s z_ToXml._extract_attributesr| )r= r r" r> )r@ r rA r r= r r" r> )r= r rA r r" r> )r$ )rZ r> ri rj rA r r" rk )r@ r rA r )rz r{ r" r7 )rU
__module__ __qualname____annotations__r dictr5 r` r6 r8 r: ro r, rB rI rS rQ rg classmethodr^ r2 r2 r2 r3 r+ M s(
%
#r+ rootr>
Iterator[str]c c s6 | D ]}|d urt j||d t j|ddV qd S )Nr& r( r) )r r r0 )r r sub_elementr2 r2 r3 r/ s r/ )Nr r r F)r r r r r r r r r r r r! r" r )r r> r r r" r )!
__future__r _annotationscollections.abcr r r dataclassesr r r r r
datetimer r r
enumr typingr r xml.etreer pydanticr __all__pydantic.fieldsr r r r+ r/ r2 r2 r2 r3 s& <~