aria-code 0.52.0__py3-none-any.whl

aria_code/apps/cli/commands/session_cmds.py

@@ -0,0 +1,236 @@
+"""SessionCommandsMixin — session list/load/save/export commands."""
+
+from __future__ import annotations
+
+from ._ui import print_error
+
+import json
+
+from aria_code.apps.cli.session_export import build_session_export_payload
+
+
+import json
+import asyncio
+import datetime
+import time
+import shlex
+import sys
+import os
+from typing import Dict, Any, Optional
+
+def _arrow_select(*args, **kwargs):
+    # Re-exported by aria_cli; imported from its own module here.
+    from aria_code.ui.picker import arrow_select as fn
+    return fn(*args, **kwargs)
+def get_model_cfg(*args, **kwargs):
+    from ..model_catalog import get_model_cfg as fn
+    return fn(*args, **kwargs)
+
+import json
+import asyncio
+import datetime
+import time
+import shlex
+import sys
+import os
+from typing import Dict, Any, Optional
+
+
+class SessionCommandsMixin:
+    """Mixin: session list/load/save/export commands."""
+
+    def cmd_sessions(self, args: str):
+        keyword = args.strip().lower()
+        sessions = self.terminal.session_mgr.list_sessions()
+        if keyword:
+            sessions = [s for s in sessions if keyword in s["title"].lower()]
+        if not sessions:
+            msg = f"No sessions matching '{keyword}'" if keyword else "No saved sessions"
+            self.context.console.print(f"[dim]{msg}[/dim]" if self.context.has_rich else msg)
+            return
+        if self.context.has_rich:
+            self.context.console.print()
+            header = f"  [bold]Sessions[/bold]  [dim]({len(sessions)} found)[/dim]" if keyword else "  [bold]Sessions[/bold]"
+            self.context.console.print(header)
+            for i, s in enumerate(sessions, 1):
+                updated = s["updated"][:16] if s["updated"] else "-"
+                self.context.console.print(f"    [dim]{i}.[/dim] [bold]{s['title']}[/bold]  "
+                              f"[dim]{s['id'][:8]}  {s['messages']} msgs  {updated}[/dim]")
+            self.context.console.print()
+            self.context.console.print("  [dim]Use /load <number> to resume · /sessions <keyword> to search[/dim]")
+        else:
+            for i, s in enumerate(sessions, 1):
+                print(f"  {i}. [{s['id'][:8]}] {s['title']} ({s['messages']} msgs)")
+
+    def cmd_save(self, args: str):
+        if not self.terminal.conversation:
+            self.context.console.print("[dim]Nothing to save[/dim]" if self.context.has_rich else "Nothing to save")
+            return
+        sid = self.terminal.session_id
+        title = args.strip().strip('"').strip("'") if args.strip() else None
+        meta = {}
+        if title:
+            meta["title"] = title
+        self.terminal.session_mgr.save_session(sid, self.terminal.conversation, metadata=meta)
+        self.terminal.config["last_session_id"] = sid
+        self.context.save_config(self.terminal.config)
+        display = f"{title} ({sid[:8]})" if title else f"{sid[:8]}..."
+        self.context.console.print(f"[green]Session saved: {display}[/green]" if self.context.has_rich
+                      else f"Saved: {display}")
+
+    def cmd_rename(self, args: str):
+        """Rename current session."""
+        title = args.strip().strip('"').strip("'")
+        if not title:
+            self.context.console.print("[dim]Usage: /rename <title>[/dim]" if self.context.has_rich else "Usage: /rename <title>")
+            return
+        sid = self.terminal.session_id
+        data = self.terminal.session_mgr.load_session(sid)
+        if data:
+            meta = data.get("metadata", {})
+            meta["title"] = title
+            self.terminal.session_mgr.save_session(sid, self.terminal.conversation, metadata=meta)
+        else:
+            self.terminal.session_mgr.save_session(sid, self.terminal.conversation, metadata={"title": title})
+        self.context.console.print(f"[green]Renamed: {title}[/green]" if self.context.has_rich else f"Renamed: {title}")
+
+    def cmd_load(self, args: str):
+        session_id = args.strip()
+        if not session_id:
+            sessions = self.terminal.session_mgr.list_sessions()
+            if not sessions:
+                self.context.console.print("[dim]No sessions. Usage: /load <session_id>[/dim]" if self.context.has_rich
+                              else "No sessions")
+                return
+            options = []
+            for s in sessions[:20]:
+                title = s.get("metadata", {}).get("title", s["id"][:8])
+                ts = s.get("updated", "")[:10]
+                options.append((title, ts))
+            choice = _arrow_select(options, selected=0, title="Load Session")
+            if 0 <= choice < len(sessions):
+                session_id = sessions[choice]["id"]
+            else:
+                if self.context.has_rich:
+                    self.context.console.print("[dim]Cancelled[/dim]")
+                else:
+                    print("Cancelled")
+                return
+
+        data = self.terminal.session_mgr.load_session(session_id)
+        if data:
+            self.terminal.conversation = data.get("messages", [])
+            self.terminal.session_id = data["id"]
+            title = data.get("metadata", {}).get("title", "Untitled")
+            n = len(self.terminal.conversation)
+            self.context.console.print(f"[green]Loaded: {title} ({n} messages)[/green]" if self.context.has_rich
+                          else f"Loaded: {title} ({n} msgs)")
+        else:
+            print_error(self.context, f"Session not found: {session_id}", "session")
+
+    def cmd_recall(self, args: str):
+        """Full-text search across all saved sessions: /recall <query>"""
+        query = args.strip()
+        if not query:
+            self.context.console.print("[dim]Usage: /recall <query>[/dim]" if self.context.has_rich else "Usage: /recall <query>")
+            return
+        results = self.terminal.session_mgr.search_sessions(query)
+        if not results:
+            msg = f"No sessions found matching '{query}'"
+            self.context.console.print(f"[dim]{msg}[/dim]" if self.context.has_rich else msg)
+            return
+        if self.context.has_rich:
+            self.context.console.print()
+            self.context.console.print(f"  [bold]Recall[/bold]  [dim]{len(results)} session(s) match '{query}'[/dim]")
+            self.context.console.print()
+            for r in results[:10]:
+                updated = r["updated"][:16] if r["updated"] else ""
+                self.context.console.print(
+                    f"  [bold]{r['title']}[/bold]  "
+                    f"[dim]{r['id'][:8]}  {r['match_count']} hit(s)  {updated}[/dim]"
+                )
+                preview = r["preview"].replace("\n", " ")[:100]
+                self.context.console.print(f"    [dim]…{preview}…[/dim]")
+                self.context.console.print()
+            self.context.console.print("  [dim]Use /load <id> to resume a session[/dim]")
+        else:
+            print(f"\n{len(results)} session(s) found:")
+            for r in results[:10]:
+                print(f"  [{r['id'][:8]}] {r['title']} ({r['match_count']} hits)")
+                print(f"    ...{r['preview'][:80]}...")
+
+    async def cmd_export(self, args: str):
+        parts = args.split()
+        fmt = parts[0].lower() if parts else "json"
+        filename = parts[1] if len(parts) > 1 else None
+
+        if not self.terminal.conversation:
+            self.context.console.print("[dim]Nothing to export[/dim]" if self.context.has_rich else "Nothing to export")
+            return
+
+        try:
+            provider_health = []
+            doctor_report = None
+            mcp_status = None
+            context_health = None
+            if fmt == "bundle":
+                try:
+                    from packages.aria_services.provider_health import GLOBAL_PROVIDER_HEALTH
+                    provider_health = GLOBAL_PROVIDER_HEALTH.snapshot()
+                except Exception:
+                    provider_health = []
+                # A support bundle should carry environment health and MCP/circuit
+                # state — the two things a maintainer asks for first. Best-effort.
+                try:
+                    from doctor import run_doctor
+                    doctor_report = run_doctor(self.terminal.config)
+                except Exception:
+                    doctor_report = None
+                try:
+                    _reg = getattr(self.terminal, "_mcp_registry", None)
+                    mcp_status = _reg.status() if _reg else None
+                except Exception:
+                    mcp_status = None
+                try:
+                    from packages.aria_services.context import context_health_snapshot
+                    _mc = get_model_cfg(self.terminal.config.get("model", "qwen2.5:7b"))
+                    context_health = context_health_snapshot(
+                        self.terminal.conversation,
+                        max_tokens=int(_mc.get("num_ctx", 16384)),
+                        threshold=float(self.terminal.config.get("auto_compact_threshold", 0.78)),
+                    )
+                except Exception:
+                    context_health = None
+            content, ext, prefix = build_session_export_payload(
+                fmt,
+                self.terminal.conversation,
+                session_id=self.terminal.session_id,
+                config=self.terminal.config,
+                trace=getattr(self.terminal, "runtime_trace", None),
+                provider_health=provider_health,
+                doctor_report=doctor_report,
+                mcp_status=mcp_status,
+                context_health=context_health,
+            )
+        except ValueError as exc:
+            if fmt == "sft" and "No user→assistant pairs" in str(exc):
+                self.context.console.print("[dim]No user→assistant pairs to export[/dim]" if self.context.has_rich else "No pairs to export")
+                return
+            self.context.console.print("[dim]Format: json, csv, md, sft, or bundle[/dim]" if self.context.has_rich
+                          else "Format: json, csv, md, sft, bundle")
+            return
+
+        if fmt == "sft":
+            pairs = json.loads(content)
+            if self.context.has_rich:
+                self.context.console.print(f"[dim]{len(pairs)} training pairs extracted[/dim]")
+            else:
+                print(f"{len(pairs)} training pairs")
+
+        if not filename:
+            from datetime import datetime
+            filename = f"{prefix}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.{ext}"
+        with open(filename, "w", encoding="utf-8") as f:
+            f.write(content)
+        self.context.console.print(f"[green]Exported to {filename}[/green]" if self.context.has_rich
+                      else f"Exported: {filename}")

