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

{onex_cli-1.16.3/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.3
+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.3 → onex_cli-1.16.4}/onex/__init__.py

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

{onex_cli-1.16.3 → 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)}"}}'
 
@@ -525,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.
@@ -533,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.3 → 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.3
+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