o
    ¨ir  ã                   @  sö   U d Z ddlmZ ddlmZmZ ddlmZ ddlm	Z	m
Z
 G dd„ de
ƒZG dd	„ d	e
ƒZG d
d„ de
ƒZG dd„ de
ƒZG dd„ de
ƒZG dd„ de
ƒZdZded< ed ZG dd„ de
ƒZee Zded< G dd„ deƒZee Zded< dS )zÌType definitions of OpenTelemetry GenAI spec message parts.

Based on https://github.com/lmolkova/semantic-conventions/blob/eccd1f806e426a32c98271c3ce77585492d26de2/docs/gen-ai/non-normative/models.ipynb
é    )Úannotations)ÚLiteralÚ	TypeAlias)Ú	JsonValue)ÚNotRequiredÚ	TypedDictc                   @  ó   e Zd ZU ded< ded< dS )ÚTextPartzLiteral['text']ÚtypeúNotRequired[str]ÚcontentN©Ú__name__Ú
__module__Ú__qualname__Ú__annotations__© r   r   ú`/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/_otel_messages.pyr	      ó   
 r	   c                   @  ó6   e Zd ZU ded< ded< ded< ded< ded	< d
S )ÚToolCallPartzLiteral['tool_call']r
   ÚstrÚidÚnameúNotRequired[JsonValue]Ú	argumentsúNotRequired[bool]ÚbuiltinNr   r   r   r   r   r      ó   
 r   c                   @  r   )ÚToolCallResponsePartzLiteral['tool_call_response']r
   r   r   r   r   Úresultr   r   Nr   r   r   r   r   r      r   r   c                   @  r   )ÚMediaUrlPartz>Literal['image-url', 'audio-url', 'video-url', 'document-url']r
   r   ÚurlNr   r   r   r   r   r!   #   r   r!   c                   @  s&   e Zd ZU ded< ded< ded< dS )ÚBinaryDataPartzLiteral['binary']r
   r   Ú
media_typer   r   Nr   r   r   r   r   r#   (   s   
 r#   c                   @  r   )ÚThinkingPartzLiteral['thinking']r
   r   r   Nr   r   r   r   r   r%   .   r   r%   z]TextPart | ToolCallPart | ToolCallResponsePart | MediaUrlPart | BinaryDataPart | ThinkingPartr   ÚMessagePart)ÚsystemÚuserÚ	assistantc                   @  r   )ÚChatMessageÚRoleÚrolezlist[MessagePart]ÚpartsNr   r   r   r   r   r*   9   r   r*   ÚInputMessagesc                   @  s   e Zd ZU ded< dS )ÚOutputMessager   Úfinish_reasonNr   r   r   r   r   r/   A   s   
 r/   ÚOutputMessagesN)Ú__doc__Ú
__future__r   Útypingr   r   Úpydanticr   Útyping_extensionsr   r   r	   r   r   r!   r#   r%   r&   r   r+   r*   Úlistr.   r/   r1   r   r   r   r   Ú<module>   s"    