highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-05 17:53:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

Extension creation in MT migration

Browse Source
This commit is contained in:
joachim-danswer
2025-05-28 15:47:40 -07:00
parent 450c29f4c6
commit c99f28652e
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
backend/alembic/versions/495cb26ce93e_create_knowlege_graph_tables.py
View File

@@ -27,9 +27,6 @@ depends_on = None
def upgrade() -> None:
# Enable pg_trgm extension if not already enabled
op.execute("CREATE EXTENSION IF NOT EXISTS pg_trgm")
# Create a new permission-less user to be later used for knowledge graph queries.
# The user will later get temporary read priviledges for a specific view that will be
# ad hoc generated specific to a knowledge graph query.
@@ -38,6 +35,10 @@ def upgrade() -> None:
# environment variables MUST be set. Otherwise, an exception will be raised.
if not MULTI_TENANT:
# Enable pg_trgm extension if not already enabled
op.execute("CREATE EXTENSION IF NOT EXISTS pg_trgm")
# Create read-only db user here only in single tenant mode. For multi-tenant mode,
# the user is created in the alembic_tenants migration.
if not (DB_READONLY_USER and DB_READONLY_PASSWORD):

4
backend/alembic_tenants/versions/3b9f09038764_add_read_only_kg_user.py
View File

@@ -23,6 +23,10 @@ depends_on = None
def upgrade() -> None:
if MULTI_TENANT:
# Enable pg_trgm extension if not already enabled
op.execute("CREATE EXTENSION IF NOT EXISTS pg_trgm")
# Create read-only db user here only in multi-tenant mode. For single-tenant mode,
# the user is created in the standard migration.
if not (DB_READONLY_USER and DB_READONLY_PASSWORD):