ensure ToolCalled retains custom render

rgbkrk · Mar 7, 2024 · f038bc0 · f038bc0
1 parent 9a4722a
commit f038bc0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/chatlab/views/tools.py b/chatlab/views/tools.py
@@ -110,7 +110,8 @@ def append_arguments(self, arguments: str):
     def apply_result(self, result: str):
         """Replaces the existing display with a new one that shows the result of the tool being called."""
         tc = ToolCalled(
-            id=self.id, name=self.name, arguments=self.arguments, result=result, display_id=self.display_id
+            id=self.id, name=self.name, arguments=self.arguments, result=result, display_id=self.display_id,
+            custom_render=self.custom_render
         )
         tc.update()
         return tc