APEX/BrowserUse_and_ComputerUse_skills
5
0
Fork 0
Code Issues 6 Pull requests 1 Projects 1 Releases Packages Wiki Activity Actions

fix: allow tool_choice, parallel_tool_calls, prompt_cache_key in codex preflight

Browse source 
_preflight_codex_api_kwargs rejected these three fields as unsupported,
but _build_api_kwargs adds them to every codex request. This caused a
ValueError before _interruptible_api_call was reached, which was caught
by the retry loop and retried with exponential backoff — appearing as
an infinite hang in tests (275s total backoff across 6 retries).

The fix adds these keys to allowed_keys and passes them through to the
normalized request dict.

This fixes the hanging test_cron_run_job_codex_path_handles_internal_401_refresh
test (now passes in 2.6s instead of timing out).
This commit is contained in:
teknium1 2026-03-11 07:00:14 -07:00
parent 59b53f0a23
commit d2dee43825
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
run_agent.py
View file

@ -1781,6 +1781,7 @@ class AIAgent:
allowed_keys = { allowed_keys = {
"model", "instructions", "input", "tools", "store", "model", "instructions", "input", "tools", "store",
"reasoning", "include", "max_output_tokens", "temperature", "reasoning", "include", "max_output_tokens", "temperature",
"tool_choice", "parallel_tool_calls", "prompt_cache_key",
} }
normalized: Dict[str, Any] = { normalized: Dict[str, Any] = {
"model": model, "model": model,
@ -1806,6 +1807,12 @@ class AIAgent:
if isinstance(temperature, (int, float)): if isinstance(temperature, (int, float)):
normalized["temperature"] = float(temperature) normalized["temperature"] = float(temperature)
# Pass through tool_choice, parallel_tool_calls, prompt_cache_key
for passthrough_key in ("tool_choice", "parallel_tool_calls", "prompt_cache_key"):
val = api_kwargs.get(passthrough_key)
if val is not None:
normalized[passthrough_key] = val
if allow_stream: if allow_stream:
stream = api_kwargs.get("stream") stream = api_kwargs.get("stream")
if stream is not None and stream is not True: if stream is not None and stream is not True: