ai_git 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 36409c80888ac303814accb279c6908e55693271fc92b22c2eb151a0539285a4
-  data.tar.gz: b17d56e53f81da126bdd231d12964b799555231eab76dc2cf7002c3a4d01e368
+  metadata.gz: f3d2a98e6fef0c0ce3deb757a48e43701f1800d8c524f0cea525a92c8b4ae575
+  data.tar.gz: 054d32afa6032a6bdfb2d4946673466b2b65ec6628dfecfb359bf67442c54e4e
 SHA512:
-  metadata.gz: e760bbd37b63c7ba1cfcb486e084742718033924990b590f03e51fd9c3ec31ece730c287c0f364680fff34962a68a9d4046f66611d2f9b350c8ac520a76f165d
-  data.tar.gz: e2d581350863a6b9dd6ff67a537b57aa126233f9cbfc921dcd63cd5f64b79cd404befff9e4157940f3ee921a4c47c4f749f37e12ed3d750fdb41f0e5303624c9
+  metadata.gz: b9064aeb4a13c6719a3847a41e07e3a933d2404cd7be169cf7806d6efa5ade9d5c3b0f1ebeb12c9951ac1386ec9fe40af83bc8cba5017829f788e1211bc9f566
+  data.tar.gz: 6eacdb5d1ad3be96c5893466e02e28f2e5886091e5f8ca19fcd6814d31fc2629938bfc18447c3e07b98d6fde28e7582b63f483b0e3631deb0829f2e424d619ad

data/bin/ai_git

@@ -1,5 +1,6 @@
 #!/usr/bin/env ruby
+# frozen_string_literal: true
 
 require_relative "../lib/ai_git"
 
-AIGit.run
+AIGit.start(ARGV)

data/lib/ai_git/default.rb

@@ -0,0 +1,152 @@
+# 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/git.rb

@@ -1,3 +1,6 @@
+# frozen_string_literal: true
+
+require "English"
 module AIGit
   module Git
     module_function
@@ -17,7 +20,7 @@ module AIGit
 
     def run_command(cmd, args)
       system("#{cmd} #{args} > /dev/null 2>&1")
-      raise "Command failed: #{cmd} #{args}" if $?.exitstatus != 0
+      raise "Command failed: #{cmd} #{args}" if $CHILD_STATUS.exitstatus != 0
     end
   end
 end

data/lib/ai_git/review.rb

@@ -0,0 +1,148 @@
+# 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

data/lib/ai_git/version.rb

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

data/lib/ai_git.rb

@@ -1,40 +1,23 @@
-require 'benchmark'
-require_relative 'ai_git/version'
-require_relative 'ai_git/git'
-require_relative 'ai_git/ollama'
+# frozen_string_literal: true
+
+require_relative "ai_git/version"
+require_relative "ai_git/git"
+require_relative "ai_git/review"
+require_relative "ai_git/default"
 
 module AIGit
   module_function
 
-  def run
-    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 = AIGit::Ollama.generate_commit_message(diff, model_name)
-      message = message.gsub(/\n{2,}/, "\n")
-
-      puts "\e[1mCommit Message:\e[0m\n\n#{message}\n"
+  SUBCOMMANDS = {
+    "review" => AIGit::Review,
+    "default" => AIGit::Default
+  }.freeze
 
-      escaped_msg = message.gsub(/[\\"`$]/) { |c| "\\#{c}" }
-      AIGit::Git.run_command('git', "commit -m \"#{escaped_msg}\"")
-      puts "\e[1mGit Commited\e[0m"
+  def start(args)
+    command = args.first || "default"
 
-      AIGit::Git.run_command('git', 'push')
-      puts "\e[1mGit Pushed\e[0m"
-    end
+    raise "Unknown subcommand: #{command}" unless SUBCOMMANDS.key?(command)
 
-    puts "\e[1mBenchmark\e[0m"
-    puts result
+    SUBCOMMANDS[command].call
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai_git
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Kaíque Kandy Koga
@@ -9,7 +9,7 @@ bindir: bin
 cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies: []
-description: Generates intelligent commit messages using Ollama LLMs
+description: AI‑powered Git using SLMs
 executables:
 - ai_git
 extensions: []
@@ -17,8 +17,9 @@ extra_rdoc_files: []
 files:
 - bin/ai_git
 - lib/ai_git.rb
+- lib/ai_git/default.rb
 - lib/ai_git/git.rb
-- lib/ai_git/ollama.rb
+- lib/ai_git/review.rb
 - lib/ai_git/version.rb
 homepage: https://github.com/kaiquekandykoga/ai_git
 licenses:
@@ -40,5 +41,5 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 rubygems_version: 4.0.9
 specification_version: 4
-summary: AI-powered git commit and push tool using Ollama
+summary: AI‑powered Git using SLMs
 test_files: []

data/lib/ai_git/ollama.rb

@@ -1,121 +0,0 @@
-require 'net/http'
-require 'uri'
-require 'json'
-
-module AIGit
-  module Ollama
-    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,
-        # These parameters help a lot with strictness:
-        temperature: 0.3,
-        top_p: 0.9,
-        stop: ["\n\n\n", "```", "Here is", "The commit message"],
-        num_predict: 400   # Limit output length
-      }.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
-
-      # Aggressive cleaning
-      message = message.gsub(/^(Here is|The commit message is|```|json|markdown)/i, '')
-        .gsub(/^>\s*/, '')
-        .gsub(/\\n/, "\n")
-        .strip
-
-      # Remove common unwanted prefixes/suffixes
-      lines = message.lines.map(&:strip)
-      lines.reject! { |line| line.match?(/^(Here|Output|Generated|Based on|The changes)/i) }
-
-      message = lines.join("\n").strip
-
-      # Ensure it has at least a title
-      if message.lines.count < 1 || message.strip.empty?
-        message = "chore: update code"  # fallback
-      end
-
-      message
-    end 
-  end
-end