feat: ai plugin

Author: twlite

apps/test-bot/commandkit.config.ts

@@ -3,12 +3,14 @@ import { legacy } from '@commandkit/legacy';
 import { i18n } from '@commandkit/i18n';
 import { devtools } from '@commandkit/devtools';
 import { cache } from '@commandkit/cache';
+import { ai } from '@commandkit/ai';
 
 export default defineConfig({
   plugins: [
     i18n(),
     legacy({ skipBuiltInValidations: true }),
     devtools(),
     cache(),
+    ai(),
   ],
 });

apps/test-bot/package.json

@@ -10,15 +10,18 @@
     "node": "node"
   },
   "dependencies": {
+    "@ai-sdk/google": "^1.2.19",
+    "@commandkit/ai": "workspace:*",
     "@commandkit/cache": "workspace:*",
     "@commandkit/devtools": "workspace:*",
     "@commandkit/i18n": "workspace:*",
     "@commandkit/legacy": "workspace:*",
     "commandkit": "workspace:*",
     "discord.js": "^14.19.1",
-    "dotenv": "^16.4.7"
+    "dotenv": "^16.4.7",
+    "zod": "^3.25.56"
   },
   "devDependencies": {
     "tsx": "^4.7.0"
   }
-}
+}

apps/test-bot/src/ai.ts

@@ -0,0 +1,19 @@
+import { createGoogleGenerativeAI } from '@ai-sdk/google';
+import { configureAI } from '@commandkit/ai';
+
+const google = createGoogleGenerativeAI({
+  apiKey: process.env.GOOGLE_API_KEY,
+});
+
+const model = google.languageModel('gemini-2.0-flash');
+
+configureAI({
+  selectAiModel: async () => {
+    return { model };
+  },
+  messageFilter: async (message) => {
+    return (
+      message.inGuild() && message.mentions.users.has(message.client.user.id)
+    );
+  },
+});

apps/test-bot/src/app.ts

@@ -1,5 +1,6 @@
 import { Client } from 'discord.js';
 import { Logger, commandkit } from 'commandkit';
+import './ai';
 
 const client = new Client({
   intents: [

apps/test-bot/src/app/commands/(leveling)/xp.ts

@@ -1,12 +1,22 @@
 import { ChatInputCommandContext, CommandData } from 'commandkit';
-import { database } from '../../../database/store.ts';
+import { database } from '@/database/store.ts';
 import { cacheTag } from '@commandkit/cache';
+import { AiConfig, AiContext } from '@commandkit/ai';
+import { z } from 'zod';
 
 export const command: CommandData = {
   name: 'xp',
   description: 'This is an xp command.',
 };
 
+export const aiConfig: AiConfig = {
+  description: 'Get the XP of a user in a guild.',
+  parameters: z.object({
+    guildId: z.string().describe('The ID of the guild.'),
+    userId: z.string().describe('The ID of the user.'),
+  }),
+};
+
 async function getUserXP(guildId: string, userId: string) {
   'use cache';
 
@@ -39,3 +49,26 @@ export async function chatInput({ interaction }: ChatInputCommandContext) {
     ],
   });
 }
+
+export async function ai(ctx: AiContext) {
+  const message = ctx.message;
+
+  if (!message.inGuild()) {
+    return {
+      error: 'This tool can only be used in a guild.',
+    };
+  }
+
+  const { guildId, userId } = ctx.params as {
+    guildId: string;
+    userId: string;
+  };
+
+  const xp = await getUserXP(guildId, userId);
+
+  return {
+    userId,
+    guildId,
+    xp,
+  };
+}

packages/ai/README.md

@@ -0,0 +1,19 @@
+# `@commandkit/ai`
+
+Supercharge your CommandKit project with AI capabilities.
+
+## Installation
+
+```bash
+npm install @commandkit/ai
+```
+
+## Usage
+
+```ts
+import { ai } from '@commandkit/ai';
+
+export default defineConfig({
+  plugins: [ai()],
+})
+```

packages/ai/package.json

@@ -0,0 +1,38 @@
+{
+  "name": "@commandkit/ai",
+  "version": "0.1.0",
+  "description": "Supercharge your CommandKit bot with AI capabilities",
+  "files": [
+    "dist"
+  ],
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "scripts": {
+    "lint": "tsc --noEmit",
+    "build": "tsc"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/underctrl-io/commandkit.git"
+  },
+  "keywords": [
+    "commandkit",
+    "ai"
+  ],
+  "author": "twilight <[email protected]>",
+  "license": "MIT",
+  "bugs": {
+    "url": "https://github.com/underctrl-io/commandkit/issues"
+  },
+  "homepage": "https://github.com/underctrl-io/commandkit#readme",
+  "devDependencies": {
+    "commandkit": "workspace:*",
+    "discord.js": "^14.19.3",
+    "tsconfig": "workspace:*",
+    "typescript": "^5.7.3"
+  },
+  "dependencies": {
+    "ai": "^4.3.16",
+    "zod": "^3.25.48"
+  }
+}

packages/ai/src/ai-context-worker.ts

@@ -0,0 +1,25 @@
+import { Message } from 'discord.js';
+import { AsyncLocalStorage } from 'node:async_hooks';
+import { AiContext } from './context';
+
+const worker = new AsyncLocalStorage<{ message: Message; ctx: AiContext }>();
+
+export function getAiWorkerContext(): { message: Message; ctx: AiContext } {
+  const ctx = worker.getStore();
+
+  if (!ctx) {
+    throw new Error(
+      'AI context is not available. Ensure you are using AI in a CommandKit environment.',
+    );
+  }
+
+  return ctx;
+}
+
+export function runInAiWorkerContext<R, F extends (...args: any[]) => R>(
+  ctx: AiContext,
+  message: Message,
+  callback: F,
+): R {
+  return worker.run({ message, ctx }, callback);
+}

packages/ai/src/context.ts

@@ -0,0 +1,30 @@
+import type { CommandKit } from 'commandkit';
+import { Client, Message } from 'discord.js';
+
+export interface AiContextOptions<
+  T extends Record<string, unknown> = Record<string, unknown>,
+> {
+  message: Message;
+  params: T;
+  commandkit: CommandKit;
+}
+
+export class AiContext<
+  T extends Record<string, unknown> = Record<string, unknown>,
+> {
+  public params!: T;
+  public message!: Message;
+  public client!: Client;
+  public commandkit!: CommandKit;
+
+  public constructor(options: AiContextOptions<T>) {
+    this.params = options.params;
+    this.message = options.message;
+    this.commandkit = options.commandkit;
+    this.client = options.commandkit.client;
+  }
+
+  public setParams(params: T): void {
+    this.params = params;
+  }
+}

packages/ai/src/index.ts

@@ -0,0 +1,10 @@
+import { AiPlugin } from './plugin';
+import { AiPluginOptions } from './types';
+
+export function ai(options?: AiPluginOptions) {
+  return new AiPlugin(options ?? {});
+}
+
+export * from './types';
+export * from './plugin';
+export * from './context';