llamapass 0.1.0__tar.gz

llamapass-0.1.0/.gitignore

@@ -0,0 +1,15 @@
+__pycache__/
+*.py[cod]
+*.so
+db.sqlite3
+.env
+staticfiles/
+*.egg-info/
+dist/
+build/
+.venv/
+venv/
+ollama-tunnel.sh
+/*.png
+!static/*.png
+!screens/*.png

llamapass-0.1.0/PKG-INFO

@@ -0,0 +1,33 @@
+Metadata-Version: 2.4
+Name: llamapass
+Version: 0.1.0
+Summary: CLI client for LlamaPass - Ollama gateway with authentication
+License-Expression: MIT
+Requires-Python: >=3.10
+Requires-Dist: httpx>=0.27
+Description-Content-Type: text/markdown
+
+# LlamaPass CLI
+
+CLI client for [LlamaPass](https://llamapass.org) - an Ollama gateway with authentication.
+
+## Install
+
+```bash
+pip install llamapass
+```
+
+## Setup
+
+```bash
+llamapass config set-url https://llamapass.org
+llamapass config set-key oah_your_api_key
+```
+
+## Usage
+
+```bash
+llamapass run gemma3       # interactive chat
+llamapass list             # list available models
+llamapass config show      # show current config
+```

llamapass-0.1.0/README.md

@@ -0,0 +1,24 @@
+# LlamaPass CLI
+
+CLI client for [LlamaPass](https://llamapass.org) - an Ollama gateway with authentication.
+
+## Install
+
+```bash
+pip install llamapass
+```
+
+## Setup
+
+```bash
+llamapass config set-url https://llamapass.org
+llamapass config set-key oah_your_api_key
+```
+
+## Usage
+
+```bash
+llamapass run gemma3       # interactive chat
+llamapass list             # list available models
+llamapass config show      # show current config
+```

llamapass-0.1.0/llamapass_cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

llamapass-0.1.0/llamapass_cli/__main__.py

@@ -0,0 +1,3 @@
+from llamapass_cli.cli import main
+
+main()

llamapass-0.1.0/llamapass_cli/cli.py

@@ -0,0 +1,100 @@
+import argparse
+import sys
+
+from llamapass_cli import __version__
+from llamapass_cli.client import chat_stream, list_models
+from llamapass_cli.config import get, load, set_value
+
+
+def cmd_run(args):
+    model = args.model
+    messages = []
+    print(f"LlamaPass - {model} (type /bye to quit)\n")
+
+    while True:
+        try:
+            user_input = input(">>> ")
+        except (KeyboardInterrupt, EOFError):
+            print("\nBye!")
+            break
+
+        if user_input.strip().lower() in ("/bye", "/exit", "/quit"):
+            print("Bye!")
+            break
+
+        if not user_input.strip():
+            continue
+
+        messages.append({"role": "user", "content": user_input})
+        response = chat_stream(model, messages)
+        if response:
+            messages.append({"role": "assistant", "content": response})
+
+
+def cmd_list(args):
+    list_models()
+
+
+def cmd_config(args):
+    action = args.action
+
+    if action == "show":
+        cfg = load()
+        print(f"  url:     {cfg['url']}")
+        print(f"  api_key: {cfg['api_key'][:10]}..." if cfg["api_key"] else "  api_key: (not set)")
+
+    elif action == "set-url":
+        if not args.value:
+            print("Usage: llamapass config set-url <url>")
+            sys.exit(1)
+        set_value("url", args.value)
+        print(f"URL set to: {args.value}")
+
+    elif action == "set-key":
+        if not args.value:
+            print("Usage: llamapass config set-key <key>")
+            sys.exit(1)
+        set_value("api_key", args.value)
+        print(f"API key saved.")
+
+    else:
+        print(f"Unknown action: {action}")
+        print("Available: show, set-url, set-key")
+        sys.exit(1)
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        prog="llamapass",
+        description="CLI client for LlamaPass",
+    )
+    parser.add_argument("-v", "--version", action="version", version=f"%(prog)s {__version__}")
+
+    subparsers = parser.add_subparsers(dest="command")
+
+    # run
+    run_parser = subparsers.add_parser("run", help="Chat with a model")
+    run_parser.add_argument("model", help="Model name (e.g. gemma3)")
+
+    # list
+    subparsers.add_parser("list", help="List available models")
+    subparsers.add_parser("ls", help="List available models")
+
+    # config
+    config_parser = subparsers.add_parser("config", help="Manage configuration")
+    config_parser.add_argument("action", help="show | set-url | set-key")
+    config_parser.add_argument("value", nargs="?", help="Value to set")
+
+    args = parser.parse_args()
+
+    if args.command is None:
+        parser.print_help()
+        sys.exit(0)
+
+    commands = {
+        "run": cmd_run,
+        "list": cmd_list,
+        "ls": cmd_list,
+        "config": cmd_config,
+    }
+    commands[args.command](args)

llamapass-0.1.0/llamapass_cli/client.py

@@ -0,0 +1,99 @@
+import json
+import sys
+
+import httpx
+
+from llamapass_cli.config import load
+
+
+def _get_client():
+    cfg = load()
+    url = cfg["url"].rstrip("/")
+    api_key = cfg.get("api_key", "")
+    if not api_key:
+        print("Error: no API key configured. Run: llamapass config set-key <key>")
+        sys.exit(1)
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json",
+    }
+    return url, headers
+
+
+def list_models():
+    url, headers = _get_client()
+    try:
+        resp = httpx.get(f"{url}/ollama/api/tags", headers=headers, timeout=30)
+        resp.raise_for_status()
+        data = resp.json()
+        models = data.get("models", [])
+        if not models:
+            print("No models available.")
+            return
+        print(f"{'NAME':<40} {'SIZE':<12} {'MODIFIED'}")
+        for m in models:
+            name = m.get("name", "?")
+            size_gb = m.get("size", 0) / (1024**3)
+            modified = m.get("modified_at", "?")[:19]
+            print(f"{name:<40} {size_gb:<12.1f}GB {modified}")
+    except httpx.ConnectError:
+        print(f"Error: cannot connect to {url}")
+        sys.exit(1)
+    except httpx.HTTPStatusError as e:
+        _handle_http_error(e.response)
+
+
+def chat_stream(model, messages):
+    url, headers = _get_client()
+    payload = {
+        "model": model,
+        "messages": messages,
+        "stream": True,
+    }
+    try:
+        with httpx.stream(
+            "POST",
+            f"{url}/ollama/api/chat",
+            json=payload,
+            headers=headers,
+            timeout=None,
+        ) as resp:
+            if resp.status_code != 200:
+                _handle_http_error(resp)
+                return ""
+            full_response = []
+            for line in resp.iter_lines():
+                if not line:
+                    continue
+                try:
+                    data = json.loads(line)
+                    content = data.get("message", {}).get("content", "")
+                    if content:
+                        print(content, end="", flush=True)
+                        full_response.append(content)
+                    if data.get("done"):
+                        break
+                except json.JSONDecodeError:
+                    continue
+            print()
+            return "".join(full_response)
+    except httpx.ConnectError:
+        print(f"\nError: cannot connect to {url}")
+        sys.exit(1)
+
+
+def _handle_http_error(resp):
+    try:
+        data = resp.json()
+        error = data.get("error", resp.status_code)
+    except Exception:
+        error = resp.status_code
+
+    messages = {
+        401: "Invalid API key. Run: llamapass config set-key <key>",
+        403: f"Access denied: {error}",
+        429: "Rate limited. Try again later.",
+        502: "LlamaPass server cannot reach upstream Ollama.",
+    }
+    print(f"Error: {messages.get(resp.status_code, f'HTTP {resp.status_code}: {error}')}")
+    sys.exit(1)

llamapass-0.1.0/llamapass_cli/config.py

@@ -0,0 +1,33 @@
+import json
+from pathlib import Path
+
+CONFIG_DIR = Path.home() / ".config" / "llamapass"
+CONFIG_FILE = CONFIG_DIR / "config.json"
+
+DEFAULTS = {
+    "url": "https://llamapass.org",
+    "api_key": "",
+}
+
+
+def load():
+    if CONFIG_FILE.exists():
+        with open(CONFIG_FILE) as f:
+            return {**DEFAULTS, **json.load(f)}
+    return dict(DEFAULTS)
+
+
+def save(cfg):
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    with open(CONFIG_FILE, "w") as f:
+        json.dump(cfg, f, indent=2)
+
+
+def get(key):
+    return load().get(key)
+
+
+def set_value(key, value):
+    cfg = load()
+    cfg[key] = value
+    save(cfg)

llamapass-0.1.0/pyproject.toml

@@ -0,0 +1,20 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "llamapass"
+version = "0.1.0"
+description = "CLI client for LlamaPass - Ollama gateway with authentication"
+readme = "README.md"
+requires-python = ">=3.10"
+license = "MIT"
+dependencies = [
+    "httpx>=0.27",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["llamapass_cli"]
+
+[project.scripts]
+llamapass = "llamapass_cli.cli:main"