PyPI - weio-cli - Versions diffs - 0.3.0__tar.gz → 0.4.0__tar.gz - Mend

weio-cli 0.3.0tar.gz → 0.4.0tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (21) hide show

{weio_cli-0.3.0/src/weio_cli.egg-info → weio_cli-0.4.0}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weio-cli
-Version: 0.3.0
+Version: 0.4.0
 Summary: Weio — an agentic coding assistant that routes inference through your Weio account.
 Author: We I/O Labs
 License: MIT
@@ -35,14 +35,29 @@ Requires Python 3.9+.
 ## Authenticate
-Generate an API key in **Settings → API & CLI** on weio.ai, then:
+```bash
+weio login            # opens your browser, sign in (incl. Google), key is created automatically
+```
+`weio login` starts a one-time local handshake, opens weio.ai in your browser to
+sign in, mints an API key for this device, and saves it to `~/.weio/config.json`.
+No copy-paste.
+Prefer to paste a key yourself? Generate one in **Settings → API & CLI** on
+weio.ai and:
 ```bash
-weio login                  # paste your weio_sk_… key (saved to ~/.weio/config.json)
-# or, per-session:
+weio login --no-browser     # paste your weio_sk_… key
+# or, per session:
 export WEIO_API_KEY="weio_sk_…"
 ```
+Check what you're using:
+```bash
+weio usage            # tier, tokens used today, remaining, reset time
+```
 ## Use
 ### Interactive TUI (default)

{weio_cli-0.3.0 → weio_cli-0.4.0}/README.md RENAMED Viewed

@@ -13,14 +13,29 @@ Requires Python 3.9+.
 ## Authenticate
-Generate an API key in **Settings → API & CLI** on weio.ai, then:
+```bash
+weio login            # opens your browser, sign in (incl. Google), key is created automatically
+```
+`weio login` starts a one-time local handshake, opens weio.ai in your browser to
+sign in, mints an API key for this device, and saves it to `~/.weio/config.json`.
+No copy-paste.
+Prefer to paste a key yourself? Generate one in **Settings → API & CLI** on
+weio.ai and:
 ```bash
-weio login                  # paste your weio_sk_… key (saved to ~/.weio/config.json)
-# or, per-session:
+weio login --no-browser     # paste your weio_sk_… key
+# or, per session:
 export WEIO_API_KEY="weio_sk_…"
 ```
+Check what you're using:
+```bash
+weio usage            # tier, tokens used today, remaining, reset time
+```
 ## Use
 ### Interactive TUI (default)

{weio_cli-0.3.0 → weio_cli-0.4.0}/pyproject.toml RENAMED Viewed

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "weio-cli"
-version = "0.3.0"
+version = "0.4.0"
 description = "Weio — an agentic coding assistant that routes inference through your Weio account."
 readme = "README.md"
 requires-python = ">=3.9"

{weio_cli-0.3.0 → weio_cli-0.4.0}/src/weio_cli/__init__.py RENAMED Viewed

@@ -1,3 +1,3 @@
 """Weio CLI — an agentic coding assistant that routes inference through your Weio account."""
-__version__ = "0.3.0"
+__version__ = "0.4.0"

weio_cli-0.4.0/src/weio_cli/browser_login.py ADDED Viewed

@@ -0,0 +1,88 @@
+"""Browser-based login: open Weio in the browser, sign in (incl. Google), and
+receive a freshly-minted API key on a local loopback server. No manual key paste.
+Standard OAuth-CLI loopback pattern (like gcloud / gh): we start a localhost
+server, open the browser to the Weio authorize page with our port + a random
+state, the page mints a key and redirects back to the loopback with it.
+"""
+from __future__ import annotations
+import http.server
+import secrets
+import socketserver
+import sys
+import time
+import urllib.parse
+import webbrowser
+from typing import Optional
+def _web_base_from_api(api_base: str) -> str:
+    """https://weio.ai/v1 -> https://weio.ai"""
+    b = (api_base or "").rstrip("/")
+    if b.endswith("/v1"):
+        b = b[:-3]
+    return b.rstrip("/")
+_DONE_HTML = (
+    b"<!doctype html><html><head><meta charset='utf-8'><title>Weio CLI</title></head>"
+    b"<body style='font-family:-apple-system,Segoe UI,sans-serif;background:#0d1117;"
+    b"color:#e6edf3;display:flex;align-items:center;justify-content:center;height:100vh;margin:0'>"
+    b"<div style='text-align:center'><h2 style='color:#3fb950'>&#10003; Weio CLI authorized</h2>"
+    b"<p>You can close this tab and return to your terminal.</p></div></body></html>"
+)
+_FAIL_HTML = (
+    b"<!doctype html><html><body style='font-family:sans-serif'>"
+    b"<h2>Authorization failed.</h2><p>Please run <code>weio login</code> again.</p></body></html>"
+)
+def browser_login(api_base: str, timeout: float = 180.0) -> Optional[str]:
+    """Open the browser and return the minted API key, or None on failure/timeout."""
+    state = secrets.token_urlsafe(16)
+    result: dict = {}
+    class Handler(http.server.BaseHTTPRequestHandler):
+        def do_GET(self):  # noqa: N802
+            params = urllib.parse.parse_qs(urllib.parse.urlparse(self.path).query)
+            key = (params.get("key") or [None])[0]
+            st = (params.get("state") or [None])[0]
+            ok = bool(key) and st == state
+            self.send_response(200)
+            self.send_header("Content-Type", "text/html; charset=utf-8")
+            self.end_headers()
+            self.wfile.write(_DONE_HTML if ok else _FAIL_HTML)
+            if ok:
+                result["key"] = key
+        def log_message(self, *args):  # silence
+            pass
+    try:
+        httpd = socketserver.TCPServer(("127.0.0.1", 0), Handler)
+    except OSError as e:
+        print(f"Could not start local auth server: {e}", file=sys.stderr)
+        return None
+    httpd.timeout = 1
+    port = httpd.server_address[1]
+    auth_url = (f"{_web_base_from_api(api_base)}/cli/auth"
+                f"?port={port}&state={urllib.parse.quote(state)}")
+    print("Opening your browser to sign in to Weio…")
+    print(f"  {auth_url}")
+    print("  (If it doesn't open, copy that URL into your browser.)")
+    try:
+        webbrowser.open(auth_url)
+    except Exception:
+        pass
+    deadline = time.time() + timeout
+    try:
+        while "key" not in result and time.time() < deadline:
+            httpd.handle_request()
+    except KeyboardInterrupt:
+        pass
+    finally:
+        httpd.server_close()
+    return result.get("key")

