@@ -62,6 +62,8 @@
     from .codeexecutionresultstep import (
         CodeExecutionResultStep,
         CodeExecutionResultStepParam,
+        CodeExecutionResultStepResult,
+        CodeExecutionResultStepResultParam,
     )
     from .codemenderagentconfig import CodeMenderAgentConfig, CodeMenderAgentConfigParam
     from .computeruse import (
@@ -481,6 +483,8 @@
     "CodeExecutionResultDeltaTypedDict",
     "CodeExecutionResultStep",
     "CodeExecutionResultStepParam",
+    "CodeExecutionResultStepResult",
+    "CodeExecutionResultStepResultParam",
     "CodeMenderAgentConfig",
     "CodeMenderAgentConfigParam",
     "ComputerUse",
@@ -892,6 +896,8 @@
     "CodeExecutionResultDeltaTypedDict": ".codeexecutionresultdelta",
     "CodeExecutionResultStep": ".codeexecutionresultstep",
     "CodeExecutionResultStepParam": ".codeexecutionresultstep",
+    "CodeExecutionResultStepResult": ".codeexecutionresultstep",
+    "CodeExecutionResultStepResultParam": ".codeexecutionresultstep",
     "CodeMenderAgentConfig": ".codemenderagentconfig",
     "CodeMenderAgentConfigParam": ".codemenderagentconfig",
     "ComputerUse": ".computeruse",
 
@@ -32,20 +32,33 @@
     current_filtered_dump,
     current_serialization_methods_dump,
 )
+from .content import Content, ContentParam
 import pydantic
 from pydantic import model_serializer
 from pydantic.functional_validators import AfterValidator
-from typing import Any, Literal, Optional, Union
-from typing_extensions import Annotated, NotRequired, TypedDict
+from typing import Any, List, Literal, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+
+
+CodeExecutionResultStepResultParam = TypeAliasType(
+    "CodeExecutionResultStepResultParam", Union[List[ContentParam], str]
+)
+r"""The output of the code execution."""
+
+
+CodeExecutionResultStepResult = TypeAliasType(
+    "CodeExecutionResultStepResult", Union[List[Content], str]
+)
+r"""The output of the code execution."""
 
 
 class CodeExecutionResultStepParam(TypedDict):
     r"""Code execution result step."""
 
     call_id: str
     r"""Required. ID to match the ID from the function call block."""
-    result: str
-    r"""Required. The output of the code execution."""
+    result: CodeExecutionResultStepResultParam
+    r"""The output of the code execution."""
     is_error: NotRequired[bool]
     r"""Whether the code execution resulted in an error."""
     signature: NotRequired[Union[str, Base64FileInput]]
@@ -59,8 +72,8 @@ class CodeExecutionResultStep(BaseModel):
     call_id: str
     r"""Required. ID to match the ID from the function call block."""
 
-    result: str
-    r"""Required. The output of the code execution."""
+    result: CodeExecutionResultStepResult
+    r"""The output of the code execution."""
 
     is_error: Optional[bool] = None
     r"""Whether the code execution resulted in an error."""