@ai-sdk/moonshotai 0.0.1

package/CHANGELOG.md

@@ -0,0 +1,7 @@
+# @ai-sdk/moonshotai
+
+## 0.0.1
+
+### Patch Changes
+
+- 2adcbf7: feat(provider): add Moonshot AI provider

package/LICENSE

@@ -0,0 +1,13 @@
+Copyright 2023 Vercel, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

package/README.md

@@ -0,0 +1,64 @@
+# AI SDK - Moonshot AI Provider
+
+The **[Moonshot AI provider](https://ai-sdk.dev/providers/ai-sdk-providers/moonshotai)** for the [AI SDK](https://ai-sdk.dev/docs) contains language model support for the [Moonshot AI](https://platform.moonshot.cn) platform, including the Kimi model series.
+
+## Setup
+
+The Moonshot AI provider is available in the `@ai-sdk/moonshotai` module. You can install it with
+
+```bash
+npm i @ai-sdk/moonshotai
+```
+
+## Skill for Coding Agents
+
+If you use coding agents such as Claude Code or Cursor, we highly recommend adding the AI SDK skill to your repository:
+
+```shell
+npx skills add vercel/ai
+```
+
+## Provider Instance
+
+You can import the default provider instance `moonshotai` from `@ai-sdk/moonshotai`:
+
+```ts
+import { moonshotai } from '@ai-sdk/moonshotai';
+```
+
+## Language Model Example
+
+```ts
+import { moonshotai } from '@ai-sdk/moonshotai';
+import { generateText } from 'ai';
+
+const { text } = await generateText({
+  model: moonshotai('kimi-k2.5'),
+  prompt: 'Write a JavaScript function that sorts a list:',
+});
+```
+
+## Thinking Mode Example (Kimi K2 Thinking)
+
+```ts
+import { moonshotai } from '@ai-sdk/moonshotai';
+import { generateText } from 'ai';
+
+const { text } = await generateText({
+  model: moonshotai('kimi-k2.5'),
+  prompt: 'Invent a new holiday and describe its traditions.',
+  providerOptions: {
+    moonshotai: {
+      thinking: {
+        type: 'enabled',
+        budgetTokens: 2048,
+      },
+      reasoningHistory: 'interleaved',
+    } as MoonshotAIProviderOptions,
+  },
+});
+```
+
+## Documentation
+
+Please check out the **[Moonshot AI provider](https://ai-sdk.dev/providers/ai-sdk-providers/moonshotai)** for more information.

package/dist/index.d.mts

@@ -0,0 +1,59 @@
+import { ProviderV2, LanguageModelV2 } from '@ai-sdk/provider';
+import { FetchFunction } from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+
+type MoonshotAIChatModelId = 'moonshot-v1-8k' | 'moonshot-v1-32k' | 'moonshot-v1-128k' | 'kimi-k2' | 'kimi-k2-0905' | 'kimi-k2-thinking' | 'kimi-k2-thinking-turbo' | 'kimi-k2-turbo' | 'kimi-k2.5' | (string & {});
+declare const moonshotaiProviderOptions: z.ZodObject<{
+    thinking: z.ZodOptional<z.ZodObject<{
+        type: z.ZodOptional<z.ZodEnum<{
+            enabled: "enabled";
+            disabled: "disabled";
+        }>>;
+        budgetTokens: z.ZodOptional<z.ZodNumber>;
+    }, z.core.$strip>>;
+    reasoningHistory: z.ZodOptional<z.ZodEnum<{
+        disabled: "disabled";
+        interleaved: "interleaved";
+        preserved: "preserved";
+    }>>;
+}, z.core.$strip>;
+type MoonshotAIProviderOptions = z.infer<typeof moonshotaiProviderOptions>;
+
+interface MoonshotAIProviderSettings {
+    /**
+     * Moonshot API key. Default value is taken from the `MOONSHOT_API_KEY`
+     * environment variable.
+     */
+    apiKey?: string;
+    /**
+     * Base URL for the API calls.
+     */
+    baseURL?: string;
+    /**
+     * Custom headers to include in the requests.
+     */
+    headers?: Record<string, string>;
+    /**
+     * Custom fetch implementation. You can use it as a middleware to intercept requests,
+     * or to provide a custom fetch implementation for e.g. testing.
+     */
+    fetch?: FetchFunction;
+}
+interface MoonshotAIProvider extends ProviderV2 {
+    /**
+     * Creates a model for text generation.
+     */
+    (modelId: MoonshotAIChatModelId): LanguageModelV2;
+    /**
+     * Creates a chat model for text generation.
+     */
+    chatModel(modelId: MoonshotAIChatModelId): LanguageModelV2;
+    /**
+     * Creates a language model for text generation.
+     */
+    languageModel(modelId: MoonshotAIChatModelId): LanguageModelV2;
+}
+declare function createMoonshotAI(options?: MoonshotAIProviderSettings): MoonshotAIProvider;
+declare const moonshotai: MoonshotAIProvider;
+
+export { type MoonshotAIChatModelId, type MoonshotAIProvider, type MoonshotAIProviderOptions, type MoonshotAIProviderSettings, createMoonshotAI, moonshotai };

package/dist/index.d.ts

@@ -0,0 +1,59 @@
+import { ProviderV2, LanguageModelV2 } from '@ai-sdk/provider';
+import { FetchFunction } from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+
+type MoonshotAIChatModelId = 'moonshot-v1-8k' | 'moonshot-v1-32k' | 'moonshot-v1-128k' | 'kimi-k2' | 'kimi-k2-0905' | 'kimi-k2-thinking' | 'kimi-k2-thinking-turbo' | 'kimi-k2-turbo' | 'kimi-k2.5' | (string & {});
+declare const moonshotaiProviderOptions: z.ZodObject<{
+    thinking: z.ZodOptional<z.ZodObject<{
+        type: z.ZodOptional<z.ZodEnum<{
+            enabled: "enabled";
+            disabled: "disabled";
+        }>>;
+        budgetTokens: z.ZodOptional<z.ZodNumber>;
+    }, z.core.$strip>>;
+    reasoningHistory: z.ZodOptional<z.ZodEnum<{
+        disabled: "disabled";
+        interleaved: "interleaved";
+        preserved: "preserved";
+    }>>;
+}, z.core.$strip>;
+type MoonshotAIProviderOptions = z.infer<typeof moonshotaiProviderOptions>;
+
+interface MoonshotAIProviderSettings {
+    /**
+     * Moonshot API key. Default value is taken from the `MOONSHOT_API_KEY`
+     * environment variable.
+     */
+    apiKey?: string;
+    /**
+     * Base URL for the API calls.
+     */
+    baseURL?: string;
+    /**
+     * Custom headers to include in the requests.
+     */
+    headers?: Record<string, string>;
+    /**
+     * Custom fetch implementation. You can use it as a middleware to intercept requests,
+     * or to provide a custom fetch implementation for e.g. testing.
+     */
+    fetch?: FetchFunction;
+}
+interface MoonshotAIProvider extends ProviderV2 {
+    /**
+     * Creates a model for text generation.
+     */
+    (modelId: MoonshotAIChatModelId): LanguageModelV2;
+    /**
+     * Creates a chat model for text generation.
+     */
+    chatModel(modelId: MoonshotAIChatModelId): LanguageModelV2;
+    /**
+     * Creates a language model for text generation.
+     */
+    languageModel(modelId: MoonshotAIChatModelId): LanguageModelV2;
+}
+declare function createMoonshotAI(options?: MoonshotAIProviderSettings): MoonshotAIProvider;
+declare const moonshotai: MoonshotAIProvider;
+
+export { type MoonshotAIChatModelId, type MoonshotAIProvider, type MoonshotAIProviderOptions, type MoonshotAIProviderSettings, createMoonshotAI, moonshotai };

package/dist/index.js

@@ -0,0 +1,132 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  createMoonshotAI: () => createMoonshotAI,
+  moonshotai: () => moonshotai
+});
+module.exports = __toCommonJS(src_exports);
+
+// src/moonshotai-provider.ts
+var import_provider = require("@ai-sdk/provider");
+var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_v4 = require("zod/v4");
+
+// src/moonshotai-chat-language-model.ts
+var import_openai_compatible = require("@ai-sdk/openai-compatible");
+
+// src/convert-moonshotai-chat-usage.ts
+function convertMoonshotAIChatUsage(usage) {
+  var _a, _b, _c, _d, _e, _f, _g, _h;
+  if (usage == null) {
+    return {
+      inputTokens: void 0,
+      outputTokens: void 0,
+      totalTokens: void 0
+    };
+  }
+  const cachedInputTokens = (_c = (_b = usage.cached_tokens) != null ? _b : (_a = usage.prompt_tokens_details) == null ? void 0 : _a.cached_tokens) != null ? _c : void 0;
+  const reasoningTokens = (_e = (_d = usage.completion_tokens_details) == null ? void 0 : _d.reasoning_tokens) != null ? _e : void 0;
+  return {
+    inputTokens: (_f = usage.prompt_tokens) != null ? _f : void 0,
+    outputTokens: (_g = usage.completion_tokens) != null ? _g : void 0,
+    totalTokens: (_h = usage.total_tokens) != null ? _h : void 0,
+    reasoningTokens,
+    cachedInputTokens
+  };
+}
+
+// src/moonshotai-chat-language-model.ts
+var MoonshotAIChatLanguageModel = class extends import_openai_compatible.OpenAICompatibleChatLanguageModel {
+  constructor(modelId, config) {
+    super(modelId, config);
+  }
+  async doGenerate(options) {
+    var _a, _b;
+    const result = await super.doGenerate(options);
+    const usage = (_b = (_a = result.response) == null ? void 0 : _a.body) == null ? void 0 : _b.usage;
+    return {
+      ...result,
+      usage: convertMoonshotAIChatUsage(usage)
+    };
+  }
+};
+
+// src/version.ts
+var VERSION = "0.0.1";
+
+// src/moonshotai-provider.ts
+var moonshotaiErrorSchema = import_v4.z.object({
+  error: import_v4.z.object({
+    message: import_v4.z.string(),
+    type: import_v4.z.string().nullish()
+  })
+});
+var moonshotaiErrorStructure = {
+  errorSchema: moonshotaiErrorSchema,
+  errorToMessage: (data) => data.error.message
+};
+var defaultBaseURL = "https://api.moonshot.ai/v1";
+function createMoonshotAI(options = {}) {
+  var _a;
+  const baseURL = (0, import_provider_utils.withoutTrailingSlash)((_a = options.baseURL) != null ? _a : defaultBaseURL);
+  const getHeaders = () => (0, import_provider_utils.withUserAgentSuffix)(
+    {
+      Authorization: `Bearer ${(0, import_provider_utils.loadApiKey)({
+        apiKey: options.apiKey,
+        environmentVariableName: "MOONSHOT_API_KEY",
+        description: "Moonshot API key"
+      })}`,
+      ...options.headers
+    },
+    `ai-sdk/moonshotai/${VERSION}`
+  );
+  const getCommonModelConfig = (modelType) => ({
+    provider: `moonshotai.${modelType}`,
+    url: ({ path }) => `${baseURL}${path}`,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
+  const createChatModel = (modelId) => {
+    return new MoonshotAIChatLanguageModel(modelId, {
+      ...getCommonModelConfig("chat"),
+      errorStructure: moonshotaiErrorStructure
+    });
+  };
+  const provider = (modelId) => createChatModel(modelId);
+  provider.specificationVersion = "v2";
+  provider.chatModel = createChatModel;
+  provider.languageModel = createChatModel;
+  provider.textEmbeddingModel = (modelId) => {
+    throw new import_provider.NoSuchModelError({ modelId, modelType: "textEmbeddingModel" });
+  };
+  provider.imageModel = (modelId) => {
+    throw new import_provider.NoSuchModelError({ modelId, modelType: "imageModel" });
+  };
+  return provider;
+}
+var moonshotai = createMoonshotAI();
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  createMoonshotAI,
+  moonshotai
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/moonshotai-provider.ts","../src/moonshotai-chat-language-model.ts","../src/convert-moonshotai-chat-usage.ts","../src/version.ts"],"sourcesContent":["export { createMoonshotAI, moonshotai } from './moonshotai-provider';\nexport type {\n  MoonshotAIProvider,\n  MoonshotAIProviderSettings,\n} from './moonshotai-provider';\nexport type {\n  MoonshotAIChatModelId,\n  MoonshotAIProviderOptions,\n} from './moonshotai-chat-options';\n","import { ProviderErrorStructure } from '@ai-sdk/openai-compatible';\nimport {\n  LanguageModelV2,\n  NoSuchModelError,\n  ProviderV2,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n  withUserAgentSuffix,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport { MoonshotAIChatLanguageModel } from './moonshotai-chat-language-model';\nimport { MoonshotAIChatModelId } from './moonshotai-chat-options';\nimport { VERSION } from './version';\n\nexport type MoonshotAIErrorData = z.infer<typeof moonshotaiErrorSchema>;\n\nconst moonshotaiErrorSchema = z.object({\n  error: z.object({\n    message: z.string(),\n    type: z.string().nullish(),\n  }),\n});\n\nconst moonshotaiErrorStructure: ProviderErrorStructure<MoonshotAIErrorData> = {\n  errorSchema: moonshotaiErrorSchema,\n  errorToMessage: data => data.error.message,\n};\n\nexport interface MoonshotAIProviderSettings {\n  /**\n   * Moonshot API key. Default value is taken from the `MOONSHOT_API_KEY`\n   * environment variable.\n   */\n  apiKey?: string;\n  /**\n   * Base URL for the API calls.\n   */\n  baseURL?: string;\n  /**\n   * Custom headers to include in the requests.\n   */\n  headers?: Record<string, string>;\n  /**\n   * Custom fetch implementation. You can use it as a middleware to intercept requests,\n   * or to provide a custom fetch implementation for e.g. testing.\n   */\n  fetch?: FetchFunction;\n}\n\nexport interface MoonshotAIProvider extends ProviderV2 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: MoonshotAIChatModelId): LanguageModelV2;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: MoonshotAIChatModelId): LanguageModelV2;\n\n  /**\n   * Creates a language model for text generation.\n   */\n  languageModel(modelId: MoonshotAIChatModelId): LanguageModelV2;\n}\n\nconst defaultBaseURL = 'https://api.moonshot.ai/v1';\n\nexport function createMoonshotAI(\n  options: MoonshotAIProviderSettings = {},\n): MoonshotAIProvider {\n  const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n  const getHeaders = () =>\n    withUserAgentSuffix(\n      {\n        Authorization: `Bearer ${loadApiKey({\n          apiKey: options.apiKey,\n          environmentVariableName: 'MOONSHOT_API_KEY',\n          description: 'Moonshot API key',\n        })}`,\n        ...options.headers,\n      },\n      `ai-sdk/moonshotai/${VERSION}`,\n    );\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const getCommonModelConfig = (modelType: string): CommonModelConfig => ({\n    provider: `moonshotai.${modelType}`,\n    url: ({ path }) => `${baseURL}${path}`,\n    headers: getHeaders,\n    fetch: options.fetch,\n  });\n\n  const createChatModel = (modelId: MoonshotAIChatModelId) => {\n    return new MoonshotAIChatLanguageModel(modelId, {\n      ...getCommonModelConfig('chat'),\n      errorStructure: moonshotaiErrorStructure,\n    });\n  };\n\n  const provider = (modelId: MoonshotAIChatModelId) => createChatModel(modelId);\n\n  provider.specificationVersion = 'v2' as const;\n  provider.chatModel = createChatModel;\n  provider.languageModel = createChatModel;\n\n  provider.textEmbeddingModel = (modelId: string) => {\n    throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n  };\n\n  provider.imageModel = (modelId: string) => {\n    throw new NoSuchModelError({ modelId, modelType: 'imageModel' });\n  };\n\n  return provider;\n}\n\nexport const moonshotai = createMoonshotAI();\n","import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport { OpenAICompatibleChatConfig } from '@ai-sdk/openai-compatible/internal';\nimport { LanguageModelV2 } from '@ai-sdk/provider';\nimport { convertMoonshotAIChatUsage } from './convert-moonshotai-chat-usage';\nimport { MoonshotAIChatModelId } from './moonshotai-chat-options';\n\nexport class MoonshotAIChatLanguageModel extends OpenAICompatibleChatLanguageModel {\n  constructor(\n    modelId: MoonshotAIChatModelId,\n    config: OpenAICompatibleChatConfig,\n  ) {\n    super(modelId, config);\n  }\n\n  async doGenerate(\n    options: Parameters<LanguageModelV2['doGenerate']>[0],\n  ): Promise<Awaited<ReturnType<LanguageModelV2['doGenerate']>>> {\n    const result = await super.doGenerate(options);\n\n    // @ts-expect-error accessing response body from parent result\n    const usage = result.response?.body?.usage;\n\n    return {\n      ...result,\n      usage: convertMoonshotAIChatUsage(usage),\n    };\n  }\n}\n","import { LanguageModelV2Usage } from '@ai-sdk/provider';\n\nexport function convertMoonshotAIChatUsage(\n  usage:\n    | {\n        prompt_tokens?: number | null;\n        completion_tokens?: number | null;\n        total_tokens?: number | null;\n        cached_tokens?: number | null;\n        prompt_tokens_details?: {\n          cached_tokens?: number | null;\n        } | null;\n        completion_tokens_details?: {\n          reasoning_tokens?: number | null;\n        } | null;\n      }\n    | undefined\n    | null,\n): LanguageModelV2Usage {\n  if (usage == null) {\n    return {\n      inputTokens: undefined,\n      outputTokens: undefined,\n      totalTokens: undefined,\n    };\n  }\n\n  const cachedInputTokens =\n    usage.cached_tokens ??\n    usage.prompt_tokens_details?.cached_tokens ??\n    undefined;\n\n  const reasoningTokens =\n    usage.completion_tokens_details?.reasoning_tokens ?? undefined;\n\n  return {\n    inputTokens: usage.prompt_tokens ?? undefined,\n    outputTokens: usage.completion_tokens ?? undefined,\n    totalTokens: usage.total_tokens ?? undefined,\n    reasoningTokens,\n    cachedInputTokens,\n  };\n}\n","declare const __PACKAGE_VERSION__: string;\n\nexport const VERSION = __PACKAGE_VERSION__;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,sBAIO;AACP,4BAKO;AACP,gBAAkB;;;ACZlB,+BAAkD;;;ACE3C,SAAS,2BACd,OAesB;AAlBxB;AAmBE,MAAI,SAAS,MAAM;AACjB,WAAO;AAAA,MACL,aAAa;AAAA,MACb,cAAc;AAAA,MACd,aAAa;AAAA,IACf;AAAA,EACF;AAEA,QAAM,qBACJ,iBAAM,kBAAN,aACA,WAAM,0BAAN,mBAA6B,kBAD7B,YAEA;AAEF,QAAM,mBACJ,iBAAM,8BAAN,mBAAiC,qBAAjC,YAAqD;AAEvD,SAAO;AAAA,IACL,cAAa,WAAM,kBAAN,YAAuB;AAAA,IACpC,eAAc,WAAM,sBAAN,YAA2B;AAAA,IACzC,cAAa,WAAM,iBAAN,YAAsB;AAAA,IACnC;AAAA,IACA;AAAA,EACF;AACF;;;ADpCO,IAAM,8BAAN,cAA0C,2DAAkC;AAAA,EACjF,YACE,SACA,QACA;AACA,UAAM,SAAS,MAAM;AAAA,EACvB;AAAA,EAEA,MAAM,WACJ,SAC6D;AAhBjE;AAiBI,UAAM,SAAS,MAAM,MAAM,WAAW,OAAO;AAG7C,UAAM,SAAQ,kBAAO,aAAP,mBAAiB,SAAjB,mBAAuB;AAErC,WAAO;AAAA,MACL,GAAG;AAAA,MACH,OAAO,2BAA2B,KAAK;AAAA,IACzC;AAAA,EACF;AACF;;;AEzBO,IAAM,UAAU;;;AHiBvB,IAAM,wBAAwB,YAAE,OAAO;AAAA,EACrC,OAAO,YAAE,OAAO;AAAA,IACd,SAAS,YAAE,OAAO;AAAA,IAClB,MAAM,YAAE,OAAO,EAAE,QAAQ;AAAA,EAC3B,CAAC;AACH,CAAC;AAED,IAAM,2BAAwE;AAAA,EAC5E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK,MAAM;AACrC;AAwCA,IAAM,iBAAiB;AAEhB,SAAS,iBACd,UAAsC,CAAC,GACnB;AAzEtB;AA0EE,QAAM,cAAU,6CAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,UACjB;AAAA,IACE;AAAA,MACE,eAAe,cAAU,kCAAW;AAAA,QAClC,QAAQ,QAAQ;AAAA,QAChB,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC,CAAC;AAAA,MACF,GAAG,QAAQ;AAAA,IACb;AAAA,IACA,qBAAqB,OAAO;AAAA,EAC9B;AASF,QAAM,uBAAuB,CAAC,eAA0C;AAAA,IACtE,UAAU,cAAc,SAAS;AAAA,IACjC,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,IACpC,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB;AAEA,QAAM,kBAAkB,CAAC,YAAmC;AAC1D,WAAO,IAAI,4BAA4B,SAAS;AAAA,MAC9C,GAAG,qBAAqB,MAAM;AAAA,MAC9B,gBAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CAAC,YAAmC,gBAAgB,OAAO;AAE5E,WAAS,uBAAuB;AAChC,WAAS,YAAY;AACrB,WAAS,gBAAgB;AAEzB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iCAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,WAAS,aAAa,CAAC,YAAoB;AACzC,UAAM,IAAI,iCAAiB,EAAE,SAAS,WAAW,aAAa,CAAC;AAAA,EACjE;AAEA,SAAO;AACT;AAEO,IAAM,aAAa,iBAAiB;","names":[]}

package/dist/index.mjs

@@ -0,0 +1,110 @@
+// src/moonshotai-provider.ts
+import {
+  NoSuchModelError
+} from "@ai-sdk/provider";
+import {
+  loadApiKey,
+  withoutTrailingSlash,
+  withUserAgentSuffix
+} from "@ai-sdk/provider-utils";
+import { z } from "zod/v4";
+
+// src/moonshotai-chat-language-model.ts
+import { OpenAICompatibleChatLanguageModel } from "@ai-sdk/openai-compatible";
+
+// src/convert-moonshotai-chat-usage.ts
+function convertMoonshotAIChatUsage(usage) {
+  var _a, _b, _c, _d, _e, _f, _g, _h;
+  if (usage == null) {
+    return {
+      inputTokens: void 0,
+      outputTokens: void 0,
+      totalTokens: void 0
+    };
+  }
+  const cachedInputTokens = (_c = (_b = usage.cached_tokens) != null ? _b : (_a = usage.prompt_tokens_details) == null ? void 0 : _a.cached_tokens) != null ? _c : void 0;
+  const reasoningTokens = (_e = (_d = usage.completion_tokens_details) == null ? void 0 : _d.reasoning_tokens) != null ? _e : void 0;
+  return {
+    inputTokens: (_f = usage.prompt_tokens) != null ? _f : void 0,
+    outputTokens: (_g = usage.completion_tokens) != null ? _g : void 0,
+    totalTokens: (_h = usage.total_tokens) != null ? _h : void 0,
+    reasoningTokens,
+    cachedInputTokens
+  };
+}
+
+// src/moonshotai-chat-language-model.ts
+var MoonshotAIChatLanguageModel = class extends OpenAICompatibleChatLanguageModel {
+  constructor(modelId, config) {
+    super(modelId, config);
+  }
+  async doGenerate(options) {
+    var _a, _b;
+    const result = await super.doGenerate(options);
+    const usage = (_b = (_a = result.response) == null ? void 0 : _a.body) == null ? void 0 : _b.usage;
+    return {
+      ...result,
+      usage: convertMoonshotAIChatUsage(usage)
+    };
+  }
+};
+
+// src/version.ts
+var VERSION = "0.0.1";
+
+// src/moonshotai-provider.ts
+var moonshotaiErrorSchema = z.object({
+  error: z.object({
+    message: z.string(),
+    type: z.string().nullish()
+  })
+});
+var moonshotaiErrorStructure = {
+  errorSchema: moonshotaiErrorSchema,
+  errorToMessage: (data) => data.error.message
+};
+var defaultBaseURL = "https://api.moonshot.ai/v1";
+function createMoonshotAI(options = {}) {
+  var _a;
+  const baseURL = withoutTrailingSlash((_a = options.baseURL) != null ? _a : defaultBaseURL);
+  const getHeaders = () => withUserAgentSuffix(
+    {
+      Authorization: `Bearer ${loadApiKey({
+        apiKey: options.apiKey,
+        environmentVariableName: "MOONSHOT_API_KEY",
+        description: "Moonshot API key"
+      })}`,
+      ...options.headers
+    },
+    `ai-sdk/moonshotai/${VERSION}`
+  );
+  const getCommonModelConfig = (modelType) => ({
+    provider: `moonshotai.${modelType}`,
+    url: ({ path }) => `${baseURL}${path}`,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
+  const createChatModel = (modelId) => {
+    return new MoonshotAIChatLanguageModel(modelId, {
+      ...getCommonModelConfig("chat"),
+      errorStructure: moonshotaiErrorStructure
+    });
+  };
+  const provider = (modelId) => createChatModel(modelId);
+  provider.specificationVersion = "v2";
+  provider.chatModel = createChatModel;
+  provider.languageModel = createChatModel;
+  provider.textEmbeddingModel = (modelId) => {
+    throw new NoSuchModelError({ modelId, modelType: "textEmbeddingModel" });
+  };
+  provider.imageModel = (modelId) => {
+    throw new NoSuchModelError({ modelId, modelType: "imageModel" });
+  };
+  return provider;
+}
+var moonshotai = createMoonshotAI();
+export {
+  createMoonshotAI,
+  moonshotai
+};
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/moonshotai-provider.ts","../src/moonshotai-chat-language-model.ts","../src/convert-moonshotai-chat-usage.ts","../src/version.ts"],"sourcesContent":["import { ProviderErrorStructure } from '@ai-sdk/openai-compatible';\nimport {\n  LanguageModelV2,\n  NoSuchModelError,\n  ProviderV2,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n  withUserAgentSuffix,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport { MoonshotAIChatLanguageModel } from './moonshotai-chat-language-model';\nimport { MoonshotAIChatModelId } from './moonshotai-chat-options';\nimport { VERSION } from './version';\n\nexport type MoonshotAIErrorData = z.infer<typeof moonshotaiErrorSchema>;\n\nconst moonshotaiErrorSchema = z.object({\n  error: z.object({\n    message: z.string(),\n    type: z.string().nullish(),\n  }),\n});\n\nconst moonshotaiErrorStructure: ProviderErrorStructure<MoonshotAIErrorData> = {\n  errorSchema: moonshotaiErrorSchema,\n  errorToMessage: data => data.error.message,\n};\n\nexport interface MoonshotAIProviderSettings {\n  /**\n   * Moonshot API key. Default value is taken from the `MOONSHOT_API_KEY`\n   * environment variable.\n   */\n  apiKey?: string;\n  /**\n   * Base URL for the API calls.\n   */\n  baseURL?: string;\n  /**\n   * Custom headers to include in the requests.\n   */\n  headers?: Record<string, string>;\n  /**\n   * Custom fetch implementation. You can use it as a middleware to intercept requests,\n   * or to provide a custom fetch implementation for e.g. testing.\n   */\n  fetch?: FetchFunction;\n}\n\nexport interface MoonshotAIProvider extends ProviderV2 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: MoonshotAIChatModelId): LanguageModelV2;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: MoonshotAIChatModelId): LanguageModelV2;\n\n  /**\n   * Creates a language model for text generation.\n   */\n  languageModel(modelId: MoonshotAIChatModelId): LanguageModelV2;\n}\n\nconst defaultBaseURL = 'https://api.moonshot.ai/v1';\n\nexport function createMoonshotAI(\n  options: MoonshotAIProviderSettings = {},\n): MoonshotAIProvider {\n  const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n  const getHeaders = () =>\n    withUserAgentSuffix(\n      {\n        Authorization: `Bearer ${loadApiKey({\n          apiKey: options.apiKey,\n          environmentVariableName: 'MOONSHOT_API_KEY',\n          description: 'Moonshot API key',\n        })}`,\n        ...options.headers,\n      },\n      `ai-sdk/moonshotai/${VERSION}`,\n    );\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const getCommonModelConfig = (modelType: string): CommonModelConfig => ({\n    provider: `moonshotai.${modelType}`,\n    url: ({ path }) => `${baseURL}${path}`,\n    headers: getHeaders,\n    fetch: options.fetch,\n  });\n\n  const createChatModel = (modelId: MoonshotAIChatModelId) => {\n    return new MoonshotAIChatLanguageModel(modelId, {\n      ...getCommonModelConfig('chat'),\n      errorStructure: moonshotaiErrorStructure,\n    });\n  };\n\n  const provider = (modelId: MoonshotAIChatModelId) => createChatModel(modelId);\n\n  provider.specificationVersion = 'v2' as const;\n  provider.chatModel = createChatModel;\n  provider.languageModel = createChatModel;\n\n  provider.textEmbeddingModel = (modelId: string) => {\n    throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });\n  };\n\n  provider.imageModel = (modelId: string) => {\n    throw new NoSuchModelError({ modelId, modelType: 'imageModel' });\n  };\n\n  return provider;\n}\n\nexport const moonshotai = createMoonshotAI();\n","import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';\nimport { OpenAICompatibleChatConfig } from '@ai-sdk/openai-compatible/internal';\nimport { LanguageModelV2 } from '@ai-sdk/provider';\nimport { convertMoonshotAIChatUsage } from './convert-moonshotai-chat-usage';\nimport { MoonshotAIChatModelId } from './moonshotai-chat-options';\n\nexport class MoonshotAIChatLanguageModel extends OpenAICompatibleChatLanguageModel {\n  constructor(\n    modelId: MoonshotAIChatModelId,\n    config: OpenAICompatibleChatConfig,\n  ) {\n    super(modelId, config);\n  }\n\n  async doGenerate(\n    options: Parameters<LanguageModelV2['doGenerate']>[0],\n  ): Promise<Awaited<ReturnType<LanguageModelV2['doGenerate']>>> {\n    const result = await super.doGenerate(options);\n\n    // @ts-expect-error accessing response body from parent result\n    const usage = result.response?.body?.usage;\n\n    return {\n      ...result,\n      usage: convertMoonshotAIChatUsage(usage),\n    };\n  }\n}\n","import { LanguageModelV2Usage } from '@ai-sdk/provider';\n\nexport function convertMoonshotAIChatUsage(\n  usage:\n    | {\n        prompt_tokens?: number | null;\n        completion_tokens?: number | null;\n        total_tokens?: number | null;\n        cached_tokens?: number | null;\n        prompt_tokens_details?: {\n          cached_tokens?: number | null;\n        } | null;\n        completion_tokens_details?: {\n          reasoning_tokens?: number | null;\n        } | null;\n      }\n    | undefined\n    | null,\n): LanguageModelV2Usage {\n  if (usage == null) {\n    return {\n      inputTokens: undefined,\n      outputTokens: undefined,\n      totalTokens: undefined,\n    };\n  }\n\n  const cachedInputTokens =\n    usage.cached_tokens ??\n    usage.prompt_tokens_details?.cached_tokens ??\n    undefined;\n\n  const reasoningTokens =\n    usage.completion_tokens_details?.reasoning_tokens ?? undefined;\n\n  return {\n    inputTokens: usage.prompt_tokens ?? undefined,\n    outputTokens: usage.completion_tokens ?? undefined,\n    totalTokens: usage.total_tokens ?? undefined,\n    reasoningTokens,\n    cachedInputTokens,\n  };\n}\n","declare const __PACKAGE_VERSION__: string;\n\nexport const VERSION = __PACKAGE_VERSION__;\n"],"mappings":";AACA;AAAA,EAEE;AAAA,OAEK;AACP;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,SAAS;;;ACZlB,SAAS,yCAAyC;;;ACE3C,SAAS,2BACd,OAesB;AAlBxB;AAmBE,MAAI,SAAS,MAAM;AACjB,WAAO;AAAA,MACL,aAAa;AAAA,MACb,cAAc;AAAA,MACd,aAAa;AAAA,IACf;AAAA,EACF;AAEA,QAAM,qBACJ,iBAAM,kBAAN,aACA,WAAM,0BAAN,mBAA6B,kBAD7B,YAEA;AAEF,QAAM,mBACJ,iBAAM,8BAAN,mBAAiC,qBAAjC,YAAqD;AAEvD,SAAO;AAAA,IACL,cAAa,WAAM,kBAAN,YAAuB;AAAA,IACpC,eAAc,WAAM,sBAAN,YAA2B;AAAA,IACzC,cAAa,WAAM,iBAAN,YAAsB;AAAA,IACnC;AAAA,IACA;AAAA,EACF;AACF;;;ADpCO,IAAM,8BAAN,cAA0C,kCAAkC;AAAA,EACjF,YACE,SACA,QACA;AACA,UAAM,SAAS,MAAM;AAAA,EACvB;AAAA,EAEA,MAAM,WACJ,SAC6D;AAhBjE;AAiBI,UAAM,SAAS,MAAM,MAAM,WAAW,OAAO;AAG7C,UAAM,SAAQ,kBAAO,aAAP,mBAAiB,SAAjB,mBAAuB;AAErC,WAAO;AAAA,MACL,GAAG;AAAA,MACH,OAAO,2BAA2B,KAAK;AAAA,IACzC;AAAA,EACF;AACF;;;AEzBO,IAAM,UAAU;;;AHiBvB,IAAM,wBAAwB,EAAE,OAAO;AAAA,EACrC,OAAO,EAAE,OAAO;AAAA,IACd,SAAS,EAAE,OAAO;AAAA,IAClB,MAAM,EAAE,OAAO,EAAE,QAAQ;AAAA,EAC3B,CAAC;AACH,CAAC;AAED,IAAM,2BAAwE;AAAA,EAC5E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK,MAAM;AACrC;AAwCA,IAAM,iBAAiB;AAEhB,SAAS,iBACd,UAAsC,CAAC,GACnB;AAzEtB;AA0EE,QAAM,UAAU,sBAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,MACjB;AAAA,IACE;AAAA,MACE,eAAe,UAAU,WAAW;AAAA,QAClC,QAAQ,QAAQ;AAAA,QAChB,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC,CAAC;AAAA,MACF,GAAG,QAAQ;AAAA,IACb;AAAA,IACA,qBAAqB,OAAO;AAAA,EAC9B;AASF,QAAM,uBAAuB,CAAC,eAA0C;AAAA,IACtE,UAAU,cAAc,SAAS;AAAA,IACjC,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,IACpC,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB;AAEA,QAAM,kBAAkB,CAAC,YAAmC;AAC1D,WAAO,IAAI,4BAA4B,SAAS;AAAA,MAC9C,GAAG,qBAAqB,MAAM;AAAA,MAC9B,gBAAgB;AAAA,IAClB,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CAAC,YAAmC,gBAAgB,OAAO;AAE5E,WAAS,uBAAuB;AAChC,WAAS,YAAY;AACrB,WAAS,gBAAgB;AAEzB,WAAS,qBAAqB,CAAC,YAAoB;AACjD,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,qBAAqB,CAAC;AAAA,EACzE;AAEA,WAAS,aAAa,CAAC,YAAoB;AACzC,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,aAAa,CAAC;AAAA,EACjE;AAEA,SAAO;AACT;AAEO,IAAM,aAAa,iBAAiB;","names":[]}

