
    <i                    j   S SK rS SKrS SKrS SKrS SKJr  S SKJrJ	r	J
r
JrJrJrJrJr  S SKrS SKJr  S SKJrJr  S SKJr  S SKJrJr  S SKJr  S S	KJr  S S
KJr  S SK J!r!  S SK"J#r#J$r$  S SK%J&r'  S SK(J)r*  S SK+J,r,  S SK-J.r.J/r/J0r0  S SK1J2r2J3r3J4r4  S SK5J6r6  S SK7J8r8  S SK9J:r:  S SK;J<r<   " S S\!5      r=g)    N)get_all_start_methods)Any	AwaitableCallableIterableMappingSequenceTypeget_args)Compression)Url	parse_url)urljoin)show_warningshow_warning_once)grpc)	construct)
BearerAuth)AsyncQdrantBase)is_compatibleget_server_version)get_async_channel)common_types)get_args_subscribed)
GrpcToRest
RestToGrpc!grpc_payload_schema_to_field_type)AsyncApiClient	AsyncApismodels)ParallelWorkerPool)GrpcBatchUploader)RestBatchUploader)BaseUploaderc            '          ^  \ rS rSrSrSr             SS\S-  S\S-  S\S\S	\S-  S
\S-  S\S-  S\S-  S\S-  S\	\\
4   S-  S\/ \4   \/ \\   4   -  S-  S\S\S-  S\
4U 4S jjjr\S\4S j5       rSS\S-  S\
SS4S jjr\S\S\\S-  \\S-  \S-  4   4S j5       rS\4S jrSS jrSS jrSS jrSS jrSS jrS\4S jr\S\R:                  4S  j5       r\S\R>                  4S! j5       r \S\RB                  4S" j5       r"\S\RF                  4S# j5       r$\S\%\&   4S$ j5       r'\S\%\&   4S% j5       r(              SS&\S'\)RT                  \+\   -  \+\+\      -  \)RX                  -  \)RZ                  -  \)R\                  -  \)R^                  -  \)R`                  -  \)Rb                  -  S-  S(\S-  S)\)Rd                  \+\)Rd                     -  S-  S*\)Rf                  S-  S+\)Rh                  S-  S,\S-\S-  S.\\5\   -  \)Rl                  -  S/\\5\   -  S0\S-  S1\)Rn                  S-  S2\)Rp                  S-  S3\)Rr                  S-  S\S-  S\
S\)Rt                  4"S4 jjr;  SS&\S5\5\)Rx                     S2\)Rp                  S-  S\S-  S\
S\+\)Rt                     4S6 jjr=               SS&\S7\S'\)RT                  \+\   -  \+\+\      -  \)RX                  -  \)RZ                  -  \)R\                  -  \)R^                  -  \)R`                  -  \)Rb                  -  S-  S(\S-  S)\)Rd                  \+\)Rd                     -  S-  S*\)Rf                  S-  S+\)Rh                  S-  S,\S8\S.\\5\   -  \)Rl                  -  S/\\5\   -  S0\S-  S9\)R|                  S-  S1\)Rn                  S-  S2\)Rp                  S-  S3\)Rr                  S-  S\S-  S\
S\)R~                  4&S: jjr@       SS&\S*\)Rf                  S-  S,\S;\S(\S-  S2\)Rp                  S-  S3\)Rr                  S-  S\S-  S\
S\)R                  4S< jjrB       SS&\S*\)Rf                  S-  S,\S;\S(\S-  S2\)Rp                  S-  S3\)Rr                  S-  S\S-  S\
S\)R                  4S= jjrD         SS&\S>\)Rf                  S-  S,\S?\)R                  S-  S-\)RT                  S-  S.\\5\   -  \)Rl                  -  S/\\5\   -  S2\)Rp                  S-  S3\)Rr                  S-  S\S-  S\
S\\+\)R                     \)RT                  S-  4   4S@ jjrG     SS&\SA\)Rf                  S-  SB\S3\)Rr                  S-  S\S-  S2\)Rp                  S-  S\
S\)R                  4SC jjrI      SS&\SD\SE\)Rf                  S-  S,\SB\S\S-  S2\)Rp                  S-  S3\)Rr                  S-  S\
S\)R                  4SF jjrK    SS&\SG\)R                  SH\SI\)R                  S-  S3\)Rr                  S-  SJ\)Rf                  S-  S\
S\)R                  4SK jjrO    SS&\SG\5\)R                     SH\SI\)R                  S-  S3\)Rr                  S-  SJ\)Rf                  S-  S\
S\)R                  4SL jjrQ   SS&\SM\5\   SG\)R                  SH\SI\)R                  S-  S3\)Rr                  S-  S\
S\)R                  4SN jjrS     SS&\SO\5\)RT                     S.\\5\   -  \)Rl                  -  S/\\5\   -  S2\)Rp                  S-  S3\)Rr                  S-  S\S-  S\
S\+\)R                     4SP jjrT\USG\)R                  S\\R                  \Rr                  S-  4   4SQ j5       rV\USG\)R                  S3\)Rr                  S-  S\WR                  4SR j5       rX\USS\R                  S\+\RT                     4ST j5       rY\USG\)R                  S\\+\WR                     S-  \WRf                  S-  4   4SU j5       r[   SS&\SS\)R                  SH\SI\)R                  S-  S3\)Rr                  S-  S\
S\)R                  4SV jjr\    SS&\SW\)R                  SG\)R                  SD\S-  SH\SI\)R                  S-  S3\)Rr                  S-  S\
S\)R                  4SX jjr^   SS&\SW\)R                  SG\)R                  SH\SI\)R                  S-  S3\)Rr                  S-  S\
S\)R                  4SY jjr_   SS&\SZ\5\   SG\)R                  SH\SI\)R                  S-  S3\)Rr                  S-  S\
S\)R                  4S[ jjr`   SS&\SS\)R                  SH\SI\)R                  S-  S3\)Rr                  S-  S\
S\)R                  4S\ jjra  SS&\S]\5\)R                     SH\SI\)R                  S-  S\
S\+\)R                     4S^ jjrc SS_\5\)R                     S\S-  S\
S\4S` jjreS&\S\
S\)R                  4Sa jrgS\
S\)R                  4Sb jrhS\
S\)R                  4Sc jrjS&\S\
S\)R                  4Sd jrlS&\S\
S\4Se jrm         SS&\Sf\)R                  S-  Sg\)R                  S-  Sh\)R                  S-  Si\)R                  S-  Sj\)R                  S-  S\S-  Sk\s\\)R                  4   S-  Sl\)R                  S-  Sm\)R                  S-  S\
S\4Sn jjrv SS&\S\S-  S\
S\4So jjrw              SS&\Sh\)R                  \s\\)R                  4   -  S-  Sp\S-  Sq\S-  Sr\S-  Ss\S-  Si\)R                  S-  Sf\)R                  S-  St\)R                  S-  Sj\)R                  S-  S\S-  Sk\s\\)R                  4   S-  Su\)R                  S-  Sl\)R                  S-  Sm\)R                  S-  S\
S\4"Sv jjr|              SS&\Sh\)R                  \s\\)R                  4   -  S-  Sp\S-  Sq\S-  Sr\S-  Ss\S-  Si\)R                  S-  Sf\)R                  S-  St\)R                  S-  Sj\)R                  S-  S\S-  Sk\s\\)R                  4   S-  Su\)R                  S-  Sl\)R                  S-  Sm\)R                  S-  S\
S\4"Sw jjr}\S\~\   4Sx j5       r     SSy\S&\Sz\S{\S|\S-  SH\S3\)Rr                  S-  SJ\)Rf                  S-  SS4S} jjr       SS&\SG\\)GR                     S~\S{\S|\S-  Sz\SH\S3\)Rr                  S-  SJ\)Rf                  S-  S\
SS4S jjr         SS&\SM\	\\)R\                  4   \)R\                  -  \\)GR
                     -  SW\\	\
