fix: test relace artifacts

Author: rushilpatel0

src/codegen/extensions/langchain/tools.py

@@ -1054,9 +1054,9 @@ class RelaceEditTool(BaseTool):
     def __init__(self, codebase: Codebase) -> None:
         super().__init__(codebase=codebase)
 
-    def _run(self, filepath: str, edit_snippet: str) -> str:
+    def _run(self, filepath: str, edit_snippet: str, tool_call_id: str) -> ToolMessage:
         result = relace_edit(self.codebase, filepath, edit_snippet)
-        return result.render()
+        return result.render(tool_call_id=tool_call_id)
 
 
 class ReflectionInput(BaseModel):

src/codegen/extensions/tools/relace_edit.py

@@ -7,10 +7,12 @@
 import requests
 from pydantic import Field
 
+from codegen.extensions.tools.tool_output_types import RelaceEditArtifacts
 from codegen.sdk.core.codebase import Codebase
 
 from .observation import Observation
 from .view_file import add_line_numbers
+from langchain_core.messages import ToolMessage
 
 
 class RelaceEditObservation(Observation):
@@ -35,6 +37,34 @@ class RelaceEditObservation(Observation):
     str_template: ClassVar[str] = "Edited file {filepath} using Relace Instant Apply"
 
 
+    def render(self, tool_call_id: str) -> ToolMessage:
+        """Render the relace edit observation as a ToolMessage."""
+
+        artifacts: RelaceEditArtifacts = {
+            "filepath": self.filepath,
+            "diff": self.diff,
+            "new_content": self.new_content,
+            "line_count": self.line_count,
+            "error": self.error,
+        }
+
+        if self.status == "error":
+            return ToolMessage(
+                content=f"[ERROR EDITING FILE]: {self.filepath}: {self.error}",
+                status=self.status,
+                name="relace_edit",
+                artifact=artifacts,
+                tool_call_id=tool_call_id,
+            )
+
+        return ToolMessage(
+            content=self.render_as_string(),
+            status=self.status,
+            tool_call_id=tool_call_id,
+            artifact=artifacts,
+        )
+
+
 def generate_diff(original: str, modified: str) -> str:
     """Generate a unified diff between two strings.
 

src/codegen/extensions/tools/tool_output_types.py

@@ -89,3 +89,17 @@ class SemanticEditArtifacts(TypedDict, total=False):
     new_content: Optional[str]  # New content of the file after edits
     line_count: Optional[int]  # Total number of lines in the edited file
     error: Optional[str]  # Error message (only present on error)
+
+
+class RelaceEditArtifacts(TypedDict, total=False):
+    """Artifacts for relace edit operations.
+
+    All fields are optional to support both success and error cases.
+    Includes metadata useful for UI diff view and file content.
+    """
+
+    filepath: str  # Path to the edited file
+    diff: Optional[str]  # Unified diff of changes made to the file
+    new_content: Optional[str]  # New content of the file after edits
+    line_count: Optional[int]  # Total number of lines in the edited file
+    error: Optional[str]  # Error message (only present on error)