ReviewLlama/src/reviewllama/configs.py

from dataclasses import dataclass
from pathlib import Path
from typing import List


@dataclass(frozen=True)
class OllamaConfig:
    """Configuration for Ollama client."""

    model: str
    base_url: str
    system_prompt: str


@dataclass(frozen=True)
class ReviewConfig:
    """Complete configuration for ReviewLlama."""

    paths: List[Path]
    ollama: OllamaConfig
    base_branch: str


def create_ollama_config(
    model: str, server_url: str, system_prompt: str
) -> OllamaConfig:
    """Create OllamaConfig with validated parameters."""
    return OllamaConfig(model=model, base_url=server_url, system_prompt=system_prompt)


def create_review_config(
    paths: List[Path], ollama_config: OllamaConfig, base_branch: str
) -> ReviewConfig:
    """Create complete ReviewConfig from validated components."""
    return ReviewConfig(paths=paths, ollama=ollama_config, base_branch=base_branch)


def create_config_from_vars(
    paths: List[Path],
    model: str,
    server_url: str,
    system_prompt: str,
    base_branch: str,
):
    ollama_config = OllamaConfig(
        model=model, base_url=server_url, system_prompt=system_prompt
    )

    return create_review_config(paths, ollama_config, base_branch)