PyPI - xgae - Versions diffs - 0.1.6__tar.gz → 0.1.8__tar.gz - Mend

xgae 0.1.6tar.gz → 0.1.8tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of xgae might be problematic. Click here for more details.

Files changed (45) hide show

{xgae-0.1.6 → xgae-0.1.8}/.env RENAMED Viewed

@@ -1,6 +1,7 @@
 # LOG
 LOG_LEVEL=INFO
 LOG_FILE=log/xgae.log
+LOG_ENABLE=True
 # LANGFUSE
 LANGFUSE_PUBLIC_KEY=
@@ -15,6 +16,8 @@ LLM_MAX_TOKENS=16384
 LLM_TEMPERATURE=0.7
 LLM_MAX_RETRIES=2
+# TASK
+MAX_AUTO_RUN = 15

{xgae-0.1.6 → xgae-0.1.8}/.idea/misc.xml RENAMED Viewed

@@ -3,5 +3,5 @@
   <component name="Black">
     <option name="sdkName" value="Python 3.13 (xgae)" />
   </component>
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.13 virtualenv at ~/DevelopSpace/xgae/.venv" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.13 virtualenv at ~/DevProjects/xga/xgae/.venv" project-jdk-type="Python SDK" />
 </project>

{xgae-0.1.6 → xgae-0.1.8}/.idea/workspace.xml RENAMED Viewed

@@ -25,26 +25,37 @@
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <component name="PropertiesComponent">{
-  &quot;keyToString&quot;: {
-    &quot;ModuleVcsDetector.initialDetectionPerformed&quot;: &quot;true&quot;,
-    &quot;Python.llm_client.executor&quot;: &quot;Run&quot;,
-    &quot;Python.run_xga_engine.executor&quot;: &quot;Run&quot;,
-    &quot;Python.setup_env.executor&quot;: &quot;Run&quot;,
-    &quot;Python.utils.executor&quot;: &quot;Run&quot;,
-    &quot;Python.xga_engine.executor&quot;: &quot;Run&quot;,
-    &quot;Python.xga_mcp_tool_box.executor&quot;: &quot;Run&quot;,
-    &quot;RunOnceActivity.ShowReadmeOnStart&quot;: &quot;true&quot;,
-    &quot;last_opened_file_path&quot;: &quot;/Users/goosezzy/DevelopSpace/xgae&quot;,
-    &quot;node.js.detected.package.eslint&quot;: &quot;true&quot;,
-    &quot;node.js.detected.package.tslint&quot;: &quot;true&quot;,
-    &quot;node.js.selected.package.eslint&quot;: &quot;(autodetect)&quot;,
-    &quot;node.js.selected.package.tslint&quot;: &quot;(autodetect)&quot;,
-    &quot;nodejs_package_manager_path&quot;: &quot;npm&quot;,
-    &quot;settings.editor.selected.configurable&quot;: &quot;preferences.pluginManager&quot;,
-    &quot;vue.rearranger.settings.migration&quot;: &quot;true&quot;
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "ModuleVcsDetector.initialDetectionPerformed": "true",
+    "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_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.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",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "last_opened_file_path": "/Users/sharkystar/DevProjects/xga/xgae",
+    "node.js.detected.package.eslint": "true",
+    "node.js.detected.package.tslint": "true",
+    "node.js.selected.package.eslint": "(autodetect)",
+    "node.js.selected.package.tslint": "(autodetect)",
+    "nodejs_package_manager_path": "npm",
+    "settings.editor.selected.configurable": "com.intellij.pycharm.community.ide.impl.configuration.PythonContentEntriesConfigurable",
+    "vue.rearranger.settings.migration": "true"
   }
-}</component>
+}]]></component>
   <component name="RecentsManager">
     <key name="MoveFile.RECENT_KEYS">
       <recent name="$PROJECT_DIR$/src/xgae/engine/responser" />
