Fix bug

PaddlePaddle · Dec 11, 2023 · 6f8adea · 6f8adea
1 parent aa1ba26
commit 6f8adea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/erniebot-agent/tests/unit_tests/agents/test_agent_response_annotations.py b/erniebot-agent/tests/unit_tests/agents/test_agent_response_annotations.py
@@ -3,13 +3,13 @@
 from typing import List, Literal
 
 from erniebot_agent.agents.schema import AgentFile, AgentResponse
-from erniebot_agent.file_io import get_file_manager
+from erniebot_agent.file_io import get_global_file_manager
 
 
 class TestAgentResponseAnnotations(unittest.TestCase):
     def setUp(self):
         self.test = ""
-        self.file_manager = get_file_manager()
+        self.file_manager = get_global_file_manager()
         self.file1 = asyncio.run(self.file_manager.create_file_from_bytes(b"test1", "test1.txt"))
         self.file2 = asyncio.run(self.file_manager.create_file_from_bytes(b"test2", "test2.txt"))