aria-code 0.52.0__py3-none-any.whl

aria_code/apps/cli/commands/orchestrator_cmds.py

@@ -0,0 +1,179 @@
+"""User-visible task graph previews for complex work."""
+
+from __future__ import annotations
+
+
+import json
+import asyncio
+import datetime
+import time
+import shlex
+from typing import Dict, Any, Optional
+
+
+import json
+import asyncio
+import datetime
+import time
+import shlex
+import sys
+import os
+from typing import Dict, Any, Optional
+
+
+import json
+import asyncio
+import datetime
+import time
+import shlex
+import sys
+import os
+from typing import Dict, Any, Optional
+
+
+class OrchestratorCommandsMixin:
+    def cmd_orchestrate(self, args: str):
+        """Preview the verifiable Agent task graph for a request."""
+        from runtime.task_orchestrator import build_task_graph
+
+        request = args.strip()
+        if not request:
+            message = "Usage: /orchestrate <任务描述>"
+            self.context.console.print(f"[yellow]{message}[/yellow]") if self.context.has_rich else print(message)
+            return
+        graph = build_task_graph(request)
+        if self.context.has_rich:
+            from rich.table import Table
+
+            table = Table(title=f"任务编排 · {graph.kind}", header_style="bold cyan")
+            table.add_column("步骤", no_wrap=True)
+            table.add_column("目标")
+            table.add_column("依赖")
+            table.add_column("验证门禁")
+            for stage in graph.stages:
+                table.add_row(
+                    stage.title, stage.objective, "、".join(stage.depends_on) or "—",
+                    stage.verification or "—",
+                )
+            self.context.console.print(table)
+            ready = "、".join(stage.title for stage in graph.ready(()))
+            self.context.console.print(f"[dim]可立即并行：{ready}。写入操作仍受权限与 worktree 隔离约束。[/dim]")
+        else:
+            for stage in graph.stages:
+                deps = ", ".join(stage.depends_on) or "none"
+                print(f"- {stage.title} [{deps}]: {stage.objective}")
+
+    async def cmd_route(self, args: str):
+        """Dynamic Agent Routing (Orchestrator) using LLM and interactive streaming."""
+        from aria_code.runtime.orchestrator import dynamic_agent_orchestration
+        from aria_code.apps.cli.providers.base import ConfiguredProvider
+        from aria_code.agents.registry import get_registry
+        
+        request = args.strip()
+        if not request:
+            message = "Usage: /route <任务描述>"
+            self.context.console.print(f"[yellow]{message}[/yellow]") if self.context.has_rich else print(message)
+            return
+            
+        if self.context.has_rich:
+            provider = ConfiguredProvider(self.context.config, "gemini-2.5-flash")
+        
+        with self.context.console.status("[dim]Orchestrator 正在编排智能体网络...[/dim]"): 
+            result = await dynamic_agent_orchestration(request, provider)
+        agents_to_run = result.get("agents", [])
+        
+        if self.context.has_rich:
+            from rich.panel import Panel
+            from rich.live import Live
+            from rich.console import Group
+            from rich.text import Text
+            from rich.markdown import Markdown
+            
+            plan = result.get("plan", "No plan provided.")
+            content = f"""[bold]路由计划:[/bold]
+{plan}
+
+[bold]调度的智能体:[/bold]
+"""
+            for a in agents_to_run:
+                content += f" - {a}\n"
+                
+            self.context.console.print(Panel(content, title="动态路由编排 (Orchestrator)", border_style="dim"))
+            
+            if not agents_to_run:
+                self.context.console.print("[yellow]没有可用的智能体处理此请求。[/yellow]")
+                return
+
+            registry = get_registry()
+            import re as _re
+            tickers = _re.findall(r'\b[A-Z]{2,5}\b', request)
+            symbol = tickers[0] if tickers else ""
+            
+            upstream_contexts = []
+            
+            for agent_name in agents_to_run:
+                agent_cls = registry.get(agent_name)
+                if not agent_cls:
+                    self.context.console.print(f"[red]找不到智能体: {agent_name}[/red]")
+                    continue
+                
+                self.context.console.print(f"\n[bold]● {agent_name}[/bold]")
+                
+                lines = []
+                def update_display(live_context, new_line=None):
+                    if new_line:
+                        lines.append(new_line)
+                    display_lines = lines[-15:]
+                    group = Group(*[Text.from_markup(l) for l in display_lines])
+                    live_context.update(group)
+                
+                with Live(auto_refresh=True, console=self.context.console) as live:
+                    def on_thought(text):
+                        update_display(live, f"  [dim]│ 思考:[/dim] {text.strip()}")
+                        
+                    def on_tool_start(name, params):
+                        update_display(live, f"  [dim]├ 工具:[/dim] {name} {params}")
+                        
+                    def on_tool_end(name, result):
+                        res_str = str(result)[:60] + "..." if len(str(result)) > 60 else str(result)
+                        update_display(live, f"  [dim]└ 返回:[/dim] {name} -> {res_str}")
+                        
+                    def on_token(token):
+                        pass
+                        
+                    agent = agent_cls(
+                        llm_provider=provider,
+                        on_thought=on_thought,
+                        on_tool_start=on_tool_start,
+                        on_tool_end=on_tool_end,
+                        on_token=on_token
+                    )
+                    
+
+                    try:
+                        data = {"company_name": symbol, "request": request, "upstream_context": list(upstream_contexts)}
+                        
+                        # Some agents use self._current_data, so we set it here for compatibility
+                        agent._current_data = data 
+                        
+                        agent_result = await agent.analyze(symbol, data)
+                        
+                        # Add this agent's analysis to upstream_contexts for the next agent!
+                        upstream_contexts.append(f"【{agent_name} 的分析结论】: {agent_result.analysis}")
+                        
+                        update_display(live, f"  [dim]✓ 完成.[/dim]")
+
+                    except Exception as e:
+                        update_display(live, f"  [red]✗ 失败: {e}[/red]")
+                        continue
+                        
+                self.context.console.print(Panel(
+                    Markdown(agent_result.analysis), 
+                    title=f"{agent_name} 分析报告", 
+                    border_style="dim"
+                ))
+
+        else:
+            print("动态路由编排结果:")
+            print(f"计划: {result.get('plan')}")
+            print(f"智能体: {result.get('agents')}")

