j!ddlZddlmZmZmZddlmZmZddlm Z edZ edZ GddZ dS)N)DictOptionalTypeVar)CollectionNotExistExceptionExceptionsMessage)ConfigIndex CollectionceZdZdedededefdZdZe de fdZ e defdZ e defd Z e defd Zd edefd Zd ZddeefdZdS)r collection field_name index_paramsreturnc ddlm}t||sttj||_||_||_| dtj |_ | ddrdS| }|j|jj|j|jfi|||jj}|D]!}|j|jkr|j|_ dS"dS)aCreates index on a specified field according to the index parameters. Args: collection(Collection): The collection in which the index is created field_name(str): The name of the field to create an index for. index_params(dict): Indexing parameters. kwargs: * *index_name* (``str``) -- The name of index which will be created. If no index name is specified, default index name will be used. Raises: MilvusException: If anything goes wrong. Examples: >>> from pymilvus import * >>> from pymilvus.schema import * >>> from pymilvus.types import DataType >>> connections.connect() >>> field1 = FieldSchema("int64", DataType.INT64, is_primary=True) >>> field2 = FieldSchema("fvec", DataType.FLOAT_VECTOR, is_primary=False, dim=128) >>> schema = CollectionSchema(fields=[field1, field2]) >>> collection = Collection(name='test_collection', schema=schema) >>> # insert some data >>> index_params = { ... "index_type": "IVF_FLAT", ... "metric_type": "L2", ... "params": {"nlist": 128}} >>> index = Index(collection, "fvec", index_params) >>> index.params {'index_type': 'IVF_FLAT', 'metric_type': 'L2', 'params': {'nlist': 128}} >>> index.collection_name test_collection >>> index.field_name fvec >>> index.drop() )r )message index_nameconstruct_onlyFN)r r isinstancerrCollectionType _collection _field_name _index_paramsgetr IndexName _index_name_get_connection create_indexname list_indexesr r) selfr r rkwargsr connindexesindexs \/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/pymilvus/orm/index.py__init__zIndex.__init__s#Z +******j11 X-6G6VWWW W%%)!::lF4DEE ::& . .  F##%%$*/1A4CU``Y_```##D$4$9::  E4#333#(#3 4  c4|jSN)rrr!s r&rzIndex._get_connectionXs//111r(c4tj|jS)zdict: The index parameters)copydeepcopyrr+s r&paramsz Index.params[s}T/000r(c|jjS)z&str: The corresponding collection name)rrr+s r&collection_namezIndex.collection_name`s$$r(c|jS)z"str: The corresponding field name.)rr+s r&r zIndex.field_namee r(c|jS)z"str: The corresponding index name.)rr+s r&rzIndex.index_namejr3r(othercV||kS)z4The order of the fields of index must be consistent.)to_dict)r!r5s r&__eq__z Index.__eq__os||~~00r(cB|jj|j|j|jdS)z;Put collection name, field name and index params into dict.)r fieldr index_param)r_namerrr/r+s r&r7z Index.to_dictss-*0%*;    r(Ntimeoutc |}||jj|j|j|dS)aDrop an index and its corresponding index files. Args: timeout(float, optional): 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 )r1r rr=N)r drop_indexrrr r)r!r=r"r#s r&dropz Index.drop|sP##%%  ,1      r(r*)__name__ __module__ __qualname__r strrr r'rpropertydictr/r1r rboolr8r7rfloatr@r(r&r r sO>>> >  >>>>@2221111X1%%%%X% C   X  C   X 1E1d1111     HUO      r() r-typingrrrpymilvus.exceptionsrrpymilvus.settingsrr r rIr(r&rMs **********NNNNNNNN$$$$$$ W\ " " s s s s s s s s s s r(