From d4355123b4615c6f4720ee6bdacbba4052994df6 Mon Sep 17 00:00:00 2001 From: yepeiwen01 Date: Wed, 22 Jan 2025 16:29:13 +0800 Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=E8=BE=93=E5=85=A5dict?= =?UTF-8?q?=E5=88=9B=E5=BB=BAjson=E8=BE=93=E5=87=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- python/core/component.py | 2 ++ python/tests/test_base_component.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/python/core/component.py b/python/core/component.py index 6ebedc73..0dd75f6d 100644 --- a/python/core/component.py +++ b/python/core/component.py @@ -559,6 +559,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 index a560cf78..26d9b407 100644 --- 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={"key": "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):