Add tests

Author: josefarias

lib/ruby_llm/aliases.json

@@ -1,4 +1,49 @@
 {
+  "adirik/realvisxl-v3.0-turbo": {
+    "replicate": "3dc73c805b11b4b01a60555e532fd3ab3f0e60d26f6584d9b8ba7e1b95858243"
+  },
+  "ai-forever/kandinsky-2": {
+    "replicate": "3c6374e7a9a17e01afe306a5218cc67de55b19ea536466d6ea2602cfecea40a9"
+  },
+  "ai-forever/kandinsky-2.2": {
+    "replicate": "ad9d7879fbffa2874e1d909d1d37d9bc682889cc65b31f7bb00d2362619f194a"
+  },
+  "black-forest-labs/flux-1.1-pro": {
+    "replicate": "80a09d66baa990429c2f5ae8a4306bf778a1b3775afd01cc2cc8bdbe9033769c"
+  },
+  "black-forest-labs/flux-1.1-pro-ultra": {
+    "replicate": "c6e5086a542c99e7e523a83d3017654e8618fe64ef427c772a1def05bb599f0c"
+  },
+  "black-forest-labs/flux-dev": {
+    "replicate": "6e4a938f85952bdabcc15aa329178c4d681c52bf25a0342403287dc26944661d"
+  },
+  "black-forest-labs/flux-dev-lora": {
+    "replicate": "ae0d7d645446924cf1871e3ca8796e8318f72465d2b5af9323a835df93bf0917"
+  },
+  "black-forest-labs/flux-kontext-max": {
+    "replicate": "f72e27297d9c05a36b7fd8faff393d31e3b368543e0bc44bde521886700e166c"
+  },
+  "black-forest-labs/flux-kontext-pro": {
+    "replicate": "2dfe45debca13e5ecfad755ef6ca9943fc56a6effb306f4c6e2ea4762df6e53e"
+  },
+  "black-forest-labs/flux-pro": {
+    "replicate": "1e237aa703bf3a8ab480d5b595563128807af649c50afc0b4f22a9174e90d1d6"
+  },
+  "black-forest-labs/flux-schnell": {
+    "replicate": "c846a69991daf4c0e5d016514849d14ee5b2e6846ce6b9d6f21369e564cfe51e"
+  },
+  "bria/image-3.2": {
+    "replicate": "143afb950d3186f914f24b9eab35c297b3bbb0ddf9d77fb8ad47784e0a66053b"
+  },
+  "bytedance/sdxl-lightning-4step": {
+    "replicate": "6f7a773af6fc3e8de9d5a3c00be77c17308914bf67772726aff83496ba1e3bbe"
+  },
+  "bytedance/seedream-3": {
+    "replicate": "8eb3d789024549c43fb53ba6ad21450e4ee4e1ff87951fabf619b6b7a6a436ec"
+  },
+  "bytedance/seedream-4": {
+    "replicate": "254faac883c3a411e95cc95d0fb02274a81e388aaa4394b3ce5b7d2a9f7a6569"
+  },
   "chatgpt-4o": {
     "openai": "chatgpt-4o-latest",
     "openrouter": "openai/chatgpt-4o-latest"
@@ -46,10 +91,37 @@
     "openrouter": "anthropic/claude-sonnet-4",
     "bedrock": "us.anthropic.claude-sonnet-4-20250514-v1:0"
   },
+  "datacte/proteus-v0.2": {
+    "replicate": "06775cd262843edbde5abab958abdbb65a0a6b58ca301c9fd78fa55c775fc019"
+  },
+  "datacte/proteus-v0.3": {
+    "replicate": "b28b79d725c8548b173b6a19ff9bffd16b9b80df5b18b8dc5cb9e1ee471bfa48"
+  },
   "deepseek-chat": {
     "deepseek": "deepseek-chat",
     "openrouter": "deepseek/deepseek-chat"
   },
+  "fermatresearch/sdxl-controlnet-lora": {
+    "replicate": "3bb13fe1c33c35987b33792b01b71ed6529d03f165d1c2416375859f09ca9fef"
+  },
+  "fofr/any-comfyui-workflow": {
+    "replicate": "67ed4ba04ce0842446e16c428b1be131452815d01810861f71d171f63e8ba8f0"
+  },
+  "fofr/latent-consistency-model": {
+    "replicate": "683d19dc312f7a9f0428b04429a9ccefd28dbf7785fef083ad5cf991b65f406f"
+  },
+  "fofr/realvisxl-v3-multi-controlnet-lora": {
+    "replicate": "90a4a3604cd637cb9f1a2bdae1cfa9ed869362ca028814cdce310a78e27daade"
+  },
+  "fofr/sdxl-emoji": {
+    "replicate": "dee76b5afde21b0f01ed7925f0665b7e879c50ee718c5f78a9d38e04d523cc5e"
+  },
+  "fofr/sdxl-multi-controlnet-lora": {
+    "replicate": "89eb212b3d1366a83e949c12a4b45dfe6b6b313b594cb8268e864931ac9ffb16"
+  },
+  "fofr/sticker-maker": {
+    "replicate": "4acb778eb059772225ec213948f0660867b2e03f277448f18cf1800b96a65a1a"
+  },
   "gemini-1.5-flash": {
     "gemini": "gemini-1.5-flash",
     "vertexai": "gemini-1.5-flash"
@@ -139,6 +211,24 @@
     "gemini": "gemma-3n-e4b-it",
     "openrouter": "google/gemma-3n-e4b-it"
   },
+  "google/imagen-3": {
+    "replicate": "3e0b477743ff405dc5b77dd73456d7ef96705a9aad479c668d57743ce1d4ca96"
+  },
+  "google/imagen-3-fast": {
+    "replicate": "5125f62df7e6d1ccb0ce5ac60f141646486c0cb74a2e3d9e8b63ed486c48313e"
+  },
+  "google/imagen-4": {
+    "replicate": "1ddba0fad36734cd24c85ecc783d67c8973da8ea05044b56db4eb3b0cde5394d"
+  },
+  "google/imagen-4-fast": {
+    "replicate": "4e5534950275f5d29aa1ea263a442018f34d90254bb89ba9a08de44bc68c5a33"
+  },
+  "google/imagen-4-ultra": {
+    "replicate": "33db85c7f42cdfee3a7658d76a9097b8d856981aba0020b323d3e83683d2e2a2"
+  },
+  "google/nano-banana": {
+    "replicate": "1b7b945e8f7edf7a034eba6cb2c20f2ab5dc7d090eea1c616e96da947be76aee"
+  },
   "gpt-3.5-turbo": {
     "openai": "gpt-3.5-turbo",
     "openrouter": "openai/gpt-3.5-turbo"
@@ -227,6 +317,60 @@
     "openai": "gpt-5-nano",
     "openrouter": "openai/gpt-5-nano"
   },
+  "ideogram-ai/ideogram-v2": {
+    "replicate": "39ec2c0b3642ba5bc9357978a8fd060c7e363643c192776e38dd9b2b2c403442"
+  },
+  "ideogram-ai/ideogram-v2-turbo": {
+    "replicate": "35eacd3dbd088d6421f7ee27646701b5e03ec5a9a0f68f43112fa228d6fc2522"
+  },
+  "ideogram-ai/ideogram-v2a": {
+    "replicate": "1751e040eb5e766b3eea4f9e69160987e1854c28a87a08882c930ecc6cef0305"
+  },
+  "ideogram-ai/ideogram-v2a-turbo": {
+    "replicate": "8b8447f69a32c73b5dc94c1aca2909b7aa702940281496d4ea2c9b260725a81a"
+  },
+  "ideogram-ai/ideogram-v3-balanced": {
+    "replicate": "37b6362ab309164a8b5528a8b384d3788a9465d1095fa7cc3cb6fbc9e65b8702"
+  },
+  "ideogram-ai/ideogram-v3-quality": {
+    "replicate": "c3b86e5c2118ee3469fd3fead8a90dc72cf23ea31e24eed8f8c359ebe7e546ba"
+  },
+  "ideogram-ai/ideogram-v3-turbo": {
+    "replicate": "c238db25556f295de040a59b462cc212bb8ceb282bf2fc6bee6fbaf4ab014f23"
+  },
+  "jagilley/controlnet-scribble": {
+    "replicate": "435061a1b5a4c1e26740464bf786efdfa9cb3a3ac488595a2de23e143fdb0117"
+  },
+  "leonardoai/lucid-origin": {
+    "replicate": "97190d506c4a659bc58c9c8deaa96415592c7fdd42d4b1b624649c6211d83f1c"
+  },
+  "lucataco/dreamshaper-xl-turbo": {
+    "replicate": "0a1710e0187b01a255302738ca0158ff02a22f4638679533e111082f9dd1b615"
+  },
+  "lucataco/open-dalle-v1.1": {
+    "replicate": "1c7d4c8dec39c7306df7794b28419078cb9d18b9213ab1c21fdc46a1deca0144"
+  },
+  "lucataco/realistic-vision-v5.1": {
+    "replicate": "2c8e954decbf70b7607a4414e5785ef9e4de4b8c51d50fb8b8b349160e0ef6bb"
+  },
+  "lucataco/ssd-1b": {
+    "replicate": "b19e3639452c59ce8295b82aba70a231404cb062f2eb580ea894b31e8ce5bbb6"
+  },
+  "luma/photon": {
+    "replicate": "fe142c037cf359764f2058c3a42ef0dc750d908311d02868cbc7769fe800b648"
+  },
+  "luma/photon-flash": {
+    "replicate": "48705cf0e6223fbc49e81781ba235623c100713035f9866e04a6236b54e1a8d3"
+  },
+  "minimax/image-01": {
+    "replicate": "47ca89ad46682c1dd0ca335601cd7ea2eb10fb94ce4e0a5abafa7e74f23ae7b6"
+  },
+  "nvidia/sana": {
+    "replicate": "c6b5d2b7459910fec94432e9e1203c3cdce92d6db20f714f1355747990b52fa6"
+  },
+  "nvidia/sana-sprint-1.6b": {
+    "replicate": "038aee6907b53a5c148780983e39a50ce7cd0747b4e2642e78387f48cf36039a"
+  },
   "o1": {
     "openai": "o1",
     "openrouter": "openai/o1"
@@ -259,8 +403,56 @@
     "openai": "o4-mini",
     "openrouter": "openai/o4-mini"
   },
+  "playgroundai/playground-v2.5-1024px-aesthetic": {
+    "replicate": "a45f82a1382bed5c7aeb861dac7c7d191b0fdf74d8d57c4a0e6ed7d4d0bf7d24"
+  },
+  "prunaai/flux.1-dev": {
+    "replicate": "b0306d92aa025bb747dc74162f3c27d6ed83798e08e5f8977adf3d859d0536a3"
+  },
+  "prunaai/hidream-l1-dev": {
+    "replicate": "4dfcd146c0def4812455415f55556f6bc84025dcb15193cf1977f01bd384d191"
+  },
+  "prunaai/hidream-l1-fast": {
+    "replicate": "06898b39cb00e42d31666b0dc8b9904f326169768129d756184f65ecf1986c8f"
+  },
+  "prunaai/hidream-l1-full": {
+    "replicate": "03d58532fd29e39fd2ed80e86c3da1cebec28ef2734081cf1366710d30388f42"
+  },
+  "prunaai/sdxl-lightning": {
+    "replicate": "123d4264aec4d1c81d8bf142e0d90d6fd4c346f1a869406019f6d1653882d7c3"
+  },
+  "prunaai/wan-2.2-image": {
+    "replicate": "16e15e913fcc71c1a5defb335ea84739f99731fa1ee17995117c7d9adc6d176c"
+  },
+  "qwen/qwen-image": {
+    "replicate": "905e345fe1dfe10d628daac2140dd8dea471c0d99793ef0fdc46a15c688b62fb"
+  },
+  "recraft-ai/recraft-v3": {
+    "replicate": "0fea59248a8a1ddb8197792577f6627ec65482abc49f50c6e9da40ca8729d24d"
+  },
+  "recraft-ai/recraft-v3-svg": {
+    "replicate": "81deed12bf1c7e762e46f53df3f85b4369abf876b1df1777ce795e0d7d5849e4"
+  },
+  "stability-ai/sdxl": {
+    "replicate": "7762fd07cf82c948538e41f63f77d685e02b063e37e496e96eefd46c929f9bdc"
+  },
+  "stability-ai/stable-diffusion": {
+    "replicate": "ac732df83cea7fff18b8472768c88ad041fa750ff7682a21affe81863cbe77e4"
+  },
+  "stability-ai/stable-diffusion-3.5-large": {
+    "replicate": "feb36d70a9f2791ed9df99f795ea5684da18f52b5f6b488dfdef635ca19d6bd3"
+  },
+  "stability-ai/stable-diffusion-3.5-large-turbo": {
+    "replicate": "8170d81b43c5b823dd15b0132720fdae8dadf9c46c7da484a24b8e98e6b161a2"
+  },
+  "stability-ai/stable-diffusion-3.5-medium": {
+    "replicate": "e069811fe9b5815a6db8874c3054c15be99c8019d07ee5ad72aa63cea3cfffe3"
+  },
   "text-embedding-004": {
     "gemini": "text-embedding-004",
     "vertexai": "text-embedding-004"
+  },
+  "tstramer/material-diffusion": {
+    "replicate": "a42692c54c0f407f803a0a8a9066160976baedb77c91171a01730f9b0d7beeff"
   }
 }

