@rimori/client 2.5.18-next.0 → 2.5.18-next.2

package/dist/plugin/module/AIModule.d.ts

@@ -91,7 +91,7 @@ export declare class AIModule {
      * @param cache Whether to cache the result (default: false).
      * @param model The model to use for generation.
      */
-    getSteamedText(messages: Message[], onMessage: OnLLMResponse, tools?: Tool[], cache?: boolean, model?: string, knowledgeId?: string): Promise<void>;
+    getSteamedText(messages: Message[], onMessage: OnLLMResponse, tools?: Tool[], cache?: boolean, model?: string, knowledgeId?: string): Promise<string>;
     /**
      * Generate voice audio from text using AI.
      * @param text The text to convert to voice.
@@ -151,7 +151,7 @@ export declare class AIModule {
         tools?: Tool[];
         model?: string;
         knowledgeId?: string;
-    }): Promise<void>;
+    }): Promise<T>;
     private streamObject;
     private sendToolResult;
 }

package/dist/plugin/module/AIModule.js

@@ -108,6 +108,7 @@ export class AIModule {
                 onResult: ({ result }) => onMessage(messageId, result, false),
             });
             onMessage(messageId, result, true);
+            return result;
         });
     }
     /**
@@ -208,7 +209,7 @@ export class AIModule {
     getStreamedObject(params) {
         return __awaiter(this, void 0, void 0, function* () {
             const { systemPrompt, responseSchema, userPrompt, onResult, cache = false, tools = [], model = undefined, knowledgeId, } = params;
-            yield this.streamObject({
+            return yield this.streamObject({
                 responseSchema,
                 messages: this.getChatMessage(systemPrompt, userPrompt),
                 onResult,

package/dist/worker/WorkerSetup.d.ts

@@ -2,6 +2,7 @@ import { RimoriClient } from '../plugin/RimoriClient';
 /**
  * Sets up the web worker for the plugin to be able receive and send messages to Rimori.
  * @param pluginId - The id of the plugin to setup the worker for.
- * @param init - The function containing the initialization logic.
+ * @param init - The function containing the initialization logic. The init must be completed within 5s. For long running tasks use the init event (e.g. onboarding.triggerInitPlugin) or run the work async.
+ * @returns A promise that resolves when the worker is setup.
  */
 export declare function setupWorker(pluginId: string, init: (client: RimoriClient) => void | Promise<void>): Promise<void>;

package/dist/worker/WorkerSetup.js

@@ -12,7 +12,8 @@ import { EventBusHandler } from '../fromRimori/EventBus';
 /**
  * Sets up the web worker for the plugin to be able receive and send messages to Rimori.
  * @param pluginId - The id of the plugin to setup the worker for.
- * @param init - The function containing the initialization logic.
+ * @param init - The function containing the initialization logic. The init must be completed within 5s. For long running tasks use the init event (e.g. onboarding.triggerInitPlugin) or run the work async.
+ * @returns A promise that resolves when the worker is setup.
  */
 export function setupWorker(pluginId, init) {
     return __awaiter(this, void 0, void 0, function* () {
@@ -27,10 +28,15 @@ export function setupWorker(pluginId, init) {
         };
         // Assign the mock to globalThis.
         Object.assign(globalThis, { window: mockWindow });
-        EventBusHandler.getInstance('Worker EventBus');
+        EventBusHandler.getInstance('Worker ' + pluginId + ' EventBus');
         const rimoriClient = yield RimoriClient.getInstance(pluginId);
         console.debug('[Worker] RimoriClient initialized.');
-        yield init(rimoriClient);
+        const timoutError = new Error('[Worker ' +
+            pluginId +
+            '] Worker setup must complete within 5s. Use init event (e.g. onboarding.triggerInitPlugin) or run work async.');
+        const initPromise = Promise.resolve(init(rimoriClient));
+        const timeout = new Promise((_, reject) => setTimeout(() => reject(timoutError), 5000));
+        yield Promise.race([initPromise, timeout]);
         console.debug('[Worker] Worker initialized.');
         rimoriClient.event.emit('self.rimori.triggerInitFinished');
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rimori/client",
-  "version": "2.5.18-next.0",
+  "version": "2.5.18-next.2",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "repository": {