jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: missing sqlite dialect (#4599)

Browse source 
fix: missing sqlite dialect import
This commit is contained in:
Ítalo Johnny 2024-11-14 13:30:55 -03:00 • committed by GitHub
commit 7d9315b356
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/services/database/service.py
View file

@ -13,6 +13,7 @@ from alembic import command, util
from alembic.config import Config from alembic.config import Config
from loguru import logger from loguru import logger
from sqlalchemy import event, inspect from sqlalchemy import event, inspect
from sqlalchemy.dialects import sqlite as dialect_sqlite
from sqlalchemy.engine import Engine from sqlalchemy.engine import Engine
from sqlalchemy.exc import OperationalError from sqlalchemy.exc import OperationalError
from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
@ -117,9 +118,7 @@ class DatabaseService(Service):
return connect_args return connect_args
def on_connection(self, dbapi_connection, _connection_record) -> None: def on_connection(self, dbapi_connection, _connection_record) -> None:
if isinstance( if isinstance(dbapi_connection, sqlite3.Connection | dialect_sqlite.aiosqlite.AsyncAdapt_aiosqlite_connection):
dbapi_connection, sqlite3.Connection | sa.dialects.sqlite.aiosqlite.AsyncAdapt_aiosqlite_connection
):
pragmas: dict = self.settings_service.settings.sqlite_pragmas or {} pragmas: dict = self.settings_service.settings.sqlite_pragmas or {}
pragmas_list = [] pragmas_list = []
for key, val in pragmas.items(): for key, val in pragmas.items():