基于 LangChain 构建智能日程规划机器人

pip install langchain openai python-dotenv

pip install dateparser  # 处理自然语言时间表达
pip install gradio      # 构建简单对话界面

OPENAI_API_KEY=你的OpenAI密钥

from dotenv import load_dotenv
load_dotenv()

calendar_agent/
├── tools/
│   ├── calendar_tools.py      # 自定义工具
├── agent_calendar.py          # Agent 主体
├── calendar_data.py           # 模拟数据库
├── app.py                     # UI入口（Gradio）
├── .env

# calendar_data.py
calendar_db = []

def add_event(date, description):
    calendar_db.append({"date": date, "description": description})
    return f"已添加：{date} - {description}"

def view_events():
    if not calendar_db:
        return "目前暂无任何日程安排。"
    return "\n".join([f"{item['date']}: {item['description']}" for item in calendar_db])

def remove_event(description):
    for i, item in enumerate(calendar_db):
        if description in item['description']:
            calendar_db.pop(i)
            return f"已删除日程：{item['description']}"
    return "未找到相关日程。"

# tools/calendar_tools.py
from langchain.tools import Tool
from calendar_data import add_event, view_events, remove_event
import dateparser

# 添加日程
def add_event_tool(input: str):
    # 解析时间和描述
    dt = dateparser.parse(input)
    if dt:
        return add_event(str(dt), input)
    else:
        return "无法解析时间，请尝试更清晰的描述。"

# 查看日程
def view_events_tool(_: str = ""):
    return view_events()

# 删除日程
def remove_event_tool(input: str):
    return remove_event(input)

tools = [
    Tool(
        name="Add Calendar Event",
        func=add_event_tool,
        description="用于添加新的日程，用户会描述内容与时间"
    ),
    Tool(
        name="View Calendar",
        func=view_events_tool,
        description="查看所有日程"
    ),
    Tool(
        name="Delete Calendar Event",
        func=remove_event_tool,
        description="删除指定的日程内容"
    ),
]

# agent_calendar.py
from langchain.chat_models import ChatOpenAI
from langchain.agents import initialize_agent, AgentType
from langchain.memory import ConversationBufferMemory
from tools.calendar_tools import tools
import os
from dotenv import load_dotenv
load_dotenv()

llm = ChatOpenAI(temperature=0)

memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)

agent_executor = initialize_agent(
    tools=tools,
    llm=llm,
    agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION,
    memory=memory,
    verbose=True
)

def ask_agent(query):
    return agent_executor.run(query)

# app.py
import gradio as gr
from agent_calendar import ask_agent

def chat(input_text, history):
    output = ask_agent(input_text)
    history.append((input_text, output))
    return "", history

gr.ChatInterface(fn=chat).launch()

python app.py

已添加：2025-04-11 09:00:00 - 请帮我安排一个周五早上9点的英语口语课。

2025-04-11 09:00:00: 请帮我安排一个周五早上9点的英语口语课。

已删除日程：请帮我安排一个周五早上9点的英语口语课。
已添加：2025-04-11 10:00:00 - 改成10点吧

用户输入 → LLM 生成 Thought → 判断 Action → 调用 Tool → Observation → LLM 生成回答