
    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  eded	   
      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 G d d      Ze G d de             Ze G d de             Zy)    )	dataclass)TypeVarUnion)ContextModel)SnippetModel)BaseDataclass)	FileModel)TokenCountsRefType)TextReferencePdfReferenceMarkdownReferenceJsonReference)boundc                   (    e Zd ZU eed<   ed        Zy)BaseReferencetypec                     t         )N)NotImplementedError)clsvalues     /var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/pinecone_plugins/assistant/models/context_responses.pyfrom_openapizBaseReference.from_openapi   s    !!    N)__name__
__module____qualname__str__annotations__classmethodr    r   r   r   r      s    
I" "r   r   c                   B    e Zd ZU ee   ed<   eed<   ededd fd       Z	y)r   pagesfileref_dictreturnc                 P     | |d   |d   t        j                  |d               S Nr   r#   r$   )r   r#   r$   r	   	from_dictr   r%   s     r   r   zPdfReference.from_openapi"   2    &!7#$$Xf%56
 	
r   N
r   r   r   listintr   r	   r    dictr   r!   r   r   r   r      1    9
O
D 
^ 
 
r   r   c                   B    e Zd ZU ee   ed<   eed<   ededdfd       Z	y)DocxReferencer#   r$   r%   r&   r   c                 P     | |d   |d   t        j                  |d               S r(   r)   r+   s     r   r   zDocxReference.from_openapi0   r,   r   Nr-   r!   r   r   r3   r3   +   r1   r   r3   c                   2    e Zd ZU eed<   ededd fd       Zy)r   r$   r%   r&   c                 H     | |d   t        j                  |d               S Nr   r$   )r   r$   r)   r+   s     r   r   zTextReference.from_openapi=   $    (y/B/B8FCS/TUUr   Nr   r   r   r	   r   r    r0   r   r!   r   r   r   r   9   +    
OVD V_ V Vr   r   c                   2    e Zd ZU eed<   ededd fd       Zy)r   r$   r%   r&   c                 H     | |d   t        j                  |d               S r7   r)   r+   s     r   r   zMarkdownReference.from_openapiF   r8   r   Nr9   r!   r   r   r   r   B   s,    
OVD V-@ V Vr   r   c                   2    e Zd ZU eed<   ededd fd       Zy)r   r$   r%   r&   c                 H     | |d   t        j                  |d               S r7   r)   r+   s     r   r   zJsonReference.from_openapiO   r8   r   Nr9   r!   r   r   r   r   K   r:   r   r   c                   &    e Zd Zededefd       Zy)TypedReferencedr&   c                 l    |d   }t         t        t        t        t        d}||   j                  |      S )Nr   )textdoc_xpdfmarkdownjson)r   r3   r   r   r   r   )r   rA   type_ref_maps       r   r   zTypedReference.from_openapiU   s8    &	!")!
 u~**1--r   N)r   r   r   r    r0   r   r   r!   r   r   r@   r@   T   s"    	.T 	.g 	. 	.r   r@   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)Snippetr   contentscore	referencesnippetc                      | |j                   |j                  |j                  t        j	                  |j
                              S )N)r   rL   rM   rN   )r   rL   rM   r@   r   rN   )r   rO   s     r   r   zSnippet.from_openapii   s8    OO--$11'2C2CD	
 	
r   N)
r   r   r   r   r   floatr   r    OpenAPISnippetModelr   r!   r   r   rK   rK   b   s1    
ILL
#6 
 
r   rK   c                   H    e Zd ZU eed<   ee   ed<   eed<   ede	fd       Z
y)ContextResponseidsnippetsusagectxc           	           | |j                   |j                  D cg c]  }t        j                  |       c}t	        j                  |j
                              S c c}w )N)rU   rV   rW   )rU   rV   rK   r   r
   rW   )r   rX   rO   s      r   r   zContextResponse.from_openapiy   sH    vvCF<<Pg**73P**3995
 	
Ps   A
N)r   r   r   r   r   r.   rK   r
   r    OpenAPIContextModelr   r!   r   r   rT   rT   s   s1    G7m
2 
 
r   rT   N)dataclassesr   typingr   r   ?pinecone_plugins.assistant.data.core.client.model.context_modelr   rZ   ?pinecone_plugins.assistant.data.core.client.model.snippet_modelr   rR   0pinecone_plugins.assistant.models.core.dataclassr   ,pinecone_plugins.assistant.models.file_modelr	   (pinecone_plugins.assistant.models.sharedr
   r   r   r   r3   r   r   r   r@   rK   rT   r!   r   r   <module>rb      s"   ! ! K B @

U
V "M " " 

= 

 

 

M 

 

 VM V V V V V VM V V. . 
m 
 
  
m 
 
r   