mirror of
https://github.com/hwchase17/langchain.git
synced 2026-06-09 18:50:33 +00:00
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:
@@ -776,7 +776,8 @@ 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 (
|
||||||
|
all(
|
||||||
getattr(self, key, None) is None
|
getattr(self, key, None) is None
|
||||||
for key in (
|
for key in (
|
||||||
"stream_usage",
|
"stream_usage",
|
||||||
@@ -790,6 +791,8 @@ class BaseChatOpenAI(BaseChatModel):
|
|||||||
"http_client",
|
"http_client",
|
||||||
"http_async_client",
|
"http_async_client",
|
||||||
)
|
)
|
||||||
|
)
|
||||||
|
and "OPENAI_BASE_URL" not in os.environ
|
||||||
):
|
):
|
||||||
self.stream_usage = True
|
self.stream_usage = True
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user