**如何优雅地合并连续相同类型的消息——探秘merge_message

本文链接：https://blog.youkuaiyun.com/tt_jishu/article/details/143772900

引言

在开发聊天机器人或自然语言处理应用时，我们常常需要处理一系列的消息。这些消息可能是用户输入、系统响应或AI生成的对话。这些消息有时会连成“连续的相同类型消息”（Runs），而某些模型可能不支持这种形式的输入。merge_message_runs工具能够帮助我们优雅地解决这一问题。本文将带你了解如何使用这一工具，合并连续的同类型消息，提高代码的整洁性和运行效率。

主要内容

什么是`merge_message_runs`？

merge_message_runs是一个用于合并连续相同类型消息的工具。如果你有一系列的系统消息或用户消息，这个工具会将这些消息合并为单独的一条消息。这样可以确保你的应用在使用某些API时能够顺利运行，不会因为消息格式问题报错。

`merge_message_runs`的使用场景

简化消息流：减少消息处理的复杂度。
兼容性：确保与不支持连续相同类型消息的API兼容。
提高代码可读性：减少多余的可读性障碍，让代码更易于维护。

代码示例

下面是一个使用merge_message_runs的示例代码：

from langchain_core.messages import (
    AIMessage,
    HumanMessage,
    SystemMessage,
    merge_message_runs,
)

# 定义一系列消息
messages = [
    SystemMessage("you're a good assistant."),
    SystemMessage("you always respond with a joke."),
    HumanMessage([{"type": "text", "text": "i wonder why it's called langchain"}]),
    HumanMessage("and who is harrison chasing anyways"),
    AIMessage(
        'Well, I guess they thought "WordRope" and "SentenceString" just didn\'t have the same ring to it!'
    ),
    AIMessage("Why, he's probably chasing after the last cup of coffee in the office!"),
]

# 合并消息
merged = merge_message_runs(messages)

# 打印合并后的消息
print("\n\n".join([repr(x) for x in merged]))

输出结果

SystemMessage(content="you're a good assistant.\nyou always respond with a joke.")
HumanMessage(content=[{'type': 'text', 'text': "i wonder why it's called langchain"}, 'and who is harrison chasing anyways'])
AIMessage(content='Well, I guess they thought "WordRope" and "SentenceString" just didn\'t have the same ring to it!\nWhy, he\'s probably chasing after the last cup of coffee in the office!')