breadcrumb-cli 0.1.0__py3-none-any.whl

breadcrumb/__init__.py

@@ -0,0 +1,7 @@
+"""
+Bread Crumb — Chat with your codebase.
+
+Multi-provider AI support (Anthropic, OpenAI, Gemini, Ollama).
+"""
+
+__version__ = "0.1.0"

breadcrumb/ai/__init__.py

@@ -0,0 +1 @@
+"""AI package initialization."""

breadcrumb/ai/prompts.py

@@ -0,0 +1,60 @@
+"""
+System prompts and prompt templates for various commands.
+"""
+
+SYSTEM_PROMPT_DEFAULT = """You are Bread Crumb, an expert AI code reviewer and architect.
+You analyze code repositories with clarity and precision.
+Always be concise but thorough. Suggest improvements when relevant.
+Format code blocks with language specifiers for syntax highlighting."""
+
+SYSTEM_PROMPT_AUDIT = """You are a security and architecture auditor.
+Analyze this codebase for:
+- Security vulnerabilities (SQL injection, XSS, auth issues, etc.)
+- Performance bottlenecks
+- Architectural anti-patterns
+- Dependencies vulnerabilities
+- Code quality issues
+
+Be specific about WHAT is wrong and HOW to fix it.
+Prioritize critical issues first."""
+
+SYSTEM_PROMPT_COMMIT = """Generate a conventional commit message for the staged changes.
+Format: type(scope): description
+Types: feat, fix, docs, style, refactor, perf, test, chore, ci
+Keep under 72 characters.
+Return ONLY the commit message, no explanation."""
+
+SYSTEM_PROMPT_EXPLAIN_ERROR = """Analyze this error or stack trace and provide:
+1. Plain English explanation of what went wrong
+2. Root cause
+3. Specific steps to fix it
+Be concise and practical."""
+
+SYSTEM_PROMPT_DIFF = """Review this git diff and provide:
+1. Summary of changes
+2. Any potential issues or bugs
+3. Suggestions for improvement
+4. Impact assessment
+Be critical but constructive."""
+
+SYSTEM_PROMPT_DIGEST = """Summarize these git commits into a concise daily digest.
+What changed? What was fixed? Any notable improvements?
+Keep it under 200 words. Use bullet points."""
+
+
+def get_system_prompt(command: str, custom: str = "") -> str:
+    """Get system prompt for a command, with optional custom override."""
+    if custom:
+        return custom
+
+    prompts = {
+        "chat": SYSTEM_PROMPT_DEFAULT,
+        "ask": SYSTEM_PROMPT_DEFAULT,
+        "audit": SYSTEM_PROMPT_AUDIT,
+        "commit": SYSTEM_PROMPT_COMMIT,
+        "explain-error": SYSTEM_PROMPT_EXPLAIN_ERROR,
+        "diff": SYSTEM_PROMPT_DIFF,
+        "digest": SYSTEM_PROMPT_DIGEST,
+    }
+
+    return prompts.get(command, SYSTEM_PROMPT_DEFAULT)

breadcrumb/ai/router.py

