weio-cli 0.2.0__tar.gz → 0.4.0__tar.gz

{weio_cli-0.2.0/src/weio_cli.egg-info → weio_cli-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weio-cli
-Version: 0.2.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)
@@ -93,6 +108,19 @@ weio ping
 Edits are shown as a diff and require confirmation before anything is written
 (use `-y`/`--yes` to apply automatically). New files are created as needed.
 
+## Updating
+
+weio-cli checks PyPI once a day (fail-silent) and prints a one-line notice when a
+newer version is available. To upgrade:
+
+```bash
+weio update           # upgrades in place via pip
+# or
+pip install -U weio-cli
+```
+
+Disable the check with `WEIO_NO_UPDATE_CHECK=1`.
+
 ## Configuration
 
 | Setting | Flag | Env | Config file |

{weio_cli-0.2.0 → weio_cli-0.4.0}/README.md

@@ -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)
@@ -71,6 +86,19 @@ weio ping
 Edits are shown as a diff and require confirmation before anything is written
 (use `-y`/`--yes` to apply automatically). New files are created as needed.
 
+## Updating
+
+weio-cli checks PyPI once a day (fail-silent) and prints a one-line notice when a
+newer version is available. To upgrade:
+
+```bash
+weio update           # upgrades in place via pip
+# or
+pip install -U weio-cli
+```
+
+Disable the check with `WEIO_NO_UPDATE_CHECK=1`.
+
 ## Configuration
 
 | Setting | Flag | Env | Config file |

{weio_cli-0.2.0 → weio_cli-0.4.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "weio-cli"
-version = "0.2.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.2.0 → weio_cli-0.4.0}/src/weio_cli/__init__.py

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

weio_cli-0.4.0/src/weio_cli/browser_login.py

@@ -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.2.0 → weio_cli-0.4.0}/src/weio_cli/cli.py

@@ -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,42 @@ 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)
+    if not latest:
+        print(_c(f"weio-cli {__version__} is up to date.", "32"))
+        return 0
+    print(_c(f"Updating weio-cli {__version__} → {latest}…", "36"))
+    return updater.self_update()
+
+
 def cmd_tui(args) -> int:
     try:
         client = _make_client(args)
@@ -226,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")
 
@@ -239,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")
@@ -257,6 +315,12 @@ 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)
+
     sp = sub.add_parser("code", parents=[common], help="Run a coding task in a directory")
     sp.add_argument("message")
     sp.add_argument("--file", "-f", action="append", help="Add a file to the context (repeatable)")
@@ -271,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"}
+    known = {"login", "ping", "ask", "chat", "code", "tui", "update", "usage"}
     # Bare `weio` (no args) → interactive coding TUI, like Claude Code.
     if not argv:
         argv = ["tui"]
@@ -284,10 +348,21 @@ def main(argv=None) -> int:
         parser.print_help()
         return 1
     try:
-        return args.func(args)
+        rc = args.func(args)
     except KeyboardInterrupt:
         print()
         return 130
+    # Throttled, fail-silent update notice (skipped for tui — it shows its own
+    # in the header — and for the update command itself).
+    if args.command not in ("update", "tui"):
+        try:
+            from . import updater
+            notice = updater.update_notice()
+            if notice:
+                print(_c("\n" + notice, "33"), file=sys.stderr)
+        except Exception:
+            pass
+    return rc
 
 
 if __name__ == "__main__":

{weio_cli-0.2.0 → weio_cli-0.4.0}/src/weio_cli/client.py

@@ -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.2.0 → weio_cli-0.4.0}/src/weio_cli/config.py

@@ -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.2.0 → weio_cli-0.4.0}/src/weio_cli/tui.py

@@ -283,6 +283,13 @@ class Repl:
         print(_c(f"  {self.client.base}  ·  " +
                  ("connected" if ok else "⚠ cannot reach gateway / bad key"),
                  GREEN if ok else RED))
+        try:
+            from .updater import update_notice
+            notice = update_notice()
+            if notice:
+                print(_c("  ⬆ " + notice, YELLOW))
+        except Exception:
+            pass
         print(_c("  Type a request, or /help. Ctrl-D to exit.\n", DIM))
         while True:
             text = self._read()

weio_cli-0.4.0/src/weio_cli/updater.py

