跳到主要内容
Open In ColabOpen on GitHub

Telegram

此 notebook 展示了如何使用 Telegram 聊天加载器。此类帮助将导出的 Telegram 对话映射到 LangChain 聊天消息。

该过程包含三个步骤

  1. 通过从 Telegram 应用程序复制聊天内容并将其粘贴到本地计算机上的文件中来导出聊天 .txt 文件
  2. 使用指向 json 文件或 JSON 文件目录的文件路径创建 TelegramChatLoader
  3. 调用 loader.load()（或 loader.lazy_load()）以执行转换。可以选择使用 merge_chat_runs 合并来自同一发送者的顺序消息，和/或 map_ai_messages 将来自指定发送者的消息转换为“AIMessage”类。

1. 创建消息转储​

目前 (2023/08/23) 此加载器最支持通过从 Telegram Desktop App 导出聊天记录生成的 json 文件格式。

重要提示： 有“lite”版本的 Telegram，例如“Telegram for MacOS”，它们缺少导出功能。请确保您使用正确的应用程序导出文件。

要进行导出

  1. 下载并打开 Telegram 桌面版
  2. 选择一个对话
  3. 导航到对话设置（当前是右上角的三个点）
  4. 点击“导出聊天记录”
  5. 取消选择照片和其他媒体。选择“机器可读 JSON”格式进行导出。

以下是一个示例

%%writefile telegram_conversation.json
{
 "name": "Jiminy",
 "type": "personal_chat",
 "id": 5965280513,
 "messages": [
  {
   "id": 1,
   "type": "message",
   "date": "2023-08-23T13:11:23",
   "date_unixtime": "1692821483",
   "from": "Jiminy Cricket",
   "from_id": "user123450513",
   "text": "You better trust your conscience",
   "text_entities": [
    {
     "type": "plain",
     "text": "You better trust your conscience"
    }
   ]
  },
  {
   "id": 2,
   "type": "message",
   "date": "2023-08-23T13:13:20",
   "date_unixtime": "1692821600",
   "from": "Batman & Robin",
   "from_id": "user6565661032",
   "text": "What did you just say?",
   "text_entities": [
    {
     "type": "plain",
     "text": "What did you just say?"
    }
   ]
  }
 ]
}
Overwriting telegram_conversation.json

2. 创建聊天加载器​

所有需要的是文件路径。您可以选择指定映射到 ai 消息的用户名，以及配置是否合并消息运行。

from langchain_community.chat_loaders.telegram import TelegramChatLoader
API 参考：TelegramChatLoader
loader = TelegramChatLoader(
    path="./telegram_conversation.json",
)

3. 加载消息​

load()（或 lazy_load）方法返回“ChatSessions”列表，该列表目前仅包含每个加载对话的消息列表。

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()
# Merge consecutive messages from the same sender into a single message
merged_messages = merge_chat_runs(raw_messages)
# Convert messages from "Jiminy Cricket" to AI messages
messages: List[ChatSession] = list(
    map_ai_messages(merged_messages, sender="Jiminy Cricket")
)

后续步骤​

然后您可以按照您认为合适的方式使用这些消息，例如微调模型、少样本示例选择，或直接对下一条消息进行预测

from langchain_openai import ChatOpenAI

llm = ChatOpenAI()

for chunk in llm.stream(messages[0]["messages"]):
    print(chunk.content, end="", flush=True)
API 参考：ChatOpenAI
I said, "You better trust your conscience."

此页面是否有所帮助？