utilitas 1998.2.37 → 1998.2.38

package/lib/alan.mjs

@@ -54,7 +54,7 @@ const [
     CLAUDE_35_HAIKU, CLOUD_37_SONNET, AUDIO, WAV, CHATGPT_MINI, ATTACHMENTS,
     CHAT, OPENAI_VOICE, MEDIUM, LOW, HIGH, GPT_REASONING_EFFORT, THINK,
     THINK_STR, THINK_END, AZURE, TOOLS_STR, TOOLS_END, TOOLS, TEXT, THINKING,
-    OK,
+    OK, FUNC,
 ] = [
         'OPENAI', 'GEMINI', 'CHATGPT', 'OPENAI_EMBEDDING', 'GEMINI_EMEDDING',
         'OPENAI_TRAINING', 'OLLAMA', 'CLAUDE', 'gpt-4o-mini', 'gpt-4o', 'o1',
@@ -66,7 +66,7 @@ const [
         'claude-3-7-sonnet@20250219', 'audio', 'wav', 'CHATGPT_MINI',
         '[ATTACHMENTS]', 'CHAT', 'OPENAI_VOICE', 'medium', 'low', 'high',
         'medium', 'think', '<think>', '</think>', 'AZURE', '<tools>',
-        '</tools>', 'tools', 'text', 'thinking', 'OK',
+        '</tools>', 'tools', 'text', 'thinking', 'OK', 'function',
     ];
 
 const [
@@ -90,7 +90,7 @@ const [tool, provider, messages, text] = [
 const [name, user, system, assistant, MODEL, JSON_OBJECT, TOOL, silent]
     = ['Alan', 'user', 'system', 'assistant', 'model', 'json_object', 'tool', true];
 const [CODE_INTERPRETER, RETRIEVAL, FUNCTION]
-    = ['code_interpreter', 'retrieval', 'function'].map(tool);
+    = ['code_interpreter', 'retrieval', FUNC].map(tool);
 const [NOT_INIT, INVALID_FILE]
     = ['AI engine has not been initialized.', 'Invalid file data.'];
 const chatConfig
@@ -442,7 +442,10 @@ const tools = [
                 parameters: {
                     type: 'object',
                     properties: {
-                        keyword: { type: 'string', description: 'The keyword you need to search for.' }
+                        keyword: { type: 'string', description: 'The keyword you need to search for.' },
+                        num: { type: 'integer', description: 'The number of search results you need, default `10`.' },
+                        start: { type: 'integer', description: 'The start index of the search results, default `1`.' },
+                        image: { type: 'boolean', description: 'Whether to search for images, default `false`.' },
                     },
                     required: ['keyword'],
                     additionalProperties: false
@@ -792,9 +795,11 @@ const handleToolsCall = async (msg, options) => {
         for (const fn of calls) {
             switch (options?.flavor) {
                 case CLAUDE:
-                    input = fn.input = String.isString(fn?.input) ? parseJson(fn.input) : fn?.input;
+                    input = fn.input = String.isString(fn?.input)
+                        ? parseJson(fn.input) : fn?.input;
                     packMsg = (content, is_error) => ({
-                        type: 'tool_result', tool_use_id: fn.id, content, is_error,
+                        type: 'tool_result', tool_use_id: fn.id,
+                        content, is_error,
                     });
                     break;
                 case GEMINI:
@@ -802,7 +807,8 @@ const handleToolsCall = async (msg, options) => {
                     packMsg = (t, e) => ({
                         functionResponse: {
                             name: fn?.functionCall?.name, response: {
-                                name: fn?.functionCall?.name, content: e ? `[Error] ${t}` : t,
+                                name: fn?.functionCall?.name,
+                                content: e ? `[Error] ${t}` : t,
                             }
                         }
                     });
@@ -842,13 +848,18 @@ const handleToolsCall = async (msg, options) => {
                 log(rt);
             }
         }
-        switch (options?.flavor) {
-            case CLAUDE: content = content.length ? [{ role: user, content }] : []; break;
-            case GEMINI: content = [{ role: user, parts: content }]; break;
+        if (content.length) {
+            switch (options?.flavor) {
+                case CLAUDE: content = [{ role: user, content }]; break;
+                case GEMINI: content = [{ role: FUNC, parts: content }]; break;
+            }
         }
         responded && await resp(`\n${TOOLS_END}`);
     }
-    return { toolsResult: [...content.length ? preRes : [], ...content], toolsResponse };
+    return {
+        toolsResult: [...content.length ? preRes : [], ...content],
+        toolsResponse,
+    };
 };
 
 const mergeMsgs = (resp, calls) => [resp, ...calls.length ? [
@@ -1089,18 +1100,18 @@ const promptGemini = async (content, options = {}) => {
         options?.toolsResult?.[options?.toolsResult?.length]?.parts
         || buildGeminiMessage(content, options)
     );
-    let [result, references, functionCalls]
+    let [result, references, functionCalls, responded]
         = [options?.result ?? '', null, null];
     for await (const chunk of resp.stream) {
         functionCalls || (functionCalls = chunk.functionCalls);
-        const delta = chunk?.text?.() || '';
         const rfc = packGeminiReferences(
             chunk.candidates[0]?.groundingMetadata?.groundingChunks,
             chunk.candidates[0]?.groundingMetadata?.groundingSupports
         );
-        if (delta === '' && !rfc) { continue; }
+        rfc && (references = rfc);
+        let delta = chunk?.text?.() || '';
+        delta && (responded = responded || (delta = `\n\n${delta}`));
         result += delta;
-        references = rfc;
         await streamResp({ text: options?.delta ? delta : result }, options);
     }
     const _resp = await resp.response;
@@ -1109,14 +1120,14 @@ const promptGemini = async (content, options = {}) => {
         { role: MODEL, parts: functionCalls },
         { ...options, result, flavor: GEMINI }
     );
-    if (functionCalls.length && countToolCalls(toolsResponse) < MAX_TOOL_RECURSION) {
+    if (toolsResult.length && countToolCalls(toolsResponse) < MAX_TOOL_RECURSION) {
         return promptGemini(content, {
             ...options || {}, toolsResult: [...options?.toolsResult || [],
             ...toolsResult], result: toolsResponse,
         });
     }
     return await packGptResp({
-        text: mergeMsgs(toolsResponse, functionCalls), references,
+        text: mergeMsgs(toolsResponse, toolsResult), references,
     }, options);
 };
 

package/lib/manifest.mjs

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

package/lib/shot.mjs

@@ -5,6 +5,7 @@ import { sha256 } from './encryption.mjs';
 import { distillHtml } from './web.mjs';
 
 import {
+    ensureInt,
     ensureString, extract, ignoreErrFunc, inBrowser, parseJson, parseVersion,
     throwError, which
 } from './utilitas.mjs';
@@ -160,66 +161,28 @@ const initSearch = (options) => {
 };
 
 const search = async (query, options) => {
+    const [key, cx, min, max]
+        = [options?.apiKey || googleApiKey, options?.cx || googleCx, 1, 10];
     assert(query, 'Query is required.');
-    const [GOOGLE, DUCKDUCKGO] = ['GOOGLE', 'DUCKDUCKGO'];
-    const provider = ensureString(
-        options?.provider || searchProvider || DUCKDUCKGO, { case: 'UP' }
-    );
-    let apiKey = options?.apiKey || googleApiKey, cx = options?.cx || googleCx,
-        url, parser;
-    switch (provider) {
-        case GOOGLE:
-            assert(apiKey, 'API key is required.');
-            assert(cx, 'CX is required.');
-            url = 'https://www.googleapis.com/customsearch/v1'
-                + `?key=${encodeURIComponent(apiKey)}`
-                + `&cx=${encodeURIComponent(cx)}`
-                + `&q=${encodeURIComponent(query)}`;
-            break;
-        case DUCKDUCKGO:
-            url = 'https://api.duckduckgo.com/'
-                + `?q=${encodeURIComponent(query)}&format=json&skip_disambig=1`;
-            parser = x => x.FirstURL ? {
-                title: x.FirstURL.replace(/^.*\/([^\/]*)$/, '$1').replace(/_/g, ' '),
-                link: x.FirstURL, snippet: x.Text,
-                image: `https://duckduckgo.com${x.Icon.URL}`, source: null,
-            } : null;
-            break;
-        default:
-            throwError(`Invalid provider: ${provider}.`);
-    }
+    assert(key, 'API key is required.');
+    assert(cx, 'CX is required.');
+    const num = ensureInt(options?.num || max, { min, max });
+    const start = ensureInt(options?.start || min, { min });
+    assert(start + num <= 100, 'Reached maximum search limit.');
+    const url = 'https://www.googleapis.com/customsearch/v1'
+        + `?key=${encodeURIComponent(key)}&cx=${encodeURIComponent(cx)}`
+        + `&q=${encodeURIComponent(query)}&num=${num}&start=${start}`
+        + (options?.image ? `&searchType=image` : '');
     const resp = await get(url, { encode: _JSON, ...options || {} });
-    let result = [];
-    if (options?.raw) {
-        result = resp.content;
-    } else if (provider === GOOGLE) {
-        result.push(...resp?.content?.items.map(x => ({
-            title: x.title, link: x.link, snippet: x.snippet,
-            image: x.pagemap?.cse_image?.[0]?.src || null,
-        })));
-    } else if (provider === DUCKDUCKGO) {
-        const cnt = resp?.content;
-        if (cnt?.Abstract) {
-            result.push({
-                title: cnt?.Heading + (cnt?.Entity ? ` (${cnt.Entity})` : ''),
-                link: cnt?.AbstractURL, snippet: cnt?.AbstractText,
-                image: cnt?.Image ? `https://duckduckgo.com${cnt.Image}` : null,
-                source: cnt?.AbstractSource,
-            });
-            if (cnt?.Results) {
-                result.push(...cnt.Results.map(x => ({
-                    title: x.Text, link: x.FirstURL, snippet: null,
-                    image: x.Icon ? `https://duckduckgo.com${x.Icon.URL}` : null,
-                    source: null,
-                })));
-            }
-        } else if (cnt?.RelatedTopics?.[0] && !options?.no_recurse) {
-            result = await search(cnt.RelatedTopics[0].FirstURL.replace(
-                /^.*\/([^\/]*)$/, '$1'
-            ).replace(/_/g, ' '), { ...options, no_recurse: true });
-        }
-    }
-    return result;
+    return options?.raw ? resp.content : {
+        totalResults: resp?.content?.searchInformation?.totalResults || 0,
+        startIndex: resp?.content?.queries?.request?.[0]?.startIndex || 1,
+        items: resp?.content?.items.map(x => ({
+            title: x.title, link: options?.image ? null : x.link,
+            snippet: x.snippet,
+            image: (options?.image ? x.link : x.pagemap?.cse_image?.[0]?.src) || null,
+        })),
+    };
 };
 
 export default get;

package/package.json

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