ailine/tools/test/reset_qdrant.py

#!/usr/bin/env python3
"""
删除 Qdrant 集合并重新索引
"""

import asyncio
import os
import sys

from backend.rag_core import QdrantVectorStore
from backend.app.model_services import get_embedding_service


async def delete_and_recreate():
    """删除并重新创建集合"""
    print("="*70)
    print("删除旧集合并重新创建...")
    print("="*70)
    
    embeddings = get_embedding_service()
    vs = QdrantVectorStore(collection_name="rag_documents", embeddings=embeddings)
    
    # 删除旧集合
    try:
        vs.delete_collection()
        print("✅ 旧集合已删除")
    except Exception as e:
        print(f"⚠️ 删除集合时出错（可能不存在）: {e}")
    
    # 重新创建
    vs.create_collection()
    print("✅ 新集合已创建")


if __name__ == "__main__":
    asyncio.run(delete_and_recreate())