diff --git a/letta/orm/__init__.py b/letta/orm/__init__.py index b69737ac65..733ce81613 100644 --- a/letta/orm/__init__.py +++ b/letta/orm/__init__.py @@ -1,4 +1,5 @@ from letta.orm.base import Base +from letta.orm.file import FileMetadata from letta.orm.organization import Organization from letta.orm.source import Source from letta.orm.tool import Tool diff --git a/tests/test_managers.py b/tests/test_managers.py index 01af5c6dad..a5d8528a0c 100644 --- a/tests/test_managers.py +++ b/tests/test_managers.py @@ -3,7 +3,7 @@ import letta.utils as utils from letta.functions.functions import derive_openai_json_schema, parse_source_code -from letta.orm import Organization, Source, Tool, User +from letta.orm import FileMetadata, Organization, Source, Tool, User from letta.schemas.agent import CreateAgent from letta.schemas.embedding_config import EmbeddingConfig from letta.schemas.file import FileMetadata as PydanticFileMetadata @@ -38,6 +38,7 @@ def clear_tables(server: SyncServer): """Fixture to clear the organization table before each test.""" with server.organization_manager.session_maker() as session: + session.execute(delete(FileMetadata)) session.execute(delete(Source)) session.execute(delete(Tool)) # Clear all records from the Tool table session.execute(delete(User)) # Clear all records from the user table