augint-shell 0.1.0__tar.gz

augint_shell-0.1.0/PKG-INFO

@@ -0,0 +1,124 @@
+Metadata-Version: 2.3
+Name: augint-shell
+Version: 0.1.0
+Summary: Launch AI coding tools and local LLMs in Docker containers
+Author: svange
+Requires-Dist: docker>=7.0.0
+Requires-Dist: click>=8.1.0
+Requires-Dist: rich>=14.0
+Requires-Dist: pytest>=7.4.0 ; extra == 'dev'
+Requires-Dist: pytest-cov>=4.1.0 ; extra == 'dev'
+Requires-Dist: ruff>=0.8.0 ; extra == 'dev'
+Requires-Dist: mypy>=1.8.0 ; extra == 'dev'
+Requires-Dist: python-semantic-release>=10.3.1 ; extra == 'dev'
+Requires-Dist: pre-commit>=4.0.0 ; extra == 'dev'
+Requires-Dist: bandit>=1.7.0 ; extra == 'dev'
+Requires-Dist: pip-audit>=2.7.0 ; extra == 'dev'
+Requires-Dist: pip-licenses>=5.0.0 ; extra == 'dev'
+Requires-Dist: pytest-html>=4.0.0 ; extra == 'dev'
+Requires-Python: >=3.12
+Provides-Extra: dev
+Description-Content-Type: text/markdown
+
+# augint-shell
+
+Launch AI coding tools and local LLMs in Docker containers.
+
+## Installation
+
+```bash
+pip install augint-shell
+```
+
+Or as a dev dependency:
+
+```bash
+uv add --dev augint-shell
+```
+
+## Quick Start
+
+```bash
+# Launch Claude Code in the current project
+ai-shell claude
+
+# Launch with extra args
+ai-shell claude -- --debug
+
+# Set up local LLM stack (first time)
+ai-shell llm setup
+
+# Launch opencode with local LLM
+ai-shell opencode
+```
+
+## Commands
+
+### AI Tools
+
+| Command | Description |
+|---|---|
+| `ai-shell claude` | Launch Claude Code |
+| `ai-shell claude-x` | Claude Code with skip-permissions |
+| `ai-shell codex` | Launch Codex |
+| `ai-shell opencode` | Launch opencode |
+| `ai-shell aider` | Launch aider with local LLM |
+| `ai-shell shell` | Bash shell in dev container |
+
+### LLM Stack
+
+| Command | Description |
+|---|---|
+| `ai-shell llm up` | Start Ollama + Open WebUI |
+| `ai-shell llm down` | Stop LLM stack |
+| `ai-shell llm pull` | Pull configured models |
+| `ai-shell llm setup` | First-time setup (up + pull + configure) |
+| `ai-shell llm status` | Show status and available models |
+| `ai-shell llm logs` | Tail LLM stack logs |
+| `ai-shell llm shell` | Shell into Ollama container |
+
+### Container Management
+
+| Command | Description |
+|---|---|
+| `ai-shell manage status` | Show dev container status |
+| `ai-shell manage stop` | Stop dev container |
+| `ai-shell manage clean` | Remove container and volumes |
+| `ai-shell manage logs` | Tail dev container logs |
+| `ai-shell manage pull` | Pull latest Docker image |
+
+## Configuration
+
+Optional `ai-shell.toml` in your project root:
+
+```toml
+[container]
+image = "svange/augint-shell"
+image_tag = "latest"
+extra_env = { MY_VAR = "value" }
+
+[llm]
+primary_model = "qwen3.5:27b"
+fallback_model = "qwen3-coder-next"
+context_size = 32768
+ollama_port = 11434
+webui_port = 3000
+
+[aider]
+model = "ollama_chat/qwen3.5:27b"
+```
+
+Global config at `~/.config/ai-shell/config.toml` is also supported.
+
+## How It Works
+
+- Pulls a pre-built Docker image from Docker Hub (`svange/augint-shell`)
+- Creates per-project containers named `augint-shell-{project}-dev`
+- Mounts your project directory, SSH keys, AWS credentials, and tool configs
+- Runs AI tools interactively inside the container
+- Supports concurrent instances across multiple projects
+
+## Requirements
+
+- Docker
+- Python >= 3.12

