@witqq/agent-sdk 0.5.2 → 0.6.1

package/dist/backends/vercel-ai.d.cts

@@ -1,37 +1,37 @@
-import { V as VercelAIBackendOptions, I as IAgentService } from '../types-CQEyww7V.cjs';
+import { V as VercelAIBackendOptions, I as IAgentService } from '../types-BvwNzZCj.cjs';
 import 'zod';
 
 /** @internal Vercel AI SDK tool result */
 interface SDKToolDefinition {
     description: string;
-    parameters: any;
-    execute?: (args: any, options: any) => Promise<any>;
-    needsApproval?: boolean | ((args: any, options: any) => Promise<boolean>);
+    inputSchema: unknown;
+    execute?: (input: unknown, options: unknown) => Promise<unknown>;
+    needsApproval?: boolean | ((input: unknown, options: unknown) => Promise<boolean>);
 }
-/** @internal Vercel AI SDK generateText result */
+/** @internal Vercel AI SDK v6 generateText result */
 interface SDKGenerateTextResult {
     text: string;
     toolCalls: Array<{
         toolCallId: string;
         toolName: string;
-        args: any;
+        input: unknown;
     }>;
     toolResults: Array<{
         toolCallId: string;
         toolName: string;
-        result: any;
+        output: unknown;
     }>;
     steps: Array<{
         text: string;
         toolCalls: Array<{
             toolCallId: string;
             toolName: string;
-            args: any;
+            input: unknown;
         }>;
         toolResults: Array<{
             toolCallId: string;
             toolName: string;
-            result: any;
+            output: unknown;
         }>;
         usage: {
             inputTokens?: number;
@@ -45,12 +45,12 @@ interface SDKGenerateTextResult {
     };
     finishReason: string;
     response: {
-        messages: any[];
+        messages: unknown[];
     };
 }
 /** @internal Vercel AI SDK generateObject result */
 interface SDKGenerateObjectResult {
-    object: any;
+    object: unknown;
     usage: {
         inputTokens?: number;
         outputTokens?: number;
@@ -65,24 +65,66 @@ interface SDKStreamTextResult {
     }>;
     text: PromiseLike<string>;
 }
-/** @internal Vercel AI SDK stream part union */
-interface SDKStreamPart {
+/** @internal Vercel AI SDK v6 stream part union */
+type SDKStreamPart = {
+    type: "text-delta";
+    text: string;
+} | {
+    type: "tool-call";
+    toolCallId: string;
+    toolName: string;
+    input: unknown;
+} | {
+    type: "tool-result";
+    toolCallId: string;
+    toolName: string;
+    output: unknown;
+} | {
+    type: "tool-error";
+    toolCallId: string;
+    toolName: string;
+    error: unknown;
+} | {
+    type: "reasoning-start";
+} | {
+    type: "reasoning-end";
+} | {
+    type: "reasoning-delta";
+    text: string;
+} | {
+    type: "finish-step";
+    usage: {
+        inputTokens?: number;
+        outputTokens?: number;
+    };
+    finishReason: string;
+} | {
+    type: "finish";
+    finishReason: string;
+    totalUsage: {
+        inputTokens?: number;
+        outputTokens?: number;
+    };
+} | {
+    type: "error";
+    error: unknown;
+} | {
     type: string;
-    [key: string]: any;
-}
-/** @internal Vercel AI SDK LanguageModel */
-type SDKLanguageModel = any;
+};
+/** @internal Vercel AI SDK LanguageModel — opaque type from SDK */
+type SDKLanguageModel = Record<string, unknown>;
 /** @internal SDK module shape */
 interface SDKModule {
-    generateText: (options: any) => Promise<SDKGenerateTextResult>;
-    streamText: (options: any) => SDKStreamTextResult;
-    generateObject: (options: any) => Promise<SDKGenerateObjectResult>;
-    tool: (options: any) => SDKToolDefinition;
-    jsonSchema: (schema: any) => any;
+    generateText: (options: Record<string, unknown>) => Promise<SDKGenerateTextResult>;
+    streamText: (options: Record<string, unknown>) => SDKStreamTextResult;
+    generateObject: (options: Record<string, unknown>) => Promise<SDKGenerateObjectResult>;
+    tool: (options: Record<string, unknown>) => SDKToolDefinition;
+    jsonSchema: (schema: unknown) => unknown;
+    stepCountIs: (count: number) => unknown;
 }
 /** @internal OpenAI-compatible module shape */
 interface SDKCompatModule {
-    createOpenAICompatible: (options: any) => {
+    createOpenAICompatible: (options: Record<string, unknown>) => {
         chatModel: (modelId: string) => SDKLanguageModel;
         languageModel: (modelId: string) => SDKLanguageModel;
     };

package/dist/backends/vercel-ai.d.ts

@@ -1,37 +1,37 @@
-import { V as VercelAIBackendOptions, I as IAgentService } from '../types-CQEyww7V.js';
+import { V as VercelAIBackendOptions, I as IAgentService } from '../types-BvwNzZCj.js';
 import 'zod';
 
 /** @internal Vercel AI SDK tool result */
 interface SDKToolDefinition {
     description: string;
-    parameters: any;
-    execute?: (args: any, options: any) => Promise<any>;
-    needsApproval?: boolean | ((args: any, options: any) => Promise<boolean>);
+    inputSchema: unknown;
+    execute?: (input: unknown, options: unknown) => Promise<unknown>;
+    needsApproval?: boolean | ((input: unknown, options: unknown) => Promise<boolean>);
 }
-/** @internal Vercel AI SDK generateText result */
+/** @internal Vercel AI SDK v6 generateText result */
 interface SDKGenerateTextResult {
     text: string;
     toolCalls: Array<{
         toolCallId: string;
         toolName: string;
-        args: any;
+        input: unknown;
     }>;
     toolResults: Array<{
         toolCallId: string;
         toolName: string;
-        result: any;
+        output: unknown;
     }>;
     steps: Array<{
         text: string;
         toolCalls: Array<{
             toolCallId: string;
             toolName: string;
-            args: any;
+            input: unknown;
         }>;
         toolResults: Array<{
             toolCallId: string;
             toolName: string;
-            result: any;
+            output: unknown;
         }>;
         usage: {
             inputTokens?: number;
@@ -45,12 +45,12 @@ interface SDKGenerateTextResult {
     };
     finishReason: string;
     response: {
-        messages: any[];
+        messages: unknown[];
     };
 }
 /** @internal Vercel AI SDK generateObject result */
 interface SDKGenerateObjectResult {
-    object: any;
+    object: unknown;
     usage: {
         inputTokens?: number;
         outputTokens?: number;
@@ -65,24 +65,66 @@ interface SDKStreamTextResult {
     }>;
     text: PromiseLike<string>;
 }
-/** @internal Vercel AI SDK stream part union */
-interface SDKStreamPart {
+/** @internal Vercel AI SDK v6 stream part union */
+type SDKStreamPart = {
+    type: "text-delta";
+    text: string;
+} | {
+    type: "tool-call";
+    toolCallId: string;
+    toolName: string;
+    input: unknown;
+} | {
+    type: "tool-result";
+    toolCallId: string;
+    toolName: string;
+    output: unknown;
+} | {
+    type: "tool-error";
+    toolCallId: string;
+    toolName: string;
+    error: unknown;
+} | {
+    type: "reasoning-start";
+} | {
+    type: "reasoning-end";
+} | {
+    type: "reasoning-delta";
+    text: string;
+} | {
+    type: "finish-step";
+    usage: {
+        inputTokens?: number;
+        outputTokens?: number;
+    };
+    finishReason: string;
+} | {
+    type: "finish";
+    finishReason: string;
+    totalUsage: {
+        inputTokens?: number;
+        outputTokens?: number;
+    };
+} | {
+    type: "error";
+    error: unknown;
+} | {
     type: string;
-    [key: string]: any;
-}
-/** @internal Vercel AI SDK LanguageModel */
-type SDKLanguageModel = any;
+};
+/** @internal Vercel AI SDK LanguageModel — opaque type from SDK */
+type SDKLanguageModel = Record<string, unknown>;
 /** @internal SDK module shape */
 interface SDKModule {
-    generateText: (options: any) => Promise<SDKGenerateTextResult>;
-    streamText: (options: any) => SDKStreamTextResult;
-    generateObject: (options: any) => Promise<SDKGenerateObjectResult>;
-    tool: (options: any) => SDKToolDefinition;
-    jsonSchema: (schema: any) => any;
+    generateText: (options: Record<string, unknown>) => Promise<SDKGenerateTextResult>;
+    streamText: (options: Record<string, unknown>) => SDKStreamTextResult;
+    generateObject: (options: Record<string, unknown>) => Promise<SDKGenerateObjectResult>;
+    tool: (options: Record<string, unknown>) => SDKToolDefinition;
+    jsonSchema: (schema: unknown) => unknown;
+    stepCountIs: (count: number) => unknown;
 }
 /** @internal OpenAI-compatible module shape */
 interface SDKCompatModule {
-    createOpenAICompatible: (options: any) => {
+    createOpenAICompatible: (options: Record<string, unknown>) => {
         chatModel: (modelId: string) => SDKLanguageModel;
         languageModel: (modelId: string) => SDKLanguageModel;
     };

package/dist/backends/vercel-ai.js

@@ -375,10 +375,10 @@ function mapToolsToSDK(sdk, tools, config, sessionApprovals, permissionStore, si
     const jsonSchema = zodToJsonSchema(ourTool.parameters);
     toolMap[ourTool.name] = sdk.tool({
       description: ourTool.description,
-      parameters: sdk.jsonSchema(jsonSchema),
+      inputSchema: sdk.jsonSchema(jsonSchema),
       execute: wrapToolExecute(ourTool, supervisor, sessionApprovals, permissionStore, signal),
       ...ourTool.needsApproval && supervisor?.onPermission ? {
-        needsApproval: async (_args) => {
+        needsApproval: async (_input) => {
           if (permissionStore && await permissionStore.isApproved(ourTool.name)) return false;
           if (sessionApprovals.has(ourTool.name)) return false;
           return true;
@@ -390,7 +390,7 @@ function mapToolsToSDK(sdk, tools, config, sessionApprovals, permissionStore, si
     const onAskUser = supervisor.onAskUser;
     toolMap["ask_user"] = sdk.tool({
       description: "Ask the user a question and wait for their response",
-      parameters: sdk.jsonSchema({
+      inputSchema: sdk.jsonSchema({
         type: "object",
         properties: {
           question: { type: "string", description: "The question to ask the user" }
@@ -468,46 +468,60 @@ function messagesToSDK(messages) {
 }
 function mapStreamPart(part) {
   switch (part.type) {
-    case "text-delta":
-      return { type: "text_delta", text: part.text ?? "" };
-    case "tool-call":
+    case "text-delta": {
+      const p = part;
+      return { type: "text_delta", text: p.text ?? "" };
+    }
+    case "tool-call": {
+      const p = part;
       return {
         type: "tool_call_start",
-        toolCallId: String(part.toolCallId ?? ""),
-        toolName: part.toolName ?? "unknown",
-        args: part.args ?? {}
+        toolCallId: String(p.toolCallId ?? ""),
+        toolName: p.toolName ?? "unknown",
+        args: p.input ?? {}
       };
-    case "tool-result":
+    }
+    case "tool-result": {
+      const p = part;
       return {
         type: "tool_call_end",
-        toolCallId: String(part.toolCallId ?? ""),
-        toolName: part.toolName ?? "unknown",
-        result: part.result ?? null
+        toolCallId: String(p.toolCallId ?? ""),
+        toolName: p.toolName ?? "unknown",
+        result: p.output ?? null
       };
-    case "tool-error":
+    }
+    case "tool-error": {
+      const p = part;
       return {
         type: "error",
-        error: part.error instanceof Error ? part.error.message : String(part.error ?? "Tool execution failed"),
+        error: p.error instanceof Error ? p.error.message : String(p.error ?? "Tool execution failed"),
         recoverable: true
       };
+    }
     case "reasoning-start":
       return { type: "thinking_start" };
     case "reasoning-end":
       return { type: "thinking_end" };
-    case "reasoning-delta":
-      return { type: "thinking_delta", text: part.text ?? "" };
-    case "finish-step":
+    case "reasoning-delta": {
+      const p = part;
+      return { type: "thinking_delta", text: p.text ?? "" };
+    }
+    case "finish-step": {
+      const p = part;
       return {
         type: "usage_update",
-        promptTokens: Number(part.usage?.inputTokens ?? 0),
-        completionTokens: Number(part.usage?.outputTokens ?? 0)
+        promptTokens: Number(p.usage?.inputTokens ?? 0),
+        completionTokens: Number(p.usage?.outputTokens ?? 0)
       };
-    case "error":
+    }
+    case "error": {
+      const p = part;
       return {
         type: "error",
-        error: part.error instanceof Error ? part.error.message : String(part.error ?? "Unknown error"),
+        error: p.error instanceof Error ? p.error.message : String(p.error ?? "Unknown error"),
         recoverable: false
       };
+    }
     default:
       return null;
   }
@@ -551,7 +565,7 @@ var VercelAIAgent = class extends BaseAgent {
       system: this.config.systemPrompt,
       messages: sdkMessages,
       tools: hasTools ? tools : void 0,
-      maxSteps: maxTurns,
+      stopWhen: sdk.stepCountIs(maxTurns),
       abortSignal: signal,
       ...this.config.modelParams?.temperature !== void 0 && {
         temperature: this.config.modelParams.temperature
@@ -561,6 +575,9 @@ var VercelAIAgent = class extends BaseAgent {
       },
       ...this.config.modelParams?.topP !== void 0 && {
         topP: this.config.modelParams.topP
+      },
+      ...this.config.providerOptions && {
+        providerOptions: this.config.providerOptions
       }
     });
     const toolCalls = [];
@@ -571,8 +588,8 @@ var VercelAIAgent = class extends BaseAgent {
         );
         toolCalls.push({
           toolName: tc.toolName,
-          args: tc.args ?? {},
-          result: matchingResult?.result ?? null,
+          args: tc.input ?? {},
+          result: matchingResult?.output ?? null,
           approved: true
         });
       }
@@ -581,13 +598,15 @@ var VercelAIAgent = class extends BaseAgent {
       promptTokens: Number(result.totalUsage?.inputTokens ?? 0),
       completionTokens: Number(result.totalUsage?.outputTokens ?? 0)
     };
+    const lastStep = result.steps.length > 0 ? result.steps[result.steps.length - 1] : null;
+    const outputText = lastStep?.text || null;
     return {
-      output: result.text || null,
+      output: outputText,
       structuredOutput: void 0,
       toolCalls,
       messages: [
         ...messages,
-        ...result.text ? [{ role: "assistant", content: result.text }] : []
+        ...outputText ? [{ role: "assistant", content: outputText }] : []
       ],
       usage
     };
@@ -612,6 +631,9 @@ var VercelAIAgent = class extends BaseAgent {
       },
       ...this.config.modelParams?.maxTokens !== void 0 && {
         maxTokens: this.config.modelParams.maxTokens
+      },
+      ...this.config.providerOptions && {
+        providerOptions: this.config.providerOptions
       }
     });
     let structuredOutput;
@@ -648,7 +670,7 @@ var VercelAIAgent = class extends BaseAgent {
       system: this.config.systemPrompt,
       messages: sdkMessages,
       tools: hasTools ? tools : void 0,
-      maxSteps: maxTurns,
+      stopWhen: sdk.stepCountIs(maxTurns),
       abortSignal: signal,
       ...this.config.modelParams?.temperature !== void 0 && {
         temperature: this.config.modelParams.temperature
@@ -658,6 +680,9 @@ var VercelAIAgent = class extends BaseAgent {
       },
       ...this.config.modelParams?.topP !== void 0 && {
         topP: this.config.modelParams.topP
+      },
+      ...this.config.providerOptions && {
+        providerOptions: this.config.providerOptions
       }
     });
     let finalText = "";
@@ -669,6 +694,12 @@ var VercelAIAgent = class extends BaseAgent {
         if (part.type === "text-delta") {
           finalText += part.text ?? "";
         }
+        if (part.type === "finish-step") {
+          const p = part;
+          if (p.finishReason === "tool-calls") {
+            finalText = "";
+          }
+        }
       }
       const totalUsage = await result.totalUsage;
       yield {