fix(openai): add max_retries parameter to ChatOpenAI for handling 503 capacity errors (#32286)

mdrxy · web-flow · commit e79e0bd6b4a4 · 2025-07-28T13:58:23.000-04:00
Some integration tests were failing
diff --git a/libs/partners/openai/tests/integration_tests/chat_models/test_base.py b/libs/partners/openai/tests/integration_tests/chat_models/test_base.py
@@ -221,6 +221,7 @@ def test_openai_invoke() -> None:
     llm = ChatOpenAI(
         model="o4-mini",
         service_tier="flex",  # Also test service_tier
+        max_retries=3,  # Add retries for 503 capacity errors
     )
 
     result = llm.invoke("Hello", config=dict(tags=["foo"]))

Original file line number	Diff line number	Diff line change
`@@ -221,6 +221,7 @@ def test_openai_invoke() -> None:`
`221`	`221`	`llm = ChatOpenAI(`
`222`	`222`	`model="o4-mini",`
`223`	`223`	`service_tier="flex", # Also test service_tier`
	`224`	`+ max_retries=3, # Add retries for 503 capacity errors`
`224`	`225`	`)`
`225`	`226`
`226`	`227`	`result = llm.invoke("Hello", config=dict(tags=["foo"]))`