test: modify test_rag_agent

2026-02-09 01:59:19 +08:00 · 2024-04-26 17:23:33 +08:00
parent 3f649e271b
commit bca4e2a1a3
1 changed files with 4 additions and 0 deletions
--- a/tests/test_agentuniverse/unit/agent/test_rag_agent.py
+++ b/tests/test_agentuniverse/unit/agent/test_rag_agent.py
@@ -21,3 +21,7 @@ class TestAgent(unittest.TestCase):
        instance: Agent = AgentManager().get_instance_obj('rag_agent')
        output_object: OutputObject = instance.run(input='分析下先锋领航退出中国的原因')
        print(output_object.get_data('output'))
+
+
+if __name__ == "__main__":
+    unittest.main()