package/package.json

@@ -0,0 +1,73 @@
+{
+  "name": "@ai-sdk/moonshotai",
+  "version": "0.0.1",
+  "license": "Apache-2.0",
+  "sideEffects": false,
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist/**/*",
+    "src",
+    "!src/**/*.test.ts",
+    "!src/**/*.test-d.ts",
+    "!src/**/__snapshots__",
+    "!src/**/__fixtures__",
+    "CHANGELOG.md",
+    "README.md"
+  ],
+  "exports": {
+    "./package.json": "./package.json",
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    }
+  },
+  "dependencies": {
+    "@ai-sdk/provider": "2.0.1",
+    "@ai-sdk/provider-utils": "3.0.20",
+    "@ai-sdk/openai-compatible": "1.0.32"
+  },
+  "devDependencies": {
+    "@types/node": "20.17.24",
+    "tsup": "^8",
+    "typescript": "5.8.3",
+    "zod": "3.25.76",
+    "@ai-sdk/test-server": "0.0.2",
+    "@vercel/ai-tsconfig": "0.0.0"
+  },
+  "peerDependencies": {
+    "zod": "^3.25.76 || ^4.1.8"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "homepage": "https://ai-sdk.dev/docs",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/vercel/ai.git"
+  },
+  "bugs": {
+    "url": "https://github.com/vercel/ai/issues"
+  },
+  "keywords": [
+    "ai"
+  ],
+  "scripts": {
+    "build": "pnpm clean && tsup --tsconfig tsconfig.build.json",
+    "build:watch": "pnpm clean && tsup --watch",
+    "clean": "del-cli dist *.tsbuildinfo",
+    "lint": "eslint \"./**/*.ts*\"",
+    "type-check": "tsc --build",
+    "prettier-check": "prettier --check \"./**/*.ts*\"",
+    "test": "pnpm test:node && pnpm test:edge",
+    "test:update": "pnpm test:node -u",
+    "test:watch": "vitest --config vitest.node.config.js",
+    "test:edge": "vitest --config vitest.edge.config.js --run",
+    "test:node": "vitest --config vitest.node.config.js --run"
+  }
+}

