支持输入dict创建json输出 (#725)

* 将get_filename_from_url实现迁移至utils * 回滚版本号 * 支持输入dict创建json输出 * 支持输入dict创建json输出 * 跳过vdb单测 --------- Co-authored-by: yepeiwen01 <[email protected]>
baidubce · Jan 22, 2025 · ebfcbaa · ebfcbaa
1 parent b38dd18
commit ebfcbaa
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/python/core/component.py b/python/core/component.py
@@ -560,6 +560,8 @@ def create_output(cls, type, text, role="tool", name="", visible_scope="all", ra
                 key_list = ["detail", "steps"]
             elif type == "function_call":
                 key_list = ["thought", "name", "arguments"]
+            elif type == "json":
+                key_list = ["data"]
             else:
                 raise ValueError("Unknown type: {}".format(type))
             assert all(key in text for key in key_list), "all keys:{} must be included in the text field".format(key_list)

diff --git a/python/tests/test_base_component.py b/python/tests/test_base_component.py
@@ -33,6 +33,7 @@ def test_valid_output_with_dict(self):
         output9 = self.component.create_output(type="plan", text={"detail": "hello", "steps":[{"name": "1", "arguments": {"query": "a", "chat_history": "world"}}]})
         output10 = self.component.create_output(type="function_call", text={"thought": "hello", "name": "AppBuilder", "arguments": {"query": "a", "chat_history": "world"}})
         output11 = self.component.create_output(type="references", text={"type": "engine", "doc_id": "1", "content": "hello, world", "title": "Have a nice day", "source": "bing", "extra": {"key": "value"}})
+        output12 = self.component.create_output(type="json", text={"data": "value"})
         self.assertIsInstance(output1, ComponentOutput)
         self.assertIsInstance(output2, ComponentOutput)
         self.assertIsInstance(output3, ComponentOutput)
@@ -44,6 +45,7 @@ def test_valid_output_with_dict(self):
         self.assertIsInstance(output9, ComponentOutput)
         self.assertIsInstance(output10, ComponentOutput)
         self.assertIsInstance(output11, ComponentOutput)
+        self.assertIsInstance(output12, ComponentOutput)
         self.assertEqual(output11.content[0].text.extra["key"], "value")
 
     def test_valid_output_type_with_same_key(self):

diff --git a/python/tests/test_core_components_baidu_vdb_retriever.py b/python/tests/test_core_components_baidu_vdb_retriever.py
@@ -21,7 +21,7 @@
 from appbuilder import BaiduVDBVectorStoreIndex,TableParams
 from appbuilder.core.component import Message 
 
-@unittest.skipUnless(os.getenv("TEST_CASE", "UNKNOWN") == "CPU_PARALLEL", "")
+@unittest.skip(reason="vdb欠费了,跳过")
 class TestBaiduVbdRetriever(unittest.TestCase):
     def setUp(self):
         self.instance_id=os.getenv("INSTANCE_ID", "UNKNOWN")