xai-review 0.3.0__py3-none-any.whl

ai_review/services/diff/renderers.py

@@ -0,0 +1,149 @@
+"""
+Renderers for diff views.
+
+Supported build modes:
+- FULL_FILE_CURRENT          snapshot after changes (+ markers for added)
+- FULL_FILE_PREVIOUS         snapshot before changes (+ markers for removed)
+- FULL_FILE_DIFF             full unified diff (+, -, unchanged)
+- ONLY_ADDED                 only added lines
+- ONLY_REMOVED               only removed lines
+- ADDED_AND_REMOVED          added + removed
+- ONLY_ADDED_WITH_CONTEXT    added + surrounding unchanged lines
+- ONLY_REMOVED_WITH_CONTEXT  removed + surrounding unchanged lines
+- ADDED_AND_REMOVED_WITH_CONTEXT added + removed + surrounding unchanged lines
+"""
+from enum import Enum
+from typing import Iterable, Optional
+
+from ai_review.libs.diff.models import DiffFile, DiffLineType
+from ai_review.services.diff.tools import normalize_file_path, marker_for_line, read_snapshot
+
+
+class MarkerType(Enum):
+    ADDED = "added"
+    REMOVED = "removed"
+
+
+def build_full_file_current(file: Optional[DiffFile], file_path: str, head_sha: str | None) -> str:
+    text = read_snapshot(file_path, head_sha=head_sha)
+    if text is None:
+        return f"# Failed to read current snapshot for {file_path}"
+
+    added_new = file.added_line_numbers() if file else set()
+    return render_plain_numbered(text.splitlines(), added_new, marker_type=MarkerType.ADDED)
+
+
+def build_full_file_previous(file: Optional[DiffFile], file_path: str, base_sha: str | None) -> str:
+    text = read_snapshot(file_path, base_sha=base_sha)
+    if text is None:
+        return f"# Failed to read previous snapshot for {file_path} (base_sha missing or file absent)"
+
+    removed_old = file.removed_line_numbers() if file else set()
+    return render_plain_numbered(text.splitlines(), removed_old, marker_type=MarkerType.REMOVED)
+
+
+def build_full_file_diff(file: DiffFile) -> str:
+    return render_unified(file, include_added=True, include_removed=True, include_unchanged=True, context=0)
+
+
+def build_only_added(file: DiffFile) -> str:
+    return render_unified(file, include_added=True, include_removed=False, include_unchanged=False, context=0)
+
+
+def build_only_removed(file: DiffFile) -> str:
+    return render_unified(file, include_added=False, include_removed=True, include_unchanged=False, context=0)
+
+
+def build_added_and_removed(file: DiffFile) -> str:
+    return render_unified(file, include_added=True, include_removed=True, include_unchanged=False, context=0)
+
+
+def build_only_added_with_context(file: DiffFile, context: int) -> str:
+    return render_unified(file, include_added=True, include_removed=False, include_unchanged=True, context=context)
+
+
+def build_only_removed_with_context(file: DiffFile, context: int) -> str:
+    return render_unified(file, include_added=False, include_removed=True, include_unchanged=True, context=context)
+
+
+def build_added_and_removed_with_context(file: DiffFile, context: int) -> str:
+    return render_unified(file, include_added=True, include_removed=True, include_unchanged=True, context=context)
+
+
+def render_plain_numbered(lines: Iterable[str], changed: set[int], marker_type: MarkerType) -> str:
+    def choose_marker(line_no: int) -> str:
+        if line_no not in changed:
+            return ""
+        if marker_type is MarkerType.ADDED:
+            return marker_for_line(added=True)
+        if marker_type is MarkerType.REMOVED:
+            return marker_for_line(removed=True)
+        return ""
+
+    return "\n".join(
+        f"{line_no}: {content}{choose_marker(line_no)}"
+        for line_no, content in enumerate(lines, start=1)
+    )
+
+
+def render_unified(
+        file: DiffFile,
+        *,
+        include_added: bool,
+        include_removed: bool,
+        include_unchanged: bool,
+        context: int,
+) -> str:
+    """
+    Render unified diff view.
+
+    Each line is prefixed with:
+      '+' for added lines,
+      '-' for removed lines,
+      ' ' for unchanged lines.
+
+    Context controls how many unchanged lines around modifications are shown.
+    """
+    lines_out: list[str] = []
+
+    added_new_positions = file.added_line_numbers()
+    removed_old_positions = file.removed_line_numbers()
+
+    def in_context(old_no: Optional[int], new_no: Optional[int]) -> bool:
+        """Check if an unchanged line falls within context radius."""
+        if context <= 0:
+            return False
+        if include_added and new_no is not None:
+            if any(abs(new_no - a) <= context for a in added_new_positions):
+                return True
+        if include_removed and old_no is not None:
+            if any(abs(old_no - r) <= context for r in removed_old_positions):
+                return True
+        return False
+
+    for hunk in file.hunks:
+        old_no = hunk.orig_range.start
+        new_no = hunk.new_range.start
+
+        for line in hunk.lines:
+            if line.type is DiffLineType.ADDED:
+                if include_added:
+                    lines_out.append(f"+{new_no}: {line.content}{marker_for_line(DiffLineType.ADDED)}")
+                new_no += 1
+
+            elif line.type is DiffLineType.REMOVED:
+                if include_removed:
+                    lines_out.append(f"-{old_no}: {line.content}{marker_for_line(DiffLineType.REMOVED)}")
+                old_no += 1
+
+            else:
+                if include_unchanged and (context == 0 or in_context(old_no, new_no)):
+                    lines_out.append(f" {new_no}: {line.content}")
+                old_no += 1
+                new_no += 1
+
+    if not lines_out:
+        header = normalize_file_path(file.new_name or file.orig_name)
+        return f"# No matching lines for mode in {header}"
+
+    return "\n".join(lines_out)

