Fix FAL.AI models list endpoint (#4236) The FAL.AI endpoint seems to have switched to a paginated format where the results are nested under an `items` key. This updates the endpoint handler to fetch all pages and extract the model options correctly.

b45477578ffd3c135a9a4745fa3cf38af0db76f1

Jenner La Fave <jenner@jfave.com>

Signed

Parents: e1a55408b60e068b648f2d0c53992aa26e357a75

src/endpoints/stable-diffusion.js+32 -16 Viewed

 falai.post('/models', async (_request, response) => {
     try {
         const modelsUrl = new URL('https://fal.ai/api/models?categories=text-to-image');
-        const result = await fetch(modelsUrl);
+        let page = 1;
+        let modelsResponse;
+        let models = [];
-        if (!result.ok) {
+        do {
-            console.warn('FAL.AI returned an error.', result.status, result.statusText);
+            modelsUrl.searchParams.set('page', page.toString());
-            throw new Error('FAL.AI request failed.');
+            const result = await fetch(modelsUrl);
-        }
-        const data = await result.json();
+            if (!result.ok) {
+                console.warn('FAL.AI returned an error.', result.status, result.statusText);
+                throw new Error('FAL.AI request failed.');
+            }
-        if (!Array.isArray(data)) {
-            console.warn('FAL.AI returned invalid data.');
-            throw new Error('FAL.AI request failed.');
-        }
-        const models = data
+            modelsResponse = await result.json();
-            .filter(x => !x.title.toLowerCase().includes('inpainting') &&
+            if (!('items' in modelsResponse) || !Array.isArray(modelsResponse.items)) {
-                !x.title.toLowerCase().includes('control') &&
+                console.warn('FAL.AI returned invalid data.');
-                !x.title.toLowerCase().includes('upscale') &&
+                throw new Error('FAL.AI request failed.');
-                !x.title.toLowerCase().includes('lora'))
+            }
+            models = models.concat(
+                modelsResponse.items.filter(
+                    x => (
+                        !x.title.toLowerCase().includes('inpainting') &&
+                        !x.title.toLowerCase().includes('control') &&
+                        !x.title.toLowerCase().includes('upscale') &&
+                        !x.title.toLowerCase().includes('lora')
+                    ),
+                ),
+            );
+            page = modelsResponse.page + 1;
+        } while (modelsResponse != null && page < modelsResponse.pages);
+        const modelOptions = models
             .sort((a, b) => a.title.localeCompare(b.title))
             .map(x => ({ value: x.modelUrl.split('fal-ai/')[1], text: x.title }));
-        return response.send(models);
+        return response.send(modelOptions);
     } catch (error) {
         console.error(error);
         return response.sendStatus(500);