xgae 0.1.9__py3-none-any.whl → 0.1.10__py3-none-any.whl

xgae/__init__.py

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

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/engine/responser/non_stream_responser.py

@@ -82,6 +82,9 @@ class NonStreamTaskResponser(TaskResponseProcessor):
                         finish_reason = "completed"
                         break
                     tool_index += 1
+            else:
+                finish_reason = "non_tool_call"
+                logging.warning(f"NonStreamTask: tool_calls is empty, No Tool need to call !")
 
             if finish_reason:
                 finish_content = {"status_type": "finish", "finish_reason": finish_reason}

xgae/engine/task_engine.py

@@ -117,7 +117,7 @@ class XGATaskEngine:
 
             self.task_prompt = self.prompt_builder.build_task_prompt(self.model_name, general_tool_schemas, custom_tool_schemas)
 
-            logging.info("*" * 30 + f"   XGATaskEngine Task'{self.task_id}' Initialized   " + "*" * 30)
+            logging.info("*" * 10 + f"   XGATaskEngine Task'{self.task_id}' Initialized   " + "*" * 10)
             logging.info(f"model_name={self.model_name}, is_stream={self.is_stream}")
             logging.info(f"general_tools={general_tools}, custom_tools={custom_tools}")
 
@@ -159,6 +159,10 @@ class XGATaskEngine:
                                     logging.warning(f"run_task_auto: Detected finish_reason='xml_tool_limit_reached', stop auto-continue")
                                     auto_continue = False
                                     break
+                                elif finish_reason == 'non_tool_call':
+                                    logging.warning(f"run_task_auto: Detected finish_reason='non_tool_call', stop auto-continue")
+                                    auto_continue = False
+                                    break
                                 elif finish_reason == 'stop' or finish_reason == 'length': # 'length' never occur
                                     auto_continue = True
                                     auto_continue_count += 1
@@ -191,14 +195,14 @@ class XGATaskEngine:
             }
             llm_messages.append(temp_assistant_message)
 
-        llm_count = continuous_state.get("auto_continue_count")
-        langfuse_metadata = self.task_langfuse.create_llm_langfuse_meta(llm_count)
+        auto_count = continuous_state.get("auto_continue_count")
+        langfuse_metadata = self.task_langfuse.create_llm_langfuse_meta(auto_count)
 
         llm_response = await self.llm_client.create_completion(llm_messages, langfuse_metadata)
         response_processor = self._create_response_processer()
 
         async for chunk in response_processor.process_response(llm_response, llm_messages, continuous_state):
-            self._logging_reponse_chunk(chunk)
+            self._logging_reponse_chunk(chunk, auto_count)
             yield chunk
 
     def _parse_final_result(self, chunks: List[Dict[str, Any]]) -> XGATaskResult:
@@ -238,13 +242,21 @@ class XGATaskEngine:
                             result_type = "answer" if success else "error"
                             result_content = f"Task execute '{tool_name}' {result_type}: {output}"
                             final_result = XGATaskResult(type=result_type, content=result_content)
-                elif chunk_type == "assistant_complete" and finish_reason == 'stop':
+                elif chunk_type == "assistant_complete" and finish_reason == 'non_tool_call':
                     assis_content = chunk.get('content', {})
                     result_content = assis_content.get("content", "LLM output is empty")
                     final_result = XGATaskResult(type="answer", content=result_content)
 
                 if final_result is not None:
                     break
+
+            if final_result and finish_reason == "completed":
+                logging.info(f"✅ FINAL_RESULT: finish_reason={finish_reason}, final_result={final_result}")
+            elif final_result is not None:
+                logging.warning(f"⚠️ FINAL_RESULT: finish_reason={finish_reason}, final_result={final_result}")
+            else:
+                logging.warning(f"❌ FINAL_RESULT: LLM Result is EMPTY, finish_reason={finish_reason}")
+                final_result = XGATaskResult(type="error", content="LLM has no answer")
         except Exception as e:
             logging.error(f"parse_final_result: Final result pass error: {str(e)}")
             final_result = XGATaskResult(type="error", content="Parse final result failed!")
@@ -328,7 +340,7 @@ class XGATaskEngine:
         return XGATaskLangFuse(self.session_id, self.task_id, self.task_run_id, self.task_no, self.agent_id)
 
 
-    def _logging_reponse_chunk(self, chunk):
+    def _logging_reponse_chunk(self, chunk, auto_count: int) -> None:
         chunk_type = chunk.get('type')
         prefix = ""
 
@@ -342,17 +354,20 @@ class XGATaskEngine:
             tool_name = tool_execution.get('function_name')
             prefix = "-" + tool_name
 
