@rimori/client 2.5.14 → 2.5.15

package/dist/modules.d.ts

@@ -0,0 +1,4 @@
+export * from './plugin/module/AIModule';
+export * from './fromRimori/PluginTypes';
+export * from './fromRimori/EventBus';
+export * from './cli/types/DatabaseTypes';

package/dist/modules.js

@@ -0,0 +1,5 @@
+// Re-export everything
+export * from './plugin/module/AIModule';
+export * from './fromRimori/PluginTypes';
+export * from './fromRimori/EventBus';
+export * from './cli/types/DatabaseTypes';

package/dist/plugin/RimoriClient.js

@@ -29,7 +29,7 @@ export class RimoriClient {
         };
         this.rimoriInfo = info;
         this.sharedContent = new SharedContentController(supabase, this);
-        this.ai = new AIModule(controller, info);
+        this.ai = new AIModule(info.backendUrl, () => this.rimoriInfo.token);
         this.event = new EventModule(info.pluginId);
         this.db = new DbModule(supabase, controller, info);
         this.plugin = new PluginModule(supabase, controller, info, this.ai);

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

@@ -1,5 +1,4 @@
 import { Tool } from '../../fromRimori/PluginTypes';
-import { RimoriCommunicationHandler, RimoriInfo } from '../CommunicationHandler';
 import { Language } from '../../controller/SettingsController';
 export type OnStreamedObjectResult<T = any> = (result: T, isLoading: boolean) => void;
 type PrimitiveType = 'string' | 'number' | 'boolean';
