AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-11 18:49:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix wrong VectorType match case (#8857)

Browse Source
This commit is contained in:
ice yao 2024-09-28 10:54:04 +08:00 committed by GitHub
parent 55e6123db9
commit 27e33fb15c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/controllers/console/datasets/datasets.py
View File

@ -563,10 +563,10 @@ class DatasetRetrievalSettingApi(Resource):
case ( case (
VectorType.MILVUS VectorType.MILVUS
| VectorType.RELYT | VectorType.RELYT
| VectorType.PGVECTOR
| VectorType.TIDB_VECTOR | VectorType.TIDB_VECTOR
| VectorType.CHROMA | VectorType.CHROMA
| VectorType.TENCENT | VectorType.TENCENT
| VectorType.PGVECTO_RS
): ):
return {"retrieval_method": [RetrievalMethod.SEMANTIC_SEARCH.value]} return {"retrieval_method": [RetrievalMethod.SEMANTIC_SEARCH.value]}
case ( case (
@ -577,6 +577,7 @@ class DatasetRetrievalSettingApi(Resource):
| VectorType.MYSCALE | VectorType.MYSCALE
| VectorType.ORACLE | VectorType.ORACLE
| VectorType.ELASTICSEARCH | VectorType.ELASTICSEARCH
| VectorType.PGVECTOR
): ):
return { return {
"retrieval_method": [ "retrieval_method": [

2
api/core/rag/datasource/vdb/tencent/tencent_vector.py
View File

@ -56,7 +56,7 @@ class TencentVector(BaseVector):
return self._client.create_database(database_name=self._client_config.database) return self._client.create_database(database_name=self._client_config.database)
def get_type(self) -> str: def get_type(self) -> str:
return "tencent" return VectorType.TENCENT
def to_index_struct(self) -> dict: def to_index_struct(self) -> dict:
return {"type": self.get_type(), "vector_store": {"class_prefix": self._collection_name}} return {"type": self.get_type(), "vector_store": {"class_prefix": self._collection_name}}