-        logging.info(f"TASK_RESP_CHUNK[{chunk_type}{prefix}]: {chunk}")
+        logging.info(f"TASK_RESP_CHUNK[{auto_count}]<{chunk_type}{prefix}>: {chunk}")
 
 
 if __name__ == "__main__":
     import asyncio
     from xgae.utils.misc import read_file
+    from xgae.utils.setup_env import setup_logging
+
+    setup_logging()
 
     async def main():
         # 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/engine/task_langfuse.py

@@ -2,20 +2,22 @@
 from typing import Any, Dict, Optional
 from langfuse import Langfuse
 
-from xgae.utils.setup_env import setup_langfuse, setup_env_logging
+from xgae.utils.setup_env import setup_langfuse
 from xgae.utils.llm_client import LangfuseMetadata
 from xgae.engine.engine_base import XGATaskResult
 
-setup_env_logging()
-langfuse:Langfuse = setup_langfuse()
-
 class XGATaskLangFuse:
+    langfuse: Langfuse = None
+
     def __init__(self,
                  session_id: str,
                  task_id:str,
                  task_run_id: str,
                  task_no: int,
                  agent_id: str) -> None:
+        if XGATaskLangFuse.langfuse is None:
+            XGATaskLangFuse.langfuse =  setup_langfuse()
+
         self.session_id = session_id
         self.task_id = task_id
         self.task_run_id = task_run_id
@@ -35,9 +37,9 @@ class XGATaskLangFuse:
             trace = None
             if trace_id:
                 self.trace_id = trace_id
-                trace = langfuse.trace(id=trace_id)
+                trace = XGATaskLangFuse.langfuse.trace(id=trace_id)
             else:
-                trace = langfuse.trace(name="xga_task_engine")
+                trace = XGATaskLangFuse.langfuse.trace(name="xga_task_engine")
                 self.trace_id = trace.id
 
             metadata = {"task_id": self.task_id, "session_id": self.session_id, "agent_id": self.agent_id}

xgae/utils/__init__.py

@@ -12,4 +12,4 @@ def to_bool(value: str) -> bool:
     if value is None:
         return False
 
-    return True if value.lower() == "true" else False
+    return value.lower() == "true"

xgae/utils/llm_client.py

@@ -8,6 +8,7 @@ from typing import Union, Dict, Any, Optional, List, TypedDict
 from openai import OpenAIError
 from litellm.utils import ModelResponse, CustomStreamWrapper
 
+from xgae.utils import to_bool
 from xgae.utils.setup_env import setup_langfuse
 
 class LLMConfig(TypedDict, total=False):
@@ -46,6 +47,7 @@ class LLMClient:
     def __init__(self, llm_config: LLMConfig=None):
         litellm.modify_params = True
         litellm.drop_params = True
+
         self._init_langfuse()
 
         llm_config = llm_config or LLMConfig()
@@ -78,22 +80,27 @@ class LLMClient:
         self.is_stream = llm_config_params['stream']
 
         self.lite_llm_params = self._prepare_llm_params(llm_config_params)
-        logging.info(f"📡 LLMClient initialed : model={self.model_name}, is_stream={self.is_stream}, enable thinking={self.lite_llm_params['enable_thinking']}")
+        logging.info(f"=== LLMClient initialed : model={self.model_name}, is_stream={self.is_stream}, enable thinking={self.lite_llm_params['enable_thinking']}")
 
     @staticmethod
     def _init_langfuse():
         if not LLMClient.langfuse_inited:
             LLMClient.langfuse_inited =True
-            env_langfuse = setup_langfuse()
-            if env_langfuse and env_langfuse.enabled:
-                litellm.success_callback = ["langfuse"]
-                litellm.failure_callback = ["langfuse"]
-                LLMClient.langfuse_enabled = True
-                logging.info("=== LiteLLM Langfuse is enable !")
+
+            env_llm_langfuse_enable = to_bool(os.getenv("LLM_LANGFUSE_ENABLE", False))
+            if env_llm_langfuse_enable:
+                env_langfuse = setup_langfuse()
+                if env_langfuse and env_langfuse.enabled:
+                    litellm.success_callback = ["langfuse"]
+                    litellm.failure_callback = ["langfuse"]
+                    LLMClient.langfuse_enabled = True
+                    logging.info("🛠️ LiteLLM Langfuse is enable !")
+                else:
+                    LLMClient.langfuse_enabled = False
+                    logging.warning("🛠️ LiteLLM Langfuse is disable, langfuse.enabled=false !")
             else:
                 LLMClient.langfuse_enabled = False
