context-forge-cli 0.4.10__tar.gz → 0.4.11__tar.gz

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: context-forge-cli
-Version: 0.4.10
+Version: 0.4.11
 Summary: Session manager and context bridge for agentic CLI tools
 Project-URL: Homepage, https://github.com/emmver/contextforge
 Project-URL: Repository, https://github.com/emmver/contextforge
@@ -169,6 +169,7 @@ The method is chosen automatically based on token count and whether a target ses
 | Claude Desktop | `~/Library/Application Support/Claude/local-agent-mode-sessions/` | `--system-prompt` |
 | Codex | `~/.codex/state_5.sqlite` | `resume` / `fork` |
 | Gemini | `~/.gemini/tmp/<project_hash>/chats/` JSON | New session / `/resume` |
+| Antigravity | `~/.gemini/antigravity-cli/history.jsonl` | `-i` / `--conversation` |
 | altimate-code | `~/.local/share/altimate-code/opencode.db` | `run -s` / `import` |
 
 ## Session Summaries

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/PLAN.md

@@ -87,6 +87,7 @@ Agents and contributors: update this file as you work.
 - [ ] Install docs: `uv tool install contextforge`
 - [ ] GitHub Actions CI: `uv run pytest --cov=contextforge`
 - [ ] `open-claw` adapter (if/when CLI interface is documented)
+- [x] Antigravity adapter (`adapters/antigravity.py`)
 - [ ] Session search: `cf ls --search <query>`
 - [ ] Session archiving: `cf archive <id>`
 

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/README.md

@@ -140,6 +140,7 @@ The method is chosen automatically based on token count and whether a target ses
 | Claude Desktop | `~/Library/Application Support/Claude/local-agent-mode-sessions/` | `--system-prompt` |
 | Codex | `~/.codex/state_5.sqlite` | `resume` / `fork` |
 | Gemini | `~/.gemini/tmp/<project_hash>/chats/` JSON | New session / `/resume` |
+| Antigravity | `~/.gemini/antigravity-cli/history.jsonl` | `-i` / `--conversation` |
 | altimate-code | `~/.local/share/altimate-code/opencode.db` | `run -s` / `import` |
 
 ## Session Summaries

context_forge_cli-0.4.11/contextforge/adapters/antigravity.py

