-
Notifications
You must be signed in to change notification settings - Fork 4.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'v2-develop' into tcm/elevnlabs-plugin
- Loading branch information
Showing
11 changed files
with
959 additions
and
407 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,37 @@ | ||
| import { describe, it, expect } from 'vitest'; | ||
| import { validateDiscordConfig } from '../src/environment'; | ||
| import type { IAgentRuntime } from '@elizaos/core'; | ||
|
|
||
| // Mock runtime environment | ||
| const mockRuntime: IAgentRuntime = { | ||
| env: { | ||
| DISCORD_API_TOKEN: 'mocked-discord-token', | ||
| }, | ||
| getEnv: function (key: string) { | ||
| return this.env[key] || null; | ||
| }, | ||
| getSetting: function (key: string) { | ||
| return this.env[key] || null; | ||
| } | ||
| } as unknown as IAgentRuntime; | ||
|
|
||
| describe('Discord Environment Configuration', () => { | ||
| it('should validate correct configuration', async () => { | ||
| const config = await validateDiscordConfig(mockRuntime); | ||
| expect(config).toBeDefined(); | ||
| expect(config.DISCORD_API_TOKEN).toBe('mocked-discord-token'); | ||
| }); | ||
|
|
||
| it('should throw an error when DISCORD_API_TOKEN is missing', async () => { | ||
| const invalidRuntime = { | ||
| ...mockRuntime, | ||
| env: { | ||
| DISCORD_API_TOKEN: undefined, | ||
| }, | ||
| } as IAgentRuntime; | ||
|
|
||
| await expect(validateDiscordConfig(invalidRuntime)).rejects.toThrowError( | ||
| 'Discord configuration validation failed:\nDISCORD_API_TOKEN: Expected string, received null' | ||
| ); | ||
| }); | ||
| }); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,211 @@ | ||
| import { describe, it, expect, vi, beforeEach } from "vitest"; | ||
| import { MessageManager } from "../src/messages.ts"; | ||
| import { ChannelType, Client, Collection } from "discord.js"; | ||
| import { type IAgentRuntime } from "@elizaos/core"; | ||
| import type { VoiceManager } from "../src/voice"; | ||
|
|
||
| vi.mock("@elizaos/core", () => ({ | ||
| logger: { | ||
| info: vi.fn(), | ||
| error: vi.fn(), | ||
| debug: vi.fn(), | ||
| }, | ||
| stringToUuid: (str: string) => str, | ||
| messageCompletionFooter: | ||
| "# INSTRUCTIONS: Choose the best response for the agent.", | ||
| shouldRespondFooter: "# INSTRUCTIONS: Choose if the agent should respond.", | ||
| generateMessageResponse: vi.fn(), | ||
| generateShouldRespond: vi.fn().mockResolvedValue("IGNORE"), // Prevent API calls by always returning "IGNORE" | ||
| composeContext: vi.fn(), | ||
| ModelClass: { | ||
| TEXT_SMALL: "TEXT_SMALL", | ||
| }, | ||
| ServiceType: { | ||
| VIDEO: "VIDEO", | ||
| BROWSER: "BROWSER", | ||
| }, | ||
| })); | ||
|
|
||
| describe("Discord MessageManager", () => { | ||
| let mockRuntime: IAgentRuntime; | ||
| let mockClient: Client; | ||
| let mockDiscordClient: { client: Client; runtime: IAgentRuntime }; | ||
| let mockVoiceManager: VoiceManager; | ||
| let mockMessage: any; | ||
| let messageManager: MessageManager; | ||
|
|
||
| beforeEach(() => { | ||
| vi.clearAllMocks(); | ||
| mockRuntime = { | ||
| character: { | ||
| name: "TestBot", | ||
| templates: {}, | ||
| clientConfig: { | ||
| discord: { | ||
| allowedChannelIds: ["mock-channal-id"], | ||
| shouldIgnoreBotMessages: true, | ||
| shouldIgnoreDirectMessages: true, | ||
| }, | ||
| }, | ||
| }, | ||
| evaluate: vi.fn(), | ||
| composeState: vi.fn(), | ||
| ensureConnection: vi.fn(), | ||
| ensureUserExists: vi.fn(), | ||
| messageManager: { | ||
| createMemory: vi.fn(), | ||
| addEmbeddingToMemory: vi.fn(), | ||
| }, | ||
| databaseAdapter: { | ||
| getParticipantUserState: vi.fn().mockResolvedValue("ACTIVE"), | ||
| log: vi.fn(), | ||
| }, | ||
| processActions: vi.fn(), | ||
| } as unknown as IAgentRuntime; | ||
|
|
||
| mockClient = new Client({ intents: [] }); | ||
| mockClient.user = { | ||
| id: "mock-bot-id", | ||
| username: "MockBot", | ||
| tag: "MockBot#0001", | ||
| displayName: "MockBotDisplay", | ||
| } as any; | ||
|
|
||
| mockDiscordClient = { | ||
| client: mockClient, | ||
| runtime: mockRuntime, | ||
| }; | ||
|
|
||
| mockVoiceManager = { | ||
| playAudioStream: vi.fn(), | ||
| } as unknown as VoiceManager; | ||
|
|
||
| messageManager = new MessageManager(mockDiscordClient, mockVoiceManager); | ||
|
|
||
| const guild = { | ||
| members: { | ||
| cache: { | ||
| get: vi.fn().mockReturnValue({ | ||
| nickname: "MockBotNickname", | ||
| permissions: { | ||
| has: vi.fn().mockReturnValue(true), // Bot has permissions | ||
| }, | ||
| }), | ||
| }, | ||
| }, | ||
| }; | ||
| mockMessage = { | ||
| content: "Hello, MockBot!", | ||
| author: { | ||
| id: "mock-user-id", | ||
| username: "MockUser", | ||
| bot: false, | ||
| }, | ||
| guild, | ||
| channel: { | ||
| id: "mock-channal-id", | ||
| type: ChannelType.GuildText, | ||
| send: vi.fn(), | ||
| guild, | ||
| client: { | ||
| user: mockClient.user, | ||
| }, | ||
| permissionsFor: vi.fn().mockReturnValue({ | ||
| has: vi.fn().mockReturnValue(true), | ||
| }), | ||
| }, | ||
| id: "mock-message-id", | ||
| createdTimestamp: Date.now(), | ||
| mentions: { | ||
| has: vi.fn().mockReturnValue(false), | ||
| }, | ||
| reference: null, | ||
| attachments: [], | ||
| }; | ||
| }); | ||
|
|
||
| it("should initialize MessageManager", () => { | ||
| expect(messageManager).toBeDefined(); | ||
| }); | ||
|
|
||
| it("should process user messages", async () => { | ||
| // Prevent further message processing after response check | ||
| vi.spyOn( | ||
| Object.getPrototypeOf(messageManager), | ||
| "_shouldRespond" | ||
| ).mockReturnValueOnce(false); | ||
|
|
||
| await messageManager.handleMessage(mockMessage); | ||
| expect(mockRuntime.ensureConnection).toHaveBeenCalled(); | ||
| expect(mockRuntime.messageManager.createMemory).toHaveBeenCalled(); | ||
| }); | ||
|
|
||
| it("should ignore bot messages", async () => { | ||
| mockMessage.author.bot = true; | ||
| await messageManager.handleMessage(mockMessage); | ||
| expect(mockRuntime.ensureConnection).not.toHaveBeenCalled(); | ||
| }); | ||
|
|
||
| it("should ignore messages from restricted channels", async () => { | ||
| mockMessage.channel.id = "undefined-channel-id"; | ||
| await messageManager.handleMessage(mockMessage); | ||
| expect(mockRuntime.ensureConnection).not.toHaveBeenCalled(); | ||
| }); | ||
|
|
||
| it.each([ | ||
| ["Hey MockBot, are you there?", "username"], | ||
| ["MockBot#0001, respond please.", "tag"], | ||
| ["MockBotNickname, can you help?", "nickname"], | ||
| ["MoCkBoT, can you help?", "mixed case mention"], | ||
| ])( | ||
| "should respond if the bot name is included in the message", | ||
| async (content) => { | ||
| mockMessage.content = content; | ||
|
|
||
| const result = await messageManager["_shouldRespond"]( | ||
| mockMessage, | ||
| {} as any | ||
| ); | ||
| expect(result).toBe(true); | ||
| } | ||
| ); | ||
|
|
||
| it("should process audio attachments", async () => { | ||
| vi.spyOn( | ||
| Object.getPrototypeOf(messageManager), | ||
| "_shouldRespond" | ||
| ).mockReturnValueOnce(false); | ||
| vi.spyOn(messageManager, "processMessageMedia").mockReturnValueOnce( | ||
| Promise.resolve({ processedContent: "", attachments: [] }) | ||
| ); | ||
|
|
||
| const myVariable = new Collection<string, any>([ | ||
| [ | ||
| "mock-attachment-id", | ||
| { | ||
| attachment: "https://www.example.mp3", | ||
| name: "mock-attachment.mp3", | ||
| contentType: "audio/mpeg", | ||
| }, | ||
| ], | ||
| ]); | ||
|
|
||
| mockMessage.attachments = myVariable; | ||
|
|
||
| const processAttachmentsMock = vi.fn().mockResolvedValue([]); | ||
|
|
||
| const mockAttachmentManager = { | ||
| processAttachments: processAttachmentsMock, | ||
| } as unknown as (typeof messageManager)["attachmentManager"]; | ||
|
|
||
| // Override the private property with a mock | ||
| Object.defineProperty(messageManager, "attachmentManager", { | ||
| value: mockAttachmentManager, | ||
| writable: true, | ||
| }); | ||
|
|
||
| await messageManager.handleMessage(mockMessage); | ||
|
|
||
| expect(processAttachmentsMock).toHaveBeenCalled(); | ||
| }); | ||
| }); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.