对Chroma有疑问

Question

老师好，对Chroma的使用有疑问，之前在做demo的时候，都是将文本向量直接实例成本地的文件夹，没有启动Chroma并连接Chroma这个步骤，导致我以为向量数据库是不能存内容的，但是看了你的实例，所以Chroma和mysql这种数据库是一样的能直接把向量存进去？

之前做demo学习用的是from langchain_chroma import Chroma
查询用的是as_retriever

这里有点晕，不太明白

Answer 1

阿基米口 2025-03-10 19:22:47

你好，chroma是支持持久化（本地存储）

langchain_chroma是对chroma api的封装

from langchain_chroma import Chroma

• chroma 支持client/server模式

server模式启动的chroma
chroma run --path ./data

数据会持久化在data目录

首次使用添加完数据以后

chroma_client = chromadb.HttpClient(host="localhost", port=8000)
embedding_db = Chroma.from_documents(documents,
                                     embedding_cls.get_embedding_fun(),
                                     client=chroma_client,
                                     collection_name="zhidu_db2",
                                     )

第二次使用，直接输入集合名就可以获取（确保chroma服务启动状态）

import chromadb
from langchain_chroma import Chroma
chroma_client = chromadb.HttpClient(host="localhost", port=8000)
zhidu_db = Chroma("zhidu_db2", 
                  embedding_cls.get_embedding_fun(), 
                  client=chroma_client)

• 另外一种是本地持久化，就是上面你说的本地文件目录

embedding_db = Chroma.from_documents(documents,
                                     embedding_cls.get_embedding_fun(),
                                     persist_directory='./db',
                                     collection_name="zhidu_db2",
                                     )

第二次使用

zhidu_db = Chroma("zhidu_db2", 
                  embedding_cls.get_embedding_fun(), 
                  persist_directory='./data')

创建完向量集合之后，会通过collection集合命令来获取已经存储在本地或者存储远程服务器上的向量数据

可以用上面两种方式，都试一下

如果有任何疑问，都可以提出来