github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(app): ensure workflow record get_many stmt is terminated

Browse Source
This commit is contained in:
psychedelicious
2025-03-04 15:05:04 +10:00
parent 84e86819b8
commit 5073c7d0a3
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/app/services/workflow_records/workflow_records_sqlite.py
View File

@@ -168,10 +168,10 @@ class SqliteWorkflowRecordsStorage(WorkflowRecordsStorageBase):
wildcard_query = "%" + stripped_query + "%"
if categories:
main_query += " AND (name LIKE ? OR description LIKE ?) "
count_query += " AND (name LIKE ? OR description LIKE ?);"
count_query += " AND (name LIKE ? OR description LIKE ?)"
else:
main_query += " WHERE name LIKE ? OR description LIKE ? "
count_query += " WHERE name LIKE ? OR description LIKE ?;"
count_query += " WHERE name LIKE ? OR description LIKE ?"
main_params.extend([wildcard_query, wildcard_query])
count_params.extend([wildcard_query, wildcard_query])
@@ -181,6 +181,9 @@ class SqliteWorkflowRecordsStorage(WorkflowRecordsStorageBase):
main_query += " LIMIT ? OFFSET ?"
main_params.extend([per_page, page * per_page])
main_query += ";"
count_query += ";"
cursor = self._conn.cursor()
cursor.execute(main_query, main_params)
rows = cursor.fetchall()