-                logging.warning("*** LiteLLM Langfuse is disable !")
-
+                logging.warning("🛠️ LiteLLM Langfuse is disable, LLM_LANGFUSE_ENABLE=False !")
 
     def _prepare_llm_params(self, llm_config_params: Dict[str, Any]) -> Dict[str, Any]:
         prepared_llm_params = llm_config_params.copy()
@@ -244,14 +251,16 @@ if __name__ == "__main__":
     setup_logging()
     langfuse = setup_langfuse()
 
-    async def llm_completion():
+    async def main():
         llm_client = LLMClient(LLMConfig(stream=False))
 
         messages = [{"role": "user", "content": "1+1="}]
         trace_id = langfuse.trace(name = "xgae_litellm_test").trace_id
+        await asyncio.sleep(1)
+
         meta = LangfuseMetadata(
                 generation_name="llm_completion_test",
-                generation_id="generation_id",
+                generation_id="generation_id_0",
                 existing_trace_id=trace_id,
                 session_id="session_0",
         )
@@ -269,7 +278,6 @@ if __name__ == "__main__":
         else:
             print(response.choices[0].message.content)
 
-
-    asyncio.run(llm_completion())
+    asyncio.run(main())
 
 

xgae/utils/setup_env.py

@@ -52,7 +52,7 @@ def setup_logging(log_file: str=None, log_level: str="INFO") :
 
     logger.setLevel(logging_level)
 
-    logging.info(f"📡 XGAE_LOGGING is initialized, log_level={log_level}, log_file={log_file}")
+    logging.info(f"🛠️ XGA_LOGGING is initialized, log_level={log_level}, log_file={log_file}")
 
 def setup_env_logging():
     log_enable = to_bool(os.getenv("LOG_ENABLE", True))
@@ -71,10 +71,10 @@ def setup_langfuse() -> Langfuse:
                             secret_key=env_secret_key,
                             host=env_host)
 
-        logging.info("📡 XGAE_LANGFUSE initialized Successfully by Key !")
+        logging.info("🛠️ XGA_LANGFUSE initialized Successfully by Key !")
     else:
         _langfuse = Langfuse(enabled=False)
-        logging.warning("📡 XGAE_LANGFUSE Not set key, Langfuse is disabled!")
+        logging.warning("🛠️ XGA_LANGFUSE Not set key, Langfuse is disabled!")
 
     return _langfuse
 

{xgae-0.1.9.dist-info → xgae-0.1.10.dist-info}/METADATA

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

xgae-0.1.10.dist-info/RECORD

@@ -0,0 +1,21 @@
+xgae/__init__.py,sha256=OEUd9y9AoGBd3xYerdTTpz9xl4NWkmXeq1a2eil7Qro,72
+xgae/cli_app.py,sha256=ePis7gYYZrevEArnCQOhaN7z4C8Y5yJSOIov8z-lGBs,3157
+xgae/engine/engine_base.py,sha256=ioywuTpDMHEmyVcd6BInoU-vR70PhQStE2MVRWoEiJg,1768
+xgae/engine/mcp_tool_box.py,sha256=ZSCBSXRWhISwyZ1uEIbt3esjesM46g-ktv6CxvyPVDU,10030
+xgae/engine/prompt_builder.py,sha256=X9bS7YIms6LYplCpNHeUmi74xFP5MwFXmXNqOt1Xz-Q,4356
+xgae/engine/task_engine.py,sha256=LAo55FKmmO8Jbo5geEUYr8kFCaVigTb-Jm06XuYKYyY,19010
+xgae/engine/task_langfuse.py,sha256=b0aJ_Di-WDcYzi0TFCvcKWxkBz7PYP2jx3N52OptQMs,2349
+xgae/engine/responser/non_stream_responser.py,sha256=9YCCUedbotH-TPPbTh2Mv1qNVYvznHYFPgAnQB7NJSE,6510
+xgae/engine/responser/responser_base.py,sha256=8PcsvQHP68FEhu6v3dT9hDCc_rLKs38i4txWLcJD4ck,29851
+xgae/engine/responser/stream_responser.py,sha256=oPGtrT1nedGMjiBAwPzUlu6Z_rPWeVSODC1xQ6D8cTY,52055
+xgae/tools/without_general_tools_app.py,sha256=FGMV6njcOKwwfitc0j_nUov0RC-eWlhO1IP8_KHz1tQ,3788
+xgae/utils/__init__.py,sha256=_-TTNq5FanrA-jl_w3-4xp-BnRM7SLwfYQcFyvepcW0,332
+xgae/utils/json_helpers.py,sha256=6BkqiyEF3jV3Irb4Z6-wGY2_FNaLlxE1WKlMJHHT6E0,4645
+xgae/utils/llm_client.py,sha256=hvEDb4DBaWVQTXMjXOd6KrFwJFBcI-YXEQD4f_AhG7Q,14008
+xgae/utils/misc.py,sha256=M8lMXYp1pHiY6Ee8ZTUG88GpOAsE5fbYoRO_hcBFUCE,953
+xgae/utils/setup_env.py,sha256=HweQ-WAyxfV3KYjGYi-rRQAbI_SXoimduOLpQPbHfl8,2619
+xgae/utils/xml_tool_parser.py,sha256=I9xAZC_ElwBY19PNUq-WLXe9FSIJMeAv2Xs-VlajI7Y,4782
+xgae-0.1.10.dist-info/METADATA,sha256=SGnhZrr3DDZ600FDMep9ihERmhwFspEtUKRv1THqQsk,310
+xgae-0.1.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+xgae-0.1.10.dist-info/entry_points.txt,sha256=vClvL_WBJyF2x3wJCz5CNJ_BJG-dWUh7h2YbAoskHsc,162
+xgae-0.1.10.dist-info/RECORD,,

