Merge pull request #16 from SergoHUH/chore/update-ruby-llm

chore(deps): update ruby_llm to 1.16.0

Author: DenisDenis9331

Gemfile

@@ -5,7 +5,7 @@ ruby '3.1.3'
 
 gem 'dotenv', '~> 3.1'
 gem 'faraday', '~> 2.14'
-gem 'ruby_llm', '1.15.0'
+gem 'ruby_llm', '1.16.0'
 
 group :development, :test do
   gem 'rspec', '~> 3.13'

Gemfile.lock

@@ -60,7 +60,7 @@ GEM
       parser (>= 3.3.7.2)
       prism (~> 1.7)
     ruby-progressbar (1.13.0)
-    ruby_llm (1.15.0)
+    ruby_llm (1.16.0)
       base64
       event_stream_parser (~> 1)
       faraday (>= 1.10.0)
@@ -85,7 +85,7 @@ DEPENDENCIES
   faraday (~> 2.14)
   rspec (~> 3.13)
   rubocop (~> 1.75)
-  ruby_llm (= 1.15.0)
+  ruby_llm (= 1.16.0)
 
 RUBY VERSION
    ruby 3.1.3p185

lib/aireview/reviewer.rb

@@ -77,13 +77,25 @@ def perform_llm_request(context:, stage:, system:, user:, options:)
       model = options[:model]
       temperature = options[:temperature]
       @logger.info("LLM #{stage} request started (model=#{model}, temperature=#{temperature})")
-      chat = context.chat(model: model, provider: options[:provider].to_sym).with_temperature(temperature.to_f)
+      chat = build_chat(context: context, stage: stage, model: model, provider: options[:provider])
+        .with_temperature(temperature.to_f)
       chat.with_instructions(system)
       response = Timeout.timeout(@config.llm_timeout.to_f) { chat.ask(user) }
       @logger.info("LLM #{stage} request completed (model=#{model})")
       response
     end
 
+    def build_chat(context:, stage:, model:, provider:)
+      context.chat(model: model, provider: provider.to_sym)
+    rescue RubyLLM::ModelNotFoundError
+      @logger.warn(
+        "LLM #{stage}: model not found in RubyLLM registry; " \
+        "using fallback with incomplete model metadata " \
+        "(model=#{model}, provider=#{provider})"
+      )
+      context.chat(model: model, provider: provider.to_sym, assume_model_exists: true)
+    end
+
     def request_with_retries(stage:, model:)
       attempt = 0
 

spec/reviewer_spec.rb

@@ -70,6 +70,7 @@
     stub_const('RubyLLM::OverloadedError', Class.new(RubyLLM::Error))
     stub_const('RubyLLM::RateLimitError', Class.new(RubyLLM::Error))
     stub_const('RubyLLM::ContextLengthExceededError', Class.new(RubyLLM::Error))
+    stub_const('RubyLLM::ModelNotFoundError', Class.new(StandardError))
     RubyLLM.singleton_class.attr_accessor :config
     RubyLLM.config = global_ruby_config
 
@@ -83,6 +84,12 @@
       allow(context).to receive(:chat)
         .with(model: 'gemini-2.5-flash-lite', provider: anything)
         .and_return(critique_chat)
+      allow(context).to receive(:chat)
+        .with(model: 'gemini-3.6-flash', provider: anything)
+        .and_raise(RubyLLM::ModelNotFoundError)
+      allow(context).to receive(:chat)
+        .with(model: 'gemini-3.6-flash', provider: anything, assume_model_exists: true)
+        .and_return(generate_chat)
       context_configs << context_config
       contexts << context
       context
@@ -107,12 +114,14 @@
     expect(context_configs.first.gemini_api_base).to eq('https://llm.example.test')
   end
 
-  it 'uses generate model and temperature for the generate pass' do
+  it 'uses a known model without bypassing the registry' do
     reviewer = described_class.new(config: config, logger: logger)
     result = reviewer.generate(system_prompt: 'system prompt', user_prompt: 'user prompt')
 
     expect(result).to eq('generate body')
-    expect(contexts.first).to have_received(:chat).with(model: 'gemini-2.5-pro', provider: :gemini)
+    expect(contexts.first)
+      .to have_received(:chat)
+      .with(model: 'gemini-2.5-pro', provider: :gemini)
     expect(generate_chat).to have_received(:with_temperature).with(0.3)
   end
 
@@ -121,7 +130,9 @@
     result = reviewer.critique(system_prompt: 'system prompt', user_prompt: 'user prompt')
 
     expect(result).to eq('critique body')
-    expect(contexts.first).to have_received(:chat).with(model: 'gemini-2.5-flash-lite', provider: :gemini)
+    expect(contexts.first)
+      .to have_received(:chat)
+      .with(model: 'gemini-2.5-flash-lite', provider: :gemini)
     expect(critique_chat).to have_received(:with_temperature).with(0.0)
   end
 
@@ -135,6 +146,26 @@
     expect(context_configs.first).not_to equal(context_configs.last)
   end
 
+  it 'retries an unknown model with an explicit provider' do
+    allow(config).to receive(:generate_model).and_return('gemini-3.6-flash')
+    log_output = StringIO.new
+    reviewer = described_class.new(config: config, logger: Logger.new(log_output))
+
+    reviewer.generate(system_prompt: 'system prompt', user_prompt: 'user prompt')
+
+    expect(contexts.first)
+      .to have_received(:chat)
+      .with(model: 'gemini-3.6-flash', provider: :gemini)
+    expect(contexts.first)
+      .to have_received(:chat)
+      .with(model: 'gemini-3.6-flash', provider: :gemini, assume_model_exists: true)
+    expect(log_output.string).to include(
+      'LLM generate: model not found in RubyLLM registry; ' \
+      'using fallback with incomplete model metadata ' \
+      '(model=gemini-3.6-flash, provider=gemini)'
+    )
+  end
+
   it 'sets the RubyLLM request timeout from LLM_TIMEOUT' do
     reviewer = described_class.new(config: config, logger: logger)