ChromaDB 向量库优化技巧实战

chroma 一步步使用

安装

# 安装chromadb
pip install chromadb,sentence_transformers

# 不启动服务会出现sock.connect(sa)TimeoutError: timed out
chroma run

服务启动后，您将看到类似以下输出：

建立连接

部署完成后，需要建立与Chroma服务的连接：

import chromadb

# 创建了临时客户端。程序终止时，您提取的任何数据都会丢失
client = chromadb.Client()

创建向量集合

chroma连接若直接创建集合(“Collection”（集合）是存储向量的基本单位，类似于关系型数据库中的表)，会使用内置的嵌入模型all-MiniLM-L6-v2。

# 首先导入embedding模型
from chromadb.utils import embedding_functions



collection = chroma_client.create_collection(name="my_collection")
collection.add(
    documents=[