aigcm 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4a5fa862b115a855444f79940d5de35b98fe28b833d7f48974879ebc441a3f8f
-  data.tar.gz: 61773de06d7776ae2caeeb85e69a65f85c969125cf326fdb5ec69422fa138296
+  metadata.gz: 2cc53c8ffef5c597adb196deffc67ad3c0f89d5a47e9bcbfc617ece6f6a73600
+  data.tar.gz: 6262260bc17b80ef51b668e19a7014c8e9cd4429f22f2654b8682303fb79ea39
 SHA512:
-  metadata.gz: dcb6a39095d7c6eb9382c4fe6a74040aa7d0bf7b265345a57244650657dd60dd9672bc7d74bcfe3f4e5083b4dfd8a4ddce5ebef04bf205d354f4354ae0677a25
-  data.tar.gz: 7b906da5bb75a7340e69701d997e02f6a8aad8494cfe5f145126120864592693eaab91b36051829f8dd0cbb0e454689140fe8b4c94af0e48b0cffa188f514bfe
+  metadata.gz: c5b256a8478183fd3bd53188916e36f2e48639927251ea128111f28baf0dc42910ba7104f95d9a8fd98f6f2df804ea9403eb6c8c1f179b22b1f1675979ed2e15
+  data.tar.gz: 063a9e30564ad0b791ea9707e2e57c333d9372330e4784cc4858431725ac6f0184ffa6a331cdb8651d97e50fe2d0a4e363a83cea7d96e9577251c30761069055

data/lib/aigcm/commit_message_generator.rb

@@ -1,7 +1,7 @@
 require "net/http"
 require "json"
 require "open3"
-require "ai_client"
+require "ruby_llm"
 
 module Aigcm
   class CommitMessageGenerator
@@ -15,10 +15,11 @@ module Aigcm
       @max_tokens = max_tokens
       @force_external = force_external
       @amend = amend
-      validate_model_provider_combination
+      infer_provider_from_model
       check_provider_availability
 
-      @client = AiClient.new(@model, provider: @provider)
+      configure_ruby_llm
+      @chat = RubyLLM.chat(model: @model)
     rescue StandardError => e
       raise Error, "Failed to initialize AI client: #{e.message}"
     end
@@ -38,8 +39,8 @@ module Aigcm
       prompt = build_prompt(diff, style_guide, processed_context)
 
       begin
-        response = @client.chat(prompt)
-        response.to_s.strip
+        response = @chat.ask(prompt)
+        response.content.strip
       rescue StandardError => e
         "Error generating commit message: #{e.message}"
       end
@@ -47,11 +48,40 @@ module Aigcm
 
     private
 
-    def validate_model_provider_combination
-      client = AiClient.new(@model)
-      @provider ||= client.provider
-    rescue ArgumentError => e
-      raise Error, "Invalid model/provider combination: #{e.message}"
+    def infer_provider_from_model
+      return if @provider
+
+      model_lower = @model.to_s.downcase
+      @provider = if model_lower.include?('gpt') || model_lower.include?('o1') || model_lower.include?('o3')
+                    :openai
+                  elsif model_lower.include?('claude')
+                    :anthropic
+                  elsif model_lower.include?('gemini')
+                    :google
+                  elsif model_lower.include?('mistral') || model_lower.include?('mixtral')
+                    :mistral
+                  elsif model_lower.include?('llama') || model_lower.include?('qwen') || model_lower.include?('deepseek')
+                    :ollama
+                  else
+                    :openai
+                  end
+    end
+
+    def configure_ruby_llm
+      RubyLLM.configure do |config|
+        case @provider
+        when :ollama
+          config.ollama_api_base = 'http://localhost:11434/v1'
+        when :openai
+          config.openai_api_key = ENV.fetch('OPENAI_API_KEY', nil)
+        when :anthropic
+          config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY', nil)
+        when :google
+          config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil)
+        when :mistral
+          config.mistral_api_key = ENV.fetch('MISTRAL_API_KEY', nil)
+        end
+      end
     end
 
     def check_provider_availability

data/lib/aigcm/version.rb

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

data/lib/aigcm.rb

@@ -2,7 +2,7 @@ require 'debug_me'
 include DebugMe
 
 require 'optparse'
-require 'ai_client'
+require 'ruby_llm'
 require 'time'
 
 require_relative 'aigcm/version'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aigcm
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.1
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -10,19 +10,19 @@ cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: ai_client
+  name: ruby_llm
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 0.4.0
+        version: '0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 0.4.0
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: minitest
   requirement: !ruby/object:Gem::Requirement