travel-agent-cli 0.2.0 → 0.2.1

package/python/agents/product_agent.py

@@ -0,0 +1,672 @@
+"""Product Analysis Agent - 旅行产品可行性分析专家
+
+负责分析特定话题作为旅行产品的潜力，生成详细可行性分析报告
+"""
+from typing import Dict, Any, List, Optional
+import json
+from datetime import datetime
+
+from agents.base import BaseAgent
+from agents.collector_agent import CollectionAgent
+from agents.analysis_agent import AnalysisAgent
+
+
+class ProductAnalysisAgent(BaseAgent):
+    """旅行产品可行性分析 Agent
+
+    职责：
+    - 接收用户给定的话题（如"北欧极光"、"东南亚潜水"等）
+    - 在社交媒体和新闻中搜索相关资料
+    - 分析该话题作为旅行产品的市场潜力
+    - 生成详细的可行性分析报告
+
+    分析维度：
+    1. 市场需求分析（社交媒体热度、搜索趋势）
+    2. 资源供给分析（航线、酒店、景点）
+    3. 竞争格局分析（现有产品、价格区间）
+    4. 风险评估（政策、安全、季节性）
+    5. 商业价值评估（利润空间、目标客群）
+    """
+
+    name = "product_analysis_agent"
+    role = "旅行产品可行性分析专家"
+    goal = "深度分析特定旅行话题的商业潜力，输出专业的可行性分析报告"
+
+    def __init__(
+        self,
+        provider: Optional[str] = None,
+        model: Optional[str] = None,
+        use_tools: bool = False,
+    ):
+        super().__init__(provider, model, use_tools)
+
+        # 子 Agent 用于数据采集和初步分析
+        self.collector = CollectionAgent(provider, model, use_tools)
+        self.analyst = AnalysisAgent(provider, model, use_tools)
+
+    async def execute_local(self, task: str, context: Dict[str, Any]) -> str:
+        """本地执行（降级模式）"""
+        topic = context.get("topic", "未知话题")
+        return f"Product Analysis Agent 本地执行：分析话题'{topic}'的可行性（模拟报告）"
+
+    async def analyze_topic(
+        self,
+        topic: str,
+        include_news: bool = True,
+        include_ota: bool = True,
+        verbose: bool = True
+    ) -> Dict[str, Any]:
+        """分析指定话题的旅行产品可行性
+
+        Args:
+            topic: 分析话题，如"北欧极光"、"东南亚潜水"、"日本赏樱"等
+            include_news: 是否包含新闻资讯分析
+            include_ota: 是否包含 OTA 数据分析
+            verbose: 是否输出详细日志
+
+        Returns:
+            完整分析结果字典
+        """
+        result = {
+            "topic": topic,
+            "status": "analyzing",
+            "started_at": datetime.now().isoformat(),
+            "stages": {}
+        }
+
+        if verbose:
+            print(f"\n[{self.name}] 🚀 开始分析话题：[bold]{topic}[/bold]")
+
+        # ========== Stage 1: 数据采集 ==========
+        if verbose:
+            print(f"\n[{self.name}] 📥 Stage 1: 数据采集")
+
+        # 构建搜索关键词
+        keywords = self._expand_keywords(topic)
+
+        collection_result = await self.collector.collect_all(
+            keyword=keywords["primary"],
+            include_ota=include_ota
+        )
+
+        result["stages"]["collection"] = {
+            "keywords_used": keywords,
+            "social_media_count": len(collection_result.get("social_media", {}).get("posts", [])),
+            "wenlv_count": len(collection_result.get("wenlv", {}).get("infos", [])),
+            "flight_count": len(collection_result.get("ota", {}).get("flights", [])),
+            "hotel_count": len(collection_result.get("ota", {}).get("hotels", [])),
+        }
+
+        if verbose:
+            print(f"  ✓ 搜索关键词：{keywords['primary']}")
+            print(f"  ✓ 社交媒体：{result['stages']['collection']['social_media_count']} 条")
+            print(f"  ✓ 文旅信息：{result['stages']['collection']['wenlv_count']} 条")
+            if include_ota:
+                print(f"  ✓ 航班：{result['stages']['collection']['flight_count']} 条")
+                print(f"  ✓ 酒店：{result['stages']['collection']['hotel_count']} 条")
+
+        # ========== Stage 2: 市场需求分析 ==========
+        if verbose:
+            print(f"\n[{self.name}] 📊 Stage 2: 市场需求分析")
+
+        posts = collection_result.get("social_media", {}).get("posts", [])
+        wenlv_infos = collection_result.get("wenlv", {}).get("infos", [])
+
+        # 分析社交媒体热度
+        social_hotspot_analysis = await self._analyze_social_hotspot(posts, topic)
+        result["stages"]["market_demand"] = social_hotspot_analysis
+
+        if verbose:
+            print(f"  ✓ 社交媒体热度：{social_hotspot_analysis.get('heat_level', 'unknown')}")
+            print(f"  ✓ 互动总量：{social_hotspot_analysis.get('total_engagements', 0):,}")
+            print(f"  ✓ 相关内容：{social_hotspot_analysis.get('content_count', 0)} 条")
+
+        # ========== Stage 3: 资源供给分析 ==========
+        if verbose:
+            print(f"\n[{self.name}] 🏨 Stage 3: 资源供给分析")
+
+        resource_analysis = await self._analyze_resources(
+            topic=topic,
+            flights=collection_result.get("ota", {}).get("flights", []),
+            hotels=collection_result.get("ota", {}).get("hotels", []),
+            wenlv_infos=wenlv_infos
+        )
+        result["stages"]["resources"] = resource_analysis
+
+        if verbose:
+            print(f"  ✓ 航线丰富度：{resource_analysis.get('flight_availability', 'unknown')}")
+            print(f"  ✓ 酒店选择：{resource_analysis.get('hotel_options', 'unknown')}")
+            print(f"  ✓ 预估费用：{resource_analysis.get('estimated_cost_range', '未知')}")
+
+        # ========== Stage 4: 综合评估 ==========
+        if verbose:
+            print(f"\n[{self.name}] 📈 Stage 4: 综合评估")
+
+        assessment = await self._comprehensive_assessment(
+            topic=topic,
+            market_demand=social_hotspot_analysis,
+            resources=resource_analysis,
+            posts=posts,
+            wenlv_infos=wenlv_infos
+        )
+        result["stages"]["assessment"] = assessment
+
+        if verbose:
+            print(f"  ✓ 市场潜力：{assessment.get('market_potential', 'unknown')}")
+            print(f"  ✓ 推荐指数：{assessment.get('recommendation_score', 0)}/10")
+            print(f"  ✓ 风险等级：{assessment.get('risk_level', 'unknown')}")
+
+        # ========== Stage 5: 生成报告 ==========
+        if verbose:
+            print(f"\n[{self.name}] 📝 Stage 5: 生成分析报告")
+
+        report = await self._generate_report(
+            topic=topic,
+            collection=result["stages"]["collection"],
+            market_demand=result["stages"]["market_demand"],
+            resources=result["stages"]["resources"],
+            assessment=result["stages"]["assessment"]
+        )
+        result["stages"]["report"] = {
+            "content": report,
+            "generated_at": datetime.now().isoformat()
+        }
+
+        if verbose:
+            print(f"  ✓ 报告长度：{len(report)} 字符")
+
+        # ========== 完成 ==========
+        result["status"] = "completed"
+        result["completed_at"] = datetime.now().isoformat()
+
+        if verbose:
+            print(f"\n[{self.name}] ✅ 分析完成！")
+
+        return result
+
+    def _expand_keywords(self, topic: str) -> Dict[str, str]:
+        """扩展搜索关键词
+
+        Args:
+            topic: 原始话题
+
+        Returns:
+            关键词字典
+        """
+        # 简单扩展逻辑，实际可接入 NLP 模型
+        keyword_map = {
+            "极光": {"primary": "极光旅行", "related": "北极光 南极光 极光摄影"},
+            "潜水": {"primary": "潜水度假", "related": "水肺潜水 浮潜 潜水考证"},
+            "滑雪": {"primary": "滑雪旅行", "related": "滑雪场 滑雪度假 粉雪"},
+            "樱花": {"primary": "赏樱旅行", "related": "樱花季 樱花前线 日本樱花"},
+            "海岛": {"primary": "海岛度假", "related": "海滩 浮潜 海岛游"},
+        }
+
+        for key, value in keyword_map.items():
+            if key in topic:
+                return {
+                    "primary": value["primary"],
+                    "related": value["related"],
+                    "original": topic
+                }
+
+        return {
+            "primary": topic,
+            "related": f"{topic}攻略 {topic}推荐",
+            "original": topic
+        }
+
+    async def _analyze_social_hotspot(
+        self,
+        posts: List[Dict],
+        topic: str
+    ) -> Dict[str, Any]:
+        """分析社交媒体热度
+
+        Args:
+            posts: 帖子列表
+            topic: 话题
+
+        Returns:
+            热度分析结果
+        """
+        if not posts:
+            return {
+                "heat_level": "低",
+                "total_engagements": 0,
+                "content_count": 0,
+                "trend": "平稳",
+                "key_topics": []
+            }
+
+        # 计算总互动量
+        total_likes = sum(p.get("likes", 0) for p in posts)
+        total_comments = sum(p.get("comments", 0) for p in posts)
+        total_shares = sum(p.get("shares", 0) for p in posts)
+        total_engagements = total_likes + total_comments + total_shares
+
+        # 判断热度等级
+        if total_engagements > 100000:
+            heat_level = "非常高"
+        elif total_engagements > 10000:
+            heat_level = "高"
+        elif total_engagements > 1000:
+            heat_level = "中等"
+        else:
+            heat_level = "低"
+
+        # 提取关键话题/标签
+        tags = []
+        for post in posts:
+            if post.get("tags"):
+                tags.extend(post["tags"][:3])
+
+        # 统计高频标签
+        from collections import Counter
+        tag_counts = Counter(tags)
+        key_topics = [tag for tag, _ in tag_counts.most_common(5)]
+
+        return {
+            "heat_level": heat_level,
+            "total_engagements": total_engagements,
+            "total_likes": total_likes,
+            "total_comments": total_comments,
+            "total_shares": total_shares,
+            "content_count": len(posts),
+            "trend": "上升" if total_engagements > 5000 else "平稳",
+            "key_topics": key_topics,
+            "avg_engagement_per_post": total_engagements / len(posts) if posts else 0
+        }
+
+    async def _analyze_resources(
+        self,
+        topic: str,
+        flights: List[Dict],
+        hotels: List[Dict],
+        wenlv_infos: List[Dict]
+    ) -> Dict[str, Any]:
+        """分析资源供给情况
+
+        Args:
+            topic: 话题
+            flights: 航班数据
+            hotels: 酒店数据
+            wenlv_infos: 文旅信息
+
+        Returns:
+            资源分析结果
+        """
+        # 航班分析
+        if flights:
+            flight_prices = [f.get("price", 0) for f in flights if f.get("price")]
+            avg_flight_price = sum(flight_prices) / len(flight_prices) if flight_prices else 0
+            flight_availability = "丰富" if len(flights) > 20 else "一般" if len(flights) > 5 else "较少"
+        else:
+            avg_flight_price = 0
+            flight_availability = "未知"
+
+        # 酒店分析
+        if hotels:
+            hotel_prices = [h.get("price", 0) for h in hotels if h.get("price")]
+            avg_hotel_price = sum(hotel_prices) / len(hotel_prices) if hotel_prices else 0
+            hotel_options = "丰富" if len(hotels) > 50 else "一般" if len(hotels) > 10 else "较少"
+        else:
+            avg_hotel_price = 0
+            hotel_options = "未知"
+
+        # 文旅政策分析
+        policy_support = "有" if wenlv_infos and len(wenlv_infos) > 3 else "暂无"
+
+        # 预估费用范围
+        if avg_flight_price > 0 and avg_hotel_price > 0:
+            min_cost = avg_flight_price + avg_hotel_price * 3
+            max_cost = avg_flight_price * 2 + avg_hotel_price * 7
+            estimated_cost_range = f"¥{int(min_cost)}-{int(max_cost)}"
+        else:
+            estimated_cost_range = "需进一步调研"
+
+        return {
+            "flight_availability": flight_availability,
+            "flight_count": len(flights),
+            "avg_flight_price": avg_flight_price,
+            "hotel_options": hotel_options,
+            "hotel_count": len(hotels),
+            "avg_hotel_price": avg_hotel_price,
+            "policy_support": policy_support,
+            "wenlv_count": len(wenlv_infos),
+            "estimated_cost_range": estimated_cost_range
+        }
+
+    async def _comprehensive_assessment(
+        self,
+        topic: str,
+        market_demand: Dict[str, Any],
+        resources: Dict[str, Any],
+        posts: List[Dict],
+        wenlv_infos: List[Dict]
+    ) -> Dict[str, Any]:
+        """综合评估
+
+        Args:
+            topic: 话题
+            market_demand: 市场需求分析
+            resources: 资源分析
+            posts: 帖子列表
+            wenlv_infos: 文旅信息
+
+        Returns:
+            综合评估结果
+        """
+        # 市场潜力评分（1-10）
+        heat_scores = {
+            "非常高": 9,
+            "高": 7,
+            "中等": 5,
+            "低": 3
+        }
+        market_score = heat_scores.get(market_demand.get("heat_level", "低"), 5)
+
+        # 资源成熟度评分
+        resource_score = 5
+        if resources.get("flight_availability") == "丰富":
+            resource_score += 2
+        elif resources.get("flight_availability") == "一般":
+            resource_score += 1
+
+        if resources.get("hotel_options") == "丰富":
+            resource_score += 2
+        elif resources.get("hotel_options") == "一般":
+            resource_score += 1
+
+        if resources.get("policy_support") == "有":
+            resource_score += 1
+
+        # 综合推荐指数
+        recommendation_score = round((market_score + resource_score) / 2, 1)
+
+        # 风险等级评估
+        if recommendation_score >= 8:
+            risk_level = "低"
+            market_potential = "非常高"
+        elif recommendation_score >= 6:
+            risk_level = "中"
+            market_potential = "高"
+        else:
+            risk_level = "高"
+            market_potential = "一般"
+
+        # 目标客群分析
+        target_audience = self._analyze_target_audience(posts)
+
+        # SWOT 分析要点
+        swot = await self._generate_swot(topic, market_demand, resources)
+
+        return {
+            "market_score": market_score,
+            "resource_score": resource_score,
+            "recommendation_score": recommendation_score,
+            "market_potential": market_potential,
+            "risk_level": risk_level,
+            "target_audience": target_audience,
+            "swot": swot
+        }
+
+    def _analyze_target_audience(self, posts: List[Dict]) -> Dict[str, Any]:
+        """分析目标客群画像"""
+        # 简化实现，实际可接入用户画像分析
+        if not posts:
+            return {
+                "age_group": "未知",
+                "interests": [],
+                "consumption_level": "未知"
+            }
+
+        # 基于互动数据推测
+        avg_engagement = sum(p.get("likes", 0) + p.get("comments", 0) for p in posts) / len(posts)
+
+        if avg_engagement > 1000:
+            consumption_level = "高"
+        elif avg_engagement > 100:
+            consumption_level = "中等"
+        else:
+            consumption_level = "大众"
+
+        return {
+            "age_group": "25-40 岁",
+            "interests": ["旅行", "摄影", "美食", "文化体验"],
+            "consumption_level": consumption_level
+        }
+
+    async def _generate_swot(
+        self,
+        topic: str,
+        market_demand: Dict[str, Any],
+        resources: Dict[str, Any]
+    ) -> Dict[str, List[str]]:
+        """生成 SWOT 分析"""
+        swot = {
+            "strengths": [],
+            "weaknesses": [],
+            "opportunities": [],
+            "threats": []
+        }
+
+        # Strengths
+        if market_demand.get("heat_level") in ["高", "非常高"]:
+            swot["strengths"].append("社交媒体关注度高，市场需求旺盛")
+        if resources.get("flight_availability") == "丰富":
+            swot["strengths"].append("航线资源丰富，交通便利")
+        if resources.get("hotel_options") == "丰富":
+            swot["strengths"].append("住宿选择多样，接待能力强")
+
+        # Weaknesses
+        if resources.get("flight_availability") in ["较少", "未知"]:
+            swot["weaknesses"].append("航线资源有限，交通可达性待提升")
+        if market_demand.get("total_engagements", 0) < 1000:
+            swot["weaknesses"].append("市场认知度较低，需要加强推广")
+
+        # Opportunities
+        if resources.get("policy_support") == "有":
+            swot["opportunities"].append("文旅政策支持，有官方背书")
+        swot["opportunities"].append("后疫情时代旅行需求反弹")
+        swot["opportunities"].append("社交媒体传播效应显著")
+
+        # Threats
+        swot["threats"].append("同类旅行产品竞争")
+        swot["threats"].append("季节性因素影响")
+        swot["threats"].append("政策变化风险")
+
+        return swot
+
+    async def _generate_report(
+        self,
+        topic: str,
+        collection: Dict[str, Any],
+        market_demand: Dict[str, Any],
+        resources: Dict[str, Any],
+        assessment: Dict[str, Any]
+    ) -> str:
+        """生成完整分析报告"""
+        report = f"""# 旅行产品可行性分析报告
+
+## 分析话题：{topic}
+
+**报告生成时间**: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}
+
+---
+
+## 📊 一、执行摘要
+
+### 核心结论
+
+| 评估维度 | 评分 | 等级 |
+|---------|------|------|
+| 市场潜力 | {assessment.get('market_score', 0)}/10 | {assessment.get('market_potential', '未知')} |
+| 资源成熟度 | {assessment.get('resource_score', 0)}/10 | - |
+| **推荐指数** | **{assessment.get('recommendation_score', 0)}/10** | {assessment.get('market_potential', '未知')} |
+| 风险等级 | - | {assessment.get('risk_level', '未知')} |
+
+### 采集数据概览
+
+- 社交媒体内容：{collection.get('social_media_count', 0)} 条
+- 文旅政策信息：{collection.get('wenlv_count', 0)} 条
+- 航班信息：{collection.get('flight_count', 0)} 条
+- 酒店信息：{collection.get('hotel_count', 0)} 条
+
+---
+
+## 🔥 二、市场需求分析
+
+### 社交媒体热度
+
+| 指标 | 数值 |
+|------|------|
+| 热度等级 | {market_demand.get('heat_level', '未知')} |
+| 总互动量 | {market_demand.get('total_engagements', 0):,} |
+| 总点赞数 | {market_demand.get('total_likes', 0):,} |
+| 总评论数 | {market_demand.get('total_comments', 0):,} |
+| 总分享数 | {market_demand.get('total_shares', 0):,} |
+| 篇均互动 | {market_demand.get('avg_engagement_per_post', 0):.1f} |
+| 趋势判断 | {market_demand.get('trend', '平稳')} |
+
+### 热门话题标签
+
+{', '.join(assessment.get('key_topics', []) or market_demand.get('key_topics', ['暂无数据']))}
+
+---
+
+## 🏨 三、资源供给分析
+
+### 交通资源
+
+| 指标 | 数值 |
+|------|------|
+| 航线丰富度 | {resources.get('flight_availability', '未知')} |
+| 航班数量 | {resources.get('flight_count', 0)} 条 |
+| 平均机票价格 | ¥{resources.get('avg_flight_price', 0):.0f} |
+
+### 住宿资源
+
+| 指标 | 数值 |
+|------|------|
+| 酒店选择 | {resources.get('hotel_options', '未知')} |
+| 酒店数量 | {resources.get('hotel_count', 0)} 家 |
+| 平均酒店价格 | ¥{resources.get('avg_hotel_price', 0):.0f}/晚 |
+
+### 政策支持
+
+- 文旅政策信息：{resources.get('wenlv_count', 0)} 条
+- 政策支持：{resources.get('policy_support', '暂无')}
+
+### 预估费用范围
+
+**{resources.get('estimated_cost_range', '需进一步调研')}**（按 3-7 天行程估算）
+
+---
+
+## 👥 四、目标客群画像
+
+| 维度 | 特征 |
+|------|------|
+| 年龄段 | {assessment.get('target_audience', {}).get('age_group', '未知')} |
+| 消费水平 | {assessment.get('target_audience', {}).get('consumption_level', '未知')} |
+| 兴趣标签 | {', '.join(assessment.get('target_audience', {}).get('interests', []))} |
+
+---
+
+## 📈 五、SWOT 分析
+
+### 优势 (Strengths)
+{chr(10).join('✓ ' + s for s in assessment.get('swot', {}).get('strengths', ['暂无']))}
+
+### 劣势 (Weaknesses)
+{chr(10).join('✗ ' + w for w in assessment.get('swot', {}).get('weaknesses', ['暂无']))}
+
+### 机会 (Opportunities)
+{chr(10).join('✓ ' + o for o in assessment.get('swot', {}).get('opportunities', ['暂无']))}
+
+### 威胁 (Threats)
+{chr(10).join('⚠ ' + t for t in assessment.get('swot', {}).get('threats', ['暂无']))}
+
+---
+
+## 💡 六、产品开发建议
+
+### 推荐产品类型
+
+基于分析结果，建议开发以下类型的旅行产品：
+
+1. **核心产品**: 围绕"{topic}"主题的深度体验产品
+2. **差异化亮点**: 结合社交媒体热门元素，打造网红打卡点
+3. **价格策略**: {resources.get('estimated_cost_range', '待调研')}的中高端定位
+
+### 开发优先级
+
+| 优先级 | 事项 | 说明 |
+|--------|------|------|
+| P0 | 供应链确认 | 落实航线、酒店、地接资源 |
+| P1 | 产品设计 | 规划行程路线和体验项目 |
+| P2 | 定价策略 | 基于成本和竞品定价 |
+| P3 | 营销素材 | 制作社交媒体推广内容 |
+
+### 风险缓释措施
+
+1. **季节性风险**: 设计多季节产品线，降低单一季节依赖
+2. **政策风险**: 密切关注文旅政策变化，保持灵活调整
+3. **竞争风险**: 打造差异化体验，建立品牌壁垒
+
+---
+
+## 🎯 七、结论与建议
+
+### 最终结论
+
+**推荐指数：{assessment.get('recommendation_score', 0)}/10**
+
+话题"{topic}"作为旅行产品的开发可行性评估为：**{assessment.get('market_potential', '未知')}**
+
+- 市场热度：{market_demand.get('heat_level', '未知')}
+- 资源成熟度：{resources.get('flight_availability', '未知')} / {resources.get('hotel_options', '未知')}
+- 风险等级：{assessment.get('risk_level', '未知')}
+
+### 行动建议
+
+{
+"建议立即启动产品开发" if assessment.get('recommendation_score', 0) >= 7 else
+"建议进一步市场调研后决策" if assessment.get('recommendation_score', 0) >= 5 else
+"建议谨慎评估，暂不优先开发"
+}
+
+---
+
+*本报告由 AI 旅行产品分析 Agent 生成，仅供参考。实际决策请结合专业市场调研和商业判断。*
+"""
+        return report
+
+    async def quick_analysis(self, topic: str) -> str:
+        """快速分析（简化版）
+
+        Args:
+            topic: 分析话题
+
+        Returns:
+            简化版分析报告
+        """
+        result = await self.analyze_topic(
+            topic=topic,
+            include_ota=False,
+            verbose=False
+        )
+
+        assessment = result.get("stages", {}).get("assessment", {})
+        return f"""【快速分析】话题：{topic}
+
+推荐指数：{assessment.get('recommendation_score', 0)}/10
+市场潜力：{assessment.get('market_potential', '未知')}
+风险等级：{assessment.get('risk_level', '未知')}
+
+核心结论：{
+"建议重点开发" if assessment.get('recommendation_score', 0) >= 7 else
+"建议进一步调研" if assessment.get('recommendation_score', 0) >= 5 else
+"建议谨慎评估"
+}"""