augint_shell-0.1.0/README.md

@@ -0,0 +1,102 @@
+# augint-shell
+
+Launch AI coding tools and local LLMs in Docker containers.
+
+## Installation
+
+```bash
+pip install augint-shell
+```
+
+Or as a dev dependency:
+
+```bash
+uv add --dev augint-shell
+```
+
+## Quick Start
+
+```bash
+# Launch Claude Code in the current project
+ai-shell claude
+
+# Launch with extra args
+ai-shell claude -- --debug
+
+# Set up local LLM stack (first time)
+ai-shell llm setup
+
+# Launch opencode with local LLM
+ai-shell opencode
+```
+
+## Commands
+
+### AI Tools
+
+| Command | Description |
+|---|---|
+| `ai-shell claude` | Launch Claude Code |
+| `ai-shell claude-x` | Claude Code with skip-permissions |
+| `ai-shell codex` | Launch Codex |
+| `ai-shell opencode` | Launch opencode |
+| `ai-shell aider` | Launch aider with local LLM |
+| `ai-shell shell` | Bash shell in dev container |
+
+### LLM Stack
+
+| Command | Description |
+|---|---|
+| `ai-shell llm up` | Start Ollama + Open WebUI |
+| `ai-shell llm down` | Stop LLM stack |
+| `ai-shell llm pull` | Pull configured models |
+| `ai-shell llm setup` | First-time setup (up + pull + configure) |
+| `ai-shell llm status` | Show status and available models |
+| `ai-shell llm logs` | Tail LLM stack logs |
+| `ai-shell llm shell` | Shell into Ollama container |
+
+### Container Management
+
+| Command | Description |
+|---|---|
+| `ai-shell manage status` | Show dev container status |
+| `ai-shell manage stop` | Stop dev container |
+| `ai-shell manage clean` | Remove container and volumes |
+| `ai-shell manage logs` | Tail dev container logs |
+| `ai-shell manage pull` | Pull latest Docker image |
+
+## Configuration
+
+Optional `ai-shell.toml` in your project root:
+
+```toml
+[container]
+image = "svange/augint-shell"
+image_tag = "latest"
+extra_env = { MY_VAR = "value" }
+
+[llm]
+primary_model = "qwen3.5:27b"
+fallback_model = "qwen3-coder-next"
+context_size = 32768
+ollama_port = 11434
+webui_port = 3000
+
+[aider]
+model = "ollama_chat/qwen3.5:27b"
+```
+
+Global config at `~/.config/ai-shell/config.toml` is also supported.
+
+## How It Works
+
+- Pulls a pre-built Docker image from Docker Hub (`svange/augint-shell`)
+- Creates per-project containers named `augint-shell-{project}-dev`
+- Mounts your project directory, SSH keys, AWS credentials, and tool configs
+- Runs AI tools interactively inside the container
+- Supports concurrent instances across multiple projects
+
+## Requirements
+
+- Docker
+- Python >= 3.12

augint_shell-0.1.0/pyproject.toml

