FIX: Cloudflare Workers AI embeddings

Regressed on 534b0df
discourse · Dec 20, 2024 · b6ea353 · b6ea353
1 parent 6a7a45f
commit b6ea353
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/inference/cloudflare_workers_ai.rb b/lib/inference/cloudflare_workers_ai.rb
@@ -27,10 +27,12 @@ def perform!(content)
           "Authorization" => "Bearer #{api_token}",
         }
 
+        payload = { text: [content] }
+
         endpoint = "https://api.cloudflare.com/client/v4/accounts/#{account_id}/ai/run/@cf/#{model}"
 
         conn = Faraday.new { |f| f.adapter FinalDestination::FaradayAdapter }
-        response = conn.post(endpoint, content.to_json, headers)
+        response = conn.post(endpoint, payload.to_json, headers)
 
         case response.status
         when 200