github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 23:29:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(jinachat): related to #9197 (#9200)

Browse Source 
related to: https://github.com/langchain-ai/langchain/issues/9197

---------

Co-authored-by: qianjun.wqj <qianjun.wqj@alibaba-inc.com>
Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
Clark 2023-08-15 02:04:20 +08:00 committed by GitHub
parent 4f1feaca83
commit 03ea0762a1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libs/langchain/langchain/chat_models/jinachat.py
View File

@ -35,6 +35,7 @@ from langchain.schema import (
ChatGeneration, ChatGeneration,
ChatMessage, ChatMessage,
ChatResult, ChatResult,
FunctionMessage,
HumanMessage, HumanMessage,
SystemMessage, SystemMessage,
) )
@ -125,6 +126,12 @@ def _convert_message_to_dict(message: BaseMessage) -> dict:
message_dict = {"role": "assistant", "content": message.content} message_dict = {"role": "assistant", "content": message.content}
elif isinstance(message, SystemMessage): elif isinstance(message, SystemMessage):
message_dict = {"role": "system", "content": message.content} message_dict = {"role": "system", "content": message.content}
elif isinstance(message, FunctionMessage):
message_dict = {
"role": "function",
"name": message.name,
"content": message.content,
}
else: else:
raise ValueError(f"Got unknown type {message}") raise ValueError(f"Got unknown type {message}")
if "name" in message.additional_kwargs: if "name" in message.additional_kwargs: