aselimov/ReviewLlama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update cli to remove uneeded options

Browse source
This commit is contained in:
Alex Selimov 2025-06-18 14:26:21 -04:00
parent 9173399ce8
commit a6cdbf1761
5 changed files with 517 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/reviewllama/configs.py
View file

@ -8,9 +8,8 @@ class OllamaConfig:
"""Configuration for Ollama client."""
model: str
server_url: str
timeout: int
max_retries: int
base_url: str
system_prompt: str
@dataclass(frozen=True)
@ -23,15 +22,10 @@ class ReviewConfig:
def create_ollama_config(
model: str, server_url: str, timeout: int, max_retries: int
model: str, server_url: str, system_prompt: str
) -> OllamaConfig:
"""Create OllamaConfig with validated parameters."""
return OllamaConfig(
model=model,
server_url=server_url,
timeout=timeout,
max_retries=max_retries,
)
return OllamaConfig(model=model, base_url=server_url, system_prompt=system_prompt)
def create_review_config(
@ -45,15 +39,11 @@ def create_config_from_vars(
paths: List[Path],
model: str,
server_url: str,
timeout: int,
max_retries: int,
system_prompt: str,
base_branch: str,
):
ollama_config = OllamaConfig(
model=model,
server_url=server_url,
timeout=timeout,
max_retries=max_retries,
model=model, base_url=server_url, system_prompt=system_prompt
)
return create_review_config(paths, ollama_config, base_branch)