Parcourir la source

fix: Formatting

Rain6435 il y a 1 mois
Parent
commit
a1e62ab422
1 fichiers modifiés avec 12 ajouts et 4 suppressions
  1. 12 4
      backend/open_webui/retrieval/vector/dbs/pgvector.py

+ 12 - 4
backend/open_webui/retrieval/vector/dbs/pgvector.py

@@ -112,26 +112,34 @@ class PgvectorClient(VectorDBBase):
         try:
             # Ensure the pgvector extension is available
             # Use a conditional check to avoid permission issues on Azure PostgreSQL
-            self.session.execute(text("""
+            self.session.execute(
+                text(
+                    """
                 DO $$
                 BEGIN
                    IF NOT EXISTS (SELECT 1 FROM pg_extension WHERE extname = 'vector') THEN
                       CREATE EXTENSION IF NOT EXISTS vector;
                    END IF;
                 END $$;
-            """))
+            """
+                )
+            )
 
             if PGVECTOR_PGCRYPTO:
                 # Ensure the pgcrypto extension is available for encryption
                 # Use a conditional check to avoid permission issues on Azure PostgreSQL
-                self.session.execute(text("""
+                self.session.execute(
+                    text(
+                        """
                     DO $$
                     BEGIN
                        IF NOT EXISTS (SELECT 1 FROM pg_extension WHERE extname = 'pgcrypto') THEN
                           CREATE EXTENSION IF NOT EXISTS pgcrypto;
                        END IF;
                     END $$;
-                """))
+                """
+                    )
+                )
 
                 if not PGVECTOR_PGCRYPTO_KEY:
                     raise ValueError(