\
4      S-  SO\\)RT                     S-  S~\S{\S|\S-  Sz\SH\S3\)Rr                  S-  SJ\)Rf                  S-  S\
SS4S jjr    SS&\S\S\)GR                  S-  S\)GR                  S-  SH\SI\)R                  S-  S\
S\)R                  4S jjr  SS&\S\SH\SI\)R                  S-  S\
S\)R                  4S jjrS&\S\
S\+\)GR                     4S jr SS&\SH\S\
S\)GR                  S-  4S jjr SS&\S\SH\S\
S\S-  4
S jjrS\
S\+\)GR                     4S jr SSH\S\
S\)GR                  4S jjr SS\SH\S\
S\S-  4S jjr    SS&\S\S
\S-  S\S-  S\)GR"                  S-  SH\S\
S\S-  4S jjrS&\S\S\
S\+\)GR                     4S jr SS&\S\SH\S\
S\)GR                  S-  4
S jjr SS&\S\S\SH\S\
S\S-  4S jjr    SS&\S\S\S
\S-  S\S-  S\)GR"                  S-  SH\S\
S\S-  4S jjr     SS&\S\)GR.                  S\S-  Sq\S-  S\+\   S-  S\)GR0                  S-  S\S-  S\
S\4S jjr SS&\S\)GR.                  S\S-  S\
S\4
S jjrS\)GR6                  4S jr SS&\S\)GR:                  S\S-  S\
S\4
S jjrS\)GR>                  4S jrS\4S jr SS\S\S-  S\S-  S\
S\4
S jjrS&\S\)GRF                  4S jrSrU =r$ )AsyncQdrantRemote+         Nurlport	grpc_portprefer_grpchttpsapi_keyprefixtimeouthostgrpc_optionsauth_token_providercheck_compatibility	pool_sizekwargsc                   > [         TU ]  " S40 UD6  X@l        X0l        U
=(       d    0 U l        Ub  UOUS LU l        U R
                  (       a  SOSU l        S U l        Ub  [        SU5      nXl        U=(       d    SU l	        [        U R                  5      S:  a'  U R                  S   S:w  a  SU R                   3U l	        Ub  U	b  [        SU SU	 35      eU	b7  U	R                  S	5      (       d  U	R                  S
5      (       a  [        S5      eU(       Ga+  UR                  S5      (       a  SU 3n[        U5      nUR                  UR                  sU l        U l        UR$                  (       a%  UR$                  S:H  U l        UR$                  U l        U R"                  (       a  U R"                  OUU l        U R                  (       a,  UR&                  (       a  [        SU SUR&                   35      eUR&                  (       a  UR&                  U l	        U R                  S;  a  [        SU R                   35      eOU	=(       d    SU l        X l        Ub  [(        R*                  " U5      OS nX`l        Xl        UR1                  SS 5      nUcR  U R                   S;   a  [2        R4                  " S SS9nOJU R                  b  [2        R4                  " U R                  S9nOU R                  b  [        SU SU 35      eUR1                  SS5      n/ U l        UR1                  S0 5      R9                  5        VVs0 s H	  u  nnUU_M     snnU l        UbJ  U R                  S:X  a  [=        S[>        SS9  X`R:                  S'   U R6                  RA                  SU45        [B        RD                  RG                  S5      n[H        RJ                  " 5       nS U S!U 3nUU R:                  S"'   UU R                  S#'   UR1                  S$S 5      nUb,  [M        U[N        5      (       d  [Q        S%[S        U5       35      eU[N        RT                  :X  a  [        S&5      eUU l+        U R"                  b  U R                    S'U R"                   3OU R                   nU R                   S(U 3n[Y        UU R                  5      U l-        U R:                  US).UEU l.        Ub  UU R\                  S'   Ub  UU R\                  S*'   UU l/        OU R`                  U l/        U R.                  bC  U R                  S:X  a  [=        S+[>        SS9  [c        U R.                  5      nUU R\                  S,'   [e        S4S-U RZ                  0U R\                  D6U l3        / U l4        S U l5        S U l6        S U l7        S U l8        SU l9        SU l:        S U l;        U(       a   [B        RD                  RG                  S5      n[y        U RZ                  U R:                  U R\                  R{                  S,5      5      U l;        U Rv                  (       d  [=        S.[>        SS9  g [}        UU Rv                  5      (       d!  [=        S/U S0U Rv                   S13[>        SS9  g g g s  snnf ! [~         a$  n[        R                  " S2U S335         S nAg S nAff = f)5Nr.   http    r   /z+Only one of (url, host) can be set. url is z
, host is zhttp://zhttps://zk`host` param is not expected to contain protocol (http:// or https://). Try to use `url` parameter instead.	localhostz//z9Prefix can be set either in `url` or in `prefix`. url is z, prefix is )r9   r.   zUnknown scheme: limits)r=   z	127.0.0.1)max_connectionsmax_keepalive_connections)r?   z>`pool_size` and `limits` are mutually exclusive. `pool_size`: z, `limit`: http2Fmetadataz,Api key is used with an insecure connection.   messagecategory
stacklevelzapi-keyzqdrant-clientzpython-client/z python/z
User-Agentzgrpc.primary_user_agentgrpc_compressionzLExpected 'grpc_compression' to be of type grpc.Compression or None, but got zfgrpc.Compression.Deflate is not supported. Try grpc.Compression.Gzip or grpc.Compression.NoCompression:z://)headersrA   r1   z8Auth token provider is used with an insecure connection.authr2   zFailed to obtain server version. Unable to check client-server compatibility. Set check_compatibility=False to skip version check.zQdrant client version z% is incompatible with server version z. Major versions should match and minor version difference must not exceed 1. Set check_compatibility=False to skip version check.zUnable to get server version: z!, server version defaults to None )Bsuper__init___prefer_grpc
_grpc_port_grpc_options_https_scheme
_pool_sizemax_prefixlen
ValueError
startswithr   r2   r+   _host_portschemepathmathceil_api_key_auth_token_providerpophttpxLimits_grpc_headersitems_rest_headersr   UserWarningappend	importlibrB   versionplatformpython_version
isinstancer   	TypeErrortypeDeflate_grpc_compressionr   rest_uri
_rest_args_timeoutDEFAULT_GRPC_TIMEOUTr   r   openapi_client_grpc_channel_pool_grpc_points_client_pool_grpc_collections_client_pool_grpc_snapshots_client_pool_grpc_root_client_pool_grpc_client_next_index_closedserver_versionr   getr   	Exceptionloggingdebug)selfr*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   
parsed_urlru   r>   rA   kvclient_versionrm   
user_agentrH   addressbase_urlbearer_auther	__class__s                                [/home/james-whalen/.local/lib/python3.13/site-packages/qdrant_client/async_qdrant_remote.pyrN   AsyncQdrantRemote.__init__/   sM   " 	"6"'#)/R$0egT6I"&++w6&* Ay)I'O|t||q T\\!_%;t||n-DL?t/J3%zZ^Y_`aa!;!;tz?Z?Z}  ~~k**3%j'nJ(2$TZ  (//7:)00'+zztDJ||
 OPSuT`akapap`qr  )||#44 #3DLL>!BCC 5 ,DJJ)0)<499W%$$7!Hd+>zz77dVWX,dooF__(PQZP[[fgmfno  

7E*17J1K1Q1Q1ST1Sv1ad1ST||v%J( 
 -4y)%%y'&:;"++33OD!002%n%5Xn=MN
+5<(8B45/5zz:Ld/S'<Lk1Z1Z^_cdt_u^vw  {222x  "226**2HTZZL$**.djjll^3wi0$,,7&*&8&85SFS(.DOOH%)1DOOI&$DM 55DM$$0||v%V( 
 %T%>%>?K&1DOOF#9B :
:
"&//:
 79FJ%PT*LP(DH#,-$""!*!3!3!;!;O!L&8MM4#5#5t7J7J67R'# **  !e!,#$
 '~t7J7JKK "88HHmnr  oB  oB  nC  CE  !F!,#$ L q UT  4RD8YZ s%   ZA>Z ;Z 
[![  [returnc                     U R                   $ N)r~   r   s    r   closedAsyncQdrantRemote.closed   s    ||    
grpc_gracec                   #    [        U S5      (       aD  [        U R                  5      S:  a+  U R                   H  n UR                  US9I S h  vN   M      U R                  R                  5       I S h  vN   SU l        g  N3! [         a    [        S[        SS9   Mh  [         a     Mt  f = f N:! [         a    [        S[        SS9   NSf = f7f)	Nrx   r   )gracezKUnable to close grpc_channel. Connection was interrupted on the server siderC   rD   zNUnable to close http connection. Connection was interrupted on the server sideT)hasattrrW   rx   closeAttributeErrorr   rh   RuntimeErrorr9   acloser   r~   )r   r   r7   channels       r   r   AsyncQdrantRemote.close   s     4-..3t7N7N3ORS3S22	!--j-999 3	))""$$$ # :%   m!,#$
 $  % 	h$	s|   :CBBBCB3 6B17B3 ;CBB.C!	B.*C-B..C1B3 3CCCCc                     [        U 5      nUR                  UR                  UR                  UR                  4u  p#pEX#XE4$ r   )r   r\   r2   r+   r]   )r*   parse_resultr\   r2   r+   r0   s         r   
_parse_urlAsyncQdrantRemote._parse_url   sH    %cN	(
$t d++r   c                 L    U R                   b  U R                   $ U R                  $ )z
Returns the pool size to use for GRPC connection pool.
This method should be preferred over accessing `self._pool_size` directly as it applies the
default value if no pool_size was provided.
)rT   DEFAULT_GRPC_POOL_SIZEr   s    r   _get_grpc_pool_size%AsyncQdrantRemote._get_grpc_pool_size   s$     ??&??"...r   c                    U R                   (       a  [        S5      e / n[        U R                  5      S:X  a  [	        U R                  5       5       Hi  n[        U R                  U R                  U R                  U R                  U R                  U R                  U R                  S9nUR                  U5        Mk     Xl        g g ! [         a  n[        SU 35      eS nAff = f)Nz=Client was closed. Please create a new QdrantClient instance.r   )r2   r+   sslrB   optionscompressionr4   z+Error initializing the grpc connection(s): )r~   r   rW   rx   ranger   get_channelrZ   rP   rR   re   rQ   rr   ra   ri   r   )r   channel_pool_r   es        r   _init_grpc_channel$AsyncQdrantRemote._init_grpc_channel   s    <<^__	RL4**+q0t779:A)!ZZ!__ KK!%!3!3 $ 2 2$($:$:,0,E,EG !''0 ; +7' 1  	R!LQCPQQ	Rs   B'C 
C$CC$c                     U R                  5         U R                   Vs/ s H  n[        R                  " U5      PM     snU l        g s  snf r   )r   rx   r   
PointsStubry   r   r   s     r   _init_grpc_points_client*AsyncQdrantRemote._init_grpc_points_client  s?    !484K4K)
4KDOOG$4K)
% )
    Ac                     U R                  5         U R                   Vs/ s H  n[        R                  " U5      PM     snU l        g s  snf r   )r   rx   r   CollectionsStubrz   r   s     r   _init_grpc_collections_client/AsyncQdrantRemote._init_grpc_collections_client  sA    !9=9P9P.
9PgD  )9P.
* .
r   c                     U R                  5         U R                   Vs/ s H  n[        R                  " U5      PM     snU l        g s  snf r   )r   rx   r   SnapshotsStubr{   r   s     r   _init_grpc_snapshots_client-AsyncQdrantRemote._init_grpc_snapshots_client  sA    !7;7N7N,
7NGDw'7N,
( ,
r   c                     U R                  5         U R                   Vs/ s H  n[        R                  " U5      PM     snU l        g s  snf r   )r   rx   r   
QdrantStubr|   r   s     r   _init_grpc_root_client(AsyncQdrantRemote._init_grpc_root_client  s?    !484K4K'
4KDOOG$4K'
# '
r   c                 h    U R                   nU R                   S-   U R                  5       -  U l         U$ )Nr:   )r}   r   )r   current_indexs     r   _next_grpc_client#AsyncQdrantRemote._next_grpc_client#  s9    44((1,$$&('$ r   c                     U R                   c  U R                  5         U R                   c   eU R                   U R                  5          $ )zjgRPC client for collections methods

Returns:
    An instance of raw gRPC client, generated from Protobuf
)rz   r   r   r   s    r   grpc_collections"AsyncQdrantRemote.grpc_collections*  sH     --5..011===11$2H2H2JKKr   c                     U R                   c  U R                  5         U R                   c   eU R                   U R                  5          $ )zegRPC client for points methods

Returns:
    An instance of raw gRPC client, generated from Protobuf
)ry   r   r   r   s    r   grpc_pointsAsyncQdrantRemote.grpc_points6  sH     ((0))+,,888,,T-C-C-EFFr   c                     U R                   c  U R                  5         U R                   c   eU R                   U R                  5          $ )zhgRPC client for snapshots methods

Returns:
    An instance of raw gRPC client, generated from Protobuf
)r{   r   r   r   s    r   grpc_snapshots AsyncQdrantRemote.grpc_snapshotsB  sH     ++3,,.//;;;//0F0F0HIIr   c                     U R                   c  U R                  5         U R                   c   eU R                   U R                  5          $ )zcgRPC client for info methods

Returns:
    An instance of raw gRPC client, generated from Protobuf
)r|   r   r   r   s    r   	grpc_rootAsyncQdrantRemote.grpc_rootN  sH     &&.'')**666**4+A+A+CDDr   c                     U R                   $ z\REST Client

Returns:
    An instance of raw REST API client, generated from OpenAPI schema
rw   r   s    r   restAsyncQdrantRemote.restZ       """r   c                     U R                   $ r   r   r   s    r   r9   AsyncQdrantRemote.httpc  r   r   collection_namequeryusingprefetchquery_filtersearch_paramslimitoffsetwith_payloadwith_vectorsscore_thresholdlookup_fromconsistencyshard_key_selectorc                 d  #    U R                   (       Ga  Ub  [        R                  " U5      n[        U[        R
                  5      (       a  [        R                  " U5      /n[        U[        5      (       aG  U Vs/ s H:  n[        U[        R
                  5      (       a  [        R                  " U5      OUPM<     nn[        U[        R                  5      (       a  [        R                  " US9n[        U[        R                  5      (       a  [        R                  " U5      n[        U	[        [        R                  5      5      (       a  [        R                  " U	5      n	[        U
[        [        R                  5      5      (       a  [        R                   " U
5      n
[        U[        R"                  5      (       a  [        R$                  " U5      n[        U[        [        R&                  5      5      (       a  [        R(                  " U5      n[        U[        [        R*                  5      5      (       a  [        R,                  " U5      nU R.                  R1                  [2        R4                  " UUUUUUU
U	UUUUUUUS9Ub  UOU R6                  S9I S h  vN nUR8                   Vs/ s H  n[:        R<                  " U5      PM     nn[        R>                  " US9$ [        U[2        R                  5      (       a  [:        R                  " US9n[        U[2        R                  5      (       a  [:        R                  " U5      n[        U	[2        R@                  5      (       a  [:        RB                  " U	5      n	[        U[2        R"                  5      (       a  [:        R$                  " U5      n[        RD                  " UUUUUUUUUU
U	US9nU RF                  RH                  RK                  UUUUS9I S h  vN nUR8                  nUc   S5       eU$ s  snf  GNs  snf  N)7f)Nmodel)r   r   r   filterr   r   r   r   paramsr   r   r   r1   r   read_consistencyr1   points)	shard_keyr   r   r   r   r   r   r   r   with_vectorr   r   )r   r   r1   query_requestzSearch returned None)&rO   r   convert_queryrn   r    Prefetchconvert_prefetch_querylistFilterconvert_filterSearchParamsconvert_search_paramsr   WithPayloadInterfaceconvert_with_payload_interface
WithVectorconvert_with_vectorsLookupLocationconvert_lookup_locationReadConsistencyconvert_read_consistencyShardKeySelectorconvert_shard_key_selectorr   Queryr   QueryPointsru   resultr   convert_scored_pointQueryResponseWithPayloadSelectorconvert_with_payload_selectorQueryRequestr9   
search_apiquery_points)r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r1   r7   preshitscored_pointsr   query_resultr  s                           r   r  AsyncQdrantRemote.query_pointsl  s}    8  "007(FOO44&==hGH(D)) &% =Gq&//<Z<ZJ55a8`aa%   ,66)88|L-)<)<== * @ @ O,(;F<W<W(XYY)HHV,(;F<M<M(NOO)>>|L+v'<'<==(@@M+':6;Q;Q'RSS(AA+N,.A&BYBY.Z[[%/%J%JK]%^",0,<,<,B,B  $3%'!!-!-($3 +#'9%0" $+#6DMM% -C - 'C( NQZZXZcZ<<SAZMX''}==,44)88|L-):):;; * @ @ O,(@(@AA)GGU+t':':;;(@@M"//,!#$ /()'M "&!5!5!B!B /'+	 "C " L 3?2E2EF%='==%MO$'( Y2s?   A;P0=AP!>G#P0!P&"P04 P)D0P0P."P0)P0requestsc                 X  #    U R                   (       Ga+  U Vs/ s H:  n[        U[        R                  5      (       a  [        R
                  " Xa5      OUPM<     nn[        U[        [        R                  5      5      (       a  [        R                  " U5      nU R                  R                  [        R                  " UUUUS9Ub  UOU R                  S9I S h  vN nUR                   VVs/ s HF  n[        R                  " UR                   Vs/ s H  n[         R"                  " U5      PM     snS9PMH     snn$ U R$                  R&                  R)                  UUU[        R*                  " US9S9I S h  vN R                  n	U	c   S5       eU	$ s  snf  Ns  snf s  snnf  N,7f)N)r   r  r   r1   r   r   )searches)r   r   r1   query_request_batchzQuery batch returned None)rO   rn   r    r  r   convert_query_requestr   r  r  r   
QueryBatchr   QueryBatchPointsru   r  r  r   r  r9   r  query_batch_pointsQueryRequestBatch)
r   r   r  r   r1   r7   rgrpc_resr  http_ress
             r   r"  $AsyncQdrantRemote.query_batch_points  s     
 "	 "A a!4!455 00D "	   +':6;Q;Q'RSS(AA+N6:6F6F6Q6Q%%$3!)%0#	 $+#6DMM 7R 7 1H "	 )A $$LMHHUHSJ;;C@HU )	  ii**==$3 +#(.(@(@((S	 >   f  'D)DD'OC1 VsN   F*AFBF*FF*/$F" F3	F"<?F*;F(< F*F""F*group_by
group_sizewith_lookupc                 	  #    U R                   (       Ga4  Ub  [        R                  " U5      n[        U[        R
                  5      (       a  [        R                  " U5      /n[        U[        5      (       aG  U Vs/ s H:  n[        U[        R
                  5      (       a  [        R                  " U5      OUPM<     nn[        U[        R                  5      (       a  [        R                  " US9n[        U[        R                  5      (       a  [        R                  " U5      n[        U
[        [        R                  5      5      (       a  [        R                  " U
5      n
[        U[        [        R                  5      5      (       a  [        R                   " U5      n[        U[        R"                  5      (       a  [        R$                  " U5      n[        U[&        5      (       a  [(        R"                  " US9n[        U[        R*                  5      (       a  [        R,                  " U5      n[        U[        [        R.                  5      5      (       a  [        R0                  " U5      n[        U[        [        R2                  5      5      (       a  [        R4                  " U5      nU R6                  R9                  [(        R:                  " S0 SU_SU_SU_SU_SU_SU_S	U
_S
U_SU_SU_SU_SU	_SU_SU_SU_SU_SU_6Ub  UOU R<                  S9I S h  vN R>                  n[@        RB                  " U5      $ [        U[(        R                  5      (       a  [@        R                  " US9n[        U[(        R                  5      (       a  [@        R                  " U5      n[        U
[(        RD                  5      (       a  [@        RF                  " U
5      n
[        U[(        R*                  5      (       a  [@        R,                  " U5      n[        RH                  " UUUUUUUUUU	UU
UUS9nU RJ                  RL                  RO                  UUUUS9I S h  vN nUc   S5       eUR>                  $ s  snf  GNa N"7f)Nr   )
collectionr   r   r   r   r   r   r   r   r   r   r(  r)  r*  r   r1   r   r   r   )r   r   r   r   r   r   r   r   r(  r)  r   r   r*  r   )r   r   r1   query_groups_requestz%Query points groups API returned NonerL   )(rO   r   r   rn   r    r   r   r   r   r   r   r   r   r  r  r  r  
WithLookupconvert_with_lookupstrr   r  r  r  r  r	  r
  r   QueryGroupsQueryPointGroupsru   r  r   convert_groups_resultr  r  QueryGroupsRequestr9   r  query_points_groups)r   r   r(  r   r   r   r   r   r   r)  r   r   r   r*  r   r   r   r1   r7   r  r  r   r  s                          r   r5  %AsyncQdrantRemote.query_points_groups  s    <  "007(FOO44&==hGH(D)) &% =Gq&//<Z<ZJ55a8`aa%   ,66)88|L-)<)<== * @ @ O,(;F<W<W(XYY)HHV,(;F<M<M(NOO)>>|L+v'8'899(<<[I+s++"ooE+v'<'<==(@@M+':6;Q;Q'RSS(AA+N,.A&BYBY.Z[[%/%J%JK]%^"&&22)) (7# "*  ,	
 $ &2 &2  - )8 $ "* $. %0 %0 !(  ,>!" *5#& (/':G) 3  , f/ 0 33F;;,44)88|L-):):;; * @ @ O,(@(@AA)GGU+t':':;;(@@M"55,!#$ /!%()''M  "&!5!5!I!I /'%2	 "J " L  +T-TT+&&&_.bs3   A;R
=AR >I&R
$R%E R
%R& R
R
samplec	                   #    U R                   (       Ga  [        U[        R                  5      (       a  [        R
                  " US9n[        U[        [        R                  5      5      (       a  [        R                  " U5      n[        U[        [        R                  5      5      (       a  [        R                  " U5      nU R                  R                  [        R                  " UUUUUUUUS9Ub  UOU R                  S9I S h  vN n
[         R"                  " U
R$                  5      $ [        U[        R                  5      (       a  [         R
                  " US9nU R&                  R(                  R+                  UUU[        R,                  " UUUUUS9S9I S h  vN R$                  nUc   S5       eU$  N N7f)Nr   r   r   r7  r   r   r1   r   r   r   r   r   r7  r   r   r   r   r1   search_matrix_requestz(Search matrix pairs returned None result)rO   rn   r    r   r   r   r   r	  r
  r  r  r   SearchMatrixPairsr   SearchMatrixPointsru   r   convert_search_matrix_pairsr  rw   r  search_matrix_pairsSearchMatrixRequestr   r   r   r   r7  r   r   r   r1   r7   responsesearch_matrix_results               r   r@  %AsyncQdrantRemote.search_matrix_pairsx  s     ,66)88|L,.A&BYBY.Z[[%/%J%JK]%^"+':6;Q;Q'RSS(AA+N!--??''$3'!#%0'9	 $+#6DMM @  H 99(//JJlDKK00%44<HL%%00DD /'&,&@&@0!''	 E   & 	 $/[1[[/##?"%   DF?	F;
BF? F=!F?=F?c	                   #    U R                   (       Ga  [        U[        R                  5      (       a  [        R
                  " US9n[        U[        [        R                  5      5      (       a  [        R                  " U5      n[        U[        [        R                  5      5      (       a  [        R                  " U5      nU R                  R                  [        R                  " UUUUUUUUS9Ub  UOU R                  S9I S h  vN n
[         R"                  " U
R$                  5      $ [        U[        R                  5      (       a  [         R
                  " US9nU R&                  R(                  R+                  UUU[        R,                  " UUUUUS9S9I S h  vN R$                  nUc   S5       eU$  N N7f)Nr   r9  r   r:  r;  z*Search matrix offsets returned None result)rO   rn   r    r   r   r   r   r	  r
  r  r  r   SearchMatrixOffsetsr   r>  ru   r   convert_search_matrix_offsetsr  rw   r  search_matrix_offsetsrA  rB  s               r   rJ  'AsyncQdrantRemote.search_matrix_offsets  s     ,66)88|L,.A&BYBY.Z[[%/%J%JK]%^"+':6;Q;Q'RSS(AA+N!--AA''$3'!#%0'9	 $+#6DMM B  H ;;HOOLLlDKK00%44<HL%%00FF /'&,&@&@0!''	 G   & 	 $/]1]]/##?"rF  scroll_filterorder_byc                   #    U R                   (       Ga]  [        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        U[        R                  5      (       a  [
        R                  " US9n[        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        U	[        [        R                  5      5      (       a  [
        R                   " U	5      n	[        U[        [        R"                  5      5      (       a  [
        R$                  " U5      nU R&                  R)                  [*        R,                  " UUUUUUUUU	U
S9
U
b  U
OU R.                  S9I S h  vN nUR0                   Vs/ s H  n[2        R4                  " U5      PM     snUR7                  S5      (       a!  [2        R8                  " UR:                  5      4$ S 4$ [        U[*        R<                  5      (       a  [2        R8                  " U5      n[        U[*        R                  5      (       a  [2        R                  " US9n[        U[*        R>                  5      (       a  [2        R@                  " U5      n[        U[*        RB                  5      (       a  [2        RD                  " U5      nU RF                  RH                  RK                  UU[        RL                  " UUUUUUU	S9U
S9I S h  vN R0                  nUc   S5       eURN                  UR:                  4$  GNs  snf  N87f)Nr   )
r   r   rM  r   r   r   r   r   r   r1   r   next_page_offset)r   r   rM  r   r   r   r   )r   r   scroll_requestr1   z&Scroll points API returned None result)(rO   rn   r   r    ExtendedPointIdr   convert_extended_point_idr   r   r  r  r  r  r  r  r	  r
  OrderByInterfaceconvert_order_by_interfacer   Scrollr   ScrollPointsru   r  r   convert_retrieved_pointHasFieldconvert_point_idrO  PointIdOrderByconvert_order_byr  r  rw   
points_apiscroll_pointsScrollRequestr   )r   r   rL  r   rM  r   r   r   r   r   r1   r7   r  pointscroll_results                  r   scrollAsyncQdrantRemote.scroll  s     &"5f6L6L"MNN#==fE-77 * 9 9 N,(;F<W<W(XYY)HHV,(;F<M<M(NOO)>>|L+':6;Q;Q'RSS(AA+N,.A&BYBY.Z[[%/%J%JK]%^"($78O8O$PQQ%@@J-1-=-=-D-D!!$3(%!!-!-%0'9# $+#6DMM .E . (C  IL

S
u33E:
S<< 233 ++C,@,@A  	  &$,,//#44V<-55 * 9 9 N(DLL11%66x@,(@(@AA)GGU))44BB$3 +#)#7#7,#!)%%1$0"4$ $ C   f   !,V.VV,!((-*H*HII_(  Ts1   HO N6O  N96EO N>1O 9O count_filterexactc                 x  #    U R                   (       Ga  [        U[        R                  5      (       a  [        R
                  " US9n[        U[        [        R                  5      5      (       a  [        R                  " U5      n[        U[        [        R                  5      5      (       a  [        R                  " U5      nU R                  R                  [        R                  " UUUUUUS9Ub  UOU R                  S9I S h  vN R                   n["        R$                  " U5      $ [        U[        R                  5      (       a  ["        R
                  " US9nU R&                  R(                  R+                  U[        R,                  " X#US9UUS9I S h  vN R                   n	U	c   S5       eU	$  N N7f)Nr   )r   r   re  r   r1   r   r   )r   re  r   )r   count_requestr   r1   z!Count points returned None result)rO   rn   r    r   r   r   r   r	  r
  r  r  r   Countr   CountPointsru   r  r   convert_count_resultrw   r]  count_pointsCountRequest)
r   r   rd  re  r   r1   r   r7   rC  count_results
             r   countAsyncQdrantRemote.count.  s     ,66)88|L,.A&BYBY.Z[[%/%J%JK]%^"+':6;Q;Q'RSS(AA+N&&,,$$(7+#+= ')4 (/':G - 
 
 f  228<<lDKK00%44<HL%%00== /$11'@R ( >   & 	 'L)LL'5
 s%   DF:F6BF:F8F:8F:keyfacet_filterc	                   #    U R                   (       Ga<  [        U[        R                  5      (       a  [        R
                  " US9n[        U[        [        R                  5      5      (       a  [        R                  " U5      n[        U[        [        R                  5      5      (       a  [        R                  " U5      nU R                  R                  [        R                  " UUUUUUUUS9Ub  UOU R                  S9I S h  vN n
[         R"                  " U
R$                   Vs/ s H  n[&        R(                  " U5      PM     snS9$ [        U[        R                  5      (       a  [&        R
                  " US9nU R*                  R,                  R/                  UUU[        R0                  " UUUUUS9S9I S h  vN R2                  nUc   S5       eU$  Ns  snf  N!7f)Nr   )r   rp  r   r   re  r1   r   r   r   )hits)r   rp  r   r   re  )r   r   r1   facet_requestz!Facet points returned None result)rO   rn   r    r   r   r   r   r	  r
  r  r  r   Facetr   FacetCountsru   typesFacetResponsers  r   convert_facet_value_hitrw   r]  facetFacetRequestr  )r   r   rp  rq  r   re  r1   r   r   r7   rC  r  facet_results                r   rz  AsyncQdrantRemote.facet\  s     ,66)88|L,.A&BYBY.Z[[%/%J%JK]%^"+':6;Q;Q'RSS(AA+N!--33  $3'#%0'9	 $+#6DMM 4  H &&IQW#j88=W  lDKK00%44<HL%%0066 /'$110'	 7   & 	 'L)LL'C X
s1   DG'	G
"G', G A7G'G%G' G'r   waitorderingupdate_filterc                 <  #    U R                   (       GaW  [        U[        R                  5      (       a  [        R
                  " UR                  [        UR                  5      5      n[        [        UR                  5      5       V	s/ s Hl  n	[        R                  " [        R                  " UR                  U	   5      X   UR                  b#  [        R                  " UR                  U	   5      OS S9PMn     nn	[        U[        5      (       aG  U V
s/ s H:  n
[        U
[        R                  5      (       a  [        R                   " U
5      OU
PM<     nn
[        U[        R"                  5      (       a  [        R$                  " U5      n[        U['        [        R(                  5      5      (       a  [        R*                  " U5      n[        U[        R,                  5      (       a  [        R.                  " US9nU R0                  R3                  [        R4                  " UUUUUUS9U R6                  S9I S h  vN R8                  nUc   S5       e[:        R<                  " U5      $ [        U[        R,                  5      (       a  [:        R.                  " US9n[        U[        5      (       a\  U V
s/ s H:  n
[        U
[        R                  5      (       a  [:        R                   " U
5      OU
PM<     nn
[        R>                  " X%US9n[        U[        R                  5      (       a  [        R@                  " X%US9nU RB                  RD                  RG                  UUUUS9I S h  vN R8                  nUc   S5       eU$ s  sn	f s  sn
f  GNUs  sn
f  N,7f)	N)idvectorspayloadr   r   r~  r   r  r   r  r   Upsert returned None resultr   r   r  )batchr   r  )r   r~  point_insert_operationsr  )$rO   rn   r    Batchr   convert_batch_vector_structr  rW   idsr   r   PointStructrR  payloadsconvert_payloadr   convert_point_structWriteOrderingconvert_write_orderingr   r	  r
  r   r   r   UpsertUpsertPointsru   r  r   convert_update_result
PointsListPointsBatchrw   r]  upsert_points)r   r   r   r~  r  r   r  r7   vectors_batchidxr`  grpc_resulthttp_results                r   upsertAsyncQdrantRemote.upsert  s:     &&,,//4>4Z4ZNNC

O5  %S_5	  6 $$%??

3P - 2!??6 !+ : :6??3;O P!  6  	 &$''
 "(	 "( "%););<< 33E: "(	   (F$8$899%<<XF,.A&BYBY.Z[[%/%J%JK]%^"-77 * 9 9 N&&--%%(7!%!)+=&3 !MM . 
 
 f  *I,II*33K@@-55 * 9 9 N&$''
 "(	 "( "%)9)9:: 33E: "(	    **!} &&,,//++ m ))44BB$3,2%	 C   f  *I,II*A	
$sN   BNA3N;NANC&N=N>A:N8AN9A4N-N.%NNc                   #    U R                   (       Ga9  U Vs/ s H  n[        R                  " U5      PM     nn[        U[        R
                  5      (       a  [        R                  " U5      n[        U[        [        R                  5      5      (       a  [        R                  " U5      n[        U[        R                  5      (       a  [        R                  " US9nU R                  R                  [        R                  " UUUUUUS9U R                   S9I S h  vN R"                  n	U	c   S5       e[$        R&                  " U	5      $ [        U[        R                  5      (       a  [$        R                  " US9nU R(                  R*                  R-                  UU[        R                  " X%US9US9I S h  vN R"                  $ s  snf  N N7f)Nr   r  r   r  r  )r   r~  update_vectorsr  )rO   r   convert_point_vectorsrn   r    r  r  r   r	  r
  r   r   r   UpdateVectorsr   UpdatePointVectorsru   r  r   r  rw   r]  r  )
r   r   r   r~  r  r   r  r7   r`  r  s
             r   r   AsyncQdrantRemote.update_vectors  s     KQR6%j66u=6FR(F$8$899%<<XF,.A&BYBY.Z[[%/%J%JK]%^"-77 * 9 9 N&&44++(7!%!)+=&3 !MM 5 
 
 f  *I,II*33K@@-55 * 9 9 N))44CC$3#)#7#7%S`$ & D   f	3 S
$s/   G GC&GG BG=G>GGr  c                   #    U R                   (       Ga  U R                  U5      u  pU=(       d    U	n[        U[        R                  5      (       a  [
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      nU R                  R                  [        R                  " UU[        R                  " US9UUUS9U R                  S9I S h  vN R                   n
U
c   S5       e["        R$                  " U
5      $ U R'                  U5      u  pU R(                  R*                  R-                  UUU[/        [        R                  UUUUS9S9I S h  vN R                   $  N N7f)N)names)r   r~  r  points_selectorr  r   r   z#Delete vectors returned None result)vectorr   r   r   )r   r~  r  delete_vectors)rO   _try_argument_to_grpc_selectorrn   r    r  r   r  r   r	  r
  r   DeleteVectorsr   DeletePointVectorsVectorsSelectorru   r  r   r  '_try_argument_to_rest_points_and_filterrw   r]  r  r   )r   r   r  r   r~  r  r   r7   r  opt_shard_key_selectorr  _points_filters                r   r   AsyncQdrantRemote.delete_vectors  sy     8<8[8[\b8c5_!3!M7M(F$8$899%<<XF,.A&BYBY.Z[[%/%J%JK]%^"&&44++(7! $ 4 47 C(7!)+= !MM 5 
 
 f  *Q,QQ*33K@@!%!M!Mf!UW))44CC$3%#,,,&&&"4$	 D   f!
"s%   C4F6F	7BF:F;FFr  c                 H  #    U R                   (       Ga  [        U[        [        R                  5      5      (       a  [
        R                  " U5      nU V	s/ s HC  n	[        U	[        [        R                  5      5      (       a  [
        R                  " U	5      OU	PME     nn	[
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      nU R                  R                  [         R"                  " UUUUUUUS9Ub  UOU R$                  S9I S h  vN R&                  n
U
c   S5       eU
 Vs/ s H  n[(        R*                  " U5      PM     sn$ [        U[         R,                  5      (       a  [(        R.                  " U5      nU V	s/ s H:  n	[        U	[         R0                  5      (       a  [(        R2                  " U	5      OU	PM<     nn	U R4                  R6                  R9                  UU[        R:                  " UUUUS9US9I S h  vN R&                  nUc   S5       eU$ s  sn	f  GNs  snf s  sn	f  N,7f)N)r   r  r   r   r   r   r1   r   zRetrieve returned None result)r  r   r   r   )r   r   point_requestr1   z!Retrieve API returned None result)rO   rn   r   r    r  r   r  rQ  rR  r  r  r  r	  r
  r   Getr   	GetPointsru   r  r   rW  r  r  rZ  rY  rw   r]  
get_pointsPointRequest)r   r   r  r   r   r   r   r1   r7   r  r  recordr  s                r   retrieveAsyncQdrantRemote.retrieveC  sy     ,(;F<W<W(XYY)HHV
 	 C c#6v7M7M#NOO 44S9 	   &::<HL+':6;Q;Q'RSS(AA+N,.A&BYBY.Z[[%/%J%JK]%^"&&**NN(7%1%1)4+= ' (/':G +   f  %F'FF%MSTV6J66v>VTT,(@(@AA)GGU C 5?sDLL4Q4Q
++C0WZZ  
 ))44??$3 +"("5"5%1$0"4	# $ @ 
 
 f  *O,OO*a U

sL   AJ"A
J!CJ"9J:J" J6<J"2AJ3A J"3J 4 J"J"c                    S n[        U[        5      (       ax  [        R                  " [        R                  " U Vs/ s HC  n[        U[        [        R                  5      5      (       a  [        R                  " U5      OUPME     snS9S9nXB4$ [        U[        R                  5      (       a  UnXB4$ [        U[        [        R                  5      5      (       aF  UR                  b   [        R                  " UR                  5      n[        R                  " U5      nXB4$ [        U[        R                  5      (       a/  [        R                  " [        [        R                   US95      nXB4$ [        U[        R                  5      (       a  [        R                  " US9nXB4$ [#        S[%        U5       35      es  snf )N)r  r   )r   "Unsupported points selector type: )rn   r   r   PointsSelectorPointsIdsListr   r    rQ  r   rR  r   r   r
  convert_points_selectorr   r   FilterSelectorrX   rp   )clsr   r   r  r  s        r   r  0AsyncQdrantRemote._try_argument_to_grpc_selector  s    "fd##"11))
 $*	 $*C &c+>v?U?U+VWW #<<SA ! $*		O0  44  3 344$O  44 )>)> ?@@+%/%J%J6K[K[%\"(@@HO  44 ..(@@&//?O  44	 ,,"11@O  44 A$v,PQQ+s   A
F=c                    [        U[        5      (       aa  U Vs/ s H:  n[        U[        R                  5      (       a  [        R
                  " U5      OUPM<     nn[        [        R                  XBS9nU$ [        U[        R                  5      (       a  [        R                  " U5      nX%l        U$ [        U[        [        R                  5      5      (       a
  UnX%l        U$ [        U[        R                  5      (       a  [        [        R                  XS9nU$ [        U[        R                  5      (       a/  [        [        R                  [        R                  " U5      US9nU$ [!        S[#        U5       35      es  snf )N)r   r   )r   r   r  )rn   r   r   rZ  r   rY  r   r    PointIdsListr  r  r   r   r   r  r   rX   rp   )r  r   r   r  r  r  s         r   _try_argument_to_rest_selector0AsyncQdrantRemote._try_argument_to_rest_selector  s`    fd## "!C 5?sDLL4Q4Q
++C0WZZ!   (##GO* %  3 344(@@HO(:%   )>)> ?@@$O(:%  ..'%%fO  ,,'%%!008,O  A$v,PQQ1s   AFr  c                 ~    UR                  S5      nUc  / $ [        X5      nUS:X  a  [        UR                  5      $ / $ )Npoints_selector_one_ofr   )
WhichOneofgetattrr   r  )r  r  namevals       r   _points_selector_to_points_list1AsyncQdrantRemote._points_selector_to_points_list  sD     ))*BC<Io,8= 	r   c                    S nS n[        U[        5      (       aJ  U Vs/ s H:  n[        U[        R                  5      (       a  [        R
                  " U5      OUPM<     nnX#4$ [        U[        R                  5      (       ar  [        R                  " U5      n[        U[        R                  5      (       a  UR                  nX#4$ [        U[        R                  5      (       a  UR                  nX#4$ [        U[        R                  5      (       a  UR                  nX#4$ [        U[        R                  5      (       a  UR                  nX#4$ [        U[        R                  5      (       a  UnX#4$ [        U[        R                  5      (       a  [        R                  " U5      nX#4$ [        S[!        U5       35      es  snf )Nr  )rn   r   r   rZ  r   rY  r  r  r    r  r   r  r   r   r   rX   rp   )r  r   r  r  r  selectors         r   r  9AsyncQdrantRemote._try_argument_to_rest_points_and_filter  s    fd## "!C 5?sDLL4Q4Q
++C0WZZ!  ( !!!  3 344!99&AH(F$7$788"// !! Hf&;&;<<"// !!  3 344mmG !!  5 566mmG !! ..G
 !!	 ,, //7G !! A$v,PQQ's   AGc                   #    U R                   (       a  U R                  U5      u  p'U=(       d    Un[        U[        R                  5      (       a  [
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        R                  " U R                  R                  [        R                  " UUUUUS9U R                   S9I S h  vN R"                  5      $ U R%                  X%5      nU R&                  R(                  R+                  UUUUS9I S h  vN R"                  nUc   S5       eU$  Ng N7f)Nr   r~  r   r  r   r   )r   r~  r  r  zDelete points returned None)rO   r  rn   r    r  r   r  r   r	  r
  r   r  r   Deleter   DeletePointsru   r  r  rw   r]  delete_points	r   r   r  r~  r  r   r7   r  r  s	            r   deleteAsyncQdrantRemote.delete  sX     8<8[8[95_ "4!M7M(F$8$899%<<XF,.A&BYBY.Z[[%/%J%JK]%^"33**11)),;!%#2%-/A !% 2 	 	 &  #AAO ))44BB$3$3%	 C   f  %D'DD%M3	"%   C0E2E3AE?E EEr  c                   #    U R                   (       Ga  U R                  U5      u  pU=(       d    U
n[        U[        R                  5      (       a  [
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        R                  " U R                  R                  [        R                  " UU[
        R                   " U5      U	UUUS9U R"                  S9I S h  vN R$                  5      $ U R'                  U5      u  pU R(                  R*                  R-                  UUU[        R                  " UUUUUS9S9I S h  vN R$                  nUc   S5       eU$  N N7f)N)r   r~  r  r  r  r   rp  r   )r  r   r   r   rp  r   r~  r  set_payloadzSet payload returned None)rO   r  rn   r    r  r   r  r   r	  r
  r   r  r   
SetPayloadr   SetPayloadPointsr  ru   r  r  rw   r]  r  )r   r   r  r   rp  r~  r  r   r7   r  r  r  r  r  s                 r   r  AsyncQdrantRemote.set_payload  s     8<8[8[\b8c5_!3!M7M(F$8$899%<<XF,.A&BYBY.Z[[%/%J%JK]%^"33**55--,;!%$.$>$>w$G,;%-/A # !% 6   & " "&!M!Mf!UW))44@@$3% & 1 1 '&&"4!	 A   f  %B'BB%M?"s%   DF	F	
A$F.F/FFc                   #    U R                   (       Ga  U R                  U5      u  pU=(       d    U	n[        U[        R                  5      (       a  [
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        R                  " U R                  R                  [        R                  " UU[
        R                   " U5      UUUS9U R"                  S9I S h  vN R$                  5      $ U R'                  U5      u  pU R(                  R*                  R-                  UUU[        R.                  " UU
UUS9S9I S h  vN R$                  nUc   S5       eU$  N~ N7f)N)r   r~  r  r  r  r   r   )r  r   r   r   r  zOverwrite payload returned None)rO   r  rn   r    r  r   r  r   r	  r
  r   r  r   OverwritePayloadr   r  r  ru   r  r  rw   r]  overwrite_payloadr  )r   r   r  r   r~  r  r   r7   r  r  r  r  r  s                r   r  #AsyncQdrantRemote.overwrite_payloadP  sz     8<8[8[\b8c5_!3!M7M(F$8$899%<<XF,.A&BYBY.Z[[%/%J%JK]%^"33**;;--,;!%$.$>$>w$G,;%-/A !% < 
 
 &   "&!M!Mf!UW))44FF$3% & 1 1 '&&"4	!	 G 
 
 f  %H'HH%M;
 
s%   DFF	A#F,F	-F	Fkeysc                   #    U R                   (       a  U R                  U5      u  pU=(       d    U	n[        U[        R                  5      (       a  [
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        R                  " U R                  R                  [        R                  " UUUUUUS9U R                   S9I S h  vN R"                  5      $ U R%                  U5      u  pU R&                  R(                  R+                  UUU[        R                  " X*XS9S9I S h  vN R"                  nUc   S5       eU$  N| N7f)N)r   r~  r  r  r  r   r   )r  r   r   r   )r   r~  r  delete_payloadzDelete payload returned None)rO   r  rn   r    r  r   r  r   r	  r
  r   r  r   DeletePayloadr   DeletePayloadPointsru   r  r  rw   r]  r  )r   r   r  r   r~  r  r   r7   r  r  r  r  r  s                r   r   AsyncQdrantRemote.delete_payload  se     8<8[8[\b8c5_!3!M7M(F$8$899%<<XF,.A&BYBY.Z[[%/%J%JK]%^"33**8800,;!%!%,;%-/A !% 9 
 
 &   "&!M!Mf!UW))44CC$3%#)#7#7!'$	 D   f  %E'EE%M5
 s%   C1E43E04A!E4E2E42E4c                   #    U R                   (       a  U R                  U5      u  p'U=(       d    Un[        U[        R                  5      (       a  [
        R                  " U5      n[        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        R                  " U R                  R                  [        R                  " UUUUUS9U R                   S9I S h  vN R"                  5      $ U R%                  X%5      nU R&                  R(                  R+                  UUUUS9I S h  vN R"                  nUc   S5       eU$  Ng N7f)Nr  r   )r   r~  r  r  zClear payload returned None)rO   r  rn   r    r  r   r  r   r	  r
  r   r  r   ClearPayloadr   ClearPayloadPointsru   r  r  rw   r]  clear_payloadr  s	            r   r  AsyncQdrantRemote.clear_payload  sX     8<8[8[95_ "4!M7M(F$8$899%<<XF,.A&BYBY.Z[[%/%J%JK]%^"33**77//,;!%#2%-/A !% 8 	 	 &  #AAO ))44BB$3%$3	 C   f  %D'DD%M3	"r  update_operationsc           	        #    U R                   (       a  U Vs/ s H  n[        R                  " U5      PM     nn[        U[        R
                  5      (       a  [        R                  " U5      nU R                  R                  [        R                  " UUUUS9U R                  S9I S h  vN R                   Vs/ s H  n[        R                  " U5      PM     sn$ U R                  R                   R#                  UUU[        R$                  " US9S9I S h  vN R                  nUc   S5       eU$ s  snf  Ns  snf  N&7f)N)r   r~  
operationsr  r   )r  )r   r~  r  r  z!Batch update points returned None)rO   r   convert_update_operationrn   r    r  r  r   UpdateBatchr   UpdateBatchPointsru   r  r   r  rw   r]  batch_updateUpdateOperations)r   r   r  r~  r  r7   	operationr  s           r   batch_update_points%AsyncQdrantRemote.batch_update_points  sP     Pa!Pa9
33I>Pa  ! (F$8$899%<<XF **66..,;!%'8%-	 !% 7   &

F 008
   ))44AA$3%&,&=&=IZ&[	 B   f  %J'JJ%M;! s:   E D5A3E+D:,E< D<>EE E<Echange_aliases_operationsc                   #    U R                   (       a  U Vs/ s H:  n[        U[        R                  5      (       d  [        R
                  " U5      OUPM<     nnU R                  R                  [        R                  " X%S9Ub  UOU R                  S9I S h  vN R                  $ U Vs/ s H:  n[        U[        R                  5      (       a  [        R
                  " U5      OUPM<     nnU R                  R                  R                  U[        R                   " US9S9I S h  vN R                  nUc   S5       eU$ s  snf  Ns  snf  N&7f)N)r1   actionsr   )r  )r1   change_aliases_operationzUpdate aliases returned None)rO   rn   r   AliasOperationsr   convert_alias_operationsr   UpdateAliasesChangeAliasesru   r  r   r9   aliases_apiupdate_aliasesr    ChangeAliasesOperation)r   r  r1   r7   r  r  r  s          r   update_collection_aliases+AsyncQdrantRemote.update_collection_aliases  se     
 ";	( ";I ")T-A-ABB 33I> ";	 % ( ++99&&wY'.':G :   f 7	$
 7	 )T%9%9:: //	: 7	 	! $
 ))''66)/)F)F4* 7   & 	 !A#AA!7(
$
s=   EAEA EEE,AE
-;E(E) E
Ec                   #    U R                   (       a  U R                  R                  [        R                  " US9U R
                  S9I S h  vN R                  n[        R                  " U Vs/ s H  n[        R                  " U5      PM     snS9$ U R                  R                  R                  US9I S h  vN R                  nUc   S5       eU$  Ns  snf  N!7f)Nr   r   aliasesz$Get collection aliases returned None)rO   r   ListCollectionAliasesr   ListCollectionAliasesRequestru   r  rw  CollectionsAliasesResponser   convert_alias_descriptionr9   r  get_collection_aliasesr  )r   r   r7   rC  descriptionr  s         r   r  (AsyncQdrantRemote.get_collection_aliases,  s      ++AA55oV MM B   g  33YaYa+J88EYa  ))''>>>__
& 	 !I#II!
 `s0   ACC"C0 C+C;C<CCc                   #    U R                   (       a  U R                  R                  [        R                  " 5       U R
                  S9I S h  vN R                  n[        R                  " U Vs/ s H  n[        R                  " U5      PM     snS9$ U R                  R                  R                  5       I S h  vN R                  nUc   S5       eU$  Ns  snf  N!7f)Nr   r  zGet aliases returned None)rO   r   ListAliasesr   ListAliasesRequestru   r  rw  r  r   r  r9   r  get_collections_aliasesr  r   r7   rC  r  r  s        r   get_aliasesAsyncQdrantRemote.get_aliasesA  s     ++77++-t}} 8   g	 
 33YaYa+J88EYa  ))''??AA
& 	 !>#>>!

 B0   AC!C"C!1 C,C!=C>C!C!c                   #    U R                   (       a  U R                  R                  [        R                  " 5       U R
                  S9I S h  vN R                  n[        R                  " U Vs/ s H  n[        R                  " U5      PM     snS9$ U R                  R                  R                  5       I S h  vN R                  nUc   S5       eU$  Ns  snf  N!7f)Nr   )collectionszGet collections returned None)rO   r   Listr   ListCollectionsRequestru   r  rw  CollectionsResponser   convert_collection_descriptionr9   collections_apiget_collectionsr  r  s        r   r#  !AsyncQdrantRemote.get_collectionsS  s     ++00//14== 1   k	 
 ,, (0'/ ==kJ'/  ))++;;==
& 	 !B#BB!
 >r  c                 n  #    U R                   (       a\  [        R                  " U R                  R	                  [
        R                  " US9U R                  S9I S h  vN R                  5      $ U R                  R                  R                  US9I S h  vN R                  nUc   S5       eU$  NS N7f)Nr
  r   zGet collection returned None)rO   r   convert_collection_infor   r  r   GetCollectionInfoRequestru   r  r9   r"  get_collectionr   r   r7   r  s       r   r(   AsyncQdrantRemote.get_collectionf  s     55//3355oV $ 4   &  ))++::?:[[
& 	 !A#AA! \s$   AB5B18B5B3B53B5c                 n  #    U R                   (       aR  U R                  R                  [        R                  " US9U R
                  S9I S h  vN R                  R                  $ U R                  R                  R                  US9I S h  vN R                  nUc   S5       eUR                  $  Nc N&7f)Nr
  r   zCollection exists returned None)rO   r   CollectionExistsr   CollectionExistsRequestru   r  existsr9   r"  collection_existsr)  s       r   r/  #AsyncQdrantRemote.collection_existsv  s     ++<<00Q MM =   fVV ))++==o=^^
& 	 !D#DD!}} _s$   AB5B1>B5B3%B53B5optimizers_configcollection_paramsvectors_confighnsw_configquantization_configsparse_vectors_configstrict_mode_configrB   c                   #    U R                   (       Ga  [        U[        R                  5      (       a  [        R
                  " U5      n[        U[        R                  5      (       a  [        R                  " U5      n[        U[        5      (       a  [        R                  " U5      n[        U[        R                  5      (       a  [        R                  " U5      n[        U[        [        R                  5      5      (       a  [        R                  " U5      n[        U[        5      (       a  [        R                  " U5      n[        U	[        R                   5      (       a  [        R"                  " U	5      n	[        U
[        5      (       a  [        R$                  " U
5      n
U R&                  R)                  [*        R,                  " UUUUUUUU	UU
S9
Ub  UOU R.                  S9I S h  vN R0                  $ [        U[*        R                  5      (       a  [2        R
                  " U5      n[        U[*        R                  5      (       a  [2        R                  " U5      n[        U[*        R4                  5      (       a  [2        R                  " U5      n[        U[*        R                  5      (       a  [2        R                  " U5      n[        U[*        R                  5      (       a  [2        R                  " U5      nU R6                  R8                  R;                  U[        R,                  " UUUUUUU	U
S9US9I S h  vN R0                  nUc   S5       eU$  GNt N7f)N)
r   r1  r   r3  r4  r5  r6  r7  r1   rB   r   )r1  r   r  r4  r5  sparse_vectorsr7  rB   )update_collectionr1   zUpdate collection returned None)rO   rn   r    OptimizersConfigDiffr   convert_optimizers_config_diffCollectionParamsDiffconvert_collection_params_diffdictconvert_vectors_config_diffHnswConfigDiffconvert_hnsw_config_diffr   QuantizationConfigDiff convert_quantization_config_diffconvert_sparse_vector_configStrictModeConfigconvert_strict_mode_configr  r   Updater   UpdateCollectionru   r  r   VectorsConfigDiffr9   r"  r:  )r   r   r1  r2  r3  r4  r5  r1   r6  r7  rB   r7   r  s                r   r:  #AsyncQdrantRemote.update_collection  s     +V-H-HII$.$M$MN_$`!+V-H-HII$.$M$MN_$`!.$//!+!G!G!W+v'<'<==(AA+N-x8U8U/VWW&0&Q&Q''# /66(2(O(O))% ,f.E.EFF%/%J%JK]%^"(D))%55h?++22))(7*;0'5$/,?.C+= '!) (/':G 3   f!" ')B)BCC * I IJ[ \')B)BCC * I IJ[ \nd&<&<=='CCNSNk4#6#677$==kJK)4+F+FGG","M"MNa"b))++=="("9"9&7,* +(;#8'9%	#   >   & 	  !D#DD!W6s%   G-M(/M#0EM(M&	M(&M(c                 P  #    U R                   (       aM  U R                  R                  [        R                  " XS9Ub  UOU R
                  S9I S h  vN R                  $ U R                  R                  R                  XS9I S h  vN R                  nUc   S5       eU$  NO N7f)Nr   r1   r   zDelete collection returned None)
rO   r   r  r   DeleteCollectionru   r  r9   r"  delete_collection)r   r   r1   r7   r  s        r   rO  #AsyncQdrantRemote.delete_collection  s      ++22))/['.':G 3   f ))++==o=__
& 	 !D#DD! `s$   AB&B"4B&B$B&$B&shard_numberreplication_factorwrite_consistency_factoron_disk_payload
wal_configsharding_methodc                   #    U R                   (       Ga0  [        U[        R                  [        45      (       a  [
        R                  " U5      n[        U[        R                  5      (       a  [
        R                  " U5      n[        U[        R                  5      (       a  [
        R                  " U5      n[        U	[        R                  5      (       a  [
        R                  " U	5      n	[        U
[        [        R                  5      5      (       a  [
        R                  " U
5      n
[        U[        5      (       a  [
        R                   " U5      n[        U[        R"                  5      (       a  [
        R$                  " U5      n[        U[        R&                  5      (       a  [
        R(                  " U5      n[        U[        5      (       a  [
        R*                  " U5      n[,        R.                  " UUU	UUUUUUUU
UUUUS9nU R0                  R3                  UU R4                  S9I S h  vN R6                  $ [        U[,        R                  5      (       a  [8        R                  " U5      n[        U[,        R                  5      (       a  [8        R                  " U5      n[        U	[,        R                  5      (       a  [8        R                  " U	5      n	[        U
[,        R                  5      (       a  [8        R                  " U
5      n
[        R.                  " UUUUUUUU	U
UUUUS9nU R:                  R<                  R?                  UUUS9I S h  vN R6                  nUc   S5       eU$  GNF N7f)N)r   r4  rU  r1  rQ  rT  r1   r3  rR  rS  r5  r6  rV  r7  rB   r   )r  rQ  rR  rS  rT  r4  r1  rU  r5  r9  rV  r7  rB   )r   create_collectionr1   zCreate collection returned None) rO   rn   r    VectorParamsr?  r   convert_vectors_configrA  rB  r;  r<  WalConfigDiffconvert_wal_config_diffr   QuantizationConfigconvert_quantization_configrE  ShardingMethodconvert_sharding_methodrF  rG  r  r   CreateCollectionr   Createru   r  r   r9   r"  rX  )r   r   r3  rQ  rR  rS  rT  r4  r1  rU  r5  r1   r6  rV  r7  rB   r7   rX  create_collection_requestr  s                       r   rX  #AsyncQdrantRemote.create_collection  s    & .6+>+>*EFF!+!B!B>!R+v'<'<==(AA+N+V-H-HII$.$M$MN_$`!*f&:&:;;'??
K
-x8Q8Q/RSS&0&L&LM`&a#/66(2(O(O))% /6+@+@AA","D"D_"U,f.E.EFF%/%J%JK]%^"(D))%55h? $ 5 5 /'%"3) /-#5)A$7&; /#5!!$ ++223Ddmm2\\f k4#6#677$==kJK')B)BCC * I IJ[ \j$"4"455#;;JGJ)4+B+BCC","H"HI\"]$*$;$;"%1%=+#/! 30+1%
!  ))++== /"; >  
 & 	 !D#DD!C ]4s%   H4N6M<7D*N!M?"N?Nc                    #    U R                  XS9I S h  vN   U R                  UUUUUUUUU	U
UUUUUS9I S h  vN $  N) N7f)Nr   )r   r3  rQ  rR  rS  rT  r4  r1  rU  r5  r1   r6  rV  r7  rB   )rO  rX  )r   r   r3  rQ  rR  rS  rT  r4  r1  rU  r5  r1   r6  rV  r7  rB   r7   s                    r   recreate_collection%AsyncQdrantRemote.recreate_collectionB  ss     & $$_$FFF+++)%1%=+#/! 3"7+1 , 
 
 	
 	G
s   A?$AAAAc                 <    U R                   (       a  [        $ [        $ r   )rO   r"   r#   r   s    r   _updater_class AsyncQdrantRemote._updater_classh  s    $$$$r   batches_iteratormax_retriesparallelmethodc	                 J   Ub*  U[        5       ;   a  Un	O+[        SU S[        5        35      eS[        5       ;   a  SOSn	U R                  (       aK  UU R                  U R                  UU R
                  U R                  UUU R                  U R                  US.n
OUU R                  UUUUS.U R                  En
US:X  a5  U R                  R                  " S	0 U
D6nUR                  U5       H  nM     g [        X@R                  U	S9nUR                  " U40 U
D6 H  nM     g )
NzStart methods z& is not available, available methods: 
forkserverspawn)r   r2   r+   rl  r   rB   r~  r   r   r1   r  )r   urirl  r~  r   r  r:   )start_methodrL   )r   rX   rO   rZ   rP   rR   re   rQ   ru   rs   rt   ri  startprocessr!   unordered_map)r   rk  r   rl  rm  rn  r~  r   r  rs  updater_kwargsupdaterr   pools                 r   _upload_collection$AsyncQdrantRemote._upload_collectiono  s:    .00% $VH,RShSjRkl  ,8;P;R+R<X_L#2

*{{ ..&8--==!.N $3}}*&8!. //N q=))//A.AG__%56 7 &h0C0CR^_D''(8KNK Lr   
batch_sizec
                 b    U R                   R                  X#S9nU R                  UUUUUUUU	S9  g )N)recordsr|  rk  r   rl  rm  rn  r~  r   r  )ri  iterate_records_batchesrz  )r   r   r   r|  rm  rn  rl  r~  r   r  r7   rk  s               r   upload_pointsAsyncQdrantRemote.upload_points  sQ      ..FF G 
 	-+#1' 	  		
r   c                 d    U R                   R                  X#XES9nU R                  UUUUUU	U
US9  g )N)r  r  r  r|  r  )ri  iterate_batchesrz  )r   r   r  r  r  r|  rm  rn  rl  r~  r   r  r7   rk  s                 r   upload_collection#AsyncQdrantRemote.upload_collection  sR      ..>># ? 
 	-+#1' 	  		
r   
field_namefield_schema
field_typec           	      D  #    Ub  [        S[        SSS9  UnU R                  (       Ga<  S n[        U[        R
                  5      (       a  [        R                  " U5      n[        U[        5      (       a*  [        R                  " [        R
                  " U5      5      n[        U[        5      (       a  [        U5      n[        U[        [        R                  5      5      (       a  [        R                  " U5      n[        U[        R                  5      (       Ga  UnUR!                  S5      n	[#        X5      n
[        U
[        R$                  5      (       a  [        R&                  R(                  n[        U
[        R*                  5      (       a  [        R&                  R,                  n[        U
[        R.                  5      (       a  [        R&                  R0                  n[        U
[        R2                  5      (       a  [        R&                  R4                  n[        U
[        R6                  5      (       a  [        R&                  R8                  n[        U
[        R:                  5      (       a  [        R&                  R<                  n[        U
[        R>                  5      (       a  [        R&                  R@                  n[        U
[        RB                  5      (       a  [        R&                  RD                  n[        RF                  " UUUUUUS9n[H        RJ                  " U RL                  RO                  XRP                  S9I S h  vN RR                  5      $ [        U[        5      (       a  [H        R                  " U5      n[        U[        R                  5      (       a  [H        R                  " U5      nU RT                  RV                  RY                  U[        RN                  " X#S9UUS	9I S h  vN RR                  nUc   S
5       eU$  N N7f)Nz2field_type is deprecated, use field_schema insteadr(   zpayload-index-field-type)rE   rF   rG   r  index_params)r   r  r  field_index_paramsr~  r  r   )r  r  )r   create_field_indexr~  r  z Create field index returned None)-r   DeprecationWarningrO   rn   r    PayloadSchemaTyper   convert_payload_schema_typer0  intr   r   PayloadSchemaParamsconvert_payload_schema_paramsr   PayloadIndexParamsr  r  TextIndexParams	FieldTypeFieldTypeTextIntegerIndexParamsFieldTypeIntegerKeywordIndexParamsFieldTypeKeywordFloatIndexParamsFieldTypeFloatGeoIndexParamsFieldTypeGeoBoolIndexParamsFieldTypeBoolDatetimeIndexParamsFieldTypeDatetimeUuidIndexParamsFieldTypeUuidCreateFieldIndexCollectionr   r  r   CreateFieldIndexru   r  rw   indexes_apir  )r   r   r  r  r  r~  r  r7   r  r  r  requestr  s                r   create_payload_index&AsyncQdrantRemote.create_payload_index  s     !L+.	 &L!%,(@(@AA)EElS,,,)EE,,\:  ,,,@N,1K1K(LMM)GGU,(?(?@@%1")44^D&'9@lD,@,@AA#'>>#?#?LlD,C,CDD#'>>#B#BLlD,C,CDD#'>>#B#BLlD,A,ABB#'>>#@#@LlD,?,?@@#'>>#>#>LlD,@,@AA#'>>#?#?LlD,D,DEE#'>>#C#CLlD,@,@AA#'>>#?#?L55 /%'#5!G 33''88--8XX``  lC((%AA,OLlD$;$;<<%CCLQL%%11DD /#)#:#:)$ ! E   & 	 !E#EE!# Ys%   MP PB-P PP P c                 |  #    U R                   (       a`  [        R                  " UUUUS9n[        R                  " U R
                  R                  X`R                  S9I S h  vN R                  5      $ U R                  R                  R                  UUUUS9I S h  vN R                  nUc   S5       eU$  NV N7f)N)r   r  r~  r  r   z Delete field index returned None)rO   r   DeleteFieldIndexCollectionr   r  r   DeleteFieldIndexru   r  rw   r  delete_field_index)r   r   r  r~  r  r7   r  r  s           r   delete_payload_index&AsyncQdrantRemote.delete_payload_index*  s      55 /%!	G 33''88--8XX``  %%11DD /%!	 E   & 	 !E#EE! Ys$   AB<!B8";B<B:B<:B<c                   #    U R                   (       ao  U R                  R                  [        R                  " US9U R
                  S9I S h  vN R                  nU Vs/ s H  n[        R                  " U5      PM     sn$ U R                  R                  R                  US9I S h  vN R                  nUc   S5       eU$  Nvs  snf  N!7f)Nr
  r   'List snapshots API returned None result)rO   r   r  r   ListSnapshotsRequestru   snapshot_descriptionsr   convert_snapshot_descriptionrw   snapshots_apilist_snapshotsr  )r   r   r7   	snapshotssnapshots        r   r   AsyncQdrantRemote.list_snapshotsG  s      ))..--oN MM /   $#  W``V_(J;;HEV_``%%33BBSbBcc
& 	 $O&OO$
 acs0   ACCC  C )C)C*CCc                 Z  #    U R                   (       a^  U R                  R                  [        R                  " US9U R
                  S9I S h  vN R                  n[        R                  " U5      $ U R                  R                  R                  XS9I S h  vN R                  $  NY N7f)Nr
  r   )r   r~  )rO   r   rb  r   CreateSnapshotRequestru   snapshot_descriptionr   r  rw   r  create_snapshotr  )r   r   r~  r7   r  s        r   r  !AsyncQdrantRemote.create_snapshotX  s      ))00..O MM 1   #"  ::8DD%%33CC / D   &		s%   AB+B'A
B+B)B+)B+snapshot_namec                   #    U R                   (       a?  U R                  R                  [        R                  " XS9U R
                  S9I S h  vN   gU R                  R                  R                  XUS9I S h  vN R                  $  N; N7f)N)r   r  r   T)r   r  r~  )
rO   r   r  r   DeleteSnapshotRequestru   rw   r  delete_snapshotr  )r   r   r  r~  r7   s        r   r  !AsyncQdrantRemote.delete_snapshoti  s      %%,,**$3 	 -    %%33CC /SW D   &		s$   ABB	,B:B;BBc                   #    U R                   (       ap  U R                  R                  [        R                  " 5       U R
                  S9I S h  vN R                  nU Vs/ s H  n[        R                  " U5      PM     sn$ U R                  R                  R                  5       I S h  vN R                  nUc   S5       eU$  Nws  snf  N!7f)Nr   z,List full snapshots API returned None result)rO   r   ListFullr   ListFullSnapshotsRequestru   r  r   r  rw   r  list_full_snapshotsr  )r   r7   r  r  s       r   r  %AsyncQdrantRemote.list_full_snapshotsz  s     ))22113T]] 3   $#	 
 W``V_(J;;HEV_``..<<PPRRZZ	$T&TT$ aRs0   ACCC! C*C+C,CCc                 \  #    U R                   (       a_  U R                  R                  [        R                  " 5       U R
                  S9I S h  vN R                  n[        R                  " U5      $ U R                  R                  R                  US9I S h  vN R                  $  NY N7f)Nr   )r~  )rO   r   
CreateFullr   CreateFullSnapshotRequestru   r  r   r  rw   r  create_full_snapshotr  )r   r~  r7   r  s       r   r  &AsyncQdrantRemote.create_full_snapshot  s      ))44224dmm 5   #"	 !
 ::;OPP))77LLRVLWW___
 Xs%   AB,B(A
B,B*B,*B,c                   #    U R                   (       a?  U R                  R                  [        R                  " US9U R
                  S9I S h  vN   gU R                  R                  R                  XS9I S h  vN R                  $  N: N7f)N)r  r   T)r  r~  )
rO   r   
DeleteFullr   DeleteFullSnapshotRequestru   rw   r  delete_full_snapshotr  )r   r  r~  r7   s       r   r  &AsyncQdrantRemote.delete_full_snapshot  s      %%00..]KUYUbUb 1    %%33HH+ I   &			
s$   ABB+B9B
:B
Blocationchecksumpriorityc                    #    U R                   R                  R                  UU[        R                  " X%XCS9S9I S h  vN R
                  $  N7f)Nr  r  r  r/   )r   r~  snapshot_recover)rw   r  recover_from_snapshotr    SnapshotRecoverr  )r   r   r  r/   r  r  r~  r7   s           r   recover_snapshot"AsyncQdrantRemote.recover_snapshot  sV      %%33II /!'!7!7%8" J   &	s   <AAAshard_idc                    #    U R                   R                  R                  XS9I S h  vN R                  nUc   S5       eU$  N7f)N)r   r  r  )rw   r  list_shard_snapshotsr  )r   r   r  r7   r  s        r   r  &AsyncQdrantRemote.list_shard_snapshots  sZ      %%33HH / I   &	 	
 $O&OO$s   'AAAc                 z   #    U R                   R                  R                  XUS9I S h  vN R                  $  N7f)N)r   r  r~  )rw   r  create_shard_snapshotr  )r   r   r  r~  r7   s        r   r  'AsyncQdrantRemote.create_shard_snapshot  sB      %%33II / J   &		s   (;9;c                 ~   #    U R                   R                  R                  UUUUS9I S h  vN R                  $  N7f)N)r   r  r  r~  )rw   r  delete_shard_snapshotr  )r   r   r  r  r~  r7   s         r   r  'AsyncQdrantRemote.delete_shard_snapshot  sI      %%33II /!+	 J   &	s   *=;=c                    #    U R                   R                  R                  UUU[        R                  " X6XTS9S9I S h  vN R
                  $  N7f)Nr  )r   r  r~  shard_snapshot_recover)rw   r  recover_shard_from_snapshotr    ShardSnapshotRecoverr  )	r   r   r  r  r/   r  r  r~  r7   s	            r   recover_shard_snapshot(AsyncQdrantRemote.recover_shard_snapshot  sY      %%33OO /!'-'B'B%8(	 P   &		s   =AA Ar   shards_number	placementinitial_statec                   #    U R                   (       a  [        U[        [        R                  5      5      (       a  [
        R                  " U5      n[        U[        R                  5      (       a  [
        R                  " U5      nU R                  R                  [        R                  " UU[        R                  " UUUU=(       d    / US9S9Ub  UOU R                  S9I S h  vN R                  $ U R                  R                   R#                  UU[        R$                  " UUUUUS9S9I S h  vN R                  n	U	c   S5       eU	$  Ng N7f)N)r   r  rR  r  r  r   r1   r  r   )r   r1   create_sharding_keyzCreate shard key returned None)rO   rn   r   r    ShardKeyr   convert_shard_keyReplicaStateconvert_replica_stater   CreateShardKeyr   CreateShardKeyRequestru   r  rw   distributed_apicreate_shard_keyCreateShardingKey)
r   r   r   r  rR  r  r  r1   r7   r  s
             r   r  "AsyncQdrantRemote.create_shard_key  sB     )%8%IJJ&88C	-)<)<== * @ @ O++::..(7 ' $ 3 3&/*7/A&/o2*7!
 (/':G ;   f$ ))99JJ$3#(.(@(@"+&3+="+&3) K 
 
 f  %G'GG%M="
s%   C$E&E'AE3E4EEc           
        #    U R                   (       a  [        U[        [        R                  5      5      (       a  [
        R                  " U5      nU R                  R                  [        R                  " UU[        R                  " US9S9Ub  UOU R                  S9I S h  vN R                  $ U R                  R                  R                  UU[        R                   " US9S9I S h  vN R                  nUc   S5       eU$  Nc N7f)N)r   r  r   )r   r1   drop_sharding_keyzDelete shard key returned None)rO   rn   r   r    r  r   r  r   DeleteShardKeyr   DeleteShardKeyRequestru   r  rw   r  delete_shard_keyDropShardingKey)r   r   r   r1   r7   r  s         r   r  "AsyncQdrantRemote.delete_shard_key 	  s      )%8%IJJ&88C	++::..(7 ' $ 3 3i H
 (/':G ;   f	 ))99JJ$3#&,&<&<y&Q K  
 f  %G'GG%M%s%   B"D$D%AD-D
.D
Dc                 N  #    U R                   (       aU  U R                  R                  [        R                  " 5       U R
                  S9I S h  vN n[        R                  " U5      $ U R                  R                  R                  5       I S h  vN nUc   S5       eU$  NR N7f)Nr   zHealthcheck returned None)rO   r   HealthCheckr   HealthCheckRequestru   r   convert_health_check_replyr   service_apiroot)r   version_infos     r   infoAsyncQdrantRemote.info?	  s     !%!;!;'')4== "< " L 88FF!YY227799'D)DD' :s%   AB%B!AB%B#B%#B%cluster_operationc                   #    U R                   (       Ga  [        R                  " U5      n0 n[        U[        R
                  5      (       a  X%S'   GO
[        U[        R                  5      (       a  X%S'   O[        U[        R                  5      (       a  X%S'   O[        U[        R                  5      (       a  X%S'   O[        U[        R                  5      (       a  X%S'   Oz[        U[        R                  5      (       a  X%S'   OV[        U[        R                  5      (       a  X%S'   O2[        U[        R                  5      (       a  X%S'   O[        S	U 35      eU R                  R                  [        R                   " SXS
.UD6Ub  UOU R"                  S9I S h  vN R$                  $ U R&                  R(                  R+                  UUUS9I S h  vN R$                  nUc   S5       eU$  NQ N7f)N
move_shardreplicate_shardabort_transferdrop_replicar  r  restart_transferreplicate_pointszUnknown cluster operation: rM  r   )r   cluster_operationsr1   z'Cluster collection update returned NonerL   )rO   r   convert_cluster_operationsrn   r   	MoveShardReplicateShardAbortShardTransferReplicar  r  RestartTransferReplicatePointsro   r   UpdateCollectionClusterSetup#UpdateCollectionClusterSetupRequestru   r  r   r  update_collection_cluster)r   r   r  r1   r7   grpc_operationupdate_results          r   cluster_collection_update+AsyncQdrantRemote.cluster_collection_updateI	  s      * E EFW XN+T^^<</@|,-t/B/BCC4E01-t/F/FGG3D/0-t||<<1B~.-t/B/BCC5F12-t/B/BCC5F12-t/C/CDD5F12-t/C/CDD5F12"=>O=P QRR++HH<< (7LZ (/':G	 I   f ))++EE /#4 F  
 & 	 (S*SS(s$   FG4G06G4G2G42G4c                    #    U R                   R                  R                  5       I S h  vN R                  nUc   S5       eU$  N7f)NzCluster status returned None)r   r  cluster_statusr  )r   status_results     r   r'   AsyncQdrantRemote.cluster_statusw	  sD     #yy88GGIIQQ(H*HH( J   (AAAc                    #    U R                   R                  R                  5       I S h  vN R                  nUc   S5       eU$  N7f)Nz"Recover current peer returned None)r   r  recover_current_peerr  )r   recover_results     r   r,  &AsyncQdrantRemote.recover_current_peer|	  sD      $		 9 9 N N PPXX)O+OO) Qr*  peer_idforcec                    #    U R                   R                  R                  XUS9I S h  vN R                  nUc   S5       eU$  N7f)N)r/  r0  r1   zRemove peer returned None)r   r  remove_peerr  )r   r/  r0  r1   r7   r#  s         r   r2  AsyncQdrantRemote.remove_peer	  sY      ))++77g 8   &	 	
 (E*EE(r*  c                 ^  #    U R                   (       aT  U R                  R                  [        R                  " US9U R
                  S9I S h  vN n[        R                  " U5      $ U R                  R                  R                  US9I S h  vN R                  nUc   S5       eU$  N[ N7f)Nr
  r   z%Collection cluster info returned None)rO   r   CollectionClusterInfor   CollectionClusterInfoRequestru   r   convert_collection_cluster_infor   r  collection_cluster_infor  )r   r   collection_infos      r   r8  )AsyncQdrantRemote.collection_cluster_info	  s     $($9$9$O$O11/R %P % O ==oNN))++CC / D   &	 	
 *S,SS*s%   AB-B)A B-B+B-+B-)r`   ra   r~   rx   r}   rz   rr   re   rQ   ry   rP   r|   r{   rZ   rR   rT   r[   rO   rV   rt   rg   rS   ru   rw   rs   r   )Ni  i  FNNNNNNNTNr   )r   N)NNNNN
   NTFNNNNN)NN)NNNNNr;  r)   TFNNNNNN)Nr)   r;  NNNN)	Nr;  NNTFNNN)NTNNN)Nr;  FNNN)TNNN)TNN)TFNNN)NTNN)TN)	NNNNNNNNN)NNNNNNNNNNNNNN)r:   NFNN)@   r:   Nr)   FNN)	NNr<  r:   Nr)   FNN)NNTN)T)NNNT)NNNNN)__name__
__module____qualname____firstlineno__rv   r   r0  r  boolr?  r   r   r   rN   propertyr   floatr   staticmethodtupler   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r9   rw  rZ  r   SparseVectorr  
NumpyArrayDocumentImageInferenceObjectr   r   r   r	   PayloadSelectorr  r  r	  r  r  r  r"  WithLookupInterfaceGroupsResultr5  SearchMatrixPairsResponser@  SearchMatrixOffsetsResponserJ  r[  Recordrb  CountResultrn  rx  rz  Pointsr  UpdateResultr  PointVectorsr  r  r  r  classmethodr  r    r  r  rQ  r  r  Payloadr  r  r  r  UpdateOperationr  r   r  r  r  r  r   r#  CollectionInfor(  r/  r;  r=  rJ  rA  rC  r   SparseVectorParamsrF  r:  rO  rY  r[  r]  r_  rX  rf  r
   r$   ri  r   rz  r  r  VectorStructr  r  r  r  SnapshotDescriptionr  r  r  r  r  r  SnapshotPriorityr  r  r  r  r  r  r  r  r  VersionInfor  ClusterOperationsr$  ClusterStatusr'  r,  r2  r5  r8  __static_attributes____classcell__)r   s   @r   r&   r&   +   s    !!"!".2W[$( $T4ZT DjT 	T
 T d{T tT d
T tT DjT 38nt+T &b#g."in:L1MMPTTT "T :T T Tl   edl S T . , ,cDj#sTz3:&M N , ,	/S 	/R*



3  	L$"6"6 	L 	L 	GT__ 	G 	G 	J 2 2 	J 	J 	E4?? 	E 	E #i/ # # #i/ # #(  AE,037!EI-2(,3748<@"3ii }}
u+
tE{
 

 ++	
 

 .. ++ 

  	i Tzi ..4#77$>i llT)i  ))D0!i" #i$ d
%i& Xc]*U-B-BB'i( Xc]*)i* +i, ))D0-i. **T1/i0 "22T91i2 t3i4 5i6 
		7i^ 59"** 5--.* **T1	*
 t* * 
e!!	"*r  AE,037EI-2(,8<3748<@"7s's' s' }}
u+
tE{
 

 ++	
 

 .. ++ 

  		s' Tzs' ..4#77$>s'  llT)!s'" ))D0#s'$ %s'& 's'( Xc]*U-B-BB)s'* Xc]*+s', -s'. ..5/s'0 ))D01s'2 **T13s'4 "22T95s'6 t7s'8 9s': 
		;s'p -1 48<@"2$2$ llT)2$ 	2$
 2$ Tz2$ **T12$ "22T92$ t2$ 2$ 
	(	(2$n -1 48<@"2$2$ llT)2$ 	2$
 2$ Tz2$ **T12$ "22T92$ t2$ 2$ 
	*	*2$n .2)-'+EI-248<@"LJLJ ||d*LJ 	LJ
 --$&LJ $LJ Xc]*U-B-BBLJ Xc]*LJ **T1LJ "22T9LJ tLJ LJ 
tELL!5==4#77	8LJb -1<@"48,, llT), 	,
 "22T9, t, **T1, , 
		,d -1"48<@44 4 llT)	4
 4 4 t4 **T14 "22T94 4 
		4t /3<@-1OO O 	O
 %%,O "22T9O ||d*O O 
		Oj /3<@-1-- ++,- 	-
 %%,- "22T9- ||d*- - 
		-h /3<@// #/ $$	/
 / %%,/ "22T9/ / 
		/j FJ-248<@">> emm$> Xc]*U-B-BB	>
 Xc]*> **T1> "22T9> t> > 
ell	>@ 5))5	t""D$9$9D$@@	A5 5> ))?D?U?UX\?\			 > 	"11		dll		 	 "))"	tF**+d2FMMD4HH	I" "@ /3<@-- --- 	-
 %%,- "22T9- - 
		-h /3<@33 3 $$	3
 4Z3 3 %%,3 "22T93 3 
		3t /3<@00 0 $$	0
 0 %%,0 "22T90 0 
		0n /3<@-- sm- $$	-
 - %%,- "22T9- - 
		-f /3<@-- --- 	-
 %%,- "22T9- - 
		-f /3&& $E$9$9:& 	&
 %%,& & 
e  	!&V #"#+E,A,A#B" t" 	"
 
"H".1		)	)*# %2R2R $c e6O6O &C 3 5K_K_  s c d " @D?C9=37CG"OS<@)-OO !55<O !55<	O
 //$6O ))D0O #99D@O tO  'sE,D,D'DELO "22T9O --$&O O 
Od ;?"-04ZJM	& X\#')-/3'+37?C15?C"OS7;<@)-![[ **WS%:L:L5L-MMPTT[ Dj	[
  $J[ #&*[ [ ))D0[ !55<[ ''$.[ #55<[ t[  'sE,D,D'DEL[ --4[ "22T9[  --$&![" #[$ 
%[@ X\#')-/3'+37?C15?C"OS7;<@)-!$
$
 **WS%:L:L5L-MMPTT$
 Dj	$

  $J$
 #&*$
 $
 ))D0$
 !55<$
 ''$.$
 #55<$
 t$
  'sE,D,D'DEL$
 --4$
 "22T9$
  --$&!$
" #$
$ 
%$
L %\ 2 % % !<@-13"3 3 	3
 3 d
3 3 "22T93 ||d*3 
3r !<@-1

 **+
 	

 
 d

 
 
 "22T9
 ||d*
 
 

> 48.2!<@-1

 c5+++,u/?/??(5K]K]B^^
 $sCx.)D0	

 emm$t+
 
 
 d

 
 
 "22T9
 ||d*
 
 

B 8<59/3LL L --4	L
 ++d2L L %%,L L 
		Ld /3  	
 %%,  
		:".1	e''	($ 26"*.AD		"	"T	)$ FJ"36>BUX	"
# 
$u?X?X:Y 
  
`
`+.
`		"	"
` 04 (,?B	$ ##26  t	
 * ((4/   
(	"	.1	=@		e''	(	 AE".19=PS		"	"T	)   	
   
, ##26  	
 t * ((4/   
4 %))-&*37"// >>/ Tz	/
  $J/ 9t#/ ))D0/ t/ / 
/j #	 >> t	
  
>E--  #	,, !22, t	,
 , 
,\e&9&9 
D  NR		#'$;	@Cd
	]`			S UE`E`  r   r&   )>importlib.metadatarj   r   r^   rl   multiprocessingr   typingr   r   r   r   r   r	   r
   r   rc   r   r   urllib3.utilr   r   urllib.parser   $qdrant_client.common.client_warningsr   r   qdrant_clientqdrant_client._pydantic_compatr   qdrant_client.authr   qdrant_client.async_client_baser   "qdrant_client.common.version_checkr   r   qdrant_client.connectionr   r   qdrant_client.conversionsr   rw  &qdrant_client.conversions.common_typesr   $qdrant_client.conversions.conversionr   r   r   qdrant_client.httpr   r   r     qdrant_client.parallel_processorr!   $qdrant_client.uploader.grpc_uploaderr"   $qdrant_client.uploader.rest_uploaderr#   qdrant_client.uploader.uploaderr$   r&   rL   r   r   <module>rv     ss        1 X X X   '   P & 4 ) ; P E ; F 
 A @ ? B B 8n% n%r   