xgae 0.3.9__tar.gz → 0.4.1__tar.gz

{xgae-0.3.9 → xgae-0.4.1}/.env

@@ -21,7 +21,7 @@ LLM_MAX_RETRIES=1
 LLM_STREAM=True
 LLM_ENABLE_THINKING=False
 
-LLM_LANGFUSE_ENABLE=False
+LLM_LANGFUSE_ENABLE=True
 
 
 # TASK_ENGINE

{xgae-0.3.9 → xgae-0.4.1}/CHANGELOG.md

@@ -1,3 +1,32 @@
+## [0.4.1] - 2025-11-15
+### Modified
+- XGATaskEngine : Fix Bug, allow 'no tool call' occur, after LLM summary, will call 'complete' 
+- general_prompt_template: Optimize prompt template
+- engine examples update with more clear example, show how to use General Agent and Custom Agent on different scenario
+
+
+## [0.4.0] - 2025-11-11
+- ✅ Overall Refact
+### Added
+- LLMClient : Add 'completion' sync LLM call function
+- XGASystemTools: Use local system tools, abandoned 'stdio' mode MCP system tool
+### Modified
+- XGAMcpToolBox:  Separate 'system_tool' from ’general_tool'
+- XGAPromptBuilder: Support two model system_prompt, distinguish between 'system' and 'general' mode
+- templates: Merge tool prompt template with system prompt template, Remove unclear template
+- XGA MCP Server Config: all merge to xga_mcp_servers.json
+- GAIA2 ARE Example: Refact with xgae engine new version
+- All Examples modified by xgae engine new version
+
+
+## [0.3.10] - 2025-11-6
+### Modified
+- XMLToolParser : Fix Bug, when string like '+2478' will convert to int 
+- GAIA2 ARE Example XGAAreToolBox: Fix Bug, ARE tool use full tool name 
+- GAIA2 ARE Example XGAAreAgent: Fix Bug, add ObservationLog log
+- GAIA2 ARE Example are_agent_factory : Fix Bug, add xga_termination_step, avoid exit on calling wait_for_notification tool
+
+
 ## [0.3.9] - 2025-11-4
 ### Added
 - ✅ GAIA2 ARE Example:  Add new prompt template, Leading over ARE 'default agent'

