test(agent): update custom runtime mock contract

This commit is contained in:
fangliquanflq
2026-09-06 21:27:18 +08:00
committed by Teknium
parent e3ff3e78d1
commit a4f2e42fbe
+1 -1
View File
@@ -793,7 +793,7 @@ class TestAuxiliaryClientProviderPriority:
from agent.auxiliary_client import get_text_auxiliary_client
with patch("agent.auxiliary_client._read_nous_auth", return_value=None), \
patch("agent.auxiliary_client._resolve_custom_runtime",
return_value=("http://localhost:1234/v1", "local-key")), \
return_value=("http://localhost:1234/v1", "local-key", "openai-compatible")), \
patch("agent.auxiliary_client.OpenAI") as mock:
client, model = get_text_auxiliary_client()
assert mock.call_args.kwargs["base_url"] == "http://localhost:1234/v1"