
    BTh                        d dl mZ d dlmZmZ 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 d dlmZmZ e G d	 d
e             Ze G d de             Ze G d de             Ze G d de             Z G d de      Ze G d de             Ze G d de             Ze G d de             Ze G d de             Ze G d de             Ze G d de             Zy)     )	dataclass)ListOptional)	ChatModel)CitationModel)ReferenceModel)BaseDataclass)	FileModel)MessageUsagec                   N    e Zd ZU eed<   eed<   edefd       Zedefd       Zy)	Highlighttypecontentdc                 R     | |j                  d      |j                  d            S Nr   r   )r   r   getclsr   s     x/var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/pinecone_plugins/assistant/models/chat.py	from_dictzHighlight.from_dict   s&    vEE)$
 	
    highlight_modelc                 R     | |j                  d      |j                  d            S r   r   )r   r   s     r   from_openapizHighlight.from_openapi   s+     $$V,#''	2
 	
r   N)	__name__
__module____qualname__str__annotations__classmethoddictr   r    r   r   r   r      s@    
IL
$ 
 
 
4 
 
r   r   c                   d    e Zd ZU eed<   ee   ed<   ee   ed<   e	de
fd       Ze	defd       Zy)		Referencefilepages	highlightr   c           	           | |j                  d      t        j                  |j                  d            |j                  d      r)t        j                  |j                  d                  S d       S )Nr)   r(   r*   r)   r(   r*   )r   r
   r   r   r   s     r   r   zReference.from_dict%   sc    %%.$$QUU6]3AB{ASi))!%%*<=
 	
 Z^
 	