ai_review/services/diff/schema.py

@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+
+
+class DiffFileSchema(BaseModel):
+    file: str
+    diff: str

ai_review/services/diff/service.py

@@ -0,0 +1,96 @@
+from ai_review.config import settings
+from ai_review.libs.config.review import ReviewMode
+from ai_review.libs.diff.models import Diff
+from ai_review.libs.diff.parser import DiffParser
+from ai_review.libs.logger import get_logger
+from ai_review.services.diff.renderers import (
+    build_full_file_diff,
+    build_full_file_current,
+    build_full_file_previous,
+    build_only_added,
+    build_only_removed,
+    build_added_and_removed,
+    build_only_added_with_context,
+    build_only_removed_with_context,
+    build_added_and_removed_with_context
+)
+from ai_review.services.diff.schema import DiffFileSchema
+from ai_review.services.diff.tools import find_diff_file
+from ai_review.services.git.types import GitServiceProtocol
+
+logger = get_logger("DIFF_SERVICE")
+
+
+class DiffService:
+    @classmethod
+    def parse(cls, raw_diff: str) -> Diff:
+        if not raw_diff.strip():
+            logger.debug("Received empty diff string")
+            return Diff(files=[], raw=raw_diff)
+
+        try:
+            return DiffParser.parse(raw_diff)
+        except Exception as error:
+            logger.exception(f"Failed to parse diff: {error}")
+            raise
+
+    @classmethod
+    def render_file(
+            cls,
+            file: str,
+            raw_diff: str,
+            base_sha: str | None = None,
+            head_sha: str | None = None,
+    ) -> DiffFileSchema:
+        diff = cls.parse(raw_diff)
+        target = find_diff_file(diff, file)
+
+        match settings.review.mode:
+            case ReviewMode.FULL_FILE_CURRENT:
+                file_diff = build_full_file_current(target, file, head_sha)
+            case ReviewMode.FULL_FILE_PREVIOUS:
+                file_diff = build_full_file_previous(target, file, base_sha)
+            case ReviewMode.FULL_FILE_DIFF:
+                file_diff = build_full_file_diff(target)
+            case ReviewMode.ONLY_ADDED:
+                file_diff = build_only_added(target)
+            case ReviewMode.ONLY_REMOVED:
+                file_diff = build_only_removed(target)
+            case ReviewMode.ADDED_AND_REMOVED:
+                file_diff = build_added_and_removed(target)
+            case ReviewMode.ONLY_ADDED_WITH_CONTEXT:
+                file_diff = build_only_added_with_context(target, settings.review.context_lines)
+            case ReviewMode.ONLY_REMOVED_WITH_CONTEXT:
+                file_diff = build_only_removed_with_context(target, settings.review.context_lines)
+            case ReviewMode.ADDED_AND_REMOVED_WITH_CONTEXT:
+                file_diff = build_added_and_removed_with_context(target, settings.review.context_lines)
+            case _:
+                file_diff = f"# Unsupported mode: {settings.review.mode}"
+
+        return DiffFileSchema(diff=file_diff, file=file)
+
+    @classmethod
+    def render_files(
+            cls,
+            git: GitServiceProtocol,
+            files: list[str],
+            base_sha: str,
+            head_sha: str,
+    ) -> list[DiffFileSchema]:
+        annotated: list[DiffFileSchema] = []
+        for file in files:
+            raw_diff = git.get_diff_for_file(base_sha, head_sha, file)
+            if not raw_diff.strip():
+                logger.debug(f"No diff for {file}, skipping")
+                continue
+
+            annotated.append(
+                cls.render_file(
+                    file=file,
+                    base_sha=base_sha,
+                    head_sha=head_sha,
+                    raw_diff=raw_diff,
+                )
+            )
+
+        return annotated

