fix: patch test

letta-ai · Dec 21, 2024 · c681f66 · c681f66
1 parent ed887cb
commit c681f66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_client_legacy.py b/tests/test_client_legacy.py
@@ -24,12 +24,12 @@
     SystemMessage,
     ToolCallMessage,
     ToolReturnMessage,
+    UsageMessage,
     UserMessage,
 )
 from letta.schemas.letta_response import LettaResponse, LettaStreamingResponse
 from letta.schemas.llm_config import LLMConfig
 from letta.schemas.message import MessageCreate
-from letta.schemas.usage import LettaUsageStatistics
 from letta.services.helpers.agent_manager_helper import initialize_message_sequence
 from letta.services.organization_manager import OrganizationManager
 from letta.services.user_manager import UserManager