@ai-sdk/deepseek 0.1.0 → 0.1.2

package/CHANGELOG.md

@@ -1,5 +1,27 @@
 # @ai-sdk/deepseek
 
+## 0.1.2
+
+### Patch Changes
+
+- ed012d2: feat (provider/deepseek): extract cache usage as provide metadata
+- Updated dependencies [ed012d2]
+- Updated dependencies [3a58a2e]
+  - @ai-sdk/openai-compatible@0.1.2
+  - @ai-sdk/provider-utils@2.1.2
+  - @ai-sdk/provider@1.0.6
+
+## 0.1.1
+
+### Patch Changes
+
+- 0a699f1: feat: add reasoning token support
+- Updated dependencies [e7a9ec9]
+- Updated dependencies [0a699f1]
+  - @ai-sdk/provider-utils@2.1.1
+  - @ai-sdk/openai-compatible@0.1.1
+  - @ai-sdk/provider@1.0.5
+
 ## 0.1.0
 
 ### Minor Changes

package/dist/index.d.mts

@@ -3,7 +3,7 @@ import { FetchFunction } from '@ai-sdk/provider-utils';
 import { OpenAICompatibleChatSettings } from '@ai-sdk/openai-compatible';
 export { OpenAICompatibleErrorData as DeepSeekErrorData } from '@ai-sdk/openai-compatible';
 
-type DeepSeekChatModelId = 'deepseek-chat' | (string & {});
+type DeepSeekChatModelId = 'deepseek-chat' | 'deepseek-reasoner' | (string & {});
 interface DeepSeekChatSettings extends OpenAICompatibleChatSettings {
 }
 

package/dist/index.d.ts

@@ -3,7 +3,7 @@ import { FetchFunction } from '@ai-sdk/provider-utils';
 import { OpenAICompatibleChatSettings } from '@ai-sdk/openai-compatible';
 export { OpenAICompatibleErrorData as DeepSeekErrorData } from '@ai-sdk/openai-compatible';
 
-type DeepSeekChatModelId = 'deepseek-chat' | (string & {});
+type DeepSeekChatModelId = 'deepseek-chat' | 'deepseek-reasoner' | (string & {});
 interface DeepSeekChatSettings extends OpenAICompatibleChatSettings {
 }
 

package/dist/index.js

@@ -28,14 +28,69 @@ module.exports = __toCommonJS(src_exports);
 // src/deepseek-provider.ts
 var import_openai_compatible = require("@ai-sdk/openai-compatible");
 var import_provider = require("@ai-sdk/provider");
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
+
+// src/deepseek-metadata-extractor.ts
 var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_zod = require("zod");
