diff --git a/src/coding_agent_bench/helpers/codex.py b/src/coding_agent_bench/helpers/codex.py index a4d3b49..71d22f0 100644 --- a/src/coding_agent_bench/helpers/codex.py +++ b/src/coding_agent_bench/helpers/codex.py @@ -1,18 +1,17 @@ +import json from pathlib import Path -TEMPLATE = """profile = "custom-vllm" + +TEMPLATE = """model = {model_name} +model_provider = "vllm" web_search = "disabled" [model_providers.vllm] name = "vllm" -base_url = "{server_url}/v1" +base_url = {base_url} wire_api = "responses" requires_openai_auth = false -[profiles.custom-vllm] -model_provider = "vllm" -model = "{model_name}" - [features] js_repl = false multi_agent = true @@ -22,7 +21,16 @@ """ +def _toml_string(value: str) -> str: + return json.dumps(value) + + def codex_create_toml(model_name: str, server_url: str, outpath: Path): - toml = TEMPLATE.format(model_name=model_name, server_url=server_url) + base_url = f"{server_url.rstrip('/')}/v1" + toml = TEMPLATE.format( + model_name=_toml_string(model_name), + base_url=_toml_string(base_url), + ) + with open(outpath, "w") as f: f.write(toml)