ChromaDB：如何检查集合是否存在？

Question

我想写一个脚本，重新创建一个chromadb的集合，也就是先删除之前的版本，然后从头开始创建一个新的。

client.delete_collection(name=COLLECTION_NAME)
collection = client.create_collection(
    name=COLLECTION_NAME,
    embedding_function=embedding_func,
    metadata={"hnsw:space": "cosine"},
)

不过，如果这个集合不存在，我会收到一个错误提示：

 File "*/lib/python3.11/site-packages/chromadb/api/segment.py", line 347, in delete_collection
raise ValueError(f"Collection {name} does not exist.")

ValueError: 集合操作不存在。

有没有什么命令可以检查一个集合是否存在？我在文档里没找到相关的内容。

Answer 1

你可以使用下面这个函数

    collection = client.get_or_create_collection(name="test")

这个函数要么获取已有的集合，要么创建一个新的集合