@@ -21,7 +20,7 @@ interface ObjectToolParameter {
  * The value is the parameter of the tool.
  *
  */
-export type ObjectTool = {
+export type AIObjectTool = {
     [key: string]: ObjectToolParameter;
 };
 export interface ToolInvocation {
@@ -40,7 +39,7 @@ export interface ObjectRequest {
     /**
      * The tools that the AI can use.
      */
-    tool: ObjectTool;
+    tool: AIObjectTool;
     /**
      * High level instructions for the AI to follow. Behaviour, tone, restrictions, etc.
      * Example: "Act like a recipe writer."
@@ -57,10 +56,9 @@ export interface ObjectRequest {
  * Provides access to text generation, voice synthesis, and object generation.
  */
 export declare class AIModule {
-    private communicationHandler;
+    private getToken;
     private backendUrl;
-    private token;
-    constructor(communicationHandler: RimoriCommunicationHandler, info: RimoriInfo);
+    constructor(backendUrl: string, getToken: () => string);
     /**
      * Generate text from messages using AI.
      * @param messages The messages to generate text from.
@@ -78,7 +76,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): Promise<void>;
+    getSteamedText(messages: Message[], onMessage: OnLLMResponse, tools?: Tool[], cache?: boolean, model?: string, knowledgeId?: string): Promise<void>;
     /**
      * Generate voice audio from text using AI.
      * @param text The text to convert to voice.
@@ -110,11 +108,12 @@ export declare class AIModule {
      */
     getObject<T = any>(params: {
         systemPrompt: string;
-        responseSchema: ObjectTool;
+        responseSchema: AIObjectTool;
         userPrompt?: string;
         cache?: boolean;
         tools?: Tool[];
         model?: string;
+        knowledgeId?: string;
     }): Promise<T>;
     /**
      * Generate a streamed structured object from a request using AI.
@@ -126,15 +125,17 @@ export declare class AIModule {
      * @param request.cache Whether to cache the result (default: false).
      * @param request.tools The tools to use for generation.
      * @param request.model The model to use for generation.
+     * @param request.knowledgeId Optional knowledge entry ID to ground AI content in real facts.
      */
     getStreamedObject<T = any>(params: {
         systemPrompt: string;
-        responseSchema: ObjectTool;
+        responseSchema: AIObjectTool;
         userPrompt?: string;
         onResult: OnStreamedObjectResult<T>;
         cache?: boolean;
         tools?: Tool[];
         model?: string;
+        knowledgeId?: string;
     }): Promise<void>;
     private streamObject;
     private sendToolResult;

package/dist/plugin/module/AIModule.js

@@ -12,13 +12,9 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
  * Provides access to text generation, voice synthesis, and object generation.
  */
 export class AIModule {
-    constructor(communicationHandler, info) {
-        this.token = info.token;
-        this.backendUrl = info.backendUrl;
-        this.communicationHandler = communicationHandler;
-        this.communicationHandler.onUpdate((updatedInfo) => {
-            this.token = updatedInfo.token;
-        });
+    constructor(backendUrl, getToken) {
+        this.backendUrl = backendUrl;
+        this.getToken = getToken;
     }
     /**
      * Generate text from messages using AI.
@@ -53,13 +49,14 @@ export class AIModule {
      * @param model The model to use for generation.
      */
     getSteamedText(messages_1, onMessage_1, tools_1) {
-        return __awaiter(this, arguments, void 0, function* (messages, onMessage, tools, cache = false, model) {
+        return __awaiter(this, arguments, void 0, function* (messages, onMessage, tools, cache = false, model, knowledgeId) {
             const messageId = Math.random().toString(36).substring(3);
             const { result } = yield this.streamObject({
                 cache,
                 tools,
                 model,
                 messages,
+                knowledgeId,
                 responseSchema: {
                     result: {
                         type: 'string',
@@ -85,7 +82,7 @@ export class AIModule {
                 method: 'POST',
                 headers: {
                     'Content-Type': 'application/json',
-                    Authorization: `Bearer ${this.token}`,
+                    Authorization: `Bearer ${this.getToken()}`,
                 },
                 body: JSON.stringify({ input: text, voice, speed, language, cache }),
             }).then((r) => r.blob());
@@ -106,7 +103,7 @@ export class AIModule {
             }
             return yield fetch(`${this.backendUrl}/voice/stt`, {
                 method: 'POST',
-                headers: { Authorization: `Bearer ${this.token}` },
+                headers: { Authorization: `Bearer ${this.getToken()}` },
                 body: formData,
             })
                 .then((r) => r.json())
@@ -136,13 +133,14 @@ export class AIModule {
      */
     getObject(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const { systemPrompt, responseSchema, userPrompt, cache = false, tools = [], model = undefined } = params;
+            const { systemPrompt, responseSchema, userPrompt, cache = false, tools = [], model = undefined, knowledgeId, } = params;
             return yield this.streamObject({
                 responseSchema,
                 messages: this.getChatMessage(systemPrompt, userPrompt),
                 cache,
                 tools,
                 model,
+                knowledgeId,
             });
         });
     }
@@ -156,10 +154,11 @@ export class AIModule {
      * @param request.cache Whether to cache the result (default: false).
      * @param request.tools The tools to use for generation.
      * @param request.model The model to use for generation.
+     * @param request.knowledgeId Optional knowledge entry ID to ground AI content in real facts.
      */
     getStreamedObject(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const { systemPrompt, responseSchema, userPrompt, onResult, cache = false, tools = [], model = undefined } = params;
+            const { systemPrompt, responseSchema, userPrompt, onResult, cache = false, tools = [], model = undefined, knowledgeId, } = params;
             yield this.streamObject({
                 responseSchema,
                 messages: this.getChatMessage(systemPrompt, userPrompt),
@@ -167,12 +166,13 @@ export class AIModule {
                 cache,
                 tools,
                 model,
+                knowledgeId,
             });
         });
     }
     streamObject(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const { messages, responseSchema, onResult = () => null, cache = false, tools = [], model = undefined } = params;
+            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`, {
                 body: JSON.stringify({
@@ -182,9 +182,10 @@ export class AIModule {
                     responseSchema,
                     messages: chatMessages,
                     model,
+                    knowledge_id: knowledgeId,
                 }),
                 method: 'POST',
-                headers: { Authorization: `Bearer ${this.token}`, 'Content-Type': 'application/json' },
+                headers: { Authorization: `Bearer ${this.getToken()}`, 'Content-Type': 'application/json' },
             });
             if (!response.ok) {
                 throw new Error(`Failed to stream object: ${response.status} ${response.statusText}`);
@@ -260,7 +261,7 @@ export class AIModule {
                     toolCallId,
                     result: result !== null && result !== void 0 ? result : '[DONE]',
                 }),
-                headers: { Authorization: `Bearer ${this.token} `, 'Content-Type': 'application/json' },
+                headers: { Authorization: `Bearer ${this.getToken()} `, 'Content-Type': 'application/json' },
             });
         });
     }

package/package.json

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