@@ -60,6 +71,7 @@
         <env name="PYTHONUNBUFFERED" value="1" />
       </envs>
       <option name="SDK_HOME" value="" />
+      <option name="SDK_NAME" value="Python 3.13 virtualenv at ~/DevProjects/xga/xgae/.venv" />
       <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$" />
       <option name="IS_MODULE_SDK" value="false" />
       <option name="ADD_CONTENT_ROOTS" value="true" />
@@ -84,7 +96,7 @@
       <option name="INPUT_FILE" value="" />
       <method v="2" />
     </configuration>
-    <configuration name="run_xga_engine" type="PythonConfigurationType" factoryName="Python" temporary="true">
+    <configuration name="run_task_engine" type="PythonConfigurationType" factoryName="Python">
       <module name="xgae" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -117,11 +129,6 @@
       <option name="INPUT_FILE" value="" />
       <method v="2" />
     </configuration>
-    <recent_temporary>
-      <list>
-        <item itemvalue="Python.run_xga_engine" />
-      </list>
-    </recent_temporary>
   </component>
   <component name="SharedIndexes">
     <attachedChunks>
@@ -145,46 +152,53 @@
       <workItem from="1755270285722" duration="12068000" />
       <workItem from="1755283728206" duration="1511000" />
       <workItem from="1755286552901" duration="3130000" />
-      <workItem from="1755307767865" duration="337000" />
-      <workItem from="1755308113420" duration="2825000" />
-      <workItem from="1755321931380" duration="8531000" />
-      <workItem from="1755330830188" duration="229000" />
-      <workItem from="1755396985976" duration="6435000" />
-      <workItem from="1755415614018" duration="461000" />
-      <workItem from="1755416476555" duration="256000" />
-      <workItem from="1755419142521" duration="150000" />
-      <workItem from="1755429851003" duration="290000" />
-      <workItem from="1755430331676" duration="21000" />
-      <workItem from="1755529583730" duration="1643000" />
-      <workItem from="1755609955702" duration="37000" />
-      <workItem from="1755612319869" duration="621000" />
-      <workItem from="1755690345086" duration="6847000" />
-      <workItem from="1755781193226" duration="485000" />
-      <workItem from="1755781685826" duration="38000" />
-      <workItem from="1755781735460" duration="170000" />
-      <workItem from="1755817545372" duration="285000" />
-      <workItem from="1755862006389" duration="1051000" />
-      <workItem from="1755863094184" duration="553000" />
-      <workItem from="1755863979826" duration="4236000" />
-      <workItem from="1755875255916" duration="2308000" />
-      <workItem from="1755877576514" duration="2000" />
-      <workItem from="1755878805800" duration="3590000" />
-      <workItem from="1755904284653" duration="7803000" />
-      <workItem from="1755912586948" duration="1553000" />
-      <workItem from="1755915353109" duration="5872000" />
-      <workItem from="1755997335603" duration="7550000" />
+      <workItem from="1755409529770" duration="670000" />
+      <workItem from="1755478595168" duration="20000" />
+      <workItem from="1755479257156" duration="16895000" />
+      <workItem from="1755525531052" duration="13030000" />
+      <workItem from="1755585869510" duration="5796000" />
+      <workItem from="1755593112104" duration="786000" />
+      <workItem from="1755611972189" duration="13340000" />
+      <workItem from="1755668525673" duration="14877000" />
+      <workItem from="1755700523844" duration="24000" />
+      <workItem from="1755737435202" duration="48139000" />
+      <workItem from="1756044658912" duration="1248000" />
+      <workItem from="1756082326044" duration="23657000" />
+      <workItem from="1756168626188" duration="26778000" />
     </task>
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">
     <option name="version" value="3" />
   </component>
