Improve OpenRouter model lists in extensions (#5459) * fix: extensions OpenRouter model lists * fix: update JSDoc for optional mapping function parameter in fetchModelsByModality * fix: update JSDoc to clarify return type of fetchModelsByModality function * fix: encode output modality in fetchModelsByModality API request

3f72d3df80fa446bf9a9f6625306a0990037650e

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed

src/endpoints/openrouter.js+10 -10 Viewed

  * @param {string} endpoint - The API endpoint to fetch from
  * @param {string} inputModality - Required input modality
  * @param {string} outputModality - Required output modality
- * @param {boolean} [idsOnly=false] - Whether to return only model IDs
+ * @param {((model: any) => any) | null} [mapFn=null] - Optional mapping function to transform the results
- * @returns {Promise<any[]>} Filtered models or model IDs
+ * @returns {Promise<any[]>} Filtered and/or mapped models
  */
-async function fetchModelsByModality(endpoint, inputModality, outputModality, idsOnly = false) {
+async function fetchModelsByModality(endpoint, inputModality, outputModality, mapFn = null) {
-    const response = await fetch(`${API_OPENROUTER}${endpoint}`, {
+    const response = await fetch(`${API_OPENROUTER}${endpoint}?output_modalities=${encodeURIComponent(outputModality)}`, {
         method: 'GET',
         headers: { 'Accept': 'application/json' },
     });

         .filter(m => m.architecture.output_modalities.includes(outputModality))
         .sort((a, b) => a?.id && b?.id ? a.id.localeCompare(b.id) : 0);
-    return idsOnly ? filtered.map(m => m.id) : filtered;
+    return typeof mapFn === 'function' ? filtered.map(mapFn) : filtered;
+}
 router.post('/models/multimodal', async (_req, res) => {
     try {
-        const models = await fetchModelsByModality('/models', 'image', 'text', true);
+        const models = await fetchModelsByModality('/models', 'image', 'text', m => m.id);
         return res.json(models);
     } catch (error) {
         console.error(error);

 router.post('/models/embedding', async (_req, res) => {
     try {
-        const models = await fetchModelsByModality('/embeddings/models', 'text', 'embeddings');
+        const models = await fetchModelsByModality('/models', 'text', 'embeddings', m => ({ id: m.id, name: m.name }));
         return res.json(models);
     } catch (error) {
         console.error(error);

 router.post('/models/image', async (_req, res) => {
     try {
-        const models = await fetchModelsByModality('/models', 'text', 'image');
+        const models = await fetchModelsByModality('/models', 'text', 'image', m => ({ value: m.id, text: m.name || m.id }));
-        return res.json(models.map(m => ({ value: m.id, text: m.name || m.id })));
+        return res.json(models);
     } catch (error) {
         console.error(error);
         return res.sendStatus(500);

                         content: prompt,
                     },
                 ],
-                modalities: ['image', 'text'],
+                modalities: ['image'],
                 image_config: {
                     aspect_ratio: req.body.aspect_ratio || '1:1',
                 },