Initial commit with working cli prototype

This commit is contained in:
Alex Selimov 2025-06-07 22:10:38 -04:00
commit c9a0d9a225
4 changed files with 220 additions and 0 deletions

146
src/reviewllama/cli.py Normal file
View file

@ -0,0 +1,146 @@
import argparse
from dataclasses import dataclass
from typing import List, Optional
from pathlib import Path
import sys
@dataclass(frozen=True)
class OllamaConfig:
"""Configuration for Ollama client."""
model: str
server_url: str
timeout: int
max_retries: int
@dataclass(frozen=True)
class ReviewConfig:
"""Complete configuration for ReviewLlama."""
paths: List[Path]
ollama: OllamaConfig
def normalize_server_url(url: str) -> str:
"""Normalize Ollama server URL to ensure proper format."""
if not url.startswith(("http://", "https://")):
return f"http://{url}"
return url.rstrip("/")
def create_ollama_config(
model: str, server_url: str, timeout: int, max_retries: int
) -> OllamaConfig:
"""Create OllamaConfig with validated parameters."""
return OllamaConfig(
model=model,
server_url=normalize_server_url(server_url),
timeout=timeout,
max_retries=max_retries,
)
def create_review_config(
paths: List[Path], ollama_config: OllamaConfig
) -> ReviewConfig:
"""Create complete ReviewConfig from validated components."""
return ReviewConfig(paths=paths, ollama=ollama_config)
def create_argument_parser() -> argparse.ArgumentParser:
"""Create and configure the argument parser."""
parser = argparse.ArgumentParser(
prog="reviewllama",
description="AI-powered code review assistant",
formatter_class=argparse.RawDescriptionHelpFormatter,
epilog="""
Examples:
reviewllama . --model gemma3:27b --server localhost:11434
reviewllama src/ tests/ --model llama3.2:7b --timeout 60
""",
)
parser.add_argument(
"paths",
nargs="+",
metavar="PATH",
help="One or more file paths or git directories to review",
)
parser.add_argument(
"--model",
default="llama3.2:3b",
help="Ollama model to use for code review (default: %(default)s)",
)
parser.add_argument(
"--server",
dest="server_url",
default="localhost:11434",
help="Ollama server URL (default: %(default)s)",
)
parser.add_argument(
"--timeout",
type=int,
default=30,
help="Request timeout in seconds (default: %(default)s)",
)
parser.add_argument(
"--max-retries",
dest="max_retries",
type=int,
default=3,
help="Maximum number of retry attempts (default: %(default)s)",
)
return parser
def parse_raw_arguments(args: Optional[List[str]] = None) -> argparse.Namespace:
"""Parse command line arguments into raw namespace."""
parser = create_argument_parser()
return parser.parse_args(args)
def transform_namespace_to_config(namespace: argparse.Namespace) -> ReviewConfig:
"""Transform argparse namespace into ReviewConfig."""
paths = [Path(path_str) for path_str in namespace.paths]
ollama_config = create_ollama_config(
model=namespace.model,
server_url=namespace.server_url,
timeout=namespace.timeout,
max_retries=namespace.max_retries,
)
return create_review_config(paths=paths, ollama_config=ollama_config)
def parse_arguments(args: Optional[List[str]] = None) -> ReviewConfig:
"""Parse command line arguments and return validated configuration."""
raw_namespace = parse_raw_arguments(args)
return transform_namespace_to_config(raw_namespace)
def cli() -> None:
"""Main entry point for the CLI."""
try:
config = parse_arguments()
# TODO: Pass config to review engine
print(f"Reviewing {len(config.paths)} path(s) with model {config.ollama.model}")
for path in config.paths:
print(f" - {path}")
except SystemExit:
# argparse calls sys.exit on error, let it propagate
raise
except Exception as e:
print(f"Error: {e}", file=sys.stderr)
sys.exit(1)
if __name__ == "__main__":
main()