travel-agent-cli 0.2.0 → 0.2.2

package/python/main.py

@@ -0,0 +1,632 @@
+# -*- coding: utf-8 -*-
+"""旅行 Agent CLI - 新入口（参考 Claude Code 设计）
+
+设计原则：
+1. 简洁的命令定义
+2. 清晰的帮助信息
+3. 支持交互和非交互模式
+4. 统一的输出格式
+
+使用方法:
+    python main.py <command> [args]
+
+    或直接运行:
+    travel-agent <command> [args]
+"""
+import sys
+import asyncio
+from typing import Optional, List
+from pathlib import Path
+
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+from rich import box
+
+from cli.commands import CommandRegistry, Command, OutputFormatter
+from config.settings import get_settings
+
+# 创建命令注册表
+registry = CommandRegistry()
+console = Console()
+
+
+# =============================================================================
+# 命令定义（参考 Claude Code 的命令模式）
+# =============================================================================
+
+@registry.command(
+    name="run",
+    aliases=["start", "execute"],
+    description="运行完整工作流：采集数据 → 分析评分 → 路线规划 → 报告生成",
+    argument_hint="[options]",
+    immediate=True,
+)
+async def cmd_run(keyword: str = "旅行", top_n: int = 10, plan_top_n: int = 3, no_ota: bool = False):
+    """运行完整工作流"""
+    from agents.manager_agent import ManagerAgent
+
+    console.print(Panel(
+        f"关键词：[bold]{keyword}[/bold]\n"
+        f"推荐数量：[bold]{top_n}[/bold]\n"
+        f"路线规划：[bold]{plan_top_n}[/bold]\n"
+        f"OTA 采集：[bold]{'关闭' if no_ota else '开启'}[/bold]",
+        title="🚀 运行工作流",
+        border_style="blue",
+    ))
+
+    manager = ManagerAgent()
+    result = await manager.run_workflow(
+        keyword=keyword,
+        include_ota=not no_ota,
+        top_n=top_n,
+        plan_top_n=plan_top_n,
+        verbose=True,
+    )
+
+    # 显示结果摘要
+    _show_result_summary(result)
+
+
+@registry.command(
+    name="analyze",
+    aliases=["research"],
+    description="分析旅行产品可行性：市场调研 → 资源评估 → SWOT 分析 → 生成报告",
+    argument_hint="<topic> [options]",
+    immediate=True,
+)
+async def cmd_analyze(topic: str, no_ota: bool = False, output: Optional[str] = None):
+    """分析旅行产品可行性"""
+    from agents.product_agent import ProductAnalysisAgent
+
+    console.print(Panel(
+        f"分析话题：[bold]{topic}[/bold]\n"
+        f"OTA 采集：[bold]{'关闭' if no_ota else '开启'}[/bold]",
+        title="🔍 产品可行性分析",
+        border_style="cyan",
+    ))
+
+    agent = ProductAnalysisAgent()
+    result = await agent.analyze_topic(
+        topic=topic,
+        include_news=True,
+        include_ota=not no_ota,
+        verbose=True,
+    )
+
+    _show_product_analysis_result(result)
+
+    # 保存报告
+    report_content = result.get("stages", {}).get("report", {}).get("content", "")
+    if report_content and output:
+        output_path = Path(output)
+        output_path.parent.mkdir(parents=True, exist_ok=True)
+        with open(output_path, "w", encoding="utf-8") as f:
+            f.write(report_content)
+        console.print(f"\n[green]✓ 报告已保存：{output_path}[/green]")
+
+
+@registry.command(
+    name="model",
+    description="管理 LLM 模型配置：查看/切换 提供商和模型",
+    argument_hint="<subcommand> [options]",
+    immediate=True,
+)
+async def cmd_model(action: Optional[str] = None, provider: Optional[str] = None, model_name: Optional[str] = None):
+    """管理 LLM 模型配置"""
+    settings = get_settings()
+
+    if action == "list" or action is None:
+        _show_model_list(settings)
+    elif action == "status":
+        _show_model_status(settings)
+    elif action == "use":
+        if not provider:
+            console.print("[red]错误：请使用 -p 指定提供商[/red]")
+            return
+        _switch_model(settings, provider, model_name)
+    else:
+        console.print(f"[red]未知操作：{action}[/red]")
+        console.print("使用 `travel-agent model list` 查看所有支持的模型")
+
+
+@registry.command(
+    name="config",
+    aliases=["settings"],
+    description="管理配置：显示当前配置或初始化配置文件",
+    argument_hint="[options]",
+    immediate=True,
+)
+async def cmd_config(show: bool = False, init: bool = False):
+    """管理配置"""
+    settings = get_settings()
+
+    if init:
+        _init_config()
+
+    if show:
+        _show_config(settings)
+
+    if not show and not init:
+        console.print("使用方法：travel-agent config [选项]")
+        console.print("  --show  显示当前配置")
+        console.print("  --init  初始化配置文件")
+
+
+@registry.command(
+    name="status",
+    aliases=["health"],
+    description="显示系统状态：版本、模型、API 配置、数据目录",
+    immediate=True,
+)
+async def cmd_status():
+    """显示系统状态"""
+    settings = get_settings()
+
+    console.print(Panel(
+        f"版本：[bold]0.2.0[/bold]\n"
+        f"提供商：[bold]{settings.llm_provider}[/bold]\n"
+        f"模型：[bold]{settings.get_active_model()}[/bold]\n"
+        f"API 已配置：{'是' if settings.get_active_api_key() else '否'}\n"
+        f"可用提供商：{', '.join(settings.get_configured_providers()) or '无'}",
+        title="📊 系统状态",
+        border_style="green",
+    ))
+
+
+@registry.command(
+    name="flyai",
+    aliases=["search", "travel"],
+    description="FlyAI 旅行搜索：搜索航班、酒店、景点和行程规划",
+    argument_hint="<query> [options]",
+    immediate=True,
+)
+async def cmd_flyai(query: Optional[str] = None, destination: Optional[str] = None, duration: Optional[str] = None, budget: Optional[str] = None):
+    """FlyAI 旅行搜索
+
+    使用自然语言搜索旅行相关信息，包括：
+    - 🔍 综合旅行搜索
+    - ✈️ 航班价格查询
+    - 🏨 酒店搜索
+    - 🎭 景点推荐
+
+    示例:
+        travel-agent flyai "5 天日本行程规划"
+        travel-agent flyai "北京到上海航班" --destination 上海
+        travel-agent flyai "三亚酒店" --destination 三亚
+        travel-agent flyai "东京景点推荐" --destination 东京
+    """
+    from tools.flyai_tools import FlyAIToolHandlers, build_flyai_tools
+
+    if not query:
+        console.print("[red]错误：请提供搜索查询[/red]")
+        console.print("示例：travel-agent flyai \"5 天日本行程规划\"")
+        return
+
+    console.print(Panel(
+        f"查询：[bold]{query}[/bold]\n"
+        f"目的地：[bold]{destination or '未指定'}[/bold]\n"
+        f"行程：[bold]{duration or '未指定'}[/bold]\n"
+        f"预算：[bold]{budget or '未指定'}[/bold]",
+        title="🔍 FlyAI 旅行搜索",
+        border_style="blue",
+    ))
+
+    handler = FlyAIToolHandlers()
+
+    # 根据查询内容自动选择工具
+    query_lower = query.lower()
+    if any(kw in query_lower for kw in ["航班", "flight", "fly", "机票", "机场"]):
+        # 航班搜索
+        result = await handler.handle_search_flights(
+            origin=query.replace("航班", "").replace("flight", "").strip(),
+            destination=destination or "未指定",
+            departure_date=duration,
+        )
+    elif any(kw in query_lower for kw in ["酒店", "hotel", "住宿", "民宿"]):
+        # 酒店搜索
+        result = await handler.handle_search_hotels(
+            destination=destination or query.replace("酒店", "").replace("hotel", "").strip(),
+            check_in=duration,
+        )
+    elif any(kw in query_lower for kw in ["景点", "attraction", "play", "玩", "旅游"]):
+        # 景点搜索
+        result = await handler.handle_search_attractions(
+            destination=destination or query,
+        )
+    else:
+        # 综合旅行搜索
+        result = await handler.handle_travel_search(
+            query=query,
+            destination=destination,
+            duration=duration,
+            budget=budget,
+        )
+
+    console.print(result)
+
+
+@registry.command(
+    name="agents",
+    aliases=["team"],
+    description="查看 Agent 团队信息",
+    argument_hint="[action]",
+    immediate=True,
+)
+async def cmd_agents(action: Optional[str] = "info"):
+    """查看 Agent 团队信息"""
+    from agents.manager_agent import ManagerAgent
+
+    if action == "info":
+        manager = ManagerAgent()
+        team_status = manager.get_team_status()
+        _show_team_info(team_status)
+    else:
+        console.print("使用方法：travel-agent agents [info]")
+
+
+@registry.command(
+    name="help",
+    description="显示帮助信息",
+    argument_hint="[command]",
+    immediate=True,
+    supports_non_interactive=True,
+)
+async def cmd_help(command_name: Optional[str] = None):
+    """显示帮助信息"""
+    if command_name:
+        cmd = registry.get(command_name)
+        if cmd:
+            console.print(OutputFormatter.format_command_help(cmd))
+        else:
+            console.print(f"[red]未知命令：{command_name}[/red]")
+    else:
+        _show_help()
+
+
+# =============================================================================
+# 主入口
+# =============================================================================
+
+def main():
+    """CLI 主入口"""
+    args = sys.argv[1:]
+
+    # 处理全局选项
+    if "--version" in args or "-v" in args:
+        console.print("travel-agent v0.2.0")
+        return
+
+    # 检查是否请求特定命令的帮助
+    if len(args) >= 2 and ("--help" in args or "-h" in args):
+        # 例如：python main.py analyze --help
+        cmd_name = args[0]
+        cmd = registry.get(cmd_name)
+        if cmd:
+            console.print(OutputFormatter.format_command_help(cmd))
+            return
+
+    if "--help" in args or "-h" in args or len(args) == 0:
+        asyncio.run(cmd_help.handler())
+        return
+
+    # 解析命令和参数
+    command_name = args[0]
+    command_args = _parse_args(args[1:], command_name)
+
+    # 执行命令
+    cmd = registry.get(command_name)
+    if not cmd:
+        console.print(f"[red]未知命令：{command_name}[/red]")
+        console.print("")
+        console.print("使用 `travel-agent help` 查看所有可用命令")
+        sys.exit(1)
+
+    # 检查非交互模式支持
+    if not sys.stdin.isatty() and not cmd.supports_non_interactive:
+        console.print("[red]错误：此命令不支持非交互模式[/red]")
+        sys.exit(1)
+
+    # 执行命令
+    try:
+        asyncio.run(cmd.handler(**command_args))
+    except Exception as e:
+        console.print(f"[red]错误：{e}[/red]")
+        sys.exit(1)
+
+
+def _parse_args(args: List[str], command_name: str = "") -> dict:
+    """解析命令行参数"""
+    result = {}
+    i = 0
+    while i < len(args):
+        arg = args[i]
+        if arg.startswith("--"):
+            key = arg[2:].replace("-", "_")
+            # 布尔标志
+            if i + 1 >= len(args) or args[i + 1].startswith("--"):
+                result[key] = True
+            else:
+                result[key] = args[i + 1]
+                i += 1
+        elif arg.startswith("-") and len(arg) == 2:
+            short_key = arg[1]
+            # 为 run 命令映射短选项
+            if command_name == "run":
+                if short_key == "k":
+                    short_key = "keyword"
+                elif short_key == "n":
+                    short_key = "top_n"
+                elif short_key == "p":
+                    short_key = "plan_top_n"
+            # 为 model 命令映射短选项
+            elif command_name == "model":
+                if short_key == "p":
+                    short_key = "provider"
+                elif short_key == "m":
+                    short_key = "model_name"
+            if i + 1 < len(args) and not args[i + 1].startswith("-"):
+                result[short_key] = args[i + 1]
+                i += 1
+            else:
+                result[short_key] = True
+        else:
+            # 位置参数：根据命令类型分配
+            if command_name == "analyze":
+                result["topic"] = arg
+            elif command_name == "model":
+                if "action" not in result:
+                    result["action"] = arg
+                elif "model_name" not in result and arg.startswith("-m") is False:
+                    result["model_name"] = arg
+            elif command_name == "help":
+                result["command_name"] = arg
+            elif command_name == "agents":
+                result["action"] = arg
+            elif command_name == "flyai":
+                # flyai 命令的参数
+                if "query" not in result:
+                    result["query"] = arg
+                elif "destination" not in result:
+                    result["destination"] = arg
+                elif "duration" not in result:
+                    result["duration"] = arg
+                elif "budget" not in result:
+                    result["budget"] = arg
+            elif command_name == "run":
+                # run 命令的第一个位置参数是 keyword
+                if "keyword" not in result:
+                    result["keyword"] = arg
+                else:
+                    result["top_n"] = arg
+            else:
+                # 默认：第一个位置参数作为 topic 或 keyword
+                if "topic" not in result:
+                    result["topic"] = arg
+                elif "keyword" not in result:
+                    result["keyword"] = arg
+        i += 1
+    return result
+
+
+# =============================================================================
+# 辅助函数
+# =============================================================================
+
+def _show_help():
+    """显示帮助信息"""
+    console.print(Panel(
+        "AI 驱动的旅行目的地推荐 Agent",
+        title="🤖 travel-agent",
+        border_style="blue",
+    ))
+    console.print("")
+
+    commands = registry.list_commands()
+    console.print(OutputFormatter.format_command_list(commands))
+
+    console.print("")
+    console.print("[dim]使用 `travel-agent help <command>` 查看单个命令的详细帮助[/dim]")
+
+
+def _show_model_list(settings):
+    """显示模型列表"""
+    from config.settings import LLM_PROVIDERS
+
+    console.print("[bold]支持的 LLM 提供商和模型：[/bold]\n")
+
+    for prov, info in LLM_PROVIDERS.items():
+        configured = "✓" if settings.is_provider_configured(prov) else " "
+        console.print(f"[{configured}] {prov}:")
+        for model in info["models"]:
+            console.print(f"    - {model}")
+        console.print()
+
+
+def _show_model_status(settings):
+    """显示模型状态"""
+    config = settings.get_llm_config()
+
+    console.print(Panel(
+        f"当前提供商：[bold]{config['provider']}[/bold]\n"
+        f"当前模型：[bold]{config['model']}[/bold]\n"
+        f"API 已配置：{'是' if config['api_key_configured'] else '否'}\n"
+        f"可用提供商：{', '.join(config['available_providers']) or '无'}",
+        title="🤖 LLM 模型状态",
+        border_style="blue",
+    ))
+
+
+def _switch_model(settings, provider: str, model_name: Optional[str]):
+    """切换模型"""
+    from config.settings import LLM_PROVIDERS
+
+    if provider not in LLM_PROVIDERS:
+        console.print(f"[red]错误：不支持的提供商 {provider}[/red]")
+        console.print(f"支持的提供商：{', '.join(LLM_PROVIDERS.keys())}")
+        return
+
+    if not settings.is_provider_configured(provider):
+        console.print(f"[red]错误：{provider} 未配置[/red]")
+        console.print("请先在 .env 文件中配置对应的 API Key")
+        return
+
+    # 更新配置
+    _update_model_config(provider, model_name)
+
+
+def _update_model_config(provider: str, model_name: Optional[str]):
+    """更新模型配置"""
+    env_path = Path(".env")
+    if not env_path.exists():
+        console.print("[yellow].env 文件不存在，正在创建...[/yellow]")
+        import shutil
+        shutil.copy(".env.example", env_path)
+
+    config_lines = []
+    with open(env_path, "r") as f:
+        config_lines = f.readlines()
+
+    updated = False
+    new_lines = []
+    for line in config_lines:
+        if line.startswith("LLM_PROVIDER="):
+            new_lines.append(f"LLM_PROVIDER={provider}\n")
+            updated = True
+        elif line.startswith("LLM_MODEL="):
+            if model_name:
+                new_lines.append(f"LLM_MODEL={model_name}\n")
+                updated = True
+            else:
+                new_lines.append("LLM_MODEL=\n")
+                updated = True
+        else:
+            new_lines.append(line)
+
+    if not updated:
+        new_lines.append(f"\nLLM_PROVIDER={provider}\n")
+        if model_name:
+            new_lines.append(f"LLM_MODEL={model_name}\n")
+        else:
+            new_lines.append("LLM_MODEL=\n")
+
+    with open(env_path, "w") as f:
+        f.writelines(new_lines)
+
+    msg = f"✓ 已切换到 {provider}"
+    if model_name:
+        msg += f"/{model_name}"
+    else:
+        msg += " (使用默认模型)"
+    console.print(f"[green]{msg}[/green]")
+
+
+def _init_config():
+    """初始化配置文件"""
+    import shutil
+    src = Path(".env.example")
+    dst = Path(".env")
+    if not dst.exists():
+        shutil.copy(src, dst)
+        console.print("[green]已创建 .env 配置文件[/green]")
+    else:
+        console.print("[yellow].env 文件已存在[/yellow]")
+
+
+def _show_config(settings):
+    """显示配置"""
+    console.print("[bold]当前配置：[/bold]")
+    console.print(f"数据库路径：{settings.database_path}")
+    console.print(f"输出目录：{settings.output_dir}")
+    console.print()
+    console.print("[bold]LLM 配置：[/bold]")
+    console.print(f"当前提供商：{settings.llm_provider}")
+    console.print(f"当前模型：{settings.get_active_model()}")
+    console.print()
+    console.print("[bold]API Key 配置状态：[/bold]")
+    providers = {
+        "anthropic": ("Anthropic", settings.anthropic_api_key),
+        "openai": ("OpenAI", settings.openai_api_key),
+        "deepseek": ("DeepSeek", settings.deepseek_api_key),
+        "azure": ("Azure OpenAI", settings.azure_openai_api_key and settings.azure_openai_endpoint),
+        "qwen": ("阿里云通义千问", settings.dashscope_api_key),
+        "ollama": ("Ollama (本地)", "✓"),
+    }
+    for prov, (name, configured) in providers.items():
+        status = "✓ 已配置" if configured else "✗ 未配置"
+        marker = "→" if prov == settings.llm_provider else " "
+        console.print(f"  {marker} {name}: {status}")
+
+
+def _show_team_info(team_status: dict):
+    """显示 Agent 团队信息"""
+    settings = get_settings()
+
+    console.print("\n[bold cyan]🤖 Agent 团队信息[/bold cyan]")
+
+    table = Table(show_header=True, header_style="bold magenta", box=box.SIMPLE)
+    table.add_column("Agent", style="cyan")
+    table.add_column("角色", style="green")
+    table.add_column("职责", style="yellow")
+
+    m = team_status["manager"]
+    table.add_row("Manager", m["role"], m["goal"])
+
+    for name, info in team_status["team_members"].items():
+        table.add_row(name.capitalize(), info["role"], info["goal"])
+
+    console.print(table)
+
+    console.print()
+    console.print(f"LLM 提供商：[bold]{settings.llm_provider}[/bold]")
+    console.print(f"模型：[bold]{settings.get_active_model()}[/bold]")
+    console.print(f"可用提供商：{', '.join(settings.get_configured_providers()) or '无'}")
+
+
+def _show_result_summary(result: dict):
+    """显示结果摘要"""
+    console.print("\n[bold green]✓ 工作流完成！[/bold green]")
+
+    stages = result.get("stages", {})
+
+    collection = stages.get("collection", {})
+    console.print(f"  • 采集数据：社交媒体 {collection.get('social_media_count', 0)} 条，"
+                  f"文旅 {collection.get('wenlv_count', 0)} 条")
+
+    analysis = stages.get("analysis", {})
+    top_dests = analysis.get("top_destinations", [])
+    if top_dests:
+        console.print(f"  • TOP 目的地：{', '.join(top_dests[:5])}")
+
+    routes = stages.get("routes", [])
+    console.print(f"  • 规划路线：{len(routes)} 条")
+
+    report = stages.get("report", {})
+    console.print(f"  • 报告生成：{len(report.get('content', ''))} 字符")
+
+
+def _show_product_analysis_result(result: dict):
+    """显示产品分析结果摘要"""
+    console.print("\n[bold green]✓ 分析完成！[/bold green]")
+
+    stages = result.get("stages", {})
+
+    collection = stages.get("collection", {})
+    console.print(f"  • 采集数据：社交媒体 {collection.get('social_media_count', 0)} 条，"
+                  f"文旅 {collection.get('wenlv_count', 0)} 条")
+
+    market_demand = stages.get("market_demand", {})
+    console.print(f"  • 社交媒体热度：{market_demand.get('heat_level', '未知')}")
+
+    resources = stages.get("resources", {})
+    console.print(f"  • 航线资源：{resources.get('flight_availability', '未知')}")
+    console.print(f"  • 酒店资源：{resources.get('hotel_options', '未知')}")
+
+    assessment = stages.get("assessment", {})
+    console.print(f"  • 推荐指数：{assessment.get('recommendation_score', 0)}/10")
+    console.print(f"  • 市场潜力：{assessment.get('market_potential', '未知')}")
+    console.print(f"  • 风险等级：{assessment.get('risk_level', '未知')}")
+
+
+if __name__ == "__main__":
+    main()

package/python/pyproject.toml

@@ -0,0 +1,51 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "travel-agent"
+version = "0.2.1"
+description = "AI-powered travel destination recommendation agent"
+authors = [{ name = "Your Name", email = "your.email@example.com" }]
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "anthropic>=0.18.0",
+    "openai>=1.0.0",
+    "playwright>=1.40.0",
+    "beautifulsoup4>=4.12.0",
+    "typer>=0.9.0",
+    "APScheduler>=3.10.0",
+    "pydantic>=2.5.0",
+    "pydantic-settings>=2.1.0",
+    "jinja2>=3.1.0",
+    "python-dotenv>=1.0.0",
+    "httpx>=0.26.0",
+    "aiosqlite>=0.19.0",
+    "rich>=13.7.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.4.0",
+    "pytest-asyncio>=0.21.0",
+    "black>=23.12.0",
+    "ruff>=0.1.0",
+]
+pdf = [
+    "weasyprint>=60.0",
+]
+
+[project.scripts]
+travel-agent = "main:app"
+
+[tool.setuptools.packages.find]
+where = ["."]
+
+[tool.ruff]
+line-length = 100
+target-version = "py310"
+
+[tool.black]
+line-length = 100
+target-version = ["py310"]