ai_git 0.1.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f3d2a98e6fef0c0ce3deb757a48e43701f1800d8c524f0cea525a92c8b4ae575
-  data.tar.gz: 054d32afa6032a6bdfb2d4946673466b2b65ec6628dfecfb359bf67442c54e4e
+  metadata.gz: e698533dcbb5389f8ada1834ad78bbc65cbd6671e37874203de67c7376b11b51
+  data.tar.gz: 858e8d1de39b0034506da5fbd7858029e6ca77e922c551ceee4da1c2b4feb732
 SHA512:
-  metadata.gz: b9064aeb4a13c6719a3847a41e07e3a933d2404cd7be169cf7806d6efa5ade9d5c3b0f1ebeb12c9951ac1386ec9fe40af83bc8cba5017829f788e1211bc9f566
-  data.tar.gz: 6eacdb5d1ad3be96c5893466e02e28f2e5886091e5f8ca19fcd6814d31fc2629938bfc18447c3e07b98d6fde28e7582b63f483b0e3631deb0829f2e424d619ad
+  metadata.gz: 330b7566ca8ec91486faffd05ebefe651e543df74f39b796001d568793342cfa53689699baec35be9ec6a15b1f11505c60a06e17297e696593e752661465a6c2
+  data.tar.gz: 357d164f5a63cfb02bd173902bee1c098a3b24590a2347f33a97b0bff259c786c87aee1fca5e79937ccf7e916df9214d21cd280884370c877e520dd8dc2aea5b

data/LICENSE

@@ -0,0 +1,28 @@
+BSD 3-Clause License
+
+Copyright (c) 2026, Kaíque Kandy Koga
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

data/README.md

