diff --git a/bin/console b/bin/console index f32609139..458a8436e 100755 --- a/bin/console +++ b/bin/console @@ -15,6 +15,7 @@ RubyLLM.configure do |config| config.bedrock_session_token = ENV.fetch('AWS_SESSION_TOKEN', nil) config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil) config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil) + config.gemini_api_base = ENV.fetch('GEMINI_API_BASE', nil) config.gpustack_api_base = ENV.fetch('GPUSTACK_API_BASE', nil) config.gpustack_api_key = ENV.fetch('GPUSTACK_API_KEY', nil) config.mistral_api_key = ENV.fetch('MISTRAL_API_KEY', nil) diff --git a/lib/ruby_llm/configuration.rb b/lib/ruby_llm/configuration.rb index eda2c3354..2593c3fdc 100644 --- a/lib/ruby_llm/configuration.rb +++ b/lib/ruby_llm/configuration.rb @@ -10,6 +10,7 @@ class Configuration :openai_use_system_role, :anthropic_api_key, :gemini_api_key, + :gemini_api_base, :vertexai_project_id, :vertexai_location, :deepseek_api_key, diff --git a/lib/ruby_llm/providers/gemini.rb b/lib/ruby_llm/providers/gemini.rb index 30e90b449..c80609da4 100644 --- a/lib/ruby_llm/providers/gemini.rb +++ b/lib/ruby_llm/providers/gemini.rb @@ -13,7 +13,7 @@ class Gemini < Provider include Gemini::Media def api_base - 'https://generativelanguage.googleapis.com/v1beta' + @config.gemini_api_base || 'https://generativelanguage.googleapis.com/v1beta' end def headers