package/src/convert-moonshotai-chat-usage.ts

@@ -0,0 +1,43 @@
+import { LanguageModelV2Usage } from '@ai-sdk/provider';
+
+export function convertMoonshotAIChatUsage(
+  usage:
+    | {
+        prompt_tokens?: number | null;
+        completion_tokens?: number | null;
+        total_tokens?: number | null;
+        cached_tokens?: number | null;
+        prompt_tokens_details?: {
+          cached_tokens?: number | null;
+        } | null;
+        completion_tokens_details?: {
+          reasoning_tokens?: number | null;
+        } | null;
+      }
+    | undefined
+    | null,
+): LanguageModelV2Usage {
+  if (usage == null) {
+    return {
+      inputTokens: undefined,
+      outputTokens: undefined,
+      totalTokens: undefined,
+    };
+  }
+
+  const cachedInputTokens =
+    usage.cached_tokens ??
+    usage.prompt_tokens_details?.cached_tokens ??
+    undefined;
+
+  const reasoningTokens =
+    usage.completion_tokens_details?.reasoning_tokens ?? undefined;
+
+  return {
+    inputTokens: usage.prompt_tokens ?? undefined,
+    outputTokens: usage.completion_tokens ?? undefined,
+    totalTokens: usage.total_tokens ?? undefined,
+    reasoningTokens,
+    cachedInputTokens,
+  };
+}