+  <component name="XDebuggerManager">
+    <breakpoint-manager>
+      <default-breakpoints>
+        <breakpoint type="python-exception">
+          <properties notifyOnTerminate="true" exception="BaseException">
+            <option name="notifyOnTerminate" value="true" />
+          </properties>
+        </breakpoint>
+      </default-breakpoints>
+    </breakpoint-manager>
+  </component>
   <component name="com.intellij.coverage.CoverageDataManagerImpl">
-    <SUITE FILE_PATH="coverage/xgae$xga_mcp_tool_box.coverage" NAME="xga_mcp_tool_box Coverage Results" MODIFIED="1755864064753" 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="1755247632274" 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="1755876041210" 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_xga_engine Coverage Results" MODIFIED="1756003050644" 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="1755876180376" 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_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$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_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$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$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$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$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$" />
   </component>
 </project>

{xgae-0.1.6 → xgae-0.1.8}/.idea/xgae.iml RENAMED Viewed

@@ -5,7 +5,7 @@
       <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
       <excludeFolder url="file://$MODULE_DIR$/.venv" />
     </content>
-    <orderEntry type="jdk" jdkName="Python 3.13 virtualenv at ~/DevelopSpace/xgae/.venv" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="Python 3.13 virtualenv at ~/DevProjects/xga/xgae/.venv" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
 </module>

xgae-0.1.8/PKG-INFO ADDED Viewed

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: xgae
+Version: 0.1.8
+Summary: Extreme General Agent Engine
+Requires-Python: >=3.13
+Requires-Dist: colorlog==6.9.0
+Requires-Dist: langchain-mcp-adapters==0.1.9
+Requires-Dist: langfuse==2.60.9
+Requires-Dist: langgraph==0.6.5
+Requires-Dist: litellm==1.74.15
+Requires-Dist: mcp==1.13.0

xgae-0.1.8/mcpservers/xga_server.json ADDED Viewed

@@ -0,0 +1,11 @@
+{
+    "mcpServers":{
+        "xga_general": {
+      		"command": "uv",
+      		"args": [
+        		"run",
+        		"xgae-tools"
+      		]
+        }
+    }
+}

{xgae-0.1.6 → xgae-0.1.8}/pyproject.toml RENAMED Viewed

@@ -1,16 +1,16 @@
 [project]
 name = "xgae"
-version = "0.1.6"
+version = "0.1.8"
 description = "Extreme General Agent Engine"
 readme = "README.md"
 requires-python = ">=3.13"
 dependencies = [
-    "colorlog>=6.9.0",
-    "langchain-mcp-adapters>=0.1.4",
-    "langgraph>=0.3.21",
-    "litellm>=1.74.8",
-    "mcp>=1.12.1",
-    "langfuse>=2.60.5",
+    "colorlog==6.9.0",
+    "langchain-mcp-adapters==0.1.9",
+    "langgraph==0.6.5",
+    "litellm==1.74.15",
+    "mcp==1.13.0",
+    "langfuse==2.60.9",
 ]
 [build-system]
@@ -21,3 +21,4 @@ build-backend = "hatchling.build"
 exclude = ["log/*"]
 [project.scripts]
+xgae-tools = "xgae.tools.without_general_tools_app:main"

xgae-0.1.8/src/examples/run_human_in_loop.py ADDED Viewed

@@ -0,0 +1,29 @@
+import asyncio
+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
+async def main() -> None:
+    tool_box = XGAMcpToolBox(custom_mcp_server_file="mcpservers/custom_servers.json")
+    system_prompt = read_file("templates/example_user_prompt.txt")
+    engine = XGATaskEngine(tool_box=tool_box,
+                           general_tools=[],
+                           custom_tools=["*"],
+                           llm_config=LLMConfig(stream=False),
+                           system_prompt=system_prompt,
+                           max_auto_run=8)
+    user_input =  "locate fault and solution"
+    final_result = await engine.run_task_with_final_answer(task_message={"role": "user", "content": user_input})
+    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})
+        print("FINAL RESULT:", final_result)
+asyncio.run(main())