aria_code/apps/cli/commands/pdf_export_cmds.py

@@ -0,0 +1,125 @@
+"""PdfExportCommandsMixin — /export-pdf: 结构化报告 markdown -> 设计过的 PDF。
+
+跟 apps/cli/commands/report.py 的 /report --pdf（单股票分析报告）是两条不同的
+路：那条走 report_generator._build_html() 的写死布局；这条走 apps/cli/pdf_report.py
+的通用 Document/Section/Theme 引擎，给的是任意结构化报告（先支持 shortterm 一种，
+新增类型只需要在 pdf_report.PARSERS 里加一个解析函数）+ 可选主题 + 可选 section
+取舍。两条路都复用同一个 report_generator.export_pdf() 做最终的 HTML→PDF。
+"""
+
+from __future__ import annotations
+
+
+import json
+import asyncio
+import datetime
+import time
+import shlex
+from typing import Dict, Any, Optional
+
+
+import json
+import asyncio
+import datetime
+import time
+import shlex
+import sys
+import os
+from typing import Dict, Any, Optional
+
+
+import json
+import asyncio
+import datetime
+import time
+import shlex
+import sys
+import os
+from typing import Dict, Any, Optional
+
+
+class PdfExportCommandsMixin:
+    """Mixin: /export-pdf <report.md> [--type=shortterm] [--theme=institutional|bloomberg] [--sections=a,b] [--exclude=x,y]"""
+
+    @staticmethod
+    def _parse_export_pdf_args(args: str) -> dict:
+        # 模块级的自由函数在 aria_cli.py 的 _rebind_mixin_globals() 之后会失联
+        # ——它把这个类每个方法的 __globals__ 换成 aria_cli.py 自己的模块命名空间
+        # （为了让 self.context.console/self.context.has_rich 这类裸名字能解析到），代价是原模块里定义的
+        # 兄弟函数就不再可见了。写成 staticmethod 走 self._xxx 属性查找，不受影响。
+        parts = args.split()
+        out = {"path": None, "type": "shortterm", "theme": "institutional", "include": None, "exclude": None}
+        for p in parts:
+            if p.startswith("--theme="):
+                out["theme"] = p.split("=", 1)[1].strip()
+            elif p.startswith("--type="):
+                out["type"] = p.split("=", 1)[1].strip()
+            elif p.startswith("--sections="):
+                out["include"] = [s.strip() for s in p.split("=", 1)[1].split(",") if s.strip()]
+            elif p.startswith("--exclude="):
+                out["exclude"] = [s.strip() for s in p.split("=", 1)[1].split(",") if s.strip()]
+            elif not p.startswith("--") and out["path"] is None:
+                out["path"] = p
+        return out
+
+    async def cmd_export_pdf(self, args: str):
+        # 全部用局部 import：模块顶层的 import 在 _rebind_mixin_globals() 之后
+        # 同样会失联（原因同上面 staticmethod 那条注释），局部 import 绑定到
+        # 函数自己的局部命名空间，不受影响。
+        from pathlib import Path
+        from apps.cli.pdf_report import PARSERS, THEMES, render_document
+        from artifacts import user_generated_dir
+
+        opts = self._parse_export_pdf_args(args.strip())
+        if not opts["path"]:
+            self.context.console.print("[dim]用法: /export-pdf <report.md> [--type=shortterm] [--theme=institutional|bloomberg] [--sections=a,b] [--exclude=x,y][/dim]"
+                          if self.context.has_rich else "用法: /export-pdf <report.md> [--type=...] [--theme=...] [--sections=...] [--exclude=...]")
+            return
+
+        src = Path(opts["path"]).expanduser()
+        if not src.exists():
+            self.context.console.print(f"[red]找不到文件: {src}[/red]" if self.context.has_rich else f"找不到文件: {src}")
+            return
+
+        parser = PARSERS.get(opts["type"])
+        if parser is None:
+            avail = ", ".join(PARSERS.keys())
+            self.context.console.print(f"[red]未知报告类型 '{opts['type']}'，目前支持: {avail}[/red]" if self.context.has_rich
+                          else f"未知报告类型 '{opts['type']}'，目前支持: {avail}")
+            return
+
+        theme = THEMES.get(opts["theme"])
+        if theme is None:
+            avail = ", ".join(THEMES.keys())
+            self.context.console.print(f"[red]未知主题 '{opts['theme']}'，目前支持: {avail}[/red]" if self.context.has_rich
+                          else f"未知主题 '{opts['theme']}'，目前支持: {avail}")
+            return
+
+        try:
+            doc = parser(src.read_text(encoding="utf-8"))
+            if opts["include"] or opts["exclude"]:
+                doc = doc.select(include=opts["include"], exclude=opts["exclude"])
+            if not doc.sections:
+                self.context.console.print("[yellow]筛选后没有剩下任何 section，检查 --sections/--exclude 关键词是否匹配得上标题[/yellow]"
+                              if self.context.has_rich else "筛选后没有剩下任何 section，检查 --sections/--exclude 关键词")
+                return
+
+            html_doc = render_document(doc, theme)
+            out_dir = user_generated_dir()
+            html_path = out_dir / f"{src.stem}.pdfexport.html"
+            html_path.write_text(html_doc, encoding="utf-8")
+
+            from report_generator import export_pdf
+            import asyncio
+            pdf_path = await asyncio.get_event_loop().run_in_executor(None, lambda: export_pdf(html_path))
+            html_path.unlink(missing_ok=True)  # 中间产物，转完就不需要留着
+
+            if pdf_path is None:
+                self.context.console.print("[red]PDF 转换失败 —— 需要 weasyprint 或系统装了 wkhtmltopdf[/red]" if self.context.has_rich
+                              else "PDF 转换失败 —— 需要 weasyprint 或系统装了 wkhtmltopdf")
+                return
+
+            msg = f"✓ PDF 已生成 ({theme.name} 主题, {len(doc.sections)} 个 section): {pdf_path}"
+            self.context.console.print(f"[green]{msg}[/green]" if self.context.has_rich else msg)
+        except Exception as e:
+            self.context.console.print(f"[red]渲染失败: {e}[/red]" if self.context.has_rich else f"渲染失败: {e}")