mirror/nanobot
1
0
Fork 0
mirror of https://github.com/HKUDS/nanobot.git synced 2026-05-21 17:12:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(providers): disable Azure SDK retries by default

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
nanobot/providers/azure_openai_provider.py
View File

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

13
tests/providers/test_provider_sdk_retry_defaults.py
View File

@ -1,6 +1,7 @@
from unittest.mock import patch from unittest.mock import patch
from nanobot.providers.anthropic_provider import AnthropicProvider from nanobot.providers.anthropic_provider import AnthropicProvider
from nanobot.providers.azure_openai_provider import AzureOpenAIProvider
from nanobot.providers.openai_compat_provider import OpenAICompatProvider 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 kwargs = mock_client.call_args.kwargs
assert kwargs["max_retries"] == 0 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