@@ -0,0 +1,187 @@
+"""
+AI provider routing and message handling.
+Supports Anthropic, OpenAI, Gemini, and Ollama.
+"""
+
+from typing import Iterator, Optional
+
+from breadcrumb.config import Config
+
+
+class AIRouter:
+    """Routes requests to the appropriate AI provider."""
+
+    def __init__(self, provider: Optional[str] = None):
+        self.config = Config()
+        self.provider = provider or self.config.get("provider", "anthropic")
+        self.api_key = self.config.get_api_key(self.provider)
+        self.model = self.config.get_model(self.provider)
+
+    def chat(self, messages: list, system: str = "", **kwargs) -> str:
+        """Send a message and get a response."""
+        if self.provider == "anthropic":
+            return self._chat_anthropic(messages, system, **kwargs)
+        elif self.provider == "openai":
+            return self._chat_openai(messages, system, **kwargs)
+        elif self.provider == "gemini":
+            return self._chat_gemini(messages, system, **kwargs)
+        elif self.provider == "ollama":
+            return self._chat_ollama(messages, system, **kwargs)
+        else:
+            raise ValueError(f"Unknown provider: {self.provider}")
+
+    def stream(self, messages: list, system: str = "", **kwargs) -> Iterator[str]:
+        """Stream a response token by token."""
+        if self.provider == "anthropic":
+            yield from self._stream_anthropic(messages, system, **kwargs)
+        elif self.provider == "openai":
+            yield from self._stream_openai(messages, system, **kwargs)
+        elif self.provider == "gemini":
+            yield from self._stream_gemini(messages, system, **kwargs)
+        elif self.provider == "ollama":
+            yield from self._stream_ollama(messages, system, **kwargs)
+        else:
+            raise ValueError(f"Unknown provider: {self.provider}")
+
+    def count_tokens(self, text: str) -> int:
+        """Estimate token count (rough approximation)."""
+        # Rough estimate: 4 chars ≈ 1 token
+        return len(text) // 4
+
+    # ── Anthropic ──────────────────────────────────────────────────────────────
+    def _chat_anthropic(self, messages: list, system: str = "", **kwargs) -> str:
+        """Synchronous chat with Anthropic."""
+        try:
+            import anthropic
+        except ImportError:
+            raise ImportError("anthropic package required. Install: pip install anthropic")
+
+        client = anthropic.Anthropic(api_key=self.api_key)
+        response = client.messages.create(
+            model=self.model,
+            max_tokens=kwargs.get("max_tokens", 4096),
+            system=system,
+            messages=messages,
+            temperature=kwargs.get("temperature", 0.7),
+        )
+        text = getattr(response.content[0], "text", "")
+        return text or ""
+
+    def _stream_anthropic(self, messages: list, system: str = "", **kwargs) -> Iterator[str]:
+        """Stream response from Anthropic."""
+        try:
+            import anthropic
+        except ImportError:
+            raise ImportError("anthropic package required. Install: pip install anthropic")
+
+        client = anthropic.Anthropic(api_key=self.api_key)
+        with client.messages.stream(
+            model=self.model,
+            max_tokens=kwargs.get("max_tokens", 4096),
+            system=system,
+            messages=messages,
+            temperature=kwargs.get("temperature", 0.7),
+        ) as stream:
+            for text in stream.text_stream:
+                yield text
+
+    # ── OpenAI ────────────────────────────────────────────────────────────────
+    def _chat_openai(self, messages: list, system: str = "", **kwargs) -> str:
+        """Synchronous chat with OpenAI."""
+        try:
+            import openai
+        except ImportError:
+            raise ImportError("openai package required. Install: pip install openai")
+
+        client = openai.OpenAI(api_key=self.api_key)
+        if system:
+            messages = [{"role": "system", "content": system}] + messages
+        response = client.chat.completions.create(
+            model=self.model,
+            messages=messages,
+            max_tokens=kwargs.get("max_tokens", 4096),
+            temperature=kwargs.get("temperature", 0.7),
+        )
+        return response.choices[0].message.content or ""
+
+    def _stream_openai(self, messages: list, system: str = "", **kwargs) -> Iterator[str]:
+        """Stream response from OpenAI."""
+        try:
+            import openai
+        except ImportError:
+            raise ImportError("openai package required. Install: pip install openai")
+
+        client = openai.OpenAI(api_key=self.api_key)
+        if system:
+            messages = [{"role": "system", "content": system}] + messages
+        stream = client.chat.completions.create(
+            model=self.model,
+            messages=messages,
+            max_tokens=kwargs.get("max_tokens", 4096),
+            temperature=kwargs.get("temperature", 0.7),
+            stream=True,
+        )
+        for chunk in stream:
+            if chunk.choices[0].delta.content:
+                yield chunk.choices[0].delta.content
+
+    # ── Gemini ────────────────────────────────────────────────────────────────
+    def _chat_gemini(self, messages: list, system: str = "", **kwargs) -> str:
+        """Synchronous chat with Gemini."""
+        try:
+            import google.generativeai as genai
+        except ImportError:
+            raise ImportError(
+                "google-generativeai package required. Install: pip install google-generativeai"
+            )
+
+        genai.configure(api_key=self.api_key)
+        model = genai.GenerativeModel(self.model, system_instruction=system)
+        response = model.generate_content([m["content"] for m in messages])
+        return response.text
+
+    def _stream_gemini(self, messages: list, system: str = "", **kwargs) -> Iterator[str]:
+        """Stream response from Gemini."""
+        try:
+            import google.generativeai as genai
+        except ImportError:
+            raise ImportError(
+                "google-generativeai package required. Install: pip install google-generativeai"
+            )
+
+        genai.configure(api_key=self.api_key)
+        model = genai.GenerativeModel(self.model, system_instruction=system)
+        response = model.generate_content([m["content"] for m in messages], stream=True)
+        for chunk in response:
+            yield chunk.text
+
+    # ── Ollama ────────────────────────────────────────────────────────────────
+    def _chat_ollama(self, messages: list, system: str = "", **kwargs) -> str:
+        """Synchronous chat with Ollama."""
+        try:
+            import ollama
+        except ImportError:
+            raise ImportError("ollama package required. Install: pip install ollama")
+
+        url = self.config.get("ollama_url", "http://localhost:11434")
+        client = ollama.Client(host=url)
+        if system:
+            messages = [{"role": "system", "content": system}] + messages
+        response = client.chat(model=self.model, messages=messages)
+        return response["message"]["content"]
+
+    def _stream_ollama(self, messages: list, system: str = "", **kwargs) -> Iterator[str]:
+        """Stream response from Ollama."""
+        try:
+            import ollama
+        except ImportError:
+            raise ImportError("ollama package required. Install: pip install ollama")
+
+        url = self.config.get("ollama_url", "http://localhost:11434")
+        client = ollama.Client(host=url)
+        if system:
+            messages = [{"role": "system", "content": system}] + messages
+        stream = client.chat(model=self.model, messages=messages, stream=True)
+        for chunk in stream:
+            if "message" in chunk:
+                yield chunk["message"]["content"]

