ai_git 0.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 7880625a0fd36e0dd667168f8bfe2dbaffc887584d6777f6675eef90202a5889
+  data.tar.gz: 17221f5e7db837d7aa155b1a19f36f2065f584ee40c45b098d8f5cf6041b9690
+SHA512:
+  metadata.gz: 24d52e06e1b4d88c63d62fa79cfa79f9feb1ea53ea13437710952a833553eba10f4db1b71ed864ff6d3935d2082e769b34fb2f651d1256827774f2be411fdcf2
+  data.tar.gz: 5e2c67e5155fb89160ac2beb0c54d5e93d9b8692be33de686589ee3977cd2505b57532c69683684bc9a053dcff4ad1fcbd860afd181b1441aa07b95e36ee764a

data/bin/ai_git

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require_relative "../lib/ai_git"
+
+AIGit.run

data/lib/ai_git/git.rb

@@ -0,0 +1,23 @@
+module AIGit
+  module Git
+    module_function
+
+    def staged_files
+      `git diff --cached --name-only`
+    end
+
+    def diff
+      `git diff --cached`
+    end
+
+    def current_branch
+      result = `git rev-parse --abbrev-ref HEAD`
+      result.chomp
+    end
+
+    def run_command(cmd, args)
+      system("#{cmd} #{args} > /dev/null 2>&1")
+      raise "Command failed: #{cmd} #{args}" if $?.exitstatus != 0
+    end
+  end
+end

data/lib/ai_git/ollama.rb

@@ -0,0 +1,91 @@
+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)
+      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.
+
+    Here are the changes:
+    #{diff}
+
+    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.
+
+    Examples of good output:
+    feat: add user authentication flow
+
+    Implemented JWT-based login with refresh tokens. Added protected routes middleware.
+
+    fix: prevent null pointer on missing metadata
+
+    Added nil check in ReportGenerator#process before accessing user preferences.
+
+    Now generate the commit message:
+      PROMPT
+
+      json_body = {
+        # model: 'qwen:14b',
+        model: 'ministral-3:8b',
+        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

data/lib/ai_git/version.rb

@@ -0,0 +1,3 @@
+module AIGit
+  VERSION = "0.0.0"
+end

data/lib/ai_git.rb

@@ -0,0 +1,37 @@
+require 'benchmark'
+require_relative 'ai_git/version'
+require_relative 'ai_git/git'
+require_relative 'ai_git/ollama'
+
+module AIGit
+  module_function
+
+  def run
+    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[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 = 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

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: ai_git
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+platform: ruby
+authors:
+- Kaíque Kandy Koga
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies: []
+description: Generates intelligent commit messages using Ollama LLMs
+executables:
+- ai_git
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/ai_git
+- lib/ai_git.rb
+- lib/ai_git/git.rb
+- lib/ai_git/ollama.rb
+- lib/ai_git/version.rb
+homepage: https://github.com/koga/ai_git
+licenses:
+- BSD-3-Clause
+metadata: {}
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '4.0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 4.0.9
+specification_version: 4
+summary: AI-powered git commit and push tool using Ollama
+test_files: []