ai_review/services/diff/tools.py

@@ -0,0 +1,59 @@
+from pathlib import Path
+
+from ai_review.config import settings
+from ai_review.libs.diff.models import Diff, DiffFile, DiffLineType
+from ai_review.libs.logger import get_logger
+from ai_review.services.git.service import GitService
+
+logger = get_logger("DIFF_TOOLS")
+
+
+def normalize_file_path(file_path: str) -> str:
+    """Normalize a git diff file path (remove a/ b/ prefixes, convert slashes)."""
+    if not file_path:
+        return ""
+
+    file_path = file_path.replace("\\", "/").lstrip("./")
+    if file_path.startswith("a/") or file_path.startswith("b/"):
+        return file_path[2:]
+
+    return file_path
+
+
+def find_diff_file(diff: Diff, file_path: str) -> DiffFile | None:
+    target = normalize_file_path(file_path)
+    for file in diff.files:
+        if normalize_file_path(file.new_name) == target or normalize_file_path(file.orig_name) == target:
+            return file
+
+    return None
+
+
+def read_snapshot(file_path: str, *, base_sha: str | None = None, head_sha: str | None = None) -> str | None:
+    git = GitService()
+    try:
+        if head_sha:
+            text = git.get_file_at_commit(file_path, head_sha)
+            if text is not None:
+                return text
+
+        if base_sha:
+            text = git.get_file_at_commit(file_path, base_sha)
+            if text is not None:
+                return text
+    except Exception as e:
+        logger.warning(f"Git snapshot read failed for {file_path}: {e}")
+
+    try:
+        return Path(file_path).read_text(encoding="utf-8")
+    except Exception as e:
+        logger.warning(f"Workspace read failed for {file_path}: {e}")
+        return None
+
+
+def marker_for_line(line_type: DiffLineType | None = None, *, added: bool = False, removed: bool = False) -> str:
+    if (line_type is DiffLineType.ADDED) or added:
+        return settings.review.review_added_marker
+    if (line_type is DiffLineType.REMOVED) or removed:
+        return settings.review.review_removed_marker
+    return ""

ai_review/services/git/service.py

