如何使用Slack聊天加载器转换聊天记录为AI消息

llzwxh888

于 2024-10-08 17:00:51 发布

阅读量243

点赞数 2

文章标签：人工智能 python

本文链接：https://blog.csdn.net/ppoojjj/article/details/142764230

版权

引言

在现代工作环境中，Slack已成为团队沟通的重要工具。如果你想将Slack聊天记录用于AI应用，如训练模型或生成对话，你需要将其转换为适合的格式。本篇文章将介绍如何使用Slack聊天加载器，将导出的Slack对话转换为LangChain中的聊天消息。

主要内容

1. 导出聊天记录

首先，你需要从Slack导出所需的对话线程。具体步骤可以参考Slack的官方说明。通常，这将得到一个包含多个JSON文件的zip目录。

2. 创建聊天加载器

接下来，使用SlackChatLoader类来加载你的聊天记录。将文件路径指向导出的JSON文件或JSON文件目录。

from langchain_community.chat_loaders.slack import SlackChatLoader

# 使用API代理服务提高访问稳定性
loader = SlackChatLoader(path="slack_dump.zip")

3. 加载和转换消息

调用load()或lazy_load()方法进行转换。你可以选择合并同一发送者的连续消息，并将特定发送者的消息转换为"AIMessage"类。

from typing import List
from langchain_community.chat_loaders.utils import (
    map_ai_messages,
    merge_chat_runs,
)
from langchain_core.chat_sessions import ChatSession

raw_messages = loader.lazy_load()
# 合并同一发送者的连续消息
merged_messages = merge_chat_runs(raw_messages)
# 将"U0500003428"的消息转换为AI消息
messages: List[ChatSession] = list(
    map_ai_messages(merged_messages, sender="U0500003428")
)

代码示例

以下是一个完整示例：

import requests

# 下载示例数据
permalink = "https://raw.githubusercontent.com/langchain-ai/langchain/342087bdfa3ac31d622385d0f2d09cf5e06c8db3/libs/langchain/tests/integration_tests/examples/slack_export.zip"
response = requests.get(permalink)
with open("slack_dump.zip", "wb") as f:
    f.write(response.content)

# 加载并转换消息
loader = SlackChatLoader(path="slack_dump.zip")
raw_messages = loader.lazy_load()
merged_messages = merge_chat_runs(raw_messages)
messages: List[ChatSession] = list(
    map_ai_messages(merged_messages, sender="U0500003428")
)