feature: 提升codeexcute的执行稳定性

Author: raychen911

trpc_agent_sdk/agents/_llm_agent.py

@@ -517,20 +517,31 @@ def accumulate_content(event: Event) -> None:
                         if event.is_final_response():
                             self._save_output_to_state(ctx, event)
 
-                        # Process code execution responses if code executor is configured
+                        # Process code execution responses if code executor is configured.
+                        # We collect code execution events first (this mutates event.content in place,
+                        # stripping executable_code parts but keeping text/function_call), then yield
+                        # the main event BEFORE the code execution events so the causal order in
+                        # session is preserved: assistant declaration → code execution → result.
+                        pending_code_events: list[Event] = []
                         if self.code_executor and event.content:
                             async for code_event in CodeExecutionResponseProcessor.run_async(ctx, event):
-                                # Check if this is a code execution result event
                                 if code_event.content and code_event.content.parts:
                                     for part in code_event.content.parts:
                                         if part.code_execution_result or part.executable_code:
                                             code_was_executed = True
                                             break
-                                yield code_event
-
-                        # Yield LLM response events directly
-                        yield event
-                        accumulate_content(event)
+                                pending_code_events.append(code_event)
+
+                        # Yield the main LLM response event first (now stripped of executable_code
+                        # but still carrying text and function_call parts).
+                        # Skip empty events (content became None after all parts were consumed).
+                        if event.content is not None:
+                            yield event
+                            accumulate_content(event)
+
+                        # Then yield code execution events in order.
+                        for code_event in pending_code_events:
+                            yield code_event
                     else:
                         # Yield other events directly
                         yield event

trpc_agent_sdk/agents/core/_code_execution_processor.py

@@ -249,10 +249,6 @@ async def _run_post_processor(
         return
 
     code_executor_context = CodeExecutorContext(invocation_context.session.state)
-    if (code_executor.execute_once_per_invocation
-            and code_executor_context.has_executed_in_invocation(invocation_context.invocation_id)):
-        return
-
     # Skip if the error count exceeds the max retry attempts.
     if code_executor_context.get_error_count(invocation_context.invocation_id) >= code_executor.error_retry_attempts:
         return
@@ -285,7 +281,6 @@ async def _run_post_processor(
         code_blocks,
         code_execution_result,
     )
-    code_executor_context.mark_executed_in_invocation(invocation_context.invocation_id)
 
     # Generate events for code execution results
     # Event 1: Code execution event

trpc_agent_sdk/code_executors/_base_code_executor.py

@@ -59,14 +59,6 @@ class BaseCodeExecutor(BaseModel):
     error_retry_attempts: int = 2
     """The number of attempts to retry on consecutive code execution errors. Default to 2."""
 
-    execute_once_per_invocation: bool = False
-    """Whether to execute model-extracted code at most once per invocation.
-
-    When enabled, post-processing code execution runs only for the first
-    detected code block in a single ``invocation_id`` and skips subsequent
-    auto-execution attempts for that invocation.
-    """
-
     code_block_delimiters: list[CodeBlockDelimiter] = [
         CodeBlockDelimiter(start="```tool_code\n", end="\n```"),
         CodeBlockDelimiter(start="```python\n", end="\n```"),

trpc_agent_sdk/code_executors/_code_executor_context.py

@@ -39,7 +39,6 @@ def _ensure_code_execution_state(self) -> None:
                 "execution_id": None,
                 "error_counts": {},
                 "code_execution_results": {},
-                "executed_invocations": {},
             }
 
     def get_input_files(self) -> List[CodeFile]:
@@ -139,14 +138,6 @@ def update_code_execution_result(self, invocation_id: str, code_blocks: List[Cod
             code_execution_result.model_dump(),
         })
 
-    def has_executed_in_invocation(self, invocation_id: str) -> bool:
-        """Whether code has already been executed in a given invocation."""
-        return bool(self.session_state["code_execution"]["executed_invocations"].get(invocation_id, False))
-
-    def mark_executed_in_invocation(self, invocation_id: str) -> None:
-        """Mark that code execution has happened in a given invocation."""
-        self.session_state["code_execution"]["executed_invocations"][invocation_id] = True
-
     def get_state_delta(self) -> Dict:
         """Get state delta for the current execution.