ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix api

Browse Source 
Former-commit-id: dcbd6d86dfc49f12529b02ec331e3e5c05740061
This commit is contained in:
hiyouga
2024-06-22 00:00:38 +08:00
parent af2cb33bb2
commit f29c1ac6e5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/api/chat.py
View File

@@ -93,7 +93,7 @@ def _process_request(
if message.role == Role.ASSISTANT and isinstance(message.tool_calls, list) and len(message.tool_calls): if message.role == Role.ASSISTANT and isinstance(message.tool_calls, list) and len(message.tool_calls):
tool_calls = [ tool_calls = [
{"name": tool_call.function.name, "argument": tool_call.function.arguments} {"name": tool_call.function.name, "arguments": tool_call.function.arguments}
for tool_call in message.tool_calls for tool_call in message.tool_calls
] ]
content = json.dumps(tool_calls, ensure_ascii=False) content = json.dumps(tool_calls, ensure_ascii=False)