treemind组件开发

baidubce · Nov 7, 2024 · 718773c · 718773c
1 parent 27920ac
commit 718773c
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/appbuilder/tests/test_treemind.py b/appbuilder/tests/test_treemind.py
@@ -27,9 +27,6 @@ def setUp(self):
         Returns:
             None.
         """
-        # 从环境变量中提取CarExpert组件的TOKEN
-        os.environ["APPBUILDER_TOKEN"] = "Bearer bce-v3/ALTAK-6AGZK6hjSpZmEclEuAWje/6d2d2ffc438f9f2ba66e23b21de69d96e7e5713a"
-        os.environ["GATEWAY_URL"] = "http://10.138.138.29"
         self.tm = TreeMind()
         self.query = "生成一份年度总结的思维导图"
 
@@ -39,7 +36,7 @@ def test_treemind_component_tool_eval(self):
         result = self.tm.tool_eval(query=self.query)
         self.assertIsNotNone(result)
         for r in result:
-            # print(r)
+            print(r)
             self.assertIsNotNone(r)
             self.assertIn(r["type"],["text", "urls"])
 
@@ -48,7 +45,7 @@ def test_treemind_component_run(self):
         """
         msg = Message(content=self.query)
         result = self.tm.run(msg)
-        # print(result)
+        print(result)
         self.assertIsNotNone(result)
         self.assertIsInstance(result, Message)