如何使用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
