@ai-sdk/moonshotai 0.0.0-b66d09a8-20260328011513

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/CHANGELOG.md ADDED
@@ -0,0 +1,212 @@
1
+ # @ai-sdk/moonshotai
2
+
3
+ ## 0.0.0-b66d09a8-20260328011513
4
+
5
+ ### Major Changes
6
+
7
+ - 8359612: Start v7 pre-release
8
+
9
+ ### Patch Changes
10
+
11
+ - 8f3e1da: chore(openai-compat): update v3 specs to v4
12
+ - Updated dependencies [74d520f]
13
+ - Updated dependencies [531251e]
14
+ - Updated dependencies [45b3d76]
15
+ - Updated dependencies [1f509d4]
16
+ - Updated dependencies [5c2a5a2]
17
+ - Updated dependencies [f7d4f01]
18
+ - Updated dependencies [776b617]
19
+ - Updated dependencies [8f3e1da]
20
+ - Updated dependencies [f7295cb]
21
+ - Updated dependencies [3887c70]
22
+ - Updated dependencies [61753c3]
23
+ - Updated dependencies [8359612]
24
+ - @ai-sdk/openai-compatible@0.0.0-b66d09a8-20260328011513
25
+ - @ai-sdk/provider-utils@0.0.0-b66d09a8-20260328011513
26
+ - @ai-sdk/provider@0.0.0-b66d09a8-20260328011513
27
+
28
+ ## 3.0.0-beta.10
29
+
30
+ ### Patch Changes
31
+
32
+ - Updated dependencies [1f509d4]
33
+ - @ai-sdk/provider-utils@5.0.0-beta.7
34
+ - @ai-sdk/provider@4.0.0-beta.5
35
+ - @ai-sdk/openai-compatible@3.0.0-beta.10
36
+
37
+ ## 3.0.0-beta.9
38
+
39
+ ### Patch Changes
40
+
41
+ - Updated dependencies [74d520f]
42
+ - @ai-sdk/openai-compatible@3.0.0-beta.9
43
+
44
+ ## 3.0.0-beta.8
45
+
46
+ ### Patch Changes
47
+
48
+ - Updated dependencies [3887c70]
49
+ - @ai-sdk/provider-utils@5.0.0-beta.6
50
+ - @ai-sdk/provider@4.0.0-beta.4
51
+ - @ai-sdk/openai-compatible@3.0.0-beta.8
52
+
53
+ ## 3.0.0-beta.7
54
+
55
+ ### Patch Changes
56
+
57
+ - Updated dependencies [776b617]
58
+ - @ai-sdk/provider-utils@5.0.0-beta.5
59
+ - @ai-sdk/provider@4.0.0-beta.3
60
+ - @ai-sdk/openai-compatible@3.0.0-beta.7
61
+
62
+ ## 3.0.0-beta.6
63
+
64
+ ### Patch Changes
65
+
66
+ - Updated dependencies [61753c3]
67
+ - @ai-sdk/provider-utils@5.0.0-beta.4
68
+ - @ai-sdk/openai-compatible@3.0.0-beta.6
69
+
70
+ ## 3.0.0-beta.5
71
+
72
+ ### Patch Changes
73
+
74
+ - Updated dependencies [f7d4f01]
75
+ - @ai-sdk/provider-utils@5.0.0-beta.3
76
+ - @ai-sdk/provider@4.0.0-beta.2
77
+ - @ai-sdk/openai-compatible@3.0.0-beta.5
78
+
79
+ ## 3.0.0-beta.4
80
+
81
+ ### Patch Changes
82
+
83
+ - Updated dependencies [5c2a5a2]
84
+ - @ai-sdk/provider@4.0.0-beta.1
85
+ - @ai-sdk/openai-compatible@3.0.0-beta.4
86
+ - @ai-sdk/provider-utils@5.0.0-beta.2
87
+
88
+ ## 3.0.0-beta.3
89
+
90
+ ### Patch Changes
91
+
92
+ - 8f3e1da: chore(openai-compat): update v3 specs to v4
93
+ - Updated dependencies [8f3e1da]
94
+ - @ai-sdk/openai-compatible@3.0.0-beta.3
95
+
96
+ ## 3.0.0-beta.2
97
+
98
+ ### Patch Changes
99
+
100
+ - Updated dependencies [45b3d76]
101
+ - Updated dependencies [f7295cb]
102
+ - @ai-sdk/openai-compatible@3.0.0-beta.2
103
+
104
+ ## 3.0.0-beta.1
105
+
106
+ ### Patch Changes
107
+
108
+ - Updated dependencies [531251e]
109
+ - @ai-sdk/provider-utils@5.0.0-beta.1
110
+ - @ai-sdk/openai-compatible@3.0.0-beta.1
111
+
112
+ ## 3.0.0-beta.0
113
+
114
+ ### Major Changes
115
+
116
+ - 8359612: Start v7 pre-release
117
+
118
+ ### Patch Changes
119
+
120
+ - Updated dependencies [8359612]
121
+ - @ai-sdk/openai-compatible@3.0.0-beta.0
122
+ - @ai-sdk/provider@4.0.0-beta.0
123
+ - @ai-sdk/provider-utils@5.0.0-beta.0
124
+
125
+ ## 2.0.10
126
+
127
+ ### Patch Changes
128
+
129
+ - Updated dependencies [ad4cfc2]
130
+ - @ai-sdk/provider-utils@4.0.19
131
+ - @ai-sdk/openai-compatible@2.0.35
132
+
133
+ ## 2.0.9
134
+
135
+ ### Patch Changes
136
+
137
+ - Updated dependencies [824b295]
138
+ - @ai-sdk/provider-utils@4.0.18
139
+ - @ai-sdk/openai-compatible@2.0.34
140
+
141
+ ## 2.0.8
142
+
143
+ ### Patch Changes
144
+
145
+ - Updated dependencies [89caf28]
146
+ - @ai-sdk/openai-compatible@2.0.33
147
+
148
+ ## 2.0.7
149
+
150
+ ### Patch Changes
151
+
152
+ - Updated dependencies [08336f1]
153
+ - @ai-sdk/provider-utils@4.0.17
154
+ - @ai-sdk/openai-compatible@2.0.32
155
+
156
+ ## 2.0.6
157
+
158
+ ### Patch Changes
159
+
160
+ - Updated dependencies [58bc42d]
161
+ - @ai-sdk/provider-utils@4.0.16
162
+ - @ai-sdk/openai-compatible@2.0.31
163
+
164
+ ## 2.0.5
165
+
166
+ ### Patch Changes
167
+
168
+ - Updated dependencies [4024a3a]
169
+ - @ai-sdk/provider-utils@4.0.15
170
+ - @ai-sdk/openai-compatible@2.0.30
171
+
172
+ ## 2.0.4
173
+
174
+ ### Patch Changes
175
+
176
+ - 99fbed8: feat: normalize provider specific model options type names and ensure they are exported
177
+ - Updated dependencies [99fbed8]
178
+ - @ai-sdk/openai-compatible@2.0.29
179
+
180
+ ## 2.0.3
181
+
182
+ ### Patch Changes
183
+
184
+ - d999bdf: fix (provider/moonshotai): include usage when streaming
185
+
186
+ ## 2.0.2
187
+
188
+ ### Patch Changes
189
+
190
+ - Updated dependencies [7168375]
191
+ - @ai-sdk/provider@3.0.8
192
+ - @ai-sdk/openai-compatible@2.0.28
193
+ - @ai-sdk/provider-utils@4.0.14
194
+
195
+ ## 2.0.1
196
+
197
+ ### Patch Changes
198
+
199
+ - Updated dependencies [9e490ad]
200
+ - @ai-sdk/openai-compatible@2.0.27
201
+
202
+ ## 2.0.0
203
+
204
+ ### Major Changes
205
+
206
+ - a57c1df: feat(provider): add Moonshot AI provider
207
+
208
+ ## 1.0.0
209
+
210
+ ### Major Changes
211
+
212
+ - 36268ff: feat(provider): add Moonshot AI provider
package/LICENSE ADDED
@@ -0,0 +1,13 @@
1
+ Copyright 2023 Vercel, Inc.
2
+
3
+ Licensed under the Apache License, Version 2.0 (the "License");
4
+ you may not use this file except in compliance with the License.
5
+ You may obtain a copy of the License at
6
+
7
+ http://www.apache.org/licenses/LICENSE-2.0
8
+
9
+ Unless required by applicable law or agreed to in writing, software
10
+ distributed under the License is distributed on an "AS IS" BASIS,
11
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ See the License for the specific language governing permissions and
13
+ limitations under the License.
package/README.md ADDED
@@ -0,0 +1,62 @@
1
+ # AI SDK - Moonshot AI Provider
2
+
3
+ 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.
4
+
5
+ ## Setup
6
+
7
+ The Moonshot AI provider is available in the `@ai-sdk/moonshotai` module. You can install it with
8
+
9
+ ```bash
10
+ npm i @ai-sdk/moonshotai
11
+ ```
12
+
13
+ ## Skill for Coding Agents
14
+
15
+ If you use coding agents such as Claude Code or Cursor, we highly recommend adding the AI SDK skill to your repository:
16
+
17
+ ```shell
18
+ npx skills add vercel/ai
19
+ ```
20
+
21
+ ## Provider Instance
22
+
23
+ You can import the default provider instance `moonshotai` from `@ai-sdk/moonshotai`:
24
+
25
+ ```ts
26
+ import { moonshotai } from '@ai-sdk/moonshotai';
27
+ ```
28
+
29
+ ## Language Model Example
30
+
31
+ ```ts
32
+ import { moonshotai } from '@ai-sdk/moonshotai';
33
+ import { generateText } from 'ai';
34
+
35
+ const { text } = await generateText({
36
+ model: moonshotai('kimi-k2.5'),
37
+ prompt: 'Write a JavaScript function that sorts a list:',
38
+ });
39
+ ```
40
+
41
+ ## Thinking Mode Example (Kimi K2 Thinking)
42
+
43
+ ```ts
44
+ import { moonshotai } from '@ai-sdk/moonshotai';
45
+ import { generateText } from 'ai';
46
+
47
+ const { text } = await generateText({
48
+ model: moonshotai('kimi-k2-thinking'),
49
+ prompt: 'Solve this problem step by step: What is 15% of 240?',
50
+ moonshotai: {
51
+ thinking: {
52
+ type: 'enabled',
53
+ budgetTokens: 2048,
54
+ },
55
+ reasoningHistory: 'interleaved',
56
+ },
57
+ });
58
+ ```
59
+
60
+ ## Documentation
61
+
62
+ Please check out the **[Moonshot AI provider](https://ai-sdk.dev/providers/ai-sdk-providers/moonshotai)** for more information.
@@ -0,0 +1,59 @@
1
+ import { ProviderV4, LanguageModelV4 } from '@ai-sdk/provider';
2
+ import { FetchFunction } from '@ai-sdk/provider-utils';
3
+ import { z } from 'zod/v4';
4
+
5
+ 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 & {});
6
+ declare const moonshotaiLanguageModelOptions: z.ZodObject<{
7
+ thinking: z.ZodOptional<z.ZodObject<{
8
+ type: z.ZodOptional<z.ZodEnum<{
9
+ enabled: "enabled";
10
+ disabled: "disabled";
11
+ }>>;
12
+ budgetTokens: z.ZodOptional<z.ZodNumber>;
13
+ }, z.core.$strip>>;
14
+ reasoningHistory: z.ZodOptional<z.ZodEnum<{
15
+ disabled: "disabled";
16
+ interleaved: "interleaved";
17
+ preserved: "preserved";
18
+ }>>;
19
+ }, z.core.$strip>;
20
+ type MoonshotAILanguageModelOptions = z.infer<typeof moonshotaiLanguageModelOptions>;
21
+
22
+ interface MoonshotAIProviderSettings {
23
+ /**
24
+ * Moonshot API key. Default value is taken from the `MOONSHOT_API_KEY`
25
+ * environment variable.
26
+ */
27
+ apiKey?: string;
28
+ /**
29
+ * Base URL for the API calls.
30
+ */
31
+ baseURL?: string;
32
+ /**
33
+ * Custom headers to include in the requests.
34
+ */
35
+ headers?: Record<string, string>;
36
+ /**
37
+ * Custom fetch implementation. You can use it as a middleware to intercept requests,
38
+ * or to provide a custom fetch implementation for e.g. testing.
39
+ */
40
+ fetch?: FetchFunction;
41
+ }
42
+ interface MoonshotAIProvider extends ProviderV4 {
43
+ /**
44
+ * Creates a model for text generation.
45
+ */
46
+ (modelId: MoonshotAIChatModelId): LanguageModelV4;
47
+ /**
48
+ * Creates a chat model for text generation.
49
+ */
50
+ chatModel(modelId: MoonshotAIChatModelId): LanguageModelV4;
51
+ /**
52
+ * Creates a language model for text generation.
53
+ */
54
+ languageModel(modelId: MoonshotAIChatModelId): LanguageModelV4;
55
+ }
56
+ declare function createMoonshotAI(options?: MoonshotAIProviderSettings): MoonshotAIProvider;
57
+ declare const moonshotai: MoonshotAIProvider;
58
+
59
+ export { type MoonshotAIChatModelId, type MoonshotAILanguageModelOptions, type MoonshotAIProvider, type MoonshotAILanguageModelOptions as MoonshotAIProviderOptions, type MoonshotAIProviderSettings, createMoonshotAI, moonshotai };
@@ -0,0 +1,59 @@
1
+ import { ProviderV4, LanguageModelV4 } from '@ai-sdk/provider';
2
+ import { FetchFunction } from '@ai-sdk/provider-utils';
3
+ import { z } from 'zod/v4';
4
+
5
+ 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 & {});
6
+ declare const moonshotaiLanguageModelOptions: z.ZodObject<{
7
+ thinking: z.ZodOptional<z.ZodObject<{
8
+ type: z.ZodOptional<z.ZodEnum<{
9
+ enabled: "enabled";
10
+ disabled: "disabled";
11
+ }>>;
12
+ budgetTokens: z.ZodOptional<z.ZodNumber>;
13
+ }, z.core.$strip>>;
14
+ reasoningHistory: z.ZodOptional<z.ZodEnum<{
15
+ disabled: "disabled";
16
+ interleaved: "interleaved";
17
+ preserved: "preserved";
18
+ }>>;
19
+ }, z.core.$strip>;
20
+ type MoonshotAILanguageModelOptions = z.infer<typeof moonshotaiLanguageModelOptions>;
21
+
22
+ interface MoonshotAIProviderSettings {
23
+ /**
24
+ * Moonshot API key. Default value is taken from the `MOONSHOT_API_KEY`
25
+ * environment variable.
26
+ */
27
+ apiKey?: string;
28
+ /**
29
+ * Base URL for the API calls.
30
+ */
31
+ baseURL?: string;
32
+ /**
33
+ * Custom headers to include in the requests.
34
+ */
35
+ headers?: Record<string, string>;
36
+ /**
37
+ * Custom fetch implementation. You can use it as a middleware to intercept requests,
38
+ * or to provide a custom fetch implementation for e.g. testing.
39
+ */
40
+ fetch?: FetchFunction;
41
+ }
42
+ interface MoonshotAIProvider extends ProviderV4 {
43
+ /**
44
+ * Creates a model for text generation.
45
+ */
46
+ (modelId: MoonshotAIChatModelId): LanguageModelV4;
47
+ /**
48
+ * Creates a chat model for text generation.
49
+ */
50
+ chatModel(modelId: MoonshotAIChatModelId): LanguageModelV4;
51
+ /**
52
+ * Creates a language model for text generation.
53
+ */
54
+ languageModel(modelId: MoonshotAIChatModelId): LanguageModelV4;
55
+ }
56
+ declare function createMoonshotAI(options?: MoonshotAIProviderSettings): MoonshotAIProvider;
57
+ declare const moonshotai: MoonshotAIProvider;
58
+
59
+ export { type MoonshotAIChatModelId, type MoonshotAILanguageModelOptions, type MoonshotAIProvider, type MoonshotAILanguageModelOptions as MoonshotAIProviderOptions, type MoonshotAIProviderSettings, createMoonshotAI, moonshotai };
package/dist/index.js ADDED
@@ -0,0 +1,190 @@
1
+ "use strict";
2
+ var __defProp = Object.defineProperty;
3
+ var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
4
+ var __getOwnPropNames = Object.getOwnPropertyNames;
5
+ var __hasOwnProp = Object.prototype.hasOwnProperty;
6
+ var __export = (target, all) => {
7
+ for (var name in all)
8
+ __defProp(target, name, { get: all[name], enumerable: true });
9
+ };
10
+ var __copyProps = (to, from, except, desc) => {
11
+ if (from && typeof from === "object" || typeof from === "function") {
12
+ for (let key of __getOwnPropNames(from))
13
+ if (!__hasOwnProp.call(to, key) && key !== except)
14
+ __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
15
+ }
16
+ return to;
17
+ };
18
+ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
19
+
20
+ // src/index.ts
21
+ var index_exports = {};
22
+ __export(index_exports, {
23
+ createMoonshotAI: () => createMoonshotAI,
24
+ moonshotai: () => moonshotai
25
+ });
26
+ module.exports = __toCommonJS(index_exports);
27
+
28
+ // src/moonshotai-provider.ts
29
+ var import_provider = require("@ai-sdk/provider");
30
+ var import_provider_utils = require("@ai-sdk/provider-utils");
31
+ var import_v4 = require("zod/v4");
32
+
33
+ // src/moonshotai-chat-language-model.ts
34
+ var import_openai_compatible = require("@ai-sdk/openai-compatible");
35
+
36
+ // src/convert-moonshotai-chat-usage.ts
37
+ function convertMoonshotAIChatUsage(usage) {
38
+ var _a, _b, _c, _d, _e, _f, _g;
39
+ if (usage == null) {
40
+ return {
41
+ inputTokens: {
42
+ total: void 0,
43
+ noCache: void 0,
44
+ cacheRead: void 0,
45
+ cacheWrite: void 0
46
+ },
47
+ outputTokens: {
48
+ total: void 0,
49
+ text: void 0,
50
+ reasoning: void 0
51
+ },
52
+ raw: void 0
53
+ };
54
+ }
55
+ const promptTokens = (_a = usage.prompt_tokens) != null ? _a : 0;
56
+ const completionTokens = (_b = usage.completion_tokens) != null ? _b : 0;
57
+ const cacheReadTokens = (_e = (_d = usage.cached_tokens) != null ? _d : (_c = usage.prompt_tokens_details) == null ? void 0 : _c.cached_tokens) != null ? _e : 0;
58
+ const reasoningTokens = (_g = (_f = usage.completion_tokens_details) == null ? void 0 : _f.reasoning_tokens) != null ? _g : 0;
59
+ return {
60
+ inputTokens: {
61
+ total: promptTokens,
62
+ noCache: promptTokens - cacheReadTokens,
63
+ cacheRead: cacheReadTokens,
64
+ cacheWrite: void 0
65
+ },
66
+ outputTokens: {
67
+ total: completionTokens,
68
+ text: completionTokens - reasoningTokens,
69
+ reasoning: reasoningTokens
70
+ },
71
+ raw: usage
72
+ };
73
+ }
74
+
75
+ // src/moonshotai-chat-language-model.ts
76
+ var MoonshotAIChatLanguageModel = class extends import_openai_compatible.OpenAICompatibleChatLanguageModel {
77
+ constructor(modelId, config) {
78
+ super(modelId, config);
79
+ }
80
+ async doGenerate(options) {
81
+ var _a, _b;
82
+ const result = await super.doGenerate(options);
83
+ const usage = (_b = (_a = result.response) == null ? void 0 : _a.body) == null ? void 0 : _b.usage;
84
+ return {
85
+ ...result,
86
+ usage: convertMoonshotAIChatUsage(usage)
87
+ };
88
+ }
89
+ async doStream(options) {
90
+ const result = await super.doStream(options);
91
+ return {
92
+ ...result,
93
+ stream: result.stream.pipeThrough(
94
+ new TransformStream({
95
+ transform(chunk, controller) {
96
+ if (chunk.type === "finish" && chunk.usage) {
97
+ controller.enqueue({
98
+ ...chunk,
99
+ usage: convertMoonshotAIChatUsage(chunk.usage.raw)
100
+ });
101
+ } else {
102
+ controller.enqueue(chunk);
103
+ }
104
+ }
105
+ })
106
+ )
107
+ };
108
+ }
109
+ };
110
+
111
+ // src/version.ts
112
+ var VERSION = "0.0.0-b66d09a8-20260328011513";
113
+
114
+ // src/moonshotai-provider.ts
115
+ var moonshotaiErrorSchema = import_v4.z.object({
116
+ error: import_v4.z.object({
117
+ message: import_v4.z.string(),
118
+ type: import_v4.z.string().nullish()
119
+ })
120
+ });
121
+ var moonshotaiErrorStructure = {
122
+ errorSchema: moonshotaiErrorSchema,
123
+ errorToMessage: (data) => data.error.message
124
+ };
125
+ var defaultBaseURL = "https://api.moonshot.ai/v1";
126
+ function createMoonshotAI(options = {}) {
127
+ var _a;
128
+ const baseURL = (0, import_provider_utils.withoutTrailingSlash)((_a = options.baseURL) != null ? _a : defaultBaseURL);
129
+ const getHeaders = () => (0, import_provider_utils.withUserAgentSuffix)(
130
+ {
131
+ Authorization: `Bearer ${(0, import_provider_utils.loadApiKey)({
132
+ apiKey: options.apiKey,
133
+ environmentVariableName: "MOONSHOT_API_KEY",
134
+ description: "Moonshot API key"
135
+ })}`,
136
+ ...options.headers
137
+ },
138
+ `ai-sdk/moonshotai/${VERSION}`
139
+ );
140
+ const getCommonModelConfig = (modelType) => ({
141
+ provider: `moonshotai.${modelType}`,
142
+ url: ({ path }) => `${baseURL}${path}`,
143
+ headers: getHeaders,
144
+ fetch: options.fetch
145
+ });
146
+ const createChatModel = (modelId) => {
147
+ return new MoonshotAIChatLanguageModel(modelId, {
148
+ ...getCommonModelConfig("chat"),
149
+ includeUsage: true,
150
+ errorStructure: moonshotaiErrorStructure,
151
+ transformRequestBody: (args) => {
152
+ const thinking = args.thinking;
153
+ const reasoningHistory = args.reasoningHistory;
154
+ const { thinking: _, reasoningHistory: __, ...rest } = args;
155
+ return {
156
+ ...rest,
157
+ ...thinking && {
158
+ thinking: {
159
+ type: thinking.type,
160
+ ...thinking.budgetTokens !== void 0 && {
161
+ budget_tokens: thinking.budgetTokens
162
+ }
163
+ }
164
+ },
165
+ ...reasoningHistory && {
166
+ reasoning_history: reasoningHistory
167
+ }
168
+ };
169
+ }
170
+ });
171
+ };
172
+ const provider = (modelId) => createChatModel(modelId);
173
+ provider.specificationVersion = "v4";
174
+ provider.chatModel = createChatModel;
175
+ provider.languageModel = createChatModel;
176
+ provider.embeddingModel = (modelId) => {
177
+ throw new import_provider.NoSuchModelError({ modelId, modelType: "embeddingModel" });
178
+ };
179
+ provider.imageModel = (modelId) => {
180
+ throw new import_provider.NoSuchModelError({ modelId, modelType: "imageModel" });
181
+ };
182
+ return provider;
183
+ }
184
+ var moonshotai = createMoonshotAI();
185
+ // Annotate the CommonJS export names for ESM import in node:
186
+ 0 && (module.exports = {
187
+ createMoonshotAI,
188
+ moonshotai
189
+ });
190
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
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 MoonshotAILanguageModelOptions,\n /** @deprecated Use `MoonshotAILanguageModelOptions` instead. */\n MoonshotAILanguageModelOptions as MoonshotAIProviderOptions,\n} from './moonshotai-chat-options';\n","import { ProviderErrorStructure } from '@ai-sdk/openai-compatible';\nimport {\n LanguageModelV4,\n NoSuchModelError,\n ProviderV4,\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 ProviderV4 {\n /**\n * Creates a model for text generation.\n */\n (modelId: MoonshotAIChatModelId): LanguageModelV4;\n\n /**\n * Creates a chat model for text generation.\n */\n chatModel(modelId: MoonshotAIChatModelId): LanguageModelV4;\n\n /**\n * Creates a language model for text generation.\n */\n languageModel(modelId: MoonshotAIChatModelId): LanguageModelV4;\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 includeUsage: true,\n errorStructure: moonshotaiErrorStructure,\n transformRequestBody: (args: Record<string, any>) => {\n const thinking = args.thinking as\n | { type?: string; budgetTokens?: number }\n | undefined;\n const reasoningHistory = args.reasoningHistory as string | undefined;\n\n const { thinking: _, reasoningHistory: __, ...rest } = args;\n\n return {\n ...rest,\n ...(thinking && {\n thinking: {\n type: thinking.type,\n ...(thinking.budgetTokens !== undefined && {\n budget_tokens: thinking.budgetTokens,\n }),\n },\n }),\n ...(reasoningHistory && {\n reasoning_history: reasoningHistory,\n }),\n };\n },\n });\n };\n\n const provider = (modelId: MoonshotAIChatModelId) => createChatModel(modelId);\n\n provider.specificationVersion = 'v4' as const;\n provider.chatModel = createChatModel;\n provider.languageModel = createChatModel;\n\n provider.embeddingModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'embeddingModel' });\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 {\n LanguageModelV4CallOptions,\n LanguageModelV4GenerateResult,\n LanguageModelV4StreamPart,\n LanguageModelV4StreamResult,\n} 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: LanguageModelV4CallOptions,\n ): Promise<LanguageModelV4GenerateResult> {\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 async doStream(\n options: LanguageModelV4CallOptions,\n ): Promise<LanguageModelV4StreamResult> {\n const result = await super.doStream(options);\n\n return {\n ...result,\n stream: result.stream.pipeThrough(\n new TransformStream<\n LanguageModelV4StreamPart,\n LanguageModelV4StreamPart\n >({\n transform(chunk, controller) {\n if (chunk.type === 'finish' && chunk.usage) {\n controller.enqueue({\n ...chunk,\n usage: convertMoonshotAIChatUsage(chunk.usage.raw as any),\n });\n } else {\n controller.enqueue(chunk);\n }\n },\n }),\n ),\n };\n }\n}\n","import { LanguageModelV4Usage } from '@ai-sdk/provider';\n\nexport function convertMoonshotAIChatUsage(\n usage:\n | {\n prompt_tokens?: number | null;\n completion_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): LanguageModelV4Usage {\n if (usage == null) {\n return {\n inputTokens: {\n total: undefined,\n noCache: undefined,\n cacheRead: undefined,\n cacheWrite: undefined,\n },\n outputTokens: {\n total: undefined,\n text: undefined,\n reasoning: undefined,\n },\n raw: undefined,\n };\n }\n\n const promptTokens = usage.prompt_tokens ?? 0;\n const completionTokens = usage.completion_tokens ?? 0;\n\n const cacheReadTokens =\n usage.cached_tokens ?? usage.prompt_tokens_details?.cached_tokens ?? 0;\n\n const reasoningTokens =\n usage.completion_tokens_details?.reasoning_tokens ?? 0;\n\n return {\n inputTokens: {\n total: promptTokens,\n noCache: promptTokens - cacheReadTokens,\n cacheRead: cacheReadTokens,\n cacheWrite: undefined,\n },\n outputTokens: {\n total: completionTokens,\n text: completionTokens - reasoningTokens,\n reasoning: reasoningTokens,\n },\n raw: usage,\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,OAcsB;AAjBxB;AAkBE,MAAI,SAAS,MAAM;AACjB,WAAO;AAAA,MACL,aAAa;AAAA,QACX,OAAO;AAAA,QACP,SAAS;AAAA,QACT,WAAW;AAAA,QACX,YAAY;AAAA,MACd;AAAA,MACA,cAAc;AAAA,QACZ,OAAO;AAAA,QACP,MAAM;AAAA,QACN,WAAW;AAAA,MACb;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAEA,QAAM,gBAAe,WAAM,kBAAN,YAAuB;AAC5C,QAAM,oBAAmB,WAAM,sBAAN,YAA2B;AAEpD,QAAM,mBACJ,iBAAM,kBAAN,aAAuB,WAAM,0BAAN,mBAA6B,kBAApD,YAAqE;AAEvE,QAAM,mBACJ,iBAAM,8BAAN,mBAAiC,qBAAjC,YAAqD;AAEvD,SAAO;AAAA,IACL,aAAa;AAAA,MACX,OAAO;AAAA,MACP,SAAS,eAAe;AAAA,MACxB,WAAW;AAAA,MACX,YAAY;AAAA,IACd;AAAA,IACA,cAAc;AAAA,MACZ,OAAO;AAAA,MACP,MAAM,mBAAmB;AAAA,MACzB,WAAW;AAAA,IACb;AAAA,IACA,KAAK;AAAA,EACP;AACF;;;AD/CO,IAAM,8BAAN,cAA0C,2DAAkC;AAAA,EACjF,YACE,SACA,QACA;AACA,UAAM,SAAS,MAAM;AAAA,EACvB;AAAA,EAEA,MAAM,WACJ,SACwC;AArB5C;AAsBI,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;AAAA,EAEA,MAAM,SACJ,SACsC;AACtC,UAAM,SAAS,MAAM,MAAM,SAAS,OAAO;AAE3C,WAAO;AAAA,MACL,GAAG;AAAA,MACH,QAAQ,OAAO,OAAO;AAAA,QACpB,IAAI,gBAGF;AAAA,UACA,UAAU,OAAO,YAAY;AAC3B,gBAAI,MAAM,SAAS,YAAY,MAAM,OAAO;AAC1C,yBAAW,QAAQ;AAAA,gBACjB,GAAG;AAAA,gBACH,OAAO,2BAA2B,MAAM,MAAM,GAAU;AAAA,cAC1D,CAAC;AAAA,YACH,OAAO;AACL,yBAAW,QAAQ,KAAK;AAAA,YAC1B;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AACF;;;AEzDO,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,cAAc;AAAA,MACd,gBAAgB;AAAA,MAChB,sBAAsB,CAAC,SAA8B;AACnD,cAAM,WAAW,KAAK;AAGtB,cAAM,mBAAmB,KAAK;AAE9B,cAAM,EAAE,UAAU,GAAG,kBAAkB,IAAI,GAAG,KAAK,IAAI;AAEvD,eAAO;AAAA,UACL,GAAG;AAAA,UACH,GAAI,YAAY;AAAA,YACd,UAAU;AAAA,cACR,MAAM,SAAS;AAAA,cACf,GAAI,SAAS,iBAAiB,UAAa;AAAA,gBACzC,eAAe,SAAS;AAAA,cAC1B;AAAA,YACF;AAAA,UACF;AAAA,UACA,GAAI,oBAAoB;AAAA,YACtB,mBAAmB;AAAA,UACrB;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CAAC,YAAmC,gBAAgB,OAAO;AAE5E,WAAS,uBAAuB;AAChC,WAAS,YAAY;AACrB,WAAS,gBAAgB;AAEzB,WAAS,iBAAiB,CAAC,YAAoB;AAC7C,UAAM,IAAI,iCAAiB,EAAE,SAAS,WAAW,iBAAiB,CAAC;AAAA,EACrE;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 ADDED
@@ -0,0 +1,168 @@
1
+ // src/moonshotai-provider.ts
2
+ import {
3
+ NoSuchModelError
4
+ } from "@ai-sdk/provider";
5
+ import {
6
+ loadApiKey,
7
+ withoutTrailingSlash,
8
+ withUserAgentSuffix
9
+ } from "@ai-sdk/provider-utils";
10
+ import { z } from "zod/v4";
11
+
12
+ // src/moonshotai-chat-language-model.ts
13
+ import { OpenAICompatibleChatLanguageModel } from "@ai-sdk/openai-compatible";
14
+
15
+ // src/convert-moonshotai-chat-usage.ts
16
+ function convertMoonshotAIChatUsage(usage) {
17
+ var _a, _b, _c, _d, _e, _f, _g;
18
+ if (usage == null) {
19
+ return {
20
+ inputTokens: {
21
+ total: void 0,
22
+ noCache: void 0,
23
+ cacheRead: void 0,
24
+ cacheWrite: void 0
25
+ },
26
+ outputTokens: {
27
+ total: void 0,
28
+ text: void 0,
29
+ reasoning: void 0
30
+ },
31
+ raw: void 0
32
+ };
33
+ }
34
+ const promptTokens = (_a = usage.prompt_tokens) != null ? _a : 0;
35
+ const completionTokens = (_b = usage.completion_tokens) != null ? _b : 0;
36
+ const cacheReadTokens = (_e = (_d = usage.cached_tokens) != null ? _d : (_c = usage.prompt_tokens_details) == null ? void 0 : _c.cached_tokens) != null ? _e : 0;
37
+ const reasoningTokens = (_g = (_f = usage.completion_tokens_details) == null ? void 0 : _f.reasoning_tokens) != null ? _g : 0;
38
+ return {
39
+ inputTokens: {
40
+ total: promptTokens,
41
+ noCache: promptTokens - cacheReadTokens,
42
+ cacheRead: cacheReadTokens,
43
+ cacheWrite: void 0
44
+ },
45
+ outputTokens: {
46
+ total: completionTokens,
47
+ text: completionTokens - reasoningTokens,
48
+ reasoning: reasoningTokens
49
+ },
50
+ raw: usage
51
+ };
52
+ }
53
+
54
+ // src/moonshotai-chat-language-model.ts
55
+ var MoonshotAIChatLanguageModel = class extends OpenAICompatibleChatLanguageModel {
56
+ constructor(modelId, config) {
57
+ super(modelId, config);
58
+ }
59
+ async doGenerate(options) {
60
+ var _a, _b;
61
+ const result = await super.doGenerate(options);
62
+ const usage = (_b = (_a = result.response) == null ? void 0 : _a.body) == null ? void 0 : _b.usage;
63
+ return {
64
+ ...result,
65
+ usage: convertMoonshotAIChatUsage(usage)
66
+ };
67
+ }
68
+ async doStream(options) {
69
+ const result = await super.doStream(options);
70
+ return {
71
+ ...result,
72
+ stream: result.stream.pipeThrough(
73
+ new TransformStream({
74
+ transform(chunk, controller) {
75
+ if (chunk.type === "finish" && chunk.usage) {
76
+ controller.enqueue({
77
+ ...chunk,
78
+ usage: convertMoonshotAIChatUsage(chunk.usage.raw)
79
+ });
80
+ } else {
81
+ controller.enqueue(chunk);
82
+ }
83
+ }
84
+ })
85
+ )
86
+ };
87
+ }
88
+ };
89
+
90
+ // src/version.ts
91
+ var VERSION = "0.0.0-b66d09a8-20260328011513";
92
+
93
+ // src/moonshotai-provider.ts
94
+ var moonshotaiErrorSchema = z.object({
95
+ error: z.object({
96
+ message: z.string(),
97
+ type: z.string().nullish()
98
+ })
99
+ });
100
+ var moonshotaiErrorStructure = {
101
+ errorSchema: moonshotaiErrorSchema,
102
+ errorToMessage: (data) => data.error.message
103
+ };
104
+ var defaultBaseURL = "https://api.moonshot.ai/v1";
105
+ function createMoonshotAI(options = {}) {
106
+ var _a;
107
+ const baseURL = withoutTrailingSlash((_a = options.baseURL) != null ? _a : defaultBaseURL);
108
+ const getHeaders = () => withUserAgentSuffix(
109
+ {
110
+ Authorization: `Bearer ${loadApiKey({
111
+ apiKey: options.apiKey,
112
+ environmentVariableName: "MOONSHOT_API_KEY",
113
+ description: "Moonshot API key"
114
+ })}`,
115
+ ...options.headers
116
+ },
117
+ `ai-sdk/moonshotai/${VERSION}`
118
+ );
119
+ const getCommonModelConfig = (modelType) => ({
120
+ provider: `moonshotai.${modelType}`,
121
+ url: ({ path }) => `${baseURL}${path}`,
122
+ headers: getHeaders,
123
+ fetch: options.fetch
124
+ });
125
+ const createChatModel = (modelId) => {
126
+ return new MoonshotAIChatLanguageModel(modelId, {
127
+ ...getCommonModelConfig("chat"),
128
+ includeUsage: true,
129
+ errorStructure: moonshotaiErrorStructure,
130
+ transformRequestBody: (args) => {
131
+ const thinking = args.thinking;
132
+ const reasoningHistory = args.reasoningHistory;
133
+ const { thinking: _, reasoningHistory: __, ...rest } = args;
134
+ return {
135
+ ...rest,
136
+ ...thinking && {
137
+ thinking: {
138
+ type: thinking.type,
139
+ ...thinking.budgetTokens !== void 0 && {
140
+ budget_tokens: thinking.budgetTokens
141
+ }
142
+ }
143
+ },
144
+ ...reasoningHistory && {
145
+ reasoning_history: reasoningHistory
146
+ }
147
+ };
148
+ }
149
+ });
150
+ };
151
+ const provider = (modelId) => createChatModel(modelId);
152
+ provider.specificationVersion = "v4";
153
+ provider.chatModel = createChatModel;
154
+ provider.languageModel = createChatModel;
155
+ provider.embeddingModel = (modelId) => {
156
+ throw new NoSuchModelError({ modelId, modelType: "embeddingModel" });
157
+ };
158
+ provider.imageModel = (modelId) => {
159
+ throw new NoSuchModelError({ modelId, modelType: "imageModel" });
160
+ };
161
+ return provider;
162
+ }
163
+ var moonshotai = createMoonshotAI();
164
+ export {
165
+ createMoonshotAI,
166
+ moonshotai
167
+ };
168
+ //# sourceMappingURL=index.mjs.map
@@ -0,0 +1 @@
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 LanguageModelV4,\n NoSuchModelError,\n ProviderV4,\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 ProviderV4 {\n /**\n * Creates a model for text generation.\n */\n (modelId: MoonshotAIChatModelId): LanguageModelV4;\n\n /**\n * Creates a chat model for text generation.\n */\n chatModel(modelId: MoonshotAIChatModelId): LanguageModelV4;\n\n /**\n * Creates a language model for text generation.\n */\n languageModel(modelId: MoonshotAIChatModelId): LanguageModelV4;\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 includeUsage: true,\n errorStructure: moonshotaiErrorStructure,\n transformRequestBody: (args: Record<string, any>) => {\n const thinking = args.thinking as\n | { type?: string; budgetTokens?: number }\n | undefined;\n const reasoningHistory = args.reasoningHistory as string | undefined;\n\n const { thinking: _, reasoningHistory: __, ...rest } = args;\n\n return {\n ...rest,\n ...(thinking && {\n thinking: {\n type: thinking.type,\n ...(thinking.budgetTokens !== undefined && {\n budget_tokens: thinking.budgetTokens,\n }),\n },\n }),\n ...(reasoningHistory && {\n reasoning_history: reasoningHistory,\n }),\n };\n },\n });\n };\n\n const provider = (modelId: MoonshotAIChatModelId) => createChatModel(modelId);\n\n provider.specificationVersion = 'v4' as const;\n provider.chatModel = createChatModel;\n provider.languageModel = createChatModel;\n\n provider.embeddingModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'embeddingModel' });\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 {\n LanguageModelV4CallOptions,\n LanguageModelV4GenerateResult,\n LanguageModelV4StreamPart,\n LanguageModelV4StreamResult,\n} 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: LanguageModelV4CallOptions,\n ): Promise<LanguageModelV4GenerateResult> {\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 async doStream(\n options: LanguageModelV4CallOptions,\n ): Promise<LanguageModelV4StreamResult> {\n const result = await super.doStream(options);\n\n return {\n ...result,\n stream: result.stream.pipeThrough(\n new TransformStream<\n LanguageModelV4StreamPart,\n LanguageModelV4StreamPart\n >({\n transform(chunk, controller) {\n if (chunk.type === 'finish' && chunk.usage) {\n controller.enqueue({\n ...chunk,\n usage: convertMoonshotAIChatUsage(chunk.usage.raw as any),\n });\n } else {\n controller.enqueue(chunk);\n }\n },\n }),\n ),\n };\n }\n}\n","import { LanguageModelV4Usage } from '@ai-sdk/provider';\n\nexport function convertMoonshotAIChatUsage(\n usage:\n | {\n prompt_tokens?: number | null;\n completion_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): LanguageModelV4Usage {\n if (usage == null) {\n return {\n inputTokens: {\n total: undefined,\n noCache: undefined,\n cacheRead: undefined,\n cacheWrite: undefined,\n },\n outputTokens: {\n total: undefined,\n text: undefined,\n reasoning: undefined,\n },\n raw: undefined,\n };\n }\n\n const promptTokens = usage.prompt_tokens ?? 0;\n const completionTokens = usage.completion_tokens ?? 0;\n\n const cacheReadTokens =\n usage.cached_tokens ?? usage.prompt_tokens_details?.cached_tokens ?? 0;\n\n const reasoningTokens =\n usage.completion_tokens_details?.reasoning_tokens ?? 0;\n\n return {\n inputTokens: {\n total: promptTokens,\n noCache: promptTokens - cacheReadTokens,\n cacheRead: cacheReadTokens,\n cacheWrite: undefined,\n },\n outputTokens: {\n total: completionTokens,\n text: completionTokens - reasoningTokens,\n reasoning: reasoningTokens,\n },\n raw: usage,\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,OAcsB;AAjBxB;AAkBE,MAAI,SAAS,MAAM;AACjB,WAAO;AAAA,MACL,aAAa;AAAA,QACX,OAAO;AAAA,QACP,SAAS;AAAA,QACT,WAAW;AAAA,QACX,YAAY;AAAA,MACd;AAAA,MACA,cAAc;AAAA,QACZ,OAAO;AAAA,QACP,MAAM;AAAA,QACN,WAAW;AAAA,MACb;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAEA,QAAM,gBAAe,WAAM,kBAAN,YAAuB;AAC5C,QAAM,oBAAmB,WAAM,sBAAN,YAA2B;AAEpD,QAAM,mBACJ,iBAAM,kBAAN,aAAuB,WAAM,0BAAN,mBAA6B,kBAApD,YAAqE;AAEvE,QAAM,mBACJ,iBAAM,8BAAN,mBAAiC,qBAAjC,YAAqD;AAEvD,SAAO;AAAA,IACL,aAAa;AAAA,MACX,OAAO;AAAA,MACP,SAAS,eAAe;AAAA,MACxB,WAAW;AAAA,MACX,YAAY;AAAA,IACd;AAAA,IACA,cAAc;AAAA,MACZ,OAAO;AAAA,MACP,MAAM,mBAAmB;AAAA,MACzB,WAAW;AAAA,IACb;AAAA,IACA,KAAK;AAAA,EACP;AACF;;;AD/CO,IAAM,8BAAN,cAA0C,kCAAkC;AAAA,EACjF,YACE,SACA,QACA;AACA,UAAM,SAAS,MAAM;AAAA,EACvB;AAAA,EAEA,MAAM,WACJ,SACwC;AArB5C;AAsBI,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;AAAA,EAEA,MAAM,SACJ,SACsC;AACtC,UAAM,SAAS,MAAM,MAAM,SAAS,OAAO;AAE3C,WAAO;AAAA,MACL,GAAG;AAAA,MACH,QAAQ,OAAO,OAAO;AAAA,QACpB,IAAI,gBAGF;AAAA,UACA,UAAU,OAAO,YAAY;AAC3B,gBAAI,MAAM,SAAS,YAAY,MAAM,OAAO;AAC1C,yBAAW,QAAQ;AAAA,gBACjB,GAAG;AAAA,gBACH,OAAO,2BAA2B,MAAM,MAAM,GAAU;AAAA,cAC1D,CAAC;AAAA,YACH,OAAO;AACL,yBAAW,QAAQ,KAAK;AAAA,YAC1B;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AACF;;;AEzDO,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,cAAc;AAAA,MACd,gBAAgB;AAAA,MAChB,sBAAsB,CAAC,SAA8B;AACnD,cAAM,WAAW,KAAK;AAGtB,cAAM,mBAAmB,KAAK;AAE9B,cAAM,EAAE,UAAU,GAAG,kBAAkB,IAAI,GAAG,KAAK,IAAI;AAEvD,eAAO;AAAA,UACL,GAAG;AAAA,UACH,GAAI,YAAY;AAAA,YACd,UAAU;AAAA,cACR,MAAM,SAAS;AAAA,cACf,GAAI,SAAS,iBAAiB,UAAa;AAAA,gBACzC,eAAe,SAAS;AAAA,cAC1B;AAAA,YACF;AAAA,UACF;AAAA,UACA,GAAI,oBAAoB;AAAA,YACtB,mBAAmB;AAAA,UACrB;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,WAAW,CAAC,YAAmC,gBAAgB,OAAO;AAE5E,WAAS,uBAAuB;AAChC,WAAS,YAAY;AACrB,WAAS,gBAAgB;AAEzB,WAAS,iBAAiB,CAAC,YAAoB;AAC7C,UAAM,IAAI,iBAAiB,EAAE,SAAS,WAAW,iBAAiB,CAAC;AAAA,EACrE;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 ADDED
@@ -0,0 +1,71 @@
1
+ {
2
+ "name": "@ai-sdk/moonshotai",
3
+ "version": "0.0.0-b66d09a8-20260328011513",
4
+ "license": "Apache-2.0",
5
+ "sideEffects": false,
6
+ "main": "./dist/index.js",
7
+ "module": "./dist/index.mjs",
8
+ "types": "./dist/index.d.ts",
9
+ "files": [
10
+ "dist/**/*",
11
+ "src",
12
+ "!src/**/*.test.ts",
13
+ "!src/**/*.test-d.ts",
14
+ "!src/**/__snapshots__",
15
+ "!src/**/__fixtures__",
16
+ "CHANGELOG.md",
17
+ "README.md"
18
+ ],
19
+ "exports": {
20
+ "./package.json": "./package.json",
21
+ ".": {
22
+ "types": "./dist/index.d.ts",
23
+ "import": "./dist/index.mjs",
24
+ "require": "./dist/index.js"
25
+ }
26
+ },
27
+ "dependencies": {
28
+ "@ai-sdk/openai-compatible": "0.0.0-b66d09a8-20260328011513",
29
+ "@ai-sdk/provider": "0.0.0-b66d09a8-20260328011513",
30
+ "@ai-sdk/provider-utils": "0.0.0-b66d09a8-20260328011513"
31
+ },
32
+ "devDependencies": {
33
+ "@types/node": "20.17.24",
34
+ "tsup": "^8",
35
+ "typescript": "5.8.3",
36
+ "zod": "3.25.76",
37
+ "@ai-sdk/test-server": "0.0.0-b66d09a8-20260328011513",
38
+ "@vercel/ai-tsconfig": "0.0.0"
39
+ },
40
+ "peerDependencies": {
41
+ "zod": "^3.25.76 || ^4.1.8"
42
+ },
43
+ "engines": {
44
+ "node": ">=18"
45
+ },
46
+ "publishConfig": {
47
+ "access": "public"
48
+ },
49
+ "homepage": "https://ai-sdk.dev/docs",
50
+ "repository": {
51
+ "type": "git",
52
+ "url": "git+https://github.com/vercel/ai.git"
53
+ },
54
+ "bugs": {
55
+ "url": "https://github.com/vercel/ai/issues"
56
+ },
57
+ "keywords": [
58
+ "ai"
59
+ ],
60
+ "scripts": {
61
+ "build": "pnpm clean && tsup --tsconfig tsconfig.build.json",
62
+ "build:watch": "pnpm clean && tsup --watch",
63
+ "clean": "del-cli dist *.tsbuildinfo",
64
+ "type-check": "tsc --build",
65
+ "test": "pnpm test:node && pnpm test:edge",
66
+ "test:update": "pnpm test:node -u",
67
+ "test:watch": "vitest --config vitest.node.config.js",
68
+ "test:edge": "vitest --config vitest.edge.config.js --run",
69
+ "test:node": "vitest --config vitest.node.config.js --run"
70
+ }
71
+ }
@@ -0,0 +1,59 @@
1
+ import { LanguageModelV4Usage } from '@ai-sdk/provider';
2
+
3
+ export function convertMoonshotAIChatUsage(
4
+ usage:
5
+ | {
6
+ prompt_tokens?: number | null;
7
+ completion_tokens?: number | null;
8
+ cached_tokens?: number | null;
9
+ prompt_tokens_details?: {
10
+ cached_tokens?: number | null;
11
+ } | null;
12
+ completion_tokens_details?: {
13
+ reasoning_tokens?: number | null;
14
+ } | null;
15
+ }
16
+ | undefined
17
+ | null,
18
+ ): LanguageModelV4Usage {
19
+ if (usage == null) {
20
+ return {
21
+ inputTokens: {
22
+ total: undefined,
23
+ noCache: undefined,
24
+ cacheRead: undefined,
25
+ cacheWrite: undefined,
26
+ },
27
+ outputTokens: {
28
+ total: undefined,
29
+ text: undefined,
30
+ reasoning: undefined,
31
+ },
32
+ raw: undefined,
33
+ };
34
+ }
35
+
36
+ const promptTokens = usage.prompt_tokens ?? 0;
37
+ const completionTokens = usage.completion_tokens ?? 0;
38
+
39
+ const cacheReadTokens =
40
+ usage.cached_tokens ?? usage.prompt_tokens_details?.cached_tokens ?? 0;
41
+
42
+ const reasoningTokens =
43
+ usage.completion_tokens_details?.reasoning_tokens ?? 0;
44
+
45
+ return {
46
+ inputTokens: {
47
+ total: promptTokens,
48
+ noCache: promptTokens - cacheReadTokens,
49
+ cacheRead: cacheReadTokens,
50
+ cacheWrite: undefined,
51
+ },
52
+ outputTokens: {
53
+ total: completionTokens,
54
+ text: completionTokens - reasoningTokens,
55
+ reasoning: reasoningTokens,
56
+ },
57
+ raw: usage,
58
+ };
59
+ }
package/src/index.ts ADDED
@@ -0,0 +1,11 @@
1
+ export { createMoonshotAI, moonshotai } from './moonshotai-provider';
2
+ export type {
3
+ MoonshotAIProvider,
4
+ MoonshotAIProviderSettings,
5
+ } from './moonshotai-provider';
6
+ export type {
7
+ MoonshotAIChatModelId,
8
+ MoonshotAILanguageModelOptions,
9
+ /** @deprecated Use `MoonshotAILanguageModelOptions` instead. */
10
+ MoonshotAILanguageModelOptions as MoonshotAIProviderOptions,
11
+ } from './moonshotai-chat-options';
@@ -0,0 +1,60 @@
1
+ import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';
2
+ import { OpenAICompatibleChatConfig } from '@ai-sdk/openai-compatible/internal';
3
+ import {
4
+ LanguageModelV4CallOptions,
5
+ LanguageModelV4GenerateResult,
6
+ LanguageModelV4StreamPart,
7
+ LanguageModelV4StreamResult,
8
+ } from '@ai-sdk/provider';
9
+ import { convertMoonshotAIChatUsage } from './convert-moonshotai-chat-usage';
10
+ import { MoonshotAIChatModelId } from './moonshotai-chat-options';
11
+
12
+ export class MoonshotAIChatLanguageModel extends OpenAICompatibleChatLanguageModel {
13
+ constructor(
14
+ modelId: MoonshotAIChatModelId,
15
+ config: OpenAICompatibleChatConfig,
16
+ ) {
17
+ super(modelId, config);
18
+ }
19
+
20
+ async doGenerate(
21
+ options: LanguageModelV4CallOptions,
22
+ ): Promise<LanguageModelV4GenerateResult> {
23
+ const result = await super.doGenerate(options);
24
+
25
+ // @ts-expect-error accessing response body from parent result
26
+ const usage = result.response?.body?.usage;
27
+
28
+ return {
29
+ ...result,
30
+ usage: convertMoonshotAIChatUsage(usage),
31
+ };
32
+ }
33
+
34
+ async doStream(
35
+ options: LanguageModelV4CallOptions,
36
+ ): Promise<LanguageModelV4StreamResult> {
37
+ const result = await super.doStream(options);
38
+
39
+ return {
40
+ ...result,
41
+ stream: result.stream.pipeThrough(
42
+ new TransformStream<
43
+ LanguageModelV4StreamPart,
44
+ LanguageModelV4StreamPart
45
+ >({
46
+ transform(chunk, controller) {
47
+ if (chunk.type === 'finish' && chunk.usage) {
48
+ controller.enqueue({
49
+ ...chunk,
50
+ usage: convertMoonshotAIChatUsage(chunk.usage.raw as any),
51
+ });
52
+ } else {
53
+ controller.enqueue(chunk);
54
+ }
55
+ },
56
+ }),
57
+ ),
58
+ };
59
+ }
60
+ }
@@ -0,0 +1,28 @@
1
+ import { z } from 'zod/v4';
2
+
3
+ export type MoonshotAIChatModelId =
4
+ | 'moonshot-v1-8k'
5
+ | 'moonshot-v1-32k'
6
+ | 'moonshot-v1-128k'
7
+ | 'kimi-k2'
8
+ | 'kimi-k2-0905'
9
+ | 'kimi-k2-thinking'
10
+ | 'kimi-k2-thinking-turbo'
11
+ | 'kimi-k2-turbo'
12
+ | 'kimi-k2.5'
13
+ | (string & {});
14
+
15
+ export const moonshotaiLanguageModelOptions = z.object({
16
+ thinking: z
17
+ .object({
18
+ type: z.enum(['enabled', 'disabled']).optional(),
19
+ budgetTokens: z.number().int().min(1024).optional(),
20
+ })
21
+ .optional(),
22
+
23
+ reasoningHistory: z.enum(['disabled', 'interleaved', 'preserved']).optional(),
24
+ });
25
+
26
+ export type MoonshotAILanguageModelOptions = z.infer<
27
+ typeof moonshotaiLanguageModelOptions
28
+ >;
@@ -0,0 +1,151 @@
1
+ import { ProviderErrorStructure } from '@ai-sdk/openai-compatible';
2
+ import {
3
+ LanguageModelV4,
4
+ NoSuchModelError,
5
+ ProviderV4,
6
+ } from '@ai-sdk/provider';
7
+ import {
8
+ FetchFunction,
9
+ loadApiKey,
10
+ withoutTrailingSlash,
11
+ withUserAgentSuffix,
12
+ } from '@ai-sdk/provider-utils';
13
+ import { z } from 'zod/v4';
14
+ import { MoonshotAIChatLanguageModel } from './moonshotai-chat-language-model';
15
+ import { MoonshotAIChatModelId } from './moonshotai-chat-options';
16
+ import { VERSION } from './version';
17
+
18
+ export type MoonshotAIErrorData = z.infer<typeof moonshotaiErrorSchema>;
19
+
20
+ const moonshotaiErrorSchema = z.object({
21
+ error: z.object({
22
+ message: z.string(),
23
+ type: z.string().nullish(),
24
+ }),
25
+ });
26
+
27
+ const moonshotaiErrorStructure: ProviderErrorStructure<MoonshotAIErrorData> = {
28
+ errorSchema: moonshotaiErrorSchema,
29
+ errorToMessage: data => data.error.message,
30
+ };
31
+
32
+ export interface MoonshotAIProviderSettings {
33
+ /**
34
+ * Moonshot API key. Default value is taken from the `MOONSHOT_API_KEY`
35
+ * environment variable.
36
+ */
37
+ apiKey?: string;
38
+ /**
39
+ * Base URL for the API calls.
40
+ */
41
+ baseURL?: string;
42
+ /**
43
+ * Custom headers to include in the requests.
44
+ */
45
+ headers?: Record<string, string>;
46
+ /**
47
+ * Custom fetch implementation. You can use it as a middleware to intercept requests,
48
+ * or to provide a custom fetch implementation for e.g. testing.
49
+ */
50
+ fetch?: FetchFunction;
51
+ }
52
+
53
+ export interface MoonshotAIProvider extends ProviderV4 {
54
+ /**
55
+ * Creates a model for text generation.
56
+ */
57
+ (modelId: MoonshotAIChatModelId): LanguageModelV4;
58
+
59
+ /**
60
+ * Creates a chat model for text generation.
61
+ */
62
+ chatModel(modelId: MoonshotAIChatModelId): LanguageModelV4;
63
+
64
+ /**
65
+ * Creates a language model for text generation.
66
+ */
67
+ languageModel(modelId: MoonshotAIChatModelId): LanguageModelV4;
68
+ }
69
+
70
+ const defaultBaseURL = 'https://api.moonshot.ai/v1';
71
+
72
+ export function createMoonshotAI(
73
+ options: MoonshotAIProviderSettings = {},
74
+ ): MoonshotAIProvider {
75
+ const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);
76
+ const getHeaders = () =>
77
+ withUserAgentSuffix(
78
+ {
79
+ Authorization: `Bearer ${loadApiKey({
80
+ apiKey: options.apiKey,
81
+ environmentVariableName: 'MOONSHOT_API_KEY',
82
+ description: 'Moonshot API key',
83
+ })}`,
84
+ ...options.headers,
85
+ },
86
+ `ai-sdk/moonshotai/${VERSION}`,
87
+ );
88
+
89
+ interface CommonModelConfig {
90
+ provider: string;
91
+ url: ({ path }: { path: string }) => string;
92
+ headers: () => Record<string, string>;
93
+ fetch?: FetchFunction;
94
+ }
95
+
96
+ const getCommonModelConfig = (modelType: string): CommonModelConfig => ({
97
+ provider: `moonshotai.${modelType}`,
98
+ url: ({ path }) => `${baseURL}${path}`,
99
+ headers: getHeaders,
100
+ fetch: options.fetch,
101
+ });
102
+
103
+ const createChatModel = (modelId: MoonshotAIChatModelId) => {
104
+ return new MoonshotAIChatLanguageModel(modelId, {
105
+ ...getCommonModelConfig('chat'),
106
+ includeUsage: true,
107
+ errorStructure: moonshotaiErrorStructure,
108
+ transformRequestBody: (args: Record<string, any>) => {
109
+ const thinking = args.thinking as
110
+ | { type?: string; budgetTokens?: number }
111
+ | undefined;
112
+ const reasoningHistory = args.reasoningHistory as string | undefined;
113
+
114
+ const { thinking: _, reasoningHistory: __, ...rest } = args;
115
+
116
+ return {
117
+ ...rest,
118
+ ...(thinking && {
119
+ thinking: {
120
+ type: thinking.type,
121
+ ...(thinking.budgetTokens !== undefined && {
122
+ budget_tokens: thinking.budgetTokens,
123
+ }),
124
+ },
125
+ }),
126
+ ...(reasoningHistory && {
127
+ reasoning_history: reasoningHistory,
128
+ }),
129
+ };
130
+ },
131
+ });
132
+ };
133
+
134
+ const provider = (modelId: MoonshotAIChatModelId) => createChatModel(modelId);
135
+
136
+ provider.specificationVersion = 'v4' as const;
137
+ provider.chatModel = createChatModel;
138
+ provider.languageModel = createChatModel;
139
+
140
+ provider.embeddingModel = (modelId: string) => {
141
+ throw new NoSuchModelError({ modelId, modelType: 'embeddingModel' });
142
+ };
143
+
144
+ provider.imageModel = (modelId: string) => {
145
+ throw new NoSuchModelError({ modelId, modelType: 'imageModel' });
146
+ };
147
+
148
+ return provider;
149
+ }
150
+
151
+ export const moonshotai = createMoonshotAI();
package/src/version.ts ADDED
@@ -0,0 +1,3 @@
1
+ declare const __PACKAGE_VERSION__: string;
2
+
3
+ export const VERSION = __PACKAGE_VERSION__;