+var buildDeepseekMetadata = (usage) => {
+  var _a, _b;
+  return usage == null ? void 0 : {
+    deepseek: {
+      promptCacheHitTokens: (_a = usage.prompt_cache_hit_tokens) != null ? _a : NaN,
+      promptCacheMissTokens: (_b = usage.prompt_cache_miss_tokens) != null ? _b : NaN
+    }
+  };
+};
+var deepSeekMetadataExtractor = {
+  extractMetadata: ({ parsedBody }) => {
+    const parsed = (0, import_provider_utils.safeValidateTypes)({
+      value: parsedBody,
+      schema: deepSeekResponseSchema
+    });
+    return !parsed.success || parsed.value.usage == null ? void 0 : buildDeepseekMetadata(parsed.value.usage);
+  },
+  createStreamExtractor: () => {
+    let usage;
+    return {
+      processChunk: (chunk) => {
+        var _a, _b;
+        const parsed = (0, import_provider_utils.safeValidateTypes)({
+          value: chunk,
+          schema: deepSeekStreamChunkSchema
+        });
+        if (parsed.success && ((_b = (_a = parsed.value.choices) == null ? void 0 : _a[0]) == null ? void 0 : _b.finish_reason) === "stop" && parsed.value.usage) {
+          usage = parsed.value.usage;
+        }
+      },
+      buildMetadata: () => buildDeepseekMetadata(usage)
+    };
+  }
+};
+var deepSeekUsageSchema = import_zod.z.object({
+  prompt_cache_hit_tokens: import_zod.z.number().nullish(),
+  prompt_cache_miss_tokens: import_zod.z.number().nullish()
+});
+var deepSeekResponseSchema = import_zod.z.object({
+  usage: deepSeekUsageSchema.nullish()
+});
+var deepSeekStreamChunkSchema = import_zod.z.object({
+  choices: import_zod.z.array(
+    import_zod.z.object({
+      finish_reason: import_zod.z.string().nullish()
+    })
+  ).nullish(),
+  usage: deepSeekUsageSchema.nullish()
+});
+
+// src/deepseek-provider.ts
 function createDeepSeek(options = {}) {
   var _a;
-  const baseURL = (0, import_provider_utils.withoutTrailingSlash)(
+  const baseURL = (0, import_provider_utils2.withoutTrailingSlash)(
     (_a = options.baseURL) != null ? _a : "https://api.deepseek.com/v1"
   );
   const getHeaders = () => ({
-    Authorization: `Bearer ${(0, import_provider_utils.loadApiKey)({
+    Authorization: `Bearer ${(0, import_provider_utils2.loadApiKey)({
       apiKey: options.apiKey,
       environmentVariableName: "DEEPSEEK_API_KEY",
       description: "DeepSeek API key"
@@ -48,7 +103,8 @@ function createDeepSeek(options = {}) {
       url: ({ path }) => `${baseURL}${path}`,
       headers: getHeaders,
       fetch: options.fetch,
-      defaultObjectGenerationMode: "json"
+      defaultObjectGenerationMode: "json",
+      metadataExtractor: deepSeekMetadataExtractor
     });
   };
   const provider = (modelId, settings) => createLanguageModel(modelId, settings);

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/deepseek-provider.ts"],"sourcesContent":["export { createDeepSeek, deepseek } from './deepseek-provider';\nexport type {\n  DeepSeekProvider,\n  DeepSeekProviderSettings,\n} from './deepseek-provider';\nexport type { OpenAICompatibleErrorData as DeepSeekErrorData } from '@ai-sdk/openai-compatible';\n","import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport {\n  LanguageModelV1,\n  NoSuchModelError,\n  ProviderV1,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport {\n  DeepSeekChatModelId,\n  DeepSeekChatSettings,\n} from './deepseek-chat-settings';\n\nexport interface DeepSeekProviderSettings {\n  /**\nDeepSeek API key.\n*/\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n*/\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\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\nexport interface DeepSeekProvider extends ProviderV1 {\n  /**\nCreates a DeepSeek model for text generation.\n*/\n  (\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a DeepSeek model for text generation.\n*/\n  languageModel(\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a DeepSeek chat model for text generation.\n*/\n  chat(\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n}\n\nexport function createDeepSeek(\n  options: DeepSeekProviderSettings = {},\n): DeepSeekProvider {\n  const baseURL = withoutTrailingSlash(\n    options.baseURL ?? 'https://api.deepseek.com/v1',\n  );\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'DEEPSEEK_API_KEY',\n      description: 'DeepSeek API key',\n    })}`,\n    ...options.headers,\n  });\n\n  const createLanguageModel = (\n    modelId: DeepSeekChatModelId,\n    settings: DeepSeekChatSettings = {},\n  ) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n      provider: `deepseek.chat`,\n      url: ({ path }) => `${baseURL}${path}`,\n      headers: getHeaders,\n      fetch: options.fetch,\n      defaultObjectGenerationMode: 'json',\n    });\n  };\n\n  const provider = (\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ) => createLanguageModel(modelId, settings);\n\n  provider.languageModel = createLanguageModel;\n  provider.chat = createLanguageModel;\n  provider.textEmbeddingModel = (modelId: string) => {\n    throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n  };\n\n  return provider as DeepSeekProvider;\n}\n\nexport const deepseek = createDeepSeek();\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,+BAAkD;AAClD,sBAIO;AACP,4BAIO;AAoDA,SAAS,eACd,UAAoC,CAAC,GACnB;AAhEpB;AAiEE,QAAM,cAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,cAAU,kCAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,sBAAsB,CAC1B,SACA,WAAiC,CAAC,MAC/B;AACH,WAAO,IAAI,2DAAkC,SAAS,UAAU;AAAA,MAC9D,UAAU;AAAA,MACV,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,MACpC,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,MACf,6BAA6B;AAAA,IAC/B,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CACf,SACA,aACG,oBAAoB,SAAS,QAAQ;AAE1C,WAAS,gBAAgB;AACzB,WAAS,OAAO;AAChB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iCAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,SAAO;AACT;AAEO,IAAM,WAAW,eAAe;","names":[]}
+{"version":3,"sources":["../src/index.ts","../src/deepseek-provider.ts","../src/deepseek-metadata-extractor.ts"],"sourcesContent":["export { createDeepSeek, deepseek } from './deepseek-provider';\nexport type {\n  DeepSeekProvider,\n  DeepSeekProviderSettings,\n} from './deepseek-provider';\nexport type { OpenAICompatibleErrorData as DeepSeekErrorData } from '@ai-sdk/openai-compatible';\n","import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport {\n  LanguageModelV1,\n  NoSuchModelError,\n  ProviderV1,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport {\n  DeepSeekChatModelId,\n  DeepSeekChatSettings,\n} from './deepseek-chat-settings';\nimport { deepSeekMetadataExtractor } from './deepseek-metadata-extractor';\n\nexport interface DeepSeekProviderSettings {\n  /**\nDeepSeek API key.\n*/\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n*/\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\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\nexport interface DeepSeekProvider extends ProviderV1 {\n  /**\nCreates a DeepSeek model for text generation.\n*/\n  (\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a DeepSeek model for text generation.\n*/\n  languageModel(\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a DeepSeek chat model for text generation.\n*/\n  chat(\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n}\n\nexport function createDeepSeek(\n  options: DeepSeekProviderSettings = {},\n): DeepSeekProvider {\n  const baseURL = withoutTrailingSlash(\n    options.baseURL ?? 'https://api.deepseek.com/v1',\n  );\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'DEEPSEEK_API_KEY',\n      description: 'DeepSeek API key',\n    })}`,\n    ...options.headers,\n  });\n\n  const createLanguageModel = (\n    modelId: DeepSeekChatModelId,\n    settings: DeepSeekChatSettings = {},\n  ) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n      provider: `deepseek.chat`,\n      url: ({ path }) => `${baseURL}${path}`,\n      headers: getHeaders,\n      fetch: options.fetch,\n      defaultObjectGenerationMode: 'json',\n      metadataExtractor: deepSeekMetadataExtractor,\n    });\n  };\n\n  const provider = (\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ) => createLanguageModel(modelId, settings);\n\n  provider.languageModel = createLanguageModel;\n  provider.chat = createLanguageModel;\n  provider.textEmbeddingModel = (modelId: string) => {\n    throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n  };\n\n  return provider as DeepSeekProvider;\n}\n\nexport const deepseek = createDeepSeek();\n","import { MetadataExtractor } from '@ai-sdk/openai-compatible';\nimport { safeValidateTypes } from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\n\nconst buildDeepseekMetadata = (\n  usage: z.infer<typeof deepSeekUsageSchema> | undefined,\n) => {\n  return usage == null\n    ? undefined\n    : {\n        deepseek: {\n          promptCacheHitTokens: usage.prompt_cache_hit_tokens ?? NaN,\n          promptCacheMissTokens: usage.prompt_cache_miss_tokens ?? NaN,\n        },\n      };\n};\n\nexport const deepSeekMetadataExtractor: MetadataExtractor = {\n  extractMetadata: ({ parsedBody }: { parsedBody: unknown }) => {\n    const parsed = safeValidateTypes({\n      value: parsedBody,\n      schema: deepSeekResponseSchema,\n    });\n\n    return !parsed.success || parsed.value.usage == null\n      ? undefined\n      : buildDeepseekMetadata(parsed.value.usage);\n  },\n\n  createStreamExtractor: () => {\n    let usage: z.infer<typeof deepSeekUsageSchema> | undefined;\n\n    return {\n      processChunk: (chunk: unknown) => {\n        const parsed = safeValidateTypes({\n          value: chunk,\n          schema: deepSeekStreamChunkSchema,\n        });\n\n        if (\n          parsed.success &&\n          parsed.value.choices?.[0]?.finish_reason === 'stop' &&\n          parsed.value.usage\n        ) {\n          usage = parsed.value.usage;\n        }\n      },\n      buildMetadata: () => buildDeepseekMetadata(usage),\n    };\n  },\n};\n\nconst deepSeekUsageSchema = z.object({\n  prompt_cache_hit_tokens: z.number().nullish(),\n  prompt_cache_miss_tokens: z.number().nullish(),\n});\n\nconst deepSeekResponseSchema = z.object({\n  usage: deepSeekUsageSchema.nullish(),\n});\n\nconst deepSeekStreamChunkSchema = z.object({\n  choices: z\n    .array(\n      z.object({\n        finish_reason: z.string().nullish(),\n      }),\n    )\n    .nullish(),\n  usage: deepSeekUsageSchema.nullish(),\n});\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,+BAAkD;AAClD,sBAIO;AACP,IAAAA,yBAIO;;;ACTP,4BAAkC;AAClC,iBAAkB;AAElB,IAAM,wBAAwB,CAC5B,UACG;AANL;AAOE,SAAO,SAAS,OACZ,SACA;AAAA,IACE,UAAU;AAAA,MACR,uBAAsB,WAAM,4BAAN,YAAiC;AAAA,MACvD,wBAAuB,WAAM,6BAAN,YAAkC;AAAA,IAC3D;AAAA,EACF;AACN;AAEO,IAAM,4BAA+C;AAAA,EAC1D,iBAAiB,CAAC,EAAE,WAAW,MAA+B;AAC5D,UAAM,aAAS,yCAAkB;AAAA,MAC/B,OAAO;AAAA,MACP,QAAQ;AAAA,IACV,CAAC;AAED,WAAO,CAAC,OAAO,WAAW,OAAO,MAAM,SAAS,OAC5C,SACA,sBAAsB,OAAO,MAAM,KAAK;AAAA,EAC9C;AAAA,EAEA,uBAAuB,MAAM;AAC3B,QAAI;AAEJ,WAAO;AAAA,MACL,cAAc,CAAC,UAAmB;AAjCxC;AAkCQ,cAAM,aAAS,yCAAkB;AAAA,UAC/B,OAAO;AAAA,UACP,QAAQ;AAAA,QACV,CAAC;AAED,YACE,OAAO,aACP,kBAAO,MAAM,YAAb,mBAAuB,OAAvB,mBAA2B,mBAAkB,UAC7C,OAAO,MAAM,OACb;AACA,kBAAQ,OAAO,MAAM;AAAA,QACvB;AAAA,MACF;AAAA,MACA,eAAe,MAAM,sBAAsB,KAAK;AAAA,IAClD;AAAA,EACF;AACF;AAEA,IAAM,sBAAsB,aAAE,OAAO;AAAA,EACnC,yBAAyB,aAAE,OAAO,EAAE,QAAQ;AAAA,EAC5C,0BAA0B,aAAE,OAAO,EAAE,QAAQ;AAC/C,CAAC;AAED,IAAM,yBAAyB,aAAE,OAAO;AAAA,EACtC,OAAO,oBAAoB,QAAQ;AACrC,CAAC;AAED,IAAM,4BAA4B,aAAE,OAAO;AAAA,EACzC,SAAS,aACN;AAAA,IACC,aAAE,OAAO;AAAA,MACP,eAAe,aAAE,OAAO,EAAE,QAAQ;AAAA,IACpC,CAAC;AAAA,EACH,EACC,QAAQ;AAAA,EACX,OAAO,oBAAoB,QAAQ;AACrC,CAAC;;;ADPM,SAAS,eACd,UAAoC,CAAC,GACnB;AAjEpB;AAkEE,QAAM,cAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,cAAU,mCAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,sBAAsB,CAC1B,SACA,WAAiC,CAAC,MAC/B;AACH,WAAO,IAAI,2DAAkC,SAAS,UAAU;AAAA,MAC9D,UAAU;AAAA,MACV,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,MACpC,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,MACf,6BAA6B;AAAA,MAC7B,mBAAmB;AAAA,IACrB,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CACf,SACA,aACG,oBAAoB,SAAS,QAAQ;AAE1C,WAAS,gBAAgB;AACzB,WAAS,OAAO;AAChB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iCAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,SAAO;AACT;AAEO,IAAM,WAAW,eAAe;","names":["import_provider_utils"]}

package/dist/index.mjs

@@ -7,6 +7,61 @@ import {
   loadApiKey,
   withoutTrailingSlash
 } from "@ai-sdk/provider-utils";
+
+// src/deepseek-metadata-extractor.ts
+import { safeValidateTypes } from "@ai-sdk/provider-utils";
+import { z } from "zod";
+var buildDeepseekMetadata = (usage) => {
+  var _a, _b;
+  return usage == null ? void 0 : {
+    deepseek: {
+      promptCacheHitTokens: (_a = usage.prompt_cache_hit_tokens) != null ? _a : NaN,
+      promptCacheMissTokens: (_b = usage.prompt_cache_miss_tokens) != null ? _b : NaN
+    }
+  };
+};
+var deepSeekMetadataExtractor = {
+  extractMetadata: ({ parsedBody }) => {
+    const parsed = safeValidateTypes({
+      value: parsedBody,
+      schema: deepSeekResponseSchema
+    });
+    return !parsed.success || parsed.value.usage == null ? void 0 : buildDeepseekMetadata(parsed.value.usage);
+  },
+  createStreamExtractor: () => {
+    let usage;
+    return {
+      processChunk: (chunk) => {
+        var _a, _b;
+        const parsed = safeValidateTypes({
+          value: chunk,
+          schema: deepSeekStreamChunkSchema
+        });
+        if (parsed.success && ((_b = (_a = parsed.value.choices) == null ? void 0 : _a[0]) == null ? void 0 : _b.finish_reason) === "stop" && parsed.value.usage) {
+          usage = parsed.value.usage;
+        }
+      },
+      buildMetadata: () => buildDeepseekMetadata(usage)
+    };
+  }
+};
+var deepSeekUsageSchema = z.object({
+  prompt_cache_hit_tokens: z.number().nullish(),
+  prompt_cache_miss_tokens: z.number().nullish()
+});
+var deepSeekResponseSchema = z.object({
+  usage: deepSeekUsageSchema.nullish()
+});
+var deepSeekStreamChunkSchema = z.object({
+  choices: z.array(
+    z.object({
+      finish_reason: z.string().nullish()
+    })
+  ).nullish(),
+  usage: deepSeekUsageSchema.nullish()
+});
+
+// src/deepseek-provider.ts
 function createDeepSeek(options = {}) {
   var _a;
   const baseURL = withoutTrailingSlash(
@@ -26,7 +81,8 @@ function createDeepSeek(options = {}) {
       url: ({ path }) => `${baseURL}${path}`,
       headers: getHeaders,
       fetch: options.fetch,
-      defaultObjectGenerationMode: "json"
+      defaultObjectGenerationMode: "json",
+      metadataExtractor: deepSeekMetadataExtractor
     });
   };
   const provider = (modelId, settings) => createLanguageModel(modelId, settings);

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/deepseek-provider.ts"],"sourcesContent":["import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport {\n  LanguageModelV1,\n  NoSuchModelError,\n  ProviderV1,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport {\n  DeepSeekChatModelId,\n  DeepSeekChatSettings,\n} from './deepseek-chat-settings';\n\nexport interface DeepSeekProviderSettings {\n  /**\nDeepSeek API key.\n*/\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n*/\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\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\nexport interface DeepSeekProvider extends ProviderV1 {\n  /**\nCreates a DeepSeek model for text generation.\n*/\n  (\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a DeepSeek model for text generation.\n*/\n  languageModel(\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a DeepSeek chat model for text generation.\n*/\n  chat(\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n}\n\nexport function createDeepSeek(\n  options: DeepSeekProviderSettings = {},\n): DeepSeekProvider {\n  const baseURL = withoutTrailingSlash(\n    options.baseURL ?? 'https://api.deepseek.com/v1',\n  );\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'DEEPSEEK_API_KEY',\n      description: 'DeepSeek API key',\n    })}`,\n    ...options.headers,\n  });\n\n  const createLanguageModel = (\n    modelId: DeepSeekChatModelId,\n    settings: DeepSeekChatSettings = {},\n  ) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n      provider: `deepseek.chat`,\n      url: ({ path }) => `${baseURL}${path}`,\n      headers: getHeaders,\n      fetch: options.fetch,\n      defaultObjectGenerationMode: 'json',\n    });\n  };\n\n  const provider = (\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ) => createLanguageModel(modelId, settings);\n\n  provider.languageModel = createLanguageModel;\n  provider.chat = createLanguageModel;\n  provider.textEmbeddingModel = (modelId: string) => {\n    throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n  };\n\n  return provider as DeepSeekProvider;\n}\n\nexport const deepseek = createDeepSeek();\n"],"mappings":";AAAA,SAAS,yCAAyC;AAClD;AAAA,EAEE;AAAA,OAEK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AAoDA,SAAS,eACd,UAAoC,CAAC,GACnB;AAhEpB;AAiEE,QAAM,UAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,UAAU,WAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,sBAAsB,CAC1B,SACA,WAAiC,CAAC,MAC/B;AACH,WAAO,IAAI,kCAAkC,SAAS,UAAU;AAAA,MAC9D,UAAU;AAAA,MACV,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,MACpC,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,MACf,6BAA6B;AAAA,IAC/B,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CACf,SACA,aACG,oBAAoB,SAAS,QAAQ;AAE1C,WAAS,gBAAgB;AACzB,WAAS,OAAO;AAChB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,SAAO;AACT;AAEO,IAAM,WAAW,eAAe;","names":[]}
+{"version":3,"sources":["../src/deepseek-provider.ts","../src/deepseek-metadata-extractor.ts"],"sourcesContent":["import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport {\n  LanguageModelV1,\n  NoSuchModelError,\n  ProviderV1,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport {\n  DeepSeekChatModelId,\n  DeepSeekChatSettings,\n} from './deepseek-chat-settings';\nimport { deepSeekMetadataExtractor } from './deepseek-metadata-extractor';\n\nexport interface DeepSeekProviderSettings {\n  /**\nDeepSeek API key.\n*/\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n*/\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\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\nexport interface DeepSeekProvider extends ProviderV1 {\n  /**\nCreates a DeepSeek model for text generation.\n*/\n  (\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a DeepSeek model for text generation.\n*/\n  languageModel(\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a DeepSeek chat model for text generation.\n*/\n  chat(\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ): LanguageModelV1;\n}\n\nexport function createDeepSeek(\n  options: DeepSeekProviderSettings = {},\n): DeepSeekProvider {\n  const baseURL = withoutTrailingSlash(\n    options.baseURL ?? 'https://api.deepseek.com/v1',\n  );\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'DEEPSEEK_API_KEY',\n      description: 'DeepSeek API key',\n    })}`,\n    ...options.headers,\n  });\n\n  const createLanguageModel = (\n    modelId: DeepSeekChatModelId,\n    settings: DeepSeekChatSettings = {},\n  ) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n      provider: `deepseek.chat`,\n      url: ({ path }) => `${baseURL}${path}`,\n      headers: getHeaders,\n      fetch: options.fetch,\n      defaultObjectGenerationMode: 'json',\n      metadataExtractor: deepSeekMetadataExtractor,\n    });\n  };\n\n  const provider = (\n    modelId: DeepSeekChatModelId,\n    settings?: DeepSeekChatSettings,\n  ) => createLanguageModel(modelId, settings);\n\n  provider.languageModel = createLanguageModel;\n  provider.chat = createLanguageModel;\n  provider.textEmbeddingModel = (modelId: string) => {\n    throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n  };\n\n  return provider as DeepSeekProvider;\n}\n\nexport const deepseek = createDeepSeek();\n","import { MetadataExtractor } from '@ai-sdk/openai-compatible';\nimport { safeValidateTypes } from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\n\nconst buildDeepseekMetadata = (\n  usage: z.infer<typeof deepSeekUsageSchema> | undefined,\n) => {\n  return usage == null\n    ? undefined\n    : {\n        deepseek: {\n          promptCacheHitTokens: usage.prompt_cache_hit_tokens ?? NaN,\n          promptCacheMissTokens: usage.prompt_cache_miss_tokens ?? NaN,\n        },\n      };\n};\n\nexport const deepSeekMetadataExtractor: MetadataExtractor = {\n  extractMetadata: ({ parsedBody }: { parsedBody: unknown }) => {\n    const parsed = safeValidateTypes({\n      value: parsedBody,\n      schema: deepSeekResponseSchema,\n    });\n\n    return !parsed.success || parsed.value.usage == null\n      ? undefined\n      : buildDeepseekMetadata(parsed.value.usage);\n  },\n\n  createStreamExtractor: () => {\n    let usage: z.infer<typeof deepSeekUsageSchema> | undefined;\n\n    return {\n      processChunk: (chunk: unknown) => {\n        const parsed = safeValidateTypes({\n          value: chunk,\n          schema: deepSeekStreamChunkSchema,\n        });\n\n        if (\n          parsed.success &&\n          parsed.value.choices?.[0]?.finish_reason === 'stop' &&\n          parsed.value.usage\n        ) {\n          usage = parsed.value.usage;\n        }\n      },\n      buildMetadata: () => buildDeepseekMetadata(usage),\n    };\n  },\n};\n\nconst deepSeekUsageSchema = z.object({\n  prompt_cache_hit_tokens: z.number().nullish(),\n  prompt_cache_miss_tokens: z.number().nullish(),\n});\n\nconst deepSeekResponseSchema = z.object({\n  usage: deepSeekUsageSchema.nullish(),\n});\n\nconst deepSeekStreamChunkSchema = z.object({\n  choices: z\n    .array(\n      z.object({\n        finish_reason: z.string().nullish(),\n      }),\n    )\n    .nullish(),\n  usage: deepSeekUsageSchema.nullish(),\n});\n"],"mappings":";AAAA,SAAS,yCAAyC;AAClD;AAAA,EAEE;AAAA,OAEK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,OACK;;;ACTP,SAAS,yBAAyB;AAClC,SAAS,SAAS;AAElB,IAAM,wBAAwB,CAC5B,UACG;AANL;AAOE,SAAO,SAAS,OACZ,SACA;AAAA,IACE,UAAU;AAAA,MACR,uBAAsB,WAAM,4BAAN,YAAiC;AAAA,MACvD,wBAAuB,WAAM,6BAAN,YAAkC;AAAA,IAC3D;AAAA,EACF;AACN;AAEO,IAAM,4BAA+C;AAAA,EAC1D,iBAAiB,CAAC,EAAE,WAAW,MAA+B;AAC5D,UAAM,SAAS,kBAAkB;AAAA,MAC/B,OAAO;AAAA,MACP,QAAQ;AAAA,IACV,CAAC;AAED,WAAO,CAAC,OAAO,WAAW,OAAO,MAAM,SAAS,OAC5C,SACA,sBAAsB,OAAO,MAAM,KAAK;AAAA,EAC9C;AAAA,EAEA,uBAAuB,MAAM;AAC3B,QAAI;AAEJ,WAAO;AAAA,MACL,cAAc,CAAC,UAAmB;AAjCxC;AAkCQ,cAAM,SAAS,kBAAkB;AAAA,UAC/B,OAAO;AAAA,UACP,QAAQ;AAAA,QACV,CAAC;AAED,YACE,OAAO,aACP,kBAAO,MAAM,YAAb,mBAAuB,OAAvB,mBAA2B,mBAAkB,UAC7C,OAAO,MAAM,OACb;AACA,kBAAQ,OAAO,MAAM;AAAA,QACvB;AAAA,MACF;AAAA,MACA,eAAe,MAAM,sBAAsB,KAAK;AAAA,IAClD;AAAA,EACF;AACF;AAEA,IAAM,sBAAsB,EAAE,OAAO;AAAA,EACnC,yBAAyB,EAAE,OAAO,EAAE,QAAQ;AAAA,EAC5C,0BAA0B,EAAE,OAAO,EAAE,QAAQ;AAC/C,CAAC;AAED,IAAM,yBAAyB,EAAE,OAAO;AAAA,EACtC,OAAO,oBAAoB,QAAQ;AACrC,CAAC;AAED,IAAM,4BAA4B,EAAE,OAAO;AAAA,EACzC,SAAS,EACN;AAAA,IACC,EAAE,OAAO;AAAA,MACP,eAAe,EAAE,OAAO,EAAE,QAAQ;AAAA,IACpC,CAAC;AAAA,EACH,EACC,QAAQ;AAAA,EACX,OAAO,oBAAoB,QAAQ;AACrC,CAAC;;;ADPM,SAAS,eACd,UAAoC,CAAC,GACnB;AAjEpB;AAkEE,QAAM,UAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,UAAU,WAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAEA,QAAM,sBAAsB,CAC1B,SACA,WAAiC,CAAC,MAC/B;AACH,WAAO,IAAI,kCAAkC,SAAS,UAAU;AAAA,MAC9D,UAAU;AAAA,MACV,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,MACpC,SAAS;AAAA,MACT,OAAO,QAAQ;AAAA,MACf,6BAA6B;AAAA,MAC7B,mBAAmB;AAAA,IACrB,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CACf,SACA,aACG,oBAAoB,SAAS,QAAQ;AAE1C,WAAS,gBAAgB;AACzB,WAAS,OAAO;AAChB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,SAAO;AACT;AAEO,IAAM,WAAW,eAAe;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/deepseek",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -19,9 +19,9 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/openai-compatible": "0.1.0",
-    "@ai-sdk/provider": "1.0.4",
-    "@ai-sdk/provider-utils": "2.1.0"
+    "@ai-sdk/openai-compatible": "0.1.2",
+    "@ai-sdk/provider": "1.0.6",
+    "@ai-sdk/provider-utils": "2.1.2"
   },
   "devDependencies": {
     "@types/node": "^18",