augint-shell 0.90.0__tar.gz → 0.90.2__tar.gz

{augint_shell-0.90.0 → augint_shell-0.90.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: augint-shell
-Version: 0.90.0
+Version: 0.90.2
 Summary: Launch AI coding tools and local LLMs in Docker containers
 Author: svange
 Requires-Dist: docker>=7.0.0

{augint_shell-0.90.0 → augint_shell-0.90.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "augint-shell"
-version = "0.90.0"
+version = "0.90.2"
 description = "Launch AI coding tools and local LLMs in Docker containers"
 authors = [{name = "svange"}]
 readme = "README.md"

{augint_shell-0.90.0 → augint_shell-0.90.2}/src/ai_shell/__init__.py

@@ -1,6 +1,6 @@
 """augint-shell (ai-shell) - Launch AI coding tools and local LLMs in Docker containers."""
 
-__version__ = "0.90.0"
+__version__ = "0.90.2"
 
 __all__ = [
     "__version__",

{augint_shell-0.90.0 → augint_shell-0.90.2}/src/ai_shell/cli/commands/tools.py

@@ -2,9 +2,11 @@
 
 from __future__ import annotations
 
+import json
 import logging
 import subprocess
 import sys
+import tempfile
 import uuid
 from pathlib import Path
 from typing import Any
@@ -1298,6 +1300,50 @@ def _check_ollama_running(container_name: str) -> None:
         )
 
 
+def _ensure_pi_ollama_provider(container_name: str, config: AiShellConfig) -> None:
+    """Register Ollama as a Pi provider if not already configured.
+
+    Ollama is not a built-in Pi provider — it must be declared in the global
+    ``~/.pi/agent/models.json``.  This writes the file only when absent so
+    user customizations are preserved.
+    """
+    check = subprocess.run(
+        ["docker", "exec", container_name, "test", "-f", "/root/.pi/agent/models.json"],
+        capture_output=True,
+    )
+    if check.returncode == 0:
+        return
+
+    models_cfg = {
+        "providers": {
+            "ollama": {
+                "baseUrl": "http://host.docker.internal:11434/v1",
+                "api": "openai-completions",
+                "apiKey": "ollama",
+                "compat": {
+                    "supportsDeveloperRole": False,
+                    "supportsReasoningEffort": False,
+                },
+                "models": [{"id": tag} for tag in config.models_to_pull],
+            }
+        }
+    }
+    with tempfile.NamedTemporaryFile(mode="w", suffix=".json", delete=False) as tmp:
+        json.dump(models_cfg, tmp, indent=2)
+        tmp_path = tmp.name
+    try:
+        subprocess.run(
+            ["docker", "exec", container_name, "mkdir", "-p", "/root/.pi/agent"],
+            capture_output=True,
+        )
+        subprocess.run(
+            ["docker", "cp", tmp_path, f"{container_name}:/root/.pi/agent/models.json"],
+            capture_output=True,
+        )
+    finally:
+        Path(tmp_path).unlink(missing_ok=True)
+
+
 @click.command(context_settings=CONTEXT_SETTINGS)
 @click.option("--aws", "use_aws", is_flag=True, default=False, help="Use Amazon Bedrock.")
 @click.option("--profile", "cli_profile", default=None, help="AWS profile for Bedrock auth.")
@@ -1355,7 +1401,13 @@ def pi(ctx, use_aws, cli_profile, openai_profile, do_login):
 
         cmd = ["pi"]
         if not use_bedrock and not resolved_openai_profile:
-            cmd.extend(["--model", f"ollama/{config.primary_coding_model}"])
+            _ensure_pi_ollama_provider(name, config)
+            pi_config = Path(config.project_dir) / ".pi" / "settings.json"
+            if not pi_config.is_file():
+                console.print(
+                    "[yellow]Warning: No Pi project config found (.pi/settings.json). "
+                    "Run 'ai-opencodex update' to generate project config.[/yellow]"
+                )
         console.print(f"[bold]Launching pi{bedrock_label}{openai_label} in {name}...[/bold]")
     manager.exec_interactive(name, cmd, extra_env=exec_env, typeahead=typeahead.bytes())
 

{augint_shell-0.90.0 → augint_shell-0.90.2}/src/ai_shell/templates/pi/models.json

@@ -4,6 +4,10 @@
       "baseUrl": "http://host.docker.internal:11434/v1",
       "api": "openai-completions",
       "apiKey": "ollama",
+      "compat": {
+        "supportsDeveloperRole": false,
+        "supportsReasoningEffort": false
+      },
       "models": [
         { "id": "qwen3-coder:30b-a3b-q4_K_M" },
         { "id": "huihui_ai/qwen3-coder-abliterated:30b-a3b-instruct-q4_K_M" },