aria_code/apps/cli/commands/session_ux_cmds.py

@@ -0,0 +1,359 @@
+"""SessionUxCommandsMixin — clear, recap, history, compact, fork, copy commands."""
+
+from __future__ import annotations
+
+import os
+
+
+import json
+import asyncio
+import datetime
+import time
+import shlex
+from typing import Dict, Any, Optional
+
+def get_model_cfg(*args, **kwargs):
+    from ..model_catalog import get_model_cfg as fn
+    return fn(*args, **kwargs)
+def OllamaProvider(*args, **kwargs):
+    # Re-exported by aria_cli; imported from its own module here.
+    from aria_code.apps.cli.providers.base import OllamaProvider as fn
+    return fn(*args, **kwargs)
+def stream_provider_result(*args, **kwargs):
+    # Re-exported by aria_cli; imported from its own module here.
+    from aria_code.packages.aria_sdk.streaming import stream_provider_result as fn
+    return fn(*args, **kwargs)
+
+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 SessionUxCommandsMixin:
+    """Mixin: session UX and conversation management commands."""
+
+    def cmd_clear(self, args: str):
+        self.terminal.conversation = []
+        os.system("clear" if os.name == "posix" else "cls")
+        self.context.console.print("[dim]Conversation cleared[/dim]" if self.context.has_rich else "Cleared")
+
+    def cmd_btw(self, args: str):
+        q = args.strip()
+        if not q:
+            self.context.console.print("[dim]/btw <question>  — quick question without polluting history[/dim]" if self.context.has_rich else "/btw <question>")
+            return
+        conv = self.terminal.conversation
+        if not conv:
+            self.context.console.print("[dim](no conversation context yet)[/dim]" if self.context.has_rich else "(no context)")
+            return
+        _ctx_slice = conv[-6:] if len(conv) >= 6 else conv
+        _ctx = "\n".join(
+            f"{m['role'].upper()}: {str(m.get('content', ''))[:300]}"
+            for m in _ctx_slice
+        )
+        _btw_prompt = (
+            f"[Side question — answer briefly, do not reference this note]\n"
+            f"Context from conversation:\n{_ctx}\n\nQuestion: {q}"
+        )
+        if self.context.has_rich:
+            from rich.panel import Panel as _Panel
+            from rich import box as _rbox
+            self.context.console.print(_Panel(f"[dim]{q}[/dim]", title="[dim]/btw[/dim]", box=_rbox.ROUNDED, border_style="dim"))
+        import asyncio as _aio
+
+        async def _ask_btw():
+            try:
+                _result = await stream_provider_result(
+                    OllamaProvider(
+                        self.terminal.config.get("ollama_url", "http://localhost:11434"),
+                        self.terminal.config.get("model", "qwen2.5:7b"),
+                        show_market_prefetch_status=False,
+                    ),
+                    _btw_prompt,
+                    tools=[],
+                )
+                if _result.get("success"):
+                    return _result.get("response", "")
+                return f"(error: {_result.get('error', '')})"
+            except Exception as _e:
+                return f"(error: {_e})"
+
+        try:
+            loop = _aio.get_event_loop()
+            answer = loop.run_until_complete(_ask_btw()) if not loop.is_running() else "(run /btw from interactive prompt)"
+        except Exception:
+            answer = "(could not get answer)"
+        if self.context.has_rich:
+            from rich.panel import Panel as _Panel
+            from rich import box as _rbox
+            self.context.console.print(_Panel(answer.strip(), title="[dim]↩ btw[/dim]", box=_rbox.ROUNDED, border_style="dim #C08050"))
+        else:
+            print(f"\n  [btw] {answer.strip()}\n")
+
+    def cmd_recap(self, args: str):
+        conv = self.terminal.conversation
+        if not conv:
+            self.context.console.print("[dim]No conversation yet[/dim]" if self.context.has_rich else "No conversation")
+            return
+        turns = len([m for m in conv if m.get("role") == "user"])
+        topics: list[str] = []
+        for m in conv:
+            if m.get("role") == "user":
+                snippet = str(m.get("content", ""))[:60].strip()
+                if snippet:
+                    topics.append(snippet)
+        if self.context.has_rich:
+            from rich.panel import Panel as _Panel
+            from rich import box as _rbox
+            body = f"[dim]{turns} 轮对话[/dim]\n"
+            for i, t in enumerate(topics[-6:], 1):
+                body += f"  [dim]{i}.[/dim] {t}…\n"
+            self.context.console.print(_Panel(body.rstrip(), title="[bold]会话摘要[/bold]", box=_rbox.ROUNDED, border_style="dim"))
+        else:
+            print(f"Session: {turns} turns")
+            for i, t in enumerate(topics[-6:], 1):
+                print(f"  {i}. {t}…")
+
+    def cmd_history(self, args: str):
+        if not self.terminal.conversation:
+            self.context.console.print("[dim]No conversation history[/dim]" if self.context.has_rich else "No history")
+            return
+        for msg in self.terminal.conversation[-10:]:
+            role = msg["role"]
+            content = msg["content"][:120]
+            if self.context.has_rich:
+                prefix = "You" if role == "user" else "Aria"
+                style = "bold" if role == "user" else "bold"
+                self.context.console.print(f"[{style}]{prefix}:[/{style}] [dim]{content}[/dim]")
+            else:
+                print(f"{'You' if role == 'user' else 'Aria'}: {content}")
+
+    def _record_context_checkpoint(self, kind: str, envelope: dict | None = None) -> None:
+        """Persist the pre-compaction conversation (aria.context_checkpoint.v1).
+
+        Best-effort by design: checkpointing must never turn a working
+        compaction into a failure, so every error is swallowed after a debug
+        log. Retention (keep-per-session / max-age) comes from config with
+        the defaults decided for the schema.
+        """
+        try:
+            from packages.aria_services.context_checkpoints import (
+                ContextCheckpointStore,
+                default_checkpoint_root,
+            )
+            cfg = self.terminal.config
+            store = ContextCheckpointStore(
+                default_checkpoint_root(),
+                keep_per_session=int(cfg.get("context_checkpoint_keep", 5) or 5),
+                max_age_days=int(cfg.get("context_checkpoint_max_age_days", 30) or 30),
+            )
+            store.record(
+                getattr(self.terminal, "session_id", "session"),
+                self.terminal.conversation,
+                kind=kind,
+                envelope=envelope,
+            )
+        except Exception as exc:
+            import logging
+            logging.getLogger(__name__).debug("context checkpoint skipped: %s", exc)
+
+    def cmd_compact(self, args: str):
+        if "--hard" in args:
+            if len(self.terminal.conversation) > 10:
+                kept = self.terminal.conversation[-6:]
+                self._record_context_checkpoint("hard")
+                self.terminal.conversation = kept
+                self.context.console.print(f"[dim]Hard-compacted to last {len(kept)} messages[/dim]" if self.context.has_rich
+                              else f"Hard-compacted to {len(kept)} messages")
+            else:
+                self.context.console.print("[dim]Context small enough, no compaction needed[/dim]" if self.context.has_rich
+                              else "No compaction needed")
+            return
+        import asyncio as _asyncio
+        try:
+            loop = _asyncio.get_event_loop()
+            loop.run_until_complete(self._smart_compact_async(silent=False))
+        except RuntimeError:
+            if len(self.terminal.conversation) > 6:
+                from apps.cli.message_processing import compact_messages
+                compacted = compact_messages(
+                    self.terminal.conversation,
+                    model_key=self.terminal.config.get("model", "qwen2.5:7b"),
+                )
+                self._record_context_checkpoint("fallback")
+                self.terminal.conversation = (
+                    compacted
+                    if len(compacted) < len(self.terminal.conversation)
+                    else self.terminal.conversation[-8:]
+                )
+                self.context.console.print("[dim]Compacted (fallback)[/dim]")
+
+    async def _smart_compact_async(self, silent: bool = False):
+        conv = self.terminal.conversation
+        if len(conv) <= 4:
+            if not silent:
+                self.context.console.print("[dim]Context small enough — no compaction needed[/dim]" if self.context.has_rich
+                              else "Context small enough")
+            return
+
+        if not silent and self.context.has_rich:
+            self.context.console.print("[dim]Summarising conversation...[/dim]")
+
+        try:
+            max_ctx = int(get_model_cfg(self.terminal.config.get("model", "qwen2.5:7b")).get("num_ctx", 16384) or 16384)
+        except Exception:
+            max_ctx = 16384
+        from packages.aria_services.context import build_context_service
+        context_service = build_context_service(max_tokens=max_ctx)
+        summary_prompt = context_service.build_summary_prompt(conv)
+
+        summary = ""
+        try:
+            result = await stream_provider_result(
+                OllamaProvider(
+                    self.terminal.config.get("ollama_url", "http://localhost:11434"),
+                    self.terminal.config.get("model", "qwen2.5:7b"),
+                    show_market_prefetch_status=False,
+                ),
+                summary_prompt,
+                [],
+                tools=[],
+            )
+            if result.get("success") and result.get("response"):
+                summary = result["response"].strip()
+        except Exception:
+            pass
+
+        if not summary:
+            try:
+                compacted = context_service.compact_messages(conv)
+            except Exception:
+                compacted = []
+            self._record_context_checkpoint("fallback")
+            self.terminal.conversation = compacted if compacted and len(compacted) < len(conv) else conv[-8:]
+            if not silent:
+                self.context.console.print("[dim]Compacted (summary failed, used local fallback)[/dim]" if self.context.has_rich
+                              else "Compacted (summary fallback)")
+            return
+
+        envelope = context_service.build_summary_envelope(conv, summary)
+        self._record_context_checkpoint(
+            "compact",
+            envelope={
+                "old_message_count": envelope.old_message_count,
+                "new_message_count": envelope.new_message_count,
+                "tail_message_count": envelope.tail_message_count,
+            },
+        )
+        self.terminal.conversation = envelope.messages
+        new_count = len(self.terminal.conversation)
+        old_count = len(conv)
+        if not silent:
+            if self.context.has_rich:
+                self.context.console.print(
+                    f"  [dim]✓ Compacted {old_count} → {new_count} messages "
+                    f"(summary preserved context)[/dim]"
+                )
+            else:
+                print(f"Compacted {old_count} → {new_count} messages")
+
+    def cmd_fork(self, args: str):
+        import time as _t
+        name = args.strip() or f"fork-{_t.strftime('%H%M%S')}"
+        snapshot = {
+            "name": name,
+            "ts": _t.strftime("%Y-%m-%d %H:%M:%S"),
+            "conv": [dict(m) for m in self.terminal.conversation],
+            "config": dict(self.terminal.config),
+        }
+        self.terminal._forks.append(snapshot)
+        idx = len(self.terminal._forks) - 1
+        if self.context.has_rich:
+            self.context.console.print(
+                f"  [dim]↳ Forked as [bold]{name}[/bold] "
+                f"(fork #{idx}, {len(snapshot['conv'])} messages). "
+                f"Restore with /load-fork {idx}[/dim]"
+            )
+        else:
+            print(f"Forked as '{name}' (#{idx}). Restore with /load-fork {idx}")
+
+    def cmd_load_fork(self, args: str):
+        forks = self.terminal._forks
+        if not forks:
+            self.context.console.print("[dim]No forks yet — use /fork to create one[/dim]" if self.context.has_rich else "No forks")
+            return
+        try:
+            idx = int(args.strip())
+        except (ValueError, IndexError):
+            if self.context.has_rich:
+                for i, f in enumerate(forks):
+                    self.context.console.print(f"  [dim]#{i}[/dim]  {f['name']}  [dim]{f['ts']}  {len(f['conv'])} msgs[/dim]")
+            else:
+                for i, f in enumerate(forks):
+                    print(f"  #{i}  {f['name']}  {f['ts']}")
+            return
+        if idx < 0 or idx >= len(forks):
+            self.context.console.print(f"[dim]Fork #{idx} not found[/dim]" if self.context.has_rich else "Invalid index")
+            return
+        snap = forks[idx]
+        self.terminal.conversation = [dict(m) for m in snap["conv"]]
+        self.context.console.print(
+            f"  [dim]✓ Restored fork [bold]{snap['name']}[/bold] "
+            f"({len(snap['conv'])} messages)[/dim]"
+            if self.context.has_rich else f"Restored fork '{snap['name']}'"
+        )
+
+    def cmd_copy(self, args: str):
+        text = self.terminal._last_response
+        if not text:
+            self.context.console.print("[dim]No response to copy yet[/dim]" if self.context.has_rich else "Nothing to copy")
+            return
+        copied = False
+        try:
+            import subprocess as _sp
+            _sp.run(["pbcopy"], input=text.encode(), check=True, timeout=3)
+            copied = True
+        except Exception:
+            pass
+        if not copied:
+            try:
+                import subprocess as _sp
+                _sp.run(["xclip", "-selection", "clipboard"], input=text.encode(), check=True, timeout=3)
+                copied = True
+            except Exception:
+                pass
+        if not copied:
+            try:
+                import subprocess as _sp
+                _sp.run(["xdotool", "type", "--clearmodifiers", text], check=True, timeout=3)
+                copied = True
+            except Exception:
+                pass
+        if copied:
+            self.terminal._record_feedback("copy", text)
+            preview = text[:60].replace("\n", " ")
+            self.context.console.print(
+                f"  [dim]✓ Copied to clipboard: \"{preview}{'…' if len(text) > 60 else ''}\"[/dim]"
+                if self.context.has_rich else f"Copied: \"{preview}\""
+            )
+        else:
+            self.context.console.print(
+                "[yellow]Could not reach clipboard (pbcopy/xclip not found). "
+                "Here is the response:[/yellow]\n" + text
+                if self.context.has_rich else "Clipboard unavailable. Response:\n" + text
+            )