@@ -0,0 +1,122 @@
+[project]
+name = "augint-shell"
+version = "0.1.0"
+description = "Launch AI coding tools and local LLMs in Docker containers"
+authors = [{name = "svange"}]
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "docker>=7.0.0",
+    "click>=8.1.0",
+    "rich>=14.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.4.0",
+    "pytest-cov>=4.1.0",
+    "ruff>=0.8.0",
+    "mypy>=1.8.0",
+    "python-semantic-release>=10.3.1",
+    "pre-commit>=4.0.0",
+    "bandit>=1.7.0",
+    "pip-audit>=2.7.0",
+    "pip-licenses>=5.0.0",
+    "pytest-html>=4.0.0",
+]
+
+[project.scripts]
+ai-shell = "ai_shell.cli.__main__:main"
+
+[build-system]
+requires = ["uv_build>=0.9,<0.11"]
+build-backend = "uv_build"
+
+[tool.uv.build-backend]
+module-name = "ai_shell"
+
+[tool.ruff]
+line-length = 100
+target-version = "py312"
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "W", "B", "C4", "UP", "DTZ"]
+ignore = ["E501"]
+
+[tool.ruff.lint.isort]
+known-first-party = ["ai_shell"]
+
+[tool.mypy]
+python_version = "3.12"
+warn_return_any = true
+warn_unused_configs = true
+disallow_untyped_defs = true
+
+[[tool.mypy.overrides]]
+module = "ai_shell.cli.commands.*"
+disallow_untyped_defs = false
+
+[[tool.mypy.overrides]]
+module = "ai_shell.cli.__main__"
+disallow_untyped_defs = false
+
+[[tool.mypy.overrides]]
+module = "docker.*"
+ignore_missing_imports = true
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+addopts = "-ra -q --strict-markers"
+
+[tool.coverage.run]
+source = ["src"]
+omit = ["*/tests/*", "*/test_*.py", "*/cli/*"]
+
+[tool.semantic_release]
+assets = ["uv.lock"]
+commit_message = "chore(release): augint-shell {version}\n\nAutomatically generated by python-semantic-release [skip ci]"
+commit_parser = "angular"
+logging_use_named_masks = false
+major_on_zero = false
+allow_zero_version = true
+no_git_verify = false
+tag_format = "augint-shell-v{version}"
+version_toml = ["pyproject.toml:project.version"]
+version_variables = ["src/ai_shell/__init__.py:__version__"]
+build_command = "uv lock && uv build"
+
+[tool.semantic_release.branches.main]
+match = "main"
+prerelease = false
+prerelease_token = "rc"
+
+[tool.semantic_release.changelog]
+mode = "update"
+changelog_file = "CHANGELOG.md"
+exclude_commit_patterns = [
+    '''chore(?:\([^)]*?\))?: .+''',
+    '''ci(?:\([^)]*?\))?: .+''',
+    '''refactor(?:\([^)]*?\))?: .+''',
+    '''style(?:\([^)]*?\))?: .+''',
+    '''test(?:\([^)]*?\))?: .+''',
+    '''build\((?!deps\): .+)''',
+]
+
+[tool.semantic_release.remote]
+name = "origin"
+type = "github"
+ignore_token_for_push = false
+insecure = false
+
+[tool.semantic_release.remote.token]
+env = "GH_TOKEN"
+
+[tool.semantic_release.publish]
+dist_glob_patterns = ["dist/*"]
+upload_to_vcs_release = false
+
+[dependency-groups]
+dev = [
+    "augint-github>=1.3.1",
+]

augint_shell-0.1.0/src/ai_shell/__init__.py

@@ -0,0 +1,7 @@
+"""augint-shell (ai-shell) - Launch AI coding tools and local LLMs in Docker containers."""
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "__version__",
+]

augint_shell-0.1.0/src/ai_shell/cli/__main__.py

@@ -0,0 +1,46 @@
+"""ai-shell CLI entry point."""
+
+import sys
+
+import click
+
+from ai_shell import __version__
+from ai_shell.cli.commands.llm import llm_group
+from ai_shell.cli.commands.manage import manage_group
+from ai_shell.cli.commands.tools import aider, claude, claude_x, codex, opencode, shell
+
+
+@click.group()
+@click.version_option(version=__version__, prog_name="ai-shell")
+@click.option("--project", default=None, help="Override project name for container naming.")
+@click.pass_context
+def cli(ctx, project):
+    """AI Shell - Launch AI coding tools and local LLMs in Docker containers."""
+    ctx.ensure_object(dict)
+    ctx.obj["project"] = project
+
+
+# Tool subcommands
+cli.add_command(claude)
+cli.add_command(claude_x, "claude-x")
+cli.add_command(codex)
+cli.add_command(opencode)
+cli.add_command(aider)
+cli.add_command(shell)
+
+# Command groups
+cli.add_command(llm_group, "llm")
+cli.add_command(manage_group, "manage")
+
+
+def main():
+    """Main entry point."""
+    try:
+        cli()
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

