LangChain聊天机器人教程

import sys
sys.stdout.reconfigure(encoding='utf-8')  # 确保中文输出正常
from dotenv import load_dotenv
load_dotenv()  # 加载环境变量

from langchain_community.chat_models.moonshot import MoonshotChat
from langchain_community.chat_models import ChatZhipuAI
from langchain_core.messages import HumanMessage, SystemMessage, AIMessage

# 构建对话历史
messages = [
    SystemMessage("你是一名社交媒体运营专家"),  # 🎭 设定角色
    HumanMessage("你好，介绍一下如何使用YouTube进行视频创作"),  # 👤 用户问题
    AIMessage("您需要确定目标受众和内容主题呀主人~")  # 🤖 AI回复
]

# 初始化模型并获取回复
llm = ChatZhipuAI(model="GLM-4-FlashX-250414")
result = llm.invoke(messages)
print(f"GLM4-FlashX:{result.content}")

# 方式1：使用Moonshot模型
# llm = MoonshotChat(model="kimi-k2-0711-preview")

# 方式2：使用智谱GLM模型
llm = ChatZhipuAI(model="GLM-4-FlashX-250414")

# 直接存入内存的方式
messages = []  # 简单的列表存储
system_message = SystemMessage("你是一名社交媒体运营专家")
messages.append(system_message)

# 聊天循环
while True:
    query = input("You:")
    if query.lower() == "exit":
        break
    messages.append(HumanMessage(content=query))  # 添加用户消息
    
    result = model.invoke(messages)
    response = result.content
    messages.append(AIMessage(content=response))  # 添加AI回复
    print(f"AI:{response}")

PROJECT_ID = "..."        # 🏗️ Google Cloud项目ID
SESSION_ID = "user_session_new"       # 👤 用户会话ID
COLLECTION_NAME = "chat_history"      # 📁 Firestore集合名称

print("初始化Firestore客户端...")
client = firestore.Client(project=PROJECT_ID)

print("初始化Firestore聊天记录...")
chat_history = FirestoreChatMessageHistory(
    session_id=SESSION_ID,
    collection=COLLECTION_NAME,
    client=client
)

print("历史聊天记录:", chat_history.messages)  # 📜 显示历史记录

while True:
    query = input("You:")
    if query.lower() == "exit":
        break
    
    # 🔄 自动保存到Firestore
    chat_history.add_message(HumanMessage(content=query))
    
    result = model.invoke(chat_history.messages)
    response = result.content
    
    # 🔄 自动保存AI回复到Firestore
    chat_history.add_message(AIMessage(content=response))
    print(f"AI:{response}")

print("---Message History---")
print(chat_history.messages)  # 📋 显示完整历史

pip install langchain-community
pip install langchain-google-firestore
pip install google-cloud-firestore
pip install python-dotenv

# API密钥配置
ZHIPU_API_KEY=your_zhipu_api_key
MOONSHOT_API_KEY=your_moonshot_api_key

# Google Cloud配置
GOOGLE_APPLICATION_CREDENTIALS=path/to/your/service-account-key.json

python chat.py

python history.py