[高效过滤消息：使用`filter_messages`优化AI交互链]

qq_37836323

于 2024-10-03 23:42:26 发布

阅读量92

点赞数 3

文章标签：交互 microsoft python

本文链接：https://blog.csdn.net/qq_29929123/article/details/142697891

版权

# 高效过滤消息：使用`filter_messages`优化AI交互链

## 引言

在构建复杂的AI交互链时，状态管理通常通过消息列表实现。然而，随着多个模型、对话者和子链的加入，这个列表会变得冗长。我们可能需要在每次模型调用中仅传递这些消息的子集。本文将介绍如何使用`filter_messages`实用工具简化这一过程。

## 主要内容

### 理解消息类型

在`langchain_core.messages`模块中，我们常见的消息类型包括：

- `AIMessage`
- `HumanMessage`
- `SystemMessage`

这些类型有助于分类和管理消息，从而实现精准过滤。

### 基本使用方法

`filter_messages`让我们可以根据类型、ID或名称方便地过滤消息。下面是基本用法：

```python
from langchain_core.messages import (
    AIMessage,
    HumanMessage,
    SystemMessage,
    filter_messages,
)

messages = [
    SystemMessage("you are a good assistant", id="1"),
    HumanMessage("example input", id="2", name="example_user"),
    AIMessage("example output", id="3", name="example_assistant"),
    HumanMessage("real input", id="4", name="bob"),
    AIMessage("real output", id="5", name="alice"),
]

# 包含特定类型
filtered_by_type = filter_messages(messages, include_types="human")

# 排除特定名称
filtered_by_name = filter_messages(messages, exclude_names=["example_user", "example_assistant"])

# 包含特定类型并排除特定ID
filtered_by_type_and_id = filter_messages(messages, include_types=[HumanMessage, AIMessage], exclude_ids=["3"])

在链中使用

可以将filter_messages以声明式或命令式方式用于链中，与其他组件组合：

from langchain_anthropic import ChatAnthropic

llm = ChatAnthropic(model="claude-3-sonnet-20240229", temperature=0)
filter_ = filter_messages(exclude_names=["example_user", "example_assistant"])
chain = filter_ | llm

# 调用链
chain.invoke(messages)

代码示例

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

messages = [
    SystemMessage("you are a good assistant", id="1"),
    HumanMessage("example input", id="2", name="example_user"),
    AIMessage("example output", id="3", name="example_assistant"),
    HumanMessage("real input", id="4", name="bob"),
    AIMessage("real output", id="5", name="alice"),
]

# 使用API代理服务提高访问稳定性
filtered_messages = filter_messages(messages, include_types="human")
print(filtered_messages)