Просмотр исходного кода

chore: fix wrong VectorType match case (#8857)

tags/0.9.0
ice yao 1 год назад
Родитель
Сommit
27e33fb15c
Аккаунт пользователя с таким Email не найден

+ 2
- 1
api/controllers/console/datasets/datasets.py Просмотреть файл

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

+ 1
- 1
api/core/rag/datasource/vdb/tencent/tencent_vector.py Просмотреть файл

@@ -56,7 +56,7 @@ class TencentVector(BaseVector):
return self._client.create_database(database_name=self._client_config.database)

def get_type(self) -> str:
return "tencent"
return VectorType.TENCENT

def to_index_struct(self) -> dict:
return {"type": self.get_type(), "vector_store": {"class_prefix": self._collection_name}}

Загрузка…
Отмена
Сохранить