@sweetretry/ai-sdk-volcengine-adapter 0.1.0 → 0.1.1

package/dist/tool/image-generation.js

@@ -0,0 +1,79 @@
+import { createProviderToolFactoryWithOutputSchema, lazySchema, zodSchema } from "@ai-sdk/provider-utils";
+import { z } from "zod/v4";
+export const imageGenerationArgsSchema = lazySchema(()=>zodSchema(z.object({
+        model: z.string().optional(),
+        size: z.enum([
+            "2K",
+            "4K"
+        ]).optional(),
+        responseFormat: z.enum([
+            "url",
+            "b64_json"
+        ]).optional(),
+        image: z.union([
+            z.string(),
+            z.array(z.string())
+        ]).optional()
+    }).strict()));
+const imageGenerationInputSchema = lazySchema(()=>zodSchema(z.object({})));
+export const imageGenerationOutputSchema = lazySchema(()=>zodSchema(z.object({
+        model: z.string(),
+        created: z.number(),
+        data: z.array(z.object({
+            url: z.string().optional(),
+            b64_json: z.string().optional(),
+            size: z.string()
+        })),
+        usage: z.object({
+            generated_images: z.number(),
+            output_tokens: z.number(),
+            total_tokens: z.number()
+        })
+    })));
+const imageGenerationToolFactory = createProviderToolFactoryWithOutputSchema({
+    id: "volcengine.image_generation",
+    inputSchema: imageGenerationInputSchema,
+    outputSchema: imageGenerationOutputSchema
+});
+/**
+ * Volcengine Image Generation Tool
+ *
+ * Enables the model to generate images from text prompts or reference images.
+ *
+ * Supports three modes:
+ * - Text-to-Image: Generate images from text prompts only
+ * - Image-to-Image: Generate images using a single reference image
+ * - Multi-Image-to-Image: Generate images using multiple reference images
+ *
+ * @see https://www.volcengine.com/docs/82379/1824121
+ *
+ * @example
+ * ```ts
+ * // Text-to-Image
+ * volcengineTools.imageGeneration({
+ *   size: "2K",
+ * })
+ *
+ * // Image-to-Image
+ * volcengineTools.imageGeneration({
+ *   size: "2K",
+ *   image: "https://example.com/reference.png",
+ * })
+ *
+ * // Multi-Image-to-Image
+ * volcengineTools.imageGeneration({
+ *   size: "2K",
+ *   image: ["https://example.com/img1.png", "https://example.com/img2.png"],
+ * })
+ *
+ * // Get Base64 encoded image
+ * volcengineTools.imageGeneration({
+ *   size: "2K",
+ *   responseFormat: "b64_json",
+ * })
+ * ```
+ */ export const imageGeneration = (args = {})=>{
+    return imageGenerationToolFactory(args);
+};
+
+//# sourceMappingURL=image-generation.js.map

package/dist/tool/image-generation.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/tool/image-generation.ts"],"sourcesContent":["import {\n  createProviderToolFactoryWithOutputSchema,\n  lazySchema,\n  zodSchema,\n} from \"@ai-sdk/provider-utils\";\nimport { z } from \"zod/v4\";\n\nexport const imageGenerationArgsSchema = lazySchema(() =>\n  zodSchema(\n    z\n      .object({\n        model: z.string().optional(),\n        size: z.enum([\"2K\", \"4K\"]).optional(),\n        responseFormat: z.enum([\"url\", \"b64_json\"]).optional(),\n        image: z.union([z.string(), z.array(z.string())]).optional(),\n      })\n      .strict()\n  )\n);\n\nconst imageGenerationInputSchema = lazySchema(() => zodSchema(z.object({})));\n\nexport const imageGenerationOutputSchema = lazySchema(() =>\n  zodSchema(\n    z.object({\n      model: z.string(),\n      created: z.number(),\n      data: z.array(\n        z.object({\n          url: z.string().optional(),\n          b64_json: z.string().optional(),\n          size: z.string(),\n        })\n      ),\n      usage: z.object({\n        generated_images: z.number(),\n        output_tokens: z.number(),\n        total_tokens: z.number(),\n      }),\n    })\n  )\n);\n\n/**\n * Image generation input options.\n */\nexport interface ImageGenerationArgs {\n  /**\n   * Model ID for image generation.\n   * @default \"doubao-seedream-4-5-251128\"\n   */\n  model?: string;\n\n  /**\n   * Image resolution.\n   * @default \"2K\"\n   */\n  size?: \"2K\" | \"4K\";\n\n  /**\n   * Response format for the generated image.\n   * - \"url\": Returns image download URL (valid for 24 hours)\n   * - \"b64_json\": Returns Base64 encoded image data\n   * @default \"url\"\n   */\n  responseFormat?: \"url\" | \"b64_json\";\n\n  /**\n   * Reference image URL(s) for image-to-image generation.\n   * - Single URL string for single image reference\n   * - Array of URLs for multi-image reference\n   */\n  image?: string | string[];\n}\n\n/**\n * Image generation output.\n */\nexport interface ImageGenerationOutput {\n  model: string;\n  created: number;\n  data: Array<{\n    /** Image URL, returned when response_format is url */\n    url?: string;\n    /** Base64 encoded image, returned when response_format is b64_json */\n    b64_json?: string;\n    /** Image dimensions (e.g., \"3104x1312\") */\n    size: string;\n  }>;\n  usage: {\n    generated_images: number;\n    output_tokens: number;\n    total_tokens: number;\n  };\n}\n\nconst imageGenerationToolFactory = createProviderToolFactoryWithOutputSchema<\n  {},\n  ImageGenerationOutput,\n  ImageGenerationArgs\n>({\n  id: \"volcengine.image_generation\",\n  inputSchema: imageGenerationInputSchema,\n  outputSchema: imageGenerationOutputSchema,\n});\n\n/**\n * Volcengine Image Generation Tool\n *\n * Enables the model to generate images from text prompts or reference images.\n *\n * Supports three modes:\n * - Text-to-Image: Generate images from text prompts only\n * - Image-to-Image: Generate images using a single reference image\n * - Multi-Image-to-Image: Generate images using multiple reference images\n *\n * @see https://www.volcengine.com/docs/82379/1824121\n *\n * @example\n * ```ts\n * // Text-to-Image\n * volcengineTools.imageGeneration({\n *   size: \"2K\",\n * })\n *\n * // Image-to-Image\n * volcengineTools.imageGeneration({\n *   size: \"2K\",\n *   image: \"https://example.com/reference.png\",\n * })\n *\n * // Multi-Image-to-Image\n * volcengineTools.imageGeneration({\n *   size: \"2K\",\n *   image: [\"https://example.com/img1.png\", \"https://example.com/img2.png\"],\n * })\n *\n * // Get Base64 encoded image\n * volcengineTools.imageGeneration({\n *   size: \"2K\",\n *   responseFormat: \"b64_json\",\n * })\n * ```\n */\nexport const imageGeneration = (args: ImageGenerationArgs = {}) => {\n  return imageGenerationToolFactory(args);\n};\n"],"names":["createProviderToolFactoryWithOutputSchema","lazySchema","zodSchema","z","imageGenerationArgsSchema","object","model","string","optional","size","enum","responseFormat","image","union","array","strict","imageGenerationInputSchema","imageGenerationOutputSchema","created","number","data","url","b64_json","usage","generated_images","output_tokens","total_tokens","imageGenerationToolFactory","id","inputSchema","outputSchema","imageGeneration","args"],"mappings":"AAAA,SACEA,yCAAyC,EACzCC,UAAU,EACVC,SAAS,QACJ,yBAAyB;AAChC,SAASC,CAAC,QAAQ,SAAS;AAE3B,OAAO,MAAMC,4BAA4BH,WAAW,IAClDC,UACEC,EACGE,MAAM,CAAC;QACNC,OAAOH,EAAEI,MAAM,GAAGC,QAAQ;QAC1BC,MAAMN,EAAEO,IAAI,CAAC;YAAC;YAAM;SAAK,EAAEF,QAAQ;QACnCG,gBAAgBR,EAAEO,IAAI,CAAC;YAAC;YAAO;SAAW,EAAEF,QAAQ;QACpDI,OAAOT,EAAEU,KAAK,CAAC;YAACV,EAAEI,MAAM;YAAIJ,EAAEW,KAAK,CAACX,EAAEI,MAAM;SAAI,EAAEC,QAAQ;IAC5D,GACCO,MAAM,KAEX;AAEF,MAAMC,6BAA6Bf,WAAW,IAAMC,UAAUC,EAAEE,MAAM,CAAC,CAAC;AAExE,OAAO,MAAMY,8BAA8BhB,WAAW,IACpDC,UACEC,EAAEE,MAAM,CAAC;QACPC,OAAOH,EAAEI,MAAM;QACfW,SAASf,EAAEgB,MAAM;QACjBC,MAAMjB,EAAEW,KAAK,CACXX,EAAEE,MAAM,CAAC;YACPgB,KAAKlB,EAAEI,MAAM,GAAGC,QAAQ;YACxBc,UAAUnB,EAAEI,MAAM,GAAGC,QAAQ;YAC7BC,MAAMN,EAAEI,MAAM;QAChB;QAEFgB,OAAOpB,EAAEE,MAAM,CAAC;YACdmB,kBAAkBrB,EAAEgB,MAAM;YAC1BM,eAAetB,EAAEgB,MAAM;YACvBO,cAAcvB,EAAEgB,MAAM;QACxB;IACF,KAEF;AAuDF,MAAMQ,6BAA6B3B,0CAIjC;IACA4B,IAAI;IACJC,aAAab;IACbc,cAAcb;AAChB;AAEA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAqCC,GACD,OAAO,MAAMc,kBAAkB,CAACC,OAA4B,CAAC,CAAC;IAC5D,OAAOL,2BAA2BK;AACpC,EAAE"}