augint_shell-0.1.0/src/ai_shell/cli/commands/llm.py

@@ -0,0 +1,204 @@
+"""LLM stack management commands: up, down, pull, setup, status, logs, shell."""
+
+import time
+from pathlib import Path
+
+import click
+from rich.console import Console
+
+from ai_shell.config import load_config
+from ai_shell.container import ContainerManager
+from ai_shell.defaults import OLLAMA_CONTAINER, WEBUI_CONTAINER
+
+console = Console(stderr=True)
+
+
+def _get_manager(ctx) -> ContainerManager:
+    """Create ContainerManager from Click context."""
+    project = ctx.obj.get("project") if ctx.obj else None
+    config = load_config(project_override=project, project_dir=Path.cwd())
+    return ContainerManager(config)
+
+
+@click.group("llm")
+@click.pass_context
+def llm_group(ctx):
+    """Manage the local LLM stack (Ollama + Open WebUI)."""
+
+
+@llm_group.command("up")
+@click.pass_context
+def llm_up(ctx):
+    """Start the LLM stack (Ollama + Open WebUI)."""
+    manager = _get_manager(ctx)
+    console.print("[bold]Starting LLM stack...[/bold]")
+
+    manager.ensure_ollama()
+    console.print(f"  Ollama API:  http://localhost:{manager.config.ollama_port}")
+
+    manager.ensure_webui()
+    console.print(f"  Open WebUI:  http://localhost:{manager.config.webui_port}")
+
+    console.print("\n[bold green]LLM stack is running.[/bold green]")
+    console.print("If this is your first time, run: [bold]ai-shell llm setup[/bold]")
+
+
+@llm_group.command("down")
+@click.pass_context
+def llm_down(ctx):
+    """Stop the LLM stack."""
+    manager = _get_manager(ctx)
+    console.print("[bold]Stopping LLM stack...[/bold]")
+
+    for name in [WEBUI_CONTAINER, OLLAMA_CONTAINER]:
+        status = manager.container_status(name)
+        if status == "running":
+            manager.stop_container(name)
+            console.print(f"  Stopped: {name}")
+        elif status is not None:
+            console.print(f"  Already stopped: {name}")
+        else:
+            console.print(f"  Not found: {name}")
+
+    console.print("[bold green]LLM stack stopped.[/bold green]")
+
+
+@llm_group.command("pull")
+@click.pass_context
+def llm_pull(ctx):
+    """Pull LLM models into Ollama."""
+    manager = _get_manager(ctx)
+    config = manager.config
+
+    console.print(f"[bold]Pulling primary model: {config.primary_model}...[/bold]")
+    output = manager.exec_in_ollama(["ollama", "pull", config.primary_model])
+    console.print(output)
+
+    console.print(f"\n[bold]Pulling fallback model: {config.fallback_model}...[/bold]")
+    output = manager.exec_in_ollama(["ollama", "pull", config.fallback_model])
+    console.print(output)
+
+    console.print("\n[bold]Available models:[/bold]")
+    output = manager.exec_in_ollama(["ollama", "list"])
+    console.print(output)
+
+
+@llm_group.command("setup")
+@click.pass_context
+def llm_setup(ctx):
+    """First-time setup: start stack, pull models, configure context window."""
+    manager = _get_manager(ctx)
+    config = manager.config
+
+    # Start the stack
+    console.print("[bold]Starting LLM stack...[/bold]")
+    manager.ensure_ollama()
+    manager.ensure_webui()
+
+    # Wait for Ollama to be ready
+    console.print("[bold]Waiting for Ollama to be ready...[/bold]")
+    for i in range(10):
+        try:
+            output = manager.exec_in_ollama(["ollama", "list"])
+            if output is not None:
+                break
+        except Exception:
+            pass
+        console.print(f"  Waiting... ({i + 1}/10)")
+        time.sleep(2)
+    else:
+        console.print("[bold red]Ollama failed to start after 20s[/bold red]")
+        raise click.Abort()
+
+    # Pull models
+    console.print(f"\n[bold]Pulling primary model: {config.primary_model}...[/bold]")
+    output = manager.exec_in_ollama(["ollama", "pull", config.primary_model])
+    console.print(output)
+
+    console.print(f"\n[bold]Pulling fallback model: {config.fallback_model}...[/bold]")
+    output = manager.exec_in_ollama(["ollama", "pull", config.fallback_model])
+    console.print(output)
+
+    # Configure context window
+    console.print(f"\n[bold]Configuring context window ({config.context_size} tokens)...[/bold]")
+    for model in [config.primary_model, config.fallback_model]:
+        modelfile = f"FROM {model}\nPARAMETER num_ctx {config.context_size}\n"
+        # Write modelfile and create model
+        manager.exec_in_ollama(
+            [
+                "sh",
+                "-c",
+                f'printf "{modelfile}" > /tmp/Modelfile && '
+                f"ollama create {model} -f /tmp/Modelfile && rm -f /tmp/Modelfile",
+            ]
+        )
+
+    console.print("\n[bold green]============================================[/bold green]")
+    console.print("[bold green] Setup complete![/bold green]")
+    console.print(f"\n  Open WebUI:  http://localhost:{config.webui_port}")
+    console.print(f"  Ollama API:  http://localhost:{config.ollama_port}")
+    console.print(f"\n  Primary model:  {config.primary_model}")
+    console.print(f"  Fallback model: {config.fallback_model}")
+    console.print(f"  Context window: {config.context_size} tokens")
+    console.print("[bold green]============================================[/bold green]")
+
+
+@llm_group.command("status")
+@click.pass_context
+def llm_status(ctx):
+    """Show status of LLM stack and loaded models."""
+    manager = _get_manager(ctx)
+
+    console.print("[bold]Container status:[/bold]")
+    for name in [OLLAMA_CONTAINER, WEBUI_CONTAINER]:
+        status = manager.container_status(name)
+        if status == "running":
+            console.print(f"  {name}: [green]{status}[/green]")
+        elif status is not None:
+            console.print(f"  {name}: [yellow]{status}[/yellow]")
+        else:
+            console.print(f"  {name}: [red]not found[/red]")
+
+    # Show models if ollama is running
+    if manager.container_status(OLLAMA_CONTAINER) == "running":
+        console.print("\n[bold]Available models:[/bold]")
+        output = manager.exec_in_ollama(["ollama", "list"])
+        console.print(output)
+
+
+@llm_group.command("logs")
+@click.option("--follow", "-f", is_flag=True, help="Follow log output.")
+@click.pass_context
+def llm_logs(ctx, follow):
+    """Tail logs from the LLM stack."""
+    manager = _get_manager(ctx)
+    if follow:
+        # Use docker CLI for multi-container following
+        import os
+        import sys
+
+        sys.stdout.flush()
+        sys.stderr.flush()
+        os.execvp(
+            "docker",
+            ["docker", "logs", "-f", OLLAMA_CONTAINER],
+        )
+    else:
+        for name in [OLLAMA_CONTAINER, WEBUI_CONTAINER]:
+            status = manager.container_status(name)
+            if status is not None:
+                console.print(f"\n[bold]--- {name} ---[/bold]")
+                manager.container_logs(name, follow=False, tail=50)
+
+
+@llm_group.command("shell")
+@click.pass_context
+def llm_shell(ctx):
+    """Open a bash shell in the Ollama container."""
+    manager = _get_manager(ctx)
+    status = manager.container_status(OLLAMA_CONTAINER)
+    if status != "running":
+        console.print("[red]Ollama is not running. Run: ai-shell llm up[/red]")
+        raise click.Abort()
+    console.print("[bold]Opening shell in Ollama container...[/bold]")
+    manager.exec_interactive(OLLAMA_CONTAINER, ["/bin/bash"])