Added more memory management and fixed system prompt bug

1 files changed, 4 insertions, 4 deletions

diff --git a/tests/test_database.py b/tests/test_database.py
index 290b985..e387123 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -34,7 +34,7 @@ def test_migrate_records_version(tmp_path):
     with database.connect() as connection:
         rows = connection.execute("SELECT version FROM schema_migrations").fetchall()
 
-    assert rows == [(1,)]
+    assert rows == [(1,), (2,), (3,)]
 
 
 def test_migrate_is_idempotent(tmp_path):
@@ -46,13 +46,13 @@ def test_migrate_is_idempotent(tmp_path):
     with database.connect() as connection:
         rows = connection.execute("SELECT version FROM schema_migrations").fetchall()
 
-    assert rows == [(1,)]
+    assert rows == [(1,), (2,), (3,)]
 
 
 def test_failed_migration_rolls_back(tmp_path):
     database = Database(tmp_path / "test.db")
 
-    migration = database.migrations_dir / "002_failing_migration.sql"
+    migration = database.migrations_dir / "999_failing_migration.sql"
     migration.write_text(
         """
         THIS IS INVALID SQL;
@@ -68,7 +68,7 @@ def test_failed_migration_rolls_back(tmp_path):
                 "SELECT version FROM schema_migrations"
             ).fetchall()
 
-        assert migrations == [(1,)]
+        assert migrations == [(1,), (2,), (3,)]
     finally:
         migration.unlink()