@@ -0,0 +1,35 @@
+import subprocess
+from pathlib import Path
+
+from ai_review.services.git.types import GitServiceProtocol
+
+
+class GitService(GitServiceProtocol):
+    def __init__(self, repo_dir: str = "."):
+        self.repo_dir = Path(repo_dir)
+
+    def run_git(self, *args: str) -> str:
+        result = subprocess.run(
+            ["git", *args],
+            cwd=self.repo_dir,
+            capture_output=True,
+            text=True,
+            check=True,
+        )
+        return result.stdout
+
+    def get_diff(self, base_sha: str, head_sha: str, unified: int = 3) -> str:
+        return self.run_git("diff", f"--unified={unified}", base_sha, head_sha)
+
+    def get_diff_for_file(self, base_sha: str, head_sha: str, file: str, unified: int = 3) -> str:
+        return self.run_git("diff", f"--unified={unified}", base_sha, head_sha, "--", file)
+
+    def get_changed_files(self, base_sha: str, head_sha: str) -> list[str]:
+        output = self.run_git("diff", "--name-only", base_sha, head_sha)
+        return [line.strip() for line in output.splitlines() if line.strip()]
+
+    def get_file_at_commit(self, file_path: str, sha: str) -> str | None:
+        try:
+            return self.run_git("show", f"{sha}:{file_path}")
+        except subprocess.CalledProcessError:
+            return None

ai_review/services/git/types.py

@@ -0,0 +1,11 @@
+from typing import Protocol
+
+
+class GitServiceProtocol(Protocol):
+    def get_diff(self, base_sha: str, head_sha: str, unified: int = 3) -> str: ...
+
+    def get_diff_for_file(self, base_sha: str, head_sha: str, file: str, unified: int = 3) -> str: ...
+
+    def get_changed_files(self, base_sha: str, head_sha: str) -> list[str]: ...
+
+    def get_file_at_commit(self, file_path: str, sha: str) -> str | None: ...

ai_review/services/llm/claude/client.py

@@ -0,0 +1,26 @@
+from ai_review.clients.claude.client import get_claude_http_client
+from ai_review.clients.claude.schema import ClaudeChatRequestSchema, ClaudeMessageSchema
+from ai_review.config import settings
+from ai_review.services.llm.types import LLMClient, ChatResultSchema
+
+
+class ClaudeLLMClient(LLMClient):
+    def __init__(self):
+        self.http_client = get_claude_http_client()
+
+    async def chat(self, prompt: str, prompt_system: str) -> ChatResultSchema:
+        meta = settings.llm.meta
+        request = ClaudeChatRequestSchema(
+            model=meta.model,
+            system=prompt_system,
+            messages=[ClaudeMessageSchema(role="user", content=prompt)],
+            max_tokens=meta.max_tokens,
+            temperature=meta.temperature,
+        )
+        response = await self.http_client.chat(request)
+        return ChatResultSchema(
+            text=response.first_text,
+            total_tokens=response.usage.total_tokens,
+            prompt_tokens=response.usage.input_tokens,
+            completion_tokens=response.usage.output_tokens,
+        )

ai_review/services/llm/factory.py

@@ -0,0 +1,18 @@
+from ai_review.config import settings
+from ai_review.libs.constants.llm_provider import LLMProvider
+from ai_review.services.llm.claude.client import ClaudeLLMClient
+from ai_review.services.llm.gemini.client import GeminiLLMClient
+from ai_review.services.llm.openai.client import OpenAILLMClient
+from ai_review.services.llm.types import LLMClient
+
+
+def get_llm_client() -> LLMClient:
+    match settings.llm.provider:
+        case LLMProvider.OPENAI:
+            return OpenAILLMClient()
+        case LLMProvider.GEMINI:
+            return GeminiLLMClient()
+        case LLMProvider.CLAUDE:
+            return ClaudeLLMClient()
+        case _:
+            raise ValueError(f"Unsupported provider: {settings.llm.provider}")

ai_review/services/llm/gemini/client.py

@@ -0,0 +1,31 @@
+from ai_review.clients.gemini.client import get_gemini_http_client
+from ai_review.clients.gemini.schema import (
+    GeminiPartSchema,
+    GeminiContentSchema,
+    GeminiChatRequestSchema,
+    GeminiGenerationConfigSchema,
+)
+from ai_review.config import settings
+from ai_review.services.llm.types import LLMClient, ChatResultSchema
+
+
+class GeminiLLMClient(LLMClient):
+    def __init__(self):
+        self.http_client = get_gemini_http_client()
+
+    async def chat(self, prompt: str, prompt_system: str) -> ChatResultSchema:
+        request = GeminiChatRequestSchema(
+            contents=[GeminiContentSchema(parts=[GeminiPartSchema(text=prompt)])],
+            generation_config=GeminiGenerationConfigSchema(
+                temperature=settings.llm.meta.temperature,
+                max_output_tokens=settings.llm.meta.max_tokens,
+            ),
+            system_instruction=GeminiContentSchema(parts=[GeminiPartSchema(text=prompt_system)])
+        )
+        response = await self.http_client.chat(request)
+        return ChatResultSchema(
+            text=response.first_text,
+            total_tokens=response.usage.total_tokens,
+            prompt_tokens=response.usage.prompt_tokens,
+            completion_tokens=response.usage.completion_tokens,
+        )

