diff --git a/examples/openai/prompting.py b/examples/openai/prompting.py index c431a7987..173810139 100644 --- a/examples/openai/prompting.py +++ b/examples/openai/prompting.py @@ -162,6 +162,10 @@ class ChatTemplate(BaseModel): sys.stderr.write(f'Unexpected message role: {message.role}\n') current_content.append(f'[ROLE={messages[i].role}]{messages[i].content}[/ROLE]') + current_content.extend( + f'{json.dumps(tc.model_dump())}' + for tc in (message.tool_calls or []) + ) if current_content: flush()