github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-12 14:23:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improvement/add finish reason to generation info in chat open ai (#7478)

Browse Source 
Description: ChatOpenAI model does not return finish_reason in
generation_info.
Issue: #2702
Dependencies: None
Tag maintainer: @baskaryan 

Thank you

---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
tonomura 2023-07-12 07:12:57 +09:00 committed by GitHub
parent b96ac13f3d
commit cc0585af42
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/chat_models/openai.py
View File

@ -386,7 +386,10 @@ class ChatOpenAI(BaseChatModel):
generations = [] generations = []
for res in response["choices"]: for res in response["choices"]:
message = _convert_dict_to_message(res["message"]) message = _convert_dict_to_message(res["message"])
gen = ChatGeneration(message=message) gen = ChatGeneration(
message=message,
generation_info=dict(finish_reason=res.get("finish_reason")),
)
generations.append(gen) generations.append(gen)
llm_output = {"token_usage": response["usage"], "model_name": self.model_name} llm_output = {"token_usage": response["usage"], "model_name": self.model_name}
return ChatResult(generations=generations, llm_output=llm_output) return ChatResult(generations=generations, llm_output=llm_output)