ai_review/services/llm/openai/client.py

@@ -0,0 +1,28 @@
+from ai_review.clients.openai.client import get_openai_http_client
+from ai_review.clients.openai.schema import OpenAIChatRequestSchema, OpenAIMessageSchema
+from ai_review.config import settings
+from ai_review.services.llm.types import LLMClient, ChatResultSchema
+
+
+class OpenAILLMClient(LLMClient):
+    def __init__(self):
+        self.http_client = get_openai_http_client()
+
+    async def chat(self, prompt: str, prompt_system: str) -> ChatResultSchema:
+        meta = settings.llm.meta
+        request = OpenAIChatRequestSchema(
+            model=meta.model,
+            messages=[
+                OpenAIMessageSchema(role="system", content=prompt_system),
+                OpenAIMessageSchema(role="user", content=prompt),
+            ],
+            max_tokens=meta.max_tokens,
+            temperature=meta.temperature,
+        )
+        response = await self.http_client.chat(request)
+        return ChatResultSchema(
+            text=response.first_text,
+            total_tokens=response.usage.total_tokens,
+            prompt_tokens=response.usage.prompt_tokens,
+            completion_tokens=response.usage.completion_tokens,
+        )

ai_review/services/llm/types.py

@@ -0,0 +1,15 @@
+from typing import Protocol
+
+from pydantic import BaseModel
+
+
+class ChatResultSchema(BaseModel):
+    text: str
+    total_tokens: int | None = None
+    prompt_tokens: int | None = None
+    completion_tokens: int | None = None
+
+
+class LLMClient(Protocol):
+    async def chat(self, prompt: str, prompt_system: str) -> ChatResultSchema:
+        ...

ai_review/services/prompt/adapter.py

@@ -0,0 +1,25 @@
+from ai_review.services.prompt.schema import PromptContextSchema
+from ai_review.services.vcs.types import MRInfoSchema
+
+
+def build_prompt_context_from_mr_info(mr: MRInfoSchema) -> PromptContextSchema:
+    return PromptContextSchema(
+        merge_request_title=mr.title,
+        merge_request_description=mr.description,
+
+        merge_request_author_name=mr.author.name,
+        merge_request_author_username=mr.author.username,
+
+        merge_request_reviewers=[reviewer.name for reviewer in mr.reviewers],
+        merge_request_reviewers_usernames=[reviewer.username for reviewer in mr.reviewers],
+        merge_request_reviewer=mr.reviewers[0].name if mr.reviewers else None,
+
+        merge_request_assignees=[assignee.name for assignee in mr.assignees],
+        merge_request_assignees_usernames=[assignee.username for assignee in mr.assignees],
+
+        source_branch=mr.source_branch,
+        target_branch=mr.target_branch,
+
+        labels=mr.labels,
+        changed_files=mr.changed_files,
+    )

ai_review/services/prompt/schema.py

