dockerbrain 1.1.0__tar.gz → 1.2.0__tar.gz

{dockerbrain-1.1.0/dockerbrain.egg-info → dockerbrain-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dockerbrain
-Version: 1.1.0
+Version: 1.2.0
 Summary: Dockerfile and container monitoring, analysis, and optimization CLI.
 License: Apache-2.0
 Project-URL: Homepage, https://github.com/iamPulakesh/DockerBrain
@@ -24,6 +24,7 @@ Classifier: Topic :: System :: Systems Administration
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: anthropic>=0.30
 Requires-Dist: click>=8.0
 Requires-Dist: docker>=6.0
 Requires-Dist: google-genai>=1.0
@@ -37,7 +38,7 @@ Dynamic: license-file
 
 # Project Info
 
-[![pipeline](https://github.com/iamPulakesh/DockerBrain/actions/workflows/dockerbrain.yml/badge.svg)](https://github.com/iamPulakesh/DockerBrain/actions/workflows/dockerbrain.yml) [![github project](https://img.shields.io/badge/github-project-blue.svg?logo=github)](https://github.com/iamPulakesh/DockerBrain)
+[![pipeline](https://github.com/iamPulakesh/DockerBrain/actions/workflows/dockerbrain.yml/badge.svg)](https://github.com/iamPulakesh/DockerBrain/actions/workflows/dockerbrain.yml) [![github project](https://img.shields.io/badge/github-project-blue.svg?logo=github)](https://github.com/iamPulakesh/DockerBrain) [![PyPI Downloads](https://static.pepy.tech/personalized-badge/dockerbrain?period=total&units=INTERNATIONAL_SYSTEM&left_color=GRAY&right_color=RED&left_text=Downloads)](https://pepy.tech/projects/dockerbrain)
 
 ## Core components of DockerBrain
 

{dockerbrain-1.1.0 → dockerbrain-1.2.0}/PYPI_README.md

@@ -1,6 +1,6 @@
 # Project Info
 
-[![pipeline](https://github.com/iamPulakesh/DockerBrain/actions/workflows/dockerbrain.yml/badge.svg)](https://github.com/iamPulakesh/DockerBrain/actions/workflows/dockerbrain.yml) [![github project](https://img.shields.io/badge/github-project-blue.svg?logo=github)](https://github.com/iamPulakesh/DockerBrain)
+[![pipeline](https://github.com/iamPulakesh/DockerBrain/actions/workflows/dockerbrain.yml/badge.svg)](https://github.com/iamPulakesh/DockerBrain/actions/workflows/dockerbrain.yml) [![github project](https://img.shields.io/badge/github-project-blue.svg?logo=github)](https://github.com/iamPulakesh/DockerBrain) [![PyPI Downloads](https://static.pepy.tech/personalized-badge/dockerbrain?period=total&units=INTERNATIONAL_SYSTEM&left_color=GRAY&right_color=RED&left_text=Downloads)](https://pepy.tech/projects/dockerbrain)
 
 ## Core components of DockerBrain
 

{dockerbrain-1.1.0 → dockerbrain-1.2.0}/README.md

@@ -1,15 +1,22 @@
+<div align="center">
+
 # DockerBrain
 
 **AI-powered Docker container monitoring, optimization, and Dockerfile generation CLI.**
 
+[![PyPI Downloads](https://static.pepy.tech/personalized-badge/dockerbrain?period=total&units=INTERNATIONAL_SYSTEM&left_color=GRAY&right_color=RED&left_text=PyPI+downloads)](https://pepy.tech/projects/dockerbrain)
 [![Python 3.10+](https://img.shields.io/badge/python-3.10%2B-blue.svg)](https://www.python.org/downloads/)
 [![License: Apache 2.0](https://img.shields.io/badge/License-Apache_2.0-green.svg)](LICENSE)
 [![PyPI](https://img.shields.io/pypi/v/dockerbrain)](https://pypi.org/project/dockerbrain/)
 
-DockerBrain monitors your Docker containers in real time, detects resource issues, and uses LLMs to generate actionable optimizations for running containers and Dockerfiles. Supports **Gemini**, **Groq**, and **Ollama**. All configuration lives in a single `.dockerbrainrc` file.
+<p align="center">
+  DockerBrain monitors your Docker containers in real time, detects resource issues, and uses LLMs to generate actionable optimizations for running containers and Dockerfiles.
+</p>
 
 ---
 
+</div>
+
 ## Features
 
 | Feature | Command | LLM Required |
@@ -68,14 +75,16 @@ This creates `.dockerbrainrc` in `~/.dockerbrain/`. Open it with `dockerb config
 
 ```ini
 [llm]
-provider = "groq"
-model    = "openai/gpt-oss-120b"
+provider = "choose_provider"
+model    = "choose_model"
 api_key  = "your_key_here"
 ```
 
 | Provider | API Key | Link |
 |---|:---:|---|
 | Gemini | Required | [aistudio.google.com](https://aistudio.google.com/app/apikey) |
+| ChatGPT | Required | [platform.openai.com](https://platform.openai.com/api-keys) |
+| Claude | Required | [console.anthropic.com](https://console.anthropic.com/settings/keys) |
 | Groq | Required | [console.groq.com/keys](https://console.groq.com/keys) |
 | Ollama | Not Required | [ollama.com](https://ollama.com) |
 

dockerbrain-1.2.0/core/__init__.py

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

{dockerbrain-1.1.0 → dockerbrain-1.2.0}/core/cli.py

@@ -2,6 +2,7 @@ import platform
 import click
 from core import __version__
 
+
 def print_version(ctx: click.Context, _param: click.Parameter, value: bool) -> None:
     """Print version with build metadata and exit."""
     if not value or ctx.resilient_parsing:
@@ -26,6 +27,7 @@ class _HiddenHelpCommand(click.Command):
 
 class _HiddenHelpGroup(click.Group):
     """A Click Group whose subcommands all use HiddenHelpCommand."""
+
     command_class = _HiddenHelpCommand
 
 
@@ -36,7 +38,9 @@ class _HiddenHelpGroup(click.Group):
     is_eager=True,
     expose_value=False,
     hidden=True,
-    callback=lambda ctx, param, value: (click.echo(ctx.get_help(), color=ctx.color) or ctx.exit()) if value else None,
+    callback=lambda ctx, param, value: (
+        (click.echo(ctx.get_help(), color=ctx.color) or ctx.exit()) if value else None
+    ),
 )
 @click.option(
     "--version",
@@ -49,21 +53,60 @@ class _HiddenHelpGroup(click.Group):
 def cli() -> None:
     pass
 
+
 # Commands
 @cli.command()
-@click.option("--interval", "-i", default=1, show_default=True, help="Polling interval in seconds.")
-@click.option("--duration", "-d", default=None, type=int, help="Total seconds to monitor (default: unlimited).")
+@click.option(
+    "--interval",
+    "-i",
+    default=1,
+    show_default=True,
+    help="Polling interval in seconds.",
+)
+@click.option(
+    "--duration",
+    "-d",
+    default=None,
+    type=int,
+    help="Total seconds to monitor (default: unlimited).",
+)
 def monitor(interval: int, duration: int | None) -> None:
     """Display containers' stats."""
     from core.monitor import run_monitor
+
     run_monitor(interval=interval, duration=duration)
 
+
 @cli.command("suggest")
-@click.option("--container", "-c", default=None, help="Target a specific container (default: all).")
-@click.option("--window", "-w", default=30, show_default=True, help="Minutes of metric history to include.")
-@click.option("--dockerfile", "-f", default=None, type=click.Path(exists=True), help="Analyze a Dockerfile instead of containers.")
-@click.option("--no-rules", is_flag=True, default=False, help="Skip rule-based suggestions, send raw metrics only.")
-def ai_suggest(container: str | None, window: int, dockerfile: str | None, no_rules: bool) -> None:
+@click.option(
+    "--container",
+    "-c",
+    default=None,
+    help="Target a specific container (default: all).",
+)
+@click.option(
+    "--window",
+    "-w",
+    default=30,
+    show_default=True,
+    help="Minutes of metric history to include.",
+)
+@click.option(
+    "--dockerfile",
+    "-f",
+    default=None,
+    type=click.Path(exists=True),
+    help="Analyze a Dockerfile instead of containers.",
+)
+@click.option(
+    "--no-rules",
+    is_flag=True,
+    default=False,
+    help="Skip rule-based suggestions, send raw metrics only.",
+)
+def ai_suggest(
+    container: str | None, window: int, dockerfile: str | None, no_rules: bool
+) -> None:
     """Get optimization suggestions for Dockerfiles and containers.
 
     Two modes:
@@ -86,11 +129,21 @@ def ai_suggest(container: str | None, window: int, dockerfile: str | None, no_ru
         no_rules=no_rules,
     )
 
+
 @cli.command()
-@click.option("--dockerfile", "-f", default=None, type=click.Path(exists=True),
-              help="Path to a Dockerfile to auto-fix.")
-@click.option("--container", "-c", default=None,
-              help="Target a specific container (default: all).")
+@click.option(
+    "--dockerfile",
+    "-f",
+    default=None,
+    type=click.Path(exists=True),
+    help="Path to a Dockerfile to auto-fix.",
+)
+@click.option(
+    "--container",
+    "-c",
+    default=None,
+    help="Target a specific container (default: all).",
+)
 def fix(dockerfile: str | None, container: str | None) -> None:
     """Automatically diagnose and fix Docker issues.
 
@@ -109,9 +162,14 @@ def fix(dockerfile: str | None, container: str | None) -> None:
 
     run_fix(dockerfile_path=dockerfile, container_name=container)
 
+
 @cli.command()
-@click.option("--force", is_flag=True, default=False,
-              help="Overwrite existing Dockerfile and .dockerignore forcefully.")
+@click.option(
+    "--force",
+    is_flag=True,
+    default=False,
+    help="Overwrite existing Dockerfile and .dockerignore forcefully.",
+)
 def dockerize(force: bool) -> None:
     """Generate a Dockerfile and .dockerignore for your project.
 
@@ -126,18 +184,22 @@ def dockerize(force: bool) -> None:
 
     run_dockerize(project_path=".", force=force)
 
+
 class _NoHelpOptGroup(click.Group):
     """Group that hides the --help option from the help output."""
+
     def get_help_option(self, ctx: click.Context) -> click.Option | None:
         opt = super().get_help_option(ctx)
         if opt:
             opt.hidden = True
         return opt
 
+
 @cli.group(cls=_NoHelpOptGroup)
 def template() -> None:
     """Browse and use curated Dockerfile templates."""
 
+
 @template.command("list")
 def template_list() -> None:
     """Show all available Dockerfile templates."""
@@ -163,9 +225,8 @@ def template_list() -> None:
 
     console.print()
     console.print(table)
-    console.print(
-        "\n[dim]Use a template:[/] [cyan]dockerb template use <name>[/]\n"
-    )
+    console.print("\n[dim]Use a template:[/] [cyan]dockerb template use <name>[/]\n")
+
 
 @template.command("use")
 @click.argument("name")
@@ -191,15 +252,14 @@ def template_use(name: str, force: bool) -> None:
 
     if tpl is None:
         console.print(f"[red]Unknown template:[/] [bold]{name}[/]")
-        console.print(
-            f"[dim]Available: {', '.join(get_template_names())}[/]"
-        )
+        console.print(f"[dim]Available: {', '.join(get_template_names())}[/]")
         return
 
     dockerfile_path = Path("Dockerfile")
 
     if dockerfile_path.exists() and not force:
         from rich.prompt import Confirm
+
         if not Confirm.ask(
             "[yellow]Dockerfile already exists.[/] Overwrite?",
             default=False,
@@ -208,9 +268,7 @@ def template_use(name: str, force: bool) -> None:
             return
 
     dockerfile_path.write_text(tpl["dockerfile"], encoding="utf-8")
-    console.print(
-        f"[green]Created Dockerfile[/] [dim]({tpl['name']})[/]"
-    )
+    console.print(f"[green]Created Dockerfile[/] [dim]({tpl['name']})[/]")
 
     dockerignore_path = Path(".dockerignore")
     if not dockerignore_path.exists():
@@ -232,7 +290,9 @@ README.md
         console.print("[green]Created .dockerignore[/]")
 
     console.print()
-    console.print(Syntax(tpl["dockerfile"], "dockerfile", theme="monokai", line_numbers=True))
+    console.print(
+        Syntax(tpl["dockerfile"], "dockerfile", theme="monokai", line_numbers=True)
+    )
 
 
 @cli.command()
@@ -252,18 +312,31 @@ def init() -> None:
     config_path = config_dir / ".dockerbrainrc"
 
     if config_path.exists():
-        console.print(f"[yellow] {config_path} already exists.[/]")
-        return
+        from rich.prompt import Confirm
+        if not Confirm.ask(
+            f"[yellow]{config_path} already exists.[/] Overwrite?",
+            default=False,
+        ):
+            console.print("[dim]Skipped. No changes made.[/]")
+            return
 
-    config_content = '''\
+    config_content = """\
 # DockerBrain Configuration
 
 # LLM Provider & Models, To switch provider or model, edit below.
 
-# GEMINI
+# GEMINI  
 #   provider = "gemini"
 #   model = "gemini-3.1-flash-lite-preview", "gemini-flash-latest"
 
+# CHATGPT 
+#   provider = "chatgpt"
+#   model = "gpt-5.4-mini", "gpt-5.4-nano"
+
+# CLAUDE 
+#   provider = "claude"
+#   model = "claude-sonnet-4-6", "claude-haiku-4-6"
+
 # GROQ (Fast)
 #   provider = "groq"
 #   model = "openai/gpt-oss-120b", "llama-3.3-70b-versatile"   
@@ -292,11 +365,12 @@ check_base_image      = true  # Flag large base images
 check_apt_recommends  = true  # Flag apt-get without --no-install-recommends
 check_cache_busting   = true  # Flag COPY . . before dependency install
 check_dockerignore    = true  # Warn if .dockerignore is missing
-'''
+"""
 
     config_path.write_text(config_content, encoding="utf-8")
     console.print(f"[green]Created [bold]{config_path}[/bold][/]")
 
+
 @cli.command()
 def config() -> None:
     """Open the .dockerbrainrc config file in your default editor."""
@@ -310,9 +384,7 @@ def config() -> None:
     config_path = Path.home() / ".dockerbrain" / ".dockerbrainrc"
 
     if not config_path.exists():
-        console.print(
-            "[yellow]Config not found.[/] Run [cyan]dockerb init[/] first."
-        )
+        console.print("[yellow]Config not found.[/] Run [cyan]dockerb init[/] first.")
         return
 
     if sys.platform == "win32":
@@ -323,6 +395,7 @@ def config() -> None:
         editor = __import__("os").environ.get("EDITOR", "nano")
         subprocess.run([editor, str(config_path)])
 
+
 @cli.command()
 def env() -> None:
     """Check your environment and configs."""
@@ -352,7 +425,7 @@ def env() -> None:
         client = docker.from_env()
         info = client.version()
         _ok("Docker daemon", f"running (v{info.get('Version', '?')})")
-    except Exception :
+    except Exception:
         _fail("Docker daemon", f"not reachable")
 
     try:
@@ -362,10 +435,15 @@ def env() -> None:
 
     try:
         from core.llm import load_llm_config
+
         cfg = load_llm_config()
         _ok("LLM Provider", cfg.provider)
         _ok("LLM Model", cfg.model)
-        masked = cfg.api_key[:4] + "…" + cfg.api_key[-4:] if len(cfg.api_key) > 8 else "set ✓"
+        masked = (
+            cfg.api_key[:4] + "…" + cfg.api_key[-4:]
+            if len(cfg.api_key) > 8
+            else "set ✓"
+        )
         _ok("API Key", f"{masked}")
     except SystemExit:
         _fail("API Key", "not set in ~/.dockerbrain/.dockerbrainrc")
@@ -375,7 +453,9 @@ def env() -> None:
         size_kb = db_path.stat().st_size / 1024
         try:
             conn = sqlite3.connect(str(db_path))
-            row_count = conn.execute("SELECT COUNT(*) FROM container_metrics").fetchone()[0]
+            row_count = conn.execute(
+                "SELECT COUNT(*) FROM container_metrics"
+            ).fetchone()[0]
             conn.close()
             _ok("SQLite DB", f"{db_path} ({size_kb:.0f} KB, {row_count:,} rows)")
         except Exception:
@@ -383,7 +463,11 @@ def env() -> None:
     else:
         _fail("SQLite DB", f"not found at {db_path} — run: dockerb monitor")
 
-    status = "[bold green]All checks passed!" if all_ok else "[bold yellow]Some checks failed"
+    status = (
+        "[bold green]All checks passed!"
+        if all_ok
+        else "[bold yellow]Some checks failed"
+    )
     console.print(
         Panel(
             lines,

{dockerbrain-1.1.0 → dockerbrain-1.2.0}/core/llm.py

@@ -10,21 +10,30 @@ console = Console()
 
 _PROVIDER_DEFAULTS: dict[str, dict[str, str]] = {
     "gemini": {
-        "model":    "gemini-3.1-flash-lite-preview",
-        "base_url": "",  # uses google-genai SDK, not OpenAI
+        "model": "gemini-3.1-flash-lite-preview",
+        "base_url": "",  # uses google-genai SDK
+    },
+    "chatgpt": {
+        "model": "gpt-5.4-mini",
+        "base_url": "https://api.openai.com/v1",
+    },
+    "claude": {
+        "model": "claude-sonnet-4-6",
+        "base_url": "",  # uses anthropic SDK
     },
     "groq": {
-        "model":    "llama-3.3-70b-versatile",
+        "model": "llama-3.3-70b-versatile",
         "base_url": "https://api.groq.com/openai/v1",
     },
     "ollama": {
-        "model":    "llama3.1",
+        "model": "llama3.1",
         "base_url": "http://localhost:11434/v1",
     },
 }
 
 _VALID_PROVIDERS = set(_PROVIDER_DEFAULTS.keys())
 
+
 @dataclass
 class LLMConfig:
     """Resolved LLM configuration."""
@@ -104,11 +113,12 @@ def _show_missing_key_error(provider: str) -> None:
         )
     )
 
+
 def _strip_code_fences(text: str) -> str:
     """Strip markdown code fences that LLMs sometimes wrap responses in."""
     text = text.strip()
     if text.startswith("```"):
-        text = text[text.index("\n") + 1:] if "\n" in text else text
+        text = text[text.index("\n") + 1 :] if "\n" in text else text
     if text.endswith("```"):
         text = text[:-3].rstrip()
     return text
@@ -129,6 +139,8 @@ def generate(
 
     if config.provider == "gemini":
         return _generate_gemini(prompt, system_instruction, config)
+    elif config.provider == "claude":
+        return _generate_anthropic(prompt, system_instruction, config)
     else:
         return _generate_openai_compat(prompt, system_instruction, config)
 
@@ -144,6 +156,8 @@ def generate_stream(
 
     if config.provider == "gemini":
         yield from _stream_gemini(prompt, system_instruction, config)
+    elif config.provider == "claude":
+        yield from _stream_anthropic(prompt, system_instruction, config)
     else:
         yield from _stream_openai_compat(prompt, system_instruction, config)
 
@@ -163,6 +177,7 @@ def _generate_gemini(prompt: str, system_instruction: str, config: LLMConfig) ->
     )
     return _strip_code_fences(response.text)
 
+
 def _stream_gemini(prompt: str, system_instruction: str, config: LLMConfig):
     from google import genai
     from google.genai import types
@@ -180,8 +195,39 @@ def _stream_gemini(prompt: str, system_instruction: str, config: LLMConfig):
             yield chunk.text
 
 
+# Anthropic backend
+def _generate_anthropic(prompt: str, system_instruction: str, config: LLMConfig) -> str:
+    import anthropic
+
+    client = anthropic.Anthropic(api_key=config.api_key)
+    response = client.messages.create(
+        model=config.model,
+        max_tokens=4096,
+        system=system_instruction,
+        messages=[{"role": "user", "content": prompt}],
+    )
+    text = response.content[0].text if response.content else ""
+    return _strip_code_fences(text)
+
+
+def _stream_anthropic(prompt: str, system_instruction: str, config: LLMConfig):
+    import anthropic
+
+    client = anthropic.Anthropic(api_key=config.api_key)
+    with client.messages.stream(
+        model=config.model,
+        max_tokens=4096,
+        system=system_instruction,
+        messages=[{"role": "user", "content": prompt}],
+    ) as stream:
+        for text in stream.text_stream:
+            yield text
+
+
 # OpenAI-compatible backend
-def _generate_openai_compat(prompt: str, system_instruction: str, config: LLMConfig) -> str:
+def _generate_openai_compat(
+    prompt: str, system_instruction: str, config: LLMConfig
+) -> str:
     from openai import OpenAI
 
     client = OpenAI(api_key=config.api_key, base_url=config.base_url)
@@ -194,6 +240,7 @@ def _generate_openai_compat(prompt: str, system_instruction: str, config: LLMCon
     )
     return _strip_code_fences(response.choices[0].message.content or "")
 
+
 def _stream_openai_compat(prompt: str, system_instruction: str, config: LLMConfig):
     from openai import OpenAI