fix: update args schema

Author: rushilpatel0

src/codegen/extensions/langchain/tools.py

@@ -1041,6 +1041,7 @@ class RelaceEditInput(BaseModel):
 
     filepath: str = Field(..., description="Path of the file relative to workspace root")
     edit_snippet: str = Field(..., description=RELACE_EDIT_PROMPT)
+    tool_call_id: Annotated[str, InjectedToolCallId]
 
 
 class RelaceEditTool(BaseTool):

src/codegen/extensions/tools/relace_edit.py

@@ -60,6 +60,7 @@ def render(self, tool_call_id: str) -> ToolMessage:
         return ToolMessage(
             content=self.render_as_string(),
             status=self.status,
+            name="relace_edit",
             tool_call_id=tool_call_id,
             artifact=artifacts,
         )