Model provider support.

Author: crmne

docs/_config.yml

@@ -40,6 +40,18 @@ color_scheme: light
 ga_tracking:
 ga_tracking_anonymize_ip: true
 
+# Callouts
+callouts:
+  new:
+    title: New
+    color: green
+  warning:
+    title: Warning
+    color: yellow
+  note:
+    title: Note
+    color: blue
+
 # Custom plugins (GitHub Pages allows these)
 plugins:
   - jekyll-remote-theme

docs/guides/chat.md

@@ -43,6 +43,33 @@ claude_chat = RubyLLM.chat(model: 'claude-3-5-sonnet-20241022')
 chat.with_model('gemini-2.0-flash')
 ```
 
+{: .warning-title }
+> Coming in v1.1.0
+>
+> The following model aliases and provider selection features are available in the upcoming version.
+
+RubyLLM supports model aliases, so you don't need to remember specific version numbers:
+
+```ruby
+# Instead of this:
+chat = RubyLLM.chat(model: 'claude-3-5-sonnet-20241022')
+
+# You can simply write:
+chat = RubyLLM.chat(model: 'claude-3-5-sonnet')
+```
+
+You can also specify a specific provider to use with a model:
+
+```ruby
+# Use a specific provider (when the same model is available from multiple providers)
+chat = RubyLLM.chat(model: 'claude-3-5-sonnet', provider: 'bedrock')
+
+# Or set the provider after initialization
+chat = RubyLLM.chat(model: 'gpt-4o')
+       .with_provider('azure')
+```
+
+See [Working with Models]({% link guides/models.md %}) for more details on model selection.
 ## System Prompts
 
 System prompts allow you to set specific instructions or context that guide the AI's behavior throughout the conversation. These prompts are not directly visible to the user but help shape the AI's responses:

docs/guides/models.md

@@ -65,6 +65,11 @@ deepseek_models = RubyLLM.models.by_provider('deepseek')
 
 ## Using Model Aliases
 
+{: .warning-title }
+> Coming in v1.1.0
+>
+> This feature is available in the upcoming version but not in the latest release.
+
 RubyLLM provides convenient aliases for popular models, so you don't have to remember specific version numbers:
 
 ```ruby
@@ -77,20 +82,16 @@ chat = RubyLLM.chat(model: 'gpt-4o')
 chat = RubyLLM.chat(model: 'gpt-4o-2024-11-20')
 ```
 
-Aliases for common models include:
+You can also specify a different provider to use with a model:
 
-| Alias | Resolves To |
-|--------|-------------|
-| `claude-3-5-sonnet` | `claude-3-5-sonnet-20241022` |
-| `claude-3-5-haiku` | `claude-3-5-haiku-20241022` |
-| `claude-3-7-sonnet` | `claude-3-7-sonnet-20250219` |
-| `claude-3-opus` | `claude-3-opus-20240229` |
-| `gpt-4o` | `gpt-4o-2024-11-20` |
-| `gpt-4o-mini` | `gpt-4o-mini-2024-07-18` |
-| `gemini-1.5-flash` | `gemini-1.5-flash-002` |
-| `gemini-2.0-flash` | `gemini-2.0-flash-001` |
+```ruby
+# Use a specific model via a different provider
+chat = RubyLLM.chat(model: 'claude-3-5-sonnet', provider: 'bedrock')
 
-Aliases are particularly useful when you want your code to always use the latest stable version of a model without having to update your codebase when providers release new model versions.
+# Or set the provider after initialization
+chat = RubyLLM.chat(model: 'gpt-4o')
+       .with_provider('azure')
+```
 
 ## Chaining Filters
 

lib/ruby_llm.rb

@@ -28,8 +28,8 @@ module RubyLLM
   class Error < StandardError; end
 
   class << self
-    def chat(model: nil)
-      Chat.new(model: model)
+    def chat(model: nil, provider: nil)
+      Chat.new(model: model, provider: provider)
     end
 
     def embed(...)

lib/ruby_llm/chat.rb

@@ -13,9 +13,10 @@ class Chat
 
     attr_reader :model, :messages, :tools
 
-    def initialize(model: nil)
+    def initialize(model: nil, provider: nil)
       model_id = model || RubyLLM.config.default_model
       self.model = model_id
+      self.provider = provider if provider
       @temperature = 0.7
       @messages = []
       @tools = {}
@@ -52,6 +53,16 @@ def model=(model_id)
       @provider = Models.provider_for model_id
     end
 
+    def provider=(provider_slug)
+      @provider = Provider.providers[provider_slug.to_sym] ||
+                  raise(Error, "Unknown provider: #{provider_slug}")
+    end
+
+    def with_provider(provider_slug)
+      self.provider = provider_slug
+      self
+    end
+
     def with_model(model_id)
       self.model = model_id
       self