* use ruby_llm

Author: palkan

Gemfile

@@ -40,6 +40,7 @@ gem "nanoid"
 
 # Integrations
 gem "httparty"
+gem "ruby_llm", "~> 1.9.0"
 
 # Admin panel [https://docs.avohq.io/]
 gem "avo", ">= 3.2"

Gemfile.lock

@@ -166,13 +166,24 @@ GEM
     erubi (1.13.1)
     et-orbi (1.2.11)
       tzinfo
+    event_stream_parser (1.0.0)
     factory_bot (6.5.3)
       activesupport (>= 6.1.0)
     factory_bot_rails (6.4.4)
       factory_bot (~> 6.5)
       railties (>= 5.0.0)
     faker (3.5.1)
       i18n (>= 1.8.11, < 2)
+    faraday (2.14.0)
+      faraday-net_http (>= 2.0, < 3.5)
+      json
+      logger
+    faraday-multipart (1.1.1)
+      multipart-post (~> 2.0)
+    faraday-net_http (3.4.1)
+      net-http (>= 0.5.0)
+    faraday-retry (2.3.2)
+      faraday (~> 2.0)
     ferrum (0.15)
       addressable (~> 2.5)
       concurrent-ruby (~> 1.1)
@@ -283,8 +294,11 @@ GEM
     msgpack (1.8.0)
     multi_xml (0.7.2)
       bigdecimal (~> 3.1)
+    multipart-post (2.4.1)
     mutex_m (0.3.0)
     nanoid (2.0.0)
+    net-http (0.6.0)
+      uri
     net-imap (0.5.11)
       date
       net-protocol
@@ -423,6 +437,17 @@ GEM
       rubocop (~> 1.72, >= 1.72.1)
     ruby-next-core (1.1.2)
     ruby-progressbar (1.13.0)
+    ruby_llm (1.9.0)
+      base64
+      event_stream_parser (~> 1)
+      faraday (>= 1.10.0)
+      faraday-multipart (>= 1)
+      faraday-net_http (>= 1)
+      faraday-retry (>= 1)
+      marcel (~> 1.0)
+      ruby_llm-schema (~> 0.2.1)
+      zeitwerk (~> 2)
+    ruby_llm-schema (0.2.1)
     securerandom (0.4.1)
     sentry-rails (5.23.0)
       railties (>= 5.0)
@@ -535,6 +560,7 @@ DEPENDENCIES
   rails (= 8.1.0.rc1)
   rspec-rails
   rubocop-rspec
+  ruby_llm (~> 1.9.0)
   sentry-rails (~> 5.15)
   sentry-ruby (~> 5.15)
   solid_queue

app/models/cloud/card_generator.rb

@@ -1,82 +1,35 @@
 class Cloud
   class CardGenerator
-    FLASH_IMAGE_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-image:generateContent"
-
     private attr_reader :cloud, :api_key
 
     delegate :participant, to: :cloud
 
-    def initialize(cloud, api_key: GeminiConfig.api_key)
+    def initialize(cloud)
       @cloud = cloud
-      @api_key = api_key
     end
 
     def generate
-      raise ArgumentError, "Gemini API key not configured" if api_key.blank?
       raise ArgumentError, "No photo attached" unless cloud.image.attached?
 
-      parts = [
-        {
-          text: build_prompt
-        },
-        build_inline_data(cloud.image.blob.download, cloud.image.content_type),
-        build_inline_data(Rails.public_path.join("sfruby_character.png").read, "image/png"),
-        build_inline_data(Rails.public_path.join("sfruby_character_h.png").read, "image/png"),
-        build_inline_data(Rails.public_path.join("sfruby_character_blue.png").read, "image/png")
-      ]
-
-      request = {
-        contents: [
-          {
-            parts:
-          }
-        ],
-        generationConfig: {
-          responseModalities: ["image"],
-          temperature: 1.0
-        }
-      }
-
-      response = HTTParty.post(
-        "#{FLASH_IMAGE_URL}?key=#{api_key}",
-        body: request.to_json,
-        headers: {"Content-Type" => "application/json"},
-        timeout: 90
-      )
-
-      if response.success?
-        process_response(response.parsed_response)
-      else
-        raise "Gemini API error: #{response.code} - #{response.body}"
-      end
-    end
+      chat = RubyLLM.chat(model: "gemini-2.5-flash-image")
+        .with_temperature(1.0)
+        .with_params(generationConfig: {responseModalities: ["image"]})
 
-    private
+      response = chat.ask build_prompt, with: build_attachments
+      raise "LLM error: #{response.raw.body.dig("candidates", 0, "finishMessage") || "no data"}" if response.content.blank?
 
-    def build_inline_data(io, content_type)
-      {
-        inlineData: {
-          mimeType: content_type,
-          data: Base64.strict_encode64(io)
-        }
-      }
+      response.content[:attachments].first.source
     end
 
-    def process_response(response)
-      parts = response.dig("candidates", 0, "content", "parts")
-
-      if parts.nil?
-        raise "No parts found in response"
-      end
-
-      image_part = parts.find { it.dig("inlineData", "mimeType")&.start_with?("image/") }
-
-      raise "No image data found in parts. Parts structure: #{parts.map(&:keys)}" unless image_part
-
-      image_data = image_part["inlineData"]["data"]
-      decoded_image = Base64.decode64(image_data)
+    private
 
-      StringIO.new(decoded_image)
+    def build_attachments
+      [
+        cloud.image.blob,
+        Rails.public_path.join("sfruby_character.png"),
+        Rails.public_path.join("sfruby_character_h.png"),
+        Rails.public_path.join("sfruby_character_blue.png")
+      ]
     end
 
     def build_prompt

config/initializers/ruby_llm.rb

@@ -0,0 +1,4 @@
+RubyLLM.configure do |config|
+  config.gemini_api_key = GeminiConfig.api_key
+  config.log_level = Logger::DEBUG if ENV["DEBUG_LLM"].in?(%w[1 t true])
+end