{xgae-0.3.9 → xgae-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xgae
-Version: 0.3.9
+Version: 0.4.1
 Summary: Extreme General Agent Engine
 Requires-Python: >=3.11
 Requires-Dist: colorlog>=6.9.0
@@ -14,7 +14,8 @@ Requires-Dist: langchain-community==0.3.29; extra == 'examples'
 Requires-Dist: langgraph==0.6.5; extra == 'examples'
 Description-Content-Type: text/markdown
 
-## XGAE: Extreme General Agent Engine 
+# XGAE: Extreme General Agent Engine 
+## XGAE Features
 ### Functional Features
 - Support Custom prompt and external MCP Tools
 - Support Langfuse
@@ -29,8 +30,21 @@ Description-Content-Type: text/markdown
 - Architecture is lighter than SUNA Engine
 - Separate tools from Agent Engine
 
-### Examples
+
+## XGAE Run Mode
+### General Agent
+- Use General Agent Prompt
+- Use general tools mainly
+- In addition to using general tools, custom tools can also be used
+
+### Custom Agent
+- Use User Custom Prompt
+- Use custom tools mainly
+- Some general tools can also be used, like 'web_search'
+
+
+## Examples
+- engine: Use XGA Engine in various scenarios, you can find out how to use General or Custom Agent
 - langgraph:  Build React mode Langgraph Agent by XGA Engine
 - are: Build GAIA2 ARE Agent by XGA Engine
-- engine: Use XGA Engine in various scenarios
 - tools: Simulation tools for example and test

{xgae-0.3.9 → xgae-0.4.1}/README.md

@@ -1,4 +1,5 @@
-## XGAE: Extreme General Agent Engine 
+# XGAE: Extreme General Agent Engine 
+## XGAE Features
 ### Functional Features
 - Support Custom prompt and external MCP Tools
 - Support Langfuse
@@ -13,8 +14,21 @@
 - Architecture is lighter than SUNA Engine
 - Separate tools from Agent Engine
 
-### Examples
+
+## XGAE Run Mode
+### General Agent
+- Use General Agent Prompt
+- Use general tools mainly
+- In addition to using general tools, custom tools can also be used
+
+### Custom Agent
+- Use User Custom Prompt
+- Use custom tools mainly
+- Some general tools can also be used, like 'web_search'
+
+
+## Examples
+- engine: Use XGA Engine in various scenarios, you can find out how to use General or Custom Agent
 - langgraph:  Build React mode Langgraph Agent by XGA Engine
 - are: Build GAIA2 ARE Agent by XGA Engine
-- engine: Use XGA Engine in various scenarios
 - tools: Simulation tools for example and test

{xgae-0.3.9 → xgae-0.4.1}/examples/agent/are/README.md

@@ -35,8 +35,8 @@
 
 
 - Modify ARE .env, add XGAE .env config, refer to env.example 
-- Copy XGAE package 'mcpservers' and 'templates' directory to ARE project root  
-- You can use prompt template by two modes, 'prior' mode is leading over ARE 'default agent'
+- Copy XGAE package 'templates' directory to ARE project root  
+
 
 ### Run XGA Agent in ARE
   ```

{xgae-0.3.9 → xgae-0.4.1}/examples/agent/are/are/simulation/agents/xga/are_agent.py

@@ -1,6 +1,8 @@
 import re
 import asyncio
 from typing import List, Dict, Any
+
+from mammoth.results import success
 from typing_extensions import override
 
 from are.simulation.agents.are_simulation_agent import AgentStoppedException
@@ -14,7 +16,9 @@ from are.simulation.agents.agent_log import (
     SystemPromptLog,
     ThoughtLog,
     ToolCallLog,
-    FinalAnswerLog
+    FinalAnswerLog,
+    ObservationLog,
+    ErrorLog
 )
 from are.simulation.agents.default_agent.base_agent import (
     BaseAgent,
@@ -64,6 +68,10 @@ def pre_run_task_check(agent, iterations: int, llm_messages: List[Dict[str, Any]
             raise AgentStoppedException("Agent stopped.")
 
         # Begin step()
+        agent.build_history_from_logs(
+            exclude_log_types=["tool_call", "rationale", "action"]
+        )
+
         agent.append_agent_log(
             LLMInputLog(
                 content     = llm_messages,
@@ -139,7 +147,7 @@ def terminate_task_check(agent, iterations: int) -> bool:
     is_terminate_task = False
 
     try:
-        if agent.termination_step.condition:
+        if agent.termination_step and agent.termination_step.condition:
             is_terminate_task = agent.termination_step.condition(agent)
             if is_terminate_task:
                 agent.logger.info(f"terminate_task_check[{iterations}]: termination_step.condition is True")
@@ -164,7 +172,7 @@ class XGAAreAgent(BaseAgent):
         tool_box = XGAAreToolBox(self.tools)
 
         general_system_prompt = "\n\n".join(prompt for prompt in self.init_system_prompts.values())
-        prompt_builder = XGAArePromptBuilder(general_system_prompt, "prior")
+        prompt_builder = XGAArePromptBuilder(general_system_prompt)
 
         model_config = self.llm_engine.model_config
         llm_config = LLMConfig(
@@ -247,6 +255,16 @@ class XGAAreAgent(BaseAgent):
             elif chunk_type == "tool":
                 tool_content = chunk['content']
                 tool_execution = tool_content.get('tool_execution')
+                tool_result = tool_execution.get('result')
+                self.append_agent_log(
+                    ObservationLog(
+                        content     = str(tool_result),
+                        attachments = [],
+                        timestamp   = self.make_timestamp(),
+                        agent_id    = self.agent_id,
+                    )
+                )
+
                 self.append_agent_log(
                     ToolCallLog(
                         tool_name       = tool_execution.get('function_name'),

{xgae-0.3.9 → xgae-0.4.1}/examples/agent/are/are/simulation/agents/xga/are_agent_factory.py

@@ -1,11 +1,27 @@
+from functools import partial
+
+from are.simulation.agents.llm.llm_engine import LLMEngine
 from are.simulation.agents.are_simulation_agent_config import ARESimulationReactBaseAgentConfig
 from are.simulation.agents.default_agent.steps.are_simulation import get_are_simulation_update_pre_step
 from are.simulation.agents.default_agent.termination_methods.are_simulation import get_gaia2_termination_step
 from are.simulation.agents.default_agent.tools.json_action_executor import JsonActionExecutor
-from are.simulation.agents.llm.llm_engine import LLMEngine
+from are.simulation.agents.default_agent.base_agent import TerminationStep
+from are.simulation.agents.default_agent.termination_methods.are_simulation import  termination_condition_are_simulation
 
 from are.simulation.agents.xga.are_agent import XGAAreAgent
 
+termination_condition_are_simulation_send_message_to_user = partial(
+    termination_condition_are_simulation,
+    termination_tool_names=[
+        "AgentUserInterface__send_message_to_user"
+    ],
+)
+
+def get_xga_termination_step() -> TerminationStep:
+    return TerminationStep(
+        name="end_scenario",
+        condition=termination_condition_are_simulation_send_message_to_user,
+    )
 
 def xga_simulation_react_xml_agent(
     llm_engine: LLMEngine, base_agent_config: ARESimulationReactBaseAgentConfig
@@ -16,7 +32,7 @@ def xga_simulation_react_xml_agent(
         system_prompts={
             "system_prompt": str(base_agent_config.system_prompt),
         },
-        termination_step=get_gaia2_termination_step(),
+        termination_step=get_xga_termination_step(),
         max_iterations=base_agent_config.max_iterations,
         action_executor=JsonActionExecutor(  # Just for compatible BaseAgent， useless
             use_custom_logger=base_agent_config.use_custom_logger

xgae-0.4.1/examples/agent/are/are/simulation/agents/xga/are_prompt_builder.py

@@ -0,0 +1,27 @@
+import re
+from typing import Optional, List, Literal
+
+from xgae.engine.prompt_builder import XGAPromptBuilder
+from xgae.utils.misc import read_file
+
+
+class XGAArePromptBuilder(XGAPromptBuilder):
+    def __init__(self, general_system_prompt: Optional[str]):
+        are_system_prompt = self.build_are_system_prompt(general_system_prompt)
+        super().__init__(are_system_prompt)
+
+
+    def build_are_system_prompt(self, _system_prompt: str)-> str:
+        pattern = r'<general_instructions>(.*?)</general_instructions>'
+        prompt_are_general = re.search(pattern, _system_prompt, re.DOTALL)
+        prompt_header = "# CORE IDENTITY\n"
+        if prompt_are_general:
+            prompt_are_general = prompt_header + prompt_are_general.group(1).strip() + "\n\n"
+        else:
+            prompt_are_general = prompt_header + _system_prompt  + "\n\n"
+
+        are_tool_prompt = read_file("templates/are_prompt_template.md")
+        are_system_prompt = prompt_are_general + are_tool_prompt
+
+        return are_system_prompt
+

{xgae-0.3.9 → xgae-0.4.1}/examples/agent/are/are/simulation/agents/xga/are_tool_box.py

@@ -21,58 +21,34 @@ class XGAAreToolBox(XGAMcpToolBox):
 
     @override
     async def init_tool_schemas(self):
-        await self._load_mcp_tools_schema()
+        #await self._load_mcp_tools_schema()
         self._load_are_tools_schema()
 
 
     @override
     async def call_tool(self, task_id: str, tool_name: str, args: Optional[Dict[str, Any]] = None) -> XGAToolResult:
-        if tool_name == "end_task":
-            server_name = self.GENERAL_MCP_SERVER_NAME
-        else:
-            task_tool_schemas = self.task_tool_schemas.get(task_id, {})
-            tool_schema = task_tool_schemas.get(tool_name, None)
-            if tool_schema is None:
-                raise XGAError(f"MCP tool not found: '{tool_name}'")
-            server_name = tool_schema.server_name
+        task_tool_schemas = self.task_tool_schemas.get(task_id, {})
+        tool_schema = task_tool_schemas.get(tool_name, None)
+        if tool_schema is None:
+            raise XGAError(f"ARE tool not found: '{tool_name}'")
+        server_name = tool_schema.server_name
 
         tool_type = self._get_tool_type(server_name)
-        if tool_type == "custom":  # ARE Tools
-            full_tool_name = server_name + "__" + tool_name
 
+        if tool_type == "system":
+            tool_args = args or {}
+            tool_args = dict({'task_id': task_id}, **tool_args)
+            result = self.system_tools.call_tool(tool_name, tool_args)
+        elif tool_type == "custom":  # ARE Tools
             try:
-                tool_result = self.are_tools[full_tool_name](**args)
+                tool_result = self.are_tools[tool_name](**args)
                 result = convert_mcp_tool_result(str(tool_result))
             except Exception as e:
                 error = f"Call ARE Tool '{tool_name}' error: {str(e)}"
                 logging.error(f"AreToolBox call_are_tool: {error}")
                 result = XGAToolResult(success=False, output=str(e))
         else:
-            async with self._mcp_client.session(server_name) as session:
-                tools = await load_mcp_tools(session)
-                mcp_tool = next((t for t in tools if t.name == tool_name), None)
-
-                if mcp_tool:
-                    tool_args = args or {}
-
-                    if tool_type == "general" or tool_type == "agent":
-                        tool_args = dict({'task_id': task_id}, **tool_args)
-
-                    try:
-                        tool_result = await mcp_tool.arun(tool_args)
-                        if tool_type == "general":
-                            tool_result = json.loads(tool_result)
-                            result = XGAToolResult(success=tool_result['success'], output=str(tool_result['output']))
-                        else:
-                            result = XGAToolResult(success=True, output=str(tool_result))
-                    except Exception as e:
-                        error = f"Call MCP Tool '{tool_name}' error: {str(e)}"
-                        logging.error(f"AreToolBox call_mcp_tool: {error}")
-                        result = XGAToolResult(success=False, output=error)
-                else:
-                    error = f"No MCP tool found with name: {tool_name}"
-                    logging.info(f"AreToolBox call_mcp_tool: error={error}")
-                    result =  XGAToolResult(success=False, output=error)
+            raise XGAError(f"ARE tool no support type: '{tool_type}'")
 
         return result
 
@@ -86,8 +62,8 @@ class XGAAreToolBox(XGAMcpToolBox):
     def _load_are_tools_schema(self) -> None:
         if not self._is_loaded_are_tool_schemas:
             for are_tool in self.are_tools.values():
-                full_tool_name = are_tool.name
-                server_name , tool_name = full_tool_name.split("__")
+                tool_name = are_tool.name
+                server_name , unction_name = tool_name.split("__")
                 tool_type :XGAToolType = "custom"
                 input_schema = {
                     'properties': are_tool.inputs,
@@ -95,6 +71,7 @@ class XGAAreToolBox(XGAMcpToolBox):
                 }
 
                 tool_schema = XGAToolSchema(tool_name       = tool_name,
+                                            function_name   = tool_name,
                                             tool_type       = tool_type,
                                             server_name     = server_name,
                                             description     = are_tool.description,

xgae-0.4.1/examples/agent/are/templates/are_prompt_template.md

@@ -0,0 +1,57 @@
+# MetaOSSAgent System Prompt
+
+## Core Identity
+
+**Name**: MetaOSSAgent  
+**Role**: Expert Assistant for Meta Agents Research Environments  
+**Mission**: Deliver accurate, reliable, and effective task completion with minimal overhead
+
+---
+
+## Execution Philosophy
+
+### 🎯 Fundamental Principles
+
+1. **Silent Execution Mode**
+   - Work quietly and autonomously throughout task execution
+   - **Only communicate with users when**:
+     - All tasks are completely finished
+     - Tasks are genuinely impossible to complete
+     - Critical ambiguity blocks progress and requires clarification
+   - No status updates, progress reports, or interim communications
+
+2. **Complete & Thorough Delivery**
+   - Finish all tasks comprehensively before reporting
+   - Ensure all deliverables meet requirements
+   - Verify nothing is left incomplete
+
+3. **Precise Instruction Following**
+   - Adhere strictly to user specifications
+   - Respect stated constraints and preferences
+   - Maintain fidelity to original requirements
+
+4. **Proactive Problem-Solving**
+   - Exhaust multiple solution approaches before declaring failure
+   - Leverage available tools creatively
+   - Research and gather information independently
+
+5. **Self-Sufficient Information Gathering**
+   - Use available tools to find answers first
+   - Minimize unnecessary user interruptions
+   - Ask only when information is truly unavailable through other means
+
+### 🔄 Ambiguity Resolution Strategy
+
+**When encountering unclear requirements**:
+
+1. **Execute Immediately**: Complete all unambiguous portions of the task
+2. **Stop Promptly**: Halt when reaching genuine ambiguity, contradictions, or impossibilities
+3. **Clarify Explicitly**: After completing clear subtasks, use the `ask` tool to request specific clarification
+
+**Never guess or assume** when dealing with:
+- Multiple entities with identical names
+- Conflicting requirements
+- Missing critical information
+- Interpretable specifications with significant impact differences
+
+---

xgae-0.3.9/examples/agent/are/templates/are_prompt_template.md → xgae-0.4.1/examples/agent/are/templates/system_prompt_template.md

@@ -1,61 +1,3 @@
-# MetaOSSAgent System Prompt
-
-## Core Identity
-
-**Name**: MetaOSSAgent  
-**Role**: Expert Assistant for Meta Agents Research Environments  
-**Mission**: Deliver accurate, reliable, and effective task completion with minimal overhead
-
----
-
-## Execution Philosophy
-
-### 🎯 Fundamental Principles
-
-1. **Silent Execution Mode**
-   - Work quietly and autonomously throughout task execution
-   - **Only communicate with users when**:
-     - All tasks are completely finished
-     - Tasks are genuinely impossible to complete
-     - Critical ambiguity blocks progress and requires clarification
-   - No status updates, progress reports, or interim communications
-
-2. **Complete & Thorough Delivery**
-   - Finish all tasks comprehensively before reporting
-   - Ensure all deliverables meet requirements
-   - Verify nothing is left incomplete
-
-3. **Precise Instruction Following**
-   - Adhere strictly to user specifications
-   - Respect stated constraints and preferences
-   - Maintain fidelity to original requirements
-
-4. **Proactive Problem-Solving**
-   - Exhaust multiple solution approaches before declaring failure
-   - Leverage available tools creatively
-   - Research and gather information independently
-
-5. **Self-Sufficient Information Gathering**
-   - Use available tools to find answers first
-   - Minimize unnecessary user interruptions
-   - Ask only when information is truly unavailable through other means
-
-### 🔄 Ambiguity Resolution Strategy
-
-**When encountering unclear requirements**:
-
-1. **Execute Immediately**: Complete all unambiguous portions of the task
-2. **Stop Promptly**: Halt when reaching genuine ambiguity, contradictions, or impossibilities
-3. **Clarify Explicitly**: After completing clear subtasks, use the `ask` tool to request specific clarification
-
-**Never guess or assume** when dealing with:
-- Multiple entities with identical names
-- Conflicting requirements
-- Missing critical information
-- Interpretable specifications with significant impact differences
-
----
-
 ## Tool Execution Framework
 
 ### Standard Execution Cycle
@@ -121,13 +63,13 @@ THOUGHT → ACTION → OBSERVATION → [Repeat until objective achieved]
 
 ### Parameter Formatting Rules
 
-| Data Type | Format                               | Example |
-|-----------|--------------------------------------|---------|
-| String | Direct text without wrapping quotes  | `<parameter name="title">Buy groceries</parameter>` |
-| Number | Direct numeric value                 | `<parameter name="count">42</parameter>` |
-| Boolean | Lowercase true/false                 | `<parameter name="active">true</parameter>` |
-| Object | JSON string                          | `<parameter name="config">{"key": "value"}</parameter>` |
-| Array | JSON string                          | `<parameter name="items">["item1", "item2"]</parameter>` |
+| Data Type | Format                            | Example |
+|-----------|-----------------------------------|---------|
+| String | Direct text with wrapping quotes  | `<parameter name="title">"Buy groceries"</parameter>` |
+| Number | Direct numeric value              | `<parameter name="count">42</parameter>` |
+| Boolean | Lowercase true/false              | `<parameter name="active">true</parameter>` |
+| Object | JSON string                       | `<parameter name="config">{"key": "value"}</parameter>` |
+| Array | JSON string                       | `<parameter name="items">["item1", "item2"]</parameter>` |
 
 ### Complete Execution Example
 
@@ -138,8 +80,8 @@ The user wants to know the current weather in Paris. I'll use the weather query
 
 <function_calls>
 <invoke name="get_weather">
-<parameter name="city_name">Paris</parameter>
-<parameter name="units">celsius</parameter>
+<parameter name="city_name">"Paris"</parameter>
+<parameter name="units">"celsius"</parameter>
 </invoke>
 </function_calls>
 ```
@@ -236,8 +178,15 @@ The weather data has been successfully retrieved. The temperature is 20°C with
 
 ---
 
-### 📱 Application Tools
-{tool_schemas}
+ ### General Tools
+<<general_tool_schemas>>
+
+
+ ### Application Tools
+<<custom_tool_schemas>>
+
+ ### A2A Tools
+<<agent_tool_schemas>>
 
 ## Critical Operating Rules
 

{xgae-0.3.9 → xgae-0.4.1}/examples/agent/langgraph/reflection/custom_prompt_rag.py

@@ -47,7 +47,7 @@ class CustomPromptRag:
         prompt_docs = []
         prompt_docs.append(self._create_prompt_doc(
             prompt_summary="Fault location and analysis of fault causes",
-            prompt_path="templates/example/fault_user_prompt.txt"
+            prompt_path="templates/example/fault_user_prompt.md"
         ))
         return prompt_docs
 

{xgae-0.3.9 → xgae-0.4.1}/examples/agent/langgraph/reflection/reflection_agent.py

@@ -35,7 +35,7 @@ class ReflectiontAgent:
         self.graph_langfuse = None
         self.task_engine: XGATaskEngine = None
 
-        self.tool_box = XGAMcpToolBox(custom_mcp_server_file="mcpservers/custom_servers.json")
+        self.tool_box = XGAMcpToolBox(custom_mcp_server_file="xga_mcp_servers.json")
         self.result_eval_agent = TaskResultEvalAgent()
 
         if use_prompt_rag:
@@ -96,7 +96,7 @@ class ReflectiontAgent:
                 system_prompt = read_file(system_prompt_path)
         else:
             if "fault" in user_input:  # only for example
-                system_prompt = read_file("templates/example/fault_user_prompt.txt")
+                system_prompt = read_file("templates/example/fault_user_prompt.md")
         return system_prompt
 
 

{xgae-0.3.9 → xgae-0.4.1}/examples/agent/langgraph/reflection/result_eval_agent.py

@@ -29,7 +29,7 @@ class TaskResultEvalAgent:
 
         logging.info("TaskResultEvalAgent: LLM Eval result.....")
         response = await self.model_client.acompletion(messages, langfuse_metadata)
-        response_text = await self.model_client.get_response_result(response)
+        response_text = await self.model_client.get_acompletion_response(response)
 
         cleaned_text = re.sub(r'^\s*```json|```\s*$', '', response_text, flags=re.MULTILINE).strip()
         eval_result = json.loads(cleaned_text)
@@ -93,7 +93,7 @@ if __name__ == "__main__":
     async def main():
         final_result_agent = TaskResultEvalAgent()
 
-        task_plan = read_file("templates/example/fault_user_prompt.txt")
+        task_plan = read_file("templates/example/fault_user_prompt.md")
         user_input = "locate 10.2.3.4 fault and solution"
 
         answer = ("Task Summary: The fault for IP 10.2.3.4 was identified as a Business Recharge Fault (Code: F01), "

{xgae-0.3.9 → xgae-0.4.1}/examples/agent/langgraph/reflection/run_agent_app.py

@@ -12,8 +12,8 @@ async def main():
     task_no = 0
     user_inputs = [
         #"5+5", # For no tool call
-        "locate 10.2.3.4 fault and solution",  # For custom tool
-        #"locate fault and solution",   # For human append input
+        #"locate 10.2.3.4 fault and solution",  # For custom tool
+        "locate fault and solution",   # For human append input
     ]
 
     for user_input in user_inputs:

xgae-0.4.1/examples/engine/run_custom_and_agent_tools.py

@@ -0,0 +1,60 @@
+import asyncio
+
+from xgae.engine.mcp_tool_box import XGAMcpToolBox
+from xgae.engine.task_engine import XGATaskEngine
+from xgae.utils.misc import read_file
+
+from xgae.utils.setup_env import setup_logging
+
+
+is_stream = False
+if is_stream:
+    setup_logging(log_level="ERROR") # only show chunk
+else:
+    setup_logging()
+
+system_prompt = read_file("templates/example/fault_user_prompt.md")
+tool_box = XGAMcpToolBox(custom_mcp_server_file="xga_mcp_servers.json")
+
+# Before Run Exec: uv run example-fault-tools --alarmtype 2  , uv run example-a2a-tools
+# If want to use real A2A agent tool, use 'xgaproxy' project run: v run xga-a2a-proxy
+async def run_custom_plus_a2a_tools() -> None:
+    engine = XGATaskEngine(tool_box=tool_box,
+                           custom_tools=["*"],
+                           system_prompt=system_prompt
+                           )
+
+
+    user_input =  "locate 10.2.3.4 fault and solution"
+    global is_stream
+    if is_stream:
+        chunks = []
+        async for chunk in engine.run_task(task_input={"role": "user", "content": user_input}):
+            chunks.append(chunk)
+            print(chunk)
+
+        final_result = engine.parse_final_result(chunks)
+        print(f"\n\nFINAL_RESULT: {final_result}")
+    else:
+        final_result = await engine.run_task_with_final_answer(task_input={"role": "user", "content": user_input})
+        print(f"\n\nFINAL_RESULT: {final_result}")
+
+
+# Before run example execute with 'xgatools' project: uv run xgatools
+# Before run exec: uv run example-fault-tools --alarmtype 3  , uv run example-a2a-tools
+async def run_custom_plus_general_tools() -> None:
+    engine = XGATaskEngine(system_prompt=system_prompt,
+                           tool_box=tool_box,
+                           general_tools=["web_search"],
+                           custom_tools=["*"])
+
+    user_input = "locate 10.3.4.5 fault and solution"
+    final_result = await engine.run_task_with_final_answer(task_input={"role": "user", "content": user_input})
+    print("FINAL RESULT:", final_result)
+
+
+async def main() -> None:
+    #await run_custom_plus_a2a_tools()
+    await run_custom_plus_general_tools()
+
+asyncio.run(main())