feat: add upstash as a new vector database provider (#9644)

2024-10-23 09:16:35 +08:00
parent 999d3f1539
commit 8e7a752b2a
16 changed files with 365 additions and 4 deletions
--- a/api/controllers/console/datasets/datasets.py
+++ b/api/controllers/console/datasets/datasets.py
@@ -630,6 +630,7 @@ class DatasetRetrievalSettingApi(Resource):
                | VectorType.ORACLE
                | VectorType.ELASTICSEARCH
                | VectorType.PGVECTOR
+                | VectorType.UPSTASH
            ):
                return {
                    "retrieval_method": [
@@ -668,6 +669,7 @@ class DatasetRetrievalSettingMockApi(Resource):
                | VectorType.ORACLE
                | VectorType.ELASTICSEARCH
                | VectorType.PGVECTOR
+                | VectorType.UPSTASH
            ):
                return {
                    "retrieval_method": [