mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-25 16:56:22 +02:00
62e698d8aa
- Updated maximum file size limit to 500 MB per file. - Removed restrictions on the number of files per upload and total upload size. - Enhanced handling of user-mentioning documents in the knowledge base search middleware. - Improved document reading and processing logic to accommodate new features and optimizations.
50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
"""create zero_publication for zero-cache replication
|
|
|
|
Restricts zero-cache replication to only the tables the frontend
|
|
queries via Zero, instead of replicating all tables in public schema.
|
|
|
|
See: https://zero.rocicorp.dev/docs/zero-cache-config#app-publications
|
|
|
|
Revision ID: 116
|
|
Revises: 115
|
|
"""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
|
|
from alembic import op
|
|
|
|
revision: str = "116"
|
|
down_revision: str | None = "115"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
PUBLICATION_NAME = "zero_publication"
|
|
|
|
TABLES = [
|
|
"notifications",
|
|
"documents",
|
|
"folders",
|
|
"search_source_connectors",
|
|
"new_chat_messages",
|
|
"chat_comments",
|
|
"chat_session_state",
|
|
]
|
|
|
|
|
|
def upgrade() -> None:
|
|
conn = op.get_bind()
|
|
exists = conn.execute(
|
|
sa.text("SELECT 1 FROM pg_publication WHERE pubname = :name"),
|
|
{"name": PUBLICATION_NAME},
|
|
).fetchone()
|
|
if not exists:
|
|
table_list = ", ".join(TABLES)
|
|
conn.execute(
|
|
sa.text(f"CREATE PUBLICATION {PUBLICATION_NAME} FOR TABLE {table_list}")
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.execute(f"DROP PUBLICATION IF EXISTS {PUBLICATION_NAME}")
|