xgae 0.1.8__tar.gz → 0.1.10__tar.gz

{xgae-0.1.8 → xgae-0.1.10}/.env

@@ -15,6 +15,7 @@ LLM_API_KEY=
 LLM_MAX_TOKENS=16384
 LLM_TEMPERATURE=0.7
 LLM_MAX_RETRIES=2
+LLM_LANGFUSE_ENABLE=False
 
 # TASK
 MAX_AUTO_RUN = 15

xgae-0.1.10/.idea/ai_toolkit.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="AI Toolkit Settings">
+    <option name="importsOfInterestPresent" value="true" />
+  </component>
+</project>

{xgae-0.1.8 → xgae-0.1.10}/.idea/workspace.xml

@@ -28,23 +28,27 @@
   <component name="PropertiesComponent"><![CDATA[{
   "keyToString": {
     "ModuleVcsDetector.initialDetectionPerformed": "true",
+    "Python.__init__.executor": "Run",
     "Python.llm_client.executor": "Run",
     "Python.mcp_tool_box.executor": "Run",
     "Python.message_tools_app.executor": "Run",
     "Python.responser_base.executor": "Run",
     "Python.run_engine_with_human_in_loop.executor": "Run",
+    "Python.run_human_in_loop.executor": "Run",
     "Python.run_simple.executor": "Run",
     "Python.run_task_engine.executor": "Run",
     "Python.run_user_prompt.executor": "Run",
     "Python.run_xga_engine.executor": "Run",
     "Python.setup_env.executor": "Run",
     "Python.task_engine.executor": "Run",
+    "Python.task_langfuse.executor": "Debug",
     "Python.test_langfuse.executor": "Run",
     "Python.test_litellm_langfuse.executor": "Run",
     "Python.utils.executor": "Run",
     "Python.xga_engine.executor": "Run",
     "Python.xga_mcp_tool_box.executor": "Debug",
     "Python.xga_prompt_builder.executor": "Debug",
+    "Python.xgae_cli.executor": "Run",
     "RunOnceActivity.ShowReadmeOnStart": "true",
     "last_opened_file_path": "/Users/sharkystar/DevProjects/xga/xgae",
     "node.js.detected.package.eslint": "true",
@@ -58,6 +62,8 @@
 }]]></component>
   <component name="RecentsManager">
     <key name="MoveFile.RECENT_KEYS">
+      <recent name="$PROJECT_DIR$/src/examples" />
+      <recent name="$PROJECT_DIR$/src/examples/engine" />
       <recent name="$PROJECT_DIR$/src/xgae/engine/responser" />
     </key>
   </component>
@@ -164,7 +170,9 @@
       <workItem from="1755737435202" duration="48139000" />
       <workItem from="1756044658912" duration="1248000" />
       <workItem from="1756082326044" duration="23657000" />
-      <workItem from="1756168626188" duration="26778000" />
+      <workItem from="1756168626188" duration="52435000" />
+      <workItem from="1756305726553" duration="2819000" />
+      <workItem from="1756340066915" duration="14276000" />
     </task>
     <servers />
   </component>
@@ -184,21 +192,25 @@
   </component>
   <component name="com.intellij.coverage.CoverageDataManagerImpl">
     <SUITE FILE_PATH="coverage/xgae$test_litellm_langfuse.coverage" NAME="test_litellm_langfuse Coverage Results" MODIFIED="1756196476262" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$task_langfuse.coverage" NAME="task_langfuse Coverage Results" MODIFIED="1756306181167" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/xgae$xga_engine.coverage" NAME="xga_engine Coverage Results" MODIFIED="1755580277172" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
-    <SUITE FILE_PATH="coverage/xgae$run_simple.coverage" NAME="run_simple Coverage Results" MODIFIED="1756111714718" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$run_simple.coverage" NAME="run_simple Coverage Results" MODIFIED="1756359842556" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$run_human_in_loop.coverage" NAME="run_human_in_loop Coverage Results" MODIFIED="1756279131815" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/xgae$run_xga_engine.coverage" NAME="run_task_engine Coverage Results" MODIFIED="1756111613459" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/xgae$message_tools_app.coverage" NAME="message_tools_app Coverage Results" MODIFIED="1756094157566" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/xgae$run_engine_with_human_in_loop.coverage" NAME="run_engine_with_human_in_loop Coverage Results" MODIFIED="1756089269027" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$__init__.coverage" NAME="__init__ Coverage Results" MODIFIED="1756345820610" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/xgae$xga_prompt_builder.coverage" NAME="xga_prompt_builder Coverage Results" MODIFIED="1755587456555" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/xgae$test_langfuse.coverage" NAME="test_langfuse Coverage Results" MODIFIED="1756196410142" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
-    <SUITE FILE_PATH="coverage/xgae$run_task_engine.coverage" NAME="run_task_engine Coverage Results" MODIFIED="1756207048901" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$run_task_engine.coverage" NAME="run_task_engine Coverage Results" MODIFIED="1756341371676" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/xgae$responser_base.coverage" NAME="responser_base Coverage Results" MODIFIED="1756103040764" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
-    <SUITE FILE_PATH="coverage/xgae$mcp_tool_box.coverage" NAME="mcp_tool_box Coverage Results" MODIFIED="1756188784603" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$mcp_tool_box.coverage" NAME="mcp_tool_box Coverage Results" MODIFIED="1756274403389" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/xgae$utils.coverage" NAME="utils Coverage Results" MODIFIED="1755226923439" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
-    <SUITE FILE_PATH="coverage/xgae$setup_env.coverage" NAME="setup_env Coverage Results" MODIFIED="1756200209483" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
-    <SUITE FILE_PATH="coverage/xgae$run_user_prompt.coverage" NAME="run_user_prompt Coverage Results" MODIFIED="1756089624828" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$setup_env.coverage" NAME="setup_env Coverage Results" MODIFIED="1756273791782" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$run_user_prompt.coverage" NAME="run_user_prompt Coverage Results" MODIFIED="1756279512361" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$xgae_cli.coverage" NAME="xgae_cli Coverage Results" MODIFIED="1756347000850" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
     <SUITE FILE_PATH="coverage/xgae$xga_mcp_tool_box.coverage" NAME="xga_mcp_tool_box Coverage Results" MODIFIED="1755583099719" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
-    <SUITE FILE_PATH="coverage/xgae$llm_client.coverage" NAME="llm_client Coverage Results" MODIFIED="1756188100577" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
-    <SUITE FILE_PATH="coverage/xgae$task_engine.coverage" NAME="task_engine Coverage Results" MODIFIED="1756199881134" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$task_engine.coverage" NAME="task_engine Coverage Results" MODIFIED="1756308226461" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
+    <SUITE FILE_PATH="coverage/xgae$llm_client.coverage" NAME="llm_client Coverage Results" MODIFIED="1756308364265" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$" />
   </component>
 </project>

{xgae-0.1.8 → xgae-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xgae
-Version: 0.1.8
+Version: 0.1.10
 Summary: Extreme General Agent Engine
 Requires-Python: >=3.13
 Requires-Dist: colorlog==6.9.0

{xgae-0.1.8 → xgae-0.1.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "xgae"
-version = "0.1.8"
+version = "0.1.10"
 description = "Extreme General Agent Engine"
 readme = "README.md"
 requires-python = ">=3.13"
@@ -21,4 +21,6 @@ build-backend = "hatchling.build"
 exclude = ["log/*"]
 
 [project.scripts]
+xgae = "xgae.cli_app:main"
 xgae-tools = "xgae.tools.without_general_tools_app:main"
+custom_fault_tools = "examples.tools.custom_fault_tools_app:main"

xgae-0.1.10/src/examples/agent/langgraph/react/react_agent.py

@@ -0,0 +1,177 @@
+import asyncio
+import logging
+from typing import Any, Dict, List, Annotated, Sequence, TypedDict
+
+from langchain_core.messages import BaseMessage, HumanMessage, AIMessage
+from langgraph.checkpoint.memory import MemorySaver
+from langgraph.graph import END, START, StateGraph
+from langgraph.graph.message import add_messages
+
+from xgae.engine.engine_base import XGATaskResult, XGAResponseMessage
+from xgae.engine.mcp_tool_box import XGAMcpToolBox
+from xgae.utils.setup_env import setup_langfuse, setup_logging
+from xgae.utils import handle_error
+from xgae.utils.misc import read_file
+
+class TaskState(TypedDict, total=False):
+    """State definition for the agent orchestration graph"""
+    messages: Annotated[Sequence[BaseMessage], add_messages]
+    user_input: str
+    next_node: str
+    context: Dict[str, Any]
+    system_prompt: str
+    custom_tools: List[str]
+    general_tools: List[str]
+    task_result: XGATaskResult
+    formatted_result: XGATaskResult
+    iteration_count: int
+
+langfuse = setup_langfuse()
+
+class XGAReactAgent:
+    MAX_TASK_RETRY = 2
+    def __init__(self):
+        self.tool_box = XGAMcpToolBox()
+
+    async def _create_graph(self) -> StateGraph:
+        try:
+            graph_builder = StateGraph(TaskState)
+
+            # Add nodes
+            graph_builder.add_node("supervisor", self._supervisor_node)
+            graph_builder.add_node("select_tool", self._select_tool_node)
+            graph_builder.add_node("exec_task", self._exec_task_node)
+            graph_builder.add_node("eval_result", self._eval_result_node)
+            graph_builder.add_node("format_result", self._format_result_node)
+
+            # Add edges
+            graph_builder.add_edge(START, "supervisor")
+            graph_builder.add_conditional_edges(
+                "supervisor",
+                self._next_condition,
+                {
+                    "select_tool": "select_tool",
+                    "exec_task": "exec_task",
+                    "format_result": "format_result"
+                }
+            )
+
+            graph_builder.add_edge("select_tool", "exec_task")
+            graph_builder.add_edge("exec_task", "eval_result")
+
+            graph_builder.add_conditional_edges(
+                "eval_result",
+                self._next_condition,
+                {
+                    "retry": "supervisor",
+                    "format_result": "format_result",
+                }
+            )
+            
+            graph_builder.add_edge("format_result", END)
+            
+            graph = graph_builder.compile(checkpointer=MemorySaver())
+            graph.name = "XGARectAgent"
+
+            return graph
+        except Exception as e:
+            logging.error("Failed to create XGARectAgent graph: %s", str(e))
+            raise
+
+
+    async def _supervisor_node(self, state: TaskState) -> Dict[str, Any]:
+        user_input = state.get("user_input", "")
+        system_prompt = None if "fault" in user_input else read_file("templates/example/fault_user_prompt.txt")
+        return {
+            "system_prompt" : system_prompt,
+            "next_node" : "select_tool",
+        }
+
+    async def _select_tool_node(self, state: TaskState) -> Dict[str, Any]:
+        system_prompt = state.get("system_prompt",None)
+        general_tools = ["*"] if system_prompt else []
+        custom_tools = ["*"] if not system_prompt  else []
+        return {
+            "general_tools" : general_tools,
+            "custom_tools" : custom_tools,
+        }
+
+    async def _exec_task_node(self, state: TaskState) -> Dict[str, Any]:
+        task_result = XGATaskResult(type="answer", content="test task result")
+        return {
+            "task_result" : task_result
+        }
+
+    async def _eval_result_node(self, state: TaskState) -> Dict[str, Any]:
+        next_node = "end"
+        return {
+            "next_node" : next_node
+        }
+
+    async def _format_result_node(self, state: TaskState) -> Dict[str, Any]:
+        formatted_result = state.get("task_result")
+        return {
+            "formatted_result" : formatted_result,
+            "messages": state["messages"] + [AIMessage(content=f"")]
+        }
+    
+    def _next_condition(self, state: TaskState) -> str:
+        next_node = state.get("next_node")
+        return next_node
+
+
+    async def generate(self, user_input: str) -> XGATaskResult:
+        result = None
+        try:
+            logging.info("****** Start React Agent for user_input: %s", user_input)
+
+            # Create graph if not already created
+            if self.graph is None:
+                self.graph = await self._create_graph()
+
+            # Initialize state
+            initial_state = {
+                "messages": [HumanMessage(content=f"information for: {user_input}")],
+                "user_input": user_input,
+                "next_node": None,
+                "tasks": [],
+                "context": "",
+                "current_task": None,
+                "next_task": None,
+                "formatted_result": "",
+                "final_error_info": "",
+                "iteration_count": 1
+            }
+
+            # Run the retrieval graph with proper configuration
+            config = {"recursion_limit": 100,
+                      "configurable": {"thread_id": "manager_async_generate_thread"}}
+            final_state = await self.graph.ainvoke(initial_state, config=config)
+
+            # Parse and return formatted results
+            result = final_state["formatted_result"]
+
+            logging.info("=" * 100)
+            logging.info("User question: %s", user_input)
+            logging.info("User answer: %s", result)
+            logging.info("=" * 100)
+
+            return result
+        except Exception as e:
+            logging.error("### Error ManagerAgent _agent_work for user_input '%s': %s ###", user_input, str(e))
+            handle_error(e)
+            result = XGATaskResult(type="error", content="Never get result, Unexpected Error")
+            return result
+
+
+if __name__ == "__main__":
+    setup_logging()
+
+    agent = XGAReactAgent()
+    user_inputs = [
+                    "Create a function to sort a list of numbers, sort [6,8,7,5]"
+                    , "sort [3,2,7,5]"
+                ]
+    for user_input in user_inputs:
+        result = agent.generate(user_input)
+        print(result)

{xgae-0.1.8/src/examples → xgae-0.1.10/src/examples/engine}/run_human_in_loop.py

@@ -5,10 +5,16 @@ from xgae.engine.task_engine import XGATaskEngine
 from xgae.utils.llm_client import LLMConfig
 from xgae.utils.misc import read_file
 
+from xgae.utils.setup_env import setup_langfuse, setup_logging
+
+setup_logging()
+langfuse = setup_langfuse()
 
 async def main() -> None:
+    # Before Run Exec: uv run custom_fault_tools
     tool_box = XGAMcpToolBox(custom_mcp_server_file="mcpservers/custom_servers.json")
-    system_prompt = read_file("templates/example_user_prompt.txt")
+    system_prompt = read_file("templates/example/fault_user_prompt.txt")
+
     engine = XGATaskEngine(tool_box=tool_box,
                            general_tools=[],
                            custom_tools=["*"],
@@ -16,14 +22,23 @@ async def main() -> None:
                            system_prompt=system_prompt,
                            max_auto_run=8)
 
+    # Two task run in same langfuse trace
+    trace_id = langfuse.trace(name="xgae_example_run_human_in_loop").trace_id
+
     user_input =  "locate fault and solution"
-    final_result = await engine.run_task_with_final_answer(task_message={"role": "user", "content": user_input})
+    final_result = await engine.run_task_with_final_answer(
+        task_message={"role": "user", "content": user_input},
+        trace_id=trace_id
+    )
     print("FINAL RESULT:", final_result)
 
     if final_result["type"] == "ask":
         print("====== Wait for user input ... ======")
         user_input = "ip=10.0.1.1"
-        final_result = await engine.run_task_with_final_answer(task_message={"role": "user", "content": user_input})
+        final_result = await engine.run_task_with_final_answer(
+            task_message={"role": "user", "content": user_input},
+            trace_id=trace_id
+        )
         print("FINAL RESULT:", final_result)
 
 asyncio.run(main())

{xgae-0.1.8/src/examples → xgae-0.1.10/src/examples/engine}/run_simple.py

@@ -3,10 +3,18 @@ import asyncio
 from xgae.engine.task_engine import XGATaskEngine
 from xgae.utils.llm_client import LLMConfig
 
+from xgae.utils.setup_env import setup_logging
+
+setup_logging()
 
 async def main() -> None:
-    engine =  XGATaskEngine(llm_config=LLMConfig(stream=False), max_auto_run=1)
-    final_result = await engine.run_task_with_final_answer(task_message={"role": "user", "content": "1+1"})
+    engine =  XGATaskEngine(llm_config=LLMConfig(stream=False), max_auto_run=3)
+
+    final_result = await engine.run_task_with_final_answer(
+        task_message={"role": "user", "content": "1+7"}
+    )
+
     print("FINAL RESULT:", final_result)
 
+
 asyncio.run(main())

{xgae-0.1.8/src/examples → xgae-0.1.10/src/examples/engine}/run_user_prompt.py

@@ -5,10 +5,15 @@ from xgae.engine.task_engine import XGATaskEngine
 from xgae.utils.llm_client import LLMConfig
 from xgae.utils.misc import read_file
 
+from xgae.utils.setup_env import setup_logging
+
+setup_logging()
 
 async def main() -> None:
+    # Before Run Exec: uv run custom_fault_tools
     tool_box = XGAMcpToolBox(custom_mcp_server_file="mcpservers/custom_servers.json")
-    system_prompt = read_file("templates/example_user_prompt.txt")
+    system_prompt = read_file("templates/example/fault_user_prompt.txt")
+
     engine = XGATaskEngine(tool_box=tool_box,
                            general_tools=[],
                            custom_tools=["*"],

xgae-0.1.10/src/examples/tools/custom_fault_tools_app.py

@@ -0,0 +1,104 @@
+import click
+import logging
+
+from typing import Annotated
+from typing import Dict, Any
+from pydantic import Field
+
+from mcp.server.fastmcp import FastMCP
+
+mcp = FastMCP(name="Fault Location Tools")
+alarm_type = 0
+
+@mcp.tool(
+    description="Get Alarm Object",
+)
+def get_alarm(ip: Annotated[str, Field(description="Alarm Object IP Address")]) -> Dict[str, str]:
+    logging.info(f"get_alarm: ip={ip}")
+    return {"alarmId":"alm0123", "objId": "obj567"}
+
+
+@mcp.tool(
+    description="Get alarm type, return result enum: 1: Business Type Alarm; 2: Equipment Type Alarm; 3: Middleware Type Alarm",
+)
+def get_alarm_type(alarm_id: Annotated[str, Field(description="Alarm Object Id")]) -> str:
+    logging.info(f"get_alarm_type: alarm_id={alarm_id}, alarm_type={alarm_type}")
+    return f"{alarm_type}"
+
+
+@mcp.tool(
+    description="Locate Business Type Fault",
+)
+def get_busi_fault(obj_id: Annotated[str, Field(description="Alarm Object Id")]) -> Dict[str, Any]:
+    logging.info(f"get_busi_fault: obj_id={obj_id}")
+    return {"fault_code": "F01", "fault_desc": "Business Recharge Fault"}
+
+
+@mcp.tool(
+    description="Locate Equipment Type Fault",
+)
+def get_equip_fault(obj_id: Annotated[str, Field(description="Alarm Object Id")]) -> Dict[str, Any]:
+    logging.info(f"get_equip_fault: obj_id={obj_id}")
+    return {"fault_code": "F02", "fault_desc": "Host Disk Fault"}
+
+
+@mcp.tool(
+    description="Locate Middleware Type Fault",
+)
+def get_middle_fault(obj_id: Annotated[str, Field(description="Alarm Object Id")]) -> Dict[str, Any]:
+    logging.info(f"get_middle_fault: obj_id={obj_id}")
+    return {"fault_code": "F03", "fault_desc": "Redis Fault"}
+
+
+@mcp.tool(
+    description="Get Business Type Fault Solution and Cause",
+)
+async def get_busi_fault_cause(fault_code: Annotated[str, Field(description="Fault Code")]) -> str:
+    logging.info(f"get_busi_fault_cause: faultCode={fault_code}")
+
+    fault_cause = ""
+    if (fault_code == 'F01'):
+        fault_cause = "Business Recharge Fault, Fault Cause is 'Phone Recharge Application Crash' ，Solution is 'Restart Phone Recharge Application'"
+    else:
+        fault_cause = f"FaultCode '{fault_code}' is not Business Type"
+
+    return  fault_cause
+
+
+@mcp.tool(
+    description="Get Equipment Type Fault Solution and Cause",
+)
+async def get_equip_fault_cause(fault_code: Annotated[str, Field(description="Fault Code")]) -> str:
+    logging.info(f"get_equip_fault_cause: faultCode={fault_code}")
+
+    fault_cause = ""
+    if (fault_code == 'F01'):
+        fault_cause = "Host Fault, Fault Cause is 'Host Disk is Damaged' ，Solution is 'Change Host Disk'"
+    else:
+        fault_cause = f"FaultCode '{fault_code}' is not Equipment Type"
+
+    return  fault_cause
+
+
+@click.command()
+@click.option("--transport", type=click.Choice(["stdio", "sse"]), default="sse", help="Transport type")
+@click.option("--host", default="0.0.0.0", help="Host to listen on for SSE")
+@click.option("--port", default=17070, help="Port to listen on for SSE")
+@click.option("--alarmtype", default=1, help="AlarmType Set")
+def main(transport: str, host: str, port: int, alarmtype:int):
+    if transport != "stdio":
+        from xgae.utils.setup_env import setup_logging
+        setup_logging()
+        logging.info("=" * 20 + f"   Custom Fault Tools Sever Started in {transport} mode   " + "=" * 20)
+
+    global alarm_type
+    alarm_type = alarmtype
+
+    mcp.settings.host = host
+    mcp.settings.port = port
+
+    mcp.run(transport=transport)
+
+
+if __name__ == '__main__':
+    main()

xgae-0.1.10/src/xgae/__init__.py

@@ -0,0 +1,4 @@
+if __name__ == "__main__":
+    from xgae.cli_app import main
+
+    main()

xgae-0.1.10/src/xgae/cli_app.py

@@ -0,0 +1,87 @@
+import asyncio
+import sys
+
+from xgae.engine.mcp_tool_box import XGAMcpToolBox
+from xgae.engine.task_engine import XGATaskEngine
+from xgae.utils.llm_client import LLMConfig
+from xgae.utils.misc import read_file
+
+from xgae.utils.setup_env import setup_langfuse, setup_env_logging
+
+setup_env_logging()
+langfuse = setup_langfuse()
+
+def get_user_message(question)-> str:
+    while True:
+        user_message = input(f"\n💬 {question}: ")
+        if user_message.lower() == 'exit':
+            print("\n====== Extreme General Agent Engine CLI EXIT ======")
+            sys.exit()
+
+        if not user_message.strip():
+            print("\nuser message is empty， input agin ！！！\n")
+            continue
+
+        return user_message
+
+async def cli() -> None:
+    await asyncio.sleep(1)
+    print("\n====== Extreme General Agent Engine CLI START ======")
+    user_message = input("\n💬 Start Custom MCP Server and Load User Prompt (Yes/No): ")
+    tool_box = None
+    system_prompt = None
+    general_tools = []
+    custom_tools = []
+    if user_message.lower() == 'yes':
+        print(f"--- Start Custom MCP Server in custom_servers.json")
+        print(f"--- Load User Prompt in example/fault_user_prompt.txt")
+        tool_box = XGAMcpToolBox(custom_mcp_server_file="mcpservers/custom_servers.json")
+        system_prompt = read_file("templates/example/fault_user_prompt.txt")
+        custom_tools = ["*"]
+    else:
+        print(f"--- Start General Agent Server")
+        print(f"--- Load System Prompt")
+        general_tools = ["*"]
+
+    while True:
+        user_message = get_user_message("Enter your message (or 'exit' to quit)")
+
+        print("\n🔄 Running XGA Engine ...\n")
+        engine = XGATaskEngine(tool_box=tool_box,
+                               general_tools=general_tools,
+                               custom_tools=custom_tools,
+                               llm_config=LLMConfig(stream=False),
+                               system_prompt=system_prompt,
+                               max_auto_run=8)
+
+        # Two task run in same langfuse trace
+        trace_id = langfuse.trace(name="xgae_cli").trace_id
+
+        final_result = await engine.run_task_with_final_answer(
+            task_message={"role": "user", "content": user_message},
+            trace_id=trace_id
+        )
+
+        if final_result["type"] == "ask":
+            await asyncio.sleep(1)
+            print(f"\n📌 ASK INFO: {final_result['content']}")
+            user_message = get_user_message("Enter ASK information (or 'exit' to quit)")
+            final_result = await engine.run_task_with_final_answer(
+                task_message={"role": "user", "content": user_message},
+                trace_id=trace_id
+            )
+
+        await asyncio.sleep(1)
+        result_prefix = "✅" if final_result["type"] == "answer" else "❌"
+        if final_result["type"] == "ask":
+            print("\n *** IMPORTANT: XGA CLI only support showing ONE TURN ASK !")
+            result_prefix = "⚠️"
+        print(f"\n {result_prefix} FINAL RESULT: {final_result['content']}")
+
+
+def main():
+    asyncio.run(cli())
+
+
+if __name__ == "__main__":
+    main()

{xgae-0.1.8 → xgae-0.1.10}/src/xgae/engine/engine_base.py

@@ -7,7 +7,7 @@ class XGAError(Exception):
     pass
 
 XGAMsgStatusType = Literal["error", "finish", "tool_error", "tool_started", "tool_completed", "tool_failed", "thread_run_start", "thread_run_end", "assistant_response_start", "assistant_response_end"]
-XGAResponseMsgType = Literal["user", "status",  "tool", "assistant"]
+XGAResponseMsgType = Literal["user", "status",  "tool", "assistant", "assistant_complete"]
 
 class XGAResponseMessage(TypedDict, total=False):
     message_id: str

{xgae-0.1.8 → xgae-0.1.10}/src/xgae/engine/mcp_tool_box.py

@@ -190,11 +190,16 @@ class XGAMcpToolBox(XGAToolBox):
 if __name__ == "__main__":
     import asyncio
     from dataclasses import asdict
+    from xgae.utils.setup_env import setup_logging
+
+    setup_logging()
 
     async def main():
-        task_id = "task1"
+        ## Before Run Exec: uv run custom_fault_tools
         mcp_tool_box = XGAMcpToolBox(custom_mcp_server_file="mcpservers/custom_servers.json")
         #mcp_tool_box = XGAMcpToolBox()
+
+        task_id = "task1"
         await mcp_tool_box.load_mcp_tools_schema()
         await mcp_tool_box.creat_task_tool_box(task_id=task_id, general_tools=["*"], custom_tools=["bomc_fault.*"])
         tool_schemas = mcp_tool_box.get_task_tool_schemas(task_id, "general_tool")

{xgae-0.1.8 → xgae-0.1.10}/src/xgae/engine/prompt_builder.py

@@ -3,7 +3,7 @@ import datetime
 
 from typing import Optional, List
 
-from xgae.engine.engine_base import XGAToolSchema, XGAError
+from xgae.engine.engine_base import XGAToolSchema
 from xgae.utils.misc import read_file, format_file_with_args