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

community: adding tool_call_id for every ToolCall (#22323)

Browse Source 
- **Description:** This PR contains a bugfix which result in malfunction
of multi-turn conversation in QianfanChatEndpoint and adaption for
ToolCall and ToolMessage
This commit is contained in:
Dobiichi-Origami 2024-05-30 22:59:08 +08:00 committed by GitHub
parent 569d325a59
commit 10b12e1c08
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 23 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
libs/community/langchain_community/chat_models/baidu_qianfan_endpoint.py
View File

@ -1,4 +1,5 @@
import logging import logging
import uuid
from operator import itemgetter from operator import itemgetter
from typing import ( from typing import (
Any, Any,
@ -29,6 +30,7 @@ from langchain_core.messages import (
FunctionMessage, FunctionMessage,
HumanMessage, HumanMessage,
SystemMessage, SystemMessage,
ToolMessage,
) )
from langchain_core.output_parsers.base import OutputParserLike from langchain_core.output_parsers.base import OutputParserLike
from langchain_core.output_parsers.openai_tools import ( from langchain_core.output_parsers.openai_tools import (
@ -59,7 +61,7 @@ def convert_message_to_dict(message: BaseMessage) -> dict:
# If function call only, content is None not empty string # If function call only, content is None not empty string
if message_dict["content"] == "": if message_dict["content"] == "":
message_dict["content"] = None message_dict["content"] = None
elif isinstance(message, FunctionMessage): elif isinstance(message, (FunctionMessage, ToolMessage)):
message_dict = { message_dict = {
"role": "function", "role": "function",
"content": message.content, "content": message.content,
@ -81,21 +83,28 @@ def _convert_dict_to_message(_dict: Mapping[str, Any]) -> AIMessage:
additional_kwargs["function_call"].pop("thoughts") additional_kwargs["function_call"].pop("thoughts")
additional_kwargs = {**_dict.get("body", {}), **additional_kwargs} additional_kwargs = {**_dict.get("body", {}), **additional_kwargs}
return AIMessage( msg_additional_kwargs = dict(
content=content,
additional_kwargs=dict(
finish_reason=additional_kwargs.get("finish_reason", ""), finish_reason=additional_kwargs.get("finish_reason", ""),
request_id=additional_kwargs["id"], request_id=additional_kwargs["id"],
object=additional_kwargs.get("object", ""), object=additional_kwargs.get("object", ""),
search_info=additional_kwargs.get("search_info", []), search_info=additional_kwargs.get("search_info", []),
function_call=additional_kwargs.get("function_call", {}), )
tool_calls=[
if additional_kwargs.get("function_call", {}):
msg_additional_kwargs["function_call"] = additional_kwargs.get(
"function_call", {}
)
msg_additional_kwargs["tool_calls"] = [
{ {
"type": "function", "type": "function",
"function": additional_kwargs.get("function_call", {}), "function": additional_kwargs.get("function_call", {}),
"id": str(uuid.uuid4()),
} }
], ]
),
return AIMessage(
content=content,
additional_kwargs=msg_additional_kwargs,
) )