@witqq/agent-sdk 0.5.2 → 0.6.1

package/README.md

@@ -367,6 +367,18 @@ const agent = service.createAgent({
 
 Uses `generateText()` for runs, `generateObject()` for structured output, `streamText()` for streaming. Supports `supervisor.onAskUser` via an injected `ask_user` tool.
 
+Pass model-specific options via `providerOptions`:
+
+```typescript
+const agent = service.createAgent({
+  model: "google/gemini-2.0-flash",
+  systemPrompt: "Think step by step.",
+  providerOptions: {
+    google: { thinkingConfig: { thinkingBudget: 1024 } },
+  },
+});
+```
+
 ## Switching Backends
 
 All backends share the same `AgentConfig` and return the same `AgentResult`. To switch backends, change only the service creation:
@@ -492,17 +504,17 @@ interface CopilotAuthToken extends AuthToken {
 
 ## Interactive Demo
 
-A web-based interactive demo showcasing all backends with authentication is in `examples/auth-demo/`.
+An interactive demo for testing multi-turn conversations across all backends is in `examples/auth-demo/`.
 
 ```bash
-# Run in Docker (web UI at http://localhost:3456)
-docker compose -f examples/auth-demo/docker-compose.yml up
-
-# Run locally (CLI mode)
+# CLI mode
 npx tsx examples/auth-demo/index.ts
+
+# Web UI via Docker (http://localhost:3456)
+docker compose -f examples/auth-demo/docker-compose.yml up
 ```
 
-Features: provider selection, auth flows (Device Flow, OAuth+PKCE, API key), model selection, streaming chat with conversation history, token persistence across container restarts, and provider switching.
+Features: backend selection, auth flows (Device Flow, OAuth+PKCE, API key), multi-turn chat with persistent sessions, 7 keyboard shortcuts for common test messages, 3 demo tools (search, calculator, format with approval), streaming event display with ANSI colors, per-turn statistics, and provider switching.
 
 ## License
 

package/dist/backends/claude.cjs

@@ -327,6 +327,14 @@ function extractSchemaFromDef(schema) {
 }
 
 // src/backends/claude.ts
+var MCP_SERVER_NAME = "agent-sdk-tools";
+var MCP_TOOL_PREFIX = `mcp__${MCP_SERVER_NAME}__`;
+function mcpToolName(toolName) {
+  return `${MCP_TOOL_PREFIX}${toolName}`;
+}
+function stripMcpPrefix(name) {
+  return name.startsWith(MCP_TOOL_PREFIX) ? name.slice(MCP_TOOL_PREFIX.length) : name;
+}
 var sdkModule = null;
 async function loadSDK() {
   if (sdkModule) return sdkModule;
@@ -350,11 +358,13 @@ var ANTHROPIC_API_VERSION = "2023-06-01";
 var ANTHROPIC_OAUTH_BETA = "oauth-2025-04-20";
 function buildMcpServer(sdk, tools, toolResultCapture) {
   if (tools.length === 0) return void 0;
-  const mcpTools = tools.map(
-    (tool) => sdk.tool(
+  const mcpTools = tools.map((tool) => {
+    const zodSchema = tool.parameters;
+    const inputSchema = zodSchema.shape ?? zodToJsonSchema(tool.parameters);
+    return sdk.tool(
       tool.name,
       tool.description ?? "",
-      zodToJsonSchema(tool.parameters),
+      inputSchema,
       async (args) => {
         const result = await tool.execute(args);
         if (toolResultCapture) {
@@ -369,10 +379,10 @@ function buildMcpServer(sdk, tools, toolResultCapture) {
           ]
         };
       }
-    )
-  );
+    );
+  });
   return sdk.createSdkMcpServer({
-    name: "agent-sdk-tools",
+    name: MCP_SERVER_NAME,
     version: "1.0.0",
     tools: mcpTools
   });
@@ -409,7 +419,8 @@ function buildCanUseTool(config) {
   const onPermission = config.supervisor?.onPermission;
   if (!onPermission) return void 0;
   const permissionStore = config.permissionStore;
-  return async (toolName, input, options) => {
+  return async (rawToolName, input, options) => {
+    const toolName = stripMcpPrefix(rawToolName);
     if (permissionStore && await permissionStore.isApproved(toolName)) {
       return {
         behavior: "allow",
@@ -491,14 +502,10 @@ function mapSDKMessage(msg, thinkingBlockIndices, toolCallTracker) {
       const betaMessage = msg.message;
       if (!betaMessage?.content) return null;
       const events = [];
-      const textParts = betaMessage.content.filter((b) => b.type === "text" && b.text).map((b) => b.text).join("");
-      if (textParts) {
-        events.push({ type: "text_delta", text: textParts });
-      }
       for (const block of betaMessage.content) {
         if (block.type === "tool_use") {
           const toolCallId = String(block.id ?? "");
-          const toolName = block.name ?? "unknown";
+          const toolName = stripMcpPrefix(block.name ?? "unknown");
           if (toolCallTracker) {
             toolCallTracker.trackStart(toolCallId, toolName);
           }
@@ -521,7 +528,7 @@ function mapSDKMessage(msg, thinkingBlockIndices, toolCallTracker) {
     }
     case "tool_use_summary": {
       const summary = msg.summary;
-      const toolName = msg.tool_name ?? "unknown";
+      const toolName = stripMcpPrefix(msg.tool_name ?? "unknown");
       const precedingIds = msg.preceding_tool_use_ids;
       let toolCallId = "";
       if (precedingIds && precedingIds.length > 0) {
@@ -530,15 +537,12 @@ function mapSDKMessage(msg, thinkingBlockIndices, toolCallTracker) {
       } else if (toolCallTracker) {
         toolCallId = toolCallTracker.consumeToolCallId(toolName);
       }
-      if (summary) {
-        return {
-          type: "tool_call_end",
-          toolCallId,
-          toolName,
-          result: summary
-        };
-      }
-      return null;
+      return {
+        type: "tool_call_end",
+        toolCallId,
+        toolName,
+        result: summary ?? null
+      };
     }
     case "stream_event": {
       const event = msg.event;
@@ -566,10 +570,7 @@ function mapSDKMessage(msg, thinkingBlockIndices, toolCallTracker) {
       return null;
     }
     case "tool_progress": {
-      const toolName = msg.tool_name;
-      if (!toolName) return null;
-      const toolCallId = toolCallTracker?.peekToolCallId(toolName) ?? "";
-      return { type: "tool_call_start", toolCallId, toolName, args: {} };
+      return null;
     }
     case "result": {
       if (msg.subtype === "success") {
@@ -664,6 +665,9 @@ var ClaudeAgent = class extends BaseAgent {
     if (opts.canUseTool && !opts.permissionMode) {
       opts.permissionMode = "default";
     }
+    if (this.config.availableTools) {
+      opts.allowedTools = [...this.config.availableTools];
+    }
     return opts;
   }
   async buildMcpConfig(opts, toolResultCapture) {
@@ -672,8 +676,10 @@ var ClaudeAgent = class extends BaseAgent {
     const mcpServer = buildMcpServer(sdk, this.tools, toolResultCapture);
     if (mcpServer) {
       opts.mcpServers = {
-        "agent-sdk-tools": mcpServer
+        [MCP_SERVER_NAME]: mcpServer
       };
+      const mcpToolNames = this.tools.map((t) => mcpToolName(t.name));
+      opts.allowedTools = [...opts.allowedTools ?? [], ...mcpToolNames];
     }
     return opts;
   }
@@ -698,7 +704,7 @@ var ClaudeAgent = class extends BaseAgent {
           if (betaMessage?.content) {
             for (const block of betaMessage.content) {
               if (block.type === "tool_use") {
-                const toolName = block.name ?? "unknown";
+                const toolName = stripMcpPrefix(block.name ?? "unknown");
                 toolCalls.push({
                   toolName,
                   args: block.input ?? {},
@@ -758,7 +764,8 @@ var ClaudeAgent = class extends BaseAgent {
     const isResuming = this.isPersistent && this._sessionId !== void 0;
     const prompt = isResuming ? extractLastUserPrompt(messages) : buildContextualPrompt(messages);
     let opts = this.buildQueryOptions(signal);
-    opts = await this.buildMcpConfig(opts);
+    const toolResultCapture = /* @__PURE__ */ new Map();
+    opts = await this.buildMcpConfig(opts, toolResultCapture);
     const jsonSchema = zodToJsonSchema(schema.schema);
     opts.outputFormat = {
       type: "json_schema",
@@ -772,6 +779,30 @@ var ClaudeAgent = class extends BaseAgent {
     let usage;
     try {
       for await (const msg of q) {
+        if (msg.type === "assistant") {
+          const betaMessage = msg.message;
+          if (betaMessage?.content) {
+            for (const block of betaMessage.content) {
+              if (block.type === "tool_use") {
+                const toolName = stripMcpPrefix(block.name ?? "unknown");
+                toolCalls.push({
+                  toolName,
+                  args: block.input ?? {},
+                  result: toolResultCapture.get(toolName) ?? null,
+                  approved: true
+                });
+              }
+            }
+          }
+        }
+        if (msg.type === "tool_use_summary" || msg.type === "result") {
+          for (const tc of toolCalls) {
+            if (tc.result === null) {
+              const captured = toolResultCapture.get(tc.toolName);
+              if (captured !== void 0) tc.result = captured;
+            }
+          }
+        }
         if (msg.type === "result" && msg.subtype === "success") {
           const r = msg;
           output = r.result;
@@ -836,13 +867,10 @@ var ClaudeAgent = class extends BaseAgent {
     try {
       for await (const msg of q) {
         if (signal.aborted) throw new AbortError();
-        const event = mapSDKMessage(msg, thinkingBlockIndices, toolCallTracker);
-        if (event) {
-          if (Array.isArray(event)) {
-            for (const e of event) yield e;
-          } else {
-            yield event;
-          }
+        const events = mapSDKMessage(msg, thinkingBlockIndices, toolCallTracker);
+        if (events) {
+          const mapped = Array.isArray(events) ? events : [events];
+          for (const e of mapped) yield e;
         }
         if (msg.type === "result" && msg.subtype === "success") {
           const r = msg;