@futdevpro/nts-dynamo 1.15.2 → 1.15.3

package/src/_modules/assistant/_collections/ass.util.spec.ts

@@ -0,0 +1,176 @@
+
+import { DyFM_AI_Message, DyFM_AI_MessageRole } from '@futdevpro/fsm-dynamo/ai';
+import { DyNTS_Ass_Util } from './ass.util';
+import { DyNTS_Bot_MessageWrapper } from '../../bot/_models/bot-message-wrapper.interface';
+import { DyNTS_Ass_global_settings } from './ass-global-settings.const';
+
+describe('| DyNTS_Ass_Util', () => {
+  describe('| convertBotMessagesToAIConversation', () => {
+    it('| should convert bot messages to AI conversation format', () => {
+      const messages: DyNTS_Bot_MessageWrapper<any>[] = [
+        {
+          id: 'msg-1',
+          content: 'User message',
+          authorId: 'user-123',
+          authorName: 'user',
+          isBot: false,
+        } as any,
+        {
+          id: 'msg-2',
+          content: 'Bot response',
+          authorId: 'bot-123',
+          authorName: 'bot',
+          isBot: true,
+        } as any,
+      ];
+
+      const result = DyNTS_Ass_Util.convertBotMessagesToAIConversation({
+        messages: messages,
+        botClientId: 'bot-123',
+        botDisplayName: 'Bot',
+        issuer: 'issuer-123',
+      });
+
+      expect(result.length).toBe(2);
+      expect(result[0].role).toBe(DyFM_AI_MessageRole.user);
+      expect(result[0].content).toBe('User message');
+      expect(result[1].role).toBe(DyFM_AI_MessageRole.assistant);
+      expect(result[1].content).toBe('Bot response');
+    });
+
+    it('| should identify bot messages by isBot flag', () => {
+      const messages: DyNTS_Bot_MessageWrapper<any>[] = [
+        {
+          id: 'msg-1',
+          content: 'Message',
+          authorId: 'bot-123',
+          isBot: true,
+        } as any,
+      ];
+
+      const result = DyNTS_Ass_Util.convertBotMessagesToAIConversation({
+        messages: messages,
+        botClientId: 'bot-123',
+        botDisplayName: 'Bot',
+        issuer: 'issuer-123',
+      });
+
+      expect(result[0].role).toBe(DyFM_AI_MessageRole.assistant);
+    });
+
+    it('| should identify bot messages by authorId match', () => {
+      const messages: DyNTS_Bot_MessageWrapper<any>[] = [
+        {
+          id: 'msg-1',
+          content: 'Message',
+          authorId: 'bot-123',
+          isBot: false,
+        } as any,
+      ];
+
+      const result = DyNTS_Ass_Util.convertBotMessagesToAIConversation({
+        messages: messages,
+        botClientId: 'bot-123',
+        botDisplayName: 'Bot',
+        issuer: 'issuer-123',
+      });
+
+      expect(result[0].role).toBe(DyFM_AI_MessageRole.assistant);
+    });
+
+    it('| should identify bot messages by authorName match', () => {
+      const messages: DyNTS_Bot_MessageWrapper<any>[] = [
+        {
+          id: 'msg-1',
+          content: 'Message',
+          authorId: 'other-123',
+          authorName: 'Bot',
+          isBot: false,
+        } as any,
+      ];
+
+      const result = DyNTS_Ass_Util.convertBotMessagesToAIConversation({
+        messages: messages,
+        botClientId: 'bot-123',
+        botDisplayName: 'Bot',
+        issuer: 'issuer-123',
+      });
+
+      expect(result[0].role).toBe(DyFM_AI_MessageRole.assistant);
+    });
+
+    it('| should identify bot messages by authorDisplayName match', () => {
+      const messages: DyNTS_Bot_MessageWrapper<any>[] = [
+        {
+          id: 'msg-1',
+          content: 'Message',
+          authorId: 'other-123',
+          authorDisplayName: 'Bot',
+          isBot: false,
+        } as any,
+      ];
+
+      const result = DyNTS_Ass_Util.convertBotMessagesToAIConversation({
+        messages: messages,
+        botClientId: 'bot-123',
+        botDisplayName: 'Bot',
+        issuer: 'issuer-123',
+      });
+
+      expect(result[0].role).toBe(DyFM_AI_MessageRole.assistant);
+    });
+
+    it('| should convert bot messages with user translation flags to user role', () => {
+      const userTranslationFlag = DyNTS_Ass_global_settings.userTranslationFlags[0];
+      const messages: DyNTS_Bot_MessageWrapper<any>[] = [
+        {
+          id: 'msg-1',
+          content: `Message with ${userTranslationFlag} flag`,
+          authorId: 'bot-123',
+          isBot: true,
+        } as any,
+      ];
+
+      const result = DyNTS_Ass_Util.convertBotMessagesToAIConversation({
+        messages: messages,
+        botClientId: 'bot-123',
+        botDisplayName: 'Bot',
+        issuer: 'issuer-123',
+      });
+
+      expect(result[0].role).toBe(DyFM_AI_MessageRole.user);
+    });
+
+    it('| should handle empty messages array', () => {
+      const result = DyNTS_Ass_Util.convertBotMessagesToAIConversation({
+        messages: [],
+        botClientId: 'bot-123',
+        botDisplayName: 'Bot',
+        issuer: 'issuer-123',
+      });
+
+      expect(result).toEqual([]);
+    });
+
+    it('| should convert user messages to user role', () => {
+      const messages: DyNTS_Bot_MessageWrapper<any>[] = [
+        {
+          id: 'msg-1',
+          content: 'User message',
+          authorId: 'user-123',
+          isBot: false,
+        } as any,
+      ];
+
+      const result = DyNTS_Ass_Util.convertBotMessagesToAIConversation({
+        messages: messages,
+        botClientId: 'bot-123',
+        botDisplayName: 'Bot',
+        issuer: 'issuer-123',
+      });
+
+      expect(result[0].role).toBe(DyFM_AI_MessageRole.user);
+    });
+  });
+});
+

