utilitas 2000.3.32 → 2000.3.34

package/lib/alan.mjs

@@ -8,8 +8,8 @@ import {
     MIME_MOV, MIME_MPEG, MIME_MP4, MIME_MPG, MIME_AVI, MIME_WMV, MIME_MPEGPS,
     MIME_FLV, MIME_GIF, MIME_WEBP, MIME_PDF, MIME_AAC, MIME_FLAC, MIME_MP3,
     MIME_MPEGA, MIME_M4A, MIME_MPGA, MIME_OPUS, MIME_PCM, MIME_WAV, MIME_WEBM,
-    MIME_TGPP, MIME_PCM16, MIME_OGG, convert, formatDataURL,
-    getTempPath, decodeBase64DataURL,
+    MIME_TGPP, MIME_PCM16, MIME_OGG, convert, getTempPath, decodeBase64DataURL,
+    getMime,
 } from './storage.mjs';
 
 import {
@@ -583,46 +583,51 @@ const isOpenrouter = (provider, model) => insensitiveCompare(
     provider, OPENROUTER
 ) && (model ? model?.source : true);
 
-const buildMessage = (content, options) => {
+const buildMessage = async (content, options) => {
     content = content || '';
-    const attachments = (options?.attachments || []).map(x => {
+    const attachments = await Promise.all((options?.attachments || []).map(async x => {
         assert(
             MODELS[options?.model]?.supportedMimeTypes?.includes?.(x.mime_type),
             `Unsupported mime type: '${x.mime_type}'.`
         );
+        if ([
+            MIME_PNG, MIME_JPEG, MIME_GIF, MIME_WEBP
+        ].includes?.(x.mime_type)) {
+            return {
+                type: 'image_url',
+                image_url: {
+                    url: x.url || await convert(x.data, {
+                        input: BUFFER, expected: DATAURL,
+                    }), detail: 'high',
+                },
+            };
+        } else if ([
+            MIME_AAC, MIME_FLAC, MIME_MP3, MIME_MPEGA, MIME_M4A, MIME_MPGA,
+            MIME_OPUS, MIME_PCM, MIME_WAV, MIME_TGPP, MIME_PCM16, MIME_OGG,
+        ].includes?.(x.mime_type)) {
+            return {
+                type: 'input_audio',
+                input_audio: {
+                    data: await convert(x.data, {
+                        input: BUFFER, expected: BASE64,
+                    }), format: WAV,
+                },
+            };
+        }
+        [
+            MIME_TEXT, MIME_MOV, MIME_MPEG, MIME_WEBM, MIME_MP4, MIME_MPG,
+            MIME_AVI, MIME_WMV, MIME_MPEGPS, MIME_FLV, MIME_PDF,
+        ].includes?.(x.mime_type)
+            || log(`Unknown mime type: '${x.mime_type}', fallbacked.`);
         return {
-            type: 'image_url',
-            image_url: { url: x.url, detail: 'high' },
+            type: 'file', file: {
+                file_data: await convert(x.data, {
+                    input: BUFFER, expected: DATAURL,
+                }), filename: x.file_name
+                    || `${uuidv4()}.${x.mime_type.split('/')[1]}`,
+            },
         };
-        // if ([
-        //     MIME_PNG, MIME_JPEG, MIME_GIF, MIME_WEBP
-        // ].includes?.(x.mime_type)) {
-        //     return {
-        //         type: 'image_url',
-        //         image_url: { url: x.url, detail: 'high' },
-        //     };
-        // } else if ([
-        //     MIME_AAC, MIME_FLAC, MIME_MP3, MIME_MPEGA, MIME_M4A, MIME_MPGA,
-        //     MIME_OPUS, MIME_PCM, MIME_WAV, MIME_TGPP, MIME_PCM16, MIME_OGG,
-        // ].includes?.(x.mime_type)) {
-        //     return {
-        //         type: 'input_audio',
-        //         input_audio: { data: x.data, format: WAV },
-        //     };
-        // }
-        // [
-        //     MIME_TEXT, MIME_MOV, MIME_MPEG, MIME_WEBM, MIME_MP4, MIME_MPG,
-        //     MIME_AVI, MIME_WMV, MIME_MPEGPS, MIME_FLV, MIME_PDF,
-        // ].includes?.(x.mime_type)
-        //     || log(`Unknown mime type: '${x.mime_type}', fallbacked.`);
-        // return {
-        //     type: 'file', file: {
-        //         file_data: formatDataURL(x.mime_type, x.data),
-        //         filename: x.file_name
-        //             || `${uuidv4()}.${x.mime_type.split('/')[1]}`,
-        //     },
-        // };
-    });
+    }));
     const message = String.isString(content) ? {
         role: options?.role || user,
         content: content.length ? [{ type: TEXT, text: content }] : [],
@@ -783,38 +788,31 @@ const buildPrompts = async (model, input, options = {}) => {
         options.attachments?.length ? options.attachments : []
     ).map(async x => {
         if (String.isString(x)) {
-            var convResp = await convert(x, { input: FILE, expected: DATAURL, meta: true });
-            return {
-                url: convResp.content,
-                mime_type: convResp.mime,
-            }
+            var convResp = await convert(x, { input: FILE, expected: BUFFER, meta: true });
+            return { data: convResp.content, mime_type: convResp.mime };
         } else if (Buffer.isBuffer(x)) {
-            var convResp = await convert(x, { input: BUFFER, expected: DATAURL, meta: true });
-            return {
-                url: convResp.content,
-                mime_type: convResp.mime,
-            }
+            return { data: x, mime_type: (await getMime(x))?.mime }
         } else if (Object.isObject(x)) { return x; } else { return null; }
     }))).filter(x => (model?.supportedMimeTypes || []).includes(x.mime_type));
-    const systemPrompt = buildMessage(options.systemPrompt, system);
-    const msgBuilder = () => {
+    const systemPrompt = await buildMessage(options.systemPrompt, system);
+    const msgBuilder = async () => {
         [history, _history] = [[], []];
-        (options.messages?.length ? options.messages : []).map((x, i) => {
-            _history.push(buildMessage(x.request));
-            _history.push(buildMessage(x.response, _assistant));
-        });
+        await Promise.all((options.messages?.length ? options.messages : []).map(async (x, i) => {
+            _history.push(await buildMessage(x.request));
+            _history.push(await buildMessage(x.response, _assistant));
+        }));
         history = messages([
-            systemPrompt, ..._history, buildMessage(content, options),
+            systemPrompt, ..._history, await buildMessage(content, options),
             ...options.toolsResult?.length ? options.toolsResult : []
         ]);
     };
-    msgBuilder();
+    await msgBuilder();
     await trimPrompt(() => [
         systemPrompt, _history, content, options.toolsResult
-    ], () => {
+    ], async () => {
         if (options.messages?.length) {
             options.messages?.shift();
-            msgBuilder();
+            await msgBuilder();
         } else if (options.trimBeginning) {
             content = '...' + trimBeginning(trimBeginning(content).slice(1));
         } else {
@@ -1233,7 +1231,7 @@ const initChat = async (options = {}) => {
     }
     options.instructions && (chatConfig.systemPrompt = options.instructions);
     // Use Gemini instead of ChatGPT because of the longer package.
-    const [spTokens, ais] = await Promise.all([countTokens([buildMessage(
+    const [spTokens, ais] = await Promise.all([countTokens([await buildMessage(
         chatConfig.systemPrompt, system
     )]), getAi(null, { all: true })]);
     for (const ai of ais.filter(x => ![
@@ -1330,15 +1328,9 @@ const distillFile = async (attachments, o) => {
         o?.keepPaging ? '' : '- If the document has multiple pages, merge them into one page. Please do not return any paging information.',
         o?.keepDecoration ? '' : '- If the document has side notes, headers, footers, or watermarks, please ignore them.',
     ].filter(x => x).join('\n');
-    attachments = await Promise.all(ensureArray(attachments).map(async x => {
-        const convResp = await convert(
-            x, { expected: DATAURL, ...o || {}, meta: true }
-        );
-        return { url: convResp.content, mime_type: convResp.mime };
-    }));
     return await prompt(strPmt, {
         select: { vision: true, hearing: true, fast: true },
-        simple: true, ...o, attachments,
+        simple: true, ...o, attachments: ensureArray(attachments),
     });
 };
 

package/lib/manifest.mjs

@@ -1,7 +1,7 @@
 const manifest = {
     "name": "utilitas",
     "description": "Just another common utility for JavaScript.",
-    "version": "2000.3.32",
+    "version": "2000.3.34",
     "private": false,
     "homepage": "https://github.com/Leask/utilitas",
     "main": "index.mjs",

package/lib/storage.mjs

@@ -538,6 +538,7 @@ export {
     getConfigFilename,
     getGcUrlByBucket,
     getIdByGs,
+    getMime,
     getTempPath,
     handleError,
     init,

package/package.json

@@ -1,7 +1,7 @@
 {
     "name": "utilitas",
     "description": "Just another common utility for JavaScript.",
-    "version": "2000.3.32",
+    "version": "2000.3.34",
     "private": false,
     "homepage": "https://github.com/Leask/utilitas",
     "main": "index.mjs",