openai[patch]: fix json mode for Azure (#25488)

https://github.com/langchain-ai/langchain/issues/25479
https://github.com/langchain-ai/langchain/issues/25485

---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
ccurme
2024-08-16 12:50:50 -04:00
committed by GitHub
parent 1fd1c1dca5
commit 01ecd0acba
3 changed files with 81 additions and 5 deletions

View File

@@ -640,14 +640,14 @@ class AzureChatOpenAI(BaseChatOpenAI):
}
if not values.get("client"):
sync_specific = {"http_client": values["http_client"]}
values["client"] = openai.AzureOpenAI(
**client_params, **sync_specific
).chat.completions
values["root_client"] = openai.AzureOpenAI(**client_params, **sync_specific)
values["client"] = values["root_client"].chat.completions
if not values.get("async_client"):
async_specific = {"http_client": values["http_async_client"]}
values["async_client"] = openai.AsyncAzureOpenAI(
values["root_async_client"] = openai.AsyncAzureOpenAI(
**client_params, **async_specific
).chat.completions
)
values["async_client"] = values["root_async_client"].chat.completions
return values
def bind_tools(