diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_database.py | 2 | ||||
| -rw-r--r-- | tests/test_memory_store.py | 6 |
2 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_database.py b/tests/test_database.py index e33fdae..290b985 100644 --- a/tests/test_database.py +++ b/tests/test_database.py @@ -2,7 +2,7 @@ import sqlite3 import pytest -from database.database import Database +from personal_agent.database.database import Database def test_migrate_creates_schema(tmp_path): diff --git a/tests/test_memory_store.py b/tests/test_memory_store.py index f4862cb..8f3407d 100644 --- a/tests/test_memory_store.py +++ b/tests/test_memory_store.py @@ -1,9 +1,9 @@ from datetime import UTC, datetime from uuid import UUID, uuid7 -from database.database import Database -from memory import LongTermMemory, MemoryCategory -from memory_store import LongTermMemoryStore +from personal_agent.database.database import Database +from personal_agent.memory.models import LongTermMemory, MemoryCategory +from personal_agent.memory.store import LongTermMemoryStore def create_store(tmp_path): |