From d430fe9551eb4013ddca7d4eb39fde6254f3bf1b Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Tue, 10 Jun 2025 11:30:54 +0400 Subject: [PATCH] refac --- backend/open_webui/retrieval/vector/dbs/pgvector.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/backend/open_webui/retrieval/vector/dbs/pgvector.py b/backend/open_webui/retrieval/vector/dbs/pgvector.py index de4073e126..632937ef5b 100644 --- a/backend/open_webui/retrieval/vector/dbs/pgvector.py +++ b/backend/open_webui/retrieval/vector/dbs/pgvector.py @@ -92,6 +92,15 @@ class PgvectorClient(VectorDBBase): # Ensure the pgvector extension is available self.session.execute(text("CREATE EXTENSION IF NOT EXISTS vector;")) + if PGVECTOR_PGCRYPTO: + # Ensure the pgcrypto extension is available for encryption + self.session.execute(text("CREATE EXTENSION IF NOT EXISTS pgcrypto;")) + + if not PGVECTOR_PGCRYPTO_KEY: + raise ValueError( + "PGVECTOR_PGCRYPTO_KEY must be set when PGVECTOR_PGCRYPTO is enabled." + ) + # Check vector length consistency self.check_vector_length()