r   reference_modelc                      | |j                   t        j                  |j                        |j                  r$t
        j                  |j                              S d       S )Nr,   )r)   r
   r   r(   r*   r   )r   r-   s     r   r   zReference.from_openapi-   sY    !''''(<(<=KZKdKdi,,_-F-FG
 	
 ko
 	
r   N)r   r   r    r
   r"   r   intr   r   r#   r$   r   r   r   r%   r   r   r'   r'      sP    
O9	""
$ 
 
 
> 
 
r   r'   c                   T    e Zd ZU eed<   ee   ed<   edefd       Z	ede
fd       Zy)Citationposition
referencesr   c           	           | |j                  d      |j                  dg       D cg c]  }t        j                  |       c}      S c c}w )Nr2   r3   r2   r3   )r   r'   r   )r   r   reference_datas      r   r   zCitation.from_dict;   sF    UU:&RSRWRWXdfhRij	++N;j
 	
js   A
citation_modelc           	           | |j                   |j                  D cg c]  }t        j                  |       c}      S c c}w )Nr5   )r2   r3   r'   r   )r   r7   r6   s      r   r   zCitation.from_openapiB   s:    #,,UcUnUno>	..~>o
 	
os   A 
N)r   r   r    r/   r"   r   r'   r#   r$   r   r   r   r%   r   r   r1   r1   6   sE    MY
$ 
 
 
- 
 
r   r1   c                   f    e Zd ZU eed<   eed<   eed<   eed<   eed<   ee   ed<   e	de
fd       Zy	)
ChatResponseidmodelusagemessagefinish_reason	citations
chat_modelc                 .    | |j                   |j                  t        j                  |j                        t        j                  |j                        |j                  |j                  D cg c]  }t        j                  |       c}      S c c}w )N)r;   r<   r=   r>   r?   r@   )
r;   r<   r   r   r=   r   r>   r?   r@   r1   )r   rA   citation_datas      r   r   zChatResponse.from_openapiS   su    }}""$$Z%5%56((););<$22Q[QeQefx,,];f
 	
 gs   .BN)r   r   r    r!   r"   r   r   r   r1   r#   OpenAPIChatModelr   r%   r   r   r:   r:   J   sA    GJLH~
&6 
 
r   r:   c                       e Zd Zy)BaseStreamChatResponseChunkN)r   r   r    r%   r   r   rF   rF   _   s    r   rF   c                   B    e Zd ZU eed<   eed<   eed<   edefd       Zy)StreamChatResponseMessageStartr   r<   roler   c                 r     | |j                  d      |j                  d      |j                  d            S )Nr   r<   rI   )r   r<   rI   r   r   s     r   r   z(StreamChatResponseMessageStart.from_dicti   s0    v%%.v
 	
r   Nr   r   r    r!   r"   r#   r$   r   r%   r   r   rH   rH   c   s*    
IJ
I
$ 
 
r   rH   c                   .    e Zd ZU eed<   edefd       Zy)MessageDeltar   r   c                 2     | |j                  d            S )Nr   )r   r   r   s     r   r   zMessageDelta.from_dictv   s    EE)$
 	
r   NrK   r%   r   r   rM   rM   r   s     L
$ 
 
r   rM   c                   L    e Zd ZU eed<   eed<   eed<   eed<   edefd       Zy)StreamChatResponseContentDeltar;   r   r<   deltar   c           
           | |j                  d      |j                  d      |j                  d      t        j                  |j                  d                  S )Nr;   r   r<   rQ   )r;   r   r<   rQ   )r   rM   r   r   s     r   r   z(StreamChatResponseContentDelta.from_dict   sF    uuT{v%%.((w8	
 	
r   N)	r   r   r    r!   r"   rM   r#   r$   r   r%   r   r   rP   rP   }   s0    G
IJ
$ 
 
r   rP   c                   L    e Zd ZU eed<   eed<   eed<   eed<   edefd       Zy)StreamChatResponseCitationr   r;   r<   citationr   c           
           | |j                  d      |j                  d      |j                  d      t        j                  |j                  d                  S )Nr   r;   r<   rU   )r   r;   r<   rU   )r   r1   r   r   s     r   r   z$StreamChatResponseCitation.from_dict   sG    vuuT{%%.''j(9:	
 	
r   N)	r   r   r    r!   r"   r1   r#   r$   r   r%   r   r   rT   rT      s0    
IGJ
$ 
 
r   rT   c                   L    e Zd ZU eed<   eed<   eed<   eed<   edefd       Zy)StreamChatResponseMessageEndr   r<   r;   r=   r   c           
           | |j                  d      |j                  d      |j                  d      t        j                  |j                  d                  S )Nr   r<   r;   r=   )r   r<   r;   r=   )r   r   r   r   s     r   r   z&StreamChatResponseMessageEnd.from_dict   sD    v%%.uuT{//!%%.1	
 	
r   N)	r   r   r    r!   r"   r   r#   r$   r   r%   r   r   rX   rX      s/    
IJGL
$ 
 
r   rX   c                   L    e Zd ZU dZee   ed<   dZee   ed<   ede	fd       Z
y)ContextOptionsNtop_ksnippet_sizer   c                 R     | |j                  d      |j                  d            S )Nr\   r]   )r\   r]   r   r   s     r   r   zContextOptions.from_dict   s&    %%.~.
 	
r   )r   r   r    r\   r   r/   r"   r]   r#   r$   r   r%   r   r   r[   r[      s7    E8C="&L(3-&
$ 
 
r   r[   N)dataclassesr   typingr   r   <pinecone_plugins.assistant.data.core.client.model.chat_modelr   rD   @pinecone_plugins.assistant.data.core.client.model.citation_modelr   Apinecone_plugins.assistant.data.core.client.model.reference_modelr   0pinecone_plugins.assistant.models.core.dataclassr	   ,pinecone_plugins.assistant.models.file_modelr
   (pinecone_plugins.assistant.models.sharedr   r   r   r'   r1   r:   rF   rH   rM   rP   rT   rX   r[   r%   r   r   <module>rg      s;   ! ! f Z \ J B C 
 
 
$ 
 
 
, 
} 
 
& 
= 
 
(	- 	 
%@ 
 
 
= 
 
 
%@ 
 
  
!< 
 
  
#> 
 
 	
] 	
 	
r   