@wabot-dev/framework 0.4.3 → 0.5.0

package/dist/src/addon/chat-controller/cmd/CmdChannel.js

@@ -54,12 +54,12 @@ let CmdChannel = CmdChannel_1 = class CmdChannel {
                     this.auth.assign(authInfo);
                 }
             }
-            this.callBack({
+            await this.callBack({
                 chatConnection,
                 message: {
                     text: trimmedInput,
                 },
-                reply: (message) => {
+                reply: async (message) => {
                     console.log(`\n[${message.senderName}]: ${message.text}\n`);
                     this.rl.prompt();
                     if (this.auth.isAssigned()) {

package/dist/src/addon/chat-controller/socket/SocketChannel.js

@@ -48,7 +48,7 @@ let SocketChannel = SocketChannel_1 = class SocketChannel {
             constructor(auth) {
                 this.auth = auth;
             }
-            onMessage(message, socket) {
+            async onMessage(message, socket) {
                 if (!channel.callBack)
                     return;
                 const trimmedInput = message.text.trim();
@@ -60,13 +60,13 @@ let SocketChannel = SocketChannel_1 = class SocketChannel {
                     chatType: 'PRIVATE',
                     channelName: SocketChannel_1.name,
                 };
-                channel.callBack({
+                await channel.callBack({
                     chatConnection,
                     message: {
                         text: message.text,
                         senderName: message.senderName,
                     },
-                    reply: (message) => {
+                    reply: async (message) => {
                         socket.emit('message', message);
                     },
                     injectInstances: [
@@ -80,7 +80,7 @@ let SocketChannel = SocketChannel_1 = class SocketChannel {
             onSocketEvent('message'),
             __metadata("design:type", Function),
             __metadata("design:paramtypes", [SocketChannelReceivedMessage, Socket]),
-            __metadata("design:returntype", void 0)
+            __metadata("design:returntype", Promise)
         ], SocketChannelController.prototype, "onMessage", null);
         SocketChannelController = __decorate([
             socketController(channel.config.namespace),

package/dist/src/addon/chat-controller/telegram/TelegramChannel.js

@@ -21,14 +21,16 @@ let TelegramChannel = TelegramChannel_1 = class TelegramChannel {
                 chatType: ctx.message.chat.type === 'private' ? 'PRIVATE' : 'GROUP',
                 channelName: TelegramChannel_1.name,
             };
-            callback({
+            await callback({
                 chatConnection,
                 message: {
                     senderName: ctx.from.first_name,
                     text: ctx.message.text,
                 },
-                reply: (replyMessage) => {
-                    replyMessage.text && ctx.reply(replyMessage.text);
+                reply: async (replyMessage) => {
+                    if (!replyMessage.text)
+                        return;
+                    await ctx.reply(replyMessage.text);
                 },
             });
         });

package/dist/src/addon/chat-controller/whatsapp/WhatsAppChannel.js

@@ -20,11 +20,11 @@ let WhatsAppChannel = class WhatsAppChannel {
             to: this.config.number,
             listener: async (message) => {
                 try {
-                    callback({
+                    await callback({
                         chatConnection: message.chatConnection,
                         message: message.message,
-                        reply: (replyMessage) => {
-                            this.sender.sendWhatsApp({
+                        reply: async (replyMessage) => {
+                            await this.sender.sendWhatsApp({
                                 from: this.config.number,
                                 to: message.chatConnection.id,
                                 message: replyMessage,

package/dist/src/feature/chat-bot/Chat.js

@@ -24,6 +24,15 @@ class Chat extends Entity {
         }
         return false;
     }
+    getConnectionByChannel(channelName) {
+        return this.data.connections.find((a) => a.channelName === channelName) ?? null;
+    }
+    addConnection(connection) {
+        if (this.hasConnection(connection)) {
+            throw new Error('Connection already exists');
+        }
+        this.data.connections.push(connection);
+    }
     hasAssociation(type, id) {
         return this.data.associations?.some((a) => a.type === type && a.id === id) ?? false;
     }

package/dist/src/feature/chat-bot/ChatBot.js

@@ -21,10 +21,10 @@ let ChatBot = class ChatBot {
             humanMessage: message,
         });
         await this.memory.create(newChatItem);
-        this.processLoop(callback);
+        await this.processLoop(callback);
     }
     async processLoop(callback) {
-        const prevItems = await this.memory.findLastItems(10);
+        const prevItems = await this.memory.findLastItems(16);
         if (prevItems.length === 0) {
             return;
         }
@@ -57,13 +57,13 @@ let ChatBot = class ChatBot {
             const newChatItem = new ChatItem(newItemData);
             await this.memory.create(newChatItem);
             if (newItemData.type === 'botMessage') {
-                callback(newChatItem.botMessage);
+                await callback(newChatItem.botMessage);
             }
         }
         if (newItemsData.length == 0 || newItemsData[newItemsData.length - 1].type === 'botMessage') {
             return;
         }
-        this.processLoop(callback);
+        await this.processLoop(callback);
     }
 };
 ChatBot = __decorate([

package/dist/src/feature/mindset/IMindset.js

@@ -1,4 +1,7 @@
 class Mindset {
+    context() {
+        throw new Error('Method not implemented.');
+    }
     identity() {
         throw new Error('Method not implemented.');
     }

package/dist/src/feature/mindset/MindsetOperator.js

@@ -16,6 +16,9 @@ let MindsetOperator = class MindsetOperator {
         this.container = container;
         this.metadata = metadataStore.getMindsetInfo(this.mindset.constructor);
     }
+    context() {
+        return this.mindset.context();
+    }
     identity() {
         return this.mindset.identity();
     }
@@ -32,7 +35,8 @@ let MindsetOperator = class MindsetOperator {
         return this.mindset.llms();
     }
     async systemPrompt() {
-        let [identity, skills, limits, workflow] = await Promise.all([
+        let [context, identity, skills, limits, workflow] = await Promise.all([
+            this.context(),
             this.identity(),
             this.skills(),
             this.limits(),
@@ -40,33 +44,35 @@ let MindsetOperator = class MindsetOperator {
         ]);
         const language = identity.language.replaceAll('#', ' ');
         const name = identity.name.replaceAll('#', ' ');
-        const age = identity.age ? identity.age.toString().replaceAll('#', ' ') : null;
         const personality = identity.personality ? identity.personality.replaceAll('#', ' ') : null;
+        context = context.replaceAll('#', ' ');
         skills = skills.replaceAll('#', ' ');
         limits = limits.replaceAll('#', ' ');
         workflow = workflow.replaceAll('#', ' ');
         const systemPrompt = `
-         # System Instructions
-         you should act as a assistant.
-         your main language is ${language}.
-         your name is ${name}.
-         ${age ? 'you are ' + age + ' years old.' : ''}
-         
-          ${personality ? '## Personality (in your main language) \n' + personality : ''}
-  
-          ## Skills (in your main language)
-          ${skills}
+      # System Instructions
+      you should act as a assistant.
+      your main language is ${language}.
+      your name is ${name}.
 
-          ## Workflow (in your main language)
-          ${workflow}
-  
-          ## System limitations (in your main language)
-          ${limits}
-  
-          ## Chat memory
-          Next you will receive a chat history,
-          you should use this information to answer the user.
-      `;
+      ${personality ? '## Personality (in your main language) \n' + personality : ''}
+
+      ## Context (in your main language)
+      ${context}
+
+      ## Skills (in your main language)
+      ${skills}
+
+      ## Workflow (in your main language)
+      ${workflow}
+
+      ## System limitations (in your main language)
+      ${limits}
+
+      ## Chat memory
+      Next you will receive a chat history,
+      you should use this information to answer the user.
+    `;
         return systemPrompt;
     }
     tools() {

package/dist/src/index.d.ts

@@ -597,6 +597,12 @@ declare class Chat extends Entity<IChatData> {
         id: string;
     }[];
     hasConnection(connection: IChatConnection): boolean;
+    getConnectionByChannel(channelName: string): {
+        chatType: "GROUP" | "PRIVATE";
+        channelName: string;
+        id: string;
+    } | null;
+    addConnection(connection: IChatConnection): void;
     hasAssociation(type: string, id: string): boolean;
     getAssociationsByType(type: string): {
         type: string;
@@ -611,7 +617,6 @@ declare class Chat extends Entity<IChatData> {
 interface IMindsetIdentity {
     name: string;
     language: string;
-    age?: number;
     personality?: string;
     emotions?: string;
 }
@@ -620,6 +625,7 @@ interface IMindsetLlm {
     model: string;
 }
 interface IMindset {
+    context(): Promise<string>;
     identity(): Promise<IMindsetIdentity>;
     skills(): Promise<string>;
     limits(): Promise<string>;
@@ -627,6 +633,7 @@ interface IMindset {
     llms(): Promise<IMindsetLlm[]>;
 }
 declare class Mindset implements IMindset {
+    context(): Promise<string>;
     identity(): Promise<IMindsetIdentity>;
     skills(): Promise<string>;
     limits(): Promise<string>;
@@ -725,6 +732,7 @@ declare class MindsetOperator implements IMindset {
     private container;
     private metadata;
     constructor(mindset: Mindset, container: Container, metadataStore: MindsetMetadataStore);
+    context(): Promise<string>;
     identity(): Promise<IMindsetIdentity>;
     skills(): Promise<string>;
     limits(): Promise<string>;
@@ -758,6 +766,8 @@ interface IChatMessage {
     senderName?: string;
     text?: string;
     images?: IChatMessageImage[];
+    object?: object;
+    metadata?: Record<string, string>;
 }
 
 interface IFunctionCall {
@@ -834,7 +844,7 @@ declare class ChatMemory implements IChatMemory {
 }
 
 interface IChatBot {
-    sendMessage(message: IChatMessage, callback: (message: IChatMessage) => void): void | Promise<void>;
+    sendMessage(message: IChatMessage, callback: (message: IChatMessage) => Promise<void>): void | Promise<void>;
 }
 
 declare class ChatBot implements IChatBot {
@@ -842,8 +852,8 @@ declare class ChatBot implements IChatBot {
     private adapter;
     private mindset;
     constructor(memory: ChatMemory, adapter: ChatAdapter, mindset: MindsetOperator);
-    sendMessage(message: IChatMessage, callback: (message: IChatMessage) => void): Promise<void>;
-    protected processLoop(callback: (message: IChatMessage) => void): Promise<void>;
+    sendMessage(message: IChatMessage, callback: (message: IChatMessage) => Promise<void>): Promise<void>;
+    protected processLoop(callback: (message: IChatMessage) => Promise<void>): Promise<void>;
 }
 
 interface IChatRepository {
@@ -885,7 +895,7 @@ interface IChatControllerMetadata {
 
 interface IReceivedMessage {
     message: IChatMessage;
-    reply: (message: IChatMessage) => void;
+    reply: (message: IChatMessage) => Promise<Record<string, string> | void>;
 }
 
 interface IChannelMessage extends IReceivedMessage {
@@ -894,7 +904,7 @@ interface IChannelMessage extends IReceivedMessage {
 }
 
 interface IChatChannel {
-    listen(callback: (received: IChannelMessage) => void): void;
+    listen(callback: (received: IChannelMessage) => Promise<void>): void;
     connect(): void;
 }
 
@@ -1505,7 +1515,7 @@ declare class CmdChannel implements IChatChannel {
     private rl;
     private callBack;
     constructor(auth: Auth<any>);
-    listen(callback: (message: IChannelMessage) => void): void;
+    listen(callback: (message: IChannelMessage) => Promise<void>): void;
     connect(): void;
 }
 declare function writeJsonToFile<T>(filename: string, data: T): void;
@@ -1540,7 +1550,7 @@ declare class SocketChannel implements IChatChannel {
     private controller;
     constructor(config: SocketChannelConfig);
     private configController;
-    listen(callback: (message: IChannelMessage) => void): void;
+    listen(callback: (message: IChannelMessage) => Promise<void>): void;
     connect(): void;
 }
 
@@ -1558,7 +1568,7 @@ declare class TelegramChannel implements IChatChannel {
     private config;
     private bot;
     constructor(config: TelegramChannelConfig);
-    listen(callback: (message: IChannelMessage) => void): void;
+    listen(callback: (message: IChannelMessage) => Promise<void>): void;
     connect(): void;
 }
 
@@ -1723,7 +1733,7 @@ declare class WhatsAppChannel implements IChatChannel {
     private receiver;
     private logger;
     constructor(config: WhatsappChannelConfig, sender: WhatsAppSender, receiver: WhatsAppReceiver);
-    listen(callback: (message: IChannelMessage) => void): void;
+    listen(callback: (message: IChannelMessage) => Promise<void>): void;
     connect(): void;
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wabot-dev/framework",
-  "version": "0.4.3",
+  "version": "0.5.0",
   "description": "Framework for IA Chat Bots",
   "type": "module",
   "main": "dist/src/index.js",