package/src/index.ts

@@ -0,0 +1,9 @@
+export { createMoonshotAI, moonshotai } from './moonshotai-provider';
+export type {
+  MoonshotAIProvider,
+  MoonshotAIProviderSettings,
+} from './moonshotai-provider';
+export type {
+  MoonshotAIChatModelId,
+  MoonshotAIProviderOptions,
+} from './moonshotai-chat-options';

package/src/moonshotai-chat-language-model.ts

@@ -0,0 +1,28 @@
+import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';
+import { OpenAICompatibleChatConfig } from '@ai-sdk/openai-compatible/internal';
+import { LanguageModelV2 } from '@ai-sdk/provider';
+import { convertMoonshotAIChatUsage } from './convert-moonshotai-chat-usage';
+import { MoonshotAIChatModelId } from './moonshotai-chat-options';
+
+export class MoonshotAIChatLanguageModel extends OpenAICompatibleChatLanguageModel {
+  constructor(
+    modelId: MoonshotAIChatModelId,
+    config: OpenAICompatibleChatConfig,
+  ) {
+    super(modelId, config);
+  }
+
+  async doGenerate(
+    options: Parameters<LanguageModelV2['doGenerate']>[0],
+  ): Promise<Awaited<ReturnType<LanguageModelV2['doGenerate']>>> {
+    const result = await super.doGenerate(options);
+
+    // @ts-expect-error accessing response body from parent result
+    const usage = result.response?.body?.usage;
+
+    return {
+      ...result,
+      usage: convertMoonshotAIChatUsage(usage),
+    };
+  }
+}

