fix(providers): disable Azure SDK retries by default

Made-with: Cursor
This commit is contained in:
Xubin Ren 2026-04-04 04:33:20 +00:00 committed by Xubin Ren
parent dbdf7e5955
commit 7229a81594
2 changed files with 14 additions and 0 deletions

View File

@ -58,6 +58,7 @@ class AzureOpenAIProvider(LLMProvider):
api_key=api_key,
base_url=base_url,
default_headers={"x-session-affinity": uuid.uuid4().hex},
max_retries=0,
)
# ------------------------------------------------------------------

View File

@ -1,6 +1,7 @@
from unittest.mock import patch
from nanobot.providers.anthropic_provider import AnthropicProvider
from nanobot.providers.azure_openai_provider import AzureOpenAIProvider
from nanobot.providers.openai_compat_provider import OpenAICompatProvider
@ -18,3 +19,15 @@ def test_anthropic_disables_sdk_retries_by_default() -> None:
kwargs = mock_client.call_args.kwargs
assert kwargs["max_retries"] == 0
def test_azure_openai_disables_sdk_retries_by_default() -> None:
with patch("nanobot.providers.azure_openai_provider.AsyncOpenAI") as mock_client:
AzureOpenAIProvider(
api_key="sk-test",
api_base="https://example.openai.azure.com",
default_model="gpt-4.1",
)
kwargs = mock_client.call_args.kwargs
assert kwargs["max_retries"] == 0