{weio_cli-0.3.0 → weio_cli-0.4.0}/src/weio_cli/cli.py RENAMED Viewed

@@ -43,14 +43,33 @@ def _c(text: str, code: str) -> str:
 def cmd_login(args) -> int:
     cli_key = getattr(args, "key", None)
     cli_base = getattr(args, "base", None)
-    key = cli_key or input("Weio API key (weio_sk_…): ").strip()
+    base = config.resolve_base(cli_base)
+    no_browser = getattr(args, "no_browser", False)
+    key = cli_key
+    # Default: open the browser, sign in (incl. Google), auto-mint a key.
+    if not key and not no_browser:
+        try:
+            from .browser_login import browser_login
+            key = browser_login(base)
+        except Exception as e:
+            print(_c(f"Browser sign-in unavailable ({e}).", "33"), file=sys.stderr)
+        if not key:
+            print(_c("Browser sign-in didn't complete. Falling back to manual key entry.", "33"))
+    # Fallback / explicit: paste a key from Settings → API & CLI.
+    if not key:
+        try:
+            key = input("Weio API key (weio_sk_…): ").strip()
+        except (EOFError, KeyboardInterrupt):
+            print()
+            return 2
     if not key:
         print("No key provided.", file=sys.stderr)
         return 2
     path = config.save(key=key, base=cli_base)
-    # Validate
     try:
-        client = WeioClient(key, config.resolve_base(cli_base))
+        client = WeioClient(key, base)
         ok = client.ping()
     except WeioError:
         ok = False
@@ -128,6 +147,32 @@ def cmd_chat(args) -> int:
     return 0
+def cmd_usage(args) -> int:
+    try:
+        client = _make_client(args)
+    except WeioError as e:
+        print(_c(str(e), "31"), file=sys.stderr)
+        return 2
+    try:
+        u = client.usage()
+    except WeioError as e:
+        print(_c(str(e), "31"), file=sys.stderr)
+        return 1
+    used = u.get("daily_tokens_used", 0)
+    limit = u.get("daily_token_limit", 0)
+    remaining = u.get("daily_tokens_remaining", 0)
+    pct = (used / limit * 100) if limit else 0
+    bar_n = int(pct / 5)
+    bar = "█" * bar_n + "░" * (20 - bar_n)
+    print(_c(f"  account  ", "2") + f"{u.get('email','?')}  ({u.get('tier','?')})")
+    print(_c(f"  today    ", "2") + f"{used:,} / {limit:,} tokens  ({pct:.0f}%)")
+    print(_c(f"           ", "2") + _c(bar, "32" if pct < 80 else "33"))
+    print(_c(f"  remaining", "2") + f" {remaining:,} tokens")
+    if u.get("reset_at"):
+        print(_c(f"  resets   ", "2") + f"{u['reset_at']}")
+    return 0
 def cmd_update(args) -> int:
     from . import updater
     latest = updater.check_for_update(force=True)
@@ -236,7 +281,7 @@ def build_parser() -> argparse.ArgumentParser:
     common.add_argument("--key", "-k", default=argparse.SUPPRESS,
                         help="API key (or set WEIO_API_KEY / run `weio login`)")
     common.add_argument("--base", default=argparse.SUPPRESS,
-                        help="API base URL (default https://api.weio.ai/v1)")
+                        help="API base URL (default https://weio.ai/v1)")
     common.add_argument("--model", "-m", default=argparse.SUPPRESS, help="Model id (default: auto)")
     common.add_argument("--max-tokens", type=int, default=argparse.SUPPRESS, help="Max output tokens")
