feat(agent): Improve history format for code flow execution results

Significant-Gravitas · Jul 23, 2024 · 7e0b115 · 7e0b115
1 parent da9360f
commit 7e0b115
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/forge/forge/components/action_history/action_history.py b/forge/forge/components/action_history/action_history.py
@@ -102,6 +102,8 @@ async def after_execute(self, result: ActionResult) -> None:
 
     @staticmethod
     def _make_result_message(episode: Episode, result: ActionResult) -> ChatMessage:
+        from forge.components.code_flow_executor import CodeFlowExecutionComponent
+
         if result.status == "success":
             return (
                 ToolResultMessage(
@@ -110,11 +112,18 @@ def _make_result_message(episode: Episode, result: ActionResult) -> ChatMessage:
                 )
                 if episode.action.raw_message.tool_calls
                 else ChatMessage.user(
-                    f"{episode.action.use_tool.name} returned: "
+                    (
+                        "Execution result:"
+                        if (
+                            episode.action.use_tool.name
+                            == CodeFlowExecutionComponent.execute_code_flow.name
+                        )
+                        else f"{episode.action.use_tool.name} returned:"
+                    )
                     + (
-                        f"```\n{result.outputs}\n```"
+                        f"\n```\n{result.outputs}\n```"
                         if "\n" in str(result.outputs)
-                        else f"`{result.outputs}`"
+                        else f" `{result.outputs}`"
                     )
                 )
             )

diff --git a/forge/forge/components/code_flow_executor/code_flow_executor.py b/forge/forge/components/code_flow_executor/code_flow_executor.py
@@ -58,7 +58,7 @@ async def execute_code_flow(self, python_code: str, plan_text: str) -> str:
         # limit the result to limit the characters
         if len(result) > MAX_RESULT_LENGTH:
             result = result[:MAX_RESULT_LENGTH] + "...[Truncated, Content is too long]"
-        return f"Execution Plan:\n{plan_text}\n\nExecution Output:\n{result}"
+        return result
 
     def _get_available_functions(self) -> dict[str, Callable]:
         return {