@@ -0,0 +1,66 @@
+# AI Git
+
+AI-powered Git commit messages using a local LLM
+
+## Usage
+
+#### Requirements
+
+- [llama.cpp](https://github.com/ggml-org/llama.cpp) running locally (e.g. `./llama-server --port 8080`)
+
+#### Install
+
+```bash
+gem install ai_git
+```
+
+#### Environment Variables
+
+| Variable | Description | Default |
+|----------|-------------|---------|
+| `AI_GIT_MODEL_NAME` | Model name | `ggml-org/gemma-4-E4B-it-GGUF:Q8_0` |
+| `AI_GIT_BASE_URL` | Base URL of the llama.cpp server | `http://127.0.0.1:8080` |
+| `NO_COLOR` | Disable colored terminal output when set | — |
+
+Run `ai_git config` to see exactly which model, URL and endpoint are resolved from your environment.
+
+ai_git talks to a local [llama.cpp](https://github.com/ggml-org/llama.cpp) server over its OpenAI-compatible
+`/v1/chat/completions` endpoint. No API key is needed.
+
+##### Example
+
+```bash
+# Start llama.cpp's server, then run ai_git with defaults
+./llama-server --port 8080
+ai_git
+
+# Or point at a custom model/port
+export AI_GIT_MODEL_NAME=my-model
+export AI_GIT_BASE_URL=http://127.0.0.1:8081
+```
+
+#### Run
+
+```bash
+git add <files>
+ai_git
+```
+
+`ai_git` generates a commit message from your staged changes, commits, and
+pushes to the current branch's upstream — no prompts, no flags.
+
+#### Subcommands
+
+| Subcommand | Description |
+|------------|-------------|
+| `ai_git` | Generate a commit message, commit, and push staged files |
+| `ai_git config` | Show the resolved provider configuration |
+| `ai_git --help` | Show usage |
+| `ai_git --version` | Print version |
+
+## Development
+
+```bash
+gem build ai_git.gemspec
+gem install ./ai_git-$(ruby -r./lib/ai_git/version -e 'print AIGit::VERSION').gem
+```

data/bin/ai_git

@@ -3,4 +3,12 @@
 
 require_relative "../lib/ai_git"
 
-AIGit.start(ARGV)
+begin
+  AIGit.start(ARGV)
+rescue Interrupt
+  warn "\nAborted."
+  exit 130
+rescue StandardError => e
+  AIGit::UI.error("ai_git: #{e.message}")
+  exit 1
+end

data/lib/ai_git/ai_client.rb

@@ -0,0 +1,127 @@
+# frozen_string_literal: true
+
+require "json"
+require "net/http"
+require "uri"
+
+require_relative "config"
+
+module AIGit
+  module AIClient
+    module_function
+
+    READ_TIMEOUT_SECONDS = 120
+    OPEN_TIMEOUT_SECONDS = 10
+
+    # Transient failures worth retrying with backoff.
+    MAX_ATTEMPTS      = 3
+    RETRY_BASE_DELAY  = 0.5
+    TRANSIENT_STATUSES = [408, 425, 429, 500, 502, 503, 504].freeze
+    RETRYABLE_ERRORS = [
+      Errno::ECONNREFUSED, Errno::ECONNRESET, Errno::EPIPE,
+      Net::OpenTimeout, Net::ReadTimeout, SocketError, EOFError
+    ].freeze
+
+    OUTPUT_NOISE_PREFIXES = /^(Here|Output|Generated|Based on|The changes)/i.freeze
+    OUTPUT_NOISE_HEADERS  = /^(Here is|The (commit message|review) is|```|json|markdown)/i.freeze
+
+    # Sends `prompt` to the local llama.cpp server and returns the cleaned text body.
+    def complete(prompt:, model_name:, temperature:)
+      sanitize(openai_complete(prompt, model_name, temperature))
+    end
+
+    def openai_complete(prompt, model_name, temperature)
+      body = {
+        model: model_name,
+        messages: [{ role: "user", content: prompt }],
+        stream: false,
+        temperature: temperature
+      }
+
+      data = post_json(body)
+      data.dig("choices", 0, "message", "content").to_s
+    end
+
+    def transient_status?(code)
+      TRANSIENT_STATUSES.include?(code.to_i)
+    end
+
+    def retry_delay(attempt)
+      RETRY_BASE_DELAY * (2**(attempt - 1))
+    end
+
+    def post_json(body)
+      uri = URI("#{AIGit::Config.base_url}#{AIGit::Config.endpoint}")
+      attempt = 0
+
+      loop do
+        attempt += 1
+
+        begin
+          response = perform_request(uri, body)
+        rescue *RETRYABLE_ERRORS => e
+          raise connection_error_message(e) if attempt >= MAX_ATTEMPTS
+
+          sleep retry_delay(attempt)
+          next
+        end
+
+        return JSON.parse(response.body) if response.is_a?(Net::HTTPSuccess)
+
+        if transient_status?(response.code) && attempt < MAX_ATTEMPTS
+          sleep retry_delay(attempt)
+          next
+        end
+
+        raise http_error_message(uri, response)
+      end
+    end
+
+    def perform_request(uri, body)
+      request = Net::HTTP::Post.new(uri)
+      request["Content-Type"] = "application/json"
+      request.body = body.to_json
+
+      Net::HTTP.start(
+        uri.host,
+        uri.port,
+        use_ssl: uri.scheme == "https",
+        open_timeout: OPEN_TIMEOUT_SECONDS,
+        read_timeout: READ_TIMEOUT_SECONDS
+      ) { |http| http.request(request) }
+    end
+
+    def http_error_message(uri, response)
+      provider = AIGit::Config.provider
+      body = response.body.to_s.strip
+      body = "#{body[0, 500]}…" if body.length > 500
+
+      hint = response.code.to_i == 404 ? " Check the model name and base URL (see `ai_git config`)." : ""
+
+      "#{provider} returned HTTP #{response.code} at #{uri}.#{hint}" \
+        "#{body.empty? ? '' : "\n#{body}"}"
+    end
+
+    def connection_error_message(error)
+      provider = AIGit::Config.provider
+      base_url = AIGit::Config.base_url
+      hint = "Is the local server running? See `ai_git config`."
+
+      "Cannot reach #{provider} at #{base_url} after #{MAX_ATTEMPTS} attempts: #{error.message}. #{hint}"
+    end
+
+    def sanitize(text)
+      cleaned = text.to_s
+                    .gsub(OUTPUT_NOISE_HEADERS, "")
+                    .gsub(/^>\s*/, "")
+                    .gsub(/\\n/, "\n")
+                    .strip
+
+      cleaned.lines
+             .map(&:strip)
+             .reject { |line| line.match?(OUTPUT_NOISE_PREFIXES) }
+             .join("\n")
+             .strip
+    end
+  end
+end

data/lib/ai_git/commands/config.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require_relative "../ai_client"
+require_relative "../config"
+require_relative "../ui"
+
+module AIGit
+  module Commands
+    # `ai_git config` — print the resolved provider configuration so users can
+    # see exactly which provider, model, URL and key will be used.
+    module Config
+      module_function
+
+      def call(_argv = [])
+        cfg = AIGit::Config
+
+        AIGit::UI.heading("ai_git configuration")
+        AIGit::UI.kv("Provider", cfg.provider)
+        AIGit::UI.kv("Model", cfg.model_name)
+        AIGit::UI.kv("Base URL", cfg.base_url)
+        AIGit::UI.kv("Endpoint", cfg.endpoint)
+        AIGit::UI.kv("Read timeout", "#{AIGit::AIClient::READ_TIMEOUT_SECONDS}s")
+      end
+    end
+  end
+end

data/lib/ai_git/commands/default.rb

@@ -0,0 +1,145 @@
+# frozen_string_literal: true
+
+require_relative "../ai_client"
+require_relative "../config"
+require_relative "../git"
+require_relative "../ui"
+
+module AIGit
+  module Commands
+    module Default
+      module_function
+
+      def generate_commit_message(diff, model_name, temperature: 0.3)
+        raise "No staged changes to generate commit message for" if diff.to_s.strip.empty?
+
+        message = AIClient.complete(
+          prompt: build_prompt(diff),
+          model_name: model_name,
+          temperature: temperature
+        )
+
+        message = normalize_message(message)
+        message.empty? ? "chore: update code" : message
+      end
+
+      # Keep the commit body readable: collapse runs of blank lines to a single
+      # one and guarantee a blank line after the title so git sees a proper
+      # subject/body split (otherwise `git log --oneline` mashes them together).
+      def normalize_message(message)
+        text = message.to_s.gsub(/\n{3,}/, "\n\n").strip
+        lines = text.lines.map(&:chomp)
+        return text if lines.length < 2
+
+        lines.insert(1, "") unless lines[1].empty?
+        lines.join("\n").gsub(/\n{3,}/, "\n\n").strip
+      end
+
+      def call(_argv = [])
+        provider = AIGit::Config.provider
+        model_name = AIGit::Config.model_name
+
+        staged = AIGit::Git.staged_files
+        abort "Error: No staged files. Use `git add` first." if staged.to_s.strip.empty?
+
+        diff = AIGit::Git.diff
+        branch = AIGit::Git.current_branch
+
+        print_header(provider, model_name, staged, branch)
+
+        started = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+
+        AIGit::UI.info(AIGit::UI.bold("Generating commit message…"))
+        message = generate_commit_message(diff, model_name)
+        print_message(message)
+
+        AIGit::Git.commit_with_message(message)
+        AIGit::UI.success("Committed.")
+
+        AIGit::Git.push_current_branch
+        AIGit::UI.success("Pushed to origin/#{branch}.")
+
+        elapsed = Process.clock_gettime(Process::CLOCK_MONOTONIC) - started
+        AIGit::UI.kv("Done in", format("%.1fs", elapsed))
+      end
+
+      def print_header(provider, model_name, staged, branch)
+        AIGit::UI.kv("AI Provider", provider)
+        AIGit::UI.kv("Model", model_name)
+        AIGit::UI.kv("Staged Files", staged.to_s.strip.gsub("\n", ", "))
+        AIGit::UI.kv("Branch", branch)
+        puts
+      end
+
+      def print_message(message)
+        puts
+        puts AIGit::UI.bold("Commit message:")
+        puts
+        puts message
+        puts
+      end
+
+      def build_prompt(diff)
+        standard_prompt(diff)
+      end
+
+      def standard_prompt(diff)
+        <<~PROMPT
+          You are an expert Git commit message writer. Output ONLY the commit message — no explanations, no markdown, no backticks, no preamble.
+
+          Here are the changes:
+          #{diff}
+
+          STRICT OUTPUT FORMAT (follow exactly):
+
+          <short imperative title, max 72 chars>
+
+          <blank line>
+
+          ## Summary
+          <2–4 bullet points covering the most important changes. Each bullet starts with a verb.>
+
+          ## Why
+          <1–3 sentences explaining the motivation or context behind the change. Omit if the reason is obvious.>
+
+          RULES:
+          - Title line: short, specific, imperative mood (e.g. "Add JWT login with refresh token support"). Avoid vague titles like "Update stuff" or "Fix bug".
+          - Summary bullets: describe WHAT changed, not HOW the code looks. Focus on behaviour and impact.
+          - Why section: explain the problem being solved or the goal being achieved. Skip if it adds no value.
+          - No filler phrases ("this commit", "this PR", "as per discussion").
+          - No line should exceed 72 characters.
+
+          EXAMPLES OF GOOD OUTPUT:
+
+          Add JWT-based login with refresh token support
+
+          ## Summary
+          - Implement login endpoint with access and refresh token issuance
+          - Add token refresh route with rotation and expiry validation
+          - Protect private routes via middleware that verifies access tokens
+          - Store refresh tokens using encrypted HTTP-only cookies
+
+          ## Why
+          Users were being logged out on every page reload. Refresh tokens allow
+          sessions to persist securely without requiring re-authentication.
+
+          ---
+
+          Prevent nil crash when user preferences are missing
+
+          ## Summary
+          - Add nil guard in ReportGenerator#process before accessing preferences
+          - Fall back to system defaults when preferences object is absent
+
+          ## Why
+          Reports were raising NoMethodError in production for users created
+          before the preferences feature shipped.
+
+          ---
+
+          Now generate the commit message:
+        PROMPT
+      end
+    end
+  end
+end

data/lib/ai_git/config.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module AIGit
+  module Config
+    PROVIDER = "llama_cpp"
+    DEFAULT_MODEL = "ggml-org/gemma-4-E4B-it-GGUF:Q8_0"
+    DEFAULT_BASE_URL = "http://127.0.0.1:8080"
+    ENDPOINT = "/v1/chat/completions"
+
+    module_function
+
+    def provider
+      PROVIDER
+    end
+
+    def model_name
+      ENV["AI_GIT_MODEL_NAME"] || DEFAULT_MODEL
+    end
+
+    def base_url
+      ENV["AI_GIT_BASE_URL"] || DEFAULT_BASE_URL
+    end
+
+    def endpoint
+      ENDPOINT
+    end
+  end
+end

data/lib/ai_git/git.rb

@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
-require "English"
+require "open3"
+require "tempfile"
+
 module AIGit
   module Git
     module_function
@@ -14,13 +16,41 @@ module AIGit
     end
 
     def current_branch
-      result = `git rev-parse --abbrev-ref HEAD`
-      result.chomp
+      `git rev-parse --abbrev-ref HEAD`.chomp
+    end
+
+    # Run a command with arguments as an array — no shell, so values are not
+    # interpolated or word-split. Accepts either:
+    #   run_command("git", "status")                  (single string of args)
+    #   run_command("git", "commit", "-m", message)   (variadic args, preferred)
+    def run_command(cmd, *args)
+      argv =
+        if args.length == 1 && args.first.is_a?(String)
+          args.first.split
+        else
+          args.map(&:to_s)
+        end
+
+      _stdout, stderr, status = Open3.capture3(cmd, *argv)
+
+      return if status.success?
+
+      raise "Command failed: #{cmd} #{argv.join(' ')} (exit #{status.exitstatus})#{stderr.empty? ? '' : "\n#{stderr}"}"
+    end
+
+    # Commit using a temp file so the message can contain anything (quotes,
+    # backticks, dollar signs) without shell-escaping concerns.
+    def commit_with_message(message)
+      Tempfile.create("ai_git_commit_msg") do |file|
+        file.write(message)
+        file.flush
+
+        run_command("git", "commit", "-F", file.path)
+      end
     end
 
-    def run_command(cmd, args)
-      system("#{cmd} #{args} > /dev/null 2>&1")
-      raise "Command failed: #{cmd} #{args}" if $CHILD_STATUS.exitstatus != 0
+    def push_current_branch
+      run_command("git", "push", "-u", "origin", "HEAD")
     end
   end
 end

data/lib/ai_git/ui.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+module AIGit
+  # Terminal output helpers: ANSI colors (auto-disabled when output is not a
+  # TTY or NO_COLOR is set) and key/value lines.
+  module UI
+    module_function
+
+    CODES = {
+      bold: 1, dim: 2, red: 31, green: 32, yellow: 33, blue: 34, cyan: 36, gray: 90
+    }.freeze
+
+    # Colors are on only for an interactive terminal and when the user has not
+    # opted out via NO_COLOR (https://no-color.org) or AI_GIT_NO_COLOR.
+    def color?
+      return false if ENV["NO_COLOR"] && !ENV["NO_COLOR"].empty?
+      return false if ENV["AI_GIT_NO_COLOR"] && !ENV["AI_GIT_NO_COLOR"].empty?
+
+      $stdout.tty?
+    end
+
+    def paint(text, *styles)
+      return text.to_s unless color?
+
+      codes = styles.map { |style| CODES[style] }.compact
+      return text.to_s if codes.empty?
+
+      "\e[#{codes.join(';')}m#{text}\e[0m"
+    end
+
+    def bold(text)
+      paint(text, :bold)
+    end
+
+    def dim(text)
+      paint(text, :dim)
+    end
+
+    def kv(key, value)
+      puts "#{paint("#{key}:", :bold)} #{value}"
+    end
+
+    def heading(text)
+      puts paint(text, :bold, :cyan)
+    end
+
+    def info(text)
+      puts text
+    end
+
+    def success(text)
+      puts paint(text, :green)
+    end
+
+    def error(text)
+      $stderr.puts paint(text, :red) # rubocop:disable Style/StderrPuts
+    end
+  end
+end

data/lib/ai_git/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AIGit
-  VERSION = "0.1.0"
+  VERSION = "0.3.0"
 end

data/lib/ai_git.rb

@@ -1,23 +1,58 @@
 # frozen_string_literal: true
 
 require_relative "ai_git/version"
+require_relative "ai_git/config"
+require_relative "ai_git/ui"
+require_relative "ai_git/ai_client"
 require_relative "ai_git/git"
-require_relative "ai_git/review"
-require_relative "ai_git/default"
+require_relative "ai_git/commands/default"
+require_relative "ai_git/commands/config"
 
 module AIGit
   module_function
 
   SUBCOMMANDS = {
-    "review" => AIGit::Review,
-    "default" => AIGit::Default
+    "config" => AIGit::Commands::Config,
+    "default" => AIGit::Commands::Default
   }.freeze
 
+  HELP_FLAGS    = %w[-h --help help].freeze
+  VERSION_FLAGS = %w[-v --version].freeze
+
+  USAGE = <<~USAGE
+    Usage: ai_git [subcommand]
+
+    Subcommands:
+      (none)    Generate a commit message, commit, and push staged files
+      config    Show the resolved provider configuration
+
+    Flags:
+      -h, --help     Show this message
+      -v, --version  Print version
+
+    Environment variables:
+      AI_GIT_MODEL_NAME    Override the default model
+      AI_GIT_BASE_URL      Override the default base URL
+      NO_COLOR             Disable colored output
+  USAGE
+
   def start(args)
-    command = args.first || "default"
+    args = args.dup
+    first = args.first
+
+    return puts(USAGE) if first && HELP_FLAGS.include?(first)
+    return puts(VERSION) if first && VERSION_FLAGS.include?(first)
 
-    raise "Unknown subcommand: #{command}" unless SUBCOMMANDS.key?(command)
+    command = "default"
+    if first && !first.start_with?("-")
+      unless SUBCOMMANDS.key?(first)
+        warn "Unknown subcommand: #{first}"
+        warn USAGE
+        exit 1
+      end
+      command = args.shift
+    end
 
-    SUBCOMMANDS[command].call
+    SUBCOMMANDS[command].call(args)
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai_git
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Kaíque Kandy Koga
@@ -9,22 +9,31 @@ bindir: bin
 cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies: []
-description: AI‑powered Git using SLMs
+description: Generate Git commit messages from staged changes via a local llama.cpp
+  server, then commit and push.
 executables:
 - ai_git
 extensions: []
 extra_rdoc_files: []
 files:
+- LICENSE
+- README.md
 - bin/ai_git
 - lib/ai_git.rb
-- lib/ai_git/default.rb
+- lib/ai_git/ai_client.rb
+- lib/ai_git/commands/config.rb
+- lib/ai_git/commands/default.rb
+- lib/ai_git/config.rb
 - lib/ai_git/git.rb
-- lib/ai_git/review.rb
+- lib/ai_git/ui.rb
 - lib/ai_git/version.rb
 homepage: https://github.com/kaiquekandykoga/ai_git
 licenses:
 - BSD-3-Clause
-metadata: {}
+metadata:
+  source_code_uri: https://github.com/kaiquekandykoga/ai_git
+  bug_tracker_uri: https://github.com/kaiquekandykoga/ai_git/issues
+  rubygems_mfa_required: 'true'
 rdoc_options: []
 require_paths:
 - lib
@@ -32,14 +41,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '4.0'
+      version: '0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.9
+rubygems_version: 4.0.10
 specification_version: 4
-summary: AI‑powered Git using SLMs
+summary: AI-powered Git commit messages using a local LLM
 test_files: []

data/lib/ai_git/default.rb

@@ -1,152 +0,0 @@
-# frozen_string_literal: true
-
-require "benchmark"
-require "net/http"
-require "uri"
-require "json"
-require_relative "version"
-require_relative "git"
-
-module AIGit
-  module Default
-    module_function
-
-    def escape_json(string)
-      string.gsub("\\", "\\\\")
-            .gsub('"', '\"')
-            .gsub("\n", '\\n')
-            .gsub("\r", '\\r')
-            .gsub("\t", '\\t')
-    end
-
-    def generate_commit_message(diff, model_name)
-      raise "No staged changes to generate commit message for" if diff.to_s.strip.empty?
-
-      prompt = <<~PROMPT
-        You are an expert Git commit message writer. Output ONLY the commit message — no explanations, no markdown, no backticks, no preamble.
-
-        Here are the changes:
-        #{diff}
-
-        STRICT OUTPUT FORMAT (follow exactly):
-
-        <short imperative title, max 72 chars>
-
-        <blank line>
-
-        ## Summary
-        <2–4 bullet points covering the most important changes. Each bullet starts with a verb.>
-
-        ## Why
-        <1–3 sentences explaining the motivation or context behind the change. Omit if the reason is obvious.>
-
-        RULES:
-        - Title line: short, specific, imperative mood (e.g. "Add JWT login with refresh token support"). Avoid vague titles like "Update stuff" or "Fix bug".
-        - Summary bullets: describe WHAT changed, not HOW the code looks. Focus on behaviour and impact.
-        - Why section: explain the problem being solved or the goal being achieved. Skip if it adds no value.
-        - No filler phrases ("this commit", "this PR", "as per discussion").
-        - No line should exceed 72 characters.
-
-        EXAMPLES OF GOOD OUTPUT:
-
-        Add JWT-based login with refresh token support
-
-        ## Summary
-        - Implement login endpoint with access and refresh token issuance
-        - Add token refresh route with rotation and expiry validation
-        - Protect private routes via middleware that verifies access tokens
-        - Store refresh tokens using encrypted HTTP-only cookies
-
-        ## Why
-        Users were being logged out on every page reload. Refresh tokens allow
-        sessions to persist securely without requiring re-authentication.
-
-        ---
-
-        Prevent nil crash when user preferences are missing
-
-        ## Summary
-        - Add nil guard in ReportGenerator#process before accessing preferences
-        - Fall back to system defaults when preferences object is absent
-
-        ## Why
-        Reports were raising NoMethodError in production for users created
-        before the preferences feature shipped.
-
-        ---
-
-        Now generate the commit message:
-      PROMPT
-
-      json_body = {
-        model: model_name,
-        prompt: prompt,
-        stream: false,
-        temperature: 0.3,
-        top_p: 0.9,
-        stop: ["\n\n\n", "```", "Here is", "The commit message"],
-        num_predict: 400
-      }.to_json
-
-      uri = URI("http://localhost:11434/api/generate")
-      request = Net::HTTP::Post.new(uri)
-      request["Content-Type"] = "application/json"
-      request.body = json_body
-
-      response = Net::HTTP.start(uri.host, uri.port, read_timeout: 120) do |http|
-        http.request(request)
-      end
-
-      raise "Failed to connect to Ollama. Is it running?" unless response.is_a?(Net::HTTPSuccess)
-
-      data = JSON.parse(response.body)
-      message = data["response"].to_s.strip
-
-      message = message.gsub(/^(Here is|The commit message is|```|json|markdown)/i, "")
-                       .gsub(/^>\s*/, "")
-                       .gsub(/\\n/, "\n")
-                       .strip
-
-      lines = message.lines.map(&:strip)
-      lines.reject! { |line| line.match?(/^(Here|Output|Generated|Based on|The changes)/i) }
-
-      message = lines.join("\n").strip
-
-      message = "chore: update code" if message.lines.count < 1 || message.strip.empty?
-
-      message
-    end
-
-    def call
-      model_name = ENV["AI_GIT_MODEL_NAME"] || "phi4:14b"
-
-      staged = AIGit::Git.staged_files
-      abort "Error: No staged files. Use `git add` first." if staged.to_s.strip.empty?
-
-      diff = AIGit::Git.diff
-      branch = AIGit::Git.current_branch
-
-      puts "\e[1mModel Name:\e[0m #{model_name}"
-      puts "\e[1mStaged Files:\e[0m #{staged}"
-      puts "\e[1mBranch:\e[0m #{branch}"
-      puts "\e[1mAI Generating Commit Message\e[0m"
-
-      result = Benchmark.measure do
-        message = generate_commit_message(diff, model_name)
-        message = message.gsub(/\n{2,}/, "\n")
-
-        puts "\e[1mCommit Message:\e[0m\n\n#{message}\n"
-
-        escaped_msg = message.gsub(/[\\"`$]/) { |c| "\\#{c}" }
-        AIGit::Git.run_command("git", "commit -m \"#{escaped_msg}\"")
-        puts "\e[1mGit Commited\e[0m"
-
-        AIGit::Git.run_command("git", "push")
-        puts "\e[1mGit Pushed\e[0m"
-      end
-
-      puts "\e[1mBenchmark\e[0m"
-      puts result
-    end
-  end
-end

data/lib/ai_git/review.rb

@@ -1,148 +0,0 @@
-# frozen_string_literal: true
-
-require "benchmark"
-require "net/http"
-require "uri"
-require "json"
-require_relative "version"
-require_relative "git"
-
-module AIGit
-  module Review
-    module_function
-
-    def generate_review(diff, model_name)
-      raise "No staged changes to review" if diff.to_s.strip.empty?
-
-      prompt = <<~PROMPT
-        You are a senior software engineer conducting a thorough code review. Output ONLY the review — no explanations, no markdown preamble, no backticks, no preamble.
-
-        Here are the staged changes:
-        #{diff}
-
-        STRICT OUTPUT FORMAT (follow exactly):
-
-        ## Summary
-        <2–5 bullet points describing what this change does at a high level. Each bullet starts with a verb.>
-
-        ## Suggestions
-        <For each issue or improvement, use this format:>
-
-        **<File and location, e.g. "app/models/user.rb">**
-        - <Concise description of the issue or suggestion. Start with a verb. Be specific.>
-        - <Another suggestion for the same file, if applicable.>
-
-        <Repeat the file block for each file with suggestions. Omit files that look clean.>
-
-        ## Verdict
-        <One of: ✅ Looks good | ⚠️ Minor issues | 🚨 Needs attention>
-        <One sentence explaining the verdict.>
-
-        RULES:
-        - Summary: describe WHAT the change does, not HOW the code looks. Focus on behaviour and intent.
-        - Suggestions: flag real issues — bugs, edge cases, security problems, missing error handling, naming confusion, performance concerns, missing tests. Do NOT invent problems. If a file is clean, omit it entirely.
-        - Be direct and constructive. No filler like "great job" or "consider possibly maybe".
-        - No line should exceed 72 characters.
-        - If the diff is a minor or trivial change (e.g. version bump, typo fix), keep the review brief.
-
-        EXAMPLES OF GOOD OUTPUT:
-
-        ## Summary
-        - Add password reset flow with tokenised email verification
-        - Introduce PasswordResetMailer with expiry-aware token links
-        - Add DB migration for reset_token and reset_token_expires_at columns
-
-        ## Suggestions
-
-        **app/models/user.rb**
-        - Ensure reset_token is invalidated after successful use to
-          prevent token reuse attacks.
-        - Add an index on reset_token column for fast lookup queries.
-
-        **app/controllers/passwords_controller.rb**
-        - Handle the case where the token has expired with a user-facing
-          error message rather than a silent redirect.
-
-        ## Verdict
-        ⚠️ Minor issues
-        Core logic is solid but token invalidation and expiry feedback
-        need addressing before merge.
-
-        ---
-
-        ## Summary
-        - Fix nil guard in ReportGenerator when preferences are absent
-
-        ## Suggestions
-
-        ## Verdict
-        ✅ Looks good
-        Safe defensive fix with no side effects.
-
-        ---
-
-        Now review the staged changes:
-      PROMPT
-
-      json_body = {
-        model: model_name,
-        prompt: prompt,
-        stream: false,
-        temperature: 0.2,
-        top_p: 0.9,
-        stop: ["\n\n\n", "```", "Here is", "The review"],
-        num_predict: 600
-      }.to_json
-
-      uri = URI("http://localhost:11434/api/generate")
-      request = Net::HTTP::Post.new(uri)
-      request["Content-Type"] = "application/json"
-      request.body = json_body
-
-      response = Net::HTTP.start(uri.host, uri.port, read_timeout: 120) do |http|
-        http.request(request)
-      end
-
-      raise "Failed to connect to Ollama. Is it running?" unless response.is_a?(Net::HTTPSuccess)
-
-      data = JSON.parse(response.body)
-      review = data["response"].to_s.strip
-
-      review = review.gsub(/^(Here is|The review is|```|json|markdown)/i, "")
-                     .gsub(/^>\s*/, "")
-                     .gsub(/\\n/, "\n")
-                     .strip
-
-      lines = review.lines.map(&:strip)
-      lines.reject! { |line| line.match?(/^(Here|Output|Generated|Based on|The changes)/i) }
-
-      review = lines.join("\n").strip
-      review = "No review generated." if review.empty?
-
-      review
-    end
-
-    def call
-      model_name = ENV["AI_GIT_MODEL_NAME"] || "phi4:14b"
-
-      staged = AIGit::Git.staged_files
-      abort "Error: No staged files. Use `git add` first." if staged.to_s.strip.empty?
-
-      diff = AIGit::Git.diff
-      branch = AIGit::Git.current_branch
-
-      puts "\e[1mModel Name:\e[0m #{model_name}"
-      puts "\e[1mStaged Files:\e[0m #{staged}"
-      puts "\e[1mBranch:\e[0m #{branch}"
-      puts "\e[1mAI Reviewing Changes\e[0m"
-
-      result = Benchmark.measure do
-        review = generate_review(diff, model_name)
-        puts "\e[1mCode Review:\e[0m\n\n#{review}\n"
-      end
-
-      puts "\e[1mBenchmark\e[0m"
-      puts result
-    end
-  end
-end