@wabot-dev/framework 0.1.0-beta.33 → 0.1.0-beta.35

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

@@ -0,0 +1,125 @@
+import { Logger } from '../../../core/logger/Logger.js';
+import { GoogleGenAI } from '@google/genai';
+
+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/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/index.d.ts

@@ -195,6 +195,8 @@ type IModelValidatorsInfo<V> = {
 };
 
 interface IValidateArrayOptions {
+    minLength?: number;
+    maxLength?: number;
 }
 interface IValidateArrayOptionsWithItemsValidators extends IValidateArrayOptions {
     itemsValidator?: {
@@ -829,6 +831,21 @@ 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;
@@ -1288,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 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 };
+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';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wabot-dev/framework",
-  "version": "0.1.0-beta.33",
+  "version": "0.1.0-beta.35",
   "description": "Framework for IA Chat Bots",
   "type": "module",
   "main": "dist/src/index.js",
@@ -36,6 +36,7 @@
   },
   "peerDependencies": {
     "@anthropic-ai/sdk": "^0.60.0",
+    "@google/genai": "^1.16.0",
     "@types/debug": "^4.1.12",
     "@types/express": "^5.0.1",
     "@types/jsonwebtoken": "^9.0.10",