@@ -0,0 +1,233 @@
+"""Adapter for Google Antigravity CLI sessions."""
+from __future__ import annotations
+
+import json
+import shlex
+from datetime import datetime, timezone
+from pathlib import Path
+
+import tiktoken
+
+from contextforge.adapters.base import ToolAdapter
+from contextforge.models.session import Message, Session
+
+_AGY_HISTORY = Path.home() / ".gemini" / "antigravity-cli" / "history.jsonl"
+_AGY_CONVERSATIONS = Path.home() / ".gemini" / "antigravity-cli" / "conversations"
+_AGY_ALT_HISTORY = Path.home() / ".gemini" / "antigravity" / "history.jsonl"
+_AGY_ALT_CONVERSATIONS = Path.home() / ".gemini" / "antigravity" / "conversations"
+
+
+def _count_tokens(text: str) -> int:
+    """Count tokens in text using tiktoken for Claude models."""
+    try:
+        enc = tiktoken.encoding_for_model("claude-3-5-sonnet-20241022")
+        return len(enc.encode(text))
+    except Exception:
+        # Fallback: rough estimate (~4 chars per token)
+        return len(text) // 4
+
+
+def _ms_to_dt(ts: int) -> datetime:
+    """Convert a millisecond timestamp to UTC datetime."""
+    return datetime.fromtimestamp(ts / 1000, tz=timezone.utc)
+
+
+class AntigravityAdapter(ToolAdapter):
+    tool_name = "antigravity"
+    default_paths = [_AGY_HISTORY, _AGY_ALT_HISTORY]
+
+    def _history_path(self) -> Path | None:
+        """Return the first existing history.jsonl path."""
+        for p in (_AGY_HISTORY, _AGY_ALT_HISTORY):
+            if p.exists():
+                return p
+        return None
+
+    def _conversations_dir(self) -> Path | None:
+        """Return the conversations directory that matches the history path."""
+        history = self._history_path()
+        if history is not None:
+            conversations = history.parent / "conversations"
+            if conversations.exists():
+                return conversations
+        for p in (_AGY_CONVERSATIONS, _AGY_ALT_CONVERSATIONS):
+            if p.exists():
+                return p
+        return None
+
+    def discover_sessions(self) -> list[Session]:
+        """Discover Antigravity sessions from history.jsonl.
+
+        history.jsonl contains one JSON object per user prompt.  Prompts that
+        belong to the same saved conversation share a ``conversationId``; we
+        group those into a single ``Session``.
+        """
+        sessions: list[Session] = []
+        history_path = self._history_path()
+        if history_path is None:
+            return sessions
+
+        # conversationId -> list of history entries
+        groups: dict[str, list[dict]] = {}
+
+        with history_path.open() as f:
+            for line in f:
+                line = line.strip()
+                if not line:
+                    continue
+                try:
+                    entry = json.loads(line)
+                except json.JSONDecodeError:
+                    continue
+
+                conversation_id = entry.get("conversationId")
+                if not conversation_id:
+                    # Skip un-persisted one-off prompts.
+                    continue
+                groups.setdefault(conversation_id, []).append(entry)
+
+        for conversation_id, entries in groups.items():
+            if not entries:
+                continue
+            # Use earliest timestamp as created_at, latest as updated_at.
+            timestamps = [e.get("timestamp", 0) for e in entries if e.get("timestamp")]
+            created_ts = min(timestamps) if timestamps else 0
+            updated_ts = max(timestamps) if timestamps else 0
+
+            created_at = _ms_to_dt(created_ts) if created_ts else datetime.now(timezone.utc)
+            updated_at = _ms_to_dt(updated_ts) if updated_ts else created_at
+
+            # Title from the first entry's display prompt.
+            first_display = entries[0].get("display", "")
+            title = first_display[:200] if first_display else None
+            cwd = entries[0].get("workspace") or None
+
+            sessions.append(
+                Session(
+                    id=conversation_id,
+                    tool=self.tool_name,
+                    title=title,
+                    cwd=cwd,
+                    created_at=created_at,
+                    updated_at=updated_at,
+                    raw_path=str(history_path),
+                    status="unknown",
+                )
+            )
+
+        return sessions
+
+    def load_messages(self, session_id: str) -> list[Message]:
+        """Load messages for a given Antigravity session ID.
+
+        Antigravity stores full conversations as encrypted protobuf (``.pb``)
+        files under ``conversations/``.  ContextForge cannot currently parse
+        those encrypted files, so this method attempts the following, in order:
+
+        1. Look for a JSON sidecar ``<session_id>.json`` (e.g. from a test
+           fixture or future export).
+        2. Read the history JSONL entries with this ``conversationId`` and
+           synthesise minimal ``Message`` objects from the user prompts.
+        """
+        conv_dir = self._conversations_dir()
+        if conv_dir is not None:
+            json_path = conv_dir / f"{session_id}.json"
+            if json_path.exists():
+                return self._parse_json_conversation(json_path)
+
+        # Fallback: reconstruct user turns from history.jsonl.
+        history_path = self._history_path()
+        if history_path is None:
+            return []
+
+        messages: list[Message] = []
+        with history_path.open() as f:
+            for line in f:
+                line = line.strip()
+                if not line:
+                    continue
+                try:
+                    entry = json.loads(line)
+                except json.JSONDecodeError:
+                    continue
+                if entry.get("conversationId") != session_id:
+                    continue
+
+                text = entry.get("display", "")
+                if not text:
+                    continue
+
+                ts = entry.get("timestamp")
+                timestamp = _ms_to_dt(ts) if ts else None
+                messages.append(
+                    Message(
+                        role="user",
+                        content=text,
+                        timestamp=timestamp,
+                        token_count=_count_tokens(text),
+                    )
+                )
+
+        return messages
+
+    def _parse_json_conversation(self, path: Path) -> list[Message]:
+        """Parse a JSON-sidecar conversation file into Messages.
+
+        Expected shape::
+
+            {
+                "messages": [
+                    {"role": "user", "content": "...", "timestamp": 1234567890000},
+                    {"role": "assistant", "content": "...", "timestamp": 1234567891000}
+                ]
+            }
+        """
+        messages: list[Message] = []
+        try:
+            with path.open() as f:
+                data = json.load(f)
+        except (json.JSONDecodeError, OSError):
+            return messages
+
+        for msg in data.get("messages", []):
+            role = msg.get("role", "")
+            if role not in ("user", "assistant"):
+                continue
+            content = str(msg.get("content", ""))
+            if not content:
+                continue
+            ts = msg.get("timestamp")
+            timestamp = _ms_to_dt(ts) if isinstance(ts, int) else None
+            messages.append(
+                Message(
+                    role=role,
+                    content=content,
+                    timestamp=timestamp,
+                    token_count=_count_tokens(content),
+                )
+            )
+
+        return messages
+
+    def build_inject_command(
+        self,
+        context: str,
+        target_session_id: str | None = None,
+        cwd: str | None = None,
+        method: str = "system_prompt",
+    ) -> str:
+        """Build a shell command to start/resume an Antigravity CLI session.
+
+        ``agy -i <prompt>`` starts an interactive session seeded with the given
+        prompt.  ``agy --conversation <id>`` resumes an existing conversation.
+        """
+        safe_ctx = shlex.quote(context)
+
+        if method == "resume" and target_session_id:
+            return f"agy --conversation {target_session_id}"
+
+        # Default: new interactive session with the context as the opening prompt.
+        cmd = f"agy -i {safe_ctx}"
+        if cwd:
+            cmd = f"cd {shlex.quote(cwd)} && {cmd}"
+        return cmd

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/contextforge/adapters/registry.py

