@wabot-dev/framework 0.1.0-beta.32 → 0.1.0-beta.34

package/dist/src/addon/chat-bot/claude/ClaudeChatAdapter.js

@@ -94,12 +94,13 @@ class ClaudeChatAdapter {
         };
     }
     mapResponse(response) {
+        let chatItem;
         const content = response.content[0];
         if (content.type === 'text') {
-            return { type: 'botMessage', botMessage: { text: content.text } };
+            chatItem = { type: 'botMessage', botMessage: { text: content.text } };
         }
         else if (content.type === 'tool_use') {
-            return {
+            chatItem = {
                 type: 'functionCall',
                 functionCall: {
                     id: content.id,
@@ -111,6 +112,17 @@ class ClaudeChatAdapter {
         else {
             throw new Error('Not supported Claude Response');
         }
+        let usage;
+        if (response.usage) {
+            usage = {
+                inputTokens: response.usage.input_tokens,
+                outputTokens: response.usage.output_tokens,
+            };
+        }
+        else {
+            throw new Error('Unable to found usage info');
+        }
+        return { chatItem, usage };
     }
 }
 

package/dist/src/addon/chat-bot/deepseek/DeepSeekChatAdapter.js

@@ -102,12 +102,13 @@ class DeepSeekChatAdapter {
         };
     }
     mapResponse(response) {
+        let chatItem;
         const { tool_calls: responseFunctionCall, content: responseText } = response.choices?.[0]?.message ?? {};
         if (responseText) {
-            return { type: 'botMessage', botMessage: { text: responseText } };
+            chatItem = { type: 'botMessage', botMessage: { text: responseText } };
         }
         else if (responseFunctionCall && responseFunctionCall[0]?.type === 'function') {
-            return {
+            chatItem = {
                 type: 'functionCall',
                 functionCall: {
                     id: responseFunctionCall[0].id,
@@ -119,6 +120,17 @@ class DeepSeekChatAdapter {
         else {
             throw new Error('Not supported DeepSeek Response');
         }
+        let usage;
+        if (response.usage) {
+            usage = {
+                inputTokens: response.usage.prompt_tokens,
+                outputTokens: response.usage.completion_tokens,
+            };
+        }
+        else {
+            throw new Error('Unable to found usage info');
+        }
+        return { chatItem, usage };
     }
 }
 

package/dist/src/addon/chat-bot/gemini/GeminiChatAdapter.js

@@ -0,0 +1,125 @@
+import { Logger } from '../../../core/logger/Logger.js';
+import { GoogleGenAI } from '../../../node_modules/@google/genai/dist/index.js';
+
+const SUPPORTED_MODELS = [
+    'gemini-2.0-flash-exp',
+    'gemini-1.5-flash',
+    'gemini-1.5-pro',
+    'gemini-pro',
+];
+const DEFAULT_MODEL = 'gemini-1.5-flash';
+class GeminiChatAdapter {
+    genai;
+    model;
+    logger = new Logger('wabot:gemini-chat-adapter');
+    constructor() {
+        const apiKey = process.env.GEMINI_API_KEY;
+        if (!apiKey) {
+            throw new Error('GEMINI_API_KEY env variable is required');
+        }
+        this.model = process.env.GEMINI_MODEL || DEFAULT_MODEL;
+        this.validateModel(this.model);
+        this.genai = new GoogleGenAI({ apiKey });
+    }
+    validateModel(model) {
+        if (!SUPPORTED_MODELS.includes(model)) {
+            throw new Error(`Unsupported Gemini model: ${model}. Supported models: ${SUPPORTED_MODELS.join(', ')}`);
+        }
+    }
+    async nextItem(req) {
+        const contents = this.buildContents(req.prevItems, req.systemPrompt);
+        const tools = req.tools.length > 0 ? req.tools.map(this.mapTool) : undefined;
+        const request = { model: this.model, contents, tools };
+        this.logger.debug(`Call Gemini API with Request: ${JSON.stringify(request)}`);
+        const response = await this.genai.models.generateContent(request);
+        return this.mapResponse(response);
+    }
+    buildContents(chatItems, systemPrompt) {
+        const contents = [];
+        if (systemPrompt) {
+            contents.push({ role: 'user', parts: [{ text: systemPrompt }] }, { role: 'model', parts: [{ text: 'I understand. I will follow these instructions.' }] });
+        }
+        chatItems.forEach(chatItem => {
+            switch (chatItem.type) {
+                case 'humanMessage':
+                    this.validateMessageContent(chatItem.humanMessage.text, 'User');
+                    contents.push({ role: 'user', parts: [{ text: chatItem.humanMessage.text }] });
+                    break;
+                case 'botMessage':
+                    this.validateMessageContent(chatItem.botMessage.text, 'Assistant');
+                    contents.push({ role: 'model', parts: [{ text: chatItem.botMessage.text }] });
+                    break;
+                case 'functionCall':
+                    contents.push(...this.mapFunctionCall(chatItem.functionCall));
+                    break;
+            }
+        });
+        return contents;
+    }
+    validateMessageContent(text, messageType) {
+        if (!text) {
+            throw new Error(`${messageType} message content is empty`);
+        }
+    }
+    mapFunctionCall(item) {
+        const args = JSON.parse(item.arguments || '{}');
+        return [
+            {
+                role: 'model',
+                parts: [{ functionCall: { name: item.name, args } }],
+            },
+            {
+                role: 'user',
+                parts: [{ functionResponse: { name: item.name, response: { result: item.result || 'No result' } } }],
+            },
+        ];
+    }
+    mapTool = (tool) => ({
+        functionDeclarations: [{
+                name: tool.name,
+                description: tool.description,
+                parameters: {
+                    type: 'object',
+                    properties: tool.parameters.reduce((acc, param) => ({ ...acc, [param.name]: { type: param.type, description: param.description } }), {}),
+                    required: tool.parameters.map(param => param.name),
+                },
+            }],
+    });
+    mapResponse(response) {
+        this.validateResponse(response);
+        const part = response.response.candidates[0].content.parts[0];
+        const chatItem = part.text
+            ? { type: 'botMessage', botMessage: { text: part.text } }
+            : part.functionCall
+                ? {
+                    type: 'functionCall',
+                    functionCall: {
+                        id: `gemini_${Date.now()}`,
+                        name: part.functionCall.name,
+                        arguments: JSON.stringify(part.functionCall.args || {}),
+                    },
+                }
+                : (() => { throw new Error('Not supported Gemini Response'); })();
+        const usage = {
+            inputTokens: response.response.usageMetadata?.promptTokenCount || 0,
+            outputTokens: response.response.usageMetadata?.candidatesTokenCount || 0,
+        };
+        if (!response.response.usageMetadata) {
+            throw new Error('Unable to found usage info');
+        }
+        return { chatItem, usage };
+    }
+    validateResponse(response) {
+        if (!response.response) {
+            throw new Error('Invalid Gemini response structure');
+        }
+        if (!response.response.candidates?.[0]) {
+            throw new Error('No candidates in Gemini response');
+        }
+        if (!response.response.candidates[0].content?.parts?.[0]) {
+            throw new Error('No parts in Gemini response');
+        }
+    }
+}
+
+export { GeminiChatAdapter };

package/dist/src/addon/chat-bot/openia/OpenaiChatAdapter.js

@@ -77,12 +77,12 @@ class OpenaiChatAdapter {
         };
     }
     mapResponse(response) {
-        let newItem;
+        let chatItem;
         if (response.output_text) {
-            newItem = { type: 'botMessage', botMessage: { text: response.output_text } };
+            chatItem = { type: 'botMessage', botMessage: { text: response.output_text } };
         }
         else if (response.output && response.output[0]?.type == 'function_call') {
-            newItem = {
+            chatItem = {
                 type: 'functionCall',
                 functionCall: {
                     id: response.output[0].call_id,
@@ -94,7 +94,17 @@ class OpenaiChatAdapter {
         else {
             throw new Error('Not supported OpenIA Response');
         }
-        return newItem;
+        let usage;
+        if (response.usage) {
+            usage = {
+                inputTokens: response.usage.input_tokens,
+                outputTokens: response.usage.output_tokens,
+            };
+        }
+        else {
+            throw new Error('Unable to found usage info');
+        }
+        return { chatItem, usage };
     }
 }
 

package/dist/src/core/validation/validators/validateArray.js

@@ -4,6 +4,16 @@ function validateArray(value, options) {
             error: { description: 'Should be an array', items: [] },
         };
     }
+    if (options?.minLength != null && value.length < options.minLength) {
+        return {
+            error: { description: 'exceeds the established min length limit', items: [] },
+        };
+    }
+    if (options?.maxLength != null && value.length > options.maxLength) {
+        return {
+            error: { description: 'exceeds the established max length limit', items: [] },
+        };
+    }
     const { itemsValidator } = options ?? {};
     const valueOut = [];
     const errorItems = [];

package/dist/src/core/validation/validators/validateModel.js

@@ -7,7 +7,9 @@ function validateModel(value, info) {
     for (const propertyName in info.properties) {
         const propertyInfo = info.properties[propertyName];
         const propertyValidators = propertyInfo.validators ?? [];
-        resultValue[propertyName] = value[propertyName] ?? resultValue[propertyName];
+        resultValue[propertyName] = propertyInfo.isOptional
+            ? (value[propertyName] ?? resultValue[propertyName])
+            : value[propertyName];
         if (resultValue[propertyName] == null && propertyInfo.isOptional) {
             resultValue[propertyName] = undefined;
             continue;

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

@@ -1,6 +1,6 @@
 class ChatAdapter {
     nextItem(req) {
-        throw new Error("Method not implemented.");
+        throw new Error('Method not implemented.');
     }
 }
 

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

@@ -28,7 +28,7 @@ class ChatBot {
         }
         const systemPrompt = await this.mindset.systemPrompt();
         const tools = this.mindset.tools();
-        const newItemData = await this.adapter.nextItem({
+        const { chatItem: newItemData } = await this.adapter.nextItem({
             model: 'gpt',
             systemPrompt,
             tools,

package/dist/src/index.d.ts

@@ -195,6 +195,8 @@ type IModelValidatorsInfo<V> = {
 };
 
 interface IValidateArrayOptions {
+    minLength?: number;
+    maxLength?: number;
 }
 interface IValidateArrayOptionsWithItemsValidators extends IValidateArrayOptions {
     itemsValidator?: {
@@ -514,18 +516,27 @@ type IFunctionCallItem = {
 type IChatItem = IBotMessageItem | IHumanMessageItem | IFunctionCallItem;
 type IChatItemType = (typeof chatItemTypeOptions)[number];
 
+interface ILanguageModelUsage {
+    inputTokens: number;
+    outputTokens: number;
+}
+
 interface IChatAdapterNextItemReq {
     model: string;
     systemPrompt: string;
     tools: IMindsetTool[];
     prevItems: IChatItem[];
 }
+interface IChatAdapterNextItemRes {
+    chatItem: IChatItem;
+    usage: ILanguageModelUsage;
+}
 interface IChatAdapter {
-    nextItem(req: IChatAdapterNextItemReq): Promise<IChatItem>;
+    nextItem(req: IChatAdapterNextItemReq): Promise<IChatAdapterNextItemRes>;
 }
 
 declare class ChatAdapter implements IChatAdapter {
-    nextItem(req: IChatAdapterNextItemReq): Promise<IChatItem>;
+    nextItem(req: IChatAdapterNextItemReq): Promise<IChatAdapterNextItemRes>;
 }
 
 type IChatItemData = IEntityData & IChatItem;
@@ -798,7 +809,7 @@ declare class ClaudeChatAdapter implements IChatAdapter {
     private anthropic;
     private logger;
     constructor();
-    nextItem(req: IChatAdapterNextItemReq): Promise<IChatItem>;
+    nextItem(req: IChatAdapterNextItemReq): Promise<IChatAdapterNextItemRes>;
     private mapChatItems;
     private mapHumanMessage;
     private mapBotMessage;
@@ -811,7 +822,7 @@ declare class DeepSeekChatAdapter implements IChatAdapter {
     private deepSeek;
     private logger;
     constructor();
-    nextItem(req: IChatAdapterNextItemReq): Promise<IChatItem>;
+    nextItem(req: IChatAdapterNextItemReq): Promise<IChatAdapterNextItemRes>;
     private mapChatItems;
     private mapHumanMessage;
     private mapBotMessage;
@@ -820,10 +831,25 @@ declare class DeepSeekChatAdapter implements IChatAdapter {
     private mapResponse;
 }
 
+declare class GeminiChatAdapter implements IChatAdapter {
+    private genai;
+    private model;
+    private logger;
+    constructor();
+    private validateModel;
+    nextItem(req: IChatAdapterNextItemReq): Promise<IChatAdapterNextItemRes>;
+    private buildContents;
+    private validateMessageContent;
+    private mapFunctionCall;
+    private mapTool;
+    private mapResponse;
+    private validateResponse;
+}
+
 declare class OpenaiChatAdapter implements IChatAdapter {
     private openai;
     private logger;
-    nextItem(req: IChatAdapterNextItemReq): Promise<IChatItem>;
+    nextItem(req: IChatAdapterNextItemReq): Promise<IChatAdapterNextItemRes>;
     private mapChatItems;
     private mapConectionMessage;
     private mapBotMessage;
@@ -1279,4 +1305,4 @@ interface IValidateMaxOptions {
 }
 declare function validateMax(value: any, options: IValidateMaxOptions): IValidationResult<any>;
 
-export { Async, Auth, Chat, ChatAdapter, ChatBot, ChatBotMetadataStore, ChatItem, ChatMemory, ChatRepository, ChatResolver, ClaudeChatAdapter, CmdChannel, Command, CommandMetadataStore, Container, ControllerMetadataStore, CustomError, DeepSeekChatAdapter, Entity, Env, EnvWhatsAppRepository, ExpressProvider, HtmlModule, HttpServerProvider, type IArrayValidationError, type IArrayValidationResult, type IBotMessageItem, type IChannelMessage, type IChannelMetadata, type IChatAdapter, type IChatAdapterNextItemReq, type IChatBot, type IChatBotMetadata, type IChatChannel, type IChatConnection, type IChatControllerMetadata, type IChatData, type IChatItem, type IChatItemData, type IChatItemType, type IChatMemory, type IChatMessage, type IChatRepository, type IChatType, type ICommandConfig, type ICommandHandler, type ICommandHandlerConfig, type IConstructor, type ICrudRepository, type ICustomErrorData, type IEndPointMetadata, type IEntityData, type IEnvType, type IFunctionCall, type IFunctionCallItem, type IGetConfig, type IGetWhatsAppTemplateRequest, type IHtmlModuleOptions, type IHumanMessageItem, type IJobData, type IJobEvent, type IJobEventListener, type IJobRepository, type IJwtRefreshTokenData, type IListenWhatsAppMessageRequest, type IMessageContext, type IMessageMetadata, type IMiddleware, type IMiddlewareMetadata, type IMindset, type IMindsetDecoration, type IMindsetFunctionConfig, type IMindsetFunctionDecoration, type IMindsetFunctionMetadata, type IMindsetFunctionParamMetadata, type IMindsetIdentity, type IMindsetMetadata, type IMindsetModuleConfig, type IMindsetModuleDecoration, type IMindsetModuleMetadata, type IMindsetTool, type IMindsetToolParameter, type IModelValidationError, type IModelValidationResult, type IModelValidatorsInfo, type IParamConfig, type IParamDecoration, type IPersistentData, type IPgRepositoryConfig, type IPostConfig, type IPrimitive, type IPropertyValidatorInfo, type IReceivedMessage, type IRestControllerConfig, type IRestControllerMetadata, type ISendWhatsAppRequest, type ISendWhatsAppTemplateRequest, type ISocketChannelConfig, type ISocketChannelReceivedMessage, type IStorableData, type ITelegramChannelConfig, type IValidateArrayOptions, type IValidateArrayOptionsWithItemsValidators, type IValidateMaxOptions, type IValidateMinOptions, type IValidationError, type IValidationResult, type IValidator, type IValidatorMetadata, type IWhatsAppBusinessAccount, type IWhatsAppBusinessNumber, type IWhatsAppContact, type IWhatsAppData, type IWhatsAppMessage, type IWhatsAppMessageListener, type IWhatsAppRepository, type IWhatsAppSenderOptions, type IWhatsAppTemplate, type IWhatsAppTemplateComponent, type IWhatsAppTemplateMessage, type IWhatsAppTemplateParameter, type IWhatsAppTemplateResponse, type IWhatsAppWebhookPayload, type IWhatsappChannelConfig, type IchatControllerConfig, Job, JobRepository, JobRunner, JobsEventsHub, Jwt, JwtAccessAndRefreshTokenDto, JwtConfig, JwtGuardMiddleware, JwtRefreshToken, JwtRefreshTokenRepository, JwtSigner, JwtTokenDto, Lifecycle, Logger, MINDSET_DECORATION_MINDSET, MINDSET_FUNCTION_DECORATION_FUNCTION, MINDSET_MODULE_DECORATION_MODULE, Mapper, Mindset, MindsetMetadataStore, MindsetOperator, OpenaiChatAdapter, PARAM_DECORATION_IS_OPTIONAL, PARAM_DECORATION_PARAM, Persistent, PgChatMemory, PgChatRepository, PgCrudRepository, PgJobRepository, PgRepositoryBase, PgWhatsAppRepository, RamChatMemory, RamChatRepository, Random, RestControllerMetadataStore, SocketChannel, SocketChannelConfig, SocketServerProvider, Storable, TelegramChannel, TelegramChannelConfig, ValidationMetadataStore, WhatsApp, WhatsAppChannel, WhatsAppReceiver, WhatsAppReceiverByDevConnection, WhatsAppReceiverByWebHook, WhatsAppRepository, WhatsAppSender, WhatsAppSenderByCloudApi, WhatsAppSenderByDevConnection, WhatsappChannelConfig, chatBot, chatController, chatItemTypeOptions, cmd, command, commandHandler, container, get, inject, injectable, isArray, isBoolean, isDate, isModel, isNotEmpty, isNumber, isOptional, isPresent, isString, jwtGuard, max, middleware, min, mindset, mindsetFunction, mindsetModule, modelInfo, param, post, restController, runAsyncCommandHandlers, runChatControllers, runRestControllers, scoped, singleton, socket, telegram, validate, validateArray, validateIsBoolean, validateIsDate, validateIsNotEmpty, validateIsNumber, validateIsPresent, validateIsString, validateMax, validateMin, validateModel, whatsapp };
+export { Async, Auth, Chat, ChatAdapter, ChatBot, ChatBotMetadataStore, ChatItem, ChatMemory, ChatRepository, ChatResolver, ClaudeChatAdapter, CmdChannel, Command, CommandMetadataStore, Container, ControllerMetadataStore, CustomError, DeepSeekChatAdapter, Entity, Env, EnvWhatsAppRepository, ExpressProvider, GeminiChatAdapter, HtmlModule, HttpServerProvider, type IArrayValidationError, type IArrayValidationResult, type IBotMessageItem, type IChannelMessage, type IChannelMetadata, type IChatAdapter, type IChatAdapterNextItemReq, type IChatAdapterNextItemRes, type IChatBot, type IChatBotMetadata, type IChatChannel, type IChatConnection, type IChatControllerMetadata, type IChatData, type IChatItem, type IChatItemData, type IChatItemType, type IChatMemory, type IChatMessage, type IChatRepository, type IChatType, type ICommandConfig, type ICommandHandler, type ICommandHandlerConfig, type IConstructor, type ICrudRepository, type ICustomErrorData, type IEndPointMetadata, type IEntityData, type IEnvType, type IFunctionCall, type IFunctionCallItem, type IGetConfig, type IGetWhatsAppTemplateRequest, type IHtmlModuleOptions, type IHumanMessageItem, type IJobData, type IJobEvent, type IJobEventListener, type IJobRepository, type IJwtRefreshTokenData, type ILanguageModelUsage, type IListenWhatsAppMessageRequest, type IMessageContext, type IMessageMetadata, type IMiddleware, type IMiddlewareMetadata, type IMindset, type IMindsetDecoration, type IMindsetFunctionConfig, type IMindsetFunctionDecoration, type IMindsetFunctionMetadata, type IMindsetFunctionParamMetadata, type IMindsetIdentity, type IMindsetMetadata, type IMindsetModuleConfig, type IMindsetModuleDecoration, type IMindsetModuleMetadata, type IMindsetTool, type IMindsetToolParameter, type IModelValidationError, type IModelValidationResult, type IModelValidatorsInfo, type IParamConfig, type IParamDecoration, type IPersistentData, type IPgRepositoryConfig, type IPostConfig, type IPrimitive, type IPropertyValidatorInfo, type IReceivedMessage, type IRestControllerConfig, type IRestControllerMetadata, type ISendWhatsAppRequest, type ISendWhatsAppTemplateRequest, type ISocketChannelConfig, type ISocketChannelReceivedMessage, type IStorableData, type ITelegramChannelConfig, type IValidateArrayOptions, type IValidateArrayOptionsWithItemsValidators, type IValidateMaxOptions, type IValidateMinOptions, type IValidationError, type IValidationResult, type IValidator, type IValidatorMetadata, type IWhatsAppBusinessAccount, type IWhatsAppBusinessNumber, type IWhatsAppContact, type IWhatsAppData, type IWhatsAppMessage, type IWhatsAppMessageListener, type IWhatsAppRepository, type IWhatsAppSenderOptions, type IWhatsAppTemplate, type IWhatsAppTemplateComponent, type IWhatsAppTemplateMessage, type IWhatsAppTemplateParameter, type IWhatsAppTemplateResponse, type IWhatsAppWebhookPayload, type IWhatsappChannelConfig, type IchatControllerConfig, Job, JobRepository, JobRunner, JobsEventsHub, Jwt, JwtAccessAndRefreshTokenDto, JwtConfig, JwtGuardMiddleware, JwtRefreshToken, JwtRefreshTokenRepository, JwtSigner, JwtTokenDto, Lifecycle, Logger, MINDSET_DECORATION_MINDSET, MINDSET_FUNCTION_DECORATION_FUNCTION, MINDSET_MODULE_DECORATION_MODULE, Mapper, Mindset, MindsetMetadataStore, MindsetOperator, OpenaiChatAdapter, PARAM_DECORATION_IS_OPTIONAL, PARAM_DECORATION_PARAM, Persistent, PgChatMemory, PgChatRepository, PgCrudRepository, PgJobRepository, PgRepositoryBase, PgWhatsAppRepository, RamChatMemory, RamChatRepository, Random, RestControllerMetadataStore, SocketChannel, SocketChannelConfig, SocketServerProvider, Storable, TelegramChannel, TelegramChannelConfig, ValidationMetadataStore, WhatsApp, WhatsAppChannel, WhatsAppReceiver, WhatsAppReceiverByDevConnection, WhatsAppReceiverByWebHook, WhatsAppRepository, WhatsAppSender, WhatsAppSenderByCloudApi, WhatsAppSenderByDevConnection, WhatsappChannelConfig, chatBot, chatController, chatItemTypeOptions, cmd, command, commandHandler, container, get, inject, injectable, isArray, isBoolean, isDate, isModel, isNotEmpty, isNumber, isOptional, isPresent, isString, jwtGuard, max, middleware, min, mindset, mindsetFunction, mindsetModule, modelInfo, param, post, restController, runAsyncCommandHandlers, runChatControllers, runRestControllers, scoped, singleton, socket, telegram, validate, validateArray, validateIsBoolean, validateIsDate, validateIsNotEmpty, validateIsNumber, validateIsPresent, validateIsString, validateMax, validateMin, validateModel, whatsapp };

package/dist/src/index.js

@@ -61,6 +61,7 @@ export { SocketServerProvider } from './feature/socket/SocketServerProvider.js';
 export { PgJobRepository } from './addon/async/pg/PgJobRepository.js';
 export { ClaudeChatAdapter } from './addon/chat-bot/claude/ClaudeChatAdapter.js';
 export { DeepSeekChatAdapter } from './addon/chat-bot/deepseek/DeepSeekChatAdapter.js';
+export { GeminiChatAdapter } from './addon/chat-bot/gemini/GeminiChatAdapter.js';
 export { OpenaiChatAdapter } from './addon/chat-bot/openia/OpenaiChatAdapter.js';
 export { PgChatRepository } from './addon/chat-bot/pg/PgChatRepository.js';
 export { PgChatMemory } from './addon/chat-bot/pg/PgChatMemory.js';