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

chore: reuse existing test functions with upstash vdb (#9679)

Browse source
This commit is contained in:
ice yao 2024-10-23 10:42:11 +08:00 • committed by GitHub
commit ceb2c4f3ef
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 7 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -619,6 +619,7 @@ class DatasetRetrievalSettingApi(Resource):
| VectorType.PGVECTO_RS
| VectorType.BAIDU
| VectorType.VIKINGDB
| VectorType.UPSTASH
):
return {"retrieval_method": [RetrievalMethod.SEMANTIC_SEARCH.value]}
case (
@ -630,7 +631,6 @@ class DatasetRetrievalSettingApi(Resource):
| VectorType.ORACLE
| VectorType.ELASTICSEARCH
| VectorType.PGVECTOR
| VectorType.UPSTASH
):
return {
"retrieval_method": [
@ -658,6 +658,7 @@ class DatasetRetrievalSettingMockApi(Resource):
| VectorType.PGVECTO_RS
| VectorType.BAIDU
| VectorType.VIKINGDB
| VectorType.UPSTASH
):
return {"retrieval_method": [RetrievalMethod.SEMANTIC_SEARCH.value]}
case (
@ -669,7 +670,6 @@ class DatasetRetrievalSettingMockApi(Resource):
| VectorType.ORACLE
| VectorType.ELASTICSEARCH
| VectorType.PGVECTOR
| VectorType.UPSTASH
):
return {
"retrieval_method": [