From 44a56f8a305a676bdc3536fbd84a1d3b8208191a Mon Sep 17 00:00:00 2001
From: fuwx <f2928560492@163.com>
Date: Tue, 11 Mar 2025 11:14:54 +0800
Subject: [PATCH] update tool

---
 .../providers/apo_rule/tools/root_cause.py        | 15 +++++++++++----
 .../providers/apo_rule/tools/root_cause.yaml      | 14 ++++++++++++++
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/api/core/tools/builtin_tool/providers/apo_rule/tools/root_cause.py b/api/core/tools/builtin_tool/providers/apo_rule/tools/root_cause.py
index 9e643ad300..2decdc5d56 100644
--- a/api/core/tools/builtin_tool/providers/apo_rule/tools/root_cause.py
+++ b/api/core/tools/builtin_tool/providers/apo_rule/tools/root_cause.py
@@ -19,12 +19,13 @@ class RootCauseTool(BuiltinTool):
         invoke tools
         """
         llm_text = tool_parameters.get("text")
-        node_lists, text = self._get_node_list(llm_text)
+        language = tool_parameters.get("language")
+        node_lists, text = self._get_node_list(llm_text, language)
         
         yield self.create_text_message(text)
         yield self.create_json_message({'nodelists': node_lists})
     
-    def _get_node_list(self, data: str) -> tuple[list, str]: 
+    def _get_node_list(self, data: str, language: str) -> tuple[list, str]: 
         data = data.strip('```')
         data = data.strip('json')
         text = ""
@@ -37,7 +38,10 @@ class RootCauseTool(BuiltinTool):
                 endpoint = ep.strip('"')
                 endpoint = endpoint.strip("'")
                 nodeList.append({'service': service, 'endpoint': endpoint, "isRoot": True})
-                text = text + f'造成根因的节点是{nodeinfo['nodeName']}\n'
+                if language == "zh-cn":
+                    text = text + f'造成根因的节点是{nodeinfo['nodeName']}\n'
+                else:
+                    text = text + f'The root cause node is {nodeinfo['nodeName']}\n'
     
             if "otherNodeName" in data:
                 for node in nodeinfo["otherNodeName"]:
@@ -47,7 +51,10 @@ class RootCauseTool(BuiltinTool):
                     tmpendpoint = tmpendpoint.strip("'")
                     nodeList.append({'service': tmpservice, 'endpoint': tmpendpoint, "isRoot": False})
                 otherstr = ",".join(nodeinfo["otherNodeName"])
-                text = text + f'疑似根因节点有{otherstr}\n'
+                if language == "zh-cn":
+                    text = text + f'疑似根因节点有{otherstr}\n'
+                else:
+                    text = text + f'Possible root cause nodes are {otherstr}\n'
             return nodeList, text
         except Exception as e:
             return [], ""
\ No newline at end of file
diff --git a/api/core/tools/builtin_tool/providers/apo_rule/tools/root_cause.yaml b/api/core/tools/builtin_tool/providers/apo_rule/tools/root_cause.yaml
index ad2537a6f2..57d0deec55 100644
--- a/api/core/tools/builtin_tool/providers/apo_rule/tools/root_cause.yaml
+++ b/api/core/tools/builtin_tool/providers/apo_rule/tools/root_cause.yaml
@@ -12,6 +12,20 @@ description:
     pt_BR: get root cause
   llm: get root cause node
 parameters:
+  - name: language
+    type: string
+    required: false
+    label:
+      en_US: language
+      zh_Hans: language
+      pt_BR: language
+    human_description:
+      en_US: text language
+      zh_Hans: 文本语言
+      pt_BR: text language
+    llm_description: rule
+    default: "zh-cn"
+    form: llm
   - name: text
     type: string
     required: false