Skip to content

Commit 7e64bdb

Browse files
committed
fix(chatwithtools): Expect a list of tool messages
Signed-off-by: Marcel Klehr <[email protected]>
1 parent b27d8c2 commit 7e64bdb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

lib/chatwithtools.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,9 @@ def __call__(
6969
messages.append(HumanMessage(content=inputs['input']))
7070

7171
try:
72-
tool_message = json.loads(inputs['tool_message'])
73-
messages.append(ToolMessage(content=tool_message['content'], name=tool_message['name'], tool_call_id='42'))
72+
tool_messages = json.loads(inputs['tool_message'])
73+
for tool_message in tool_messages:
74+
messages.append(ToolMessage(content=tool_message['content'], name=tool_message['name'], tool_call_id='42'))
7475
except:
7576
pass
7677

0 commit comments

Comments
 (0)