@sprucelabs/sprucebot-llm 2.3.0 → 2.4.0

package/build/bots/SprucebotLlmBotImpl.d.ts

@@ -1,18 +1,21 @@
 import { AbstractEventEmitter } from '@sprucelabs/mercury-event-emitter';
 import { Schema, SchemaValues } from '@sprucelabs/schema';
-import { BotOptions, LlmEventContract, LlmMessage, SerializedBot, SprucebotLlmBot } from '../llm.types';
+import { BotOptions, LlmEventContract, LlmMessage, SerializedBot, SprucebotLlmBot, SprucebotLLmSkill } from '../llm.types';
 export default class SprucebotLlmBotImpl<StateSchema extends Schema = Schema, State extends SchemaValues<StateSchema> = SchemaValues<StateSchema>> extends AbstractEventEmitter<LlmEventContract> implements SprucebotLlmBot<StateSchema, State> {
+    static messageMemoryLimit: number;
     private adapter;
     private youAre;
     private stateSchema?;
     protected state?: Partial<State>;
     private isDone;
     protected messages: LlmMessage[];
-    private skill?;
+    protected skill?: SprucebotLLmSkill;
     constructor(options: BotOptions<StateSchema, State>);
     markAsDone(): void;
     getIsDone(): boolean;
     serialize(): SerializedBot<StateSchema, State>;
     sendMessage(message: string): Promise<string>;
+    private trackMessage;
     updateState(newState: Partial<State>): Promise<void>;
+    setSkill(skill: SprucebotLLmSkill<any>): void;
 }

package/build/bots/SprucebotLlmBotImpl.js

@@ -41,7 +41,7 @@ class SprucebotLlmBotImpl extends mercury_event_emitter_1.AbstractEventEmitter {
     async sendMessage(message) {
         var _a, _b;
         (0, schema_1.assertOptions)({ message }, ['message']);
-        this.messages.push({
+        this.trackMessage({
             from: 'Me',
             message,
         });
@@ -55,15 +55,25 @@ class SprucebotLlmBotImpl extends mercury_event_emitter_1.AbstractEventEmitter {
         else if (state) {
             await ((_b = this.skill) === null || _b === void 0 ? void 0 : _b.updateState(state));
         }
-        this.messages.push({
+        this.trackMessage({
             from: 'You',
             message: parsedResponse,
         });
         return parsedResponse;
     }
+    trackMessage(m) {
+        if (this.messages.length === SprucebotLlmBotImpl.messageMemoryLimit) {
+            this.messages.shift();
+        }
+        this.messages.push(m);
+    }
     async updateState(newState) {
         this.state = Object.assign(Object.assign({}, this.state), newState);
         await this.emit('did-update-state');
     }
+    setSkill(skill) {
+        this.skill = skill;
+    }
 }
+SprucebotLlmBotImpl.messageMemoryLimit = 10;
 exports.default = SprucebotLlmBotImpl;

package/build/esm/bots/SprucebotLlmBotImpl.d.ts

@@ -1,18 +1,21 @@
 import { AbstractEventEmitter } from '@sprucelabs/mercury-event-emitter';
 import { Schema, SchemaValues } from '@sprucelabs/schema';
-import { BotOptions, LlmEventContract, LlmMessage, SerializedBot, SprucebotLlmBot } from '../llm.types';
+import { BotOptions, LlmEventContract, LlmMessage, SerializedBot, SprucebotLlmBot, SprucebotLLmSkill } from '../llm.types';
 export default class SprucebotLlmBotImpl<StateSchema extends Schema = Schema, State extends SchemaValues<StateSchema> = SchemaValues<StateSchema>> extends AbstractEventEmitter<LlmEventContract> implements SprucebotLlmBot<StateSchema, State> {
+    static messageMemoryLimit: number;
     private adapter;
     private youAre;
     private stateSchema?;
     protected state?: Partial<State>;
     private isDone;
     protected messages: LlmMessage[];
-    private skill?;
+    protected skill?: SprucebotLLmSkill;
     constructor(options: BotOptions<StateSchema, State>);
     markAsDone(): void;
     getIsDone(): boolean;
     serialize(): SerializedBot<StateSchema, State>;
     sendMessage(message: string): Promise<string>;
+    private trackMessage;
     updateState(newState: Partial<State>): Promise<void>;
+    setSkill(skill: SprucebotLLmSkill<any>): void;
 }

package/build/esm/bots/SprucebotLlmBotImpl.js

@@ -11,7 +11,7 @@ import { AbstractEventEmitter } from '@sprucelabs/mercury-event-emitter';
 import { assertOptions, defaultSchemaValues, } from '@sprucelabs/schema';
 import { llmEventContract, } from '../llm.types.js';
 import ResponseParser from '../parsingResponses/ResponseParser.js';
-export default class SprucebotLlmBotImpl extends AbstractEventEmitter {
+class SprucebotLlmBotImpl extends AbstractEventEmitter {
     constructor(options) {
         const { adapter, youAre, stateSchema, state, skill } = options;
         super(llmEventContract);
@@ -46,7 +46,7 @@ export default class SprucebotLlmBotImpl extends AbstractEventEmitter {
         var _a, _b;
         return __awaiter(this, void 0, void 0, function* () {
             assertOptions({ message }, ['message']);
-            this.messages.push({
+            this.trackMessage({
                 from: 'Me',
                 message,
             });
@@ -60,17 +60,28 @@ export default class SprucebotLlmBotImpl extends AbstractEventEmitter {
             else if (state) {
                 yield ((_b = this.skill) === null || _b === void 0 ? void 0 : _b.updateState(state));
             }
-            this.messages.push({
+            this.trackMessage({
                 from: 'You',
                 message: parsedResponse,
             });
             return parsedResponse;
         });
     }
+    trackMessage(m) {
+        if (this.messages.length === SprucebotLlmBotImpl.messageMemoryLimit) {
+            this.messages.shift();
+        }
+        this.messages.push(m);
+    }
     updateState(newState) {
         return __awaiter(this, void 0, void 0, function* () {
             this.state = Object.assign(Object.assign({}, this.state), newState);
             yield this.emit('did-update-state');
         });
     }
+    setSkill(skill) {
+        this.skill = skill;
+    }
 }
+SprucebotLlmBotImpl.messageMemoryLimit = 10;
+export default SprucebotLlmBotImpl;

package/build/esm/llm.types.d.ts

@@ -11,6 +11,7 @@ export interface SprucebotLlmBot<StateSchema extends Schema = Schema, State exte
     sendMessage(message: string): Promise<string>;
     serialize(): SerializedBot<StateSchema, State>;
     updateState(state: Partial<State>): Promise<void>;
+    setSkill(skill: SprucebotLLmSkill<any>): void;
 }
 export interface LlmAdapter {
     sendMessage(bot: SprucebotLlmBot<Schema>): Promise<string>;

package/build/llm.types.d.ts

@@ -11,6 +11,7 @@ export interface SprucebotLlmBot<StateSchema extends Schema = Schema, State exte
     sendMessage(message: string): Promise<string>;
     serialize(): SerializedBot<StateSchema, State>;
     updateState(state: Partial<State>): Promise<void>;
+    setSkill(skill: SprucebotLLmSkill<any>): void;
 }
 export interface LlmAdapter {
     sendMessage(bot: SprucebotLlmBot<Schema>): Promise<string>;

package/package.json

@@ -9,7 +9,7 @@
       "@sprucelabs/spruce-test-fixtures"
     ]
   },
-  "version": "2.3.0",
+  "version": "2.4.0",
   "files": [
     "build"
   ],