xgae-0.1.8/src/examples/run_simple.py ADDED Viewed

@@ -0,0 +1,12 @@
+import asyncio
+from xgae.engine.task_engine import XGATaskEngine
+from xgae.utils.llm_client import LLMConfig
+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"})
+    print("FINAL RESULT:", final_result)
+asyncio.run(main())

xgae-0.1.8/src/examples/run_user_prompt.py ADDED Viewed

@@ -0,0 +1,30 @@
+import asyncio
+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
+async def main() -> None:
+    tool_box = XGAMcpToolBox(custom_mcp_server_file="mcpservers/custom_servers.json")
+    system_prompt = read_file("templates/example_user_prompt.txt")
+    engine = XGATaskEngine(tool_box=tool_box,
+                           general_tools=[],
+                           custom_tools=["*"],
+                           llm_config=LLMConfig(stream=False),
+                           system_prompt=system_prompt,
+                           max_auto_run=8)
+    user_input =  "locate 10.2.3.4 fault and solution"
+    is_final_result = False
+    if is_final_result:
+        final_result = await engine.run_task_with_final_answer(task_message={"role": "user", "content": user_input})
+        print("FINAL RESULT:", final_result)
+    else:
+        # Get All Task Process Message
+        async for chunk in engine.run_task(task_message={"role": "user", "content": user_input}):
+            print(chunk)
+asyncio.run(main())

{xgae-0.1.6 → xgae-0.1.8}/src/xgae/engine/engine_base.py RENAMED Viewed

@@ -55,4 +55,4 @@ class XGAToolBox(ABC):
     @abstractmethod
     async def call_tool(self, task_id: str, tool_name: str, args: Optional[Dict[str, Any]] = None) -> XGAToolResult:
-        pass
+        pass

{xgae-0.1.6 → xgae-0.1.8}/src/xgae/engine/mcp_tool_box.py RENAMED Viewed

@@ -8,7 +8,6 @@ from langchain_mcp_adapters.client import MultiServerMCPClient
 from langchain_mcp_adapters.tools import load_mcp_tools
 from xgae.engine.engine_base import XGAError, XGAToolSchema, XGAToolBox, XGAToolResult
-from xgae.utils import langfuse
 class XGAMcpToolBox(XGAToolBox):
     GENERAL_MCP_SERVER_NAME = "xga_general"
@@ -46,6 +45,12 @@ class XGAMcpToolBox(XGAToolBox):
                     task_tool_schemas[tool_schema.tool_name] = tool_schema
         task_tool_schemas.pop("end_task", None)
+        if len(custom_tools) == 1 and custom_tools[0] == "*":
+            custom_tools = []
+            for server_name in self.mcp_server_names:
+                if server_name != XGAMcpToolBox.GENERAL_MCP_SERVER_NAME:
+                    custom_tools.append(f"{server_name}.*")
         for server_tool_name in custom_tools:
             parts = server_tool_name.split(".")
             if len(parts) != 2:
@@ -188,8 +193,8 @@ if __name__ == "__main__":
     async def main():
         task_id = "task1"
-        #mcp_tool_box = XGAMcpToolBox(custom_mcp_server_file="mcpservers/custom_servers.json")
-        mcp_tool_box = XGAMcpToolBox()
+        mcp_tool_box = XGAMcpToolBox(custom_mcp_server_file="mcpservers/custom_servers.json")
+        #mcp_tool_box = XGAMcpToolBox()
         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")
@@ -204,9 +209,6 @@ if __name__ == "__main__":
             print(asdict(tool_schema))
         print()
-        result = await mcp_tool_box.call_tool(task_id=task_id, tool_name="web_search", args={"task_id": task_id, "query": "查询天津天气"})
-        print(f"call web_search result: {result}")
         result = await mcp_tool_box.call_tool(task_id=task_id, tool_name="complete", args={"task_id": task_id})
         print(f"call complete result: {result}")

