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

2024-10-23 09:16:35 +08:00 · 2024-10-23 09:16:35 +08:00 · 8e7a752b2a
commit 8e7a752b2a
parent 999d3f1539
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": [