import json
from VikingDBMemoryService import VikingDBMemoryService
from Config.Config import VOLC_ACCESSKEY, VOLC_SECRETKEY

def drop_existing_collection(collection_name):
    # 初始化记忆库服务
    memory_service = VikingDBMemoryService(
        ak=VOLC_ACCESSKEY,
        sk=VOLC_SECRETKEY,
        host="api-knowledgebase.mlp.cn-beijing.volces.com",
        region="cn-beijing"
    )

    try:
        # 检查集合是否存在
        print(f"正在检查集合 '{collection_name}'...")
        memory_service.get_collection(collection_name)
        print(f"集合 '{collection_name}' 已存在，准备删除...")

        # 删除集合
        response = memory_service.drop_collection(collection_name)
        print(f"删除响应: {json.dumps(response, ensure_ascii=False, indent=2)}")
        print(f"集合 '{collection_name}' 删除成功")
        return True

    except Exception as e:
        error_msg = str(e)
        if "collection not exist" in error_msg:
            print(f"集合 '{collection_name}' 不存在，无需删除")
            return False
        else:
            print(f"操作失败: {error_msg}")
            raise

if __name__ == "__main__":
    # 可修改为目标集合名称
    TARGET_COLLECTION = "emotional_support"
    drop_existing_collection(TARGET_COLLECTION)