travel-agent-cli 0.2.0 → 0.2.2

package/python/scheduler/tasks.py

@@ -0,0 +1,77 @@
+"""定时任务调度模块"""
+from apscheduler.schedulers.blocking import BlockingScheduler
+from apscheduler.schedulers.asyncio import AsyncIOScheduler
+from apscheduler.triggers.cron import CronTrigger
+from datetime import datetime
+
+from config.settings import get_settings
+from agents.manager_agent import ManagerAgent
+
+
+def job_full_workflow():
+    """完整工作流：使用多 Agent 系统"""
+    print(f"[{datetime.now()}] ===== 开始完整工作流 =====")
+
+    async def run():
+        manager = ManagerAgent()
+        result = await manager.run_workflow(
+            keyword="旅行",
+            include_ota=True,
+            top_n=10,
+            plan_top_n=3,
+            verbose=True,
+        )
+
+        # 保存报告
+        settings = get_settings()
+        settings.ensure_dirs()
+        report_content = result.get("stages", {}).get("report", {}).get("content", "")
+        if report_content:
+            import os
+            filename = f"report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.md"
+            output_path = os.path.join(settings.output_dir, filename)
+            with open(output_path, "w", encoding="utf-8") as f:
+                f.write(report_content)
+            print(f"[{datetime.now()}] 报告已保存：{output_path}")
+
+    import asyncio
+    asyncio.run(run())
+
+    print(f"[{datetime.now()}] ===== 工作流完成 =====")
+
+
+def setup_scheduler(cron_spec: str = None) -> BlockingScheduler:
+    """设置定时调度器
+
+    Args:
+        cron_spec: Cron 表达式，如 "0 9 * * *" 表示每天 9 点
+                   默认每天凌晨 2 点执行
+    """
+    scheduler = BlockingScheduler()
+
+    # 默认每天凌晨 2 点执行
+    if cron_spec is None:
+        cron_spec = "0 2 * * *"
+
+    trigger = CronTrigger.from_crontab(cron_spec)
+
+    # 添加完整工作流任务
+    scheduler.add_job(
+        job_full_workflow,
+        trigger=trigger,
+        id="full_workflow",
+        name="旅行推荐完整工作流",
+        replace_existing=True,
+    )
+
+    # 可选：每小时采集一次（轻量级）
+    # scheduler.add_job(
+    #     job_collect,
+    #     trigger="interval",
+    #     hours=1,
+    #     id="collect_hourly",
+    #     name="每小时数据采集",
+    #     replace_existing=True,
+    # )
+
+    return scheduler

package/python/tools/fliggy_mcp.py

