
    BThW                         d dl mZ d dlmZ d dlmZ d dlm	Z
 d dlmZ e G d de             Ze G d d	e             Ze G d
 de             Z	y)    )	dataclass)MessageModel)
UsageModel)TokenCounts)BaseDataclassc                   R    e Zd ZU eed<   dZeed<   edefd       Zede	fd       Z
y)	Messagecontentuserrolemessage_modelc                 >     | |j                   |j                        S )Nr   r
   r   )clsr   s     z/var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/pinecone_plugins/assistant/models/shared.pyfrom_openapizMessage.from_openapi   s!    ##!))
 	
    dc                 T     | |j                  dd      |j                  d            S )Nr   r   r
   r   getr   r   s     r   	from_dictzMessage.from_dict   s)    vv&EE)$
 	
r   N)__name__
__module____qualname__str__annotations__r   classmethodOpenAPIMessager   dictr    r   r   r	   r	   	   sE    LD#
 
 
 
$ 
 
r   r	   c                   X    e Zd ZU eed<   eed<   eed<   edefd       Zedefd       Z	y)	Usageprompt_tokenscompletion_tokenstotal_tokensr   c                 r     | |j                  d      |j                  d      |j                  d            S )Nr%   r&   r'   r%   r&   r'   r   r   s     r   r   zUsage.from_dict#   s4    %%0ee$78~.
 	
r   usage_modelc                 T     | |j                   |j                  |j                        S Nr)   r)   )r   r*   s     r   r   zUsage.from_openapi+   s*    %33);;$11
 	
r   N)
r   r   r   intr   r   r!   r   OpenAPIUsageModelr   r"   r   r   r$   r$      sI    
$ 
 
 
'8 
 
r   r$   c                   B    e Zd ZU eed<   eed<   eed<   edefd       Zy)r   r%   r&   r'   token_countsc                 T     | |j                   |j                  |j                        S r,   r)   )r   r0   s     r   r   zTokenCounts.from_openapi:   s*    &44*<<%22
 	
r   N)r   r   r   r-   r   r   OpenAPITokenCountsr   r"   r   r   r   r   4   s.    
(: 
 
r   r   N)dataclassesr   ?pinecone_plugins.assistant.data.core.client.model.message_modelr   r    =pinecone_plugins.assistant.data.core.client.model.usage_modelr   r.   Dpinecone_plugins.assistant.evaluation.core.client.model.token_countsr   r2   0pinecone_plugins.assistant.models.core.dataclassr   r	   r$   r"   r   r   <module>r8      se    ! j i r J 
m 
 
& 
M 
 
, 
- 
 
r   