feat!: ai sdk 5

Author: twlite

apps/test-bot/package.json

@@ -10,7 +10,7 @@
     "node": "node"
   },
   "dependencies": {
-    "@ai-sdk/google": "^1.2.19",
+    "@ai-sdk/google": "^2.0.8",
     "@commandkit/ai": "workspace:*",
     "@commandkit/cache": "workspace:*",
     "@commandkit/devtools": "workspace:*",
@@ -27,4 +27,4 @@
     "@types/ms": "^2.1.0",
     "tsx": "^4.7.0"
   }
-}
+}

packages/ai/package.json

@@ -35,7 +35,7 @@
     "typescript": "catalog:build"
   },
   "dependencies": {
-    "ai": "^4.3.16",
-    "zod": "^3.25.48"
+    "ai": "^5.0.22",
+    "zod": "^4.1.0"
   }
-}
+}

packages/ai/src/plugin.ts

@@ -3,15 +3,16 @@ import { AiPluginOptions, CommandTool } from './types';
 import CommandKit, { getCommandKit, Logger } from 'commandkit';
 import { AiContext } from './context';
 import { Collection, Events, Message } from 'discord.js';
-import { tool, Tool, generateText } from 'ai';
+import { tool, Tool, generateText, stepCountIs, ModelMessage } from 'ai';
 import { getAiWorkerContext, runInAiWorkerContext } from './ai-context-worker';
 import { getAvailableCommands } from './tools/get-available-commands';
 import { getChannelById } from './tools/get-channel-by-id';
 import { getCurrentClientInfo } from './tools/get-current-client-info';
 import { getGuildById } from './tools/get-guild-by-id';
 import { getUserById } from './tools/get-user-by-id';
-import { AiMessage, getAIConfig } from './configure';
+import { getAIConfig } from './configure';
 import { augmentCommandKit } from './augmentation';
+import { ToolParameterType } from './tools/common';
 
 /**
  * Represents the configuration options for the AI plugin scoped to a specific command.
@@ -24,7 +25,7 @@ export interface AiConfig {
   /**
    * A zod schema defining the parameters that the AI command accepts.
    */
