PGVector Support in SAML (#919)

* feat: add support for pgvector in SAML * fix: typo
superagent-ai · Mar 28, 2024 · 3f60fe5 · 3f60fe5
1 parent aa91d17
commit 3f60fe5
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git a/libs/superagent/app/api/workflow_configs/saml_schema.py b/libs/superagent/app/api/workflow_configs/saml_schema.py
@@ -26,6 +26,7 @@ class SuperragDatabaseProvider(str, Enum):
     pinecone = "pinecone"
     weaviate = "weaviate"
     qdrant = "qdrant"
+    pgvector = "pgvector"
 
 
 class SuperragIndex(BaseModel):

diff --git a/libs/superagent/app/utils/helpers.py b/libs/superagent/app/utils/helpers.py
@@ -114,6 +114,8 @@ def get_superrag_compatible_credentials(credentials: dict):
         # weaviate
         "WEAVIATE_API_KEY": "api_key",
         "WEAVIATE_URL": "host",
+        # PGVector
+        "SUPABASE_DB_URL": "database_uri",
     }
 
     superrag_credentials = {}

diff --git a/libs/superagent/app/vectorstores/base.py b/libs/superagent/app/vectorstores/base.py
@@ -21,6 +21,8 @@
     "astra": "ASTRA_DB",
     "weaviate": "WEAVIATE",
     "supabase": "SUPABASE",
+    # NOTE: we used "supabase" name initially for the pgvector databases in our database
+    "pgvector": "SUPABASE",
 }
 
 REVERSE_VECTOR_DB_MAPPING = {v: k for k, v in VECTOR_DB_MAPPING.items()}

diff --git a/libs/ui/config/site.ts b/libs/ui/config/site.ts
@@ -621,16 +621,16 @@ export const siteConfig = {
       name: "Supabase",
       logo: "/supabase.png",
       description:
-        "The pgvector extension is particularly useful for tasks such as vector similarity search, retrieval, generation, and clustering",
-      formDescription: "Please enter your Supabase Pgvector credentials.",
+        "The PGVector extension is particularly useful for tasks such as vector similarity search, retrieval, generation, and clustering",
+      formDescription: "Please enter your Supabase PGVector credentials.",
       metadata: [
         {
           key: "SUPABASE_DB_URL",
           type: "input",
-          label: "Database Connection URL",
+          label: "Database Connection URI",
           placeholder: "postgres://postgres:postgres@localhost:5432/postgres",
           helpText:
-            "The connection URL for your database. You can find this in your Supabase dashboard.",
+            "The connection URI for your database. You can find this in your Supabase dashboard.",
         },
         {
           key: "SUPABASE_TABLE_NAME",