travel-agent-cli 0.2.0 → 0.2.1

package/python/agents/planning_agent.py

@@ -0,0 +1,161 @@
+"""规划 Agent - 负责旅行路线规划"""
+from typing import Dict, Any, List, Optional
+import json
+from agents.base import BaseAgent
+
+
+class PlanningAgent(BaseAgent):
+    """规划 Agent
+
+    职责：
+    - 为目的地生成详细行程
+    - 估算预算
+    - 提供旅行建议
+    """
+
+    name = "planning_agent"
+    role = "资深旅行规划师"
+    goal = "为推荐目的地设计详细、实用、高性价比的旅行路线"
+
+    def __init__(
+        self,
+        provider: Optional[str] = None,
+        model: Optional[str] = None,
+        use_tools: bool = False,
+    ):
+        super().__init__(provider, model, use_tools)
+
+    async def execute_local(self, task: str, context: Dict[str, Any]) -> str:
+        """本地执行路线规划"""
+        destination = context.get("destination", "目的地")
+        days = context.get("days", 3)
+
+        result = {
+            "destination": destination,
+            "duration_days": days,
+            "daily_plan": [
+                {
+                    "day": i + 1,
+                    "theme": f"第{i + 1}天探索",
+                    "activities": [
+                        {"time": "上午", "item": "景点游览", "duration": "3 小时"},
+                        {"time": "下午", "item": "深度体验", "duration": "4 小时"},
+                        {"time": "晚上", "item": "美食探索", "duration": "2 小时"},
+                    ]
+                }
+                for i in range(days)
+            ],
+            "budget_estimate": {
+                "accommodation": "¥300-500/晚",
+                "food": "¥100-200/天",
+                "transport": "¥50-100/天",
+                "activities": "¥100-300/天",
+                "total": f"¥{(300 + 100 + 50 + 100) * days}-{(500 + 200 + 100 + 300) * days}"
+            }
+        }
+
+        return json.dumps(result, ensure_ascii=False, indent=2)
+
+    async def plan_route(
+        self,
+        destination: Dict[str, Any],
+        days: int = None
+    ) -> Dict[str, Any]:
+        """为单个目的地规划路线
+
+        Args:
+            destination: 目的地信息（包含 name, suggested_days 等）
+            days: 游玩天数，如不传则从 destination 中获取
+
+        Returns:
+            路线规划结果
+        """
+        dest_name = destination.get("name", "未知目的地")
+        travel_days = days or destination.get("suggested_days", 3)
+
+        task = f"""请为以下目的地设计详细的 {travel_days} 天旅行路线：
+
+目的地信息：
+- 名称：{dest_name}
+- 建议天数：{travel_days}天
+- 推荐理由：{destination.get("reason", "")}
+- 亮点：{", ".join(destination.get("highlights", []))}
+
+请设计包含以下内容的路线：
+1. 每日详细行程（上午/下午/晚上的活动）
+2. 每个景点的建议游玩时长
+3. 用餐推荐（当地特色美食）
+4. 住宿区域建议
+5. 交通建议
+6. 预算估算（住宿、餐饮、交通、活动、总计）
+
+请输出 JSON 格式：
+{{
+    "destination": "{dest_name}",
+    "duration_days": {travel_days},
+    "daily_plan": [
+        {{
+            "day": 1,
+            "theme": "今日主题",
+            "activities": [
+                {{"time": "上午", "item": "景点/活动", "duration": "时长", "description": "描述"}},
+                {{"time": "下午", "item": "景点/活动", "duration": "时长", "description": "描述"}},
+                {{"time": "晚上", "item": "景点/活动", "duration": "时长", "description": "描述"}}
+            ],
+            "food_recommendation": "用餐建议",
+            "accommodation_area": "住宿区域建议",
+            "transport_tips": "交通建议"
+        }}
+    ],
+    "budget_estimate": {{
+        "accommodation": "住宿预算",
+        "food": "餐饮预算",
+        "transport": "交通预算",
+        "activities": "活动预算",
+        "tips": "省钱技巧",
+        "total": "总计"
+    }},
+    "travel_tips": ["出行提示 1", "出行提示 2"]
+}}
+"""
+
+        result_str = await self.execute(task, {"destination": destination})
+
+        # 尝试解析 JSON
+        try:
+            # 清理可能的 markdown 格式
+            clean_result = result_str.strip()
+            if clean_result.startswith("```json"):
+                clean_result = clean_result[7:]
+            if clean_result.endswith("```"):
+                clean_result = clean_result[:-3]
+            return json.loads(clean_result.strip())
+        except json.JSONDecodeError:
+            # 解析失败，返回本地规划结果
+            return json.loads(await self.execute_local(task, {
+                "destination": dest_name,
+                "days": travel_days
+            }))
+
+    async def plan_top_destinations(
+        self,
+        destinations: List[Dict[str, Any]],
+        top_n: int = 3
+    ) -> List[Dict[str, Any]]:
+        """为 TOP 目的地规划路线
+
+        Args:
+            destinations: 目的地列表
+            top_n: 规划前 N 个目的地
+
+        Returns:
+            路线规划列表
+        """
+        routes = []
+
+        for dest in destinations[:top_n]:
+            print(f"[{self.name}] 规划路线：{dest.get('name')}")
+            route = await self.plan_route(dest)
+            routes.append(route)
+
+        return routes