如何使用Python和Milvus向量数据库查看、加载和释放2-6级集合?

摘要:Collection(集合)核心运维操作 操作类型关键代码说明 查看所有集合列表 client.list_collections():返回 Milvus 中所有已创建的 Collection 名称列表 查询集合详细信息 client.
Collection(集合)核心运维操作 操作类型关键代码 / 说明 查看所有集合列表 client.list_collections():返回 Milvus 中所有已创建的 Collection 名称列表 查询集合详细信息 client.describe_collection("quick_setup"):返回指定集合的元数据(字段、索引、参数等) 查看集合加载状态 client.get_load_state("quick_setup"):检查集合是否已加载到内存(Milvus 中只有加载后的集合才能进行查询) 加载集合到内存 client.load_collection("quick_setup"):将集合加载到内存,为查询 / 搜索做准备 释放内存中的集合 client.release_collection("quick_setup"):释放集合占用的内存资源,降低服务器负载 看代码如何实现: # 过滤 pymilvus 依赖的 pkg_resources 废弃警告(setuptools≥81 版本触发) # 目的是消除版本兼容带来的无关警告,让控制台输出更整洁 import warnings warnings.filterwarnings("ignore", message=".*pkg_resources is deprecated as an API.*") from pymilvus import MilvusClient, DataType client = MilvusClient( uri="http://192.168.211.128:19530", token="root:Milvus" ) #查看collection列表 print('----查看collection列表----') res = client.list_collections() print(res) des_res = client.describe_collection( collection_name="quick_setup" ) #描述 print('----查看指定collection描述----') print(des_res) #查看加载状态 res = client.get_load_state( collection_name="quick_setup" ) print('----查看加载状态----') print(res) #加载collection client.load_collection( collection_name="quick_setup" ) res = client.get_load_state( collection_name="quick_setup" ) print('----查看加载状态----') print(res) #释放 Collections print('----释放 Collections----') client.release_collection( collection_name="quick_setup" ) print('----查看加载状态----') res = client.get_load_state( collection_name="quick_setup" ) print(res) 输出结果: ----查看collection列表---- ['two_table', 'one_talbe', 'three_table', 'quick_setup', 'custom_quick_setup'] ----查看指定collection描述---- {'collection_name': 'quick_setup', 'auto_id': False, 'num_shards': 1, 'description': '', 'fields': [{'field_id': 100, 'name': 'id', 'description': '', 'type': <DataType.INT64: 5>, 'params': {}, 'is_primary': True}, {'field_id': 101, 'name': 'vector', 'description': '', 'type': <DataType.FLOAT_VECTOR: 101>, 'params': {'dim': 5}}], 'functions': [], 'aliases': [], 'collection
阅读全文