Skip to content

Commit

Permalink
re-arrange functuon call message parse logic
Browse files Browse the repository at this point in the history
  • Loading branch information
Dobiichi-Origami committed Nov 6, 2024
1 parent 31f4fb7 commit e3b9dca
Showing 1 changed file with 3 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -66,17 +66,13 @@ def convert_message_to_dict(message: BaseMessage) -> dict:
message_dict = {"role": "user", "content": message.content}
elif isinstance(message, AIMessage):
message_dict = {"role": "assistant", "content": message.content}
if "function_call" in message.additional_kwargs:
message_dict["function_call"] = message.additional_kwargs["function_call"]
elif len(message.tool_calls) != 0:
if len(message.tool_calls) != 0:
tool_call = message.tool_calls[0]
message_dict["function_call"] = {
"name": tool_call["name"],
"args": tool_call["args"],
"arguments": json.dumps(tool_call["args"], ensure_ascii=False),
}

# If function call only, content is None not empty string
if "function_call" in message_dict and message_dict["content"] == "":
# If function call only, content is None not empty string
message_dict["content"] = None
elif isinstance(message, (FunctionMessage, ToolMessage)):
message_dict = {
Expand Down

0 comments on commit e3b9dca

Please sign in to comment.