理解与应用：如何合并连续相同类型的消息-CSDN博客

本文链接：https://blog.csdn.net/tt_jishu/article/details/142537230

理解与应用：如何合并连续相同类型的消息

在构建复杂对话系统或使用某些语言模型时，可能会遇到一个限制：无法传递连续相同类型的消息。本文将介绍如何使用merge_message_runs工具来方便地合并这些消息。我们将提供详细的用法说明、潜在的挑战以及代码示例，让你轻松掌握这一技术。

引言

在处理自然语言对话时，我们经常需要传递一系列结构化的消息给模型。有些模型对连续的相同消息类型有处理限制，因此整合这些消息成为一个重要的操作。merge_message_runs工具提供了一种便捷的方式来实现这一目标。

主要内容

什么是`merge_message_runs`

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]))

处理消息内容

注意，若要合并的消息内容是列表，则合并后的消息内容将是一个内容块列表；如果是字符串，则这些内容用换行符连接。

使用重载操作符

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]))

代码示例

以下是完整的代码示例，演示了如何合并连续的相同类型的消息：

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]))