diff --git a/agent/component/concentrator.py b/agent/component/concentrator.py index 62fd0dcea96..ee405b3dcc6 100644 --- a/agent/component/concentrator.py +++ b/agent/component/concentrator.py @@ -26,7 +26,7 @@ def __init__(self): super().__init__() def check(self): - super().check() + return True class Concentrator(ComponentBase, ABC): diff --git a/agent/test/dsl_examples/categorize.json b/agent/test/dsl_examples/categorize.json index 549d21b4f18..600c9bc3fc4 100644 --- a/agent/test/dsl_examples/categorize.json +++ b/agent/test/dsl_examples/categorize.json @@ -26,20 +26,48 @@ "category_description": { "product_related": { "description": "The question is about the product usage, appearance and how it works.", - "examples": "Why it always beaming?\nHow to install it onto the wall?\nIt leaks, what to do?" + "examples": "Why it always beaming?\nHow to install it onto the wall?\nIt leaks, what to do?", + "to": "message:0" }, "others": { "description": "The question is not about the product usage, appearance and how it works.", - "examples": "How are you doing?\nWhat is your name?\nAre you a robot?\nWhat's the weather?\nWill it rain?" + "examples": "How are you doing?\nWhat is your name?\nAre you a robot?\nWhat's the weather?\nWill it rain?", + "to": "message:1" } } } }, - "downstream": [], + "downstream": ["message:0","message:1"], "upstream": ["answer:0"] + }, + "message:0": { + "obj": { + "component_name": "Message", + "params": { + "messages": [ + "Message 0!!!!!!!" + ] + } + }, + "downstream": ["answer:0"], + "upstream": ["categorize:0"] + }, + "message:1": { + "obj": { + "component_name": "Message", + "params": { + "messages": [ + "Message 1!!!!!!!" + ] + } + }, + "downstream": ["answer:0"], + "upstream": ["categorize:0"] } }, "history": [], + "messages": [], "path": [], + "reference": [], "answer": [] -} \ No newline at end of file +} diff --git a/agent/test/dsl_examples/concentrator_message.json b/agent/test/dsl_examples/concentrator_message.json new file mode 100644 index 00000000000..98d62b58094 --- /dev/null +++ b/agent/test/dsl_examples/concentrator_message.json @@ -0,0 +1,113 @@ +{ + "components": { + "begin": { + "obj":{ + "component_name": "Begin", + "params": { + "prologue": "Hi there!" + } + }, + "downstream": ["answer:0"], + "upstream": [] + }, + "answer:0": { + "obj": { + "component_name": "Answer", + "params": {} + }, + "downstream": ["categorize:0"], + "upstream": ["begin"] + }, + "categorize:0": { + "obj": { + "component_name": "Categorize", + "params": { + "llm_id": "deepseek-chat", + "category_description": { + "product_related": { + "description": "The question is about the product usage, appearance and how it works.", + "examples": "Why it always beaming?\nHow to install it onto the wall?\nIt leaks, what to do?", + "to": "concentrator:0" + }, + "others": { + "description": "The question is not about the product usage, appearance and how it works.", + "examples": "How are you doing?\nWhat is your name?\nAre you a robot?\nWhat's the weather?\nWill it rain?", + "to": "concentrator:1" + } + } + } + }, + "downstream": ["concentrator:0","concentrator:1"], + "upstream": ["answer:0"] + }, + "concentrator:0": { + "obj": { + "component_name": "Concentrator", + "params": {} + }, + "downstream": ["message:0"], + "upstream": ["categorize:0"] + }, + "concentrator:1": { + "obj": { + "component_name": "Concentrator", + "params": {} + }, + "downstream": ["message:1_0","message:1_1","message:1_2"], + "upstream": ["categorize:0"] + }, + "message:0": { + "obj": { + "component_name": "Message", + "params": { + "messages": [ + "Message 0_0!!!!!!!" + ] + } + }, + "downstream": ["answer:0"], + "upstream": ["concentrator:0"] + }, + "message:1_0": { + "obj": { + "component_name": "Message", + "params": { + "messages": [ + "Message 1_0!!!!!!!" + ] + } + }, + "downstream": ["answer:0"], + "upstream": ["concentrator:1"] + }, + "message:1_1": { + "obj": { + "component_name": "Message", + "params": { + "messages": [ + "Message 1_1!!!!!!!" + ] + } + }, + "downstream": ["answer:0"], + "upstream": ["concentrator:1"] + }, + "message:1_2": { + "obj": { + "component_name": "Message", + "params": { + "messages": [ + "Message 1_2!!!!!!!" + ] + } + }, + "downstream": ["answer:0"], + "upstream": ["concentrator:1"] + } + }, + "history": [], + "messages": [], + "path": [], + "reference": [], + "answer": [] +} \ No newline at end of file