package/src/_modules/assistant/_services/ass-io.control-service.spec.ts

@@ -0,0 +1,140 @@
+
+import { DyNTS_Ass_IO_ControlService } from './ass-io.control-service';
+import { DyNTS_Ass_Main_ControlService } from './ass-main.control-service';
+import { DyNTS_Bot_IO_ControlService } from '../../bot/_services/bot-io.control-service';
+import { DyNTS_Bot_MessageWrapper } from '../../bot/_models/bot-message-wrapper.interface';
+import { DyNTS_AI_LLMChat_ServiceBase } from '../../ai/_services/ai-llm-chat.service-base';
+import { DyFM_Error } from '@futdevpro/fsm-dynamo';
+import { DyNTS_global_settings } from '../../../_collections/global-settings.const';
+import { DyNTS_Ass_Util } from '../_collections/ass.util';
+
+class TestAssIOService extends DyNTS_Ass_IO_ControlService {
+  protected declare mainBot_CS: any;
+  protected getMainBotControlService(): any {
+    return {
+      gatherMessagesForMessage: jasmine.createSpy('gatherMessagesForMessage').and.returnValue(
+        Promise.resolve([])
+      ),
+      defaultSystemPrompt: 'Test prompt',
+      aiProvider: {
+        requestSimpleMessageInConversation: jasmine.createSpy('requestSimpleMessageInConversation').and.returnValue(
+          Promise.resolve('AI response')
+        ),
+      } as any,
+    };
+  }
+
+  protected getCommandsControlService(): any {
+    return {} as any;
+  }
+
+  static getInstance(): TestAssIOService {
+    return TestAssIOService.getSingletonInstance();
+  }
+}
+
+describe('| DyNTS_Ass_IO_ControlService', () => {
+  let service: TestAssIOService;
+
+  beforeEach(() => {
+    service = TestAssIOService.getInstance();
+  });
+
+  it('| should be a singleton instance', () => {
+    const instance1 = TestAssIOService.getInstance();
+    const instance2 = TestAssIOService.getInstance();
+
+    expect(instance1).toBe(instance2);
+    expect(instance1).toBeInstanceOf(TestAssIOService);
+  });
+
+  it('| should get AI provider from main bot service', () => {
+    (service as any).mainBot_CS = (service as any).getMainBotControlService();
+
+    expect(service.aiProvider).toBeDefined();
+  });
+
+  describe('| handleMessage', () => {
+    it('| should handle message with AI and reply', async () => {
+      const mockMessage = {
+        provider: {
+          botId: 'bot-123',
+          botDisplayName: 'Bot',
+          replyToMessage: jasmine.createSpy('replyToMessage').and.returnValue(
+            Promise.resolve({ content: 'AI response' } as any)
+          ),
+        },
+        content: 'User message',
+      } as any as DyNTS_Bot_MessageWrapper<any>;
+
+      (service as any).mainBot_CS = (service as any).getMainBotControlService();
+      spyOn(DyNTS_Ass_Util, 'convertBotMessagesToAIConversation').and.returnValue([
+        { role: 'user' as any, content: 'User message' },
+      ]);
+
+      const result = await service.handleMessage(mockMessage, 'issuer-123');
+
+      expect((service as any).mainBot_CS.gatherMessagesForMessage).toHaveBeenCalledWith(mockMessage);
+      expect((service as any).mainBot_CS.aiProvider.requestSimpleMessageInConversation).toHaveBeenCalled();
+      expect(mockMessage.provider.replyToMessage).toHaveBeenCalledWith(mockMessage, 'AI response');
+      expect(result).toBeDefined();
+    });
+
+    it('| should send error reply when error occurs and debugLevel >= 1', async () => {
+      const mockMessage = {
+        provider: {
+          botId: 'bot-123',
+          botDisplayName: 'Bot',
+          replyToMessage: jasmine.createSpy('replyToMessage').and.returnValue(
+            Promise.resolve({} as any)
+          ),
+        },
+        content: 'User message',
+      } as any as DyNTS_Bot_MessageWrapper<any>;
+
+      (service as any).mainBot_CS = (service as any).getMainBotControlService();
+      (service as any).mainBot_CS.gatherMessagesForMessage = jasmine.createSpy('gatherMessagesForMessage').and.returnValue(
+        Promise.reject(new Error('Test error'))
+      );
+      service.dontSendErrorReply = false;
+      // Set bot_settings directly instead of using spyOn
+      const originalBotSettings = DyNTS_global_settings.bot_settings;
+      (DyNTS_global_settings as any).bot_settings = { debugLevel: 1 };
+
+      try {
+        await expectAsync(
+          service.handleMessage(mockMessage, 'issuer-123')
+        ).toBeRejected();
+
+        expect(mockMessage.provider.replyToMessage).toHaveBeenCalled();
+      } finally {
+        // Restore original bot_settings
+        (DyNTS_global_settings as any).bot_settings = originalBotSettings;
+      }
+    });
+
+    it('| should not send error reply when dontSendErrorReply is true', async () => {
+      const mockMessage = {
+        provider: {
+          botId: 'bot-123',
+          botDisplayName: 'Bot',
+          replyToMessage: jasmine.createSpy('replyToMessage'),
+        },
+        content: 'User message',
+      } as any as DyNTS_Bot_MessageWrapper<any>;
+
+      (service as any).mainBot_CS = (service as any).getMainBotControlService();
+      (service as any).mainBot_CS.gatherMessagesForMessage = jasmine.createSpy('gatherMessagesForMessage').and.returnValue(
+        Promise.reject(new Error('Test error'))
+      );
+      service.dontSendErrorReply = true;
+
+      await expectAsync(
+        service.handleMessage(mockMessage, 'issuer-123')
+      ).toBeRejected();
+
+      expect(mockMessage.provider.replyToMessage).not.toHaveBeenCalled();
+    });
+  });
+});
+

