Fix parallel tool call handling in test utils

alexander-akhmetov · alexander-akhmetov · commit 253e66426d63 · 2026-03-05T09:46:15.000+01:00
diff --git a/tests/utils.py b/tests/utils.py
@@ -90,6 +90,7 @@ async def run_llm_tool_loop(
     response = await acompletion(model=model, messages=messages, tools=tools)
 
     while response.choices and response.choices[0].message.tool_calls:
+        messages.append(response.choices[0].message)
         for tool_call in response.choices[0].message.tool_calls:
             tool_name = tool_call.function.name
             args = (
@@ -101,7 +102,6 @@ async def run_llm_tool_loop(
                 mcp_client, tool_name, args
             )
             tools_called.append(mcp_tc)
-            messages.append(response.choices[0].message)
             messages.append(
                 Message(role="tool", tool_call_id=tool_call.id, content=result_text)
             )

Original file line number	Diff line number	Diff line change
`@@ -90,6 +90,7 @@ async def run_llm_tool_loop(`
`90`	`90`	`response = await acompletion(model=model, messages=messages, tools=tools)`
`91`	`91`
`92`	`92`	`while response.choices and response.choices[0].message.tool_calls:`
	`93`	`+ messages.append(response.choices[0].message)`
`93`	`94`	`for tool_call in response.choices[0].message.tool_calls:`
`94`	`95`	`tool_name = tool_call.function.name`
`95`	`96`	`args = (`
`@@ -101,7 +102,6 @@ async def run_llm_tool_loop(`
`101`	`102`	`mcp_client, tool_name, args`
`102`	`103`	`)`
`103`	`104`	`tools_called.append(mcp_tc)`
`104`		`- messages.append(response.choices[0].message)`
`105`	`105`	`messages.append(`
`106`	`106`	`Message(role="tool", tool_call_id=tool_call.id, content=result_text)`
`107`	`107`	`)`