From 6e7b2fbfcba23ad6a9d0f463a2c750b6c035611f Mon Sep 17 00:00:00 2001
From: zhongxiaotian <zhongxiaotian@huawei.com>
Date: Mon, 14 Jul 2025 21:39:40 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E5=B7=A5=E5=85=B7?=
 =?UTF-8?q?=E7=BB=84=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

新增工具组件

#ICLT7L
---
 jiuwen/core/common/exception/status_code.py |   5 +
 jiuwen/core/component/tool_comp.py          | 123 ++++++++++++++++++++
 2 files changed, 128 insertions(+)
 create mode 100644 jiuwen/core/component/tool_comp.py

diff --git a/jiuwen/core/common/exception/status_code.py b/jiuwen/core/common/exception/status_code.py
index fbef15d..6ca6caf 100644
--- a/jiuwen/core/common/exception/status_code.py
+++ b/jiuwen/core/common/exception/status_code.py
@@ -18,6 +18,11 @@ class StatusCode(Enum):
         102053,
         "Errors occur when formatting the template content due to wrong format")
 
+    # Tool组件 101741-101770
+    TOOL_COMPONENT_PARAM_CHECK_ERROR = (101742, 'Tool component parameter check error')
+    TOOL_COMPONENT_INPUTS_ERROR = (101743, 'Tool component inputs not defined')
+    TOOL_COMPONENT_EXECUTE_ERROR = (101745, "Tool component execute error")
+
     # Prompt 模板管理 102100 - 102149
     PROMPT_TEMPLATE_DUPLICATED_ERROR = (102101, "Template duplicated")
     PROMPT_TEMPLATE_NOT_FOUND_ERROR = (102102, "Template not found")
diff --git a/jiuwen/core/component/tool_comp.py b/jiuwen/core/component/tool_comp.py
new file mode 100644
index 0000000..3d33dbd
--- /dev/null
+++ b/jiuwen/core/component/tool_comp.py
@@ -0,0 +1,123 @@
+#!/usr/bin/python3.10
+# coding: utf-8
+# Copyright (c) Huawei Technologies Co., Ltd. 2025-2025. All rights reserved
+
+from dataclasses import dataclass, field
+from typing import Dict, Any, AsyncIterator, Iterator
+
+from jiuwen.core.common.exception.exception import JiuWenBaseException
+from jiuwen.core.common.exception.status_code import StatusCode
+from jiuwen.core.component.base import ComponentConfig, WorkflowComponent
+from jiuwen.core.context.context import Context
+from jiuwen.core.graph.executable import Executable, Input, Output
+
+
+@dataclass
+class ToolComponentConfig(ComponentConfig):
+    header: Dict[str, Any] = field(default_factory=dict)
+    needValidate: bool = True
+
+
+class ToolExecutable(Executable):
+
+    def __init__(self, config: ToolComponentConfig):
+        super().__init__()
+        self._config = config
+        self._tool = None
+
+    def invoke(self, inputs: Input, context: Context) -> Output:
+        self._tool = self.get_tool(context)
+        validated = inputs.get('validate', False)
+        user_field = inputs.get('userFields', None)
+        if self._config.needValidate and not validated:
+            self.validate_require_params(user_field)
+        formatted_inputs = prepare_inputs(user_field, self.get_tool_param())
+        try:
+            response = self._tool.run(formatted_inputs)
+            return self._create_output(response)
+        except Exception as e:
+            raise JiuWenBaseException(
+                error_code=StatusCode.TOOL_COMPONENT_EXECUTE_ERROR.code,
+                message='tool component execution error'
+            ) from e
+
+    async def ainvoke(self, inputs: Input, context: Context) -> Output:
+        pass
+
+    def stream(self, inputs: Input, context: Context) -> Iterator[Output]:
+        pass
+
+    async def astream(self, inputs: Input, context: Context) -> AsyncIterator[Output]:
+        pass
+
+    def interrupt(self, message: dict):
+        pass
+
+    def _create_output(self, response):
+        return response
+
+    def get_tool(self, context: Context):
+        pass
+
+    def get_tool_param(self):
+        pass
+
+    def validate_require_params(self, user_field):
+        require_params = self.get_tool_param()
+        params_dict = {param.name: param.descrpition for param in require_params}
+        missing_params = {param for param in params_dict if param not in user_field}
+        if missing_params:
+            missing_params_dict = {param: params_dict[param] for param in missing_params}
+            interrupt_message = {
+                'type': 'MessageSubTypes.PLUGIN_PARAM_MISS.value',
+                'tool_name': self._tool.name,
+                'missing_params': missing_params_dict,
+            }
+            self.interrupt(interrupt_message)
+
+TYPE_CASTER = {
+    "str": str,
+    "integer": int,
+    "number": float,
+    "bool": bool
+}
+
+def _transform_type(value, expected_type, key):
+    expected_type = expected_type.lower()
+    caster = TYPE_CASTER.get(expected_type)
+    if caster:
+        try:
+            return caster(value)
+        except(TypeError, ValueError) as e:
+            raise JiuWenBaseException(
+                error_code=StatusCode.TOOL_COMPONENT_PARAM_CHECK_ERROR.code,
+                message=f'{StatusCode.TOOL_COMPONENT_PARAM_CHECK_ERROR.errmsg}'
+                        f'param name is {key}, expected type: {expected_type}'
+            ) from e
+    return value
+
+
+def prepare_inputs(user_field, defined_param):
+    define_dict = {}
+    formatted_inputs = {}
+    for param in defined_param:
+        define_dict[param.name] = param
+    for k, v in user_field.items():
+        if define_dict.get(k):
+            param = define_dict.get(k)
+            expected_type = param.type
+            formatted_inputs[k] = _transform_type(v, expected_type, k)
+        else:
+            raise JiuWenBaseException(
+                error_code=StatusCode.TOOL_COMPONENT_INPUTS_ERROR.code,
+                message=f'{StatusCode.TOOL_COMPONENT_INPUTS_ERROR.errmsg}, param is {k}'
+            )
+
+class ToolComponent(WorkflowComponent):
+
+    def __init__(self, config: ToolComponentConfig):
+        super().__init__()
+        self._config = config
+
+    def to_executable(self) -> Executable:
+        return ToolExecutable(self._config)
\ No newline at end of file
-- 
Gitee