github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-07 22:11:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

openai[patch]: fix async http client (#19164)

Browse Source 
Fix #19116
This commit is contained in:
Bagatur
2024-03-16 17:50:22 -07:00
committed by GitHub
parent 635b3372bd
commit 611d5a1618
6 changed files with 70 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
libs/partners/openai/langchain_openai/chat_models/azure.py
View File

@@ -185,12 +185,17 @@ class AzureChatOpenAI(ChatOpenAI):
"max_retries": values["max_retries"],
"default_headers": values["default_headers"],
"default_query": values["default_query"],
"http_client": values["http_client"],
}
values["client"] = openai.AzureOpenAI(**client_params).chat.completions
values["async_client"] = openai.AsyncAzureOpenAI(
**client_params
).chat.completions
if not values.get("client"):
sync_specific = {"http_client": values["http_client"]}
values["client"] = openai.AzureOpenAI(
**client_params, **sync_specific
).chat.completions
if not values.get("async_client"):
async_specific = {"http_client": values["http_async_client"]}
values["async_client"] = openai.AsyncAzureOpenAI(
**client_params, **async_specific
).chat.completions
return values
@property