【Spring AI】05. 向量数据库-Chroma

Chroma

本节将指导您如何设置 Chroma VectorStore ，并使用组件进行存储文档嵌入向量和相似性检索。

Chroma Container

什么是 Chroma？

---

Chroma 是开源的嵌入向量数据库。它提供了文档嵌入向量、内容和元数据存储，并配备了通过这些嵌入向量进行搜索的工具，包括元数据过滤。

先决条件

1. OpenAI 账户：在 OpenAI Signup 注册并在 API Keys 生成令牌。
2. 访问 ChromeDB。后面会介绍如何使用 Docker 容器在本地启动 ChromaDB。
   在启动时，如果尚未预配完全，则ChromaVectorStore会创建所需的组件。

---

配置

要设置 ChromaVectorStore，您需要提供您的 OpenAI API 密钥。请将其设置为环境变量，如下所示：

export SPRING_AI_OPENAI_API_KEY='Your_OpenAI_API_Key'

依赖项

将这些依赖项添加到您的项目中：

• OpenAI：用于计算嵌入的必需项。

  <dependency>
     <groupId>org.springframework.ai</groupId>
     <artifactId>spring-ai-openai-spring-boot-starter</artifactId>
  </dependency>
  * Chroma VectorStore
  <dependency>
    <groupId>org.springframework.ai</groupId>
    <artifactId>spring-ai-chroma-store</artifactId>
  </