Add Replicate provider

bin/console

@@ -23,6 +23,8 @@ RubyLLM.configure do |config|
   config.openai_api_key = ENV.fetch('OPENAI_API_KEY', nil)
   config.openrouter_api_key = ENV.fetch('OPENROUTER_API_KEY', nil)
   config.perplexity_api_key = ENV.fetch('PERPLEXITY_API_KEY', nil)
+  config.replicate_api_key = ENV.fetch('REPLICATE_API_KEY', nil)
+  config.replicate_webhook_url = ENV.fetch('REPLICATE_WEBHOOK_URL', nil)
   config.vertexai_location = ENV.fetch('GOOGLE_CLOUD_LOCATION', nil)
   config.vertexai_project_id = ENV.fetch('GOOGLE_CLOUD_PROJECT', nil)
 end

docs/_core_features/image-generation.md

@@ -119,7 +119,7 @@ image_portrait = RubyLLM.paint(
 )
 ```
 
-> Not all models support size customization. If a size is specified for a model that doesn't support it (like Google Imagen), RubyLLM may log a debug message indicating the size parameter is ignored. Check the provider's documentation or the [Available Models Guide]({% link _reference/available-models.md %}) for supported sizes.
+> Not all models support size customization. Check the provider's documentation or the [Available Models Guide]({% link _reference/available-models.md %}) for supported sizes.
 {: .note }
 
 ## Working with Generated Images
@@ -277,4 +277,4 @@ Image generation can take several seconds (typically 5-20 seconds depending on t
 
 *   [Chatting with AI Models]({% link _core_features/chat.md %}): Learn about conversational AI.
 *   [Embeddings]({% link _core_features/embeddings.md %}): Explore text vector representations.
-*   [Error Handling]({% link _advanced/error-handling.md %}): Master handling API errors.
+*   [Error Handling]({% link _advanced/error-handling.md %}): Master handling API errors.

docs/_getting_started/configuration.md

@@ -59,6 +59,8 @@ RubyLLM.configure do |config|
   config.mistral_api_key = ENV['MISTRAL_API_KEY']
   config.perplexity_api_key = ENV['PERPLEXITY_API_KEY']
   config.openrouter_api_key = ENV['OPENROUTER_API_KEY']
+  config.replicate_api_key = ENV['REPLICATE_API_KEY']
+  config.replicate_webhook_url = ENV['REPLICATE_WEBHOOK_URL']
 
   # Local providers
   config.ollama_api_base = 'http://localhost:11434/v1'
@@ -363,4 +365,4 @@ Now that you've configured RubyLLM, you're ready to:
 
 - [Start chatting with AI models]({% link _core_features/chat.md %})
 - [Work with different providers and models]({% link _advanced/models.md %})
-- [Set up Rails integration]({% link _advanced/rails.md %})
+- [Set up Rails integration]({% link _advanced/rails.md %})

lib/ruby_llm.rb

@@ -93,6 +93,7 @@ def logger
 RubyLLM::Provider.register :openrouter, RubyLLM::Providers::OpenRouter
 RubyLLM::Provider.register :perplexity, RubyLLM::Providers::Perplexity
 RubyLLM::Provider.register :vertexai, RubyLLM::Providers::VertexAI
+RubyLLM::Provider.register :replicate, RubyLLM::Providers::Replicate
 
 if defined?(Rails::Railtie)
   require 'ruby_llm/railtie'

lib/ruby_llm/configuration.rb

@@ -23,6 +23,9 @@ class Configuration
                   :gpustack_api_base,
                   :gpustack_api_key,
                   :mistral_api_key,
+                  :replicate_api_key,
+                  :replicate_webhook_url,
+                  :replicate_webhook_events_filter,
                   # Default models
                   :default_model,
                   :default_embedding_model,

lib/ruby_llm/image.rb

@@ -35,15 +35,15 @@ def self.paint(prompt, # rubocop:disable Metrics/ParameterLists
                    model: nil,
                    provider: nil,
                    assume_model_exists: false,
-                   size: '1024x1024',
-                   context: nil)
+                   context: nil,
+                   **model_params)
       config = context&.config || RubyLLM.config
       model ||= config.default_image_model
       model, provider_instance = Models.resolve(model, provider: provider, assume_exists: assume_model_exists,
                                                        config: config)
       model_id = model.id
 
-      provider_instance.paint(prompt, model: model_id, size:)
+      provider_instance.paint(prompt, model: model_id, **model_params)
     end
   end
 end