ai-reviewer 1.0.0

AI code reviewer

Git Commit File Reviewer

This project provides a Python script that retrieves files modified in the latest Git commit and displays the updated lines for each file, along with a review of the code with the updates. It is designed to be used in Continuous Integration (CI) environments, such as GitHub Actions and GitLab CI.

Features

• Get a list of files modified in the latest commit.
• Retrieve updated lines for each modified file.
• Review code changes using the LLaMA model from Ollama.
• Easy integration with CI/CD pipelines.
• New: Automatically reviews changes with the LLaMA 3.2 model after retrieving updated lines.
• New: GitHub CLI (gh) for adding comments to pull requests

Prerequisites

• Docker
• Git
• Python 3.11 or higher (for local testing)
• New: GitHub CLI (gh) for adding comments to pull requests

Environment Variables

To run this project, you need to set the following environment variables:

export OLLAMA_API_BASE_URL="https://ollama.kabla.xyz"
export IGNORED_FILES="['README.md', '.gitignore', 'requirements.txt']"
export OLLAMA_MODEL="codellama"

Setup

1. Clone the repository:

   git clone <repository-url>
   cd <repository-directory>
   

2. Build the Docker image:

   docker build -t git-commit-file-tracker .
   

3. Run the Docker container:

   docker run --rm git-commit-file-tracker
   

Usage in CI/CD

GitHub Actions

You can use the following example workflow to integrate this script into your GitHub Actions:

name: CI

on:
  push:
    branches:
      - main

jobs:
  track-commits:
    runs-on: ubuntu-latest
    steps:
      - name: Checkout code
        uses: actions/checkout@v2

      - name: Build and run Docker container
        run: |
          docker build -t git-commit-file-tracker .
          docker run --rm git-commit-file-tracker

GitLab CI

For GitLab CI, you can add the following configuration to your .gitlab-ci.yml file:

stages:
  - track

track_commits:
  stage: track
  image: python:3.11
  before_script:
    - apt-get update && apt-get install -y git
  script:
    - python main.py

Python Script: main.py

The main.py script is responsible for fetching pull request data, parsing the changes, and generating suggestions for improvements. It utilizes the LLaMA model for code review and integrates with the GitHub CLI to add comments to pull requests

Key Functions

• init_cache_db: Initializes a SQLite database for caching.
• get_pull_request: Fetches pull request data from Git.
• parse_diff: Parses the Git diff output to extract modified lines.
• propose_updates: Generates suggestions for modified lines using the LLaMA model.
• add_comments_to_pr: Adds comments to the pull request with suggestions.

License

This project is licensed under the MIT License. See the LICENSE file for details.

Contributing

Contributions are welcome! Please open an issue or submit a pull request for any improvements or bug fixes.

name: LLM Code Review

on:
  pull_request:
    types: [opened, synchronize]

jobs:
  code_review:
    runs-on: ubuntu-latest

    steps:
    - name: Checkout code
      uses: actions/checkout@v3

    - name: Set up Python
      uses: actions/setup-python@v4
      with:
        python-version: '3.x'

    - name: Install dependencies
      run: |
        pip install requests langchain_core

    - name: Run code review script
      env:
        OLLAMA_MODEL: ${{ secrets.OLLAMA_MODEL }}
        OLLAMA_API_BASE_URL: ${{ secrets.OLLAMA_API_BASE_URL }}
        IGNORED_FILES: ${{ secrets.IGNORED_FILES }}
      run: script.python