lib/ruby_llm/providers/replicate/images.rb

@@ -20,7 +20,7 @@ def render_image_payload(prompt, model:, **params)
 
           {}.tap do |payload|
             payload[:webhook] = @config.replicate_webhook_url
-            payload[:version] = model.id unless official_model?
+            payload[:version] = self.model.id unless official_model?
             payload[:input] = { prompt: prompt }.merge(params)
 
             if @config.replicate_webhook_events_filter
@@ -30,18 +30,18 @@ def render_image_payload(prompt, model:, **params)
         end
 
         def parse_image_response(response, **)
-          response
+          response.body
         end
 
         private
 
         def model_id=(id)
           @model_id = id
-          @model = Models.find(@model_id, 'replicate')
+          @model = RubyLLM::Models.find(@model_id, 'replicate')
         end
 
         def official_model?
-          model.metadata['is_official'] == true
+          model.metadata[:is_official] == true
         end
       end
     end

lib/tasks/models.rake

@@ -429,6 +429,10 @@ def generate_aliases # rubocop:disable Metrics/PerceivedComplexity
     }
   end
 
+  models['replicate'].each do |model|
+    aliases[RubyLLM.models.find(model).name] = { 'replicate' => model }
+  end
+
   sorted_aliases = aliases.sort.to_h
   File.write(RubyLLM::Aliases.aliases_file, JSON.pretty_generate(sorted_aliases))