-  parameters: any;
+  inputSchema: ToolParameterType;
 }
 
 const defaultTools: Record<string, Tool> = {
@@ -99,11 +100,19 @@ export class AiPlugin extends RuntimePlugin<AiPluginOptions> {
     await runInAiWorkerContext(ctx, message, async () => {
       const systemPrompt = await prepareSystemPrompt(ctx, message);
       const prompt = await preparePrompt(ctx, message);
-      const { model, abortSignal, maxSteps, ...modelOptions } =
-        await selectAiModel(ctx, message);
-
-      const promptOrMessage =
-        typeof prompt === 'string' ? { prompt } : { messages: prompt };
+      const {
+        model,
+        abortSignal,
+        stopWhen,
+        prompt: _prompt,
+        ...modelOptions
+      } = await selectAiModel(ctx, message);
+
+      const promptOrMessage = (
+        typeof prompt === 'string' ? { prompt } : { messages: prompt }
+      ) as
+        | { prompt: string; messages: never }
+        | { messages: ModelMessage[]; prompt: never };
 
       await onProcessingStart(ctx, message);
 
@@ -112,7 +121,7 @@ export class AiPlugin extends RuntimePlugin<AiPluginOptions> {
           model,
           abortSignal: abortSignal ?? AbortSignal.timeout(60_000),
           system: systemPrompt,
-          maxSteps: maxSteps ?? 5,
+          stopWhen: stopWhen ?? stepCountIs(5),
           ...modelOptions,
           tools: {
             // Include built-in least significant tools if not disabled
@@ -181,12 +190,12 @@ export class AiPlugin extends RuntimePlugin<AiPluginOptions> {
 
       const cmdTool = tool({
         description,
-        parameters: cmd.data.aiConfig.parameters,
+        inputSchema: cmd.data.aiConfig.inputSchema,
         async execute(params) {
           const config = getAIConfig();
           const ctx = getAiWorkerContext();
 
-          ctx.ctx.setParams(params);
+          ctx.ctx.setParams(params as Record<string, unknown>);
 
           try {
             const target = await commandkit.commandHandler.prepareCommandRun(

packages/ai/src/tools/common/index.ts

@@ -14,7 +14,7 @@ type Awaitable<T> = T | Promise<T>;
  * Type representing the parameters schema for AI tools.
  * Extracted from the first parameter of the `tool` function from the 'ai' library.
  */
-export type ToolParameterType = Parameters<typeof tool>[0]['parameters'];
+export type ToolParameterType = z.ZodType | Schema<any>;
 
 /**
  * Utility type that infers the TypeScript type from a tool parameter schema.
@@ -39,7 +39,7 @@ export interface CreateToolOptions<T extends ToolParameterType, R = unknown> {
   /** A human-readable description of what the tool does */
   description: string;
   /** The parameter schema that defines the tool's input structure */
-  parameters: T;
+  inputSchema: T;
   /** The function that executes when the tool is called */
   execute: ToolExecuteFunction<T, R>;
 }
@@ -49,12 +49,12 @@ export interface CreateToolOptions<T extends ToolParameterType, R = unknown> {
  * @template T - The parameter schema type
  * @template R - The return type of the function
  * @param ctx - The AI context containing request and response information
- * @param parameters - The validated parameters passed to the tool
+ * @param inputSchema - The validated inputSchema passed to the tool
  * @returns The result of the tool execution, which can be synchronous or asynchronous
  */
 export type ToolExecuteFunction<T extends ToolParameterType, R> = (
   ctx: AiContext,
-  parameters: InferParameters<T>,
+  inputSchema: InferParameters<T>,
 ) => Awaitable<R>;
 
 /**
@@ -92,11 +92,11 @@ export function createTool<T extends ToolParameterType, R = unknown>(
   const _tool = tool<T, R>({
     name: options.name,
     description: options.description,
-    parameters: options.parameters,
+    inputSchema: options.inputSchema,
     async execute(params) {
       const { ctx } = getAiWorkerContext();
 
-      ctx.setParams(params);
+      ctx.setParams(params as Record<string, unknown>);
 
       return options.execute(ctx, params as InferParameters<T>);
     },

packages/ai/src/tools/get-available-commands.ts

@@ -4,7 +4,7 @@ import { createTool } from './common';
 export const getAvailableCommands = createTool({
   description: 'Get a list of all available commands.',
   name: 'getAvailableCommands',
-  parameters: z.object({}),
+  inputSchema: z.object({}),
   execute(ctx, params) {
     const { commandkit } = ctx;
 

packages/ai/src/tools/get-channel-by-id.ts

@@ -5,7 +5,7 @@ import { Logger } from 'commandkit';
 export const getChannelById = createTool({
   description: 'Get a channel by its ID.',
   name: 'getChannelById',
-  parameters: z.object({
+  inputSchema: z.object({
     channelId: z.string().describe('The ID of the channel to retrieve.'),
   }),
   async execute(ctx, params) {

packages/ai/src/tools/get-current-client-info.ts

@@ -4,7 +4,7 @@ import { createTool } from './common';
 export const getCurrentClientInfo = createTool({
   name: 'getCurrentClientInfo',
   description: 'Get information about the current discord bot user',
-  parameters: z.object({}),
+  inputSchema: z.object({}),
   execute: async (ctx, params) => {
     const { client } = ctx;
     const user = client.user;

packages/ai/src/tools/get-guild-by-id.ts

@@ -5,7 +5,7 @@ import { Logger } from 'commandkit';
 export const getGuildById = createTool({
   description: 'Get a guild by its ID.',
   name: 'getGuildById',
-  parameters: z.object({
+  inputSchema: z.object({
     guildId: z.string().describe('The ID of the guild to retrieve.'),
   }),
   async execute(ctx, params) {

packages/ai/src/tools/get-user-by-id.ts

@@ -5,7 +5,7 @@ import { Logger } from 'commandkit';
 export const getUserById = createTool({
   description: 'Get a user by their ID.',
   name: 'getUserById',
-  parameters: z.object({
+  inputSchema: z.object({
     userId: z.string().describe('The ID of the user to retrieve.'),
   }),
   async execute(ctx, params) {

packages/ai/src/types.ts

@@ -44,7 +44,7 @@ export interface AiPluginOptions {}
  * Extracts the AI configuration params.
  */
 export type ExtractAiConfig<T extends Record<string, unknown>> =
-  T extends AiConfig ? InferParameters<T['parameters']> : T;
+  T extends AiConfig ? InferParameters<T['inputSchema']> : T;
 
 /**
  * Represents the context in which an AI command is executed.