@micdrop/ai-sdk - npm Package Compare versions

Comparing version 1.0.5 to 1.0.6

LICENSE

+MIT License
+
+Copyright (c) 2022 Rolebase
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

dist/index.d.mts

@@ -1,2 +0,2 @@
-import { AgentOptions, Agent } from '@micdrop/server';
+import { Agent, AgentOptions } from '@micdrop/server';
 import { LanguageModel, CallSettings } from 'ai';
@@ -3,0 +3,0 @@ import { PassThrough } from 'stream';

dist/index.d.ts

@@ -1,2 +0,2 @@
-import { AgentOptions, Agent } from '@micdrop/server';
+import { Agent, AgentOptions } from '@micdrop/server';
 import { LanguageModel, CallSettings } from 'ai';
@@ -3,0 +3,0 @@ import { PassThrough } from 'stream';

dist/index.js

@@ -133,3 +133,2 @@ "use strict";
       tools[t.name] = (0, import_ai.tool)({
-        name: t.name,
         description: t.description,
@@ -136,0 +135,0 @@ inputSchema: t.inputSchema || import_zod.z.object({}),

dist/index.js.map

@@ -1,1 +0,1 @@
-{"version":3,"sources":["../src/index.ts","../src/AiSdkAgent.ts"],"sourcesContent":["export * from './AiSdkAgent'","import { Agent, AgentOptions } from '@micdrop/server'\nimport {\n  CallSettings,\n  LanguageModel,\n  ModelMessage,\n  stepCountIs,\n  streamText,\n  tool,\n  ToolSet,\n} from 'ai'\nimport { PassThrough } from 'stream'\nimport { z } from 'zod'\n\nexport interface AiSdkAgentOptions extends AgentOptions {\n  model: LanguageModel\n  settings?: CallSettings\n  maxRetry?: number\n}\n\nconst DEFAULT_MAX_RETRY = 3\n\nexport class AiSdkAgent extends Agent<AiSdkAgentOptions> {\n  private abortController?: AbortController\n\n  protected async generateAnswer(stream: PassThrough): Promise<void> {\n    const abortController = new AbortController()\n    this.abortController = abortController\n    const signal = this.abortController.signal\n    let skipAnswer = false\n\n    // Prepare extracting\n    let extracting = false\n    const extractOptions = this.getExtractOptions()\n\n    try {\n      // Generate response using AI SDK\n      const result = streamText({\n        model: this.options.model,\n        messages: this.buildMessages(),\n        tools: this.buildTools(),\n        maxRetries: this.options.maxRetry ?? DEFAULT_MAX_RETRY,\n        stopWhen: stepCountIs(5),\n        onStepFinish: (step) => {\n          const tools = step.toolCalls.map((toolCall) =>\n            this.getTool(toolCall.toolName)\n          )\n          if (tools.some((t) => t?.skipAnswer)) {\n            skipAnswer = true\n          }\n        },\n        ...this.options.settings,\n        abortSignal: signal,\n      })\n\n      for await (const textPart of result.textStream) {\n        if (skipAnswer) return\n        this.log(`Answer chunk: \"${textPart}\"`)\n\n        // Extracting value?\n        if (extractOptions) {\n          if (!extracting) {\n            const startTagIndex = textPart.indexOf(extractOptions.startTag)\n            if (startTagIndex !== -1) {\n              extracting = true\n              const messagePart = textPart.slice(0, startTagIndex).trimEnd()\n              stream.write(messagePart)\n              continue\n            }\n          } else {\n            // Extracting, don't write to stream\n            continue\n          }\n        }\n\n        // Send chunk\n        stream.write(textPart)\n      }\n      if (skipAnswer) return\n\n      const fullText = await result.text\n      const { message, metadata } = this.extract(fullText)\n\n      // Emit message\n      this.addAssistantMessage(message, metadata)\n      this.abortController = undefined\n    } catch (error: any) {\n      if (abortController === this.abortController) {\n        console.error('[AiSdkAgent] Error answering:', error)\n      }\n    }\n  }\n\n  private buildMessages(): ModelMessage[] {\n    return this.conversation.map((message): ModelMessage => {\n      switch (message.role) {\n        case 'user':\n        case 'assistant':\n        case 'system':\n          return {\n            role: message.role,\n            content: message.content,\n          }\n        case 'tool_call':\n          return {\n            role: 'assistant',\n            content: [\n              {\n                type: 'tool-call',\n                toolCallId: message.toolCallId,\n                toolName: message.toolName,\n                input: JSON.parse(message.parameters),\n              },\n            ],\n          }\n        case 'tool_result':\n          return {\n            role: 'tool',\n            content: [\n              {\n                type: 'tool-result',\n                toolCallId: message.toolCallId,\n                toolName: message.toolName,\n                output: {\n                  type: 'json',\n                  value: JSON.parse(message.output),\n                },\n              },\n            ],\n          }\n      }\n    })\n  }\n\n  private buildTools() {\n    const tools: ToolSet = {}\n\n    // Disable tools if first message\n    const enableTools = this.conversation.length > 1\n    if (!enableTools) return tools\n\n    this.tools.forEach((t) => {\n      tools[t.name] = tool({\n        name: t.name,\n        description: t.description,\n        inputSchema: t.inputSchema || z.object({}),\n        execute: (args, { toolCallId }) =>\n          this.executeTool({\n            role: 'tool_call',\n            toolCallId,\n            toolName: t.name,\n            parameters: JSON.stringify(args),\n          }),\n      })\n    })\n    return tools\n  }\n\n  cancel() {\n    if (!this.abortController) return\n    this.log('Cancel')\n    this.abortController.abort()\n    this.abortController = undefined\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,oBAAoC;AACpC,gBAQO;AAEP,iBAAkB;AAQlB,IAAM,oBAAoB;AAEnB,IAAM,aAAN,cAAyB,oBAAyB;AAAA,EAGvD,MAAgB,eAAe,QAAoC;AACjE,UAAM,kBAAkB,IAAI,gBAAgB;AAC5C,SAAK,kBAAkB;AACvB,UAAM,SAAS,KAAK,gBAAgB;AACpC,QAAI,aAAa;AAGjB,QAAI,aAAa;AACjB,UAAM,iBAAiB,KAAK,kBAAkB;AAE9C,QAAI;AAEF,YAAM,aAAS,sBAAW;AAAA,QACxB,OAAO,KAAK,QAAQ;AAAA,QACpB,UAAU,KAAK,cAAc;AAAA,QAC7B,OAAO,KAAK,WAAW;AAAA,QACvB,YAAY,KAAK,QAAQ,YAAY;AAAA,QACrC,cAAU,uBAAY,CAAC;AAAA,QACvB,cAAc,CAAC,SAAS;AACtB,gBAAM,QAAQ,KAAK,UAAU;AAAA,YAAI,CAAC,aAChC,KAAK,QAAQ,SAAS,QAAQ;AAAA,UAChC;AACA,cAAI,MAAM,KAAK,CAAC,MAAM,GAAG,UAAU,GAAG;AACpC,yBAAa;AAAA,UACf;AAAA,QACF;AAAA,QACA,GAAG,KAAK,QAAQ;AAAA,QAChB,aAAa;AAAA,MACf,CAAC;AAED,uBAAiB,YAAY,OAAO,YAAY;AAC9C,YAAI,WAAY;AAChB,aAAK,IAAI,kBAAkB,QAAQ,GAAG;AAGtC,YAAI,gBAAgB;AAClB,cAAI,CAAC,YAAY;AACf,kBAAM,gBAAgB,SAAS,QAAQ,eAAe,QAAQ;AAC9D,gBAAI,kBAAkB,IAAI;AACxB,2BAAa;AACb,oBAAM,cAAc,SAAS,MAAM,GAAG,aAAa,EAAE,QAAQ;AAC7D,qBAAO,MAAM,WAAW;AACxB;AAAA,YACF;AAAA,UACF,OAAO;AAEL;AAAA,UACF;AAAA,QACF;AAGA,eAAO,MAAM,QAAQ;AAAA,MACvB;AACA,UAAI,WAAY;AAEhB,YAAM,WAAW,MAAM,OAAO;AAC9B,YAAM,EAAE,SAAS,SAAS,IAAI,KAAK,QAAQ,QAAQ;AAGnD,WAAK,oBAAoB,SAAS,QAAQ;AAC1C,WAAK,kBAAkB;AAAA,IACzB,SAAS,OAAY;AACnB,UAAI,oBAAoB,KAAK,iBAAiB;AAC5C,gBAAQ,MAAM,iCAAiC,KAAK;AAAA,MACtD;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,gBAAgC;AACtC,WAAO,KAAK,aAAa,IAAI,CAAC,YAA0B;AACtD,cAAQ,QAAQ,MAAM;AAAA,QACpB,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AACH,iBAAO;AAAA,YACL,MAAM,QAAQ;AAAA,YACd,SAAS,QAAQ;AAAA,UACnB;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,YAAY,QAAQ;AAAA,gBACpB,UAAU,QAAQ;AAAA,gBAClB,OAAO,KAAK,MAAM,QAAQ,UAAU;AAAA,cACtC;AAAA,YACF;AAAA,UACF;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,YAAY,QAAQ;AAAA,gBACpB,UAAU,QAAQ;AAAA,gBAClB,QAAQ;AAAA,kBACN,MAAM;AAAA,kBACN,OAAO,KAAK,MAAM,QAAQ,MAAM;AAAA,gBAClC;AAAA,cACF;AAAA,YACF;AAAA,UACF;AAAA,MACJ;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,aAAa;AACnB,UAAM,QAAiB,CAAC;AAGxB,UAAM,cAAc,KAAK,aAAa,SAAS;AAC/C,QAAI,CAAC,YAAa,QAAO;AAEzB,SAAK,MAAM,QAAQ,CAAC,MAAM;AACxB,YAAM,EAAE,IAAI,QAAI,gBAAK;AAAA,QACnB,MAAM,EAAE;AAAA,QACR,aAAa,EAAE;AAAA,QACf,aAAa,EAAE,eAAe,aAAE,OAAO,CAAC,CAAC;AAAA,QACzC,SAAS,CAAC,MAAM,EAAE,WAAW,MAC3B,KAAK,YAAY;AAAA,UACf,MAAM;AAAA,UACN;AAAA,UACA,UAAU,EAAE;AAAA,UACZ,YAAY,KAAK,UAAU,IAAI;AAAA,QACjC,CAAC;AAAA,MACL,CAAC;AAAA,IACH,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEA,SAAS;AACP,QAAI,CAAC,KAAK,gBAAiB;AAC3B,SAAK,IAAI,QAAQ;AACjB,SAAK,gBAAgB,MAAM;AAC3B,SAAK,kBAAkB;AAAA,EACzB;AACF;","names":[]}
+{"version":3,"sources":["../src/index.ts","../src/AiSdkAgent.ts"],"sourcesContent":["export * from './AiSdkAgent'","import { Agent, AgentOptions } from '@micdrop/server'\nimport {\n  CallSettings,\n  LanguageModel,\n  ModelMessage,\n  stepCountIs,\n  streamText,\n  tool,\n  ToolSet,\n} from 'ai'\nimport { PassThrough } from 'stream'\nimport { z } from 'zod'\n\nexport interface AiSdkAgentOptions extends AgentOptions {\n  model: LanguageModel\n  settings?: CallSettings\n  maxRetry?: number\n}\n\nconst DEFAULT_MAX_RETRY = 3\n\nexport class AiSdkAgent extends Agent<AiSdkAgentOptions> {\n  private abortController?: AbortController\n\n  protected async generateAnswer(stream: PassThrough): Promise<void> {\n    const abortController = new AbortController()\n    this.abortController = abortController\n    const signal = this.abortController.signal\n    let skipAnswer = false\n\n    // Prepare extracting\n    let extracting = false\n    const extractOptions = this.getExtractOptions()\n\n    try {\n      // Generate response using AI SDK\n      const result = streamText({\n        model: this.options.model,\n        messages: this.buildMessages(),\n        tools: this.buildTools(),\n        maxRetries: this.options.maxRetry ?? DEFAULT_MAX_RETRY,\n        stopWhen: stepCountIs(5),\n        onStepFinish: (step) => {\n          const tools = step.toolCalls.map((toolCall) =>\n            this.getTool(toolCall.toolName)\n          )\n          if (tools.some((t) => t?.skipAnswer)) {\n            skipAnswer = true\n          }\n        },\n        ...this.options.settings,\n        abortSignal: signal,\n      })\n\n      for await (const textPart of result.textStream) {\n        if (skipAnswer) return\n        this.log(`Answer chunk: \"${textPart}\"`)\n\n        // Extracting value?\n        if (extractOptions) {\n          if (!extracting) {\n            const startTagIndex = textPart.indexOf(extractOptions.startTag)\n            if (startTagIndex !== -1) {\n              extracting = true\n              const messagePart = textPart.slice(0, startTagIndex).trimEnd()\n              stream.write(messagePart)\n              continue\n            }\n          } else {\n            // Extracting, don't write to stream\n            continue\n          }\n        }\n\n        // Send chunk\n        stream.write(textPart)\n      }\n      if (skipAnswer) return\n\n      const fullText = await result.text\n      const { message, metadata } = this.extract(fullText)\n\n      // Emit message\n      this.addAssistantMessage(message, metadata)\n      this.abortController = undefined\n    } catch (error: any) {\n      if (abortController === this.abortController) {\n        console.error('[AiSdkAgent] Error answering:', error)\n      }\n    }\n  }\n\n  private buildMessages(): ModelMessage[] {\n    return this.conversation.map((message): ModelMessage => {\n      switch (message.role) {\n        case 'user':\n        case 'assistant':\n        case 'system':\n          return {\n            role: message.role,\n            content: message.content,\n          }\n        case 'tool_call':\n          return {\n            role: 'assistant',\n            content: [\n              {\n                type: 'tool-call',\n                toolCallId: message.toolCallId,\n                toolName: message.toolName,\n                input: JSON.parse(message.parameters),\n              },\n            ],\n          }\n        case 'tool_result':\n          return {\n            role: 'tool',\n            content: [\n              {\n                type: 'tool-result',\n                toolCallId: message.toolCallId,\n                toolName: message.toolName,\n                output: {\n                  type: 'json',\n                  value: JSON.parse(message.output),\n                },\n              },\n            ],\n          }\n      }\n    })\n  }\n\n  private buildTools() {\n    const tools: ToolSet = {}\n\n    // Disable tools if first message\n    const enableTools = this.conversation.length > 1\n    if (!enableTools) return tools\n\n    this.tools.forEach((t) => {\n      tools[t.name] = tool({\n        description: t.description,\n        inputSchema: (t.inputSchema || z.object({})) as z.ZodObject,\n        execute: (args: any, { toolCallId }: { toolCallId: string }) =>\n          this.executeTool({\n            role: 'tool_call',\n            toolCallId,\n            toolName: t.name,\n            parameters: JSON.stringify(args),\n          }),\n      })\n    })\n    return tools\n  }\n\n  cancel() {\n    if (!this.abortController) return\n    this.log('Cancel')\n    this.abortController.abort()\n    this.abortController = undefined\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,oBAAoC;AACpC,gBAQO;AAEP,iBAAkB;AAQlB,IAAM,oBAAoB;AAEnB,IAAM,aAAN,cAAyB,oBAAyB;AAAA,EAGvD,MAAgB,eAAe,QAAoC;AACjE,UAAM,kBAAkB,IAAI,gBAAgB;AAC5C,SAAK,kBAAkB;AACvB,UAAM,SAAS,KAAK,gBAAgB;AACpC,QAAI,aAAa;AAGjB,QAAI,aAAa;AACjB,UAAM,iBAAiB,KAAK,kBAAkB;AAE9C,QAAI;AAEF,YAAM,aAAS,sBAAW;AAAA,QACxB,OAAO,KAAK,QAAQ;AAAA,QACpB,UAAU,KAAK,cAAc;AAAA,QAC7B,OAAO,KAAK,WAAW;AAAA,QACvB,YAAY,KAAK,QAAQ,YAAY;AAAA,QACrC,cAAU,uBAAY,CAAC;AAAA,QACvB,cAAc,CAAC,SAAS;AACtB,gBAAM,QAAQ,KAAK,UAAU;AAAA,YAAI,CAAC,aAChC,KAAK,QAAQ,SAAS,QAAQ;AAAA,UAChC;AACA,cAAI,MAAM,KAAK,CAAC,MAAM,GAAG,UAAU,GAAG;AACpC,yBAAa;AAAA,UACf;AAAA,QACF;AAAA,QACA,GAAG,KAAK,QAAQ;AAAA,QAChB,aAAa;AAAA,MACf,CAAC;AAED,uBAAiB,YAAY,OAAO,YAAY;AAC9C,YAAI,WAAY;AAChB,aAAK,IAAI,kBAAkB,QAAQ,GAAG;AAGtC,YAAI,gBAAgB;AAClB,cAAI,CAAC,YAAY;AACf,kBAAM,gBAAgB,SAAS,QAAQ,eAAe,QAAQ;AAC9D,gBAAI,kBAAkB,IAAI;AACxB,2BAAa;AACb,oBAAM,cAAc,SAAS,MAAM,GAAG,aAAa,EAAE,QAAQ;AAC7D,qBAAO,MAAM,WAAW;AACxB;AAAA,YACF;AAAA,UACF,OAAO;AAEL;AAAA,UACF;AAAA,QACF;AAGA,eAAO,MAAM,QAAQ;AAAA,MACvB;AACA,UAAI,WAAY;AAEhB,YAAM,WAAW,MAAM,OAAO;AAC9B,YAAM,EAAE,SAAS,SAAS,IAAI,KAAK,QAAQ,QAAQ;AAGnD,WAAK,oBAAoB,SAAS,QAAQ;AAC1C,WAAK,kBAAkB;AAAA,IACzB,SAAS,OAAY;AACnB,UAAI,oBAAoB,KAAK,iBAAiB;AAC5C,gBAAQ,MAAM,iCAAiC,KAAK;AAAA,MACtD;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,gBAAgC;AACtC,WAAO,KAAK,aAAa,IAAI,CAAC,YAA0B;AACtD,cAAQ,QAAQ,MAAM;AAAA,QACpB,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AACH,iBAAO;AAAA,YACL,MAAM,QAAQ;AAAA,YACd,SAAS,QAAQ;AAAA,UACnB;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,YAAY,QAAQ;AAAA,gBACpB,UAAU,QAAQ;AAAA,gBAClB,OAAO,KAAK,MAAM,QAAQ,UAAU;AAAA,cACtC;AAAA,YACF;AAAA,UACF;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,YAAY,QAAQ;AAAA,gBACpB,UAAU,QAAQ;AAAA,gBAClB,QAAQ;AAAA,kBACN,MAAM;AAAA,kBACN,OAAO,KAAK,MAAM,QAAQ,MAAM;AAAA,gBAClC;AAAA,cACF;AAAA,YACF;AAAA,UACF;AAAA,MACJ;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,aAAa;AACnB,UAAM,QAAiB,CAAC;AAGxB,UAAM,cAAc,KAAK,aAAa,SAAS;AAC/C,QAAI,CAAC,YAAa,QAAO;AAEzB,SAAK,MAAM,QAAQ,CAAC,MAAM;AACxB,YAAM,EAAE,IAAI,QAAI,gBAAK;AAAA,QACnB,aAAa,EAAE;AAAA,QACf,aAAc,EAAE,eAAe,aAAE,OAAO,CAAC,CAAC;AAAA,QAC1C,SAAS,CAAC,MAAW,EAAE,WAAW,MAChC,KAAK,YAAY;AAAA,UACf,MAAM;AAAA,UACN;AAAA,UACA,UAAU,EAAE;AAAA,UACZ,YAAY,KAAK,UAAU,IAAI;AAAA,QACjC,CAAC;AAAA,MACL,CAAC;AAAA,IACH,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEA,SAAS;AACP,QAAI,CAAC,KAAK,gBAAiB;AAC3B,SAAK,IAAI,QAAQ;AACjB,SAAK,gBAAgB,MAAM;AAC3B,SAAK,kBAAkB;AAAA,EACzB;AACF;","names":[]}

dist/index.mjs

@@ -111,3 +111,2 @@ // src/AiSdkAgent.ts
       tools[t.name] = tool({
-        name: t.name,
         description: t.description,
@@ -114,0 +113,0 @@ inputSchema: t.inputSchema || z.object({}),

dist/index.mjs.map

@@ -1,1 +0,1 @@
-{"version":3,"sources":["../src/AiSdkAgent.ts"],"sourcesContent":["import { Agent, AgentOptions } from '@micdrop/server'\nimport {\n  CallSettings,\n  LanguageModel,\n  ModelMessage,\n  stepCountIs,\n  streamText,\n  tool,\n  ToolSet,\n} from 'ai'\nimport { PassThrough } from 'stream'\nimport { z } from 'zod'\n\nexport interface AiSdkAgentOptions extends AgentOptions {\n  model: LanguageModel\n  settings?: CallSettings\n  maxRetry?: number\n}\n\nconst DEFAULT_MAX_RETRY = 3\n\nexport class AiSdkAgent extends Agent<AiSdkAgentOptions> {\n  private abortController?: AbortController\n\n  protected async generateAnswer(stream: PassThrough): Promise<void> {\n    const abortController = new AbortController()\n    this.abortController = abortController\n    const signal = this.abortController.signal\n    let skipAnswer = false\n\n    // Prepare extracting\n    let extracting = false\n    const extractOptions = this.getExtractOptions()\n\n    try {\n      // Generate response using AI SDK\n      const result = streamText({\n        model: this.options.model,\n        messages: this.buildMessages(),\n        tools: this.buildTools(),\n        maxRetries: this.options.maxRetry ?? DEFAULT_MAX_RETRY,\n        stopWhen: stepCountIs(5),\n        onStepFinish: (step) => {\n          const tools = step.toolCalls.map((toolCall) =>\n            this.getTool(toolCall.toolName)\n          )\n          if (tools.some((t) => t?.skipAnswer)) {\n            skipAnswer = true\n          }\n        },\n        ...this.options.settings,\n        abortSignal: signal,\n      })\n\n      for await (const textPart of result.textStream) {\n        if (skipAnswer) return\n        this.log(`Answer chunk: \"${textPart}\"`)\n\n        // Extracting value?\n        if (extractOptions) {\n          if (!extracting) {\n            const startTagIndex = textPart.indexOf(extractOptions.startTag)\n            if (startTagIndex !== -1) {\n              extracting = true\n              const messagePart = textPart.slice(0, startTagIndex).trimEnd()\n              stream.write(messagePart)\n              continue\n            }\n          } else {\n            // Extracting, don't write to stream\n            continue\n          }\n        }\n\n        // Send chunk\n        stream.write(textPart)\n      }\n      if (skipAnswer) return\n\n      const fullText = await result.text\n      const { message, metadata } = this.extract(fullText)\n\n      // Emit message\n      this.addAssistantMessage(message, metadata)\n      this.abortController = undefined\n    } catch (error: any) {\n      if (abortController === this.abortController) {\n        console.error('[AiSdkAgent] Error answering:', error)\n      }\n    }\n  }\n\n  private buildMessages(): ModelMessage[] {\n    return this.conversation.map((message): ModelMessage => {\n      switch (message.role) {\n        case 'user':\n        case 'assistant':\n        case 'system':\n          return {\n            role: message.role,\n            content: message.content,\n          }\n        case 'tool_call':\n          return {\n            role: 'assistant',\n            content: [\n              {\n                type: 'tool-call',\n                toolCallId: message.toolCallId,\n                toolName: message.toolName,\n                input: JSON.parse(message.parameters),\n              },\n            ],\n          }\n        case 'tool_result':\n          return {\n            role: 'tool',\n            content: [\n              {\n                type: 'tool-result',\n                toolCallId: message.toolCallId,\n                toolName: message.toolName,\n                output: {\n                  type: 'json',\n                  value: JSON.parse(message.output),\n                },\n              },\n            ],\n          }\n      }\n    })\n  }\n\n  private buildTools() {\n    const tools: ToolSet = {}\n\n    // Disable tools if first message\n    const enableTools = this.conversation.length > 1\n    if (!enableTools) return tools\n\n    this.tools.forEach((t) => {\n      tools[t.name] = tool({\n        name: t.name,\n        description: t.description,\n        inputSchema: t.inputSchema || z.object({}),\n        execute: (args, { toolCallId }) =>\n          this.executeTool({\n            role: 'tool_call',\n            toolCallId,\n            toolName: t.name,\n            parameters: JSON.stringify(args),\n          }),\n      })\n    })\n    return tools\n  }\n\n  cancel() {\n    if (!this.abortController) return\n    this.log('Cancel')\n    this.abortController.abort()\n    this.abortController = undefined\n  }\n}\n"],"mappings":";AAAA,SAAS,aAA2B;AACpC;AAAA,EAIE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AAEP,SAAS,SAAS;AAQlB,IAAM,oBAAoB;AAEnB,IAAM,aAAN,cAAyB,MAAyB;AAAA,EAGvD,MAAgB,eAAe,QAAoC;AACjE,UAAM,kBAAkB,IAAI,gBAAgB;AAC5C,SAAK,kBAAkB;AACvB,UAAM,SAAS,KAAK,gBAAgB;AACpC,QAAI,aAAa;AAGjB,QAAI,aAAa;AACjB,UAAM,iBAAiB,KAAK,kBAAkB;AAE9C,QAAI;AAEF,YAAM,SAAS,WAAW;AAAA,QACxB,OAAO,KAAK,QAAQ;AAAA,QACpB,UAAU,KAAK,cAAc;AAAA,QAC7B,OAAO,KAAK,WAAW;AAAA,QACvB,YAAY,KAAK,QAAQ,YAAY;AAAA,QACrC,UAAU,YAAY,CAAC;AAAA,QACvB,cAAc,CAAC,SAAS;AACtB,gBAAM,QAAQ,KAAK,UAAU;AAAA,YAAI,CAAC,aAChC,KAAK,QAAQ,SAAS,QAAQ;AAAA,UAChC;AACA,cAAI,MAAM,KAAK,CAAC,MAAM,GAAG,UAAU,GAAG;AACpC,yBAAa;AAAA,UACf;AAAA,QACF;AAAA,QACA,GAAG,KAAK,QAAQ;AAAA,QAChB,aAAa;AAAA,MACf,CAAC;AAED,uBAAiB,YAAY,OAAO,YAAY;AAC9C,YAAI,WAAY;AAChB,aAAK,IAAI,kBAAkB,QAAQ,GAAG;AAGtC,YAAI,gBAAgB;AAClB,cAAI,CAAC,YAAY;AACf,kBAAM,gBAAgB,SAAS,QAAQ,eAAe,QAAQ;AAC9D,gBAAI,kBAAkB,IAAI;AACxB,2BAAa;AACb,oBAAM,cAAc,SAAS,MAAM,GAAG,aAAa,EAAE,QAAQ;AAC7D,qBAAO,MAAM,WAAW;AACxB;AAAA,YACF;AAAA,UACF,OAAO;AAEL;AAAA,UACF;AAAA,QACF;AAGA,eAAO,MAAM,QAAQ;AAAA,MACvB;AACA,UAAI,WAAY;AAEhB,YAAM,WAAW,MAAM,OAAO;AAC9B,YAAM,EAAE,SAAS,SAAS,IAAI,KAAK,QAAQ,QAAQ;AAGnD,WAAK,oBAAoB,SAAS,QAAQ;AAC1C,WAAK,kBAAkB;AAAA,IACzB,SAAS,OAAY;AACnB,UAAI,oBAAoB,KAAK,iBAAiB;AAC5C,gBAAQ,MAAM,iCAAiC,KAAK;AAAA,MACtD;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,gBAAgC;AACtC,WAAO,KAAK,aAAa,IAAI,CAAC,YAA0B;AACtD,cAAQ,QAAQ,MAAM;AAAA,QACpB,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AACH,iBAAO;AAAA,YACL,MAAM,QAAQ;AAAA,YACd,SAAS,QAAQ;AAAA,UACnB;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,YAAY,QAAQ;AAAA,gBACpB,UAAU,QAAQ;AAAA,gBAClB,OAAO,KAAK,MAAM,QAAQ,UAAU;AAAA,cACtC;AAAA,YACF;AAAA,UACF;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,YAAY,QAAQ;AAAA,gBACpB,UAAU,QAAQ;AAAA,gBAClB,QAAQ;AAAA,kBACN,MAAM;AAAA,kBACN,OAAO,KAAK,MAAM,QAAQ,MAAM;AAAA,gBAClC;AAAA,cACF;AAAA,YACF;AAAA,UACF;AAAA,MACJ;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,aAAa;AACnB,UAAM,QAAiB,CAAC;AAGxB,UAAM,cAAc,KAAK,aAAa,SAAS;AAC/C,QAAI,CAAC,YAAa,QAAO;AAEzB,SAAK,MAAM,QAAQ,CAAC,MAAM;AACxB,YAAM,EAAE,IAAI,IAAI,KAAK;AAAA,QACnB,MAAM,EAAE;AAAA,QACR,aAAa,EAAE;AAAA,QACf,aAAa,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;AAAA,QACzC,SAAS,CAAC,MAAM,EAAE,WAAW,MAC3B,KAAK,YAAY;AAAA,UACf,MAAM;AAAA,UACN;AAAA,UACA,UAAU,EAAE;AAAA,UACZ,YAAY,KAAK,UAAU,IAAI;AAAA,QACjC,CAAC;AAAA,MACL,CAAC;AAAA,IACH,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEA,SAAS;AACP,QAAI,CAAC,KAAK,gBAAiB;AAC3B,SAAK,IAAI,QAAQ;AACjB,SAAK,gBAAgB,MAAM;AAC3B,SAAK,kBAAkB;AAAA,EACzB;AACF;","names":[]}
+{"version":3,"sources":["../src/AiSdkAgent.ts"],"sourcesContent":["import { Agent, AgentOptions } from '@micdrop/server'\nimport {\n  CallSettings,\n  LanguageModel,\n  ModelMessage,\n  stepCountIs,\n  streamText,\n  tool,\n  ToolSet,\n} from 'ai'\nimport { PassThrough } from 'stream'\nimport { z } from 'zod'\n\nexport interface AiSdkAgentOptions extends AgentOptions {\n  model: LanguageModel\n  settings?: CallSettings\n  maxRetry?: number\n}\n\nconst DEFAULT_MAX_RETRY = 3\n\nexport class AiSdkAgent extends Agent<AiSdkAgentOptions> {\n  private abortController?: AbortController\n\n  protected async generateAnswer(stream: PassThrough): Promise<void> {\n    const abortController = new AbortController()\n    this.abortController = abortController\n    const signal = this.abortController.signal\n    let skipAnswer = false\n\n    // Prepare extracting\n    let extracting = false\n    const extractOptions = this.getExtractOptions()\n\n    try {\n      // Generate response using AI SDK\n      const result = streamText({\n        model: this.options.model,\n        messages: this.buildMessages(),\n        tools: this.buildTools(),\n        maxRetries: this.options.maxRetry ?? DEFAULT_MAX_RETRY,\n        stopWhen: stepCountIs(5),\n        onStepFinish: (step) => {\n          const tools = step.toolCalls.map((toolCall) =>\n            this.getTool(toolCall.toolName)\n          )\n          if (tools.some((t) => t?.skipAnswer)) {\n            skipAnswer = true\n          }\n        },\n        ...this.options.settings,\n        abortSignal: signal,\n      })\n\n      for await (const textPart of result.textStream) {\n        if (skipAnswer) return\n        this.log(`Answer chunk: \"${textPart}\"`)\n\n        // Extracting value?\n        if (extractOptions) {\n          if (!extracting) {\n            const startTagIndex = textPart.indexOf(extractOptions.startTag)\n            if (startTagIndex !== -1) {\n              extracting = true\n              const messagePart = textPart.slice(0, startTagIndex).trimEnd()\n              stream.write(messagePart)\n              continue\n            }\n          } else {\n            // Extracting, don't write to stream\n            continue\n          }\n        }\n\n        // Send chunk\n        stream.write(textPart)\n      }\n      if (skipAnswer) return\n\n      const fullText = await result.text\n      const { message, metadata } = this.extract(fullText)\n\n      // Emit message\n      this.addAssistantMessage(message, metadata)\n      this.abortController = undefined\n    } catch (error: any) {\n      if (abortController === this.abortController) {\n        console.error('[AiSdkAgent] Error answering:', error)\n      }\n    }\n  }\n\n  private buildMessages(): ModelMessage[] {\n    return this.conversation.map((message): ModelMessage => {\n      switch (message.role) {\n        case 'user':\n        case 'assistant':\n        case 'system':\n          return {\n            role: message.role,\n            content: message.content,\n          }\n        case 'tool_call':\n          return {\n            role: 'assistant',\n            content: [\n              {\n                type: 'tool-call',\n                toolCallId: message.toolCallId,\n                toolName: message.toolName,\n                input: JSON.parse(message.parameters),\n              },\n            ],\n          }\n        case 'tool_result':\n          return {\n            role: 'tool',\n            content: [\n              {\n                type: 'tool-result',\n                toolCallId: message.toolCallId,\n                toolName: message.toolName,\n                output: {\n                  type: 'json',\n                  value: JSON.parse(message.output),\n                },\n              },\n            ],\n          }\n      }\n    })\n  }\n\n  private buildTools() {\n    const tools: ToolSet = {}\n\n    // Disable tools if first message\n    const enableTools = this.conversation.length > 1\n    if (!enableTools) return tools\n\n    this.tools.forEach((t) => {\n      tools[t.name] = tool({\n        description: t.description,\n        inputSchema: (t.inputSchema || z.object({})) as z.ZodObject,\n        execute: (args: any, { toolCallId }: { toolCallId: string }) =>\n          this.executeTool({\n            role: 'tool_call',\n            toolCallId,\n            toolName: t.name,\n            parameters: JSON.stringify(args),\n          }),\n      })\n    })\n    return tools\n  }\n\n  cancel() {\n    if (!this.abortController) return\n    this.log('Cancel')\n    this.abortController.abort()\n    this.abortController = undefined\n  }\n}\n"],"mappings":";AAAA,SAAS,aAA2B;AACpC;AAAA,EAIE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AAEP,SAAS,SAAS;AAQlB,IAAM,oBAAoB;AAEnB,IAAM,aAAN,cAAyB,MAAyB;AAAA,EAGvD,MAAgB,eAAe,QAAoC;AACjE,UAAM,kBAAkB,IAAI,gBAAgB;AAC5C,SAAK,kBAAkB;AACvB,UAAM,SAAS,KAAK,gBAAgB;AACpC,QAAI,aAAa;AAGjB,QAAI,aAAa;AACjB,UAAM,iBAAiB,KAAK,kBAAkB;AAE9C,QAAI;AAEF,YAAM,SAAS,WAAW;AAAA,QACxB,OAAO,KAAK,QAAQ;AAAA,QACpB,UAAU,KAAK,cAAc;AAAA,QAC7B,OAAO,KAAK,WAAW;AAAA,QACvB,YAAY,KAAK,QAAQ,YAAY;AAAA,QACrC,UAAU,YAAY,CAAC;AAAA,QACvB,cAAc,CAAC,SAAS;AACtB,gBAAM,QAAQ,KAAK,UAAU;AAAA,YAAI,CAAC,aAChC,KAAK,QAAQ,SAAS,QAAQ;AAAA,UAChC;AACA,cAAI,MAAM,KAAK,CAAC,MAAM,GAAG,UAAU,GAAG;AACpC,yBAAa;AAAA,UACf;AAAA,QACF;AAAA,QACA,GAAG,KAAK,QAAQ;AAAA,QAChB,aAAa;AAAA,MACf,CAAC;AAED,uBAAiB,YAAY,OAAO,YAAY;AAC9C,YAAI,WAAY;AAChB,aAAK,IAAI,kBAAkB,QAAQ,GAAG;AAGtC,YAAI,gBAAgB;AAClB,cAAI,CAAC,YAAY;AACf,kBAAM,gBAAgB,SAAS,QAAQ,eAAe,QAAQ;AAC9D,gBAAI,kBAAkB,IAAI;AACxB,2BAAa;AACb,oBAAM,cAAc,SAAS,MAAM,GAAG,aAAa,EAAE,QAAQ;AAC7D,qBAAO,MAAM,WAAW;AACxB;AAAA,YACF;AAAA,UACF,OAAO;AAEL;AAAA,UACF;AAAA,QACF;AAGA,eAAO,MAAM,QAAQ;AAAA,MACvB;AACA,UAAI,WAAY;AAEhB,YAAM,WAAW,MAAM,OAAO;AAC9B,YAAM,EAAE,SAAS,SAAS,IAAI,KAAK,QAAQ,QAAQ;AAGnD,WAAK,oBAAoB,SAAS,QAAQ;AAC1C,WAAK,kBAAkB;AAAA,IACzB,SAAS,OAAY;AACnB,UAAI,oBAAoB,KAAK,iBAAiB;AAC5C,gBAAQ,MAAM,iCAAiC,KAAK;AAAA,MACtD;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,gBAAgC;AACtC,WAAO,KAAK,aAAa,IAAI,CAAC,YAA0B;AACtD,cAAQ,QAAQ,MAAM;AAAA,QACpB,KAAK;AAAA,QACL,KAAK;AAAA,QACL,KAAK;AACH,iBAAO;AAAA,YACL,MAAM,QAAQ;AAAA,YACd,SAAS,QAAQ;AAAA,UACnB;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,YAAY,QAAQ;AAAA,gBACpB,UAAU,QAAQ;AAAA,gBAClB,OAAO,KAAK,MAAM,QAAQ,UAAU;AAAA,cACtC;AAAA,YACF;AAAA,UACF;AAAA,QACF,KAAK;AACH,iBAAO;AAAA,YACL,MAAM;AAAA,YACN,SAAS;AAAA,cACP;AAAA,gBACE,MAAM;AAAA,gBACN,YAAY,QAAQ;AAAA,gBACpB,UAAU,QAAQ;AAAA,gBAClB,QAAQ;AAAA,kBACN,MAAM;AAAA,kBACN,OAAO,KAAK,MAAM,QAAQ,MAAM;AAAA,gBAClC;AAAA,cACF;AAAA,YACF;AAAA,UACF;AAAA,MACJ;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,aAAa;AACnB,UAAM,QAAiB,CAAC;AAGxB,UAAM,cAAc,KAAK,aAAa,SAAS;AAC/C,QAAI,CAAC,YAAa,QAAO;AAEzB,SAAK,MAAM,QAAQ,CAAC,MAAM;AACxB,YAAM,EAAE,IAAI,IAAI,KAAK;AAAA,QACnB,aAAa,EAAE;AAAA,QACf,aAAc,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;AAAA,QAC1C,SAAS,CAAC,MAAW,EAAE,WAAW,MAChC,KAAK,YAAY;AAAA,UACf,MAAM;AAAA,UACN;AAAA,UACA,UAAU,EAAE;AAAA,UACZ,YAAY,KAAK,UAAU,IAAI;AAAA,QACjC,CAAC;AAAA,MACL,CAAC;AAAA,IACH,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEA,SAAS;AACP,QAAI,CAAC,KAAK,gBAAiB;AAC3B,SAAK,IAAI,QAAQ;AACjB,SAAK,gBAAgB,MAAM;AAC3B,SAAK,kBAAkB;AAAA,EACzB;AACF;","names":[]}

package.json

 {
   "name": "@micdrop/ai-sdk",
-  "version": "1.0.5",
+  "version": "1.0.6",
   "description": "AI SDK implementation for @micdrop/server",
@@ -10,9 +10,2 @@ "author": "Lonestone",
   "types": "./dist/index.d.ts",
-  "scripts": {
-    "build": "tsup",
-    "dev": "tsup --watch",
-    "clean": "rm -rf dist",
-    "typecheck": "tsc --noEmit",
-    "prepublishOnly": "npm run build"
-  },
   "files": [
@@ -23,3 +16,3 @@ "dist"
     "type": "git",
-    "url": "git+https://github.com/lonestone/micdrop.git",
+    "url": "git+https://github.com/Godefroy/micdrop.git",
     "directory": "packages/ai-sdk"
@@ -39,3 +32,3 @@ },
   "dependencies": {
-    "ai": "^5.0.14"
+    "ai": "^6.0.0"
   },
@@ -47,6 +40,12 @@ "peerDependencies": {
   "devDependencies": {
-    "@types/node": "^22.13.4",
-    "tsup": "^8.3.6",
-    "typescript": "^5.7.3"
+    "@types/node": "^22.19.13",
+    "tsup": "^8.5.1",
+    "typescript": "^5.9.3"
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "clean": "rm -rf dist",
+    "typecheck": "tsc --noEmit"
   }
-}
+}

No alert changes

Improved metrics

Total package byte prevSize

28159

3.63%

Number of package files

9

12.5%

Worsened metrics

Lines of code

301

-0.66%

Dependency changes

• + Added

  @ai-sdk/gateway@3.0.125
  (transitive)

• + Added

  @ai-sdk/provider@3.0.10
  (transitive)

• + Added

  @ai-sdk/provider-utils@4.0.27
  (transitive)

• + Added

  @opentelemetry/api@1.9.1
  (transitive)

• + Added

  @vercel/oidc@3.2.0
  (transitive)

• + Added

  ai@6.0.197
  (transitive)

• - Removed

  @ai-sdk/gateway@2.0.97
  (transitive)

• - Removed

  @ai-sdk/provider@2.0.3
  (transitive)

• - Removed

  @ai-sdk/provider-utils@3.0.25
  (transitive)

• - Removed

  @opentelemetry/api@1.9.0
  (transitive)

• - Removed

  @vercel/oidc@3.1.0
  (transitive)

• - Removed

  ai@5.0.196
  (transitive)

• Updated

  ai@^6.0.0