xgae 0.1.4__tar.gz → 0.1.6__tar.gz

{xgae-0.1.4 → xgae-0.1.6}/.idea/misc.xml

@@ -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 ~/DevProjects/xga/xgae/.venv" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.13 virtualenv at ~/DevelopSpace/xgae/.venv" project-jdk-type="Python SDK" />
 </project>

{xgae-0.1.4 → xgae-0.1.6}/.idea/workspace.xml

@@ -25,27 +25,26 @@
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <component name="PropertiesComponent"><![CDATA[{
-  "keyToString": {
-    "ModuleVcsDetector.initialDetectionPerformed": "true",
-    "Python.llm_client.executor": "Run",
-    "Python.run_xga_engine.executor": "Run",
-    "Python.setup_env.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 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>
+}</component>
   <component name="RecentsManager">
     <key name="MoveFile.RECENT_KEYS">
       <recent name="$PROJECT_DIR$/src/xgae/engine/responser" />
@@ -61,7 +60,6 @@
         <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" />
@@ -86,7 +84,7 @@
       <option name="INPUT_FILE" value="" />
       <method v="2" />
     </configuration>
-    <configuration name="run_xga_engine" type="PythonConfigurationType" factoryName="Python">
+    <configuration name="run_xga_engine" type="PythonConfigurationType" factoryName="Python" temporary="true">
       <module name="xgae" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -110,7 +108,7 @@
           <ENTRY IS_ENABLED="true" PARSER="runconfig" IS_EXECUTABLE="false" />
         </ENTRIES>
       </EXTENSION>
-      <option name="SCRIPT_NAME" value="$PROJECT_DIR$/src/xgae/engine/xga_engine.py" />
+      <option name="SCRIPT_NAME" value="$PROJECT_DIR$/src/xgae/engine/task_engine.py" />
       <option name="PARAMETERS" value="" />
       <option name="SHOW_COMMAND_LINE" value="false" />
       <option name="EMULATE_TERMINAL" value="false" />
@@ -119,6 +117,11 @@
       <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>
@@ -142,40 +145,46 @@
       <workItem from="1755270285722" duration="12068000" />
       <workItem from="1755283728206" duration="1511000" />
       <workItem from="1755286552901" duration="3130000" />
-      <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="24716000" />
+      <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" />
     </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_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$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$run_xga_engine.coverage" NAME="run_xga_engine Coverage Results" MODIFIED="1755768463510" 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="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$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="1755657717310" 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="1755565705235" 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.4 → xgae-0.1.6}/.idea/xgae.iml

@@ -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 ~/DevProjects/xga/xgae/.venv" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="Python 3.13 virtualenv at ~/DevelopSpace/xgae/.venv" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
 </module>

{xgae-0.1.4 → xgae-0.1.6}/PKG-INFO

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

{xgae-0.1.4 → xgae-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "xgae"
-version = "0.1.4"
+version = "0.1.6"
 description = "Extreme General Agent Engine"
 readme = "README.md"
 requires-python = ">=3.13"

xgae-0.1.4/src/xgae/engine/xga_base.py → xgae-0.1.6/src/xgae/engine/engine_base.py

@@ -6,24 +6,20 @@ class XGAError(Exception):
     """Custom exception for errors in the XGA system."""
     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"]
 
-class XGAContextMsg(TypedDict, total=False):
-    type: Literal["user", "status",  "tool", "assistant", "assistant_response_end"]
-    content: Union[Dict[str, Any], List[Any], str]
-    is_llm_message: bool
-    metadata: Dict[str, Any]
+class XGAResponseMessage(TypedDict, total=False):
     message_id: str
-    session_id: str
-    agent_id: str
-    task_id: str
-    task_run_id: str
-    trace_id: str
-
-class XGAResponseMsg(TypedDict, total=False):
-    type: Literal["content", "status"]
+    type: XGAResponseMsgType
+    is_llm_message: bool
     content: Union[Dict[str, Any], List[Any], str]
-    status: Literal["error", "status"]
-    message: str
+    metadata: Dict[str, Any]
+
+class XGATaskResult(TypedDict, total=False):
+    type: Literal["ask", "answer", "error"]
+    content: str
+    attachments: Optional[List[str]]
 
 @dataclass
 class XGAToolSchema:
@@ -39,6 +35,7 @@ class XGAToolResult:
     success: bool
     output: str
 
+
 class XGAToolBox(ABC):
     @abstractmethod
     async def creat_task_tool_box(self, task_id: str, general_tools: List[str], custom_tools: List[str]):

xgae-0.1.4/src/xgae/engine/xga_mcp_tool_box.py → xgae-0.1.6/src/xgae/engine/mcp_tool_box.py

@@ -7,8 +7,8 @@ from typing import List, Any, Dict, Optional, Literal, override
 from langchain_mcp_adapters.client import MultiServerMCPClient
 from langchain_mcp_adapters.tools import load_mcp_tools
 
-from xgae.engine.xga_base import XGAError, XGAToolSchema, XGAToolBox, XGAToolResult
-from xgae.utils.setup_env import langfuse
+from xgae.engine.engine_base import XGAError, XGAToolSchema, XGAToolBox, XGAToolResult
+from xgae.utils import langfuse
 
 class XGAMcpToolBox(XGAToolBox):
     GENERAL_MCP_SERVER_NAME = "xga_general"
@@ -38,7 +38,7 @@ class XGAMcpToolBox(XGAToolBox):
     async def creat_task_tool_box(self, task_id: str, general_tools: List[str], custom_tools: List[str]):
         task_tool_schemas = {}
         general_tool_schemas = self.mcp_tool_schemas.get(XGAMcpToolBox.GENERAL_MCP_SERVER_NAME, {})
-        if len(general_tools) > 0 and general_tools[0] == "*":
+        if "*" in general_tools:
             task_tool_schemas = {tool_schema.tool_name: tool_schema for tool_schema in general_tool_schemas}
         else:
             for tool_schema in general_tool_schemas:
@@ -58,8 +58,8 @@ class XGAMcpToolBox(XGAToolBox):
             if custom_tool_schemas is None:
                 continue
             if custom_tool_name == "*":
-                custom_tool_schema_d = {tool_schema.tool_name: tool_schema for tool_schema in custom_tool_schemas}
-                task_tool_schemas.update(custom_tool_schema_d)
+                custom_tool_schema_dict = {tool_schema.tool_name: tool_schema for tool_schema in custom_tool_schemas}
+                task_tool_schemas.update(custom_tool_schema_dict)
             else:
                 for tool_schema in custom_tool_schemas:
                     if custom_tool_name == tool_schema.tool_name:
@@ -112,10 +112,7 @@ class XGAMcpToolBox(XGAToolBox):
             if mcp_tool:
                 tool_args = args or {}
                 if server_name == self.GENERAL_MCP_SERVER_NAME:
-                    pass
-                    #tool_args["task_id"] = task_id #xga general tool, first param must be task_id
-                else:
-                    tool_args = args
+                    tool_args = dict({"task_id": task_id}, **tool_args)
 
                 try:
                     tool_result = await mcp_tool.arun(tool_args)

xgae-0.1.4/src/xgae/engine/xga_prompt_builder.py → xgae-0.1.6/src/xgae/engine/prompt_builder.py

@@ -1,9 +1,10 @@
 import json
 import datetime
+
 from typing import Optional, List
 
-from xga_base import XGAToolSchema, XGAError
-from xgae.utils.utils import read_file, format_file_with_args
+from engine_base import XGAToolSchema, XGAError
+from xgae.utils.misc import read_file, format_file_with_args
 
 
 class XGAPromptBuilder():

xgae-0.1.6/src/xgae/engine/responser/non_stream_responser.py

@@ -0,0 +1,110 @@
+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):
+    def __init__(self, response_context: TaskResponserContext):
+        super().__init__(response_context)
+
+    @override
+    async def process_response(self,llm_response: Any,prompt_messages: List[Dict[str, Any]],
+                               continuous_state: Optional[TaskRunContinuousState] = None) -> AsyncGenerator[Dict[str, Any], None]:
+        llm_content = ""
+        parsed_xml_data = []
+        finish_reason = None
+
+        try:
+            # Extract finish_reason, content, tool calls
+            if hasattr(llm_response, 'choices') and llm_response.choices:
+                if hasattr(llm_response.choices[0], 'finish_reason'):
+                    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}"))
+
+                response_message = llm_response.choices[0].message if hasattr(llm_response.choices[0], 'message') else None
+                if response_message:
+                    if hasattr(response_message, 'content') and response_message.content:
+                        llm_content = response_message.content
+                        
+                        parsed_xml_data = self._parse_xml_tool_calls(llm_content)
+                        if self.max_xml_tool_calls > 0 and len(parsed_xml_data) > self.max_xml_tool_calls:
+                            logging.warning(f"NonStreamTask：Truncate content, parsed_xml_data length {len(parsed_xml_data)} limit over max_xml_tool_calls={self.max_xml_tool_calls}")
+                            xml_chunks = self._extract_xml_chunks(llm_content)[:self.max_xml_tool_calls]
+                            if xml_chunks:
+                                last_chunk = xml_chunks[-1]
+                                last_chunk_pos = llm_content.find(last_chunk)
+                                if last_chunk_pos >= 0:
+                                    llm_content = llm_content[:last_chunk_pos + len(last_chunk)]
+                            parsed_xml_data = parsed_xml_data[:self.max_xml_tool_calls]
+                            finish_reason = "xml_tool_limit_reached"
+                            logging.info(f"NonStreamTask：LLM response finish_reason={finish_reason}")
+                else:
+                    logging.warning(f"NonStreamTask：LLM response_message is empty")
+
+            message_data = {"role": "assistant", "content": llm_content, "index": -1} # index=-1, full llm_content
+            assistant_msg = self.add_response_message(type="assistant", content=message_data, is_llm_message=True)
+            yield assistant_msg
+
+            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)
+
+                tool_index = 0
+                for i, (returned_tool_call, tool_result) in enumerate(tool_results):
+                    parsed_xml_item = parsed_xml_data[i]
+                    tool_call = parsed_xml_item['tool_call']
+                    parsing_details = parsed_xml_item['parsing_details']
+                    assistant_msg_id = assistant_msg['message_id'] if assistant_msg else None
+
+                    tool_context = self._create_tool_context(tool_call, tool_index, assistant_msg_id, parsing_details)
+                    tool_context.result = tool_result
+
+                    tool_start_msg = self._add_tool_start_message(tool_context)
+                    yield format_for_yield(tool_start_msg)
+
+                    tool_message = self._add_tool_messsage(tool_call, tool_result, self.xml_adding_strategy, assistant_msg_id, parsing_details)
+
+                    tool_completed_msg = self._add_tool_completed_message(tool_context, tool_message['message_id'])
+                    yield format_for_yield(tool_completed_msg)
+
+                    yield format_for_yield(tool_message)
+
+                    if tool_completed_msg["metadata"].get("agent_should_terminate") == "true":
+                        finish_reason = "completed"
+                        break
+                    tool_index += 1
+
+            if finish_reason:
+                finish_content = {"status_type": "finish", "finish_reason": finish_reason}
+                finish_msg_obj = self.add_response_message(type="status", content=finish_content, is_llm_message=False)
+                if finish_msg_obj:
+                    yield format_for_yield(finish_msg_obj)
+
+        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",
+                             status_message=(f"Error processing non-streaming response: {str(e)}"))
+
+            content = {"role": "system", "status_type": "error", "message": str(e)}
+            err_msg = self.add_response_message(ype="status", content=content,is_llm_message=False)
+            if err_msg:
+                yield format_for_yield(err_msg)
+
+            # 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",
+                             status_message=(f"Re-raising error to stop further processing: {str(e)}"))
+            raise  # Use bare 'raise' to preserve the original exception with its traceback
+
+
+