121 lines
3.6 KiB
Python
121 lines
3.6 KiB
Python
|
"""
|
|||
|
|
清理 szmp@ash@2026 租户下不需要的 Qdrant collections
|
|||
|
|
保留：8559ebc9-bfaf-4211-8fe3-ee2b22a5e29c, 30c19c84-8f69-4768-9d23-7f4a5bc3627a
|
|||
|
|
删除：其他所有 collections
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import asyncio
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
|||
|
|
|
|||
|
|
from qdrant_client import AsyncQdrantClient
|
|||
|
|
from app.core.config import get_settings
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def cleanup_collections():
|
|||
|
|
"""清理 collections"""
|
|||
|
|
settings = get_settings()
|
|||
|
|
client = AsyncQdrantClient(url=settings.qdrant_url)
|
|||
|
|
|
|||
|
|
tenant_id = "szmp@ash@2026"
|
|||
|
|
safe_tenant_id = tenant_id.replace('@', '_')
|
|||
|
|
prefix = f"kb_{safe_tenant_id}"
|
|||
|
|
|
|||
|
|
# 保留的 kb_id 前缀（前8位）
|
|||
|
|
keep_kb_ids = [
|
|||
|
|
"8559ebc9",
|
|||
|
|
"30c19c84",
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
print(f"🔍 扫描租户 {tenant_id} 的 collections...")
|
|||
|
|
print(f" 前缀: {prefix}")
|
|||
|
|
print(f" 保留: {keep_kb_ids}")
|
|||
|
|
print("-" * 80)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
collections = await client.get_collections()
|
|||
|
|
|
|||
|
|
# 找出该租户的所有 collections
|
|||
|
|
tenant_collections = [
|
|||
|
|
c.name for c in collections.collections
|
|||
|
|
if c.name.startswith(prefix)
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
print(f"\n📊 找到 {len(tenant_collections)} 个 collections:")
|
|||
|
|
for name in sorted(tenant_collections):
|
|||
|
|
# 检查是否需要保留
|
|||
|
|
should_keep = any(kb_id in name for kb_id in keep_kb_ids)
|
|||
|
|
status = "✅ 保留" if should_keep else "❌ 删除"
|
|||
|
|
print(f" {status} {name}")
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 80)
|
|||
|
|
print("开始删除...")
|
|||
|
|
print("=" * 80)
|
|||
|
|
|
|||
|
|
deleted = []
|
|||
|
|
skipped = []
|
|||
|
|
|
|||
|
|
for collection_name in tenant_collections:
|
|||
|
|
# 检查是否需要保留
|
|||
|
|
should_keep = any(kb_id in collection_name for kb_id in keep_kb_ids)
|
|||
|
|
|
|||
|
|
if should_keep:
|
|||
|
|
print(f"\n⏭️ 跳过 {collection_name} (保留)")
|
|||
|
|
skipped.append(collection_name)
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
print(f"\n🗑️ 删除 {collection_name}...")
|
|||
|
|
try:
|
|||
|
|
await client.delete_collection(collection_name)
|
|||
|
|
print(f" ✅ 已删除")
|
|||
|
|
deleted.append(collection_name)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f" ❌ 删除失败: {e}")
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 80)
|
|||
|
|
print("清理完成!")
|
|||
|
|
print("=" * 80)
|
|||
|
|
print(f"\n📈 统计:")
|
|||
|
|
print(f" 保留: {len(skipped)} 个")
|
|||
|
|
for name in skipped:
|
|||
|
|
print(f" - {name}")
|
|||
|
|
print(f"\n 删除: {len(deleted)} 个")
|
|||
|
|
for name in deleted:
|
|||
|
|
print(f" - {name}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"\n❌ 错误: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
finally:
|
|||
|
|
await client.close()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# 安全确认
|
|||
|
|
print("=" * 80)
|
|||
|
|
print("⚠️ 警告: 此操作将永久删除以下 collections:")
|
|||
|
|
print(" - kb_szmp_ash_2026")
|
|||
|
|
print(" - kb_szmp_ash_2026_fa4c1d61")
|
|||
|
|
print(" - kb_szmp_ash_2026_3ddf0ce7")
|
|||
|
|
print("\n 保留:")
|
|||
|
|
print(" - kb_szmp_ash_2026_8559ebc9")
|
|||
|
|
print(" - kb_szmp_ash_2026_30c19c84")
|
|||
|
|
print("=" * 80)
|
|||
|
|
print("\n确认删除? (yes/no): ", end="")
|
|||
|
|
|
|||
|
|
# 在非交互环境自动确认
|
|||
|
|
import os
|
|||
|
|
if os.environ.get('AUTO_CONFIRM') == 'true':
|
|||
|
|
response = 'yes'
|
|||
|
|
print('yes (auto)')
|
|||
|
|
else:
|
|||
|
|
response = input().strip().lower()
|
|||
|
|
|
|||
|
|
if response in ('yes', 'y'):
|
|||
|
|
asyncio.run(cleanup_collections())
|
|||
|
|
else:
|
|||
|
|
print("\n❌ 已取消")
|