55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
重置向量数据库集合脚本
|
||
|
|
"""
|
||
|
|
|
||
|
|
import configparser
|
||
|
|
import time
|
||
|
|
from dashvector import Client
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
config = configparser.ConfigParser()
|
||
|
|
config.read('config.ini', encoding='utf-8')
|
||
|
|
|
||
|
|
print("=" * 50)
|
||
|
|
print("重置 DashVector 集合")
|
||
|
|
print("=" * 50)
|
||
|
|
|
||
|
|
api_key = config.get('dashvector', 'api_key')
|
||
|
|
endpoint = config.get('dashvector', 'endpoint')
|
||
|
|
collection_name = config.get('dashvector', 'collection_name')
|
||
|
|
dimension = config.getint('dashvector', 'vector_dimension')
|
||
|
|
|
||
|
|
client = Client(api_key=api_key, endpoint=endpoint)
|
||
|
|
|
||
|
|
# 删除
|
||
|
|
del_result = client.delete(collection_name)
|
||
|
|
if del_result.code == 0:
|
||
|
|
print(f"已删除集合: {collection_name}")
|
||
|
|
print("等待删除完成...")
|
||
|
|
time.sleep(3)
|
||
|
|
else:
|
||
|
|
print(f"集合不存在，跳过删除")
|
||
|
|
|
||
|
|
# 创建
|
||
|
|
create_result = client.create(collection_name, dimension=dimension)
|
||
|
|
if create_result.code == 0:
|
||
|
|
print(f"已创建集合: {collection_name} (维度={dimension})")
|
||
|
|
else:
|
||
|
|
print(f"创建失败: {create_result.message}")
|
||
|
|
return
|
||
|
|
|
||
|
|
# 验证
|
||
|
|
collections = client.list()
|
||
|
|
if collection_name in collections.output:
|
||
|
|
print(f"验证通过: 集合已在列表中")
|
||
|
|
else:
|
||
|
|
print(f"警告: 集合未出现在列表中，请稍后重试")
|
||
|
|
|
||
|
|
print("=" * 50)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|