@@ -0,0 +1,114 @@
+"""Update checking + self-update.
+
+A throttled, fail-silent check against PyPI (once per day) tells the user when a
+newer weio-cli is published. `weio update` upgrades in place. Disable the check
+with WEIO_NO_UPDATE_CHECK=1.
+"""
+from __future__ import annotations
+
+import json
+import os
+import subprocess
+import sys
+import time
+from pathlib import Path
+from typing import Optional
+
+from . import __version__
+
+INDEX_JSON = os.environ.get("WEIO_UPDATE_INDEX", "https://pypi.org/pypi/weio-cli/json")
+_CACHE_NAME = "update_check.json"
+_TTL_HOURS = 24
+
+
+def _cache_dir() -> Path:
+    return Path(os.environ.get("WEIO_CONFIG_DIR", str(Path.home() / ".weio")))
+
+
+def _parse_version(v: str) -> tuple:
+    """Lenient numeric version tuple, e.g. '0.10.2' -> (0, 10, 2)."""
+    out = []
+    for part in (v or "").split("."):
+        num = ""
+        for ch in part:
+            if ch.isdigit():
+                num += ch
+            else:
+                break
+        out.append(int(num) if num else 0)
+    return tuple(out) or (0,)
+
+
+def is_newer(latest: str, current: str) -> bool:
+    return _parse_version(latest) > _parse_version(current)
+
+
+def fetch_latest(timeout: float = 2.5) -> Optional[str]:
+    try:
+        import httpx
+        r = httpx.get(INDEX_JSON, timeout=timeout,
+                      headers={"User-Agent": f"weio-cli/{__version__}"})
+        if r.status_code != 200:
+            return None
+        return r.json()["info"]["version"]
+    except Exception:
+        return None
+
+
+def check_for_update(current: str = __version__, *, ttl_hours: int = _TTL_HOURS,
+                     force: bool = False) -> Optional[str]:
+    """Return the latest version string if newer than `current`, else None.
+
+    Throttled to once per ttl_hours via a cache file; never raises.
+    """
+    if os.environ.get("WEIO_NO_UPDATE_CHECK") == "1":
+        return None
+    try:
+        cache = _cache_dir() / _CACHE_NAME
+        data = {}
+        if cache.exists():
+            try:
+                data = json.loads(cache.read_text())
+            except Exception:
+                data = {}
+        now = time.time()
+        latest = data.get("latest")
+        last = float(data.get("last_check", 0) or 0)
+
+        if force or not latest or (now - last) > ttl_hours * 3600:
+            fetched = fetch_latest()
+            if fetched:
+                latest = fetched
+                try:
+                    cache.parent.mkdir(parents=True, exist_ok=True)
+                    cache.write_text(json.dumps({"last_check": now, "latest": latest}))
+                except Exception:
+                    pass
+
+        if latest and is_newer(latest, current):
+            return latest
+    except Exception:
+        pass
+    return None
+
+
+def update_notice(current: str = __version__) -> Optional[str]:
+    """One-line, color-free notice if an update is available (for printing)."""
+    latest = check_for_update(current)
+    if not latest:
+        return None
+    return (f"weio-cli {latest} is available (you have {current}). "
+            f"Update with: weio update")
+
+
+def self_update(pre: bool = False) -> int:
+    """Upgrade weio-cli in place using the running interpreter's pip."""
+    cmd = [sys.executable, "-m", "pip", "install", "--upgrade", "weio-cli"]
+    if pre:
+        cmd.append("--pre")
+    print(f"Running: {' '.join(cmd)}")
+    try:
+        return subprocess.call(cmd)
+    except Exception as e:
+        print(f"Update failed: {e}", file=sys.stderr)
+        return 1

{weio_cli-0.2.0 → weio_cli-0.4.0/src/weio_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weio-cli
-Version: 0.2.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)
@@ -93,6 +108,19 @@ weio ping
 Edits are shown as a diff and require confirmation before anything is written
 (use `-y`/`--yes` to apply automatically). New files are created as needed.
 
+## Updating
+
+weio-cli checks PyPI once a day (fail-silent) and prints a one-line notice when a
+newer version is available. To upgrade:
+
+```bash
+weio update           # upgrades in place via pip
+# or
+pip install -U weio-cli
+```
+
+Disable the check with `WEIO_NO_UPDATE_CHECK=1`.
+
 ## Configuration
 
 | Setting | Flag | Env | Config file |

{weio_cli-0.2.0 → weio_cli-0.4.0}/src/weio_cli.egg-info/SOURCES.txt

@@ -3,11 +3,13 @@ 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
 src/weio_cli/config.py
 src/weio_cli/tui.py
+src/weio_cli/updater.py
 src/weio_cli.egg-info/PKG-INFO
 src/weio_cli.egg-info/SOURCES.txt
 src/weio_cli.egg-info/dependency_links.txt