@@ -0,0 +1,71 @@
+from pydantic import BaseModel, Field
+
+from ai_review.config import settings
+
+
+class PromptContextSchema(BaseModel):
+    merge_request_title: str | None = None
+    merge_request_description: str | None = None
+
+    merge_request_author_name: str | None = None
+    merge_request_author_username: str | None = None
+
+    merge_request_reviewer: str | None = None
+    merge_request_reviewers: list[str] = Field(default_factory=list)
+    merge_request_reviewers_usernames: list[str] = Field(default_factory=list)
+
+    merge_request_assignees: list[str] = Field(default_factory=list)
+    merge_request_assignees_usernames: list[str] = Field(default_factory=list)
+
+    source_branch: str | None = None
+    target_branch: str | None = None
+
+    labels: list[str] = Field(default_factory=list)
+    changed_files: list[str] = Field(default_factory=list)
+
+    @property
+    def reviewers_format(self) -> str:
+        return ", ".join(self.merge_request_reviewers)
+
+    @property
+    def reviewers_usernames_format(self) -> str:
+        return ", ".join(self.merge_request_reviewers_usernames)
+
+    @property
+    def assignees_format(self) -> str:
+        return ", ".join(self.merge_request_assignees)
+
+    @property
+    def assignees_usernames_format(self) -> str:
+        return ", ".join(self.merge_request_assignees_usernames)
+
+    @property
+    def labels_format(self) -> str:
+        return ", ".join(self.labels)
+
+    @property
+    def changed_files_format(self) -> str:
+        return ", ".join(self.changed_files)
+
+    def apply_format(self, prompt: str) -> str:
+        return prompt.format(
+            merge_request_title=self.merge_request_title or "",
+            merge_request_description=self.merge_request_description or "",
+
+            merge_request_author_name=self.merge_request_author_name or "",
+            merge_request_author_username=self.merge_request_author_username or "",
+
+            merge_request_reviewer=self.merge_request_reviewer or "",
+            merge_request_reviewers=self.reviewers_format,
+            merge_request_reviewers_usernames=self.reviewers_usernames_format,
+
+            merge_request_assignees=self.assignees_format,
+            merge_request_assignees_usernames=self.assignees_usernames_format,
+
+            source_branch=self.source_branch or "",
+            target_branch=self.target_branch or "",
+
+            labels=self.labels_format,
+            changed_files=self.changed_files_format,
+            **settings.prompt.context
+        )

ai_review/services/prompt/service.py

@@ -0,0 +1,56 @@
+from ai_review.config import settings
+from ai_review.services.diff.schema import DiffFileSchema
+from ai_review.services.prompt.schema import PromptContextSchema
+
+
+def format_file(diff: DiffFileSchema) -> str:
+    return f"# File: {diff.file}\n{diff.diff}\n"
+
+
+class PromptService:
+    @classmethod
+    def build_inline_request(cls, diff: DiffFileSchema, context: PromptContextSchema) -> str:
+        inline_prompts = "\n\n".join(settings.prompt.load_inline())
+        prompt = (
+            f"{inline_prompts}\n\n"
+            f"## Diff\n\n"
+            f"{format_file(diff)}"
+        )
+        return context.apply_format(prompt)
+
+    @classmethod
+    def build_summary_request(cls, diffs: list[DiffFileSchema], context: PromptContextSchema) -> str:
+        changes = "\n\n".join(map(format_file, diffs))
+        summary_prompts = "\n\n".join(settings.prompt.load_summary())
+        prompt = (
+            f"{summary_prompts}\n\n"
+            f"## Changes\n\n"
+            f"{changes}\n"
+        )
+        return context.apply_format(prompt)
+
+    @classmethod
+    def build_context_request(cls, diffs: list[DiffFileSchema], context: PromptContextSchema) -> str:
+        changes = "\n\n".join(map(format_file, diffs))
+        inline_prompts = "\n\n".join(settings.prompt.load_context())
+        prompt = (
+            f"{inline_prompts}\n\n"
+            f"## Diff\n\n"
+            f"{changes}\n"
+        )
+        return context.apply_format(prompt)
+
+    @classmethod
+    def build_system_inline_request(cls, context: PromptContextSchema) -> str:
+        prompt = "\n\n".join(settings.prompt.load_system_inline())
+        return context.apply_format(prompt)
+
+    @classmethod
+    def build_system_context_request(cls, context: PromptContextSchema) -> str:
+        prompt = "\n\n".join(settings.prompt.load_system_context())
+        return context.apply_format(prompt)
+
+    @classmethod
+    def build_system_summary_request(cls, context: PromptContextSchema) -> str:
+        prompt = "\n\n".join(settings.prompt.load_system_summary())
+        return context.apply_format(prompt)