ai_git 0.0.4 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a43b04eaad5bed61c5dbe62df46fb915e83d0481e274c2f440b3be2bd7966660
-  data.tar.gz: c74c24c20d94f28a58938ec1f8d0a0d323c15a0522fd6d3596b7f8e51323182d
+  metadata.gz: f3d2a98e6fef0c0ce3deb757a48e43701f1800d8c524f0cea525a92c8b4ae575
+  data.tar.gz: 054d32afa6032a6bdfb2d4946673466b2b65ec6628dfecfb359bf67442c54e4e
 SHA512:
-  metadata.gz: 03cbbe8cf09690cc3db34f98bf42e39479672ce464d65de18a6a25d901ea9c8d35c318a0efa4eac06abc3d1f07d1848be24506ca4c6146facff7e9dcb18a1e80
-  data.tar.gz: 47a3c30cbb3ed60d128db9d68af240b5404096ffc95643766cb873296f8e3f2ccacc5cbc284be4fdc6d984dd1c310ea2ff01873c5442abb86745e1d34be59c19
+  metadata.gz: b9064aeb4a13c6719a3847a41e07e3a933d2404cd7be169cf7806d6efa5ade9d5c3b0f1ebeb12c9951ac1386ec9fe40af83bc8cba5017829f788e1211bc9f566
+  data.tar.gz: 6eacdb5d1ad3be96c5893466e02e28f2e5886091e5f8ca19fcd6814d31fc2629938bfc18447c3e07b98d6fde28e7582b63f483b0e3631deb0829f2e424d619ad

data/bin/ai_git

@@ -3,4 +3,4 @@
 
 require_relative "../lib/ai_git"
 
-AIGit.run
+AIGit.start(ARGV)

data/lib/ai_git/{ollama.rb → default.rb}

@@ -1,11 +1,14 @@
 # frozen_string_literal: true
 
+require "benchmark"
 require "net/http"
 require "uri"
 require "json"
+require_relative "version"
+require_relative "git"
 
 module AIGit
-  module Ollama
+  module Default
     module_function
 
     def escape_json(string)
@@ -79,11 +82,10 @@ module AIGit
         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
+        num_predict: 400
       }.to_json
 
       uri = URI("http://localhost:11434/api/generate")
@@ -100,24 +102,51 @@ module AIGit
       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 = "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

@@ -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,5 +1,5 @@
 # frozen_string_literal: true
 
 module AIGit
-  VERSION = "0.0.4"
+  VERSION = "0.1.0"
 end

data/lib/ai_git.rb

@@ -1,42 +1,23 @@
 # frozen_string_literal: true
 
-require "benchmark"
 require_relative "ai_git/version"
 require_relative "ai_git/git"
-require_relative "ai_git/ollama"
+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"
+  SUBCOMMANDS = {
+    "review" => AIGit::Review,
+    "default" => AIGit::Default
+  }.freeze
 
-    staged = AIGit::Git.staged_files
-    abort "Error: No staged files. Use `git add` first." if staged.to_s.strip.empty?
+  def start(args)
+    command = args.first || "default"
 
-    diff = AIGit::Git.diff
-    branch = AIGit::Git.current_branch
+    raise "Unknown subcommand: #{command}" unless SUBCOMMANDS.key?(command)
 
-    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"
-
-      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
+    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.4
+  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: AI‑powered Git commit + push tool using SLMs
+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 + push tool using SLMs
+summary: AI‑powered Git using SLMs
 test_files: []