持久化客户端与集合
持久化客户端与集合
import chromadb
client = chromadb.PersistentClient(path="./chroma_db")
col = client.get_or_create_collection(
name="kb",
metadata={"hnsw:space": "cosine"},
)
col.add(ids=["c1", "c2"], documents=["第一段文本", "第二段文本"],
metadatas=[{"source": "faq"}, {"source": "faq"}])
res = col.query(query_texts=["文本检索"], n_results=3)
print(res["ids"], res["distances"])默认 embedding 函数适合快速演示；中文场景建议显式接入自己的 embedding 函数并统一维度。