Skip to content

Commit c7cdd5f

Browse files
committed
feat(Configuration): Add log_assume_model_exists option to control warning behavior
Closes #191
1 parent 2ee1d0d commit c7cdd5f

File tree

3 files changed

+9
-4
lines changed

3 files changed

+9
-4
lines changed

docs/configuration.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ RubyLLM.configure do |config|
8383
# --- Logging Settings ---
8484
config.log_file = '/logs/ruby_llm.log'
8585
config.log_level = :debug # debug level can also be set to debug by setting RUBYLLM_DEBUG envar to true
86+
config.log_assume_model_exists = false # Silence "Assuming model exists for provider" warning
8687
end
8788
```
8889

lib/ruby_llm/configuration.rb

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,8 @@ class Configuration
3737
:http_proxy,
3838
# Logging configuration
3939
:log_file,
40-
:log_level
40+
:log_level,
41+
:log_assume_model_exists
4142

4243
def initialize
4344
# Connection configuration
@@ -56,6 +57,7 @@ def initialize
5657
# Logging configuration
5758
@log_file = $stdout
5859
@log_level = ENV['RUBYLLM_DEBUG'] ? Logger::DEBUG : Logger::INFO
60+
@log_assume_model_exists = true
5961
end
6062

6163
def inspect

lib/ruby_llm/models.rb

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def fetch_from_providers
4646
end
4747
end
4848

49-
def resolve(model_id, provider: nil, assume_exists: false)
49+
def resolve(model_id, provider: nil, assume_exists: false) # rubocop:disable Metrics/PerceivedComplexity
5050
assume_exists = true if provider && Provider.providers[provider.to_sym].local?
5151

5252
if assume_exists
@@ -61,8 +61,10 @@ def resolve(model_id, provider: nil, assume_exists: false)
6161
modalities: { input: %w[text image], output: %w[text] },
6262
metadata: { warning: 'Assuming model exists, capabilities may not be accurate' }
6363
)
64-
RubyLLM.logger.warn "Assuming model '#{model_id}' exists for provider '#{provider}'. " \
65-
'Capabilities may not be accurately reflected.'
64+
if RubyLLM.config.log_assume_model_exists
65+
RubyLLM.logger.warn "Assuming model '#{model_id}' exists for provider '#{provider}'. " \
66+
'Capabilities may not be accurately reflected.'
67+
end
6668
else
6769
model = Models.find model_id, provider
6870
provider = Provider.providers[model.provider.to_sym] || raise(Error, "Unknown provider: #{model.provider}")

0 commit comments

Comments
 (0)