Fix tests

letta/schemas/file.py

@@ -29,6 +29,3 @@ class FileMetadata(FileMetadataBase):
     created_at: Optional[datetime] = Field(default_factory=datetime.utcnow, description="The creation date of the file.")
     updated_at: Optional[datetime] = Field(default_factory=datetime.utcnow, description="The update date of the file.")
     is_deleted: bool = Field(False, description="Whether this file is deleted or not.")
-
-    class Config:
-        extra = "allow"

letta/server/server.py

@@ -1669,7 +1669,6 @@ def attach_source_to_agent(
             data_source = self.source_manager.get_source_by_name(source_name=source_name, actor=user)
         else:
             raise ValueError(f"Need to provide at least source_id or source_name to find the source.")
-
         # get connection to data source storage
         source_connector = StorageConnector.get_storage_connector(TableType.PASSAGES, self.config, user_id=user_id)
 

tests/test_client.py

@@ -12,7 +12,7 @@
 from letta.agent import initialize_message_sequence
 from letta.client.client import LocalClient, RESTClient
 from letta.constants import DEFAULT_PRESET
-from letta.orm import Source
+from letta.orm import FileMetadata, Source
 from letta.schemas.agent import AgentState
 from letta.schemas.embedding_config import EmbeddingConfig
 from letta.schemas.enums import MessageRole, MessageStreamStatus
@@ -91,6 +91,7 @@ def clear_tables():
     from letta.server.server import db_context
 
     with db_context() as session:
+        session.execute(delete(FileMetadata))
         session.execute(delete(Source))
         session.commit()
 

tests/utils.py

@@ -24,8 +24,7 @@ def __init__(self, texts: List[str]):
     def find_files(self, source) -> Iterator[FileMetadata]:
         for text in self.texts:
             file_metadata = FileMetadata(
-                user_id="",
-                source_id="",
+                source_id=source.id,
                 file_name="",
                 file_path="",
                 file_type="",