This commit is contained in:
2026-03-07 14:49:39 -05:00
parent 6ab359deda
commit f655547184
3 changed files with 14 additions and 16 deletions

View File

@@ -572,6 +572,7 @@ async def _execute_tool_calls_concurrently(
"""
monitor = performance_monitor.get_monitor()
if monitor.enabled: monitor.start_component("ai_client._execute_tool_calls_concurrently")
tier = get_current_tier()
tasks = []
for fc in calls:
if provider == "gemini":
@@ -597,7 +598,7 @@ async def _execute_tool_calls_concurrently(
else:
continue
tasks.append(_execute_single_tool_call_async(name, args, call_id, base_dir, pre_tool_callback, qa_callback, r_idx, patch_callback))
tasks.append(_execute_single_tool_call_async(name, args, call_id, base_dir, pre_tool_callback, qa_callback, r_idx, tier, patch_callback))
results = await asyncio.gather(*tasks)
if monitor.enabled: monitor.end_component("ai_client._execute_tool_calls_concurrently")
@@ -611,8 +612,11 @@ async def _execute_single_tool_call_async(
pre_tool_callback: Optional[Callable[[str, str, Optional[Callable[[str], str]]], Optional[str]]],
qa_callback: Optional[Callable[[str], str]],
r_idx: int,
tier: str | None = None,
patch_callback: Optional[Callable[[str, str], Optional[str]]] = None
) -> tuple[str, str, str, str]:
if tier:
set_current_tier(tier)
out = ""
tool_executed = False
events.emit("tool_execution", payload={"status": "started", "tool": name, "args": args, "round": r_idx})