test(agent): update custom runtime mock contract
This commit is contained in:
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user