Add xAI Provider

bin/console

@@ -25,6 +25,7 @@ RubyLLM.configure do |config|
   config.perplexity_api_key = ENV.fetch('PERPLEXITY_API_KEY', nil)
   config.vertexai_location = ENV.fetch('GOOGLE_CLOUD_LOCATION', nil)
   config.vertexai_project_id = ENV.fetch('GOOGLE_CLOUD_PROJECT', nil)
+  config.xai_api_key = ENV.fetch('XAI_API_KEY', nil)
 end
 
 IRB.start(__FILE__)

docs/_getting_started/configuration.md

@@ -59,6 +59,7 @@ 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.xai_api_key = ENV['XAI_API_KEY']
 
   # Local providers
   config.ollama_api_base = 'http://localhost:11434/v1'

lib/ruby_llm.rb

@@ -23,7 +23,8 @@
   'mistral' => 'Mistral',
   'vertexai' => 'VertexAI',
   'pdf' => 'PDF',
-  'UI' => 'UI'
+  'UI' => 'UI',
+  'xai' => 'XAI'
 )
 loader.ignore("#{__dir__}/tasks")
 loader.ignore("#{__dir__}/generators")
@@ -93,6 +94,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 :xai, RubyLLM::Providers::XAI
 
 if defined?(Rails::Railtie)
   require 'ruby_llm/railtie'

lib/ruby_llm/aliases.json

@@ -106,6 +106,14 @@
     "gemini": "gemini-2.5-flash-lite-preview-06-17",
     "openrouter": "google/gemini-2.5-flash-lite-preview-06-17"
   },
+  "gemini-2.5-flash-lite-preview-09-2025": {
+    "gemini": "gemini-2.5-flash-lite-preview-09-2025",
+    "openrouter": "google/gemini-2.5-flash-lite-preview-09-2025"
+  },
+  "gemini-2.5-flash-preview-09-2025": {
+    "gemini": "gemini-2.5-flash-preview-09-2025",
+    "openrouter": "google/gemini-2.5-flash-preview-09-2025"
+  },
   "gemini-2.5-pro": {
     "gemini": "gemini-2.5-pro",
     "openrouter": "google/gemini-2.5-pro",
@@ -115,10 +123,6 @@
     "gemini": "gemini-2.5-pro-preview-05-06",
     "openrouter": "google/gemini-2.5-pro-preview-05-06"
   },
-  "gemini-embedding-001": {
-    "gemini": "gemini-embedding-001",
-    "vertexai": "gemini-embedding-001"
-  },
   "gemini-exp-1206": {
     "gemini": "gemini-exp-1206",
     "vertexai": "gemini-exp-1206"
@@ -219,6 +223,10 @@
     "openai": "gpt-5",
     "openrouter": "openai/gpt-5"
   },
+  "gpt-5-codex": {
+    "openai": "gpt-5-codex",
+    "openrouter": "openai/gpt-5-codex"
+  },
   "gpt-5-mini": {
     "openai": "gpt-5-mini",
     "openrouter": "openai/gpt-5-mini"
@@ -227,6 +235,95 @@
     "openai": "gpt-5-nano",
     "openrouter": "openai/gpt-5-nano"
   },
+  "gpt-oss-120b": {
+    "openai": "gpt-oss-120b",
+    "openrouter": "openai/gpt-oss-120b"
+  },
+  "gpt-oss-20b": {
+    "openai": "gpt-oss-20b",
+    "openrouter": "openai/gpt-oss-20b"
+  },
+  "grok-2": {
+    "xai": "grok-2-1212"
+  },
+  "grok-2-image": {
+    "xai": "grok-2-image-1212"
+  },
+  "grok-2-image-latest": {
+    "xai": "grok-2-image-1212"
+  },
+  "grok-2-latest": {
+    "xai": "grok-2-1212"
+  },
+  "grok-2-vision": {
+    "xai": "grok-2-vision-1212"
+  },
+  "grok-2-vision-latest": {
+    "xai": "grok-2-vision-1212"
+  },
+  "grok-3": {
+    "xai": "grok-3",
+    "openrouter": "x-ai/grok-3"
+  },
+  "grok-3-beta": {
+    "xai": "grok-3"
+  },
+  "grok-3-fast": {
+    "xai": "grok-3"
+  },
+  "grok-3-fast-beta": {
+    "xai": "grok-3"
+  },
+  "grok-3-fast-latest": {
+    "xai": "grok-3"
+  },
+  "grok-3-latest": {
+    "xai": "grok-3"
+  },
+  "grok-3-mini": {
+    "xai": "grok-3-mini",
+    "openrouter": "x-ai/grok-3-mini"
+  },
+  "grok-3-mini-beta": {
+    "xai": "grok-3-mini"
+  },
+  "grok-3-mini-fast": {
+    "xai": "grok-3-mini"
+  },
+  "grok-3-mini-fast-beta": {
+    "xai": "grok-3-mini"
+  },
+  "grok-3-mini-fast-latest": {
+    "xai": "grok-3-mini"
+  },
+  "grok-3-mini-latest": {
+    "xai": "grok-3-mini"
+  },
+  "grok-4": {
+    "xai": "grok-4-0709"
+  },
+  "grok-4-fast": {
+    "xai": "grok-4-fast-reasoning"
+  },
+  "grok-4-fast-non-reasoning-latest": {
+    "xai": "grok-4-fast-non-reasoning"
+  },
+  "grok-4-fast-reasoning-latest": {
+    "xai": "grok-4-fast-reasoning"
+  },
+  "grok-4-latest": {
+    "xai": "grok-4-0709"
+  },
+  "grok-code-fast": {
+    "xai": "grok-code-fast-1"
+  },
+  "grok-code-fast-1": {
+    "xai": "grok-code-fast-1",
+    "openrouter": "x-ai/grok-code-fast-1"
+  },
+  "grok-code-fast-1-0825": {
+    "xai": "grok-code-fast-1"
+  },
   "o1": {
     "openai": "o1",
     "openrouter": "openai/o1"

lib/ruby_llm/configuration.rb

@@ -14,6 +14,7 @@ class Configuration
                   :vertexai_location,
                   :deepseek_api_key,
                   :perplexity_api_key,
+                  :xai_api_key,
                   :bedrock_api_key,
                   :bedrock_secret_key,
                   :bedrock_region,