package/src/_modules/assistant/_services/ass-main.control-service.spec.ts

@@ -0,0 +1,192 @@
+
+import { DyNTS_Ass_Main_ControlService } from './ass-main.control-service';
+import { DyNTS_Bot_Main_ControlService } from '../../bot/_services/bot-main.control-service';
+import { DyNTS_AI_LLMChat_ServiceBase } from '../../ai/_services/ai-llm-chat.service-base';
+import { DyNTS_Bot_MessageWrapper } from '../../bot/_models/bot-message-wrapper.interface';
+import { DyNTS_Bot_ChannelWrapper } from '../../bot/_models/bot-channel-wrapper.interface';
+import { DyFM_AI_Message } from '@futdevpro/fsm-dynamo/ai';
+import { DyFM_Msg_Provider_Type } from '@futdevpro/fsm-dynamo/messaging';
+import { DyNTS_Ass_Util } from '../_collections/ass.util';
+import { DyNTS_Ass_global_settings } from '../_collections/ass-global-settings.const';
+
+class TestAssMainService extends DyNTS_Ass_Main_ControlService<DyFM_Msg_Provider_Type> {
+  protected getAIProvider(): DyNTS_AI_LLMChat_ServiceBase {
+    return {
+      defaultSettings: {
+        systemPrompt: 'Test prompt',
+        useModel: 'test-model',
+      },
+    } as any;
+  }
+
+  protected getBotIOControlService(): any {
+    return {} as any;
+  }
+
+  protected getRoutinesControlService(): any {
+    return {} as any;
+  }
+
+  protected getCommandsControlService(): any {
+    return {} as any;
+  }
+
+  protected defaultProviderType: DyFM_Msg_Provider_Type = DyFM_Msg_Provider_Type.discord;
+  protected availableProviders: any[] = [];
+
+  static getInstance(): TestAssMainService {
+    return TestAssMainService.getSingletonInstance();
+  }
+}
+
+describe('| DyNTS_Ass_Main_ControlService', () => {
+  let service: TestAssMainService;
+
+  beforeEach(() => {
+    service = TestAssMainService.getInstance();
+  });
+
+  it('| should be a singleton instance', () => {
+    const instance1 = TestAssMainService.getInstance();
+    const instance2 = TestAssMainService.getInstance();
+
+    expect(instance1).toBe(instance2);
+    expect(instance1).toBeInstanceOf(TestAssMainService);
+  });
+
+  it('| should have default system prompt', () => {
+    expect(service.defaultSystemPrompt).toBe(DyNTS_Ass_global_settings.defaultSystemPrompt);
+  });
+
+  it('| should initialize AI provider in constructor', () => {
+    expect(service.aiProvider).toBeDefined();
+  });
+
+  describe('| gatherMessagesForMessage', () => {
+    it('| should gather and filter messages', async () => {
+      const mockMessage = {
+        provider: {
+          fetchMessagesForMessage: jasmine.createSpy('fetchMessagesForMessage').and.returnValue(
+            Promise.resolve([
+              { content: 'Message 1', timestamp: 1000 },
+              { content: '[NOISE] Message 2', timestamp: 2000 },
+              { content: 'Message 3', timestamp: 3000 },
+            ] as any[])
+          ),
+        },
+      } as any as DyNTS_Bot_MessageWrapper<any>;
+
+      const result = await service.gatherMessagesForMessage(mockMessage);
+
+      expect(result.length).toBe(2);
+      expect(result[0].content).toBe('Message 1');
+      expect(result[1].content).toBe('Message 3');
+      expect(mockMessage.provider.fetchMessagesForMessage).toHaveBeenCalledWith(mockMessage, 100);
+    });
+
+    it('| should use custom limit', async () => {
+      const mockMessage = {
+        provider: {
+          fetchMessagesForMessage: jasmine.createSpy('fetchMessagesForMessage').and.returnValue(
+            Promise.resolve([])
+          ),
+        },
+      } as any as DyNTS_Bot_MessageWrapper<any>;
+
+      await service.gatherMessagesForMessage(mockMessage, 50);
+
+      expect(mockMessage.provider.fetchMessagesForMessage).toHaveBeenCalledWith(mockMessage, 50);
+    });
+
+    it('| should sort messages by timestamp', async () => {
+      const mockMessage = {
+        provider: {
+          fetchMessagesForMessage: jasmine.createSpy('fetchMessagesForMessage').and.returnValue(
+            Promise.resolve([
+              { content: 'Message 3', timestamp: 3000 },
+              { content: 'Message 1', timestamp: 1000 },
+              { content: 'Message 2', timestamp: 2000 },
+            ] as any[])
+          ),
+        },
+      } as any as DyNTS_Bot_MessageWrapper<any>;
+
+      const result = await service.gatherMessagesForMessage(mockMessage);
+
+      expect(result[0].timestamp).toBe(1000);
+      expect(result[1].timestamp).toBe(2000);
+      expect(result[2].timestamp).toBe(3000);
+    });
+  });
+
+  describe('| gatherMessagesInChannel', () => {
+    it('| should gather messages from channel', async () => {
+      const mockChannel = {
+        provider: {
+          fetchMessages: jasmine.createSpy('fetchMessages').and.returnValue(
+            Promise.resolve([
+              { content: 'Message 1' },
+              { content: 'Message 2' },
+            ] as any[])
+          ),
+        },
+      } as any as DyNTS_Bot_ChannelWrapper;
+
+      const result = await service.gatherMessagesInChannel(mockChannel, 'issuer-123');
+
+      expect(result.length).toBe(2);
+      expect(mockChannel.provider.fetchMessages).toHaveBeenCalledWith(mockChannel, 100);
+    });
+  });
+
+  describe('| gatherMessagesAsAIConversation', () => {
+    it('| should convert messages to AI conversation format', async () => {
+      const mockMessage = {
+        provider: {
+          botId: 'bot-123',
+          botDisplayName: 'Bot',
+          fetchMessagesForMessage: jasmine.createSpy('fetchMessagesForMessage').and.returnValue(
+            Promise.resolve([
+              { content: 'User message', authorId: 'user-123', isBot: false },
+              { content: 'Bot message', authorId: 'bot-123', isBot: true },
+            ] as any[])
+          ),
+        },
+      } as any as DyNTS_Bot_MessageWrapper<any>;
+      spyOn(DyNTS_Ass_Util, 'convertBotMessagesToAIConversation').and.returnValue([
+        { role: 'user' as any, content: 'User message' },
+        { role: 'assistant' as any, content: 'Bot message' },
+      ]);
+
+      const result = await service.gatherMessagesAsAIConversation(mockMessage, 'issuer-123');
+
+      expect(result.length).toBe(2);
+      expect(DyNTS_Ass_Util.convertBotMessagesToAIConversation).toHaveBeenCalled();
+    });
+  });
+
+  describe('| gatherMessagesAsAIConversationInChannel', () => {
+    it('| should convert channel messages to AI conversation format', async () => {
+      const mockChannel = {
+        provider: {
+          botId: 'bot-123',
+          botDisplayName: 'Bot',
+          fetchMessages: jasmine.createSpy('fetchMessages').and.returnValue(
+            Promise.resolve([
+              { content: 'User message', authorId: 'user-123', isBot: false },
+            ] as any[])
+          ),
+        },
+      } as any as DyNTS_Bot_ChannelWrapper;
+      spyOn(DyNTS_Ass_Util, 'convertBotMessagesToAIConversation').and.returnValue([
+        { role: 'user' as any, content: 'User message' },
+      ]);
+
+      const result = await service.gatherMessagesAsAIConversationInChannel(mockChannel, 'issuer-123');
+
+      expect(result.length).toBe(1);
+      expect(DyNTS_Ass_Util.convertBotMessagesToAIConversation).toHaveBeenCalled();
+    });
+  });
+});
+