From 7d9315b356fa71b323164978fdc011979eb190a8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=8Dtalo=20Johnny?= Date: Thu, 14 Nov 2024 13:30:55 -0300 Subject: [PATCH] fix: missing sqlite dialect (#4599) fix: missing sqlite dialect import --- src/backend/base/langflow/services/database/service.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/services/database/service.py b/src/backend/base/langflow/services/database/service.py index 22d05f2e3..18e5bad93 100644 --- a/src/backend/base/langflow/services/database/service.py +++ b/src/backend/base/langflow/services/database/service.py @@ -13,6 +13,7 @@ from alembic import command, util from alembic.config import Config from loguru import logger from sqlalchemy import event, inspect +from sqlalchemy.dialects import sqlite as dialect_sqlite from sqlalchemy.engine import Engine from sqlalchemy.exc import OperationalError from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine @@ -117,9 +118,7 @@ class DatabaseService(Service): return connect_args def on_connection(self, dbapi_connection, _connection_record) -> None: - if isinstance( - dbapi_connection, sqlite3.Connection | sa.dialects.sqlite.aiosqlite.AsyncAdapt_aiosqlite_connection - ): + if isinstance(dbapi_connection, sqlite3.Connection | dialect_sqlite.aiosqlite.AsyncAdapt_aiosqlite_connection): pragmas: dict = self.settings_service.settings.sqlite_pragmas or {} pragmas_list = [] for key, val in pragmas.items():