Refactor code

Author: leeyyi

trpc_agent_sdk/code_executors/_base_code_executor.py

@@ -16,13 +16,19 @@
 from typing import Optional
 
 from pydantic import BaseModel
+from pydantic import Field
 from trpc_agent_sdk.context import InvocationContext
 
 from ._base_workspace_runtime import BaseWorkspaceRuntime
 from ._types import CodeBlockDelimiter
 from ._types import CodeExecutionInput
 from ._types import CodeExecutionResult
 
+DEFAULT_CODE_BLOCK_DELIMITERS: tuple[CodeBlockDelimiter, ...] = (
+    CodeBlockDelimiter(start="```tool_code\n", end="\n```"),
+    CodeBlockDelimiter(start="```python\n", end="\n```"),
+)
+
 
 class BaseCodeExecutor(BaseModel):
     """Abstract base class for all code executors.
@@ -59,10 +65,15 @@ class BaseCodeExecutor(BaseModel):
     error_retry_attempts: int = 2
     """The number of attempts to retry on consecutive code execution errors. Default to 2."""
 
-    code_block_delimiters: list[CodeBlockDelimiter] = [
-        CodeBlockDelimiter(start="```tool_code\n", end="\n```"),
-        CodeBlockDelimiter(start="```python\n", end="\n```"),
-    ]
+    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] = Field(default_factory=lambda: list(DEFAULT_CODE_BLOCK_DELIMITERS))
     """The list of the enclosing delimiters to identify the code blocks.
 
     For example, the delimiter ('```python\\n', '\\n```') can be

trpc_agent_sdk/code_executors/cube/_code_executor.py

@@ -21,6 +21,7 @@
 from trpc_agent_sdk.context import InvocationContext
 
 from .._base_code_executor import BaseCodeExecutor
+from .._base_code_executor import DEFAULT_CODE_BLOCK_DELIMITERS
 from .._types import CodeBlock
 from .._types import CodeBlockDelimiter
 from .._types import CodeExecutionInput
@@ -36,21 +37,6 @@
 _BASH_DELIMITER = CodeBlockDelimiter(start="```bash\n", end="\n```")
 
 
-def _cube_default_code_block_delimiters() -> list[CodeBlockDelimiter]:
-    """Default delimiters for :class:`CubeCodeExecutor`.
-
-    Reuses :attr:`BaseCodeExecutor.code_block_delimiters` so the parent's
-    defaults (currently ``tool_code`` + ``python``) stay the single
-    source of truth, then appends a ``bash`` fence so text-path
-    extraction matches what :meth:`CubeCodeExecutor.execute_code` can
-    actually run (see :meth:`_select_interpreter`). Returns a fresh list
-    on every call to keep per-instance defaults independent of one
-    another.
-    """
-    parent_default = BaseCodeExecutor.model_fields["code_block_delimiters"].default
-    return [*parent_default, _BASH_DELIMITER]
-
-
 class CubeCodeExecutor(BaseCodeExecutor):
     """A code executor that runs blocks inside a Cube/E2B remote sandbox.
 
@@ -84,13 +70,8 @@ class CubeCodeExecutor(BaseCodeExecutor):
     stateful: bool = Field(default=False, frozen=True, exclude=True)
     optimize_data_file: bool = Field(default=False, frozen=True, exclude=True)
 
-    # Extend the inherited default with a ``bash`` fence so text-path
-    # extraction matches what ``execute_code`` can actually run (see
-    # ``_select_interpreter``). The factory reads the parent's default
-    # at call time so the base list stays the single source of truth;
-    # callers may still override via the ``code_block_delimiters`` field
-    # at construction time.
-    code_block_delimiters: list[CodeBlockDelimiter] = Field(default_factory=_cube_default_code_block_delimiters, )
+    code_block_delimiters: list[CodeBlockDelimiter] = Field(
+        default_factory=lambda: [*DEFAULT_CODE_BLOCK_DELIMITERS, _BASH_DELIMITER])
 
     # `_client` is `Optional` because :meth:`close` / :meth:`destroy`
     # legitimately drop the handle post-construction. `_cfg` has no such