Skip to content

Commit 95a451e

Browse files
authored
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.
1 parent c8636a6 commit 95a451e

File tree

1 file changed

+16
-13
lines changed
  • libs/partners/openai/langchain_openai/chat_models

1 file changed

+16
-13
lines changed

libs/partners/openai/langchain_openai/chat_models/base.py

Lines changed: 16 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -776,20 +776,23 @@ def validate_environment(self) -> Self:
776776
self.openai_api_base = self.openai_api_base or os.getenv("OPENAI_API_BASE")
777777

778778
# Enable stream_usage by default if using default base URL and client
779-
if all(
780-
getattr(self, key, None) is None
781-
for key in (
782-
"stream_usage",
783-
"openai_proxy",
784-
"openai_api_base",
785-
"base_url",
786-
"client",
787-
"root_client",
788-
"async_client",
789-
"root_async_client",
790-
"http_client",
791-
"http_async_client",
779+
if (
780+
all(
781+
getattr(self, key, None) is None
782+
for key in (
783+
"stream_usage",
784+
"openai_proxy",
785+
"openai_api_base",
786+
"base_url",
787+
"client",
788+
"root_client",
789+
"async_client",
790+
"root_async_client",
791+
"http_client",
792+
"http_async_client",
793+
)
792794
)
795+
and "OPENAI_BASE_URL" not in os.environ
793796
):
794797
self.stream_usage = True
795798

0 commit comments

Comments
 (0)