org.noear.solon.ai.rag.repository

类 ChromaRepository

java.lang.Object

org.noear.solon.ai.rag.repository.ChromaRepository

所有已实现的接口:

Repository, RepositoryLifecycle, RepositoryStorable

@Preview(value="3.1")
public class ChromaRepository
extends java.lang.Object
implements RepositoryStorable, RepositoryLifecycle

Chroma 矢量存储知识库 自动检测并支持 Chroma v1 和 v2 API 版本，默认使用 v2

从以下版本开始:

3.1

嵌套类概要

嵌套类

限定符和类型	类和说明
static class	ChromaRepository.Builder

方法概要

限定符和类型	方法和说明
static ChromaRepository.Builder	builder(EmbeddingModel embeddingModel, ChromaClient client)
static ChromaRepository.Builder	builder(EmbeddingModel embeddingModel, java.lang.String baseUrl) 创建 Builder（自动检测 API 版本）
static ChromaRepository.Builder	builder(EmbeddingModel embeddingModel, java.lang.String baseUrl, java.lang.String tenant, java.lang.String database) 创建支持完整配置的 Builder（自动检测 API 版本）
void	deleteById(java.lang.String... ids) 删除指定ID的文档
void	dropRepository() 注销仓库
boolean	existsById(java.lang.String id) 检查文档是否存在
void	initRepository() 初始化仓库 如果集合不存在，则自动创建新集合
boolean	isHealthy() 检查服务是否健康
void	save(java.util.List<Document> documents, java.util.function.BiConsumer<java.lang.Integer,java.lang.Integer> progressCallback) 批量存储文档（支持更新）
java.util.List<Document>	search(QueryCondition condition) 搜索文档

从类继承的方法 java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

从接口继承的方法 org.noear.solon.ai.rag.RepositoryStorable

asyncInsert, asyncSave, delete, exists, insert, insert, insert, save, save

从接口继承的方法 org.noear.solon.ai.rag.Repository

search

方法详细资料

initRepository

public void initRepository()
                    throws java.io.IOException

初始化仓库 如果集合不存在，则自动创建新集合

指定者:

initRepository 在接口中 RepositoryLifecycle

抛出:

java.io.IOException - 如果初始化失败

isHealthy

public boolean isHealthy()

检查服务是否健康

save

public void save(java.util.List<Document> documents,
                 java.util.function.BiConsumer<java.lang.Integer,java.lang.Integer> progressCallback)
          throws java.io.IOException

批量存储文档（支持更新）

指定者:

save 在接口中 RepositoryStorable

参数:

documents - 文档集

progressCallback - 进度回调

抛出:

java.io.IOException

deleteById

public void deleteById(java.lang.String... ids)
                throws java.io.IOException

删除指定ID的文档

指定者:

deleteById 在接口中 RepositoryStorable

参数:

ids - 文档IDs

抛出:

java.io.IOException

existsById

public boolean existsById(java.lang.String id)
                   throws java.io.IOException

检查文档是否存在

指定者:

existsById 在接口中 RepositoryStorable

参数:

id - 文档ID

抛出:

java.io.IOException

search

public java.util.List<Document> search(QueryCondition condition)
                                throws java.io.IOException

搜索文档

指定者:

search 在接口中 Repository

参数:

condition - 查询条件

抛出:

java.io.IOException

dropRepository

public void dropRepository()
                    throws java.io.IOException

注销仓库

指定者:

dropRepository 在接口中 RepositoryLifecycle

抛出:

java.io.IOException - 如果注销过程发生IO错误

builder

public static ChromaRepository.Builder builder(EmbeddingModel embeddingModel,
                                               ChromaClient client)

builder

public static ChromaRepository.Builder builder(EmbeddingModel embeddingModel,
                                               java.lang.String baseUrl)

创建 Builder（自动检测 API 版本）

参数:

embeddingModel - 向量模型

baseUrl - Chroma 服务器地址

返回:

Builder 实例

builder

public static ChromaRepository.Builder builder(EmbeddingModel embeddingModel,
                                               java.lang.String baseUrl,
                                               java.lang.String tenant,
                                               java.lang.String database)

创建支持完整配置的 Builder（自动检测 API 版本）

参数:

embeddingModel - 向量模型

baseUrl - Chroma 服务器地址

tenant - 租户名称 (仅 v2 API 需要)

database - 数据库名称 (仅 v2 API 需要)

返回:

Builder 实例