breadcrumb/cli.py

@@ -0,0 +1,144 @@
+"""
+Main CLI entry point using Click.
+Wires all commands together.
+"""
+
+import sys
+from pathlib import Path
+
+import click
+
+from breadcrumb import __version__
+from breadcrumb.commands.ask import cmd_ask
+from breadcrumb.commands.audit import cmd_audit
+from breadcrumb.commands.chat import cmd_chat
+from breadcrumb.commands.commit import cmd_commit
+from breadcrumb.commands.diff import cmd_diff
+from breadcrumb.commands.explain_error import cmd_explain_error
+from breadcrumb.commands.init import cmd_init
+from breadcrumb.config import Config
+
+
+@click.group(invoke_without_command=True)
+@click.version_option(version=__version__)
+@click.pass_context
+def cli(ctx):
+    """🍞 Bread Crumb — Chat with your codebase."""
+    # If no command, show help
+    if ctx.invoked_subcommand is None:
+        click.echo(ctx.get_help())
+
+
+@cli.command()
+@click.argument("repo_path", type=click.Path(exists=True), default=".")
+@click.option("--provider", help="AI provider (anthropic, openai, gemini, ollama)")
+@click.option("--session", default="default", help="Session name")
+def chat(repo_path, provider, session):
+    """Interactive chat with your codebase."""
+    cmd_chat(Path(repo_path), provider, session)
+
+
+@cli.command()
+@click.argument("repo_path", type=click.Path(exists=True), default=".")
+@click.argument("question", required=False)
+@click.option("--provider", help="AI provider")
+@click.option("--model", help="Model name")
+@click.option("--format", type=click.Choice(["text", "markdown"]), default="text")
+@click.option("--pipe", is_flag=True, help="Read question from stdin")
+def ask(repo_path, question, provider, model, format, pipe):
+    """Ask a one-shot question about the codebase."""
+    cmd_ask(Path(repo_path), question or "", provider, model, format, pipe)
+
+
+@cli.command()
+@click.argument("repo_path", type=click.Path(exists=True), default=".")
+@click.option("--provider", help="AI provider")
+@click.option("--model", help="Model name")
+@click.option("--format", type=click.Choice(["text", "markdown"]), default="text")
+def audit(repo_path, provider, model, format):
+    """Run a security and architecture audit."""
+    cmd_audit(Path(repo_path), provider, model, format)
+
+
+@cli.command()
+@click.argument("repo_path", type=click.Path(exists=True), default=".")
+@click.argument("revision", required=False, default="")
+@click.option("--provider", help="AI provider")
+def diff(repo_path, revision, provider):
+    """Review a git diff with AI."""
+    cmd_diff(Path(repo_path), revision, provider)
+
+
+@cli.command()
+@click.argument("repo_path", type=click.Path(exists=True), default=".")
+@click.option("--provider", help="AI provider")
+@click.option("--silent", is_flag=True, help="Only output the message")
+def commit(repo_path, provider, silent):
+    """Generate a conventional commit message."""
+    cmd_commit(Path(repo_path), provider, silent)
+
+
+@cli.command()
+@click.option("--provider", help="AI provider")
+@click.argument("error", required=False)
+def explain_error(provider, error):
+    """Explain an error or stack trace."""
+    cmd_explain_error(error, provider)
+
+
+@cli.command()
+@click.argument("repo_path", type=click.Path(exists=True), default=".")
+def init(repo_path):
+    """Initialize .breadcrumb.yaml in the repository."""
+    cmd_init(Path(repo_path))
+
+
+@cli.group()
+def config():
+    """Manage global configuration."""
+    pass
+
+
+@config.command()
+@click.argument("key")
+@click.argument("value")
+def set_key(key, value):
+    """Set a configuration value."""
+    cfg = Config()
+    cfg.set(key, value)
+    click.echo(f"✓ Set {key} = {value}")
+
+
+@config.command()
+@click.argument("key")
+def get_key(key):
+    """Get a configuration value."""
+    cfg = Config()
+    value = cfg.get(key)
+    click.echo(value)
+
+
+@config.command()
+def show():
+    """Show all configuration."""
+    cfg = Config()
+    for key, value in sorted(cfg.to_dict().items()):
+        if "key" in key.lower() and value:
+            value = "***" + value[-4:]  # Hide API keys
+        click.echo(f"{key}: {value}")
+
+
+def main():
+    """Entry point for the CLI."""
+    try:
+        cli()
+    except KeyboardInterrupt:
+        click.echo("\n")
+        sys.exit(0)
+    except Exception as e:
+        click.echo(f"Error: {e}", err=True)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