{xgae-0.1.9.dist-info → xgae-0.1.10.dist-info}/entry_points.txt

@@ -1,3 +1,4 @@
 [console_scripts]
 custom_fault_tools = examples.tools.custom_fault_tools_app:main
+xgae = xgae.cli_app:main
 xgae-tools = xgae.tools.without_general_tools_app:main

xgae-0.1.9.dist-info/RECORD

@@ -1,20 +0,0 @@
-xgae/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xgae/engine/engine_base.py,sha256=ioywuTpDMHEmyVcd6BInoU-vR70PhQStE2MVRWoEiJg,1768
-xgae/engine/mcp_tool_box.py,sha256=ZSCBSXRWhISwyZ1uEIbt3esjesM46g-ktv6CxvyPVDU,10030
-xgae/engine/prompt_builder.py,sha256=X9bS7YIms6LYplCpNHeUmi74xFP5MwFXmXNqOt1Xz-Q,4356
-xgae/engine/task_engine.py,sha256=BLy32wJUbBFGkjAPfHcbSpCUPQm2uZAmLHtZVqvXwx0,18026
-xgae/engine/task_langfuse.py,sha256=-WMnlMcBGfDuJpV0ZYKnVzlO0HizGOfFVGyzVkP6Rag,2260
-xgae/engine/responser/non_stream_responser.py,sha256=80ZFcitjXpQJANxbYpQ0LlhVzIxmURNJi9Fbo953Uak,6349
-xgae/engine/responser/responser_base.py,sha256=8PcsvQHP68FEhu6v3dT9hDCc_rLKs38i4txWLcJD4ck,29851
-xgae/engine/responser/stream_responser.py,sha256=oPGtrT1nedGMjiBAwPzUlu6Z_rPWeVSODC1xQ6D8cTY,52055
-xgae/tools/without_general_tools_app.py,sha256=FGMV6njcOKwwfitc0j_nUov0RC-eWlhO1IP8_KHz1tQ,3788
-xgae/utils/__init__.py,sha256=tMSYguPrubusr7IyKZn6iPYE6qbN7sOsEVDeJPRdPa8,351
-xgae/utils/json_helpers.py,sha256=6BkqiyEF3jV3Irb4Z6-wGY2_FNaLlxE1WKlMJHHT6E0,4645
-xgae/utils/llm_client.py,sha256=nOhPlllUj4mASdj2dU5HkSmDqYZ8ZMpDa7AKQwUWJnQ,13600
-xgae/utils/misc.py,sha256=M8lMXYp1pHiY6Ee8ZTUG88GpOAsE5fbYoRO_hcBFUCE,953
-xgae/utils/setup_env.py,sha256=XD6WmGNyMFDKVzNU_ufYtqJNd2Otq4hzLs9YZ2aSz6g,2613
-xgae/utils/xml_tool_parser.py,sha256=I9xAZC_ElwBY19PNUq-WLXe9FSIJMeAv2Xs-VlajI7Y,4782
-xgae-0.1.9.dist-info/METADATA,sha256=q4_B-_Kt2hswi0kGX9B_H3uI_7C7SWiJOo6HnzoMF8g,309
-xgae-0.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-xgae-0.1.9.dist-info/entry_points.txt,sha256=bYzQ-1iabCduosArMzq1hktJrCjp8jq4_E8Sg8-gEic,137
-xgae-0.1.9.dist-info/RECORD,,