{xgae-0.1.6 → xgae-0.1.8}/src/xgae/engine/prompt_builder.py RENAMED Viewed

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

{xgae-0.1.6 → xgae-0.1.8}/src/xgae/engine/responser/non_stream_responser.py RENAMED Viewed

@@ -3,7 +3,6 @@ import logging
 from typing import List, Dict, Any, AsyncGenerator, override,Optional
 from xgae.engine.responser.responser_base import TaskResponseProcessor, TaskResponserContext, TaskRunContinuousState
-from xgae.utils import langfuse
 from xgae.utils.json_helpers import format_for_yield
 class NonStreamTaskResponser(TaskResponseProcessor):
@@ -16,6 +15,7 @@ class NonStreamTaskResponser(TaskResponseProcessor):
         llm_content = ""
         parsed_xml_data = []
         finish_reason = None
+        llm_count = continuous_state.get("auto_continue_count")
         try:
             # Extract finish_reason, content, tool calls
@@ -24,8 +24,8 @@ class NonStreamTaskResponser(TaskResponseProcessor):
                     finish_reason = llm_response.choices[0].finish_reason
                     logging.info(f"NonStreamTask：LLM response finish_reason={finish_reason}")
-                    langfuse.create_event(trace_context=self.trace_context, name="non_streaming_finish_reason", level="DEFAULT",
-                                     status_message=(f"Non-streaming finish_reason: {finish_reason}"))
+                    self.root_span.event(name=f"non_stream_processor_start[{self.task_no}]({llm_count})", level="DEFAULT",
+                                     status_message=(f"finish_reason={finish_reason}, tool_exec_strategy={self.tool_execution_strategy}"))
                 response_message = llm_response.choices[0].message if hasattr(llm_response.choices[0], 'message') else None
                 if response_message:
@@ -54,8 +54,6 @@ class NonStreamTaskResponser(TaskResponseProcessor):
             tool_calls_to_execute = [item['tool_call'] for item in parsed_xml_data]
             if  len(tool_calls_to_execute) > 0:
                 logging.info(f"NonStreamTask：Executing {len(tool_calls_to_execute)} tools with strategy: {self.tool_execution_strategy}")
-                langfuse.create_event(trace_context=self.trace_context, name="executing_tools_with_strategy", level="DEFAULT", status_message=(
-                    f"NonStreamTask Executing {len(tool_calls_to_execute)} tools with strategy: {self.tool_execution_strategy}"))
                 tool_results = await self._execute_tools(tool_calls_to_execute, self.tool_execution_strategy)
@@ -92,7 +90,7 @@ class NonStreamTaskResponser(TaskResponseProcessor):
         except Exception as e:
             logging.error(f"NonStreamTask: Error processing non-streaming response: {llm_content}")
-            langfuse.create_event(trace_context=self.trace_context, name="error_processing_non_streaming_response", level="ERROR",
+            self.root_span.event(name="error_processing_non_streaming_response", level="ERROR",
                              status_message=(f"Error processing non-streaming response: {str(e)}"))
             content = {"role": "system", "status_type": "error", "message": str(e)}
@@ -102,7 +100,7 @@ class NonStreamTaskResponser(TaskResponseProcessor):
             # Re-raise the same exception (not a new one) to ensure proper error propagation
             logging.critical(f"NonStreamTask: Re-raising error to stop further processing: {str(e)}")
-            langfuse.create_event(trace_context=self.trace_context, name="re_raising_error_to_stop_further_processing", level="CRITICAL",
+            self.root_span.event(name="re_raising_error_to_stop_further_processing", level="CRITICAL",
                              status_message=(f"Re-raising error to stop further processing: {str(e)}"))
             raise  # Use bare 'raise' to preserve the original exception with its traceback

xgae 0.1.6__tar.gz → 0.1.8__tar.gz

Potentially problematic release.

xgae 0.1.6tar.gz → 0.1.8tar.gz