@@ -0,0 +1,553 @@
+# -*- coding: utf-8 -*-
+"""Fliggy MCP 客户端 - Python 实现
+
+基于飞猪开放平台 API，提供：
+- 航班搜索
+- 酒店搜索
+- 景点/门票搜索
+- 订单管理
+
+文档参考:
+- 飞猪开放平台：https://open.alitrip.com/
+- 国内机票慧飞 API: https://open.fliggy.com/docs/doc.htm?treeId=111
+- 酒店商品 API: https://open.alitrip.com/docs/api_list.htm?cid=20752
+"""
+import httpx
+from typing import Dict, Any, List, Optional
+from datetime import datetime
+
+
+class FliggyMCPClient:
+    """Fliggy MCP 客户端"""
+
+    # API 基础 URL
+    BASE_URL = "https://api.fliggy.com"
+    SANDBOX_URL = "https://tap.fliggy.com"
+
+    def __init__(
+        self,
+        app_key: Optional[str] = None,
+        app_secret: Optional[str] = None,
+        access_token: Optional[str] = None,
+        use_sandbox: bool = True,
+    ):
+        """初始化 Fliggy MCP 客户端
+
+        Args:
+            app_key: 飞猪开放平台 App Key
+            app_secret: 飞猪开放平台 App Secret
+            access_token: 用户授权 Token（可选，部分接口需要）
+            use_sandbox: 是否使用沙箱环境
+        """
+        self.app_key = app_key
+        self.app_secret = app_secret
+        self.access_token = access_token
+        self.use_sandbox = use_sandbox
+        self.base_url = self.SANDBOX_URL if use_sandbox else self.BASE_URL
+
+    async def search_flights(
+        self,
+        origin: str,
+        destination: str,
+        departure_date: str,
+        return_date: Optional[str] = None,
+        cabin_class: str = "economy",
+        direct_only: bool = False,
+    ) -> Dict[str, Any]:
+        """搜索航班
+
+        Args:
+            origin: 出发城市（三字码，如 PEK）
+            destination: 目的地城市（三字码，如 SHA）
+            departure_date: 出发日期（YYYY-MM-DD）
+            return_date: 返程日期（可选）
+            cabin_class: 舱位等级（economy/premium/business/first）
+            direct_only: 是否只要直飞航班
+
+        Returns:
+            航班搜索结果
+        """
+        # 沙箱环境返回模拟数据
+        if self.use_sandbox or not self.app_key:
+            return self._mock_flight_search(
+                origin, destination, departure_date, return_date, cabin_class
+            )
+
+        # 实际 API 调用（需要飞猪开放平台账号）
+        async with httpx.AsyncClient() as client:
+            params = {
+                "app_key": self.app_key,
+                "method": "alitrip.travel.flight.search",
+                "v": "2.0",
+                "origin": origin,
+                "destination": destination,
+                "departure_date": departure_date,
+            }
+
+            if return_date:
+                params["return_date"] = return_date
+
+            if direct_only:
+                params["direct_only"] = "true"
+
+            response = await client.get(
+                f"{self.base_url}/router/rest",
+                params=params,
+                timeout=30,
+            )
+            response.raise_for_status()
+            return response.json()
+
+    def _mock_flight_search(
+        self,
+        origin: str,
+        destination: str,
+        departure_date: str,
+        return_date: Optional[str],
+        cabin_class: str,
+    ) -> Dict[str, Any]:
+        """模拟航班搜索（沙箱模式）"""
+        import random
+
+        airlines = [
+            {"code": "CA", "name": "中国国际航空"},
+            {"code": "MU", "name": "东方航空"},
+            {"code": "CZ", "name": "南方航空"},
+            {"code": "HU", "name": "海南航空"},
+            {"code": "ZH", "name": "深圳航空"},
+        ]
+
+        flights = []
+        base_prices = {
+            "economy": 1500,
+            "premium": 3000,
+            "business": 5000,
+            "first": 8000,
+        }
+
+        for i in range(10):
+            airline = random.choice(airlines)
+            base_price = base_prices.get(cabin_class, 1500)
+            price = base_price + random.randint(-500, 1000)
+
+            departure_hour = 6 + (i * 2) % 16
+            flight_duration = 2 + random.randint(0, 3)
+
+            flights.append({
+                "flight_no": f"{airline['code']}{random.randint(100, 9999)}",
+                "airline": airline["name"],
+                "airline_code": airline["code"],
+                "departure": {
+                    "airport": random.choice(["PEK", "PVG", "SHA", "SZX", "CAN"]),
+                    "terminal": random.choice(["T1", "T2", "T3"]),
+                    "time": f"{departure_hour:02d}:{random.randint(0, 59):02d}",
+                    "date": departure_date,
+                },
+                "arrival": {
+                    "airport": random.choice(["PEK", "PVG", "SHA", "SZX", "CAN"]),
+                    "terminal": random.choice(["T1", "T2", "T3"]),
+                    "time": f"{(departure_hour + flight_duration) % 24:02d}:{random.randint(0, 59):02d}",
+                    "date": departure_date,
+                },
+                "duration": f"{flight_duration}小时{random.randint(0, 59):02d}分",
+                "price": price,
+                "cabin_class": cabin_class,
+                "stops": 0 if random.random() > 0.3 else 1,
+                "available_seats": random.randint(1, 20),
+            })
+
+        # 按价格排序
+        flights.sort(key=lambda x: x["price"])
+
+        return {
+            "success": True,
+            "search_id": f"search_{datetime.now().strftime('%Y%m%d%H%M%S')}",
+            "origin": origin,
+            "destination": destination,
+            "departure_date": departure_date,
+            "return_date": return_date,
+            "total_results": len(flights),
+            "flights": flights[:10],  # 返回前 10 个结果
+            "currency": "CNY",
+        }
+
+    async def search_hotels(
+        self,
+        city: str,
+        check_in: str,
+        check_out: str,
+        guests: int = 2,
+        rooms: int = 1,
+        star_rating: Optional[str] = None,
+        price_min: int = 0,
+        price_max: int = 5000,
+    ) -> Dict[str, Any]:
+        """搜索酒店
+
+        Args:
+            city: 城市名称
+            check_in: 入住日期（YYYY-MM-DD）
+            check_out: 退房日期（YYYY-MM-DD）
+            guests: 入住人数
+            rooms: 房间数
+            star_rating: 星级要求（3/4/5/luxury）
+            price_min: 最低价格
+            price_max: 最高价格
+
+        Returns:
+            酒店搜索结果
+        """
+        # 沙箱环境返回模拟数据
+        if self.use_sandbox or not self.app_key:
+            return self._mock_hotel_search(
+                city, check_in, check_out, guests, rooms, star_rating, price_min, price_max
+            )
+
+        # 实际 API 调用
+        async with httpx.AsyncClient() as client:
+            params = {
+                "app_key": self.app_key,
+                "method": "alitrip.travel.hotel.search",
+                "v": "2.0",
+                "city": city,
+                "check_in": check_in,
+                "check_out": check_out,
+                "guests": guests,
+                "rooms": rooms,
+            }
+
+            response = await client.get(
+                f"{self.base_url}/router/rest",
+                params=params,
+                timeout=30,
+            )
+            response.raise_for_status()
+            return response.json()
+
+    def _mock_hotel_search(
+        self,
+        city: str,
+        check_in: str,
+        check_out: str,
+        guests: int,
+        rooms: int,
+        star_rating: Optional[str],
+        price_min: int,
+        price_max: int,
+    ) -> Dict[str, Any]:
+        """模拟酒店搜索（沙箱模式）"""
+        import random
+
+        hotel_types = {
+            "3": ["经济型酒店", "舒适型酒店"],
+            "4": ["高档型酒店", "四星级酒店"],
+            "5": ["豪华型酒店", "五星级酒店"],
+            "luxury": ["奢华酒店", "度假村", "精品酒店"],
+        }
+
+        star_names = {
+            "3": "三星级",
+            "4": "四星级",
+            "5": "五星级",
+            "luxury": "豪华型",
+        }
+
+        amenities_pool = [
+            "免费 WiFi", "游泳池", "健身房", "SPA", "停车场",
+            "餐厅", "会议室", "商务中心", "机场接送", "洗衣服务"
+        ]
+
+        hotels = []
+        target_stars = star_rating if star_rating else random.choice(["3", "4", "5"])
+        hotel_type_names = hotel_types.get(target_stars, ["酒店"])
+
+        for i in range(15):
+            price = random.randint(max(200, price_min), min(price_max, 3000))
+            rating = round(random.uniform(4.0, 5.0), 1)
+
+            hotels.append({
+                "hotel_id": f"hotel_{random.randint(10000, 99999)}",
+                "name": f"{city}{random.choice(['国际', '商务', '精品', '豪华', '经济'])}{random.choice(hotel_type_names)}",
+                "star_rating": target_stars,
+                "star_name": star_names.get(target_stars, "舒适型"),
+                "address": f"{city}{random.choice(['朝阳区', '海淀区', '浦东新区', '天河区', '南山区'])}{random.randint(1, 999)}号",
+                "location": {
+                    "latitude": 39.9 + random.uniform(-0.5, 0.5),
+                    "longitude": 116.4 + random.uniform(-0.5, 0.5),
+                },
+                "price": {
+                    "amount": price,
+                    "currency": "CNY",
+                    "per_night": True,
+                },
+                "amenities": random.sample(amenities_pool, random.randint(3, 7)),
+                "rating": rating,
+                "review_count": random.randint(100, 5000),
+                "images": [
+                    f"https://example.com/hotel_{i}_1.jpg",
+                    f"https://example.com/hotel_{i}_2.jpg",
+                ],
+                "rooms_available": random.randint(1, 20),
+            })
+
+        # 按评分和价格综合排序
+        hotels.sort(key=lambda x: x["rating"] * 100 - x["price"]["amount"] / 100, reverse=True)
+
+        return {
+            "success": True,
+            "search_id": f"search_{datetime.now().strftime('%Y%m%d%H%M%S')}",
+            "city": city,
+            "check_in": check_in,
+            "check_out": check_out,
+            "guests": guests,
+            "rooms": rooms,
+            "total_results": len(hotels),
+            "hotels": hotels[:15],
+            "currency": "CNY",
+        }
+
+    async def search_attractions(
+        self,
+        city: str,
+        category: str = "all",
+        duration: Optional[str] = None,
+    ) -> Dict[str, Any]:
+        """搜索景点/门票
+
+        Args:
+            city: 城市名称
+            category: 景点类型（all/nature/culture/entertainment）
+            duration: 游玩时长
+
+        Returns:
+            景点搜索结果
+        """
+        # 沙箱环境返回模拟数据
+        if self.use_sandbox or not self.app_key:
+            return self._mock_attraction_search(city, category, duration)
+
+        # 实际 API 调用
+        async with httpx.AsyncClient() as client:
+            params = {
+                "app_key": self.app_key,
+                "method": "alitrip.travel.attraction.search",
+                "v": "2.0",
+                "city": city,
+            }
+
+            response = await client.get(
+                f"{self.base_url}/router/rest",
+                params=params,
+                timeout=30,
+            )
+            response.raise_for_status()
+            return response.json()
+
+    def _mock_attraction_search(
+        self,
+        city: str,
+        category: str,
+        duration: Optional[str],
+    ) -> Dict[str, Any]:
+        """模拟景点搜索（沙箱模式）"""
+        import random
+
+        attraction_types = {
+            "nature": ["自然风景区", "国家公园", "湖泊", "山脉", "海滩"],
+            "culture": ["博物馆", "历史遗迹", "寺庙", "古建筑", "文化街区"],
+            "entertainment": ["主题乐园", "动物园", "水族馆", "游乐园", "演艺中心"],
+            "all": ["自然风景区", "博物馆", "主题乐园", "历史遗迹", "寺庙", "国家公园"],
+        }
+
+        category_names = {
+            "nature": "自然风光",
+            "culture": "文化历史",
+            "entertainment": "娱乐休闲",
+            "all": "全部类型",
+        }
+
+        attractions = []
+        attraction_type_names = attraction_types.get(category, attraction_types["all"])
+
+        famous_attractions = {
+            "北京": ["故宫", "长城", "天坛", "颐和园", "圆明园"],
+            "上海": ["外滩", "东方明珠", "迪士尼乐园", "豫园", "南京路"],
+            "西安": ["兵马俑", "大雁塔", "城墙", "华清池", "陕西历史博物馆"],
+            "杭州": ["西湖", "灵隐寺", "宋城", "千岛湖", "西溪湿地"],
+            "成都": ["大熊猫基地", "都江堰", "武侯祠", "杜甫草堂", "青城山"],
+            "三亚": ["亚龙湾", "天涯海角", "南山文化旅游区", "蜈支洲岛", "大小洞天"],
+        }
+
+        # 使用知名景点或生成通用景点
+        if city in famous_attractions:
+            attraction_names = famous_attractions[city]
+        else:
+            attraction_names = [f"{city}{random.choice(['公园', '景区', '博物馆', '乐园'])}" for _ in range(5)]
+
+        for i, name in enumerate(attraction_names):
+            ticket_price = random.randint(50, 500)
+            rating = round(random.uniform(4.0, 5.0), 1)
+
+            attractions.append({
+                "attraction_id": f"attraction_{random.randint(10000, 99999)}",
+                "name": name,
+                "type": random.choice(attraction_type_names),
+                "category": category_names.get(category, "综合"),
+                "description": f"{city}著名景点，{random.choice(['历史悠久', '风景优美', '设施完善', '值得一游'])}",
+                "address": f"{city}{random.choice(['朝阳区', '海淀区', '浦东新区', '西湖区', '南山区'])}",
+                "location": {
+                    "latitude": 39.9 + random.uniform(-0.5, 0.5),
+                    "longitude": 116.4 + random.uniform(-0.5, 0.5),
+                },
+                "ticket": {
+                    "price": ticket_price,
+                    "currency": "CNY",
+                    "type": random.choice(["成人票", "通票", "套票"]),
+                },
+                "rating": rating,
+                "review_count": random.randint(100, 10000),
+                "duration": random.choice(["2-3 小时", "半天", "一天"]),
+                "opening_hours": "09:00-17:00",
+                "images": [
+                    f"https://example.com/attraction_{i}_1.jpg",
+                    f"https://example.com/attraction_{i}_2.jpg",
+                ],
+            })
+
+        # 按评分排序
+        attractions.sort(key=lambda x: x["rating"], reverse=True)
+
+        return {
+            "success": True,
+            "search_id": f"search_{datetime.now().strftime('%Y%m%d%H%M%S')}",
+            "city": city,
+            "category": category_names.get(category, "综合"),
+            "total_results": len(attractions),
+            "attractions": attractions,
+            "currency": "CNY",
+        }
+
+
+# =============================================================================
+# 工具处理器集成
+# =============================================================================
+
+class FliggyToolHandlers:
+    """Fliggy 工具处理器（使用真实 API）"""
+
+    def __init__(self, client: Optional[FliggyMCPClient] = None):
+        self.client = client or FliggyMCPClient(use_sandbox=True)
+
+    async def search_flights(
+        self,
+        origin: str,
+        destination: str,
+        departure_date: Optional[str] = None,
+        return_date: Optional[str] = None,
+        cabin_class: str = "economy",
+    ) -> str:
+        """搜索航班并格式化输出"""
+        result = await self.client.search_flights(
+            origin=origin,
+            destination=destination,
+            departure_date=departure_date or datetime.now().strftime("%Y-%m-%d"),
+            return_date=return_date,
+            cabin_class=cabin_class,
+        )
+
+        if not result.get("success"):
+            return "航班搜索失败，请稍后重试"
+
+        flights = result.get("flights", [])
+        if not flights:
+            return "未找到符合条件的航班"
+
+        output = f"✈️ 航班搜索结果\n\n{origin} → {destination}\n"
+        output += f"出发日期：{departure_date}\n"
+        if return_date:
+            output += f"返程日期：{return_date}\n"
+
+        output += f"\n共找到 {len(flights)} 个航班\n\n"
+
+        for i, flight in enumerate(flights[:5], 1):
+            output += f"{i}. {flight['flight_no']} | {flight['airline']}\n"
+            output += f"   {flight['departure']['time']} - {flight['arrival']['time']} | "
+            stops_text = '直飞' if flight['stops'] == 0 else f"{flight['stops']}次中转"
+            output += f"{stops_text}\n"
+            output += f"   ¥{flight['price']:,} | 剩余{flight['available_seats']}座\n\n"
+
+        return output
+
+    async def search_hotels(
+        self,
+        destination: str,
+        check_in: Optional[str] = None,
+        check_out: Optional[str] = None,
+        guests: int = 2,
+        star_rating: str = "4",
+        price_min: int = 0,
+        price_max: int = 5000,
+    ) -> str:
+        """搜索酒店并格式化输出"""
+        today = datetime.now().strftime("%Y-%m-%d")
+        result = await self.client.search_hotels(
+            city=destination,
+            check_in=check_in or today,
+            check_out=check_out or today,
+            guests=guests,
+            star_rating=star_rating,
+            price_min=price_min,
+            price_max=price_max,
+        )
+
+        if not result.get("success"):
+            return "酒店搜索失败，请稍后重试"
+
+        hotels = result.get("hotels", [])
+        if not hotels:
+            return "未找到符合条件的酒店"
+
+        output = f"🏨 酒店搜索结果\n\n目的地：{destination}\n"
+        output += f"入住：{check_in or '未指定'} - {check_out or '未指定'}\n"
+        output += f"星级：{star_rating}星 | 预算：¥{price_min}-{price_max}\n\n"
+        output += f"共找到 {len(hotels)} 家酒店\n\n"
+
+        for i, hotel in enumerate(hotels[:5], 1):
+            output += f"{i}. 【{hotel['star_name']}】{hotel['name']}\n"
+            output += f"   评分：{hotel['rating']}/5.0 | ¥{hotel['price']['amount']:,}/晚\n"
+            output += f"   位置：{hotel['address']}\n"
+            output += f"   设施：{', '.join(hotel['amenities'][:4])}\n\n"
+
+        return output
+
+    async def search_attractions(
+        self,
+        destination: str,
+        category: str = "all",
+        duration: Optional[str] = None,
+    ) -> str:
+        """搜索景点并格式化输出"""
+        result = await self.client.search_attractions(
+            city=destination,
+            category=category,
+            duration=duration,
+        )
+
+        if not result.get("success"):
+            return "景点搜索失败，请稍后重试"
+
+        attractions = result.get("attractions", [])
+        if not attractions:
+            return "未找到符合条件的景点"
+
+        output = f"🎭 景点搜索结果\n\n目的地：{destination}\n"
+        output += f"类型：{result.get('category', '综合')}\n\n"
+        output += f"共找到 {len(attractions)} 个景点\n\n"
+
+        for i, attraction in enumerate(attractions[:5], 1):
+            output += f"{i}. ⭐⭐⭐⭐⭐ {attraction['name']}\n"
+            output += f"   类型：{attraction['type']} | 游玩：{attraction['duration']}\n"
+            output += f"   门票：¥{attraction['ticket']['price']} | 评分：{attraction['rating']}/5.0\n"
+            output += f"   亮点：{attraction['description']}\n\n"
+
+        return output