
    ug                         d dl mZ d dlmZ d dlmZmZmZmZ d dl	m
Z
mZmZmZmZ e G d d             Ze G d d             Ze G d	 d
             Ze G d d             Ze G d d             Zy)    )	dataclass)Optional)
EmbeddingsIDsIncludeIncludeEnum)
CollectionRequestVersionContextSegmentWhereWhereDocumentc                   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)Scan
collectionknnmetadatarecordreturnc                 l    t        | j                  j                  | j                  j                        S )N)collection_versionlog_position)r
   r   versionr   )selfs    [/var/www/openai/venv/lib/python3.12/site-packages/chromadb/execution/expression/operator.pyr   zScan.version   s)    $#6655
 	
    N)	__name__
__module____qualname__r	   __annotations__r   propertyr
   r    r   r   r   r      s2    	LO
. 
 
r   r   c                   J    e Zd ZU dZee   ed<   dZee   ed<   dZ	ee
   ed<   y)FilterNuser_idswherewhere_document)r   r   r   r$   r   r   r   r%   r   r&   r   r!   r   r   r#   r#      s-    "Hhsm"!E8E?!.2NH]+2r   r#   c                   "    e Zd ZU eed<   eed<   y)KNN
embeddingsfetchN)r   r   r   r   r   intr!   r   r   r(   r(      s    Jr   r(   c                   0    e Zd ZU dZeed<   dZee   ed<   y)Limitr   skipNr*   )r   r   r   r.   r+   r   r*   r   r!   r   r   r-   r-   #   s    D#ME8C=r   r-   c                   j    e Zd ZU dZeed<   dZeed<   dZeed<   dZeed<   dZ	eed<   e
defd       Zy	)

ProjectionFdocument	embeddingr   rankurir   c                    t               }| j                  r|j                  t        j                         | j
                  r|j                  t        j                         | j                  r|j                  t        j                         | j                  r|j                  t        j                         | j                  r|j                  t        j                         |S )N)listr1   appendr   	documentsr2   r)   r   	metadatasr3   	distancesr4   uris)r   includess     r   includedzProjection.included1   s    6==OOK112>>OOK223==OOK11299OOK11288OOK,,-r   N)r   r   r   r1   boolr   r2   r   r3   r4   r    r   r=   r!   r   r   r0   r0   )   sM    HdItHdD$C'  r   r0   N)dataclassesr   typingr   chromadb.api.typesr   r   r   r   chromadb.typesr	   r
   r   r   r   r   r#   r(   r-   r0   r!   r   r   <module>rC      s    !  D D [ [ 
 
 
 3 3 3   
      
   r   