ai_git 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 65259ec68cad081e1e9a8de3918eae4b338ae23211c5ccde67ef98e089d2fd29
-  data.tar.gz: 8de30045b286d41c52cd4bae828ae6236ae513a693c9fcc9dd2ba1fb4c30b197
+  metadata.gz: a43b04eaad5bed61c5dbe62df46fb915e83d0481e274c2f440b3be2bd7966660
+  data.tar.gz: c74c24c20d94f28a58938ec1f8d0a0d323c15a0522fd6d3596b7f8e51323182d
 SHA512:
-  metadata.gz: ff8d20bb06a60b58e4a8fe07b4632541e8a2a8fc00ffd3d8f2b06d22a6800a06e04f59b6f14562104374640a21e44d0621d1621f25fda66da3fcb246144ddcc6
-  data.tar.gz: 9a32d475d621cfdc506f9c2459c879d9e33de9696e3feac98c755cb86066102cde8a05db2f769b6173cac16e32577940721464224e3a80165de46d28a99f5554
+  metadata.gz: 03cbbe8cf09690cc3db34f98bf42e39479672ce464d65de18a6a25d901ea9c8d35c318a0efa4eac06abc3d1f07d1848be24506ca4c6146facff7e9dcb18a1e80
+  data.tar.gz: 47a3c30cbb3ed60d128db9d68af240b5404096ffc95643766cb873296f8e3f2ccacc5cbc284be4fdc6d984dd1c310ea2ff01873c5442abb86745e1d34be59c19

data/bin/ai_git

@@ -1,4 +1,5 @@
 #!/usr/bin/env ruby
+# frozen_string_literal: true
 
 require_relative "../lib/ai_git"
 

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/ollama.rb

@@ -1,77 +1,110 @@
-require 'net/http'
-require 'uri'
-require 'json'
+# frozen_string_literal: true
+
+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')
+      string.gsub("\\", "\\\\")
+            .gsub('"', '\"')
+            .gsub("\n", '\\n')
+            .gsub("\r", '\\r')
+            .gsub("\t", '\\t')
     end
 
-    def generate_commit_message(diff)
-      raise 'No staged changes to generate commit message for' if diff.to_s.strip.empty?
+    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. Your only job is to output a commit message — nothing else.
+        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.>
 
-    Here are the changes:
-    #{diff}
+        ## Why
+        <1–3 sentences explaining the motivation or context behind the change. Omit if the reason is obvious.>
 
-    Rules you MUST follow exactly:
-    - Output ONLY the commit message. No explanations, no quotes, no markdown, no "Here is the commit message", no backticks.
-    - Use this exact structure:
-      1. First line: Conventional commit type + short imperative title (max 72 chars, ideally < 50)
-      2. Second line: blank
-      3. Body (optional but recommended): Clear, concise explanation of WHAT changed and WHY.
+        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:
-    feat: add user authentication flow
+        EXAMPLES OF GOOD OUTPUT:
 
-    Implemented JWT-based login with refresh tokens. Added protected routes middleware.
+        Add JWT-based login with refresh token support
 
-    fix: prevent null pointer on missing metadata
+        ## 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
 
-    Added nil check in ReportGenerator#process before accessing user preferences.
+        ## Why
+        Users were being logged out on every page reload. Refresh tokens allow
+        sessions to persist securely without requiring re-authentication.
 
-    Now generate the commit message:
+        ---
+
+        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: 'phi4:14b',
+        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 # Limit output length
       }.to_json
 
-      uri = URI('http://localhost:11434/api/generate')
+      uri = URI("http://localhost:11434/api/generate")
       request = Net::HTTP::Post.new(uri)
-      request['Content-Type'] = 'application/json'
+      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)
+      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 = 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
+      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)
@@ -81,10 +114,10 @@ module AIGit
 
       # Ensure it has at least a title
       if message.lines.count < 1 || message.strip.empty?
-        message = "chore: update code"  # fallback
+        message = "chore: update code" # fallback
       end
 
       message
-    end 
+    end
   end
 end

data/lib/ai_git/version.rb

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

data/lib/ai_git.rb

@@ -1,33 +1,38 @@
-require 'benchmark'
-require_relative 'ai_git/version'
-require_relative 'ai_git/git'
-require_relative 'ai_git/ollama'
+# frozen_string_literal: true
+
+require "benchmark"
+require_relative "ai_git/version"
+require_relative "ai_git/git"
+require_relative "ai_git/ollama"
 
 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?
+    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)
+      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}\"")
+      AIGit::Git.run_command("git", "commit -m \"#{escaped_msg}\"")
       puts "\e[1mGit Commited\e[0m"
 
-      AIGit::Git.run_command('git', 'push')
+      AIGit::Git.run_command("git", "push")
       puts "\e[1mGit Pushed\e[0m"
     end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai_git
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 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 commit + push tool using SLMs
 executables:
 - ai_git
 extensions: []
@@ -40,5 +40,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 commit + push tool using SLMs
 test_files: []