@@ -6,6 +6,7 @@ from contextforge.adapters.base import ToolAdapter
 from contextforge.adapters.claude_code import ClaudeCodeAdapter
 from contextforge.adapters.claude_desktop import ClaudeDesktopAdapter
 from contextforge.adapters.codex import CodexAdapter
+from contextforge.adapters.antigravity import AntigravityAdapter
 from contextforge.adapters.gemini import GeminiAdapter
 
 ADAPTERS: dict[str, type[ToolAdapter]] = {
@@ -14,6 +15,7 @@ ADAPTERS: dict[str, type[ToolAdapter]] = {
     "codex": CodexAdapter,
     "altimate_code": AltimateCodeAdapter,
     "gemini": GeminiAdapter,
+    "antigravity": AntigravityAdapter,
 }
 
 

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/contextforge/tui/widgets/session_detail.py

@@ -16,6 +16,7 @@ TOOL_COLORS = {
     "altimate_code":  "magenta",
     "claude_desktop": "yellow",
     "gemini":         "blue",
+    "antigravity":    "white",
 }
 TOOL_LABELS = {
     "claude_code":    "◆ Claude Code",
@@ -23,6 +24,7 @@ TOOL_LABELS = {
     "altimate_code":  "⚡ Altimate",
     "claude_desktop": "◇ Claude Desktop",
     "gemini":         "✦ Gemini",
+    "antigravity":    "◉ Antigravity",
 }
 
 

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/contextforge/tui/widgets/session_table.py

@@ -19,6 +19,7 @@ TOOL_ICON = {
     "altimate_code":  "⚡",   # Altimate — lightning
     "claude_desktop": "◇",   # Claude Desktop — open diamond
     "gemini":         "✦",   # Google Gemini — four-pointed star
+    "antigravity":    "◉",   # Antigravity — circle
 }
 
 TOOL_MARKUP = {
@@ -27,6 +28,7 @@ TOOL_MARKUP = {
     "altimate_code":  "[magenta]⚡ Alt[/magenta]",
     "claude_desktop": "[yellow]◇ Desktop[/yellow]",
     "gemini":         "[blue]✦ Gemini[/blue]",
+    "antigravity":    "[white]◉ Agy[/white]",
 }
 
 TOOL_DISPLAY = {
@@ -35,6 +37,7 @@ TOOL_DISPLAY = {
     "altimate_code":  "⚡ Alt",
     "claude_desktop": "◇ Desktop",
     "gemini":         "✦ Gemini",
+    "antigravity":    "◉ Agy",
 }
 
 _FTOOL_MAP = {
@@ -43,6 +46,7 @@ _FTOOL_MAP = {
     "ftool-codex": "codex",
     "ftool-gem":   "gemini",
     "ftool-alt":   "altimate_code",
+    "ftool-agy":   "antigravity",
 }
 
 
@@ -159,6 +163,7 @@ class SessionTable(Widget):
                 yield Button("⬡ Codex", id="ftool-codex",  classes="filter-tool-btn")
                 yield Button("✦ Gem",   id="ftool-gem",    classes="filter-tool-btn")
                 yield Button("⚡ Alt",   id="ftool-alt",    classes="filter-tool-btn")
+                yield Button("◉ Agy",   id="ftool-agy",    classes="filter-tool-btn")
             yield DataTable(cursor_type="row", zebra_stripes=True, id="sessions-datatable")
 
     def on_mount(self) -> None:

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/contextforge/tui/widgets/stats_panel.py

@@ -21,6 +21,7 @@ TOOL_COLORS = {
     "altimate_code":  "magenta",
     "claude_desktop": "yellow",
     "gemini":         "blue",
+    "antigravity":    "white",
 }
 TOOL_LABELS = {
     "claude_code":    "◆ Claude Code",
@@ -28,9 +29,10 @@ TOOL_LABELS = {
     "altimate_code":  "⚡ Altimate",
     "claude_desktop": "◇ Desktop",
     "gemini":         "✦ Gemini",
+    "antigravity":    "◉ Antigravity",
 }
 
-_TOOL_ORDER = ["claude_code", "codex", "gemini", "altimate_code", "claude_desktop"]
+_TOOL_ORDER = ["claude_code", "codex", "gemini", "altimate_code", "claude_desktop", "antigravity"]
 
 WINDOW_LABELS = {
     "7d":  "7 days",

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/contextforge/tui/widgets/status_bar.py

@@ -69,6 +69,7 @@ class StatusBar(Widget):
             "altimate_code":  "⚡ Alt",
             "claude_desktop": "◇ Desktop",
             "gemini":         "✦ Gemini",
+            "antigravity":    "◉ Agy",
         }
         for tool, label in labels.items():
             n = counts.get(tool, 0)
@@ -95,7 +96,7 @@ class StatusBar(Widget):
         if text:
             parts.append(f'"{text}"')
         if tool:
-            short = {"claude_code": "◆ CC", "codex": "⬡ Codex", "altimate_code": "⚡ Alt", "claude_desktop": "◇ Desktop", "gemini": "✦ Gemini"}.get(tool, tool)
+            short = {"claude_code": "◆ CC", "codex": "⬡ Codex", "altimate_code": "⚡ Alt", "claude_desktop": "◇ Desktop", "gemini": "✦ Gemini", "antigravity": "◉ Agy"}.get(tool, tool)
             parts.append(short)
 
         self.query_one("#status-filter", Static).update(

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/contextforge/tui/widgets/tokens_panel.py

@@ -24,6 +24,7 @@ _TOOL_LABELS = {
     "altimate_code":  "⚡ Altimate",
     "claude_desktop": "◇ Claude Desktop",
     "gemini":         "✦ Gemini",
+    "antigravity":    "◉ Antigravity",
 }
 
 

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/contextforge/tui/widgets/transfer_panel.py

@@ -12,6 +12,7 @@ TOOLS = [
     ("claude_code", "Claude Code"),
     ("codex", "Codex"),
     ("altimate_code", "altimate-code"),
+    ("antigravity", "Antigravity"),
 ]
 
 STRATEGIES = [

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "context-forge-cli"
-version = "0.4.10"
+version = "0.4.11"
 description = "Session manager and context bridge for agentic CLI tools"
 readme = "README.md"
 license = { text = "MIT" }

context_forge_cli-0.4.11/tests/adapters/test_antigravity.py

@@ -0,0 +1,195 @@
+"""Tests for the Antigravity adapter using fixture data."""
+from __future__ import annotations
+
+import json
+import tempfile
+from datetime import datetime, timezone
+from pathlib import Path
+
+import pytest
+
+from contextforge.adapters.antigravity import AntigravityAdapter, _count_tokens, _ms_to_dt
+
+
+def test_ms_to_dt():
+    ts = 1779426988061  # milliseconds
+    dt = _ms_to_dt(ts)
+    assert dt.year == 2026
+    assert dt.tzinfo is not None
+
+
+def test_count_tokens():
+    text = "hello world"
+    n = _count_tokens(text)
+    assert isinstance(n, int) and n > 0
+
+
+def test_discover_sessions_from_fixture():
+    adapter = AntigravityAdapter()
+
+    with tempfile.TemporaryDirectory() as tmpdir:
+        history = Path(tmpdir) / ".gemini" / "antigravity-cli" / "history.jsonl"
+        history.parent.mkdir(parents=True)
+        entries = [
+            {
+                "display": "First prompt",
+                "timestamp": 1700000000000,
+                "workspace": "/tmp/ws1",
+                "conversationId": "conv-a",
+            },
+            {
+                "display": "Second prompt",
+                "timestamp": 1700000100000,
+                "workspace": "/tmp/ws1",
+                "conversationId": "conv-a",
+            },
+            {
+                "display": "Other prompt",
+                "timestamp": 1700001000000,
+                "workspace": "/tmp/ws2",
+                "conversationId": "conv-b",
+            },
+        ]
+        history.write_text("\n".join(json.dumps(e) for e in entries) + "\n")
+
+        import contextforge.adapters.antigravity as mod
+        original_history = mod._AGY_HISTORY
+        mod._AGY_HISTORY = history
+        try:
+            sessions = adapter.discover_sessions()
+        finally:
+            mod._AGY_HISTORY = original_history
+
+    assert len(sessions) == 2
+    ids = {s.id for s in sessions}
+    assert ids == {"conv-a", "conv-b"}
+
+    conv_a = next(s for s in sessions if s.id == "conv-a")
+    assert conv_a.title == "First prompt"
+    assert conv_a.cwd == "/tmp/ws1"
+    assert conv_a.tool == "antigravity"
+    assert conv_a.created_at == _ms_to_dt(1700000000000)
+    assert conv_a.updated_at == _ms_to_dt(1700000100000)
+
+
+def test_discover_skips_entries_without_conversation_id():
+    adapter = AntigravityAdapter()
+
+    with tempfile.TemporaryDirectory() as tmpdir:
+        history = Path(tmpdir) / ".gemini" / "antigravity-cli" / "history.jsonl"
+        history.parent.mkdir(parents=True)
+        entries = [
+            {"display": "orphan", "timestamp": 1700000000000, "workspace": "/tmp/ws"},
+            # No conversationId
+        ]
+        history.write_text("\n".join(json.dumps(e) for e in entries) + "\n")
+
+        import contextforge.adapters.antigravity as mod
+        original_history = mod._AGY_HISTORY
+        mod._AGY_HISTORY = history
+        try:
+            sessions = adapter.discover_sessions()
+        finally:
+            mod._AGY_HISTORY = original_history
+
+    assert sessions == []
+
+
+def test_load_messages_json_sidecar():
+    adapter = AntigravityAdapter()
+
+    with tempfile.TemporaryDirectory() as tmpdir:
+        history = Path(tmpdir) / ".gemini" / "antigravity-cli" / "history.jsonl"
+        history.parent.mkdir(parents=True)
+        history.write_text(json.dumps({"display": "hi", "timestamp": 1700000000000, "conversationId": "c1"}) + "\n")
+
+        conversations = history.parent / "conversations"
+        conversations.mkdir()
+        json_path = conversations / "c1.json"
+        json_path.write_text(
+            json.dumps(
+                {
+                    "messages": [
+                        {"role": "user", "content": "hello", "timestamp": 1700000000000},
+                        {"role": "assistant", "content": "world", "timestamp": 1700000010000},
+                    ]
+                }
+            )
+        )
+
+        import contextforge.adapters.antigravity as mod
+        original_history = mod._AGY_HISTORY
+        original_conv = mod._AGY_CONVERSATIONS
+        mod._AGY_HISTORY = history
+        mod._AGY_CONVERSATIONS = conversations
+        try:
+            messages = adapter.load_messages("c1")
+        finally:
+            mod._AGY_HISTORY = original_history
+            mod._AGY_CONVERSATIONS = original_conv
+
+    assert len(messages) == 2
+    assert messages[0].role == "user"
+    assert messages[0].content == "hello"
+    assert messages[1].role == "assistant"
+    assert messages[1].content == "world"
+
+
+def test_load_messages_fallback_to_history():
+    adapter = AntigravityAdapter()
+
+    with tempfile.TemporaryDirectory() as tmpdir:
+        history = Path(tmpdir) / ".gemini" / "antigravity-cli" / "history.jsonl"
+        history.parent.mkdir(parents=True)
+        entries = [
+            {
+                "display": "Prompt one",
+                "timestamp": 1700000000000,
+                "workspace": "/tmp/ws",
+                "conversationId": "c1",
+            },
+            {
+                "display": "Prompt two",
+                "timestamp": 1700000010000,
+                "workspace": "/tmp/ws",
+                "conversationId": "c1",
+            },
+        ]
+        history.write_text("\n".join(json.dumps(e) for e in entries) + "\n")
+
+        import contextforge.adapters.antigravity as mod
+        original_history = mod._AGY_HISTORY
+        mod._AGY_HISTORY = history
+        try:
+            messages = adapter.load_messages("c1")
+        finally:
+            mod._AGY_HISTORY = original_history
+
+    assert len(messages) == 2
+    assert messages[0].role == "user"
+    assert messages[0].content == "Prompt one"
+    assert messages[1].role == "user"
+    assert messages[1].content == "Prompt two"
+
+
+def test_build_inject_command_new_session():
+    adapter = AntigravityAdapter()
+    cmd = adapter.build_inject_command("some context", method="system_prompt")
+    assert "agy" in cmd
+    assert "-i" in cmd
+    assert "some context" in cmd
+
+
+def test_build_inject_command_resume():
+    adapter = AntigravityAdapter()
+    cmd = adapter.build_inject_command("ctx", target_session_id="abc-123", method="resume")
+    assert "agy" in cmd
+    assert "--conversation" in cmd
+    assert "abc-123" in cmd
+
+
+def test_build_inject_command_with_cwd():
+    adapter = AntigravityAdapter()
+    cmd = adapter.build_inject_command("ctx", cwd="/tmp/proj")
+    assert cmd.startswith("cd /tmp/proj")
+    assert "agy" in cmd

{context_forge_cli-0.4.10 → context_forge_cli-0.4.11}/uv.lock

@@ -206,7 +206,7 @@ wheels = [
 
 [[package]]
 name = "context-forge-cli"
-version = "0.2.5"
+version = "0.4.10"
 source = { editable = "." }
 dependencies = [
     { name = "anthropic" },