jddlmZddlmZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZdd lmZmZmZmZmZmZGd d Zd S) )parse) deprecated)MilvusException ParamError)Config)Mapping) is_legal_host is_legal_port) GrpcHandler)BulkInsertStateCompactionPlansCompactionStateReplicaResourceGroupInfoResourceGroupConfigc PeZdZedejejdf dWdZdejejfdZdZ e dZ e dZ dZ d Zd ZdXd ZdXd ZdXd ZdXdZdYdZdXdZdXdZdXdZdXdZdXdZdXdZdYdZdXdZdXdZdXdZdXdZdXdZ dXdZ!dXdZ"dXdZ#dZd!Z$d[d"Z%d[d#Z&d[d$Z' d\d&Z(dXd'Z)dXd(Z*dXd)Z+dXd*Z,dXd+Z-dXd,Z.dXd-Z/dXd.Z0 d]d/Z1 dXd0e2fd1Z3d^de4fd3Z5 d^d4e4de6fd5Z7 dXd4e4de6fd6Z8dXd4e4de9fd7Z:dXd0e2de;fd8Z< dXd0e2d9e2d:e=de4fd;Z>dXde?fd<Z@dXde=fd=ZAdXd>ZBdXd?ZCdXd@ZDdXdAZEdXdBZFdXdCZGdXdDZHdXdEZIdXdFZJdXdGZKdXdHZLdXdIZMdXdJZNdXdKZOdXdLZP dXdMZQdXdNZRdXdOZS dXdPeTe2eUffdQZVdXdRZWdXdSZXdXdeYfdTZZdXdUZ[ dXdVZ\dS)_MilvusNreturnc |||||_td|j|d||_|dddur|jdSdS)N)addresschannelpre_pingFT)_Milvus__get_addressrr _handlerget_wait_for_channel_ready)selfhostporturirkwargss ^/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/pymilvus/client/stub.py__init__zMilvus.__init__sv))$c:: #TDL'TTVTT ::j% ( (D 0 0 M 1 1 3 3 3 3 3 1 0c~||td|N tj|d}n)#t$r}td|d||d}~wwxYw|j|j}}t |t |}}t|rt|std|d|d|d |S) Nz Host and uri cannot both be Nonemessagetcpz Illegal uri [z]: zIllegal host [z ] or port []:) rrurlparse Exceptionhostnamer!strr r )rr r!r" parsed_uries r$ __get_addresszMilvus.__get_addresss zMilvus.get_server_type<s},,...r&c>|j|||dSr5)rreset_password)ruser old_password new_passwords r$r@zMilvus.reset_password?s" $$T<FFFFFr&ct|jtd|jd|_dS)NzClosing on closed handlerr()rrr6closer7s r$rEz Milvus.closeBs; = !*EFFF F  r&c |5}|j||fd|i|cdddS#1swxYwYdS)a:Creates a collection. :param collection_name: The name of the collection. A collection name can only include numbers, letters, and underscores, and must not begin with a number. :type collection_name: str :param fields: Field parameters. :type fields: dict ` {"fields": [ {"field": "A", "type": DataType.INT32} {"field": "B", "type": DataType.INT64}, {"field": "C", "type": DataType.FLOAT}, {"field": "Vec", "type": DataType.FLOAT_VECTOR, "params": {"dim": 128}} ], "auto_id": True}` :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :param kwargs: * *num_shards* (``int``) -- How wide to scale collection. Corresponds to how many active datanodes can be used on insert. * *shards_num* (``int``, deprecated) -- How wide to scale collection. Corresponds to how many active datanodes can be used on insert. * *consistency_level* (``str/int``) -- Which consistency level to use when searching in the collection. For details, see https://github.com/milvus-io/milvus/blob/master/docs/developer_guides/how-guarantee-ts-works.md. Note: this parameter can be overwritten by the same parameter specified in search. * *properties* (``dict``) -- :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok timeoutN)r8create_collection)rcollection_namefieldsrGr#r6s r$rHzMilvus.create_collectionHsT     a7,7,_f``g`Y_`` a a a a a a a a a a a a a a a a a a 377c|5}|||cdddS#1swxYwYdS)a` Delete a specified collection. :param collection_name: The name of the collection to delete. :type collection_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8drop_collectionrrIrGr6s r$rNzMilvus.drop_collectionu$     M7**?G*LL M M M M M M M M M M M M M M M M M M 9==c|5}|||cdddS#1swxYwYdS)a Checks whether a specified collection exists. :param collection_name: The name of the collection to check. :type collection_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: If specified collection exists :rtype: bool :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rMN)r8has_collectionrOs r$rSzMilvus.has_collections$     L7))/7)KK L L L L L L L L L L L L L L L L L LrQc|5}|||cdddS#1swxYwYdS)a Returns the schema of specified collection. Example: {'collection_name': 'create_collection_eXgbpOtn', 'auto_id': True, 'description': '', 'fields': [{'field_id': 100, 'name': 'INT32', 'description': '', 'type': 4, 'params': {}, {'field_id': 101, 'name': 'FLOAT_VECTOR', 'description': '', 'type': 101, 'params': {'dim': '128'}}]} :param collection_name: The name of the collection to describe. :type collection_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: The schema of collection to describe. :rtype: dict :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rMN)r8describe_collectionrOs r$rUzMilvus.describe_collections,     Q7...PP Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q Q QrQr c |5}|j||fd|i|cdddS#1swxYwYdS)a Loads a specified collection from disk to memory. :param collection_name: The name of the collection to load. :type collection_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :param replica_number: Number of replication in memory to load :type replica_number: int :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8load_collection)rrIreplica_numberrGr#r6s r$rWzMilvus.load_collections*     7*7*9@DJ                  rKc|5}|||cdddS#1swxYwYdS)aa Clear collection data from memory. :param collection_name: The name of collection to release. :type collection_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok )rIrGN)r8release_collectionrOs r$rZzMilvus.release_collections$     `7--oW^-__ ` ` ` ` ` ` ` ` ` ` ` ` ` ` ` ` ` `rQc |5}|j|fd|i|}d|D}t|d|d<|cdddS#1swxYwYdS)a Returns collection statistics information. Example: {"row_count": 10} :param collection_name: The name of collection. :type collection_name: str. :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: statistics information :rtype: dict :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGc(i|]}|j|jSrkeyvalue.0stats r$ z/Milvus.get_collection_stats..===tdh ===r& row_countN)r8get_collection_statsint)rrIrGr#r6statsresults r$rfzMilvus.get_collection_statss&     70G0\\'\U[\\E==u===F"%f[&9":":F;                    s6AAAc|5}||cdddS#1swxYwYdS)a8 Returns a list of all collection names. :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: List of collection names, return when operation is successful :rtype: list[str] :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rMN)r8list_collections)rrGr6s r$rkzMilvus.list_collectionss     =7++G+<< = = = = = = = = = = = = = = = = = =s 8<<c|5}||||cdddS#1swxYwYdS)a Creates a partition in a specified collection. You only need to import the parameters of partition_name to create a partition. A collection cannot hold partitions of the same tag, whilst you can insert the same tag in different collections. :param collection_name: The name of the collection to create partitions in. :type collection_name: str :param partition_name: The tag name of the partition to create. :type partition_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rMN)r8create_partitionrrIpartition_namerGr6s r$rmzMilvus.create_partitions.     ^7++O^U\+]] ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ :>>c|5}||||cdddS#1swxYwYdS)a Deletes the specified partition in a collection. Note that the default partition '_default' is not permitted to delete. When a partition deleted, all data stored in it will be deleted. :param collection_name: The name of the collection to delete partitions from. :type collection_name: str :param partition_name: The tag name of the partition to delete. :type partition_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rMN)r8drop_partitionrns r$rrzMilvus.drop_partition,s.     \7))/>SZ)[[ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \rpc|5}||||cdddS#1swxYwYdS)a Checks if a specified partition exists in a collection. :param collection_name: The name of the collection to find the partition in. :type collection_name: str :param partition_name: The tag name of the partition to check :type partition_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: Whether a specified partition exists in a collection. :rtype: bool :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rMN)r8 has_partitionrns r$rtzMilvus.has_partitionFs*     [7((.RY(ZZ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [rpc|5}|||||cdddS#1swxYwYdS)aK Load specified partitions from disk to memory. :param collection_name: The collection name which partitions belong to. :type collection_name: str :param partition_names: The specified partitions to load. :type partition_names: list[str] :param replica_number: The replication numbers to load. :type replica_number: int :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok )rIpartition_namesrXrGN)r8load_partitions)rrIrvrXrGr6s r$rwzMilvus.load_partitions^s0     7** / /- +                   ;??c|5}||||cdddS#1swxYwYdS)a Clear partitions data from memory. :param collection_name: The collection name which partitions belong to. :type collection_name: str :param partition_names: The specified partition to release. :type partition_names: list[str] :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok )rIrvrGN)r8release_partitionsrrIrvrGr6s r$rzzMilvus.release_partitions~s*     7-- /Za.                  rpc|5}|||cdddS#1swxYwYdS)a Returns a list of all partition tags in a specified collection. :param collection_name: The name of the collection to retrieve partition tags from. :type collection_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: A list of all partition tags in specified collection. :rtype: list[str] :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rMN)r8list_partitionsrOs r$r}zMilvus.list_partitionsrPrQc |5}|j||fd|i|}d|D}t|d|d<|cdddS#1swxYwYdS)a Returns partition statistics information. Example: {"row_count": 10} :param collection_name: The name of collection. :type collection_name: str. :param partition_name: The name of partition. :type partition_name: str. :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: statistics information :rtype: dict :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGc(i|]}|j|jSrr]r`s r$rcz.Milvus.get_partition_stats..rdr&reN)r8get_partition_statsrg)rrIrorGr#r6rhris r$rzMilvus.get_partition_statss,     7/G/9@DJE>=u===F"%f[&9":":F;                    s7AA Ac |5}|j||fd|i|cdddS#1swxYwYdS)a! Specify alias for a collection. Alias cannot be duplicated, you can't assign same alias to different collections. But you can specify multiple aliases for a collection, for example: before create_alias("collection_1", "bob"): collection_1's aliases = ["tom"] after create_alias("collection_1", "bob"): collection_1's aliases = ["tom", "bob"] :param collection_name: The name of collection. :type collection_name: str. :param alias: The alias of the collection. :type alias: str. :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8 create_aliasrrIaliasrGr#r6s r$rzMilvus.create_aliass6     [7'7'ZZZSYZZ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [ [rKc ~|5}|j|fd|i|cdddS#1swxYwYdS)al Delete an alias. This api no need to specify collection name because the milvus server knows which collection it belongs. For example: before drop_alias("bob"): collection_1's aliases = ["tom", "bob"] after drop_alias("bob"): collection_1's aliases = ["tom"] :param alias: The alias to be deleted. :type alias: str. :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8 drop_alias)rrrGr#r6s r$rzMilvus.drop_aliass0     H7%7%eGGWGGG H H H H H H H H H H H H H H H H H H 266c |5}|j||fd|i|cdddS#1swxYwYdS)a Change alias of a collection to another collection. If the alias doesn't exist, the api will return error. Alias cannot be duplicated, you can't assign same alias to different collections. This api can change alias owner collection, for example: before alter_alias("collection_2", "bob"): collection_1's aliases = ["bob"] collection_2's aliases = [] after alter_alias("collection_2", "bob"): collection_1's aliases = [] collection_2's aliases = ["bob"] :param collection_name: The name of collection. :type collection_name: str. :param alias: The new alias of the collection. :type alias: str. :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8 alter_aliasrs r$rzMilvus.alter_aliass:     Z7&7&YYwYRXYY Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z Z ZrKc |5}|j|||fd|i|cdddS#1swxYwYdS)a Creates an index for a field in a specified collection. Milvus does not support creating multiple indexes for a field. In a scenario where the field already has an index, if you create another one, the server will replace the existing index files with the new ones. Note that you need to call load_collection() or load_partitions() to make the new index take effect on searching tasks. :param collection_name: The name of the collection to create field indexes. :type collection_name: str :param field_name: The name of the field to create an index for. :type field_name: str :param params: Indexing parameters. :type params: dict There are examples of supported indexes: IVF_FLAT: ` { "metric_type":"L2", "index_type": "IVF_FLAT", "params":{"nlist": 1024} }` IVF_PQ: `{ "metric_type": "L2", "index_type": "IVF_PQ", "params": {"nlist": 1024, "m": 8, "nbits": 8} }` IVF_SQ8: `{ "metric_type": "L2", "index_type": "IVF_SQ8", "params": {"nlist": 1024} }` BIN_IVF_FLAT: `{ "metric_type": "JACCARD", "index_type": "BIN_IVF_FLAT", "params": {"nlist": 1024} }` HNSW: `{ "metric_type": "L2", "index_type": "HNSW", "params": {"M": 48, "efConstruction": 50} }` RHNSW_FLAT: `{ "metric_type": "L2", "index_type": "RHNSW_FLAT", "params": {"M": 48, "efConstruction": 50} }` RHNSW_PQ: `{ "metric_type": "L2", "index_type": "RHNSW_PQ", "params": {"M": 48, "efConstruction": 50, "PQM": 8} }` RHNSW_SQ: `{ "metric_type": "L2", "index_type": "RHNSW_SQ", "params": {"M": 48, "efConstruction": 50} }` ANNOY: `{ "metric_type": "L2", "index_type": "ANNOY", "params": {"n_trees": 8} }` :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :param kwargs: * *_async* (``bool``) -- Indicate if invoke asynchronously. When value is true, method returns a IndexFuture object; otherwise, method returns results from server. * *_callback* (``function``) -- The callback function which is invoked after server response successfully. It only take effect when _async is set to True. * *index_name* (``str``) -- The name of index which will be created. Then you can use the index name to check the state of index. If no index name is specified, default index name is used. :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8 create_index)rrI field_nameparamsrGr#r6s r$rzMilvus.create_index$sP     7'7'V=DHN                   488c|5}|||d|cdddS#1swxYwYdS)a Removes the index of a field in a specified collection. :param collection_name: The name of the collection to remove the field index from. :type collection_name: str :param field_name: The name of the field to remove the index of. :type field_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok )rIr index_namerGN)r8 drop_index)rrIrrGr6s r$rzMilvus.drop_indexs*     7%% /% &                  rxrc|5}||||cdddS#1swxYwYdS)aY Returns the schema of index built on specified field. Example: {'index_type': 'FLAT', 'metric_type': 'L2', 'params': {'nlist': 128}} :param collection_name: The name of the collection which field belong to. :type collection_name: str :param field_name: The name of field to describe. :type field_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: the schema of index built on specified field. :rtype: dict :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rMN)r8describe_indexrrIrrGr6s r$rzMilvus.describe_indexs,     X7))/:w)WW X X X X X X X X X X X X X X X X X Xrpc |5}|j|||fd|i|cdddS#1swxYwYdS)ut Inserts entities in a specified collection. :param collection_name: The name of the collection to insert entities in. :type collection_name: str. :param entities: The entities to insert. :type entities: list :param partition_name: The name of the partition to insert entities in. The default value is None. The server stores entities in the “_default” partition by default. :type partition_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :param kwargs: * *_async* (``bool``) -- Indicate if invoke asynchronously. When value is true, method returns a MutationFuture object; otherwise, method returns results from server. * *_callback* (``function``) -- The callback function which is invoked after server response successfully. It only take effect when _async is set to True. :return: list of ids of the inserted vectors. :rtype: list[int] :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8 batch_insert)rrIentitiesrorGr#r6s r$insertz Milvus.insertsB     7'7'>CJNT                  rc |5}|j|||fd|i|cdddS#1swxYwYdS)a Delete entities with an expression condition. And return results to show which primary key is deleted successfully :param collection_name: Name of the collection to delete entities from :type collection_name: str :param expr: The expression to specify entities to be deleted :type expr: str :param partition_name: Name of partitions that contain entities :type partition_name: str :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur :type timeout: float :return: list of ids of the deleted vectors. :rtype: list :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8delete)rrIexprrorGr#r6s r$rz Milvus.deletes2     d7!7>/4ccQXc\bcc d d d d d d d d d d d d d d d d d drc ~|5}|j|fd|i|cdddS#1swxYwYdS)a Internally, Milvus organizes data into segments, and indexes are built in a per-segment manner. By default, a segment will be sealed if it grows large enough (according to segment size configuration). If any index is specified on certain field, the index-creating task will be triggered automatically when a segment is sealed. The flush() call will seal all the growing segments immediately of the given collection, and force trigger the index-creating tasks. :param collection_names: The name of collection to flush. :type collection_names: list[str] :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :param kwargs: * *_async* (``bool``) -- Indicate if invoke asynchronously. When value is true, method returns a FlushFuture object; otherwise, method returns results from server. * *_callback* (``function``) -- The callback function which is invoked after server response successfully. It only take effect when _async is set to True. :return: None :rtype: NoneType :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8flush)rcollection_namesrGr#r6s r$rz Milvus.flush s@     N7 7=!1MM7MfMM N N N N N N N N N N N N N N N N N Nrc |5} | j||||||||f| | d| cdddS#1swxYwYdS)a% Searches a collection based on the given expression and returns query results. :param collection_name: The name of the collection to search. :type collection_name: str :param data: The vectors of search data, the length of data is number of query (nq), the dim of every vector in data must be equal to vector field's of collection. :type data: list[list[float]] :param anns_field: The vector field used to search of collection. :type anns_field: str :param param: The parameters of search, such as nprobe, etc. :type param: dict :param limit: The max number of returned record, we also called this parameter as topk. :type limit: int :param expression: The boolean expression used to filter attribute. :type expression: str :param partition_names: The names of partitions to search. :type partition_names: list[str] :param output_fields: The fields to return in the search result, not supported now. :type output_fields: list[str] :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :param round_decimal: The specified number of decimal places of returned distance :type round_decimal: int :param kwargs: * *_async* (``bool``) -- Indicate if invoke asynchronously. When value is true, method returns a SearchFuture object; otherwise, method returns results from server. * *_callback* (``function``) -- The callback function which is invoked after server response successfully. It only take effect when _async is set to True. * *consistency_level* (``str/int``) -- Which consistency level to use when searching in the collection. For details, see https://github.com/milvus-io/milvus/blob/master/docs/developer_guides/how-guarantee-ts-works.md. Note: this parameter will overwrite the same parameter user specified when creating the collection, if no consistency level was specified, search will use the collection consistency level. * *guarantee_timestamp* (``int``) -- This function instructs Milvus to see all operations performed before a provided timestamp. If no such timestamp is provided, then Milvus will search all operations performed to date. Note: only used in Customized consistency level. * *graceful_time* (``int``) -- Only used in bounded consistency level. If graceful_time is set, PyMilvus will use current timestamp minus the graceful_time as the `guarantee_timestamp`. This option is 5s by default if not set. :return: Query result. QueryResult is iterable and is a 2d-array-like class, the first dimension is the number of vectors to query (nq), the second dimension is the number of limit(topk). :rtype: QueryResult :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok ) round_decimalrGN)r8search) rrIdata anns_fieldparamlimit expressionrv output_fieldsrGrr#r6s r$rz Milvus.search,sF     7!7> ,                      rpc ~|5}|j|fd|i|cdddS#1swxYwYdS)a Notifies Proxy to return segments information from query nodes. :param collection_name: The name of the collection to get segments info. :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur. :type timeout: float :return: QuerySegmentInfo: QuerySegmentInfo is the growing segments's information in query cluster. :rtype: QuerySegmentInfo rGN)r8get_query_segment_inforrIrGr#r6s r$rzMilvus.get_query_segment_info~s     ^7171/]]7]V\]] ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^ ^rc|5}|||cdddS#1swxYwYdSz{ 'loading_progress': '100%', 'num_loaded_partitions': 3, 'not_loaded_partitions': [], } rMN)r8load_collection_progressrOs r$rzMilvus.load_collection_progresss     V733OW3UU V V V V V V V V V V V V V V V V V VrQc|5}||||cdddS#1swxYwYdSr)r8load_partitions_progressr{s r$rzMilvus.load_partitions_progresss     733'4                  rpc|5}|||cdddS#1swxYwYdSNrM)r8wait_for_loading_collectionrOs r$$wait_for_loading_collection_completez+Milvus.wait_for_loading_collection_completes      Y766PW6XX Y Y Y Y Y Y Y Y Y Y Y Y Y Y Y Y Y YrQc|5}||||cdddS#1swxYwYdSr)r8wait_for_loading_partitionsr{s r$$wait_for_loading_partitions_completez+Milvus.wait_for_loading_partitions_completes      766'7                  rpc|5}||||cdddS#1swxYwYdSr)r8get_index_build_progressrs r$rzMilvus.get_index_build_progresss      b733OZY`3aa b b b b b b b b b b b b b b b b b brpc|5}||||cdddS#1swxYwYdSr)r8wait_for_creating_indexrs r$rzMilvus.wait_for_creating_indexs      a722?JX_2`` a a a a a a a a a a a a a a a a a arpc|5}|||cdddS#1swxYwYdSr)r8dummy)r request_typerGr6s r$rz Milvus.dummys      @7==w=?? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @rQc |5}|j||||fd|i|cdddS#1swxYwYdS)a Query with a set of criteria, and results in a list of records that match the query exactly. :param collection_name: Name of the collection to retrieve entities from :type collection_name: str :param expr: The query expression :type expr: str :param output_fields: A list of fields to return :type output_fields: list[str] :param partition_names: Name of partitions that contain entities :type partition_names: list[str] :param timeout: An optional duration of time in seconds to allow for the RPC. When timeout is set to None, client waits until server response or error occur :type timeout: float :return: A list that contains all results :rtype: list :param kwargs: * *consistency_level* (``str/int``) -- Which consistency level to use during a query on the collection. For details, see https://github.com/milvus-io/milvus/blob/master/docs/developer_guides/how-guarantee-ts-works.md. Note: this parameter will overwrite the same parameter user specified when creating the collection, if no consistency level was specified, query will use the collection consistency level. * *guarantee_timestamp* (``int``) -- This function instructs Milvus to see all operations performed before a provided timestamp. If no such timestamp is specified, Milvus queries all operations performed to date. Note: only used in Customized consistency level. * *graceful_time* (``int``) -- Only used in bounded consistency level. If graceful_time is set, PyMilvus will use current timestamp minus the graceful_time as the `guarantee_timestamp`. This option is 5s by default if not set. :raises RpcError: If gRPC encounter an error :raises ParamError: If parameters are invalid :raises MilvusException: If the return result from server is not ok rGN)r8query)rrIrrrvrGr#r6s r$rz Milvus.querysb     7 7=}oOVZ`                   599rIc |5}|j||||fd|i|cdddS#1swxYwYdS)a Do load balancing operation from source query node to destination query node. :param collection_name: The collection to balance. :type collection_name: str :param src_node_id: The source query node id to balance. :type src_node_id: int :param dst_node_ids: The destination query node ids to balance. :type dst_node_ids: list[int] :param sealed_segment_ids: Sealed segment ids to balance. :type sealed_segment_ids: list[int] :param timeout: The timeout for this method, unit: second :type timeout: int :raises MilvusException: If query nodes not exist. :raises MilvusException: If sealed segments not exist. rGN)r8 load_balance)rrI src_node_id dst_node_idssealed_segment_idsrGr#r6s r$rzMilvus.load_balances:     7'7'"                       rFc |5}|j|f||d|cdddS#1swxYwYdS)a Do compaction for the collection. :param collection_name: The collection name to compact :type collection_name: str :param is_clustering: trigger clustering compaction :type is_clustering: bool :param timeout: The timeout for this method, unit: second :type timeout: int :return: the compaction ID :rtype: int :raises MilvusException: If collection name not exist.  is_clusteringrGN)r8compact)rrIrrGr#r6s r$rzMilvus.compacts$     7"7?/r@rErHrNrSrUrWrZrfrkrmrrrtrwrzr}rrrrrrrrrrrrrrrrrrrrr0rrgrrrrrrrrlistrr rrrrrrrrrrrrrrrr r r rrrrrrrrrrr"rr&r$rrsf.FOT4 444Z4"&F,<&/    $XX///GGG +a+a+a+aZMMMM*LLLL*QQQQ24````*2====$^^^^4\\\\4[[[[0@4MMMM*<[[[[<HHHH6ZZZZ@kkkkZ:XXXX2$$$$Ldddd8!N!N!N!NTPPPPd^^^^ VVVV    YYYY bbbbaaaa@@@@ 4444x %%%%%%NWZ0@D  6+/cc c cccc ZZ#ZRaZZZZ$TTCTGTTTT$OS))")47)@D) ))))V U U U U U UMMMMMM K K K Kaaaa"AAAAEEEEFFFFDDDDUUUUZZZZ]]]]RRRR\\\\RRRR  TTTT7; ;;;;KKKKCG W Ws$778 W W W WIIIIDDDD M MGX M M M M W W W WEIr&rN)urllibrpymilvus.decoratorsrpymilvus.exceptionsrrpymilvus.settingsrtypingrcheckr r grpc_handlerr typesr rrrrrrrr&r$r3s******;;;;;;;;$$$$$$////////%%%%%%ddddddddddr&