File tree Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -37,7 +37,7 @@ def run_tool_calling_llm(llm, request_params):
37
37
request_params ["tools" ] = [tool_schema ]
38
38
39
39
last_tool_id = 0
40
- for message in request_params ["messages" ]:
40
+ for i , message in enumerate ( request_params ["messages" ]) :
41
41
if "function_call" in message :
42
42
function = message .pop ("function_call" )
43
43
message ["tool_calls" ] = [
@@ -48,10 +48,16 @@ def run_tool_calling_llm(llm, request_params):
48
48
}
49
49
]
50
50
if message ["role" ] == "function" :
51
- message ["role" ] = "tool"
52
- message ["tool_call_id" ] = "toolu_" + str (last_tool_id )
51
+ if i != 0 and request_params ["messages" ][i - 1 ]["role" ] == "tool" :
52
+ request_params ["messages" ][i ]["content" ] += message ["content" ]
53
+ message = None
54
+ else :
55
+ message ["role" ] = "tool"
56
+ message ["tool_call_id" ] = "toolu_" + str (last_tool_id )
57
+
58
+ last_tool_id += 1
53
59
54
- last_tool_id += 1
60
+ request_params [ "messages" ] = [ m for m in request_params [ "messages" ] if m != None ]
55
61
56
62
# Add OpenAI's recommended function message
57
63
# request_params["messages"][0][
You can’t perform that action at this time.
0 commit comments