package/src/moonshotai-chat-options.ts

@@ -0,0 +1,28 @@
+import { z } from 'zod/v4';
+
+export type MoonshotAIChatModelId =
+  | 'moonshot-v1-8k'
+  | 'moonshot-v1-32k'
+  | 'moonshot-v1-128k'
+  | 'kimi-k2'
+  | 'kimi-k2-0905'
+  | 'kimi-k2-thinking'
+  | 'kimi-k2-thinking-turbo'
+  | 'kimi-k2-turbo'
+  | 'kimi-k2.5'
+  | (string & {});
+
+export const moonshotaiProviderOptions = z.object({
+  thinking: z
+    .object({
+      type: z.enum(['enabled', 'disabled']).optional(),
+      budgetTokens: z.number().int().min(1024).optional(),
+    })
+    .optional(),
+
+  reasoningHistory: z.enum(['disabled', 'interleaved', 'preserved']).optional(),
+});
+
+export type MoonshotAIProviderOptions = z.infer<
+  typeof moonshotaiProviderOptions
+>;

package/src/moonshotai-provider.ts

@@ -0,0 +1,127 @@
+import { ProviderErrorStructure } from '@ai-sdk/openai-compatible';
+import {
+  LanguageModelV2,
+  NoSuchModelError,
+  ProviderV2,
+} from '@ai-sdk/provider';
+import {
+  FetchFunction,
+  loadApiKey,
+  withoutTrailingSlash,
+  withUserAgentSuffix,
+} from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+import { MoonshotAIChatLanguageModel } from './moonshotai-chat-language-model';
+import { MoonshotAIChatModelId } from './moonshotai-chat-options';
+import { VERSION } from './version';
+
+export type MoonshotAIErrorData = z.infer<typeof moonshotaiErrorSchema>;
+
+const moonshotaiErrorSchema = z.object({
+  error: z.object({
+    message: z.string(),
+    type: z.string().nullish(),
+  }),
+});
+
+const moonshotaiErrorStructure: ProviderErrorStructure<MoonshotAIErrorData> = {
+  errorSchema: moonshotaiErrorSchema,
+  errorToMessage: data => data.error.message,
+};
+
+export interface MoonshotAIProviderSettings {
+  /**
+   * Moonshot API key. Default value is taken from the `MOONSHOT_API_KEY`
+   * environment variable.
+   */
+  apiKey?: string;
+  /**
+   * Base URL for the API calls.
+   */
+  baseURL?: string;
+  /**
+   * Custom headers to include in the requests.
+   */
+  headers?: Record<string, string>;
+  /**
+   * Custom fetch implementation. You can use it as a middleware to intercept requests,
+   * or to provide a custom fetch implementation for e.g. testing.
+   */
+  fetch?: FetchFunction;
+}
+
+export interface MoonshotAIProvider extends ProviderV2 {
+  /**
+   * Creates a model for text generation.
+   */
+  (modelId: MoonshotAIChatModelId): LanguageModelV2;
+
+  /**
+   * Creates a chat model for text generation.
+   */
+  chatModel(modelId: MoonshotAIChatModelId): LanguageModelV2;
+
+  /**
+   * Creates a language model for text generation.
+   */
+  languageModel(modelId: MoonshotAIChatModelId): LanguageModelV2;
+}
+
+const defaultBaseURL = 'https://api.moonshot.ai/v1';
+
+export function createMoonshotAI(
+  options: MoonshotAIProviderSettings = {},
+): MoonshotAIProvider {
+  const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);
+  const getHeaders = () =>
+    withUserAgentSuffix(
+      {
+        Authorization: `Bearer ${loadApiKey({
+          apiKey: options.apiKey,
+          environmentVariableName: 'MOONSHOT_API_KEY',
+          description: 'Moonshot API key',
+        })}`,
+        ...options.headers,
+      },
+      `ai-sdk/moonshotai/${VERSION}`,
+    );
+
+  interface CommonModelConfig {
+    provider: string;
+    url: ({ path }: { path: string }) => string;
+    headers: () => Record<string, string>;
+    fetch?: FetchFunction;
+  }
+
+  const getCommonModelConfig = (modelType: string): CommonModelConfig => ({
+    provider: `moonshotai.${modelType}`,
+    url: ({ path }) => `${baseURL}${path}`,
+    headers: getHeaders,
+    fetch: options.fetch,
+  });
+
+  const createChatModel = (modelId: MoonshotAIChatModelId) => {
+    return new MoonshotAIChatLanguageModel(modelId, {
+      ...getCommonModelConfig('chat'),
+      errorStructure: moonshotaiErrorStructure,
+    });
+  };
+
+  const provider = (modelId: MoonshotAIChatModelId) => createChatModel(modelId);
+
+  provider.specificationVersion = 'v2' as const;
+  provider.chatModel = createChatModel;
+  provider.languageModel = createChatModel;
+
+  provider.textEmbeddingModel = (modelId: string) => {
+    throw new NoSuchModelError({ modelId, modelType: 'textEmbeddingModel' });
+  };
+
+  provider.imageModel = (modelId: string) => {
+    throw new NoSuchModelError({ modelId, modelType: 'imageModel' });
+  };
+
+  return provider;
+}
+
+export const moonshotai = createMoonshotAI();

package/src/version.ts

@@ -0,0 +1,3 @@
+declare const __PACKAGE_VERSION__: string;
+
+export const VERSION = __PACKAGE_VERSION__;