package/dist/tool/web-search.d.ts

@@ -1,12 +1,60 @@
-export interface WebSearchInput {
+export declare const webSearchArgsSchema: import("@ai-sdk/provider-utils").LazySchema<{
+    maxKeyword?: number | undefined;
+    limit?: number | undefined;
+    maxToolCalls?: number | undefined;
+    sources?: string[] | undefined;
+    userLocation?: {
+        type?: "approximate" | undefined;
+        country?: string | undefined;
+        region?: string | undefined;
+        city?: string | undefined;
+    } | undefined;
+}>;
+export declare const webSearchOutputSchema: import("@ai-sdk/provider-utils").LazySchema<{
+    result: string;
+}>;
+/**
+ * Web search input options.
+ */
+export interface WebSearchArgs {
+    /**
+     * Maximum number of keywords to extract from the query.
+     */
     maxKeyword?: number;
+    /**
+     * Maximum number of search results to return.
+     * @default 10
+     */
     limit?: number;
+    /**
+     * Maximum number of tool calls.
+     * @default 3
+     */
     maxToolCalls?: number;
+    /**
+     * Sources to search from (e.g., "douyin", "toutiao").
+     */
     sources?: string[];
+    /**
+     * User location for optimizing search results.
+     */
     userLocation?: {
+        /**
+         * Location type.
+         * @default "approximate"
+         */
         type?: "approximate";
+        /**
+         * Country name (e.g., "中国").
+         */
         country?: string;
+        /**
+         * Region/Province name (e.g., "浙江").
+         */
         region?: string;
+        /**
+         * City name (e.g., "杭州").
+         */
         city?: string;
     };
 }
@@ -16,5 +64,25 @@ export interface WebSearchInput {
  * Enables the model to perform web searches to retrieve up-to-date information.
  *
  * @see https://www.volcengine.com/docs/82379/1756990
+ *
+ * @example
+ * ```ts
+ * volcengineTools.webSearch({
+ *   maxKeyword: 5,
+ *   limit: 20,
+ *   sources: ["douyin", "toutiao"],
+ *   userLocation: {
+ *     type: "approximate",
+ *     country: "中国",
+ *     region: "浙江",
+ *     city: "杭州",
+ *   },
+ * })
+ * ```
  */
-export declare const webSearch: import("@ai-sdk/provider-utils").ProviderToolFactory<WebSearchInput, WebSearchInput>;
+export declare const webSearch: (args?: WebSearchArgs) => import("@ai-sdk/provider-utils").Tool<{}, {
+    /**
+     * The search result.
+     */
+    result: string;
+}>;

package/dist/tool/web-search.js

@@ -1,23 +1,28 @@
-import { createProviderToolFactory, lazySchema, zodSchema } from "@ai-sdk/provider-utils";
-import { z } from "zod";
-/**
- * User location for optimizing search results.
- */ const userLocationSchema = z.object({
-    /**
-   * Location type.
-   * @default "approximate"
-   */ type: z.enum([
+import { createProviderToolFactoryWithOutputSchema, lazySchema, zodSchema } from "@ai-sdk/provider-utils";
+import { z } from "zod/v4";
+const userLocationSchema = z.object({
+    type: z.enum([
         "approximate"
-    ]).default("approximate"),
-    /**
-   * Country name (e.g., "中国").
-   */ country: z.string().optional(),
-    /**
-   * Region/Province name (e.g., "浙江").
-   */ region: z.string().optional(),
-    /**
-   * City name (e.g., "杭州").
-   */ city: z.string().optional()
+    ]).optional(),
+    country: z.string().optional(),
+    region: z.string().optional(),
+    city: z.string().optional()
+}).strict();
+export const webSearchArgsSchema = lazySchema(()=>zodSchema(z.object({
+        maxKeyword: z.number().int().min(1).max(50).optional(),
+        limit: z.number().int().min(1).max(50).optional(),
+        maxToolCalls: z.number().int().min(1).max(10).optional(),
+        sources: z.array(z.string()).optional(),
+        userLocation: userLocationSchema.optional()
+    }).strict()));
+const webSearchInputSchema = lazySchema(()=>zodSchema(z.object({})));
+export const webSearchOutputSchema = lazySchema(()=>zodSchema(z.object({
+        result: z.string()
+    })));
+const webSearchToolFactory = createProviderToolFactoryWithOutputSchema({
+    id: "volcengine.web_search",
+    inputSchema: webSearchInputSchema,
+    outputSchema: webSearchOutputSchema
 });
 /**
  * Volcengine Web Search Tool
@@ -25,15 +30,23 @@ import { z } from "zod";
  * Enables the model to perform web searches to retrieve up-to-date information.
  *
  * @see https://www.volcengine.com/docs/82379/1756990
- */ export const webSearch = createProviderToolFactory({
-    id: "volcengine.web_search",
-    inputSchema: lazySchema(()=>zodSchema(z.object({
-            maxKeyword: z.number().min(1).max(50).optional(),
-            limit: z.number().min(1).max(50).default(10),
-            maxToolCalls: z.number().min(1).max(10).default(3),
-            sources: z.array(z.string()).optional(),
-            userLocation: userLocationSchema.optional()
-        })))
-});
+ *
+ * @example
+ * ```ts
+ * volcengineTools.webSearch({
+ *   maxKeyword: 5,
+ *   limit: 20,
+ *   sources: ["douyin", "toutiao"],
+ *   userLocation: {
+ *     type: "approximate",
+ *     country: "中国",
+ *     region: "浙江",
+ *     city: "杭州",
+ *   },
+ * })
+ * ```
+ */ export const webSearch = (args = {})=>{
+    return webSearchToolFactory(args);
+};
 
 //# sourceMappingURL=web-search.js.map

package/dist/tool/web-search.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/tool/web-search.ts"],"sourcesContent":["import {\n  createProviderToolFactory,\n  lazySchema,\n  zodSchema,\n} from \"@ai-sdk/provider-utils\";\nimport { z } from \"zod\";\n\n/**\n * User location for optimizing search results.\n */\nconst userLocationSchema = z.object({\n  /**\n   * Location type.\n   * @default \"approximate\"\n   */\n  type: z.enum([\"approximate\"]).default(\"approximate\"),\n  /**\n   * Country name (e.g., \"中国\").\n   */\n  country: z.string().optional(),\n  /**\n   * Region/Province name (e.g., \"浙江\").\n   */\n  region: z.string().optional(),\n  /**\n   * City name (e.g., \"杭州\").\n   */\n  city: z.string().optional(),\n});\n\nexport interface WebSearchInput {\n  maxKeyword?: number;\n  limit?: number;\n  maxToolCalls?: number;\n  sources?: string[];\n  userLocation?: {\n    type?: \"approximate\";\n    country?: string;\n    region?: string;\n    city?: string;\n  };\n}\n\n/**\n * Volcengine Web Search Tool\n *\n * Enables the model to perform web searches to retrieve up-to-date information.\n *\n * @see https://www.volcengine.com/docs/82379/1756990\n */\nexport const webSearch = createProviderToolFactory<\n  WebSearchInput,\n  WebSearchInput\n>({\n  id: \"volcengine.web_search\",\n  inputSchema: lazySchema(() =>\n    zodSchema(\n      z.object({\n        maxKeyword: z.number().min(1).max(50).optional(),\n        limit: z.number().min(1).max(50).default(10),\n        maxToolCalls: z.number().min(1).max(10).default(3),\n        sources: z.array(z.string()).optional(),\n        userLocation: userLocationSchema.optional(),\n      })\n    )\n  ),\n});\n"],"names":["createProviderToolFactory","lazySchema","zodSchema","z","userLocationSchema","object","type","enum","default","country","string","optional","region","city","webSearch","id","inputSchema","maxKeyword","number","min","max","limit","maxToolCalls","sources","array","userLocation"],"mappings":"AAAA,SACEA,yBAAyB,EACzBC,UAAU,EACVC,SAAS,QACJ,yBAAyB;AAChC,SAASC,CAAC,QAAQ,MAAM;AAExB;;CAEC,GACD,MAAMC,qBAAqBD,EAAEE,MAAM,CAAC;IAClC;;;GAGC,GACDC,MAAMH,EAAEI,IAAI,CAAC;QAAC;KAAc,EAAEC,OAAO,CAAC;IACtC;;GAEC,GACDC,SAASN,EAAEO,MAAM,GAAGC,QAAQ;IAC5B;;GAEC,GACDC,QAAQT,EAAEO,MAAM,GAAGC,QAAQ;IAC3B;;GAEC,GACDE,MAAMV,EAAEO,MAAM,GAAGC,QAAQ;AAC3B;AAeA;;;;;;CAMC,GACD,OAAO,MAAMG,YAAYd,0BAGvB;IACAe,IAAI;IACJC,aAAaf,WAAW,IACtBC,UACEC,EAAEE,MAAM,CAAC;YACPY,YAAYd,EAAEe,MAAM,GAAGC,GAAG,CAAC,GAAGC,GAAG,CAAC,IAAIT,QAAQ;YAC9CU,OAAOlB,EAAEe,MAAM,GAAGC,GAAG,CAAC,GAAGC,GAAG,CAAC,IAAIZ,OAAO,CAAC;YACzCc,cAAcnB,EAAEe,MAAM,GAAGC,GAAG,CAAC,GAAGC,GAAG,CAAC,IAAIZ,OAAO,CAAC;YAChDe,SAASpB,EAAEqB,KAAK,CAACrB,EAAEO,MAAM,IAAIC,QAAQ;YACrCc,cAAcrB,mBAAmBO,QAAQ;QAC3C;AAGN,GAAG"}
+{"version":3,"sources":["../../src/tool/web-search.ts"],"sourcesContent":["import {\n  createProviderToolFactoryWithOutputSchema,\n  lazySchema,\n  zodSchema,\n} from \"@ai-sdk/provider-utils\";\nimport { z } from \"zod/v4\";\n\nconst userLocationSchema = z\n  .object({\n    type: z.enum([\"approximate\"]).optional(),\n    country: z.string().optional(),\n    region: z.string().optional(),\n    city: z.string().optional(),\n  })\n  .strict();\n\nexport const webSearchArgsSchema = lazySchema(() =>\n  zodSchema(\n    z\n      .object({\n        maxKeyword: z.number().int().min(1).max(50).optional(),\n        limit: z.number().int().min(1).max(50).optional(),\n        maxToolCalls: z.number().int().min(1).max(10).optional(),\n        sources: z.array(z.string()).optional(),\n        userLocation: userLocationSchema.optional(),\n      })\n      .strict()\n  )\n);\n\nconst webSearchInputSchema = lazySchema(() => zodSchema(z.object({})));\n\nexport const webSearchOutputSchema = lazySchema(() =>\n  zodSchema(z.object({ result: z.string() }))\n);\n\n/**\n * Web search input options.\n */\nexport interface WebSearchArgs {\n  /**\n   * Maximum number of keywords to extract from the query.\n   */\n  maxKeyword?: number;\n\n  /**\n   * Maximum number of search results to return.\n   * @default 10\n   */\n  limit?: number;\n\n  /**\n   * Maximum number of tool calls.\n   * @default 3\n   */\n  maxToolCalls?: number;\n\n  /**\n   * Sources to search from (e.g., \"douyin\", \"toutiao\").\n   */\n  sources?: string[];\n\n  /**\n   * User location for optimizing search results.\n   */\n  userLocation?: {\n    /**\n     * Location type.\n     * @default \"approximate\"\n     */\n    type?: \"approximate\";\n    /**\n     * Country name (e.g., \"中国\").\n     */\n    country?: string;\n    /**\n     * Region/Province name (e.g., \"浙江\").\n     */\n    region?: string;\n    /**\n     * City name (e.g., \"杭州\").\n     */\n    city?: string;\n  };\n}\n\nconst webSearchToolFactory = createProviderToolFactoryWithOutputSchema<\n  {},\n  {\n    /**\n     * The search result.\n     */\n    result: string;\n  },\n  WebSearchArgs\n>({\n  id: \"volcengine.web_search\",\n  inputSchema: webSearchInputSchema,\n  outputSchema: webSearchOutputSchema,\n});\n\n/**\n * Volcengine Web Search Tool\n *\n * Enables the model to perform web searches to retrieve up-to-date information.\n *\n * @see https://www.volcengine.com/docs/82379/1756990\n *\n * @example\n * ```ts\n * volcengineTools.webSearch({\n *   maxKeyword: 5,\n *   limit: 20,\n *   sources: [\"douyin\", \"toutiao\"],\n *   userLocation: {\n *     type: \"approximate\",\n *     country: \"中国\",\n *     region: \"浙江\",\n *     city: \"杭州\",\n *   },\n * })\n * ```\n */\nexport const webSearch = (args: WebSearchArgs = {}) => {\n  return webSearchToolFactory(args);\n};\n"],"names":["createProviderToolFactoryWithOutputSchema","lazySchema","zodSchema","z","userLocationSchema","object","type","enum","optional","country","string","region","city","strict","webSearchArgsSchema","maxKeyword","number","int","min","max","limit","maxToolCalls","sources","array","userLocation","webSearchInputSchema","webSearchOutputSchema","result","webSearchToolFactory","id","inputSchema","outputSchema","webSearch","args"],"mappings":"AAAA,SACEA,yCAAyC,EACzCC,UAAU,EACVC,SAAS,QACJ,yBAAyB;AAChC,SAASC,CAAC,QAAQ,SAAS;AAE3B,MAAMC,qBAAqBD,EACxBE,MAAM,CAAC;IACNC,MAAMH,EAAEI,IAAI,CAAC;QAAC;KAAc,EAAEC,QAAQ;IACtCC,SAASN,EAAEO,MAAM,GAAGF,QAAQ;IAC5BG,QAAQR,EAAEO,MAAM,GAAGF,QAAQ;IAC3BI,MAAMT,EAAEO,MAAM,GAAGF,QAAQ;AAC3B,GACCK,MAAM;AAET,OAAO,MAAMC,sBAAsBb,WAAW,IAC5CC,UACEC,EACGE,MAAM,CAAC;QACNU,YAAYZ,EAAEa,MAAM,GAAGC,GAAG,GAAGC,GAAG,CAAC,GAAGC,GAAG,CAAC,IAAIX,QAAQ;QACpDY,OAAOjB,EAAEa,MAAM,GAAGC,GAAG,GAAGC,GAAG,CAAC,GAAGC,GAAG,CAAC,IAAIX,QAAQ;QAC/Ca,cAAclB,EAAEa,MAAM,GAAGC,GAAG,GAAGC,GAAG,CAAC,GAAGC,GAAG,CAAC,IAAIX,QAAQ;QACtDc,SAASnB,EAAEoB,KAAK,CAACpB,EAAEO,MAAM,IAAIF,QAAQ;QACrCgB,cAAcpB,mBAAmBI,QAAQ;IAC3C,GACCK,MAAM,KAEX;AAEF,MAAMY,uBAAuBxB,WAAW,IAAMC,UAAUC,EAAEE,MAAM,CAAC,CAAC;AAElE,OAAO,MAAMqB,wBAAwBzB,WAAW,IAC9CC,UAAUC,EAAEE,MAAM,CAAC;QAAEsB,QAAQxB,EAAEO,MAAM;IAAG,KACxC;AAoDF,MAAMkB,uBAAuB5B,0CAS3B;IACA6B,IAAI;IACJC,aAAaL;IACbM,cAAcL;AAChB;AAEA;;;;;;;;;;;;;;;;;;;;;CAqBC,GACD,OAAO,MAAMM,YAAY,CAACC,OAAsB,CAAC,CAAC;IAChD,OAAOL,qBAAqBK;AAC9B,EAAE"}

package/dist/volcengine-chat-prompt.d.ts

@@ -21,6 +21,9 @@ export type VolcengineChatUserMessageContent = {
     video_url: {
         url: string;
     };
+} | {
+    type: 'input_file';
+    file_url: string;
 };
 export interface VolcengineChatAssistantMessage {
     role: 'assistant';

package/dist/volcengine-chat-prompt.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/volcengine-chat-prompt.ts"],"sourcesContent":["// export type MistralPrompt = Array<MistralMessage>;\nexport type VolcengineChatPrompt = Array<VolcengineChatMessage>;\n\nexport type VolcengineChatMessage =\n    | VolcengineChatSystemMessage\n    | VolcengineChatUserMessage\n    | VolcengineChatAssistantMessage\n    | VolcengineChatToolMessage;\n\nexport interface VolcengineChatSystemMessage {\n    role: 'system';\n    content: string;\n}\n\nexport interface VolcengineChatUserMessage {\n    role: 'user';\n    content: Array<VolcengineChatUserMessageContent>;\n}\n\nexport type VolcengineChatUserMessageContent =\n    | { type: 'text'; text: string }\n    | { type: 'image_url'; image_url: { url: string, } }\n    | { type: 'video_url', video_url: { url: string, } }\n\nexport interface VolcengineChatAssistantMessage {\n    role: 'assistant';\n    reasoning_content?: string;\n    content: string;\n    tool_calls?: Array<{\n        id: string;\n        type: 'function';\n        function: { name: string; arguments: string };\n    }>;\n}\n\nexport interface VolcengineChatToolMessage {\n    role: 'tool';\n    content: string;\n    tool_call_id: string;\n}"],"names":[],"mappings":"AAAA,qDAAqD;AAmCrD,WAIC"}
+{"version":3,"sources":["../src/volcengine-chat-prompt.ts"],"sourcesContent":["// export type MistralPrompt = Array<MistralMessage>;\nexport type VolcengineChatPrompt = Array<VolcengineChatMessage>;\n\nexport type VolcengineChatMessage =\n    | VolcengineChatSystemMessage\n    | VolcengineChatUserMessage\n    | VolcengineChatAssistantMessage\n    | VolcengineChatToolMessage;\n\nexport interface VolcengineChatSystemMessage {\n    role: 'system';\n    content: string;\n}\n\nexport interface VolcengineChatUserMessage {\n    role: 'user';\n    content: Array<VolcengineChatUserMessageContent>;\n}\n\nexport type VolcengineChatUserMessageContent =\n    | { type: 'text'; text: string }\n    | { type: 'image_url'; image_url: { url: string } }\n    | { type: 'video_url'; video_url: { url: string } }\n    | { type: 'input_file'; file_url: string }\n\nexport interface VolcengineChatAssistantMessage {\n    role: 'assistant';\n    reasoning_content?: string;\n    content: string;\n    tool_calls?: Array<{\n        id: string;\n        type: 'function';\n        function: { name: string; arguments: string };\n    }>;\n}\n\nexport interface VolcengineChatToolMessage {\n    role: 'tool';\n    content: string;\n    tool_call_id: string;\n}"],"names":[],"mappings":"AAAA,qDAAqD;AAoCrD,WAIC"}

package/dist/volcengine-prepare-tools.js

@@ -1,5 +1,8 @@
 import { UnsupportedFunctionalityError } from '@ai-sdk/provider';
-export function prepareTools({ tools, toolChoice }) {
+import { validateTypes } from '@ai-sdk/provider-utils';
+import { webSearchArgsSchema } from './tool/web-search';
+import { imageGenerationArgsSchema } from './tool/image-generation';
+export async function prepareTools({ tools, toolChoice }) {
     // when the tools array is empty, change it to undefined to prevent errors:
     tools = tools?.length ? tools : undefined;
     const toolWarnings = [];
@@ -13,10 +16,48 @@ export function prepareTools({ tools, toolChoice }) {
     const volcengineTools = [];
     for (const tool of tools){
         if (tool.type === 'provider') {
-            toolWarnings.push({
-                type: 'unsupported',
-                feature: `provider-defined tool ${tool.id}`
-            });
+            const toolId = tool.id;
+            switch(toolId){
+                case 'volcengine.web_search':
+                    {
+                        const args = await validateTypes({
+                            value: tool.args,
+                            schema: webSearchArgsSchema
+                        });
+                        volcengineTools.push({
+                            type: 'web_search',
+                            web_search: {
+                                max_keyword: args.maxKeyword,
+                                limit: args.limit,
+                                max_tool_calls: args.maxToolCalls
+                            }
+                        });
+                        break;
+                    }
+                case 'volcengine.image_generation':
+                    {
+                        const args = await validateTypes({
+                            value: tool.args,
+                            schema: imageGenerationArgsSchema
+                        });
+                        volcengineTools.push({
+                            type: 'image_generation',
+                            image_generation: {
+                                size: args.size,
+                                watermark: false,
+                                response_format: args.responseFormat,
+                                image: args.image,
+                                sequential_image_generation: 'disabled'
+                            }
+                        });
+                        break;
+                    }
+                default:
+                    toolWarnings.push({
+                        type: 'unsupported',
+                        feature: `provider-defined tool ${toolId}`
+                    });
+            }
         } else {
             volcengineTools.push({
                 type: 'function',

package/dist/volcengine-prepare-tools.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/volcengine-prepare-tools.ts"],"sourcesContent":["import {\n  LanguageModelV3CallOptions,\n  SharedV3Warning,\n  UnsupportedFunctionalityError,\n} from '@ai-sdk/provider';\n\nexport type VolcengineToolChoice =\n  | 'auto'\n  | 'none'\n  | 'required'\n  | { type: 'function'; function: { name: string } };\n\nexport type VolcengineTool = {\n  type: 'function';\n  function: {\n    name: string;\n    description: string | undefined;\n    parameters: unknown;\n  };\n};\n\nexport function prepareTools({\n  tools,\n  toolChoice,\n}: {\n  tools: LanguageModelV3CallOptions['tools'];\n  toolChoice?: LanguageModelV3CallOptions['toolChoice'];\n}): {\n  tools: VolcengineTool[] | undefined;\n  toolChoice: VolcengineToolChoice | undefined;\n  toolWarnings: SharedV3Warning[];\n} {\n  // when the tools array is empty, change it to undefined to prevent errors:\n  tools = tools?.length ? tools : undefined;\n\n  const toolWarnings: SharedV3Warning[] = [];\n\n  if (tools == null) {\n    return { tools: undefined, toolChoice: undefined, toolWarnings };\n  }\n\n  const volcengineTools: VolcengineTool[] = [];\n\n  for (const tool of tools) {\n    if (tool.type === 'provider') {\n      toolWarnings.push({\n        type: 'unsupported',\n        feature: `provider-defined tool ${tool.id}`,\n      });\n    } else {\n      volcengineTools.push({\n        type: 'function',\n        function: {\n          name: tool.name,\n          description: tool.description,\n          parameters: tool.inputSchema,\n        },\n      });\n    }\n  }\n\n  if (toolChoice == null) {\n    return { tools: volcengineTools, toolChoice: undefined, toolWarnings };\n  }\n\n  const type = toolChoice.type;\n\n  switch (type) {\n    case 'auto':\n    case 'none':\n      return { tools: volcengineTools, toolChoice: type, toolWarnings };\n    case 'required':\n      return { tools: volcengineTools, toolChoice: 'required', toolWarnings };\n    case 'tool':\n      return {\n        tools: volcengineTools,\n        toolChoice: {\n          type: 'function',\n          function: { name: toolChoice.toolName },\n        },\n        toolWarnings,\n      };\n    default: {\n      const _exhaustiveCheck: never = type;\n      throw new UnsupportedFunctionalityError({\n        functionality: `tool choice type: ${_exhaustiveCheck}`,\n      });\n    }\n  }\n}\n"],"names":["UnsupportedFunctionalityError","prepareTools","tools","toolChoice","length","undefined","toolWarnings","volcengineTools","tool","type","push","feature","id","function","name","description","parameters","inputSchema","toolName","_exhaustiveCheck","functionality"],"mappings":"AAAA,SAGEA,6BAA6B,QACxB,mBAAmB;AAiB1B,OAAO,SAASC,aAAa,EAC3BC,KAAK,EACLC,UAAU,EAIX;IAKC,2EAA2E;IAC3ED,QAAQA,OAAOE,SAASF,QAAQG;IAEhC,MAAMC,eAAkC,EAAE;IAE1C,IAAIJ,SAAS,MAAM;QACjB,OAAO;YAAEA,OAAOG;YAAWF,YAAYE;YAAWC;QAAa;IACjE;IAEA,MAAMC,kBAAoC,EAAE;IAE5C,KAAK,MAAMC,QAAQN,MAAO;QACxB,IAAIM,KAAKC,IAAI,KAAK,YAAY;YAC5BH,aAAaI,IAAI,CAAC;gBAChBD,MAAM;gBACNE,SAAS,CAAC,sBAAsB,EAAEH,KAAKI,EAAE,EAAE;YAC7C;QACF,OAAO;YACLL,gBAAgBG,IAAI,CAAC;gBACnBD,MAAM;gBACNI,UAAU;oBACRC,MAAMN,KAAKM,IAAI;oBACfC,aAAaP,KAAKO,WAAW;oBAC7BC,YAAYR,KAAKS,WAAW;gBAC9B;YACF;QACF;IACF;IAEA,IAAId,cAAc,MAAM;QACtB,OAAO;YAAED,OAAOK;YAAiBJ,YAAYE;YAAWC;QAAa;IACvE;IAEA,MAAMG,OAAON,WAAWM,IAAI;IAE5B,OAAQA;QACN,KAAK;QACL,KAAK;YACH,OAAO;gBAAEP,OAAOK;gBAAiBJ,YAAYM;gBAAMH;YAAa;QAClE,KAAK;YACH,OAAO;gBAAEJ,OAAOK;gBAAiBJ,YAAY;gBAAYG;YAAa;QACxE,KAAK;YACH,OAAO;gBACLJ,OAAOK;gBACPJ,YAAY;oBACVM,MAAM;oBACNI,UAAU;wBAAEC,MAAMX,WAAWe,QAAQ;oBAAC;gBACxC;gBACAZ;YACF;QACF;YAAS;gBACP,MAAMa,mBAA0BV;gBAChC,MAAM,IAAIT,8BAA8B;oBACtCoB,eAAe,CAAC,kBAAkB,EAAED,kBAAkB;gBACxD;YACF;IACF;AACF"}
+{"version":3,"sources":["../src/volcengine-prepare-tools.ts"],"sourcesContent":["import {\n  LanguageModelV3CallOptions,\n  SharedV3Warning,\n  UnsupportedFunctionalityError,\n} from '@ai-sdk/provider';\nimport { validateTypes } from '@ai-sdk/provider-utils';\nimport { webSearchArgsSchema } from './tool/web-search';\nimport { imageGenerationArgsSchema } from './tool/image-generation';\n\nexport type VolcengineToolChoice =\n  | 'auto'\n  | 'none'\n  | 'required'\n  | { type: 'function'; function: { name: string } };\n\nexport type VolcengineTool =\n  | {\n      type: 'function';\n      function: {\n        name: string;\n        description: string | undefined;\n        parameters: unknown;\n      };\n    }\n  | {\n      type: 'web_search';\n      web_search?: {\n        max_keyword?: number;\n        limit?: number;\n        max_tool_calls?: number;\n      };\n    }\n  | {\n      type: 'image_generation';\n      image_generation?: {\n        size?: '2K' | '4K';\n        watermark: false;\n        response_format?: 'url' | 'b64_json';\n        image?: string | string[];\n        sequential_image_generation: 'disabled';\n      };\n    };\n\nexport async function prepareTools({\n  tools,\n  toolChoice,\n}: {\n  tools: LanguageModelV3CallOptions['tools'];\n  toolChoice?: LanguageModelV3CallOptions['toolChoice'];\n}): Promise<{\n  tools: VolcengineTool[] | undefined;\n  toolChoice: VolcengineToolChoice | undefined;\n  toolWarnings: SharedV3Warning[];\n}> {\n  // when the tools array is empty, change it to undefined to prevent errors:\n  tools = tools?.length ? tools : undefined;\n\n  const toolWarnings: SharedV3Warning[] = [];\n\n  if (tools == null) {\n    return { tools: undefined, toolChoice: undefined, toolWarnings };\n  }\n\n  const volcengineTools: VolcengineTool[] = [];\n\n  for (const tool of tools) {\n    if (tool.type === 'provider') {\n      const toolId = tool.id;\n\n      switch (toolId) {\n        case 'volcengine.web_search': {\n          const args = await validateTypes({\n            value: tool.args,\n            schema: webSearchArgsSchema,\n          });\n\n          volcengineTools.push({\n            type: 'web_search',\n            web_search: {\n              max_keyword: args.maxKeyword,\n              limit: args.limit,\n              max_tool_calls: args.maxToolCalls,\n            },\n          });\n          break;\n        }\n\n        case 'volcengine.image_generation': {\n          const args = await validateTypes({\n            value: tool.args,\n            schema: imageGenerationArgsSchema,\n          });\n\n          volcengineTools.push({\n            type: 'image_generation',\n            image_generation: {\n              size: args.size,\n              watermark: false,\n              response_format: args.responseFormat,\n              image: args.image,\n              sequential_image_generation: 'disabled',\n            },\n          });\n          break;\n        }\n\n        default:\n          toolWarnings.push({\n            type: 'unsupported',\n            feature: `provider-defined tool ${toolId}`,\n          });\n      }\n    } else {\n      volcengineTools.push({\n        type: 'function',\n        function: {\n          name: tool.name,\n          description: tool.description,\n          parameters: tool.inputSchema,\n        },\n      });\n    }\n  }\n\n  if (toolChoice == null) {\n    return { tools: volcengineTools, toolChoice: undefined, toolWarnings };\n  }\n\n  const type = toolChoice.type;\n\n  switch (type) {\n    case 'auto':\n    case 'none':\n      return { tools: volcengineTools, toolChoice: type, toolWarnings };\n    case 'required':\n      return { tools: volcengineTools, toolChoice: 'required', toolWarnings };\n    case 'tool':\n      return {\n        tools: volcengineTools,\n        toolChoice: {\n          type: 'function',\n          function: { name: toolChoice.toolName },\n        },\n        toolWarnings,\n      };\n    default: {\n      const _exhaustiveCheck: never = type;\n      throw new UnsupportedFunctionalityError({\n        functionality: `tool choice type: ${_exhaustiveCheck}`,\n      });\n    }\n  }\n}\n"],"names":["UnsupportedFunctionalityError","validateTypes","webSearchArgsSchema","imageGenerationArgsSchema","prepareTools","tools","toolChoice","length","undefined","toolWarnings","volcengineTools","tool","type","toolId","id","args","value","schema","push","web_search","max_keyword","maxKeyword","limit","max_tool_calls","maxToolCalls","image_generation","size","watermark","response_format","responseFormat","image","sequential_image_generation","feature","function","name","description","parameters","inputSchema","toolName","_exhaustiveCheck","functionality"],"mappings":"AAAA,SAGEA,6BAA6B,QACxB,mBAAmB;AAC1B,SAASC,aAAa,QAAQ,yBAAyB;AACvD,SAASC,mBAAmB,QAAQ,oBAAoB;AACxD,SAASC,yBAAyB,QAAQ,0BAA0B;AAoCpE,OAAO,eAAeC,aAAa,EACjCC,KAAK,EACLC,UAAU,EAIX;IAKC,2EAA2E;IAC3ED,QAAQA,OAAOE,SAASF,QAAQG;IAEhC,MAAMC,eAAkC,EAAE;IAE1C,IAAIJ,SAAS,MAAM;QACjB,OAAO;YAAEA,OAAOG;YAAWF,YAAYE;YAAWC;QAAa;IACjE;IAEA,MAAMC,kBAAoC,EAAE;IAE5C,KAAK,MAAMC,QAAQN,MAAO;QACxB,IAAIM,KAAKC,IAAI,KAAK,YAAY;YAC5B,MAAMC,SAASF,KAAKG,EAAE;YAEtB,OAAQD;gBACN,KAAK;oBAAyB;wBAC5B,MAAME,OAAO,MAAMd,cAAc;4BAC/Be,OAAOL,KAAKI,IAAI;4BAChBE,QAAQf;wBACV;wBAEAQ,gBAAgBQ,IAAI,CAAC;4BACnBN,MAAM;4BACNO,YAAY;gCACVC,aAAaL,KAAKM,UAAU;gCAC5BC,OAAOP,KAAKO,KAAK;gCACjBC,gBAAgBR,KAAKS,YAAY;4BACnC;wBACF;wBACA;oBACF;gBAEA,KAAK;oBAA+B;wBAClC,MAAMT,OAAO,MAAMd,cAAc;4BAC/Be,OAAOL,KAAKI,IAAI;4BAChBE,QAAQd;wBACV;wBAEAO,gBAAgBQ,IAAI,CAAC;4BACnBN,MAAM;4BACNa,kBAAkB;gCAChBC,MAAMX,KAAKW,IAAI;gCACfC,WAAW;gCACXC,iBAAiBb,KAAKc,cAAc;gCACpCC,OAAOf,KAAKe,KAAK;gCACjBC,6BAA6B;4BAC/B;wBACF;wBACA;oBACF;gBAEA;oBACEtB,aAAaS,IAAI,CAAC;wBAChBN,MAAM;wBACNoB,SAAS,CAAC,sBAAsB,EAAEnB,QAAQ;oBAC5C;YACJ;QACF,OAAO;YACLH,gBAAgBQ,IAAI,CAAC;gBACnBN,MAAM;gBACNqB,UAAU;oBACRC,MAAMvB,KAAKuB,IAAI;oBACfC,aAAaxB,KAAKwB,WAAW;oBAC7BC,YAAYzB,KAAK0B,WAAW;gBAC9B;YACF;QACF;IACF;IAEA,IAAI/B,cAAc,MAAM;QACtB,OAAO;YAAED,OAAOK;YAAiBJ,YAAYE;YAAWC;QAAa;IACvE;IAEA,MAAMG,OAAON,WAAWM,IAAI;IAE5B,OAAQA;QACN,KAAK;QACL,KAAK;YACH,OAAO;gBAAEP,OAAOK;gBAAiBJ,YAAYM;gBAAMH;YAAa;QAClE,KAAK;YACH,OAAO;gBAAEJ,OAAOK;gBAAiBJ,YAAY;gBAAYG;YAAa;QACxE,KAAK;YACH,OAAO;gBACLJ,OAAOK;gBACPJ,YAAY;oBACVM,MAAM;oBACNqB,UAAU;wBAAEC,MAAM5B,WAAWgC,QAAQ;oBAAC;gBACxC;gBACA7B;YACF;QACF;YAAS;gBACP,MAAM8B,mBAA0B3B;gBAChC,MAAM,IAAIZ,8BAA8B;oBACtCwC,eAAe,CAAC,kBAAkB,EAAED,kBAAkB;gBACxD;YACF;IACF;AACF"}

package/dist/volcengine-provider.d.ts

@@ -1,6 +1,8 @@
-import { LanguageModelV3, ProviderV3 } from "@ai-sdk/provider";
+import { ImageModelV3, LanguageModelV3, ProviderV3 } from "@ai-sdk/provider";
 import { FetchFunction } from "@ai-sdk/provider-utils";
-import { VolcengineModelId } from "./volcengine-chat-options";
+import { VolcengineModelId } from "./chat";
+import { VolcengineImageModelId } from "./image";
+import { volcengineTools } from "./volcengine-tools";
 export interface VolcengineProviderSettings {
     /**
   Use a different URL prefix for API calls, e.g. to use proxy servers.
@@ -34,9 +36,17 @@ export interface VolcengineProvider extends ProviderV3 {
   */
     chat(modelId: VolcengineModelId): LanguageModelV3;
     /**
+  Creates a model for image generation.
+  */
+    imageModel(modelId: VolcengineImageModelId): ImageModelV3;
+    /**
   Creates a model for text embeddings.
   */
     embeddingModel(modelId: VolcengineModelId): never;
+    /**
+  Volcengine-specific tools.
+  */
+    tools: typeof volcengineTools;
 }
 /**
 Create a Volcengine provider instance.

package/dist/volcengine-provider.js

@@ -1,7 +1,9 @@
 import { NoSuchModelError } from "@ai-sdk/provider";
 import { loadApiKey, withUserAgentSuffix, withoutTrailingSlash } from "@ai-sdk/provider-utils";
 import { VERSION } from "./version";
-import { VolcengineChatLanguageModel } from "./volcengine-chat-language-model";
+import { VolcengineChatLanguageModel } from "./chat";
+import { VolcengineImageModel } from "./image";
+import { volcengineTools } from "./volcengine-tools";
 /**
 Create a Volcengine provider instance.
  */ export function createVolcengine(options = {}) {
@@ -15,34 +17,33 @@ Create a Volcengine provider instance.
             ...options.headers
         }, `ai-sdk/volcengine/${VERSION}`);
     const createChatModel = (modelId)=>new VolcengineChatLanguageModel(modelId, {
-            provider: "volcengine",
+            provider: "volcengine.chat",
             baseURL,
             headers: getHeaders,
             fetch: options.fetch,
             generateId: options.generateId
         });
-    return Object.assign((modelId)=>{
-        if (new.target) {
-            throw new Error("The Volcengine model function cannot be called with the new keyword.");
-        }
+    const createImageModel = (modelId)=>new VolcengineImageModel(modelId, {
+            provider: "volcengine.image",
+            baseURL,
+            headers: getHeaders,
+            fetch: options.fetch
+        });
+    const provider = function(modelId) {
         return createChatModel(modelId);
-    }, {
-        specificationVersion: "v3",
-        languageModel: createChatModel,
-        chat: createChatModel,
-        embeddingModel: (modelId)=>{
-            throw new NoSuchModelError({
-                modelId,
-                modelType: "embeddingModel"
-            });
-        },
-        imageModel: (modelId)=>{
-            throw new NoSuchModelError({
-                modelId,
-                modelType: "imageModel"
-            });
-        }
-    });
+    };
+    provider.specificationVersion = "v3";
+    provider.languageModel = createChatModel;
+    provider.chat = createChatModel;
+    provider.imageModel = createImageModel;
+    provider.embeddingModel = (modelId)=>{
+        throw new NoSuchModelError({
+            modelId,
+            modelType: "embeddingModel"
+        });
+    };
+    provider.tools = volcengineTools;
+    return provider;
 }
 /**
 Default Volcengine provider instance.

package/dist/volcengine-provider.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/volcengine-provider.ts"],"sourcesContent":["import {\n  LanguageModelV3,\n  NoSuchModelError,\n  ProviderV3\n} from \"@ai-sdk/provider\";\nimport {\n  FetchFunction,\n  loadApiKey,\n  withUserAgentSuffix,\n  withoutTrailingSlash\n} from \"@ai-sdk/provider-utils\";\nimport { VERSION } from \"./version\";\nimport { VolcengineChatLanguageModel } from \"./volcengine-chat-language-model\";\nimport { VolcengineModelId } from \"./volcengine-chat-options\";\n\nexport interface VolcengineProviderSettings {\n  /**\nUse a different URL prefix for API calls, e.g. to use proxy servers.\nThe default prefix is `https://ark.cn-beijing.volces.com/api/v3`.\n   */\n  baseURL?: string;\n\n  /**\nAPI key that is sent using the `Authorization` header.\nIt defaults to the `ARK_API_KEY` environment variable.\n   */\n  apiKey?: string;\n\n  /**\nCustom headers to include in the requests.\n   */\n  headers?: Record<string, string>;\n\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n   */\n  fetch?: FetchFunction;\n\n  generateId?: () => string;\n}\n\nexport interface VolcengineProvider extends ProviderV3 {\n  (modelId: VolcengineModelId): LanguageModelV3;\n\n  /**\nCreates a model for text generation using the Chat Completions API.\n*/\n  languageModel(modelId: VolcengineModelId): LanguageModelV3;\n\n  /**\nCreates a model for text generation using the Chat Completions API.\n*/\n  chat(modelId: VolcengineModelId): LanguageModelV3;\n\n  /**\nCreates a model for text embeddings.\n*/\n  embeddingModel(modelId: VolcengineModelId): never;\n}\n\n/**\nCreate a Volcengine provider instance.\n */\nexport function createVolcengine(\n  options: VolcengineProviderSettings = {}\n): VolcengineProvider {\n  const baseURL =\n    withoutTrailingSlash(options.baseURL) ??\n    \"https://ark.cn-beijing.volces.com/api/v3\";\n\n  const getHeaders = () =>\n    withUserAgentSuffix(\n      {\n        Authorization: `Bearer ${loadApiKey({\n          apiKey: options.apiKey,\n          environmentVariableName: \"ARK_API_KEY\",\n          description: \"Volcengine\"\n        })}`,\n        ...options.headers\n      },\n      `ai-sdk/volcengine/${VERSION}`\n    );\n\n  const createChatModel = (modelId: string) =>\n    new VolcengineChatLanguageModel(modelId, {\n      provider: \"volcengine\",\n      baseURL,\n      headers: getHeaders,\n      fetch: options.fetch,\n      generateId: options.generateId\n    });\n\n  return Object.assign(\n    (modelId: string) => {\n      if (new.target) {\n        throw new Error(\n          \"The Volcengine model function cannot be called with the new keyword.\"\n        );\n      }\n\n      return createChatModel(modelId);\n    },\n    {\n      specificationVersion: \"v3\" as const,\n      languageModel: createChatModel,\n      chat: createChatModel,\n      embeddingModel: (modelId: string) => {\n        throw new NoSuchModelError({ modelId, modelType: \"embeddingModel\" });\n      },\n      imageModel: (modelId: string) => {\n        throw new NoSuchModelError({ modelId, modelType: \"imageModel\" });\n      }\n    }\n  ) as VolcengineProvider;\n}\n\n/**\nDefault Volcengine provider instance.\n */\nexport const volcengine = createVolcengine();\n"],"names":["NoSuchModelError","loadApiKey","withUserAgentSuffix","withoutTrailingSlash","VERSION","VolcengineChatLanguageModel","createVolcengine","options","baseURL","getHeaders","Authorization","apiKey","environmentVariableName","description","headers","createChatModel","modelId","provider","fetch","generateId","Object","assign","Error","specificationVersion","languageModel","chat","embeddingModel","modelType","imageModel","volcengine"],"mappings":"AAAA,SAEEA,gBAAgB,QAEX,mBAAmB;AAC1B,SAEEC,UAAU,EACVC,mBAAmB,EACnBC,oBAAoB,QACf,yBAAyB;AAChC,SAASC,OAAO,QAAQ,YAAY;AACpC,SAASC,2BAA2B,QAAQ,mCAAmC;AAiD/E;;CAEC,GACD,OAAO,SAASC,iBACdC,UAAsC,CAAC,CAAC;IAExC,MAAMC,UACJL,qBAAqBI,QAAQC,OAAO,KACpC;IAEF,MAAMC,aAAa,IACjBP,oBACE;YACEQ,eAAe,CAAC,OAAO,EAAET,WAAW;gBAClCU,QAAQJ,QAAQI,MAAM;gBACtBC,yBAAyB;gBACzBC,aAAa;YACf,IAAI;YACJ,GAAGN,QAAQO,OAAO;QACpB,GACA,CAAC,kBAAkB,EAAEV,SAAS;IAGlC,MAAMW,kBAAkB,CAACC,UACvB,IAAIX,4BAA4BW,SAAS;YACvCC,UAAU;YACVT;YACAM,SAASL;YACTS,OAAOX,QAAQW,KAAK;YACpBC,YAAYZ,QAAQY,UAAU;QAChC;IAEF,OAAOC,OAAOC,MAAM,CAClB,CAACL;QACC,IAAI,YAAY;YACd,MAAM,IAAIM,MACR;QAEJ;QAEA,OAAOP,gBAAgBC;IACzB,GACA;QACEO,sBAAsB;QACtBC,eAAeT;QACfU,MAAMV;QACNW,gBAAgB,CAACV;YACf,MAAM,IAAIhB,iBAAiB;gBAAEgB;gBAASW,WAAW;YAAiB;QACpE;QACAC,YAAY,CAACZ;YACX,MAAM,IAAIhB,iBAAiB;gBAAEgB;gBAASW,WAAW;YAAa;QAChE;IACF;AAEJ;AAEA;;CAEC,GACD,OAAO,MAAME,aAAavB,mBAAmB"}
+{"version":3,"sources":["../src/volcengine-provider.ts"],"sourcesContent":["import {\n  ImageModelV3,\n  LanguageModelV3,\n  NoSuchModelError,\n  ProviderV3\n} from \"@ai-sdk/provider\";\nimport {\n  FetchFunction,\n  loadApiKey,\n  withUserAgentSuffix,\n  withoutTrailingSlash\n} from \"@ai-sdk/provider-utils\";\nimport { VERSION } from \"./version\";\nimport { VolcengineChatLanguageModel, VolcengineModelId } from \"./chat\";\nimport { VolcengineImageModel, VolcengineImageModelId } from \"./image\";\nimport { volcengineTools } from \"./volcengine-tools\";\n\nexport interface VolcengineProviderSettings {\n  /**\nUse a different URL prefix for API calls, e.g. to use proxy servers.\nThe default prefix is `https://ark.cn-beijing.volces.com/api/v3`.\n   */\n  baseURL?: string;\n\n  /**\nAPI key that is sent using the `Authorization` header.\nIt defaults to the `ARK_API_KEY` environment variable.\n   */\n  apiKey?: string;\n\n  /**\nCustom headers to include in the requests.\n   */\n  headers?: Record<string, string>;\n\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n   */\n  fetch?: FetchFunction;\n\n  generateId?: () => string;\n}\n\nexport interface VolcengineProvider extends ProviderV3 {\n  (modelId: VolcengineModelId): LanguageModelV3;\n\n  /**\nCreates a model for text generation using the Chat Completions API.\n*/\n  languageModel(modelId: VolcengineModelId): LanguageModelV3;\n\n  /**\nCreates a model for text generation using the Chat Completions API.\n*/\n  chat(modelId: VolcengineModelId): LanguageModelV3;\n\n  /**\nCreates a model for image generation.\n*/\n  imageModel(\n    modelId: VolcengineImageModelId,\n\n  ): ImageModelV3;\n\n  /**\nCreates a model for text embeddings.\n*/\n  embeddingModel(modelId: VolcengineModelId): never;\n\n  /**\nVolcengine-specific tools.\n*/\n  tools: typeof volcengineTools;\n}\n\n/**\nCreate a Volcengine provider instance.\n */\nexport function createVolcengine(\n  options: VolcengineProviderSettings = {}\n): VolcengineProvider {\n  const baseURL =\n    withoutTrailingSlash(options.baseURL) ??\n    \"https://ark.cn-beijing.volces.com/api/v3\";\n\n  const getHeaders = () =>\n    withUserAgentSuffix(\n      {\n        Authorization: `Bearer ${loadApiKey({\n          apiKey: options.apiKey,\n          environmentVariableName: \"ARK_API_KEY\",\n          description: \"Volcengine\"\n        })}`,\n        ...options.headers\n      },\n      `ai-sdk/volcengine/${VERSION}`\n    );\n\n  const createChatModel = (modelId: string) =>\n    new VolcengineChatLanguageModel(modelId, {\n      provider: \"volcengine.chat\",\n      baseURL,\n      headers: getHeaders,\n      fetch: options.fetch,\n      generateId: options.generateId\n    });\n\n  const createImageModel = (\n    modelId: string,\n  ) =>\n    new VolcengineImageModel(modelId, {\n      provider: \"volcengine.image\",\n      baseURL,\n      headers: getHeaders,\n      fetch: options.fetch\n    });\n\n  const provider = function (modelId: VolcengineModelId) {\n    return createChatModel(modelId);\n  };\n\n  provider.specificationVersion = \"v3\" as const;\n  provider.languageModel = createChatModel;\n  provider.chat = createChatModel;\n  provider.imageModel = createImageModel;\n  provider.embeddingModel = (modelId: string) => {\n    throw new NoSuchModelError({ modelId, modelType: \"embeddingModel\" });\n  };\n  provider.tools = volcengineTools;\n\n  return provider as VolcengineProvider;\n}\n\n/**\nDefault Volcengine provider instance.\n */\nexport const volcengine = createVolcengine();\n"],"names":["NoSuchModelError","loadApiKey","withUserAgentSuffix","withoutTrailingSlash","VERSION","VolcengineChatLanguageModel","VolcengineImageModel","volcengineTools","createVolcengine","options","baseURL","getHeaders","Authorization","apiKey","environmentVariableName","description","headers","createChatModel","modelId","provider","fetch","generateId","createImageModel","specificationVersion","languageModel","chat","imageModel","embeddingModel","modelType","tools","volcengine"],"mappings":"AAAA,SAGEA,gBAAgB,QAEX,mBAAmB;AAC1B,SAEEC,UAAU,EACVC,mBAAmB,EACnBC,oBAAoB,QACf,yBAAyB;AAChC,SAASC,OAAO,QAAQ,YAAY;AACpC,SAASC,2BAA2B,QAA2B,SAAS;AACxE,SAASC,oBAAoB,QAAgC,UAAU;AACvE,SAASC,eAAe,QAAQ,qBAAqB;AA6DrD;;CAEC,GACD,OAAO,SAASC,iBACdC,UAAsC,CAAC,CAAC;IAExC,MAAMC,UACJP,qBAAqBM,QAAQC,OAAO,KACpC;IAEF,MAAMC,aAAa,IACjBT,oBACE;YACEU,eAAe,CAAC,OAAO,EAAEX,WAAW;gBAClCY,QAAQJ,QAAQI,MAAM;gBACtBC,yBAAyB;gBACzBC,aAAa;YACf,IAAI;YACJ,GAAGN,QAAQO,OAAO;QACpB,GACA,CAAC,kBAAkB,EAAEZ,SAAS;IAGlC,MAAMa,kBAAkB,CAACC,UACvB,IAAIb,4BAA4Ba,SAAS;YACvCC,UAAU;YACVT;YACAM,SAASL;YACTS,OAAOX,QAAQW,KAAK;YACpBC,YAAYZ,QAAQY,UAAU;QAChC;IAEF,MAAMC,mBAAmB,CACvBJ,UAEA,IAAIZ,qBAAqBY,SAAS;YAChCC,UAAU;YACVT;YACAM,SAASL;YACTS,OAAOX,QAAQW,KAAK;QACtB;IAEF,MAAMD,WAAW,SAAUD,OAA0B;QACnD,OAAOD,gBAAgBC;IACzB;IAEAC,SAASI,oBAAoB,GAAG;IAChCJ,SAASK,aAAa,GAAGP;IACzBE,SAASM,IAAI,GAAGR;IAChBE,SAASO,UAAU,GAAGJ;IACtBH,SAASQ,cAAc,GAAG,CAACT;QACzB,MAAM,IAAIlB,iBAAiB;YAAEkB;YAASU,WAAW;QAAiB;IACpE;IACAT,SAASU,KAAK,GAAGtB;IAEjB,OAAOY;AACT;AAEA;;CAEC,GACD,OAAO,MAAMW,aAAatB,mBAAmB"}

package/dist/volcengine-tools.d.ts

@@ -38,5 +38,7 @@ export declare const volcengineTools: {
      * })
      * ```
      */
-    webSearch: import("@ai-sdk/provider-utils").ProviderToolFactory<import("./tool/web-search").WebSearchInput, import("./tool/web-search").WebSearchInput>;
+    webSearch: (args?: import("./tool/web-search").WebSearchArgs) => import("@ai-sdk/provider-utils").Tool<{}, {
+        result: string;
+    }>;
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sweetretry/ai-sdk-volcengine-adapter",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Volcengine adapter for AI SDK",
   "license": "MIT",
   "author": "SweetRetry <zimin.zhang2000@gmail.com>",
@@ -48,7 +48,7 @@
   },
   "dependencies": {
     "@ai-sdk/provider": "^3.0.2",
-    "@ai-sdk/provider-utils": "^4.0.4",
+    "@ai-sdk/provider-utils": "^4.0.5",
     "zod": "^4.3.5"
   }
 }