@rimori/client 2.5.32 → 2.5.33

package/dist/plugin/module/AIModule.js

@@ -1,32 +1,12 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
 /**
  * Controller for AI-related operations.
  * Provides access to text generation, voice synthesis, and object generation.
  */
 export class AIModule {
+    controller;
+    sessionTokenId = null;
+    onRateLimitedCb;
     constructor(controller) {
-        this.sessionTokenId = null;
-        /** Exercise session management. */
-        this.session = {
-            /** Returns the current exercise session token ID (null if no active session). */
-            get: () => this.sessionTokenId,
-            /** Sets the session token ID. */
-            set: (id) => {
-                this.sessionTokenId = id;
-            },
-            /** Clears the stored session token. */
-            clear: () => {
-                this.sessionTokenId = null;
-            },
-        };
         this.controller = controller;
     }
     /**
@@ -43,6 +23,19 @@ export class AIModule {
         }
         return name;
     }
+    /** Exercise session management. */
+    session = {
+        /** Returns the current exercise session token ID (null if no active session). */
+        get: () => this.sessionTokenId,
+        /** Sets the session token ID. */
+        set: (id) => {
+            this.sessionTokenId = id;
+        },
+        /** Clears the stored session token. */
+        clear: () => {
+            this.sessionTokenId = null;
+        },
+    };
     /** Registers a callback invoked whenever a 429 rate-limit response is received. */
     setOnRateLimited(cb) {
         this.onRateLimitedCb = cb;
@@ -56,18 +49,16 @@ export class AIModule {
      * @param params.variables Variables for the server-side prompt template.
      * @returns The generated text.
      */
-    getText(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { messages, tools, cache = false, prompt, variables } = params;
-            const { result } = yield this.streamObject({
-                cache,
-                tools,
-                messages,
-                prompt,
-                variables,
-            });
-            return result;
+    async getText(params) {
+        const { messages, tools, cache = false, prompt, variables } = params;
+        const { result } = await this.streamObject({
+            cache,
+            tools,
+            messages,
+            prompt,
+            variables,
         });
+        return result;
     }
     /**
      * Stream text generation from messages using AI.
@@ -78,21 +69,19 @@ export class AIModule {
      * @param params.prompt Server-side prompt name (e.g. 'writing.analysis').
      * @param params.variables Variables for the server-side prompt template.
      */
-    getStreamedText(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { messages, onMessage, tools, cache = false, prompt, variables, } = params;
-            const messageId = Math.random().toString(36).substring(3);
-            const { result } = yield this.streamObject({
-                cache,
-                tools,
-                messages,
-                prompt,
-                variables,
-                onResult: ({ result }) => onMessage(messageId, result, false),
-            });
-            onMessage(messageId, result, true);
-            return result;
+    async getStreamedText(params) {
+        const { messages, onMessage, tools, cache = false, prompt, variables, } = params;
+        const messageId = Math.random().toString(36).substring(3);
+        const { result } = await this.streamObject({
+            cache,
+            tools,
+            messages,
+            prompt,
+            variables,
+            onResult: ({ result }) => onMessage(messageId, result, false),
         });
+        onMessage(messageId, result, true);
+        return result;
     }
     /**
      * Generate voice audio from text using AI.
@@ -108,26 +97,23 @@ export class AIModule {
      * from the TTS backend while keeping the caller's workflow intact.
      * A warning is logged to the console in this case.
      */
-    getVoice(text_1) {
-        return __awaiter(this, arguments, void 0, function* (text, voice = 'alloy', speed = 1, language, cache = false, instructions) {
-            var _a;
-            if (!text.trim().length) {
-                console.warn('[rimori-client] getVoice called with empty text — skipping TTS request and returning empty Blob.');
-                return new Blob([], { type: 'audio/mpeg' });
-            }
-            return yield this.controller.fetchBackend('/voice/tts', {
-                method: 'POST',
-                body: JSON.stringify({
-                    input: text,
-                    voice,
-                    speed,
-                    language,
-                    cache,
-                    instructions,
-                    session_token_id: (_a = this.sessionTokenId) !== null && _a !== void 0 ? _a : undefined,
-                }),
-            }).then((r) => r.blob());
-        });
+    async getVoice(text, voice = 'alloy', speed = 1, language, cache = false, instructions) {
+        if (!text.trim().length) {
+            console.warn('[rimori-client] getVoice called with empty text — skipping TTS request and returning empty Blob.');
+            return new Blob([], { type: 'audio/mpeg' });
+        }
+        return await this.controller.fetchBackend('/voice/tts', {
+            method: 'POST',
+            body: JSON.stringify({
+                input: text,
+                voice,
+                speed,
+                language,
+                cache,
+                instructions,
+                session_token_id: this.sessionTokenId ?? undefined,
+            }),
+        }).then((r) => r.blob());
     }
     /**
      * Convert voice audio to text using AI.
@@ -135,25 +121,23 @@ export class AIModule {
      * @param language Optional language for the voice.
      * @returns The transcribed text.
      */
-    getTextFromVoice(file, language) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const formData = new FormData();
-            formData.append('file', file);
-            if (language) {
-                formData.append('language', language.code);
-            }
-            if (this.sessionTokenId) {
-                formData.append('session_token_id', this.sessionTokenId);
-            }
-            return yield this.controller.fetchBackend('/voice/stt', {
-                method: 'POST',
-                body: formData,
-            })
-                .then((r) => r.json())
-                .then((r) => {
-                // console.log("STT response: ", r);
-                return r.text;
-            });
+    async getTextFromVoice(file, language) {
+        const formData = new FormData();
+        formData.append('file', file);
+        if (language) {
+            formData.append('language', language.code);
+        }
+        if (this.sessionTokenId) {
+            formData.append('session_token_id', this.sessionTokenId);
+        }
+        return await this.controller.fetchBackend('/voice/stt', {
+            method: 'POST',
+            body: formData,
+        })
+            .then((r) => r.json())
+            .then((r) => {
+            // console.log("STT response: ", r);
+            return r.text;
         });
     }
     /**
@@ -164,16 +148,14 @@ export class AIModule {
      * @param request.variables Variables for the server-side prompt template.
      * @returns The generated object.
      */
-    getObject(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { cache = false, tools = [], prompt, variables } = params;
-            return yield this.streamObject({
-                messages: [],
-                cache,
-                tools,
-                prompt,
-                variables,
-            });
+    async getObject(params) {
+        const { cache = false, tools = [], prompt, variables } = params;
+        return await this.streamObject({
+            messages: [],
+            cache,
+            tools,
+            prompt,
+            variables,
         });
     }
     /**
@@ -184,162 +166,158 @@ export class AIModule {
      * @param request.prompt Server-side prompt name (e.g. 'writing.analysis').
      * @param request.variables Variables for the server-side prompt template.
      */
-    getStreamedObject(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { onResult, cache = false, tools = [], prompt, variables } = params;
-            return yield this.streamObject({
-                messages: [],
-                onResult,
-                cache,
-                tools,
-                prompt,
-                variables,
-            });
+    async getStreamedObject(params) {
+        const { onResult, cache = false, tools = [], prompt, variables } = params;
+        return await this.streamObject({
+            messages: [],
+            onResult,
+            cache,
+            tools,
+            prompt,
+            variables,
         });
     }
-    streamObject(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e;
-            const { messages, onResult = () => null, cache = false, tools = [], prompt, variables, } = params;
-            const chatMessages = messages.map((message, index) => (Object.assign(Object.assign({}, message), { id: `${index + 1}` })));
-            const payload = {
-                cache,
-                tools,
-                stream: true,
-                messages: chatMessages,
-                session_token_id: (_a = this.sessionTokenId) !== null && _a !== void 0 ? _a : undefined,
-            };
-            if (prompt) {
-                payload.prompt = { name: this.resolvePromptName(prompt), variables: variables !== null && variables !== void 0 ? variables : {} };
-            }
-            const response = yield this.controller.fetchBackend('/ai/llm', {
-                method: 'POST',
-                body: JSON.stringify(payload),
-            });
-            if (!response.ok) {
-                if (response.status === 429) {
-                    const body = yield response.json().catch(() => ({}));
-                    const remaining = (_b = body.exercises_remaining) !== null && _b !== void 0 ? _b : 0;
-                    (_c = this.onRateLimitedCb) === null || _c === void 0 ? void 0 : _c.call(this, remaining);
-                    throw new Error(`Rate limit exceeded: ${(_d = body.error) !== null && _d !== void 0 ? _d : 'Daily exercise limit reached'}. exercises_remaining: ${remaining}`);
-                }
-                throw new Error(`Failed to stream object: ${response.status} ${response.statusText}`);
+    async streamObject(params) {
+        const { messages, onResult = () => null, cache = false, tools = [], prompt, variables, } = params;
+        const chatMessages = messages.map((message, index) => ({
+            ...message,
+            id: `${index + 1}`,
+        }));
+        const payload = {
+            cache,
+            tools,
+            stream: true,
+            messages: chatMessages,
+            session_token_id: this.sessionTokenId ?? undefined,
+        };
+        if (prompt) {
+            payload.prompt = { name: this.resolvePromptName(prompt), variables: variables ?? {} };
+        }
+        const response = await this.controller.fetchBackend('/ai/llm', {
+            method: 'POST',
+            body: JSON.stringify(payload),
+        });
+        if (!response.ok) {
+            if (response.status === 429) {
+                const body = await response.json().catch(() => ({}));
+                const remaining = body.exercises_remaining ?? 0;
+                this.onRateLimitedCb?.(remaining);
+                throw new Error(`Rate limit exceeded: ${body.error ?? 'Daily exercise limit reached'}. exercises_remaining: ${remaining}`);
             }
-            if (!response.body) {
-                throw new Error('No response body.');
+            throw new Error(`Failed to stream object: ${response.status} ${response.statusText}`);
+        }
+        if (!response.body) {
+            throw new Error('No response body.');
+        }
+        const reader = response.body.getReader();
+        const decoder = new TextDecoder('utf-8');
+        let currentObject = {};
+        // Buffer for SSE lines that are split across network chunks.
+        // TCP/IP does not guarantee that each `read()` call delivers a complete
+        // logical line. For example, the `token:` line carrying the session token
+        // may arrive as two separate chunks:
+        //   chunk 1 → `token: {"token_id":`
+        //   chunk 2 → `"abc123"}\n`
+        // Without buffering, `JSON.parse` would throw on the partial line and the
+        // session token would be silently discarded, causing the next LLM call to
+        // start without a session (triggering an unnecessary extra round-trip via
+        // `session.ensure()`). By keeping the incomplete tail in `lineBuffer` and
+        // prepending it to the next chunk we always process whole lines.
+        let lineBuffer = '';
+        let isLoading = true;
+        while (isLoading) {
+            //wait 50ms to not overload the CPU
+            await new Promise((resolve) => setTimeout(resolve, 30));
+            const { value, done: readerDone } = await reader.read();
+            if (readerDone) {
+                isLoading = false;
+                onResult(currentObject, false);
+                return currentObject;
             }
-            const reader = response.body.getReader();
-            const decoder = new TextDecoder('utf-8');
-            let currentObject = {};
-            // Buffer for SSE lines that are split across network chunks.
-            // TCP/IP does not guarantee that each `read()` call delivers a complete
-            // logical line. For example, the `token:` line carrying the session token
-            // may arrive as two separate chunks:
-            //   chunk 1 → `token: {"token_id":`
-            //   chunk 2 → `"abc123"}\n`
-            // Without buffering, `JSON.parse` would throw on the partial line and the
-            // session token would be silently discarded, causing the next LLM call to
-            // start without a session (triggering an unnecessary extra round-trip via
-            // `session.ensure()`). By keeping the incomplete tail in `lineBuffer` and
-            // prepending it to the next chunk we always process whole lines.
-            let lineBuffer = '';
-            let isLoading = true;
-            while (isLoading) {
-                //wait 50ms to not overload the CPU
-                yield new Promise((resolve) => setTimeout(resolve, 30));
-                const { value, done: readerDone } = yield reader.read();
-                if (readerDone) {
-                    isLoading = false;
-                    onResult(currentObject, false);
-                    return currentObject;
-                }
-                //the check needs to be behind readerDone because in closed connections the value is undefined
-                if (!value)
-                    continue;
-                const chunk = decoder.decode(value, { stream: true });
-                // Prepend any incomplete line left over from the previous chunk, then
-                // split on newlines. `parts.pop()` removes (and saves) the last element
-                // which may be an incomplete line if the chunk did not end with '\n'.
-                const combined = lineBuffer + chunk;
-                const parts = combined.split('\n');
-                lineBuffer = (_e = parts.pop()) !== null && _e !== void 0 ? _e : '';
-                const lines = parts.filter((line) => line.trim());
-                for (const line of lines) {
-                    // Handle token: line (session token issued by backend on first AI call)
-                    if (line.startsWith('token:')) {
-                        try {
-                            const tokenData = JSON.parse(line.slice(6).trim());
-                            if (tokenData.token_id) {
-                                this.sessionTokenId = tokenData.token_id;
-                            }
-                        }
-                        catch (_f) {
-                            console.error('Failed to parse token: line', line);
-                        }
-                        continue;
-                    }
-                    // Handle debug: line (prompt resolution debug info, dev/local only)
-                    if (line.startsWith('debug:')) {
-                        try {
-                            const debug = JSON.parse(line.slice(6).trim());
-                            console.group(`[Rimori Prompt] ${debug.promptName}`);
-                            console.log('System prompt:\n', debug.system);
-                            console.log('User prompt:\n', debug.user);
-                            console.log('Variables:', debug.variables);
-                            console.groupEnd();
-                        }
-                        catch (_g) {
-                            // Ignore malformed debug lines
+            //the check needs to be behind readerDone because in closed connections the value is undefined
+            if (!value)
+                continue;
+            const chunk = decoder.decode(value, { stream: true });
+            // Prepend any incomplete line left over from the previous chunk, then
+            // split on newlines. `parts.pop()` removes (and saves) the last element
+            // which may be an incomplete line if the chunk did not end with '\n'.
+            const combined = lineBuffer + chunk;
+            const parts = combined.split('\n');
+            lineBuffer = parts.pop() ?? '';
+            const lines = parts.filter((line) => line.trim());
+            for (const line of lines) {
+                // Handle token: line (session token issued by backend on first AI call)
+                if (line.startsWith('token:')) {
+                    try {
+                        const tokenData = JSON.parse(line.slice(6).trim());
+                        if (tokenData.token_id) {
+                            this.sessionTokenId = tokenData.token_id;
                         }
-                        continue;
                     }
-                    const command = line.substring(0, 5);
-                    const dataStr = line.substring(5).trim();
-                    if (dataStr === '[DONE]') {
-                        isLoading = false;
-                        onResult(currentObject, false);
-                        return currentObject;
+                    catch {
+                        console.error('Failed to parse token: line', line);
                     }
-                    if (command === 'data:') {
-                        currentObject = JSON.parse(dataStr);
-                        onResult(currentObject, true);
+                    continue;
+                }
+                // Handle debug: line (prompt resolution debug info, dev/local only)
+                if (line.startsWith('debug:')) {
+                    try {
+                        const debug = JSON.parse(line.slice(6).trim());
+                        console.group(`[Rimori Prompt] ${debug.promptName}`);
+                        console.log('System prompt:\n', debug.system);
+                        console.log('User prompt:\n', debug.user);
+                        console.log('Variables:', debug.variables);
+                        console.groupEnd();
                     }
-                    else if (command === 'tool:') {
-                        const { toolCallId, toolName, args } = JSON.parse(dataStr);
-                        const tool = tools.find((tool) => tool.name === toolName);
-                        if (tool && tool.execute) {
-                            const result = yield tool.execute(args);
-                            // Send the result to the backend
-                            yield this.sendToolResult(toolCallId, result);
-                        }
-                        else if (tool && !tool.execute) {
-                            console.error('Tool found but has no execute function:', toolName);
-                        }
-                        else {
-                            console.error('Tool not found:', toolName);
-                        }
+                    catch {
+                        // Ignore malformed debug lines
                     }
-                    else if (command === 'error') {
-                        //error has 5 letters + the colon so we need to remove one character of the data string to get the error message
-                        console.error('Error:', dataStr.substring(1));
+                    continue;
+                }
+                const command = line.substring(0, 5);
+                const dataStr = line.substring(5).trim();
+                if (dataStr === '[DONE]') {
+                    isLoading = false;
+                    onResult(currentObject, false);
+                    return currentObject;
+                }
+                if (command === 'data:') {
+                    currentObject = JSON.parse(dataStr);
+                    onResult(currentObject, true);
+                }
+                else if (command === 'tool:') {
+                    const { toolCallId, toolName, args } = JSON.parse(dataStr);
+                    const tool = tools.find((tool) => tool.name === toolName);
+                    if (tool && tool.execute) {
+                        const result = await tool.execute(args);
+                        // Send the result to the backend
+                        await this.sendToolResult(toolCallId, result);
                     }
-                    else if (command === 'info:') {
-                        //ignore info messages
+                    else if (tool && !tool.execute) {
+                        console.error('Tool found but has no execute function:', toolName);
                     }
                     else {
-                        console.error('Unknown stream data:', line);
+                        console.error('Tool not found:', toolName);
                     }
                 }
+                else if (command === 'error') {
+                    //error has 5 letters + the colon so we need to remove one character of the data string to get the error message
+                    console.error('Error:', dataStr.substring(1));
+                }
+                else if (command === 'info:') {
+                    //ignore info messages
+                }
+                else {
+                    console.error('Unknown stream data:', line);
+                }
             }
-            return currentObject;
-        });
+        }
+        return currentObject;
     }
-    sendToolResult(toolCallId, result) {
-        return __awaiter(this, void 0, void 0, function* () {
-            yield this.controller.fetchBackend('/ai/llm/tool_result', {
-                method: 'POST',
-                body: JSON.stringify({ toolCallId, result: result !== null && result !== void 0 ? result : '[DONE]' }),
-            });
+    async sendToolResult(toolCallId, result) {
+        await this.controller.fetchBackend('/ai/llm/tool_result', {
+            method: 'POST',
+            body: JSON.stringify({ toolCallId, result: result ?? '[DONE]' }),
         });
     }
 }

package/dist/plugin/module/DbModule.js

@@ -1,13 +1,8 @@
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
 export class DbModule {
+    supabase;
+    communicationHandler;
+    tablePrefix;
+    schema;
     constructor(supabase, communicationHandler, info) {
         this.supabase = supabase;
         this.communicationHandler = communicationHandler;
@@ -63,18 +58,16 @@ export class DbModule {
      * @param entryId The UUID of the entry to update
      * @param publicity The desired publicity level
      */
-    setPublicity(table, entryId, publicity) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const tableName = this.getTableName(table);
-            yield this.communicationHandler.fetchBackend('/db-entry/publicity', {
-                method: 'POST',
-                body: JSON.stringify({
-                    table_name: tableName,
-                    schema: this.schema,
-                    entry_id: entryId,
-                    publicity,
-                }),
-            });
+    async setPublicity(table, entryId, publicity) {
+        const tableName = this.getTableName(table);
+        await this.communicationHandler.fetchBackend('/db-entry/publicity', {
+            method: 'POST',
+            body: JSON.stringify({
+                table_name: tableName,
+                schema: this.schema,
+                entry_id: entryId,
+                publicity,
+            }),
         });
     }
     /**
@@ -83,16 +76,14 @@ export class DbModule {
      * @param params Search parameters
      * @returns Matching rows sorted by similarity
      */
-    vectorSearch(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const response = yield this.communicationHandler.fetchBackend('/plugin-search/vector-search', {
-                method: 'POST',
-                body: JSON.stringify(params),
-            });
-            if (!response.ok) {
-                throw new Error(`Vector search failed: ${response.statusText}`);
-            }
-            return yield response.json();
+    async vectorSearch(params) {
+        const response = await this.communicationHandler.fetchBackend('/plugin-search/vector-search', {
+            method: 'POST',
+            body: JSON.stringify(params),
         });
+        if (!response.ok) {
+            throw new Error(`Vector search failed: ${response.statusText}`);
+        }
+        return await response.json();
     }
 }