Discord 集成

import os

import pandas as pd

path = input('请输入 Discord "messages" 文件夹内容的路径： ')
li = []
for f in os.listdir(path):
    expected_csv_path = os.path.join(path, f, "messages.csv")
    csv_exists = os.path.isfile(expected_csv_path)
    if csv_exists:
        df = pd.read_csv(expected_csv_path, index_col=None, header=0)
        li.append(df)

df = pd.concat(li, axis=0, ignore_index=True, sort=False)

from langchain_community.document_loaders.discord import DiscordChatLoader

loader = DiscordChatLoader(df, user_id_col="ID")
print(loader.load())