旅行智能体 (Travel Agent)

基于 LangChain 构建的智能旅行助手，支持天气查询、航班查询、景点推荐和预算计算。

功能特性

• 🌤️ 天气查询：查询北京、上海的实时天气信息
• ✈️ 航班查询：查询北京-上海之间的航班信息，支持按价格筛选
• 🏛️ 景点推荐：推荐北京、上海的热门旅游景点
• 💰 预算计算：计算机票和景点门票费用

项目结构

travel_agent/
├── data/                          # 数据文件目录
│   ├── weather.json              # 天气数据
│   ├── flights.json              # 航班数据
│   └── attractions.json          # 景点数据
├── tools/                         # 工具函数目录
│   ├── __init__.py
│   ├── weather_tools.py          # 天气查询工具
│   ├── flight_tools.py           # 航班查询工具
│   ├── attraction_tools.py       # 景点推荐工具
│   └── budget_tools.py           # 预算计算工具
├── travel_agent.py               # 主程序
├── requirements.txt              # 依赖包
└── README.md                     # 项目说明

安装依赖

pip install -r requirements.txt --index-url https://pypi.org/simple/

配置

需要设置 DeepSeek API Key：

Windows (CMD)

set OPENAI_API_KEY=your_deepseek_api_key

Windows (PowerShell)

$env:OPENAI_API_KEY='your_deepseek_api_key'

Linux/Mac

export OPENAI_API_KEY='your_deepseek_api_key'

注意：虽然环境变量名为 OPENAI_API_KEY，但实际使用的是 DeepSeek 的 API Key。程序会自动连接到 DeepSeek 的 API 端点。

使用方法

启动智能体

python travel_agent.py

交互示例

您：我想从北京去上海，帮我查一下航班
助手：[显示北京到上海的航班信息]

您：上海天气怎么样
助手：[显示上海当前天气]

您：推荐一下北京的景点
助手：[显示北京热门景点推荐]

您：我想和女朋友去上海玩3天，帮我算一下预算
助手：[计算往返机票和景点门票费用]

您：我和一朋友想下周五从上海出发去北京玩5天，预算8000元，喜欢美食和历史文化。请给我安排规划一下。
助手：[显示当天天气，显示航班信息，计算费用，景点推荐]

工具说明

1. query_weather

查询指定城市的天气信息

• 参数：city（城市名称）
• 支持：北京、上海

2. query_flights

查询两个城市之间的航班信息

• 参数：origin（出发城市）、destination（目的地城市）
• 返回：航班号、航空公司、时间、价格

3. get_cheapest_flight

获取最便宜的航班

• 参数：origin（出发城市）、destination（目的地城市）

4. recommend_attractions

推荐指定城市的旅游景点

• 参数：city（城市名称）
• 返回：景点名称、描述、票价、游玩时长

5. get_attraction_by_category

根据类别推荐景点

• 参数：city（城市名称）、category（景点类别）

6. calculate_flight_budget

计算往返机票总费用

• 参数：origin、destination、passengers（乘客人数）

7. calculate_total_budget

计算旅行总预算（机票+景点门票）

• 参数：origin、destination、passengers、attraction_count（景点数量）

数据说明

所有数据均为模拟数据：

• 航班价格范围：500-1500 元
• 每个城市包含 4 个景点
• 不包含日期相关的数据变化

注意事项

1. 需要有效的 DeepSeek API Key
2. 当前仅支持北京和上海两个城市
3. 预算计算仅包含交通和景点门票，不包含住宿、餐饮等费用

开发者

基于 LangChain 框架开发，使用 DeepSeek API 提供 LLM 能力

源码

https://github.com/chendanyu/travel_agent/tree/main/langchain_agent