breadcrumb/commands/__init__.py

@@ -0,0 +1 @@
+"""Commands package initialization."""

breadcrumb/commands/ask.py

@@ -0,0 +1,98 @@
+"""
+One-shot non-interactive query command.
+Ask a question and get an answer in one go.
+"""
+
+import sys
+from pathlib import Path
+from typing import Optional
+
+from rich.console import Console
+from rich.markdown import Markdown
+
+from breadcrumb.ai.prompts import get_system_prompt
+from breadcrumb.ai.router import AIRouter
+from breadcrumb.ingest import FileIngester
+
+console = Console()
+
+
+def cmd_ask(
+    repo_path: Path,
+    question: str,
+    provider: Optional[str] = None,
+    model: Optional[str] = None,
+    format: str = "text",
+    pipe: bool = False,
+) -> str:
+    """
+    Ask a one-shot question about the repository.
+
+    Args:
+        repo_path: Repository path
+        question: The question to ask
+        provider: AI provider (overrides config)
+        model: Model name (overrides config)
+        format: Output format ('text' or 'markdown')
+        pipe: If True, read question from stdin if not provided
+
+    Returns:
+        The AI response
+    """
+    if not question and pipe:
+        question = sys.stdin.read()
+
+    if not question:
+        console.print("[red]Error: No question provided[/red]")
+        return ""
+
+    # Get repository context
+    try:
+        ingester = FileIngester(repo_path)
+        context = ingester.get_content()
+
+        if not context:
+            console.print("[yellow]Warning: No code files found in repository[/yellow]")
+    except Exception as e:
+        console.print(f"[red]Error reading repository: {e}[/red]")
+        context = ""
+
+    # Setup AI
+    router = AIRouter(provider)
+    if model:
+        router.model = model
+
+    # Build prompt
+    system = get_system_prompt("ask")
+    prompt = f"""Repo context:
+{context}
+
+Question: {question}"""
+
+    messages = [{"role": "user", "content": prompt}]
+
+    # Show spinner and stream response
+    response_text = ""
+    try:
+        with console.status("[bold cyan]Thinking...", spinner="dots"):
+            for chunk in router.stream(messages, system):
+                response_text += chunk
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+        return ""
+
+    # Output based on format
+    if format == "markdown":
+        console.print(Markdown(response_text))
+    else:
+        console.print(response_text)
+
+    # Show token usage
+    total_tokens = (
+        router.count_tokens(context)
+        + router.count_tokens(question)
+        + router.count_tokens(response_text)
+    )
+    console.print(f"\n[dim]~{total_tokens:,} tokens used[/dim]")
+
+    return response_text

