AIMaster7 commited on
Commit
dc017da
·
verified ·
1 Parent(s): 348c3b8

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -244,7 +244,9 @@ Using tools is recommended.
244
 
245
  if "<tool_call>" in assistant_response and "</tool_call>" in assistant_response:
246
  tool_call_str = assistant_response.split("<tool_call>")[1].split("</tool_call>")[0]
247
- tool_call_json = json.loads(tool_call_str.strip())
 
 
248
 
249
 
250
  return JSONResponse(content={
 
244
 
245
  if "<tool_call>" in assistant_response and "</tool_call>" in assistant_response:
246
  tool_call_str = assistant_response.split("<tool_call>")[1].split("</tool_call>")[0]
247
+ tool_call = json.loads(tool_call_str.strip())
248
+ tool_call_json = [{"id": generate_random_id("call_"),"function": {"name": tool_call["name"], "arguments": json.dumps(tool_call["parameters"])}}]
249
+
250
 
251
 
252
  return JSONResponse(content={