fix tongyi tool calls (#6896)

This commit is contained in:
dufei 2024-08-02 10:03:43 +08:00 committed by GitHub
parent 2ab9af3b38
commit f8617db012
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -497,12 +497,13 @@ You should also complete the text started with ``` but not tell ``` directly.
content = prompt_message.content
if not content:
content = ' '
tongyi_messages.append({
message = {
'role': 'assistant',
'content': content if not rich_content else [{"text": content}],
'tool_calls': [tool_call.model_dump() for tool_call in
prompt_message.tool_calls] if prompt_message.tool_calls else None
})
'content': content if not rich_content else [{"text": content}]
}
if prompt_message.tool_calls:
message['tool_calls'] = [tool_call.model_dump() for tool_call in prompt_message.tool_calls]
tongyi_messages.append(message)
elif isinstance(prompt_message, ToolPromptMessage):
tongyi_messages.append({
"role": "tool",