
    IgP                          d Z ddlZddlmZ  G d de      Z G d de      Zd Z G d	 d
e      Z	 G d de      Z
 G d de      Zd Z G d de      Zy)zEClient and server classes corresponding to protobuf-defined services.    N)
chroma_pb2c                       e Zd ZdZd Zy)MetadataReaderStub Metadata Reader Interface 

    c                 (   |j                  dt        j                  j                  t        j                  j
                        | _        |j                  dt        j                  j                  t        j                  j
                        | _	        y)IConstructor.

        Args:
            channel: A grpc.Channel.
        $/chroma.MetadataReader/QueryMetadatarequest_serializerresponse_deserializer#/chroma.MetadataReader/CountRecordsN)
unary_unary"chromadb_dot_proto_dot_chroma__pb2QueryMetadataRequestSerializeToStringQueryMetadataResponse
FromStringQueryMetadataCountRecordsRequestCountRecordsResponseCountRecordsselfchannels     [/var/www/html/answerous/venv/lib/python3.12/site-packages/chromadb/proto/chroma_pb2_grpc.py__init__zMetadataReaderStub.__init__   sx     %006#E#Z#Z#l#l&H&^&^&i&i 1 
 $//5#E#Y#Y#k#k&H&]&]&h&h 0     N__name__
__module____qualname____doc__r    r   r   r   r          r   r   c                       e Zd ZdZd Zd Zy)MetadataReaderServicerr   c                     |j                  t        j                  j                         |j	                  d       t        d      z8Missing associated documentation comment in .proto file.zMethod not implemented!set_codegrpc
StatusCodeUNIMPLEMENTEDset_detailsNotImplementedErrorr   requestcontexts      r   r   z$MetadataReaderServicer.QueryMetadata$   5    66756!";<<r   c                     |j                  t        j                  j                         |j	                  d       t        d      r(   r)   r0   s      r   r   z#MetadataReaderServicer.CountRecords*   r3   r   N)r   r    r!   r"   r   r   r#   r   r   r&   r&          ==r   r&   c                    t        j                  | j                  t        j                  j
                  t        j                  j                        t        j                  | j                  t        j                  j
                  t        j                  j                        d}t        j                  d|      }|j                  |f       y )Nrequest_deserializerresponse_serializer)r   r   zchroma.MetadataReader)r+   unary_unary_rpc_method_handlerr   r   r   r   r   r   r   r   r   method_handlers_generic_handleradd_generic_rpc_handlersservicerserverrpc_method_handlersgeneric_handlers       r   $add_MetadataReaderServicer_to_serverrB   1   s    !@@**)K)`)`)k)k(J(`(`(r(r
 !??)))K)_)_)j)j(J(_(_(q(q ::#%8:O
##_$67r   c                   T    e Zd ZdZe	 	 	 	 	 	 	 	 dd       Ze	 	 	 	 	 	 	 	 dd       Zy)MetadataReaderr   Nc
                     t         j                  j                  | |dt        j                  j
                  t        j                  j                  ||||||||	      S )Nr	   )r+   experimentalr   r   r   r   r   r   
r1   targetoptionschannel_credentialscall_credentialsinsecurecompressionwait_for_readytimeoutmetadatas
             r   r   zMetadataReader.QueryMetadataI   sX       ,,Wf>d.CCUU.DDOO(&^Wh	X 	Xr   c
                     t         j                  j                  | |dt        j                  j
                  t        j                  j                  ||||||||	      S )Nr   )r+   rF   r   r   r   r   r   r   rG   s
             r   r   zMetadataReader.CountRecordsZ   sX       ,,Wf>c.BBTT.CCNN(&^Wh	X 	Xr   r#   NNFNNNN)r   r    r!   r"   staticmethodr   r   r#   r   r   rD   rD   D   g       $!X X    $!X Xr   rD   c                       e Zd ZdZd Zy)VectorReaderStubVector Reader Interface 

    c                 (   |j                  dt        j                  j                  t        j                  j
                        | _        |j                  dt        j                  j                  t        j                  j
                        | _	        y)r   /chroma.VectorReader/GetVectorsr
   !/chroma.VectorReader/QueryVectorsN)
r   r   GetVectorsRequestr   GetVectorsResponser   
GetVectorsQueryVectorsRequestQueryVectorsResponseQueryVectorsr   s     r   r   zVectorReaderStub.__init__q   sw     "--1#E#W#W#i#i&H&[&[&f&f . 
 $//3#E#Y#Y#k#k&H&]&]&h&h 0 r   Nr   r#   r   r   rV   rV   l   r$   r   rV   c                       e Zd ZdZd Zd Zy)VectorReaderServicerrW   c                     |j                  t        j                  j                         |j	                  d       t        d      r(   r)   r0   s      r   r]   zVectorReaderServicer.GetVectors   r3   r   c                     |j                  t        j                  j                         |j	                  d       t        d      r(   r)   r0   s      r   r`   z!VectorReaderServicer.QueryVectors   r3   r   N)r   r    r!   r"   r]   r`   r#   r   r   rb   rb      r5   r   rb   c                    t        j                  | j                  t        j                  j
                  t        j                  j                        t        j                  | j                  t        j                  j
                  t        j                  j                        d}t        j                  d|      }|j                  |f       y )Nr7   )r]   r`   zchroma.VectorReader)r+   r:   r]   r   r[   r   r\   r   r`   r^   r_   r;   r<   r=   s       r   "add_VectorReaderServicer_to_serverrf      s    =='')K)])])h)h(J(](](o(o
 !??)))K)_)_)j)j(J(_(_(q(q ::!#68O
##_$67r   c                   T    e Zd ZdZe	 	 	 	 	 	 	 	 dd       Ze	 	 	 	 	 	 	 	 dd       Zy)VectorReaderrW   Nc
                     t         j                  j                  | |dt        j                  j
                  t        j                  j                  ||||||||	      S )NrY   )r+   rF   r   r   r[   r   r\   r   rG   s
             r   r]   zVectorReader.GetVectors   sX       ,,Wf>_.@@RR.AALL(&^Wh	X 	Xr   c
                     t         j                  j                  | |dt        j                  j
                  t        j                  j                  ||||||||	      S )NrZ   )r+   rF   r   r   r^   r   r_   r   rG   s
             r   r`   zVectorReader.QueryVectors   sX       ,,Wf>a.BBTT.CCNN(&^Wh	X 	Xr   rR   )r   r    r!   r"   rS   r]   r`   r#   r   r   rh   rh      rT   r   rh   )r"   r+   chromadb.protor   r   objectr   r&   rB   rD   rV   rb   rf   rh   r#   r   r   <module>rm      se    K  K .=V =$8&%XV %XPv .=6 =$8&%X6 %Xr   