Add Volcengine VikingDB as new vector provider (#9287)

2024-10-13 21:26:05 +08:00 · 2024-10-13 21:26:05 +08:00 · d15ba3939d
commit d15ba3939d
parent 1ec83e4969
15 changed files with 627 additions and 3 deletions
--- a/api/controllers/console/datasets/datasets.py
+++ b/api/controllers/console/datasets/datasets.py
@ -618,6 +618,7 @@ class DatasetRetrievalSettingApi(Resource):
                | VectorType.TENCENT
                | VectorType.PGVECTO_RS
                | VectorType.BAIDU
+                | VectorType.VIKINGDB
            ):
                return {"retrieval_method": [RetrievalMethod.SEMANTIC_SEARCH.value]}
            case (
@ -655,6 +656,7 @@ class DatasetRetrievalSettingMockApi(Resource):
                | VectorType.TENCENT
                | VectorType.PGVECTO_RS
                | VectorType.BAIDU
+                | VectorType.VIKINGDB
            ):
                return {"retrieval_method": [RetrievalMethod.SEMANTIC_SEARCH.value]}
            case (