jedick commited on
Commit
cb0db2c
·
1 Parent(s): bcab0eb

Omit "Tool Call" in formatted message content

Browse files
Files changed (1) hide show
  1. graph.py +4 -3
graph.py CHANGED
@@ -65,9 +65,10 @@ def normalize_messages(messages):
65
  tool_messages = []
66
  count = 1
67
  while i < len(messages) and type(messages[i]) is ToolMessage:
68
- tool_msg = messages[i]
69
- formatted_content = f"## Tool Call {count}\n\n{tool_msg.content}"
70
- tool_messages.append(formatted_content)
 
71
  count += 1
72
  i += 1
73
 
 
65
  tool_messages = []
66
  count = 1
67
  while i < len(messages) and type(messages[i]) is ToolMessage:
68
+ #tool_msg = messages[i]
69
+ #formatted_content = f"## Tool Call {count}\n\n{tool_msg.content}"
70
+ #tool_messages.append(formatted_content)
71
+ tool_messages.append(messages[i])
72
  count += 1
73
  i += 1
74