@rimori/client 2.5.14-next.3 → 2.5.14-next.4

package/dist/plugin/RimoriClient.js

@@ -15,6 +15,7 @@ import { DbModule } from './module/DbModule';
 import { EventModule } from './module/EventModule';
 import { AIModule } from './module/AIModule';
 import { ExerciseModule } from './module/ExerciseModule';
+import { EventBus } from '../fromRimori/EventBus';
 export class RimoriClient {
     constructor(controller, supabase, info) {
         this.navigation = {
@@ -30,7 +31,10 @@ export class RimoriClient {
         this.rimoriInfo = info;
         this.sharedContent = new SharedContentController(supabase, this);
         this.ai = new AIModule(info.backendUrl, () => this.rimoriInfo.token);
-        this.event = new EventModule(info.pluginId);
+        this.ai.setOnRateLimited((exercisesRemaining) => {
+            EventBus.emit(info.pluginId, 'global.quota.triggerExceeded', { exercises_remaining: exercisesRemaining });
+        });
+        this.event = new EventModule(info.pluginId, info.backendUrl, () => this.rimoriInfo.token, this.ai);
         this.db = new DbModule(supabase, controller, info);
         this.plugin = new PluginModule(supabase, controller, info, this.ai);
         this.exercise = new ExerciseModule(supabase, controller, info, this.event);

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

@@ -58,7 +58,17 @@ export interface ObjectRequest {
 export declare class AIModule {
     private getToken;
     private backendUrl;
+    private sessionTokenId;
+    private onRateLimitedCb?;
     constructor(backendUrl: string, getToken: () => string);
+    /** Returns the current exercise session token ID (null if no active session). */
+    getSessionTokenId(): string | null;
+    /** Sets the session token ID (used by workers inheriting a session from the plugin). */
+    setSessionToken(id: string): void;
+    /** Clears the stored session token (called after macro accomplishment). */
+    clearSessionToken(): void;
+    /** Registers a callback invoked whenever a 429 rate-limit response is received. */
+    setOnRateLimited(cb: (exercisesRemaining: number) => void): void;
     /**
      * Generate text from messages using AI.
      * @param messages The messages to generate text from.

package/dist/plugin/module/AIModule.js

@@ -13,9 +13,26 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
  */
 export class AIModule {
     constructor(backendUrl, getToken) {
+        this.sessionTokenId = null;
         this.backendUrl = backendUrl;
         this.getToken = getToken;
     }
+    /** Returns the current exercise session token ID (null if no active session). */
+    getSessionTokenId() {
+        return this.sessionTokenId;
+    }
+    /** Sets the session token ID (used by workers inheriting a session from the plugin). */
+    setSessionToken(id) {
+        this.sessionTokenId = id;
+    }
+    /** Clears the stored session token (called after macro accomplishment). */
+    clearSessionToken() {
+        this.sessionTokenId = null;
+    }
+    /** Registers a callback invoked whenever a 429 rate-limit response is received. */
+    setOnRateLimited(cb) {
+        this.onRateLimitedCb = cb;
+    }
     /**
      * Generate text from messages using AI.
      * @param messages The messages to generate text from.
@@ -78,13 +95,14 @@ export class AIModule {
      */
     getVoice(text_1) {
         return __awaiter(this, arguments, void 0, function* (text, voice = 'alloy', speed = 1, language, cache = false) {
+            var _a;
             return yield fetch(`${this.backendUrl}/voice/tts`, {
                 method: 'POST',
                 headers: {
                     'Content-Type': 'application/json',
                     Authorization: `Bearer ${this.getToken()}`,
                 },
-                body: JSON.stringify({ input: text, voice, speed, language, cache }),
+                body: JSON.stringify({ input: text, voice, speed, language, cache, session_token_id: (_a = this.sessionTokenId) !== null && _a !== void 0 ? _a : undefined }),
             }).then((r) => r.blob());
         });
     }
@@ -101,6 +119,9 @@ export class AIModule {
             if (language) {
                 formData.append('language', language.code);
             }
+            if (this.sessionTokenId) {
+                formData.append('session_token_id', this.sessionTokenId);
+            }
             return yield fetch(`${this.backendUrl}/voice/stt`, {
                 method: 'POST',
                 headers: { Authorization: `Bearer ${this.getToken()}` },
@@ -172,6 +193,7 @@ export class AIModule {
     }
     streamObject(params) {
         return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d;
             const { messages, responseSchema, onResult = () => null, cache = false, tools = [], model = undefined, knowledgeId, } = params;
             const chatMessages = messages.map((message, index) => (Object.assign(Object.assign({}, message), { id: `${index + 1}` })));
             const response = yield fetch(`${this.backendUrl}/ai/llm`, {
@@ -183,11 +205,18 @@ export class AIModule {
                     messages: chatMessages,
                     model,
                     knowledge_id: knowledgeId,
+                    session_token_id: (_a = this.sessionTokenId) !== null && _a !== void 0 ? _a : undefined,
                 }),
                 method: 'POST',
                 headers: { Authorization: `Bearer ${this.getToken()}`, 'Content-Type': 'application/json' },
             });
             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}`);
             }
             if (!response.body) {
@@ -212,6 +241,19 @@ export class AIModule {
                 const chunk = decoder.decode(value, { stream: true });
                 const lines = chunk.split('\n').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 (_e) {
+                            console.error('Failed to parse token: line', line);
+                        }
+                        continue;
+                    }
                     const command = line.substring(0, 5);
                     const dataStr = line.substring(5).trim();
                     if (dataStr === '[DONE]') {

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

@@ -1,6 +1,7 @@
 import { MainPanelAction, SidebarAction } from '../../fromRimori/PluginTypes';
 import { AccomplishmentPayload } from '../../controller/AccomplishmentController';
 import { EventBusMessage, EventHandler, EventPayload, EventListener } from '../../fromRimori/EventBus';
+import { AIModule } from './AIModule';
 /**
  * Event module for plugin event bus operations.
  * Provides methods for emitting, listening to, and responding to events.
@@ -8,7 +9,10 @@ import { EventBusMessage, EventHandler, EventPayload, EventListener } from '../.
 export declare class EventModule {
     private pluginId;
     private accomplishmentController;
-    constructor(pluginId: string);
+    private aiModule;
+    private backendUrl;
+    private getToken;
+    constructor(pluginId: string, backendUrl: string, getToken: () => string, aiModule: AIModule);
     getGlobalEventTopic(preliminaryTopic: string): string;
     /**
      * Emit an event to Rimori or a plugin.
@@ -51,7 +55,7 @@ export declare class EventModule {
      * Emit an accomplishment.
      * @param payload The payload to emit.
      */
-    emitAccomplishment(payload: AccomplishmentPayload): void;
+    emitAccomplishment(payload: AccomplishmentPayload): Promise<void>;
     /**
      * Subscribe to an accomplishment.
      * @param accomplishmentTopic The topic to subscribe to.

package/dist/plugin/module/EventModule.js

@@ -1,3 +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());
+    });
+};
 import { AccomplishmentController } from '../../controller/AccomplishmentController';
 import { EventBus } from '../../fromRimori/EventBus';
 /**
@@ -5,8 +14,11 @@ import { EventBus } from '../../fromRimori/EventBus';
  * Provides methods for emitting, listening to, and responding to events.
  */
 export class EventModule {
-    constructor(pluginId) {
+    constructor(pluginId, backendUrl, getToken, aiModule) {
         this.pluginId = pluginId;
+        this.backendUrl = backendUrl;
+        this.getToken = getToken;
+        this.aiModule = aiModule;
         this.accomplishmentController = new AccomplishmentController(pluginId);
     }
     getGlobalEventTopic(preliminaryTopic) {
@@ -86,7 +98,24 @@ export class EventModule {
      * @param payload The payload to emit.
      */
     emitAccomplishment(payload) {
-        this.accomplishmentController.emitAccomplishment(payload);
+        return __awaiter(this, void 0, void 0, function* () {
+            if (payload.type === 'macro') {
+                const sessionId = this.aiModule.getSessionTokenId();
+                if (sessionId) {
+                    try {
+                        yield fetch(`${this.backendUrl}/ai/session/${sessionId}/complete`, {
+                            method: 'PATCH',
+                            headers: { Authorization: `Bearer ${this.getToken()}` },
+                        });
+                    }
+                    catch (_a) {
+                        // non-fatal — session will expire naturally
+                    }
+                    this.aiModule.clearSessionToken();
+                }
+            }
+            this.accomplishmentController.emitAccomplishment(payload);
+        });
     }
     /**
      * Subscribe to an accomplishment.

package/package.json

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