NgmddlZddlmZddlmZmZmZmZmZm Z m Z m Z m Z m Z mZddlmZddlmZddlmZddlmZmZddlmZdd lmZdd lmZdd lm Z Gd d eZ!dS)N)deepcopy) Any AwaitableCallableDictIterableListMappingOptionalSequenceTupleUnion)grpc) QdrantBase) common_types) ApiClientSyncApis) QdrantLocalmigrate)QdrantFastembedMixin) QdrantRemotec)L$eZdZdZ ddeedeedeed ed ed eed eed eedeedeedeededeeee fdee e gefe ge effde ffd Z ddZddeede ddfdZedejfdZedejfdZedejfdZedejfdZedeefdZedeefdZedeee ffd Z dd!ed"eej deed#eej!de de"e"ej#f d$Z$ dd!ed'e eee%ee"efej&ej'ej(fd(eej)d)eej*d*ed+eed,e eeeej+fd-e eeefd.eed/ed#eej!d0eej,deede de"ej#fd1Z- dd!ed"eej.d#eej!deede de"ej/f d2Z0 dd!ed3e ej1e"ee"e"eej2ej3ej(ej4dfd4eed5e ej5e"ej5dfd(eej)d)eej*d*ed+eed,e eeeej+fd-e eeefd.eed6eej6d#eej!d0eej,deede dej/f"d7Z7 dd!ed9ed3e ej1e"ee"e"eej2ej3ej(ej4dfd4eed5e ej5e"ej5dfd(eej)d)eej*d*ed:ed,e eeeej+fd-e eeefd.eed;eej8d6eej6d#eej!d0eej,deede dej9f&d<Z: dd!ed'e eee%ee"efej&ej'ej(fd9ed(eej)d)eej*d*ed:ed,e eeeej+fd-e eeefd.eed;eej8d#eej!d0eej,deede dej9f d>Z; dd!ed"eej<d#eej!deede de"e"ej#f d?Z= dd!edAeeej>dBeeej>d(eej)d)eej*d*ed+ed,e ee"eej+fd-e ee"efd.eed4eed6eej6dCeej?d#eej!d0eej,deede de"ej#f$dDZ@ dd!ed(eej)d*edEed4eed#eej!deed0eej,de dejAfdFZB dd!ed(eej)d*edEed4eed#eej!deed0eej,de dejCfdGZD dd!ed9edAeeej>dBeeej>d(eej)d)eej*d*ed:ed.eed,e eeeej+fd-e eeefd4eed6eej6d;eej8dCeej?d#eej!d0eej,deede dej9f(dHZE dd!edIeejFdJeeejGd(eej)d)eej*d*ed+ed,e ee"eej+fd-e ee"efd4eed6eej6d#eej!d0eej,deede de"ej#f dKZH dd!ed"eejId#eej!deede de"e"ej#f dLZJ dd!edMeej)d*edNeejKd+eej1d,e eeeej+fd-e eeefd#eej!d0eej,deede de%e"ejLeej1ffdOZM dd!edPeej)dQed0eej,deede dejNfdRZO dd!edSedTeej)d*edQed#eej!deed0eej,de dejPfdUZQ dd!edVejRdWedXeejSd0eej,de dejTfdYZU dd!edVeejVdWedXeejSd0eej,de dejTfdZZW dd!ed[eedVejXdWedXeejSd0eej,de dejTfd\ZY dd!ed]eej1d,e eeeej+fd-e eeefd#eej!d0eej,deede de"ejLfd^ZZ dd!ed_ejXdWedXeejSd0eej,de dejTfd`Z[ dd!edaej\dVejXdSeedWedXeejSd0eej,de dejTfdbZ] dd!edaej\dVejXdWedXeejSd0eej,de dejTfdcZ^ dd!eddeedVejXdWedXeejSd0eej,de dejTfdeZ_ dd!ed_ejXdWedXeejSd0eej,de dejTfdfZ` dd!edgeejadWedXeejSde de"ejTf dhZb ddieejcdeede defdjZdd!ede dejefdkZfde dejefdlZgde dejhfdmZid!ede dejjfdnZkd!ede defdoZl dd!edpeejmdqeejndreejodseejpdteejqdeedueereejsfde defdvZt dd!edeede defdwZu dd!edre ejvereejvffdueereejsfdxeedyeejwdzeed{eed|eedseejpdpeejmd}eejxdteejyd~eejzdeede def dZ{ dd!edre ejvereejvffdueereejsfdxeedyeejwdzeed{eed|eedseejpdpeejmd}eejxdteejyd~eejzdeede def dZ| dd!ede}ejLdededeededWed0eej,de ddfdZ~ dd!edVe}ejdededeededWed0eej,de ddfdZ dd!ed[e e}ejeeej(fej(fdaee}ee e fd]ee}ej1dededeededWed0eej,de ddfdZ dd!ededeejdeejdWedXeejSde dejTfdZ dd!ededWedXeejSde dejTf dZd!ede de"ejfdZ dd!edWede deejfdZ dd!ededWede deef dZde de"ejfdZ ddWede deejfdZ ddedWede deefdZ dd!eded eedeedeejdWede deefdZd!edede de"ejfdZ dd!ededWede deejf dZ dd!edededWede deef dZ dd!ededed eedeedeejdWede deefdZdede dejfdZde dejfdZde dejfdZ ddedee"edededdf dZ dd!edejdeedzeedee"ede defdZd!edejde defdZdejfdZxZS) QdrantClienta Entry point to communicate with Qdrant service via REST or gRPC API. It combines interface classes and endpoint implementation. Additionally, it provides custom implementations for frequently used methods like initial collection upload. All methods in QdrantClient accept both gRPC and REST structures as an input. Conversion will be performed automatically. .. note:: This module methods are wrappers around generated client code for gRPC and REST methods. If you need lower-level access to generated clients, use following properties: - :py:attr:`QdrantClient.grpc_points` - :py:attr:`QdrantClient.grpc_collections` - :py:attr:`QdrantClient.rest` .. note:: If you need async, please consider using Async Implementations of QdrantClient. - :class:`qdrant_client.async_qdrant_client` Args: location: If `":memory:"` - use in-memory Qdrant instance. If `str` - use it as a `url` parameter. If `None` - use default values for `host` and `port`. url: either host or str of "Optional[scheme], host, Optional[port], Optional[prefix]". Default: `None` port: Port of the REST API interface. Default: 6333 grpc_port: Port of the gRPC interface. Default: 6334 prefer_grpc: If `true` - use gPRC interface whenever possible in custom methods. https: If `true` - use HTTPS(SSL) protocol. Default: `None` api_key: API key for authentication in Qdrant Cloud. Default: `None` prefix: If not `None` - add `prefix` to the REST URL path. Example: `service/v1` will result in `http://localhost:6333/service/v1/{qdrant-endpoint}` for REST API. Default: `None` timeout: Timeout for REST and gRPC API requests. Default: 5 seconds for REST and unlimited for gRPC host: Host name of Qdrant service. If url and host are None, set to 'localhost'. Default: `None` path: Persistence path for QdrantLocal. Default: `None` force_disable_check_same_thread: For QdrantLocal, force disable check_same_thread. Default: `False` Only use this if you can guarantee that you can resolve the thread safety outside QdrantClient. auth_token_provider: Callback function to get Bearer access token. If given, the function will be called before each request to get the token. **kwargs: Additional arguments passed directly into REST client initialization NFlocationurlport grpc_port prefer_grpchttpsapi_keyprefixtimeouthostpathforce_disable_check_same_thread grpc_optionsauth_token_providerkwargsc tjdi|dtD|_|jt ||td||| | fDdkrtd|dkrt|| |_ dS| t| | |_ dS|||}td|||||||| | | |d ||_ dS) Nc"i|] \}}|dv || S))self __class__r+).0keyvalues W/var/www/html/ai-engine/env/lib/python3.11/site-packages/qdrant_client/qdrant_client.py z)QdrantClient.__init__..ks4   U999 999cg|]}|duSNr0)r1params r4 z)QdrantClient.__init__..tsKKKeT!KKKr6zDOnly one of , , or should be specified.z:memory:)rr() rrr r!r"r#r$r%r&r)r*r0) super__init__localsitems _init_optionsupdatersum ValueErrorr_clientr)r.rrrr r!r"r#r$r%r&r'r(r)r*r+r/s r4r=zQdrantClient.__init__Oss(        $hhnn..    !!(6"2"2333  KK#tT/JKKK L Lq P PV  z ! !&!0ODLLL &0ODLLL # ' #')$7    DLLLr6returnc.|dSr8)closer.s r4__del__zQdrantClient.__del__s r6 grpc_gracec Rt|dr|jjdd|i|dSdS)zCloses the connection to Qdrant Args: grpc_grace: Grace period for gRPC connection close. Default: None rDrJNr0)hasattrrDrG)r.rJr+s r4rGzQdrantClient.closesI 4 # # @ DL  ? ?* ? ? ? ? ? ? @ @r6ct|jtr |jjSt dt |j)gRPC client for collections methods Returns: An instance of raw gRPC client, generated from Protobuf !gRPC client is not supported for ) isinstancerDrgrpc_collectionsNotImplementedErrortyperHs r4rQzQdrantClient.grpc_collectionssD dlL 1 1 1<0 0!"Zd4 > 6{{a!LtFKKMM7J7J!L!L-t|-  +//)# 3"7      r6c t|dks,Jdt||jjd||d|S)aLRemoves collection and all it's data Args: collection_name: Name of the collection to delete timeout: Wait for operation commit timeout in seconds. If timeout is reached - request will return with service error. Returns: Operation result rrm)rir%r0)rorprqrDdelete_collection)r.rir%r+s r4rzQdrantClient.delete_collectionesl6{{a!LtFKKMM7J7J!L!L-t|- +W  @F   r6 shard_numbersharding_methodreplication_factorwrite_consistency_factoron_disk_payload wal_config init_fromc t|dks,Jdt||jjd|||||||| | | | | ||d|S)a Create empty collection with given parameters Args: collection_name: Name of the collection to recreate vectors_config: Configuration of the vector storage. Vector params contains size and distance for the vector storage. If dict is passed, service will create a vector storage for each key in the dict. If single VectorParams is passed, service will create a single anonymous vector storage. sparse_vectors_config: Configuration of the sparse vector storage. The service will create a sparse vector storage for each key in the dict. shard_number: Number of shards in collection. Default is 1, minimum is 1. sharding_method: Defines strategy for shard creation. Option `auto` (default) creates defined number of shards automatically. Data will be distributed between shards automatically. After creation, shards could be additionally replicated, but new shards could not be created. Option `custom` allows to create shards manually, each shard should be created with assigned unique `shard_key`. Data will be distributed between based on `shard_key` value. replication_factor: Replication factor for collection. Default is 1, minimum is 1. Defines how many copies of each shard will be created. Have effect only in distributed mode. write_consistency_factor: Write consistency factor for collection. Default is 1, minimum is 1. Defines how many replicas should apply the operation for us to consider it successful. Increasing this number will make the collection more resilient to inconsistencies, but will also make it fail if not enough replicas are available. Does not have any performance impact. Have effect only in distributed mode. on_disk_payload: If true - point`s payload will not be stored in memory. It will be read from the disk every time it is requested. This setting saves RAM by (slightly) increasing the response time. Note: those payload values that are involved in filtering and are indexed - remain in RAM. hnsw_config: Params for HNSW index optimizers_config: Params for optimizer wal_config: Params for Write-Ahead-Log quantization_config: Params for quantization, if None - quantization will be disabled init_from: Use data stored in another collection to initialize this collection timeout: Wait for operation commit timeout in seconds. If timeout is reached - request will return with service error. Returns: Operation result rrmrirrrrrrrrrrrr%rr0)rorprqrDcreate_collectionr.rirrrrrrrrrrrrr%r+s r4rzQdrantClient.create_collectionysB6{{a!LtFKKMM7J7J!L!L-t|- +)%+1%=+#/! 3"7     r6c t|dks,Jdt|tjdt d|jjd|||||||| | | | | ||d|S)a Delete and create empty collection with given parameters Args: collection_name: Name of the collection to recreate vectors_config: Configuration of the vector storage. Vector params contains size and distance for the vector storage. If dict is passed, service will create a vector storage for each key in the dict. If single VectorParams is passed, service will create a single anonymous vector storage. sparse_vectors_config: Configuration of the sparse vector storage. The service will create a sparse vector storage for each key in the dict. shard_number: Number of shards in collection. Default is 1, minimum is 1. sharding_method: Defines strategy for shard creation. Option `auto` (default) creates defined number of shards automatically. Data will be distributed between shards automatically. After creation, shards could be additionally replicated, but new shards could not be created. Option `custom` allows to create shards manually, each shard should be created with assigned unique `shard_key`. Data will be distributed between based on `shard_key` value. replication_factor: Replication factor for collection. Default is 1, minimum is 1. Defines how many copies of each shard will be created. Have effect only in distributed mode. write_consistency_factor: Write consistency factor for collection. Default is 1, minimum is 1. Defines how many replicas should apply the operation for us to consider it successful. Increasing this number will make the collection more resilient to inconsistencies, but will also make it fail if not enough replicas are available. Does not have any performance impact. Have effect only in distributed mode. on_disk_payload: If true - point`s payload will not be stored in memory. It will be read from the disk every time it is requested. This setting saves RAM by (slightly) increasing the response time. Note: those payload values that are involved in filtering and are indexed - remain in RAM. hnsw_config: Params for HNSW index optimizers_config: Params for optimizer wal_config: Params for Write-Ahead-Log quantization_config: Params for quantization, if None - quantization will be disabled init_from: Use data stored in another collection to initialize this collection timeout: Wait for operation commit timeout in seconds. If timeout is reached - request will return with service error. Returns: Operation result rrmz`recreate_collection` method is deprecated and will be removed in the future. Use `collection_exists` to check collection existence and `create_collection` instead.rXrYrr0)rorprqr[r\r]rDrecreate_collectionrs r4rz QdrantClient.recreate_collectionsB6{{a!LtFKKMM7J7J!L!L  f      0t|/ +)%+1%=+#/! 3"7     r6@records batch_sizeparallelmethod max_retriesc tjdtdt| dks,Jdt | |j||||||||S)a-Upload records to the collection Similar to `upload_collection` method, but operates with records, rather than vector and payload individually. Args: collection_name: Name of the collection to upload to records: Iterator over records to upload batch_size: How many vectors upload per-request, Default: 64 parallel: Number of parallel processes of upload method: Start method for parallel processes, Default: forkserver max_retries: maximum number of retries in case of a failure during the upload of a batch wait: Await for the results to be applied on the server side. If `true`, each update request will explicitly wait for the confirmation of completion. Might be slower. If `false`, each update request will return immediately after the confirmation of receiving. Default: `false` shard_key_selector: Defines the shard groups that should be used to write updates into. If multiple shard_keys are provided, the update will be written to each of them. Only works for collections with `custom` sharding method. This parameter overwrites shard keys written in the records. z;`upload_records` is deprecated, use `upload_points` insteadrXrYrrm)rirrrrrrr})r[r\r]rorprqrDupload_records) r.rirrrrrrr}r+s r4rzQdrantClient.upload_records, sF  I     6{{a!LtFKKMM7J7J!L!L|**+!#1+   r6c t| dks,Jdt| |j||||||||S)a)Upload points to the collection Similar to `upload_collection` method, but operates with points, rather than vector and payload individually. Args: collection_name: Name of the collection to upload to points: Iterator over points to upload batch_size: How many vectors upload per-request, Default: 64 parallel: Number of parallel processes of upload method: Start method for parallel processes, Default: forkserver max_retries: maximum number of retries in case of a failure during the upload of a batch wait: Await for the results to be applied on the server side. If `true`, each update request will explicitly wait for the confirmation of completion. Might be slower. If `false`, each update request will return immediately after the confirmation of receiving. Default: `false` shard_key_selector: Defines the shard groups that should be used to write updates into. If multiple shard_keys are provided, the update will be written to each of them. Only works for collections with `custom` sharding method. This parameter overwrites shard keys written in the records. rrm)rirrrrrrr})rorprqrD upload_points) r.rirrrrrrr}r+s r4rzQdrantClient.upload_pointsb stF6{{a!LtFKKMM7J7J!L!L|))+!#1*   r6c t| dks,Jdt| |j||||||||| |  S)aUpload vectors and payload to the collection. This method will perform automatic batching of the data. If you need to perform a single update, use `upsert` method. Note: use `upload_records` method if you want to upload multiple vectors with single payload. Args: collection_name: Name of the collection to upload to vectors: np.ndarray or an iterable over vectors to upload. Might be mmaped payload: Iterable of vectors payload, Optional, Default: None ids: Iterable of custom vectors ids, Optional, Default: None batch_size: How many vectors upload per-request, Default: 64 parallel: Number of parallel processes of upload method: Start method for parallel processes, Default: forkserver max_retries: maximum number of retries in case of a failure during the upload of a batch wait: Await for the results to be applied on the server side. If `true`, each update request will explicitly wait for the confirmation of completion. Might be slower. If `false`, each update request will return immediately after the confirmation of receiving. Default: `false` shard_key_selector: Defines the shard groups that should be used to write updates into. If multiple shard_keys are provided, the update will be written to each of them. Only works for collections with `custom` sharding method. rrm) rirrrrrrrrr})rorprqrDupload_collection) r.rirrrrrrrrr}r+s r4rzQdrantClient.upload_collection szT6{{a!LtFKKMM7J7J!L!L|--+!#1.   r6 field_name field_schema field_typec t|dks,Jdt||jjd||||||d|S)ajCreates index for a given payload field. Indexed fields allow to perform filtered search operations faster. Args: collection_name: Name of the collection field_name: Name of the payload field field_schema: Type of data to index field_type: Same as field_schema, but deprecated wait: Await for the results to be processed. - If `true`, result will be returned only when all changes are applied - If `false`, result will be returned immediately after the confirmation of receiving. ordering (Optional[WriteOrdering]): Define strategy for ordering of the points. Possible values: - `weak` (default) - write operations may be reordered, works faster - `medium` - write operations go through dynamically selected leader, may be inconsistent for a short period of time in case of leader change - `strong` - Write operations go through the permanent leader, consistent, but may be unavailable if leader is down Returns: Operation Result rrm)rirrr rrr0)rorprqrDcreate_payload_index)r.rirrr rrr+s r4r z!QdrantClient.create_payload_index s{>6{{a!LtFKKMM7J7J!L!L0t|0 +!%!      r6c t|dks,Jdt||jjd||||d|S)aRemoves index for a given payload field. Args: collection_name: Name of the collection field_name: Name of the payload field wait: Await for the results to be processed. - If `true`, result will be returned only when all changes are applied - If `false`, result will be returned immediately after the confirmation of receiving. ordering (Optional[WriteOrdering]): Define strategy for ordering of the points. Possible values: - `weak` (default) - write operations may be reordered, works faster - `medium` - write operations go through dynamically selected leader, may be inconsistent for a short period of time in case of leader change - `strong` - Write operations go through the permanent leader, consistent, but may be unavailable if leader is down Returns: Operation Result rrm)rirrrr0)rorprqrDdelete_payload_index)r.rirrrr+s r4r z!QdrantClient.delete_payload_index su46{{a!LtFKKMM7J7J!L!L0t|0 +!       r6c t|dks,Jdt||jjdd|i|S)zList all snapshots for a given collection. Args: collection_name: Name of the collection Returns: List of snapshots rrmrir0)rorprqrDlist_snapshotsrs r4rzQdrantClient.list_snapshots s\6{{a!LtFKKMM7J7J!L!L*t|*UU?UfUUUr6c t|dks,Jdt||jjd||d|S)aCreate snapshot for a given collection. Args: collection_name: Name of the collection wait: Await for the snapshot to be created. - If `true`, result will be returned only when a snapshot is created - If `false`, result will be returned immediately after the confirmation of receiving. Returns: Snapshot description rrm)rirr0)rorprqrDcreate_snapshot)r.rirr+s r4rzQdrantClient.create_snapshot) s`6{{a!LtFKKMM7J7J!L!L+t|+aORVaaZ`aaar6 snapshot_namec t|dks,Jdt||jjd|||d|S)aDelete snapshot for a given collection. Args: collection_name: Name of the collection snapshot_name: Snapshot id wait: Await for the snapshot to be deleted. - If `true`, result will be returned only when the snapshot is deleted - If `false`, result will be returned immediately after the confirmation of receiving. Returns: True if snapshot was deleted rrm)rirrr0)rorprqrDdelete_snapshot)r.rirrr+s r4rzQdrantClient.delete_snapshot< sr 6{{a!LtFKKMM7J7J!L!L+t|+ +'      r6c t|dks,Jdt||jjdi|S)z_List all snapshots for a whole storage Returns: List of snapshots rrmr0)rorprqrDlist_full_snapshotsrs r4rz QdrantClient.list_full_snapshotsU sW 6{{a!LtFKKMM7J7J!L!L/t|/99&999r6c t|dks,Jdt||jjdd|i|S)afCreate snapshot for a whole storage. Args: wait: Await for the snapshot to be created. - If `true`, result will be returned only when the snapshot is created - If `false`, result will be returned immediately after the confirmation of receiving. Returns: Snapshot description rrmrr0)rorprqrDcreate_full_snapshot)r.rr+s r4rz!QdrantClient.create_full_snapshot_ s\6{{a!LtFKKMM7J7J!L!L0t|0EEdEfEEEr6c t|dks,Jdt||jjd||d|S)aDelete snapshot for a whole storage. Args: snapshot_name: Snapshot name wait: Await for the snapshot to be deleted. - If `true`, result will be returned only when the snapshot is deleted - If `false`, result will be returned immediately after the confirmation of receiving. Returns: True if snapshot was deleted rrm)rrr0)rorprqrDdelete_full_snapshot)r.rrr+s r4rz!QdrantClient.delete_full_snapshotq s`6{{a!LtFKKMM7J7J!L!L0t|0b}SWbb[abbbr6checksumpriorityc t|dks,Jdt||jjd||||||d|S)aRecover collection from snapshot. Args: collection_name: Name of the collection location: URL of the snapshot Example: - URL `http://localhost:8080/collections/my_collection/snapshots/my_snapshot` - Local path `file:///qdrant/snapshots/test_collection/test_collection-6194298859870377-2023-11-09-15-17-51.snapshot` api_key: API key to use for accessing the snapshot on another server. checksum: Checksum of the snapshot to verify the integrity of the snapshot. priority: Defines source of truth for snapshot recovery - `replica` (default) means - prefer existing data over the snapshot - `no_sync` means - do not sync shard with other shards - `snapshot` means - prefer snapshot data over the current state wait: Await for the recovery to be done. - If `true`, result will be returned only when the recovery is done - If `false`, result will be returned immediately after the confirmation of receiving. Returns: True if snapshot was recovered rrm)rirr#rrrr0)rorprqrDrecover_snapshot)r.rirr#rrrr+s r4rzQdrantClient.recover_snapshot s|J6{{a!LtFKKMM7J7J!L!L,t|, +      r6shard_idc t|dks,Jdt||jjd||d|S)zList all snapshots of a given shard Args: collection_name: Name of the collection shard_id: Index of the shard Returns: List of snapshots rrm)rirr0)rorprqrDlist_shard_snapshots)r.rirr+s r4r!z!QdrantClient.list_shard_snapshots so6{{a!LtFKKMM7J7J!L!L0t|0 +     r6c t|dks,Jdt||jjd|||d|S)aCreate snapshot for a given shard. Args: collection_name: Name of the collection shard_id: Index of the shard wait: Await for the snapshot to be created. - If `true`, result will be returned only when the snapshot is created. - If `false`, result will be returned immediately after the confirmation of receiving. Returns: Snapshot description rrm)rirrr0)rorprqrDcreate_shard_snapshot)r.rirrr+s r4r#z"QdrantClient.create_shard_snapshot sr 6{{a!LtFKKMM7J7J!L!L1t|1 +      r6c t|dks,Jdt||jjd||||d|S)aDelete snapshot for a given shard. Args: collection_name: Name of the collection shard_id: Index of the shard snapshot_name: Snapshot id wait: Await for the snapshot to be deleted. - If `true`, result will be returned only when the snapshot is deleted - If `false`, result will be returned immediately after the confirmation of receiving. Returns: True if snapshot was deleted rrm)rirrrr0)rorprqrDdelete_shard_snapshot)r.rirrrr+s r4r%z"QdrantClient.delete_shard_snapshot su.6{{a!LtFKKMM7J7J!L!L1t|1 +'       r6c t|dks,Jdt||jjd|||||||d|S)a>Recover shard from snapshot. Args: collection_name: Name of the collection shard_id: Index of the shard location: URL of the snapshot Example: - URL `http://localhost:8080/collections/my_collection/snapshots/my_snapshot` api_key: API key to use for accessing the snapshot on another server. checksum: Checksum of the snapshot to verify the integrity of the snapshot. priority: Defines source of truth for snapshot recovery - `replica` (default) means - prefer existing data over the snapshot - `no_sync` means - do not sync shard with other shards - `snapshot` means - prefer snapshot data over the current state wait: Await for the recovery to be done. - If `true`, result will be returned only when the recovery is done - If `false`, result will be returned immediately after the confirmation of receiving. Returns: True if snapshot was recovered rrm)rirrr#rrrr0)rorprqrDrecover_shard_snapshot) r.rirrr#rrrr+s r4r'z#QdrantClient.recover_shard_snapshot sH6{{a!LtFKKMM7J7J!L!L2t|2  +      r6reasonc t|dks,Jdt||jjdd|i|S)zLock storage for writing.rrmr(r0)rorprqrD lock_storage)r.r(r+s r4r*zQdrantClient.lock_storage4 sZ6{{a!LtFKKMM7J7J!L!L(t|(AAA&AAAr6c t|dks,Jdt||jjdi|S)zUnlock storage for writing.rrmr0)rorprqrDunlock_storagers r4r,zQdrantClient.unlock_storage: sU6{{a!LtFKKMM7J7J!L!L*t|*44V444r6c t|dks,Jdt||jjdi|S)zGet current locks state.rrmr0)rorprqrD get_locksrs r4r.zQdrantClient.get_locks@ sU6{{a!LtFKKMM7J7J!L!L%t|%/////r6d dest_clientcollection_namesrecreate_on_collisionc.t|||||dS)aMigrate data from one Qdrant instance to another. Args: dest_client: Destination Qdrant instance either in local or remote mode collection_names: List of collection names to migrate. If None - migrate all collections batch_size: Batch size to be in scroll and upsert operations during migration recreate_on_collision: If True - recreate collection on destination if it already exists, otherwise raise ValueError exception )r1rr2Nr)r.r0r1rr2s r4rzQdrantClient.migrateF s4   -!"7       r6 shard_key shards_number placementc 2|jjd|||||d|S)aCreate shard key for collection. Only works for collections with `custom` sharding method. Args: collection_name: Name of the collection shard_key: Shard key to create shards_number: How many shards to create for this key replication_factor: Replication factor for this key placement: List of peers to place shards on. If None - place on all peers. Returns: Operation result )rir4r5rr6r0)rDcreate_shard_key)r.rir4r5rr6r+s r4r8zQdrantClient.create_shard_key^ s@.-t|, +'1       r6c ,|jjd||d|S)a Delete shard key for collection. Only works for collections with `custom` sharding method. Args: collection_name: Name of the collection shard_key: Shard key to delete Returns: Operation result )rir4r0)rDdelete_shard_key)r.rir4r+s r4r:zQdrantClient.delete_shard_key~ s7"-t|, +     r6c4|jS)zReturns information about the running Qdrant instance like version and commit id Returns: Title, version and optionally commit info )rDinforHs r4r<zQdrantClient.info s|  """r6)NNrrFNNNNNNFNN)rENr8)NN) NNrsNTFNTNNN)NNNNNrsNTFNNNNN)NNNNNrsrTFNNNNNN) NNrsr;TFNNNNN)NNNNrsrTFNNNNNNN)NrrsNNNN)NNNNrsr;NTFNNNNNNN) NNNNrsrTFNNNNN) NrsNNTFNNN)NTNN)NrsFNNN)TNN)TFNNN)TN)NNNNNNN) NNNNNNNNNNNN)rr;NrFN)NNrr;NrFN)NNTN)T)NNNT)Nr/F)NNN)__name__ __module__ __qualname____doc__r strintboolrrrrrr=rIfloatrGpropertyrCollectionsStubrQ PointsStubrVr^r`rrrdrfrhr types SearchRequestReadConsistencyr ScoredPointrrr NamedVectorNamedSparseVector NumpyArrayFilter SearchParamsPayloadSelectorShardKeySelectorr QueryRequest QueryResponserPointId SparseVectorQueryDocumentPrefetchLookupLocationrWithLookupInterface GroupsResultrrRecommendRequestrRecommendExampleRecommendStrategyrSearchMatrixPairsResponserSearchMatrixOffsetsResponserr TargetVectorContextExamplePairrDiscoverRequestrOrderByRecordr CountResultr FacetResponserPoints WriteOrdering UpdateResultr PointVectorsrPointsSelectorrrrPayloadrrrrUpdateOperationrAliasOperationsrCollectionsAliasesResponserrCollectionsResponserCollectionInforrOptimizersConfigDiffCollectionParamsDiffVectorsConfigDiffHnswConfigDiffQuantizationConfigDiffr SparseVectorParamsrr VectorParamsShardingMethod WalConfigDiffQuantizationConfigInitFromrrrr PointStructr VectorStructrPayloadSchemaTyper r SnapshotDescriptionrrrrrrSnapshotPriorityrr!r#r%r' LocksOptionr*r,r.rrShardKeyr8r: VersionInfor< __classcell__)r/s@r4rrs&11j#'!"! $!% $!%""0515 #DD3-Dc]Dsm D  D  D~D#D D#DsmDsmD*.DtCH~.D& (2s7#Xb)C..@%AA B D$%DDDDDDL@@@#@$@@@@ \$"6 \ \ \X \ \T_ \ \ \X \\4?\\\X\ \(<\\\X\ \hy)\\\X\  \hy) \ \ \X \"d38n"""X""&7; " " " 5./" # " e34 "  "  d5$% &" " " " \046: $JN38+/#7;?C!%)m m m  UO #tE{" #    #     m u|,m   23m m  m D(3-1FFGm D(3-/0m "%!m "#m $e34%m &%U%;<'m (#)m *+m , e -m m m m f8?0 $E$>? 0 !!89 0 e23 0 &e&BC0 #0  (U5M0M(NO0 0  0 0 0 0 f>B  " -5c] MP     0SW&*:>,026*.6:BF48BF.2!%S S S e0'#u?Q:Q2RRSS  (U5M0M(NO S sm S "%"67 S %SMS #+3-S "$S e23S $E$>?S U01S &e&>?S EN+S #S !S " #S S S S vSW&*:>,026*.6:BF48BF.2!%Z Z Z e0'#u?Q:Q2RRSZ  (U5M0M(NO Z sm Z "%"67 Z %SMZ #+3-Z "$Z e23Z $E$>?Z U01Z &e&>?Z EN+Z #Z !Z " #Z Z Z Z @ $?C4 4 4 %,'4  4  4  4 4 4 %U%;<4 4  4 4 4 4 t $?C. . . *+.  .  .  . . . %U%;<. .  . . . . p7;15 $?C7 7 7  U' ( e&& '    7 (4S>237 hu}- .7 7 7  7 7 7 %U%;<7 !7 " #7 7 7 7 z;?8<26) ) ) ) u67 ) U45 )  ) 5./) )   ) ) ) ) ^26 " " " "  " 5./ "  "   " " " " H V" V.1 V e' ( V V V V 26bb"b*.bADb %+ ,bbbb(FJ  " 36 >B UX $    2:C:D9R4S:::: FFF+.F %+ ,FFFF&04cc c(,c?Bc $cccc."&"&59/ / / / # / 3- / 512 / / /  $/ / / / b " .1 =@ e' (    *AE  " .1 9= PS %+ ,    <            $    L"&"&59/ / / /  / # / 3- / 512/ / /  $/ / / / bB3B#B%:KBBBB 5s5u/@5555 0#0%*;000015&+    #49-   $       8(,,0)-    >  }  %SM  DI&        @  >        .#e'########r6r)"r[copyrtypingrrrrrr r r r r r qdrant_clientrqdrant_client.client_baserqdrant_client.conversionsrrHqdrant_client.httprr qdrant_client.local.qdrant_localrqdrant_client.migraterqdrant_client.qdrant_fastembedrqdrant_client.qdrant_remoterrr0r6r4rsy                          '&&&&&000000;;;;;;22222222888888))))))??????444444A.#A.#A.#A.#A.#'A.#A.#A.#A.#A.#r6