breadcrumb/commands/audit.py

@@ -0,0 +1,77 @@
+"""
+Security and architecture audit command.
+"""
+
+from pathlib import Path
+from typing import Optional
+
+from rich.console import Console
+from rich.markdown import Markdown
+
+from breadcrumb.ai.prompts import get_system_prompt
+from breadcrumb.ai.router import AIRouter
+from breadcrumb.ingest import FileIngester
+
+console = Console()
+
+
+def cmd_audit(
+    repo_path: Path,
+    provider: Optional[str] = None,
+    model: Optional[str] = None,
+    format: str = "text",
+) -> str:
+    """
+    Run a security and architecture audit on the repository.
+
+    Args:
+        repo_path: Repository path
+        provider: AI provider (overrides config)
+        model: Model name (overrides config)
+        format: Output format ('text' or 'markdown')
+
+    Returns:
+        The audit report
+    """
+    # Get repository context
+    try:
+        ingester = FileIngester(repo_path)
+        context = ingester.get_content()
+
+        if not context:
+            console.print("[yellow]Warning: No code files found[/yellow]")
+    except Exception as e:
+        console.print(f"[red]Error reading repository: {e}[/red]")
+        context = ""
+
+    # Setup AI
+    router = AIRouter(provider)
+    if model:
+        router.model = model
+
+    system = get_system_prompt("audit")
+    prompt = f"""Analyze this codebase for security, architecture, and quality issues.
+
+{context}"""
+
+    messages = [{"role": "user", "content": prompt}]
+
+    try:
+        response_text = ""
+        with console.status("[bold cyan]Auditing codebase...", spinner="dots"):
+            for chunk in router.stream(messages, system):
+                response_text += chunk
+
+        if format == "markdown":
+            console.print(Markdown(response_text))
+        else:
+            console.print(response_text)
+
+        # Show token usage
+        total_tokens = router.count_tokens(context) + router.count_tokens(response_text)
+        console.print(f"\n[dim]~{total_tokens:,} tokens used[/dim]")
+
+        return response_text
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+        return ""