Update src/txagent/utils.py

src/txagent/utils.py

@@ -3,7 +3,7 @@ import json
 import hashlib
 import torch
 from typing import List
-from gradio import ChatMessage  # Ensure this is present
+
 
 def get_md5(input_str):
     # Create an MD5 hash object
@@ -13,35 +13,18 @@ def get_md5(input_str):
 
 
 def tool_result_format(function_call_messages):
-    """
-    Format tool outputs as a list of ChatMessage objects with metadata
-    so the UI can display tool names and details cleanly.
-    """
-    formatted_messages = []
-
+    current_output = "\n\n<details>\n<summary> <strong>Verified Feedback from Tools</strong>, click to see details:</summary>\n\n"
     for each_message in function_call_messages:
         if each_message['role'] == 'tool':
             try:
-                data = json.loads(each_message['content'])
-                tool_name = data.get("tool_name", "Tool Result")
-                tool_output = data.get("content", "")
-                log = data if isinstance(data, dict) else {}
-            except Exception as e:
-                # Handle malformed JSON
-                tool_name = "Tool Result"
-                tool_output = str(each_message['content'])
-                log = {"error": "Malformed tool output", "raw": tool_output}
-
-            formatted_messages.append(ChatMessage(
-                role="assistant",
-                content=tool_output,
-                metadata={
-                    "title": f"⚒️ {tool_name}",
-                    "log": json.dumps(log, indent=2)
-                }
-            ))
-
-    return formatted_messages
+                parsed = json.loads(each_message['content'])
+                tool_name = parsed.get("tool_name", "Unknown Tool")
+                tool_output = parsed.get("content", each_message['content'])
+                current_output += f"**🔧 Tool: {tool_name}**\n\n{tool_output}\n\n"
+            except Exception:
+                current_output += f"{each_message['content']}\n\n"
+    current_output += "</details>\n\n\n"
+    return current_output
 
 
 class NoRepeatSentenceProcessor:
@@ -80,15 +63,12 @@ class ReasoningTraceChecker:
             each = self.conversation[i]
             self.index = i
             if each['role'] == 'assistant':
-                print(each)
                 thought = each['content']
                 actions = each['tool_calls']
-
                 good_status, current_info = self.check_repeat_thought(thought)
                 info += current_info
                 if not good_status:
                     return False, info
-
                 good_status, current_info = self.check_repeat_action(actions)
                 info += current_info
                 if not good_status:
@@ -107,8 +87,8 @@ class ReasoningTraceChecker:
         for each_action in actions:
             if 'call_id' in each_action:
                 del each_action['call_id']
-            each_action_str = json.dumps(each_action)
-            if each_action_str in self.existing_actions:
+            each_action = json.dumps(each_action)
+            if each_action in self.existing_actions:
                 return False, "repeat_action"
-            self.existing_actions.append(each_action_str)
+            self.existing_actions.append(each_action)
         return True, ''