fix(openai): disable stream_usage in chat completions if OPENAI_BASE_URL is set (#33298)

This env var is used internally by the OpenAI client.
This commit is contained in:
ccurme
2025-10-06 10:14:43 -04:00
committed by GitHub
parent c8636a626a
commit 95a451ef2c

View File

@@ -776,20 +776,23 @@ class BaseChatOpenAI(BaseChatModel):
self.openai_api_base = self.openai_api_base or os.getenv("OPENAI_API_BASE") self.openai_api_base = self.openai_api_base or os.getenv("OPENAI_API_BASE")
# Enable stream_usage by default if using default base URL and client # Enable stream_usage by default if using default base URL and client
if all( if (
getattr(self, key, None) is None all(
for key in ( getattr(self, key, None) is None
"stream_usage", for key in (
"openai_proxy", "stream_usage",
"openai_api_base", "openai_proxy",
"base_url", "openai_api_base",
"client", "base_url",
"root_client", "client",
"async_client", "root_client",
"root_async_client", "async_client",
"http_client", "root_async_client",
"http_async_client", "http_client",
"http_async_client",
)
) )
and "OPENAI_BASE_URL" not in os.environ
): ):
self.stream_usage = True self.stream_usage = True