diff --git a/src/langbot/pkg/api/http/controller/groups/knowledge/migration.py b/src/langbot/pkg/api/http/controller/groups/knowledge/migration.py index 68faa00b..fdcf2568 100644 --- a/src/langbot/pkg/api/http/controller/groups/knowledge/migration.py +++ b/src/langbot/pkg/api/http/controller/groups/knowledge/migration.py @@ -47,9 +47,9 @@ class KnowledgeMigrationRouterGroup(group.RouterGroup): return result.scalar() else: result = await self.ap.persistence_mgr.execute_async( - sqlalchemy.text( - "SELECT name FROM sqlite_master WHERE type='table' AND name=:table_name;" - ).bindparams(table_name=table_name) + sqlalchemy.text("SELECT name FROM sqlite_master WHERE type='table' AND name=:table_name;").bindparams( + table_name=table_name + ) ) return result.first() is not None @@ -231,9 +231,7 @@ class KnowledgeMigrationRouterGroup(group.RouterGroup): ) try: - await self.ap.plugin_connector.rag_on_kb_create( - external_plugin_id, kb_uuid, retriever_config - ) + await self.ap.plugin_connector.rag_on_kb_create(external_plugin_id, kb_uuid, retriever_config) task_context.trace(f'Restored external KB: {name} ({kb_uuid})') except Exception as e: warning = f'Failed to notify plugin for external KB {name} ({kb_uuid}): {e}' diff --git a/src/langbot/pkg/persistence/migrations/dbm020_knowledge_engine_plugin_architecture.py b/src/langbot/pkg/persistence/migrations/dbm020_knowledge_engine_plugin_architecture.py index de10df9d..616cb91a 100644 --- a/src/langbot/pkg/persistence/migrations/dbm020_knowledge_engine_plugin_architecture.py +++ b/src/langbot/pkg/persistence/migrations/dbm020_knowledge_engine_plugin_architecture.py @@ -59,18 +59,14 @@ class DBMigrateKnowledgeEnginePluginArchitecture(migration.DBMigration): async def _backup_knowledge_bases(self) -> bool: """Backup knowledge_bases data. Returns True if data was backed up.""" - result = await self.ap.persistence_mgr.execute_async( - sqlalchemy.text('SELECT COUNT(*) FROM knowledge_bases;') - ) + result = await self.ap.persistence_mgr.execute_async(sqlalchemy.text('SELECT COUNT(*) FROM knowledge_bases;')) count = result.scalar() if count == 0: return False # Drop backup table if it already exists (from a previous failed migration) if await self._table_exists('knowledge_bases_backup'): - await self.ap.persistence_mgr.execute_async( - sqlalchemy.text('DROP TABLE knowledge_bases_backup;') - ) + await self.ap.persistence_mgr.execute_async(sqlalchemy.text('DROP TABLE knowledge_bases_backup;')) await self.ap.persistence_mgr.execute_async( sqlalchemy.text('CREATE TABLE knowledge_bases_backup AS SELECT * FROM knowledge_bases;') @@ -103,9 +99,7 @@ class DBMigrateKnowledgeEnginePluginArchitecture(migration.DBMigration): async def _clear_knowledge_bases(self): """Clear all rows from knowledge_bases table (preserve table structure).""" - await self.ap.persistence_mgr.execute_async( - sqlalchemy.text('DELETE FROM knowledge_bases;') - ) + await self.ap.persistence_mgr.execute_async(sqlalchemy.text('DELETE FROM knowledge_bases;')) async def _add_columns_to_knowledge_bases(self): """Add new RAG plugin architecture columns to knowledge_bases table.""" @@ -154,15 +148,11 @@ class DBMigrateKnowledgeEnginePluginArchitecture(migration.DBMigration): row = result.first() if row is not None: await self.ap.persistence_mgr.execute_async( - sqlalchemy.text( - "UPDATE metadata SET value = 'true' WHERE key = 'rag_plugin_migration_needed';" - ) + sqlalchemy.text("UPDATE metadata SET value = 'true' WHERE key = 'rag_plugin_migration_needed';") ) else: await self.ap.persistence_mgr.execute_async( - sqlalchemy.text( - "INSERT INTO metadata (key, value) VALUES ('rag_plugin_migration_needed', 'true');" - ) + sqlalchemy.text("INSERT INTO metadata (key, value) VALUES ('rag_plugin_migration_needed', 'true');") ) self.ap.logger.info('Set rag_plugin_migration_needed=true in metadata.')