jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: support Chroma vector store (#5015)

Browse source
This commit is contained in:
Bowen Liang 2024-06-13 18:02:18 +08:00 • committed by GitHub
commit cdc08a434f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
16 changed files with 1483 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/controllers/console/datasets/datasets.py
View file

@ -479,7 +479,7 @@ class DatasetRetrievalSettingApi(Resource):
vector_type = current_app.config['VECTOR_STORE']
match vector_type:
case VectorType.MILVUS | VectorType.RELYT | VectorType.PGVECTOR | VectorType.TIDB_VECTOR:
case VectorType.MILVUS | VectorType.RELYT | VectorType.PGVECTOR | VectorType.TIDB_VECTOR | VectorType.CHROMA:
return {
'retrieval_method': [
'semantic_search'
@ -501,7 +501,7 @@ class DatasetRetrievalSettingMockApi(Resource):
@account_initialization_required
def get(self, vector_type):
match vector_type:
case VectorType.MILVUS | VectorType.RELYT | VectorType.PGVECTOR | VectorType.TIDB_VECTOR:
case VectorType.MILVUS | VectorType.RELYT | VectorType.PGVECTOR | VectorType.TIDB_VECTOR | VectorType.CHROMA:
return {
'retrieval_method': [
'semantic_search'