diff --git a/vibe_bot/database.py b/vibe_bot/database.py index 9da4b5e..255716a 100644 --- a/vibe_bot/database.py +++ b/vibe_bot/database.py @@ -292,8 +292,9 @@ class ChatDatabase: """, (f"{message_id}_response",), ) - response: str = cursor.fetchone()[0] - results.append((content, response, similarity)) + response_row = cursor.fetchone() + if response_row: + results.append((content, response_row[0], similarity)) conn.close()