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: raw response headers (#24150)

Browse Source
This commit is contained in:
Erick Friis
2024-07-16 09:54:54 -07:00
committed by GitHub
parent dc42279eb5
commit 1e9cc02ed8
5 changed files with 106 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -928,7 +928,9 @@ class AzureChatOpenAI(BaseChatOpenAI):
return params
def _create_chat_result(
self, response: Union[dict, openai.BaseModel]
self,
response: Union[dict, openai.BaseModel],
generation_info: Optional[Dict] = None,
) -> ChatResult:
if not isinstance(response, dict):
response = response.model_dump()
@@ -938,7 +940,7 @@ class AzureChatOpenAI(BaseChatOpenAI):
"Azure has not provided the response due to a content filter "
"being triggered"
)
chat_result = super()._create_chat_result(response)
chat_result = super()._create_chat_result(response, generation_info)
if "model" in response:
model = response["model"]