stackfix 0.1.0__tar.gz → 0.1.2__tar.gz

{stackfix-0.1.0 → stackfix-0.1.2}/.github/workflows/stackfix.yml

@@ -19,7 +19,7 @@ jobs:
 
       - name: Install stackfix with GGUF backend
         run: |
-          pip install stackfix[gguf]
+          pip install stackfix
           pip install cfn-lint
 
       - name: Download GGUF model

{stackfix-0.1.0 → stackfix-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stackfix
-Version: 0.1.0
+Version: 0.1.2
 Summary: AI-powered Infrastructure-as-Code repair — fix broken Terraform and CloudFormation on CPU
 Author: Jon
 License-Expression: MIT
@@ -8,9 +8,9 @@ License-File: LICENSE
 Requires-Python: >=3.11
 Requires-Dist: cfn-lint>=1.0
 Requires-Dist: click>=8.0
-Requires-Dist: datasets>=3.0
 Requires-Dist: httpx>=0.27
-Requires-Dist: ollama>=0.4
+Requires-Dist: huggingface-hub>=0.20
+Requires-Dist: llama-cpp-python>=0.3
 Requires-Dist: python-hcl2>=5.0
 Requires-Dist: pyyaml>=6.0
 Requires-Dist: rich>=13.0
@@ -18,10 +18,13 @@ Provides-Extra: dev
 Requires-Dist: pytest-asyncio>=0.24; extra == 'dev'
 Requires-Dist: pytest>=8.0; extra == 'dev'
 Requires-Dist: ruff>=0.8; extra == 'dev'
-Provides-Extra: gguf
-Requires-Dist: llama-cpp-python>=0.3; extra == 'gguf'
 Provides-Extra: mlx
 Requires-Dist: mlx-lm>=0.21; extra == 'mlx'
+Provides-Extra: ollama
+Requires-Dist: ollama>=0.4; extra == 'ollama'
+Provides-Extra: training
+Requires-Dist: datasets>=3.0; extra == 'training'
+Requires-Dist: ollama>=0.4; extra == 'training'
 Description-Content-Type: text/markdown
 
 # Cloud-Gym
@@ -37,7 +40,7 @@ The `stackfix` CLI tool validates and repairs broken IaC files using fine-tuned
 ### Install
 
 ```bash
-pip install stackfix[gguf]
+pip install stackfix
 ```
 
 ### Download a Model
@@ -86,9 +89,9 @@ All models are fine-tuned Qwen2.5-Coder with LoRA, exported to GGUF. They run on
 
 | Backend | Install | Platform | Use Case |
 |---|---|---|---|
-| `gguf` | `pip install stackfix[gguf]` | Any (CPU) | CI/CD, Lambda, servers |
+| `gguf` | `pip install stackfix` | Any (CPU) | Default — CI/CD, Lambda, servers |
 | `mlx` | `pip install stackfix[mlx]` | Apple Silicon | Local dev on Mac |
-| `ollama` | `pip install stackfix` + Ollama | Any | When Ollama is already running |
+| `ollama` | `pip install stackfix[ollama]` + Ollama | Any | When Ollama is already running |
 
 ### CI/CD Integration
 
@@ -97,7 +100,7 @@ Add to your GitHub Actions workflow to catch IaC errors on every PR:
 ```yaml
 - name: Check IaC
   run: |
-    pip install stackfix[gguf]
+    pip install stackfix
     python -c "
     from huggingface_hub import hf_hub_download
     hf_hub_download('Tetsuto/iac-repair-3b-gguf', 'iac-repair-3b-q4.gguf', local_dir='.')

{stackfix-0.1.0 → stackfix-0.1.2}/README.md

@@ -11,7 +11,7 @@ The `stackfix` CLI tool validates and repairs broken IaC files using fine-tuned
 ### Install
 
 ```bash
-pip install stackfix[gguf]
+pip install stackfix
 ```
 
 ### Download a Model
@@ -60,9 +60,9 @@ All models are fine-tuned Qwen2.5-Coder with LoRA, exported to GGUF. They run on
 
 | Backend | Install | Platform | Use Case |
 |---|---|---|---|
-| `gguf` | `pip install stackfix[gguf]` | Any (CPU) | CI/CD, Lambda, servers |
+| `gguf` | `pip install stackfix` | Any (CPU) | Default — CI/CD, Lambda, servers |
 | `mlx` | `pip install stackfix[mlx]` | Apple Silicon | Local dev on Mac |
-| `ollama` | `pip install stackfix` + Ollama | Any | When Ollama is already running |
+| `ollama` | `pip install stackfix[ollama]` + Ollama | Any | When Ollama is already running |
 
 ### CI/CD Integration
 
@@ -71,7 +71,7 @@ Add to your GitHub Actions workflow to catch IaC errors on every PR:
 ```yaml
 - name: Check IaC
   run: |
-    pip install stackfix[gguf]
+    pip install stackfix
     python -c "
     from huggingface_hub import hf_hub_download
     hf_hub_download('Tetsuto/iac-repair-3b-gguf', 'iac-repair-3b-q4.gguf', local_dir='.')

{stackfix-0.1.0 → stackfix-0.1.2}/cloudgym/fixer/cli.py

@@ -30,6 +30,27 @@ stderr_console = Console(stderr=True)
 _repairer = None
 
 
+DEFAULT_GGUF_REPO = "Tetsuto/iac-repair-3b-gguf"
+DEFAULT_GGUF_FILE = "iac-repair-3b-q4.gguf"
+
+
+def _resolve_gguf_model(model: str | None) -> str:
+    """Resolve the GGUF model path, auto-downloading if needed."""
+    if model and Path(model).exists():
+        return model
+
+    # Auto-download the default model from HuggingFace
+    from huggingface_hub import hf_hub_download
+
+    repo = DEFAULT_GGUF_REPO
+    filename = DEFAULT_GGUF_FILE
+
+    console.print(f"[dim]Downloading model {filename} from {repo}...[/dim]")
+    path = hf_hub_download(repo_id=repo, filename=filename)
+    console.print(f"[dim]Model cached at {path}[/dim]")
+    return path
+
+
 def _get_repairer(backend: str, model: str | None, adapter: str | None):
     """Get or create the repairer instance."""
     global _repairer
@@ -46,12 +67,8 @@ def _get_repairer(backend: str, model: str | None, adapter: str | None):
     elif backend == "gguf":
         from cloudgym.fixer.repairer import GGUFRepairer
 
-        if not model:
-            raise click.ClickException(
-                "GGUF backend requires --model pointing to a .gguf file. "
-                "Export one with: python scripts/export_gguf.py 0.5b"
-            )
-        _repairer = GGUFRepairer(model_path=model)
+        model_path = _resolve_gguf_model(model)
+        _repairer = GGUFRepairer(model_path=model_path)
     elif backend == "ollama":
         from cloudgym.fixer.repairer import OllamaRepairer
 
@@ -63,7 +80,7 @@ def _get_repairer(backend: str, model: str | None, adapter: str | None):
 
 
 @click.group()
-@click.version_option(version="0.1.0", prog_name="stackfix")
+@click.version_option(version="0.1.2", prog_name="stackfix")
 def cli():
     """AI-powered Infrastructure-as-Code repair.
 
@@ -115,7 +132,7 @@ def check(files: tuple[str, ...], fmt: str | None):
 @click.option("-o", "--output", type=click.Path(), help="Write fixed output to this file")
 @click.option("--format", "fmt", type=click.Choice(["terraform", "cloudformation", "opentofu"]),
               default=None, help="Override auto-detected format")
-@click.option("--backend", type=click.Choice(["mlx", "gguf", "ollama"]), default="mlx",
+@click.option("--backend", type=click.Choice(["mlx", "gguf", "ollama"]), default="gguf",
               help="Model backend (default: mlx)")
 @click.option("--model", default=None, help="Override base model")
 @click.option("--adapter", default=None, help="Override adapter path")
@@ -305,7 +322,7 @@ def _show_diff(original: str, repaired: str, filename: str, color: bool):
 
 @cli.command(name="pre-commit")
 @click.argument("files", nargs=-1, type=click.Path(exists=True))
-@click.option("--backend", type=click.Choice(["mlx", "gguf", "ollama"]), default="mlx")
+@click.option("--backend", type=click.Choice(["mlx", "gguf", "ollama"]), default="gguf")
 @click.option("--model", default=None)
 @click.option("--adapter", default=None)
 def pre_commit(files: tuple[str, ...], backend: str, model: str | None, adapter: str | None):
@@ -375,7 +392,7 @@ def pre_commit(files: tuple[str, ...], backend: str, model: str | None, adapter:
 @click.argument("files", nargs=-1, required=True, type=click.Path(exists=True))
 @click.option("--format", "fmt", type=click.Choice(["terraform", "cloudformation", "opentofu"]),
               default=None, help="Override auto-detected format")
-@click.option("--backend", type=click.Choice(["mlx", "gguf", "ollama"]), default="mlx")
+@click.option("--backend", type=click.Choice(["mlx", "gguf", "ollama"]), default="gguf")
 @click.option("--model", default=None)
 @click.option("--adapter", default=None)
 def discuss(files: tuple[str, ...], fmt: str | None, backend: str, model: str | None, adapter: str | None):
@@ -411,7 +428,7 @@ def discuss(files: tuple[str, ...], fmt: str | None, backend: str, model: str |
 
 
 @cli.command(name="git-diff")
-@click.option("--backend", type=click.Choice(["mlx", "gguf", "ollama"]), default="mlx")
+@click.option("--backend", type=click.Choice(["mlx", "gguf", "ollama"]), default="gguf")
 @click.option("--model", default=None)
 @click.option("--adapter", default=None)
 @click.option("--apply", is_flag=True, help="Auto-fix and stage repaired files")

{stackfix-0.1.0 → stackfix-0.1.2}/examples/.pre-commit-config.yaml

@@ -7,7 +7,7 @@ repos:
         name: stackfix (AI-powered IaC repair)
         entry: stackfix pre-commit
         language: python
-        additional_dependencies: ['stackfix[gguf]']
+        additional_dependencies: ['stackfix']
         types_or: [terraform, yaml]
         # Only run on IaC-related files
         files: '\.(tf|yaml|yml)$'

{stackfix-0.1.0 → stackfix-0.1.2}/examples/LINKEDIN_POST.md

@@ -46,7 +46,7 @@ cat broken.tf | stackfix repair - --backend gguf --model model.gguf > fixed.tf
 - **Pre-commit hook**: Catches errors before they're committed. Fully offline.
 - **GitHub Actions**: Free CI gate — block PRs with broken IaC configs.
 - **AWS Lambda**: ~$0.0001 per invocation with the 0.5B model.
-- **Local dev**: Just `pip install stackfix[gguf]` and go.
+- **Local dev**: Just `pip install stackfix` and go.
 
 The whole pipeline is open source: data generation, training, evaluation benchmark, CLI tool, and model weights.
 
@@ -69,7 +69,7 @@ Results: 7B → 92.6% pass@1. 3B → 86.7%. 0.5B → 72.3%. Gemma-4-26B without
 
 The 3B model is 1.8 GB, loads in 0.6s, and runs in GitHub Actions for free. No API keys, no cloud costs.
 
-`pip install stackfix[gguf]` — open source, MIT licensed.
+`pip install stackfix` — open source, MIT licensed.
 
 github.com/jhammant/Cloud-Gym
 

{stackfix-0.1.0 → stackfix-0.1.2}/examples/USE_CASES.md

@@ -7,7 +7,7 @@ Runs on CPU anywhere — no GPU required, no API keys, no cloud costs.
 
 ```bash
 # Install
-pip install stackfix[gguf]
+pip install stackfix
 
 # Download the model (1.8 GB, one-time)
 python -c "
@@ -41,7 +41,7 @@ jobs:
         with: { python-version: '3.12' }
 
       - name: Install
-        run: pip install stackfix[gguf]
+        run: pip install stackfix
 
       - name: Download model
         run: |

{stackfix-0.1.0 → stackfix-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "stackfix"
-version = "0.1.0"
+version = "0.1.2"
 description = "AI-powered Infrastructure-as-Code repair — fix broken Terraform and CloudFormation on CPU"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -17,8 +17,8 @@ dependencies = [
     "python-hcl2>=5.0",
     "cfn-lint>=1.0",
     "rich>=13.0",
-    "datasets>=3.0",
-    "ollama>=0.4",
+    "llama-cpp-python>=0.3",
+    "huggingface-hub>=0.20",
 ]
 
 [project.optional-dependencies]
@@ -30,8 +30,12 @@ dev = [
 mlx = [
     "mlx-lm>=0.21",
 ]
-gguf = [
-    "llama-cpp-python>=0.3",
+training = [
+    "datasets>=3.0",
+    "ollama>=0.4",
+]
+ollama = [
+    "ollama>=0.4",
 ]
 
 [project.scripts]