onex-cli 1.16.2__tar.gz → 1.16.4__tar.gz

{onex_cli-1.16.2/onex_cli.egg-info → onex_cli-1.16.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onex-cli
-Version: 1.16.2
+Version: 1.16.4
 Summary: Official CLI for deploying and managing services on OneXEOS Platform (Windows, macOS, Linux)
 Home-page: https://github.com/onexeos/onex-cli
 Author: OneXEOS Platform Team

{onex_cli-1.16.2 → onex_cli-1.16.4}/onex/__init__.py

@@ -1,2 +1,2 @@
 # OneXEOS Services CLI
-__version__ = "1.16.2"
+__version__ = "1.16.4"

{onex_cli-1.16.2 → onex_cli-1.16.4}/onex/mcp/tools/rocket_tools.py

@@ -130,11 +130,25 @@ def _detect_git_repo() -> Optional[dict]:
 
 def register_rocket_tools(mcp, api_url: Optional[str] = None):
     """Register all onex-rocket tools on an existing FastMCP server."""
+    import time as _time
+
     base_url = _resolve_rocket_url(api_url)
     client = httpx.Client(base_url=base_url, timeout=10.0)
     _indexed_repos: set = set()  # track repos already indexed this session
     _indexing_in_progress: set = set()  # repos currently being indexed
 
+    # Session tracking for savings estimation
+    _session_start = _time.time()
+    _call_counts: dict = {}  # tool_name -> count
+    _total_response_tokens = 0  # approximate tokens from MCP responses
+
+    def _track_call(tool_name: str, response_len: int):
+        """Track a tool call for session savings estimation."""
+        nonlocal _total_response_tokens
+        _call_counts[tool_name] = _call_counts.get(tool_name, 0) + 1
+        # Rough estimate: 1 token ≈ 4 chars
+        _total_response_tokens += response_len // 4
+
     def _check_index_status() -> Optional[dict]:
         """Check server-side indexing status for repos in progress.
         Returns status dict if indexing, None if all done."""
@@ -167,6 +181,7 @@ def register_rocket_tools(mcp, api_url: Optional[str] = None):
             resp = client.get(f"/api/v1/{path}", params=params or {})
             resp.raise_for_status()
             result = resp.text
+            _track_call(path.split("/")[0], len(result))
             if result in ("null", "[]", "{}"):
                 active = _check_index_status()
                 if active:
@@ -183,7 +198,9 @@ def register_rocket_tools(mcp, api_url: Optional[str] = None):
         try:
             resp = client.post(f"/api/v1/{path}", json=data)
             resp.raise_for_status()
-            return resp.text
+            result = resp.text
+            _track_call(path, len(result))
+            return result
         except httpx.HTTPError as e:
             return f'{{"error": "{str(e)}"}}'
 
@@ -254,6 +271,75 @@ def register_rocket_tools(mcp, api_url: Optional[str] = None):
             return None
         return repo_info["graph_name"]
 
+    # GitHub org used to construct clone URLs for auto-indexing
+    GITHUB_ORG = "OneXApis"
+
+    # Map of known repo folder names to GitHub repo names
+    KNOWN_REPOS = {
+        "onexerp-platform": "onex-platform",
+        "onexerp-services": "onex-services",
+        "onexeos-graph": "onex-rocket",
+    }
+
+    def _ensure_repo_by_name(repo_name: str):
+        """Auto-index a repo by name if not already in the graph.
+
+        Constructs the GitHub URL from the known org and triggers
+        async indexing on the server. Non-blocking.
+        """
+        if not repo_name or repo_name in _indexed_repos or repo_name in _indexing_in_progress:
+            return
+
+        # Check if already in graph via index/status
+        try:
+            resp = client.get(f"/api/v1/index/status/{repo_name}", timeout=3)
+            if resp.status_code == 200:
+                status = resp.json()
+                if status.get("status") == "done":
+                    _indexed_repos.add(repo_name)
+                    return
+                if status.get("status") == "indexing":
+                    _indexing_in_progress.add(repo_name)
+                    return
+        except Exception:
+            pass
+
+        # Check graph stats as fallback
+        try:
+            resp = client.get("/api/v1/stats", timeout=3)
+            if resp.status_code == 200:
+                by_repo = resp.json().get("by_repo", {})
+                if repo_name in by_repo:
+                    _indexed_repos.add(repo_name)
+                    return
+        except Exception:
+            return
+
+        # Not in graph — construct URL and trigger async index
+        # Extract branch from name: "repo@branch" -> branch, "repo" -> "main"
+        if "@" in repo_name:
+            base_name = repo_name.split("@")[0]
+            branch = repo_name.split("@")[1]
+        else:
+            base_name = repo_name
+            branch = "main"
+
+        # Resolve GitHub repo name
+        github_name = KNOWN_REPOS.get(base_name, base_name)
+        url = f"https://github.com/{GITHUB_ORG}/{github_name}.git"
+
+        try:
+            resp = client.post("/api/v1/index", json={
+                "url": url,
+                "name": repo_name,
+                "branch": branch,
+                "async": True,
+            }, timeout=5)
+            if resp.status_code == 200:
+                _indexing_in_progress.add(repo_name)
+        except Exception:
+            pass
+
     # ================================================================
     # IMPORTANT: Graph context for AI agents
     # The code graph indexes the MAIN branch only. Results do not reflect
@@ -294,36 +380,40 @@ def register_rocket_tools(mcp, api_url: Optional[str] = None):
     def rocket_find_dead_code(repo: str = "") -> str:
         """Find functions/methods never called anywhere in the codebase.
         BEST FOR: "what code can I safely delete?". Grep can't do this — only a graph can determine zero callers.
-        If repo is omitted, auto-detects from current git repo and branch."""
-        _ensure_repo_indexed()
+        If repo is omitted, auto-detects from current git repo and branch.
+        Auto-indexes the repo if not already in the graph."""
         repo = repo or _get_current_graph_name() or "unknown"
+        _ensure_repo_by_name(repo)
         return _get(f"repo/{repo}/dead-code")
 
     @mcp.tool()
     def rocket_find_hotspots(repo: str = "", limit: int = 20) -> str:
         """Find the most depended-on symbols, ranked by caller count.
         BEST FOR: "what are the riskiest parts?", "what should we test most?".
-        If repo is omitted, auto-detects from current git repo and branch."""
-        _ensure_repo_indexed()
+        If repo is omitted, auto-detects from current git repo and branch.
+        Auto-indexes the repo if not already in the graph."""
         repo = repo or _get_current_graph_name() or "unknown"
+        _ensure_repo_by_name(repo)
         return _get(f"repo/{repo}/hotspots", {"limit": limit})
 
     @mcp.tool()
     def rocket_find_cycles(repo: str = "") -> str:
         """Find circular dependencies in the codebase.
         BEST FOR: "are there any import cycles?", architectural analysis.
-        If repo is omitted, auto-detects from current git repo and branch."""
-        _ensure_repo_indexed()
+        If repo is omitted, auto-detects from current git repo and branch.
+        Auto-indexes the repo if not already in the graph."""
         repo = repo or _get_current_graph_name() or "unknown"
+        _ensure_repo_by_name(repo)
         return _get(f"repo/{repo}/cycles")
 
     @mcp.tool()
     def rocket_get_repo_map(repo: str = "") -> str:
         """Get a bird's-eye view of the entire repo: all files with their key symbols.
         BEST FOR: understanding project structure, onboarding. Much faster than reading files one by one.
-        If repo is omitted, auto-detects from current git repo and branch."""
-        _ensure_repo_indexed()
+        If repo is omitted, auto-detects from current git repo and branch.
+        Auto-indexes the repo if not already in the graph."""
         repo = repo or _get_current_graph_name() or "unknown"
+        _ensure_repo_by_name(repo)
         return _get(f"repo/{repo}/map")
 
     @mcp.tool()
@@ -396,16 +486,17 @@ def register_rocket_tools(mcp, api_url: Optional[str] = None):
     @mcp.tool()
     def rocket_get_file_churn(repo: str = "", limit: int = 20) -> str:
         """Get files ranked by change frequency. BEST FOR: identifying unstable/buggy code.
-        If repo is omitted, auto-detects from current git repo and branch."""
-        _ensure_repo_indexed()
+        If repo is omitted, auto-detects from current git repo and branch.
+        Auto-indexes the repo if not already in the graph."""
         repo = repo or _get_current_graph_name() or "unknown"
+        _ensure_repo_by_name(repo)
         return _get(f"repo/{repo}/churn", {"limit": limit})
 
     @mcp.tool()
     def rocket_stats() -> str:
         """Get graph statistics: total nodes/edges, breakdown by kind and repo.
-        Auto-indexes the current repo if not already in the graph (non-blocking)."""
-        _ensure_repo_indexed()
+        Auto-indexes the current git repo if not already in the graph (non-blocking)."""
+        _ensure_repo_indexed()  # index cwd repo if possible
         return _get("stats")
 
     @mcp.tool()
@@ -451,6 +542,7 @@ def register_rocket_tools(mcp, api_url: Optional[str] = None):
             data["name"] = name
         return _post("index", data)
 
+    @mcp.tool()
     @mcp.tool()
     def rocket_index_status(repo: str = "") -> str:
         """Check indexing status for a repo or all repos.
@@ -459,3 +551,45 @@ def register_rocket_tools(mcp, api_url: Optional[str] = None):
         if repo:
             return _get(f"index/status/{repo}")
         return _get("index/status")
+
+    @mcp.tool()
+    def rocket_session_savings() -> str:
+        """Show how many tokens and time the code graph saved in this session.
+        Call at the end of a conversation to report savings.
+        Returns: total calls, tokens used via MCP, estimated tokens without MCP, savings percentage."""
+        elapsed = _time.time() - _session_start
+        total_calls = sum(_call_counts.values())
+
+        # Estimation: without the graph, each query would require reading ~5 files
+        # averaging ~200 lines = ~1000 tokens per file = ~5000 tokens per query.
+        # The graph returns structured data at ~1/4 of that cost.
+        est_without_mcp = total_calls * 5000
+        est_with_mcp = _total_response_tokens
+        est_saved = max(0, est_without_mcp - est_with_mcp)
+        savings_pct = (est_saved / est_without_mcp * 100) if est_without_mcp > 0 else 0
+
+        # Cost estimates (Opus pricing: ~$15/M input, ~$75/M output)
+        cost_per_token = 15.0 / 1_000_000  # input token cost
+        est_cost_mcp = est_with_mcp * cost_per_token
+        est_cost_without = est_without_mcp * cost_per_token
+        est_cost_saved = est_cost_without - est_cost_mcp
+
+        # Time estimate: ~2s per file read avoided
+        est_time_saved_sec = total_calls * 5 * 2  # 5 files * 2s each
+
+        minutes = int(elapsed // 60)
+        seconds = int(elapsed % 60)
+
+        return json.dumps({
+            "session_duration": f"{minutes}m{seconds}s",
+            "total_calls": total_calls,
+            "calls_by_tool": _call_counts,
+            "mcp_tokens_used": est_with_mcp,
+            "est_without_mcp": est_without_mcp,
+            "est_tokens_saved": est_saved,
+            "est_savings_percent": round(savings_pct, 1),
+            "est_cost_mcp_usd": round(est_cost_mcp, 5),
+            "est_cost_without_usd": round(est_cost_without, 5),
+            "est_cost_saved_usd": round(est_cost_saved, 5),
+            "est_time_saved": f"{est_time_saved_sec // 60}m{est_time_saved_sec % 60}s" if est_time_saved_sec >= 60 else f"{est_time_saved_sec}s",
+        })

{onex_cli-1.16.2 → onex_cli-1.16.4/onex_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onex-cli
-Version: 1.16.2
+Version: 1.16.4
 Summary: Official CLI for deploying and managing services on OneXEOS Platform (Windows, macOS, Linux)
 Home-page: https://github.com/onexeos/onex-cli
 Author: OneXEOS Platform Team