public class ChromaClient
extends java.lang.Object
构造器和说明 |
---|
ChromaClient(java.util.Properties properties) |
ChromaClient(java.lang.String baseUrl) |
限定符和类型 | 方法和说明 |
---|---|
void |
addDocuments(java.lang.String collectionId,
java.util.List<java.lang.String> ids,
java.util.List<java.util.List<java.lang.Float>> embeddings,
java.util.List<java.lang.String> documents,
java.util.List<java.util.Map<java.lang.String,java.lang.Object>> metadatas)
添加文档
|
CollectionResponse |
createCollection(java.lang.String name,
java.util.Map<java.lang.String,java.lang.Object> metadata)
创建集合
|
void |
deleteCollection(java.lang.String collectionId)
删除集合
|
void |
deleteDocuments(java.lang.String collectionId,
java.util.List<java.lang.String> ids)
删除文档
|
boolean |
documentExists(java.lang.String collectionId,
java.lang.String id)
检查文档是否存在
|
CollectionResponse |
getCollectionStats(java.lang.String collectionName)
获取集合信息
|
boolean |
isHealthy()
检查服务是否健康
|
CollectionsResponse |
listCollections()
获取所有集合
|
QueryResponse |
queryDocuments(java.lang.String collectionId,
java.util.List<java.lang.Float> queryEmbedding,
int limit,
java.util.Map<java.lang.String,java.lang.Object> metadataFilter)
查询文档
|
void |
setBasicAuth(java.lang.String username,
java.lang.String password)
设置基础鉴权
|
void |
setBearerAuth(java.lang.String token)
设置令牌鉴权
|
public ChromaClient(java.lang.String baseUrl)
public ChromaClient(java.util.Properties properties)
public void setBasicAuth(java.lang.String username, java.lang.String password)
public void setBearerAuth(java.lang.String token)
public boolean isHealthy()
public CollectionsResponse listCollections() throws java.io.IOException
java.io.IOException
- 如果请求失败public CollectionResponse createCollection(java.lang.String name, java.util.Map<java.lang.String,java.lang.Object> metadata) throws java.io.IOException
name
- 集合名称metadata
- 集合元数据java.io.IOException
- 如果创建失败public CollectionResponse getCollectionStats(java.lang.String collectionName) throws java.io.IOException
collectionName
- 集合Namejava.io.IOException
- 如果请求失败public void deleteCollection(java.lang.String collectionId) throws java.io.IOException
collectionId
- 集合IDjava.io.IOException
- 如果删除失败public void addDocuments(java.lang.String collectionId, java.util.List<java.lang.String> ids, java.util.List<java.util.List<java.lang.Float>> embeddings, java.util.List<java.lang.String> documents, java.util.List<java.util.Map<java.lang.String,java.lang.Object>> metadatas) throws java.io.IOException
collectionId
- 集合IDids
- 文档ID列表embeddings
- 文档向量列表documents
- 文档内容列表metadatas
- 文档元数据列表java.io.IOException
- 如果添加失败public void deleteDocuments(java.lang.String collectionId, java.util.List<java.lang.String> ids) throws java.io.IOException
collectionId
- 集合IDids
- 文档ID列表java.io.IOException
- 如果删除失败public boolean documentExists(java.lang.String collectionId, java.lang.String id) throws java.io.IOException
collectionId
- 集合IDid
- 文档IDjava.io.IOException
- 如果请求失败public QueryResponse queryDocuments(java.lang.String collectionId, java.util.List<java.lang.Float> queryEmbedding, int limit, java.util.Map<java.lang.String,java.lang.Object> metadataFilter) throws java.io.IOException
collectionId
- 集合IDqueryEmbedding
- 查询向量limit
- 结果数量限制metadataFilter
- 元数据过滤条件java.io.IOException
- 如果查询失败