diff --git a/acp_adapter/session.py b/acp_adapter/session.py index 6f1e17a07f57..995752e26ba4 100644 --- a/acp_adapter/session.py +++ b/acp_adapter/session.py @@ -640,6 +640,7 @@ def _make_agent( "api_mode": api_mode or runtime.get("api_mode"), "base_url": base_url or runtime.get("base_url"), "api_key": runtime.get("api_key"), + "credential_pool": runtime.get("credential_pool"), "command": runtime.get("command"), "args": list(runtime.get("args") or []), } diff --git a/tests/acp/test_session.py b/tests/acp/test_session.py index 199454b39dba..cc7493454d55 100644 --- a/tests/acp/test_session.py +++ b/tests/acp/test_session.py @@ -77,7 +77,9 @@ def test_get_session(self, manager): def test_get_nonexistent_session_returns_none(self, manager): assert manager.get_session("does-not-exist") is None - def test_make_agent_stamps_session_cwd_for_codex_runtime(self, monkeypatch): + def test_make_agent_stamps_cwd_and_forwards_runtime_credential_pool(self, monkeypatch): + sentinel_pool = object() + class FakeAgent: model = "fake-model" @@ -113,6 +115,7 @@ def __init__(self, **kwargs): "api_mode": "codex_app_server", "base_url": "https://example.invalid", "api_key": "test-key", + "credential_pool": sentinel_pool, }, ) monkeypatch.setattr("acp_adapter.session._register_task_cwd", lambda task_id, cwd: None) @@ -120,8 +123,7 @@ def __init__(self, **kwargs): state = SessionManager(db=None).create_session(cwd="/tmp/project") assert state.agent.session_cwd == "/tmp/project" - - + assert state.agent.kwargs["credential_pool"] is sentinel_pool # ---------------------------------------------------------------------------