@@ -249,7 +294,10 @@ def build_parser() -> argparse.ArgumentParser:
     sub = p.add_subparsers(dest="command")
-    sp = sub.add_parser("login", parents=[common], help="Save your API key")
+    sp = sub.add_parser("login", parents=[common],
+                        help="Sign in via browser (or --no-browser to paste a key)")
+    sp.add_argument("--no-browser", action="store_true",
+                    help="Skip the browser flow and paste an API key manually")
     sp.set_defaults(func=cmd_login)
     sp = sub.add_parser("ping", parents=[common], help="Check connectivity and key")
@@ -267,6 +315,9 @@ def build_parser() -> argparse.ArgumentParser:
     sp.add_argument("--dir", "-C", default=".", help="Project directory (default: cwd)")
     sp.set_defaults(func=cmd_tui)
+    sp = sub.add_parser("usage", parents=[common], help="Show your token usage and limits")
+    sp.set_defaults(func=cmd_usage)
     sp = sub.add_parser("update", parents=[common], help="Update weio-cli to the latest version")
     sp.set_defaults(func=cmd_update)
@@ -284,7 +335,7 @@ def main(argv=None) -> int:
     argv = list(sys.argv[1:] if argv is None else argv)
     parser = build_parser()
-    known = {"login", "ping", "ask", "chat", "code", "tui", "update"}
+    known = {"login", "ping", "ask", "chat", "code", "tui", "update", "usage"}
     # Bare `weio` (no args) → interactive coding TUI, like Claude Code.
     if not argv:
         argv = ["tui"]

{weio_cli-0.3.0 → weio_cli-0.4.0}/src/weio_cli/client.py RENAMED Viewed

@@ -108,6 +108,21 @@ class WeioClient:
         except httpx.HTTPError as e:
             raise WeioError(f"Network error talking to {self.base}: {e}") from e
+    def usage(self) -> dict:
+        """Fetch account usage + limits."""
+        try:
+            r = httpx.get(f"{self.base}/usage", headers=self._headers(), timeout=15.0)
+        except httpx.HTTPError as e:
+            raise WeioError(f"Network error talking to {self.base}: {e}") from e
+        if r.status_code == 401:
+            raise WeioError("Authentication failed (401). Check your API key.")
+        if r.status_code >= 400:
+            raise WeioError(f"Gateway error {r.status_code}: {r.text[:200]}")
+        try:
+            return r.json()
+        except json.JSONDecodeError as e:
+            raise WeioError(f"Unexpected response: {r.text[:200]}") from e
     def ping(self) -> bool:
         """Check the gateway is reachable and the key is valid (GET /models)."""
         try:

{weio_cli-0.3.0 → weio_cli-0.4.0}/src/weio_cli/config.py RENAMED Viewed

@@ -13,7 +13,7 @@ import os
 from pathlib import Path
 from typing import Optional
-DEFAULT_BASE = "https://api.weio.ai/v1"
+DEFAULT_BASE = "https://weio.ai/v1"
 CONFIG_DIR = Path(os.environ.get("WEIO_CONFIG_DIR", str(Path.home() / ".weio")))
 CONFIG_FILE = CONFIG_DIR / "config.json"

{weio_cli-0.3.0 → weio_cli-0.4.0/src/weio_cli.egg-info}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weio-cli
-Version: 0.3.0
+Version: 0.4.0
 Summary: Weio — an agentic coding assistant that routes inference through your Weio account.
 Author: We I/O Labs
 License: MIT
@@ -35,14 +35,29 @@ Requires Python 3.9+.
 ## Authenticate
-Generate an API key in **Settings → API & CLI** on weio.ai, then:
+```bash
+weio login            # opens your browser, sign in (incl. Google), key is created automatically
+```
+`weio login` starts a one-time local handshake, opens weio.ai in your browser to
+sign in, mints an API key for this device, and saves it to `~/.weio/config.json`.
+No copy-paste.
+Prefer to paste a key yourself? Generate one in **Settings → API & CLI** on
+weio.ai and:
 ```bash
-weio login                  # paste your weio_sk_… key (saved to ~/.weio/config.json)
-# or, per-session:
+weio login --no-browser     # paste your weio_sk_… key
+# or, per session:
 export WEIO_API_KEY="weio_sk_…"
 ```
+Check what you're using:
+```bash
+weio usage            # tier, tokens used today, remaining, reset time
+```
 ## Use
 ### Interactive TUI (default)

{weio_cli-0.3.0 → weio_cli-0.4.0}/src/weio_cli.egg-info/SOURCES.txt RENAMED Viewed

@@ -3,6 +3,7 @@ README.md
 pyproject.toml
 src/weio_cli/__init__.py
 src/weio_cli/__main__.py
+src/weio_cli/browser_login.py
 src/weio_cli/cli.py
 src/weio_cli/client.py
 src/weio_cli/coder.py