fix(providers): disable Azure SDK retries by default

Made-with: Cursor
This commit is contained in:
Xubin Ren
2026-04-04 12:36:45 +08:00
committed by Xubin Ren
parent dbdf7e5955
commit 7229a81594
2 changed files with 14 additions and 0 deletions
@@ -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,
)
# ------------------------------------------------------------------
@@ -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