@ai-sdk/moonshotai 3.0.0-beta.9 → 3.0.0-canary.38
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 +277 -0
- package/README.md +2 -0
- package/dist/index.d.ts +3 -1
- package/dist/index.js +38 -45
- package/dist/index.js.map +1 -1
- package/package.json +12 -10
- package/src/convert-moonshotai-chat-usage.ts +1 -1
- package/src/index.ts +1 -0
- package/src/moonshotai-chat-language-model.ts +21 -2
- package/src/moonshotai-provider.ts +5 -5
- package/dist/index.d.mts +0 -59
- package/dist/index.mjs +0 -168
- package/dist/index.mjs.map +0 -1
package/CHANGELOG.md
CHANGED
|
@@ -1,5 +1,282 @@
|
|
|
1
1
|
# @ai-sdk/moonshotai
|
|
2
2
|
|
|
3
|
+
## 3.0.0-canary.38
|
|
4
|
+
|
|
5
|
+
### Patch Changes
|
|
6
|
+
|
|
7
|
+
- Updated dependencies [cd9c311]
|
|
8
|
+
- @ai-sdk/openai-compatible@3.0.0-canary.38
|
|
9
|
+
|
|
10
|
+
## 3.0.0-canary.37
|
|
11
|
+
|
|
12
|
+
### Patch Changes
|
|
13
|
+
|
|
14
|
+
- 0c4c275: trigger initial canary release
|
|
15
|
+
- Updated dependencies [0c4c275]
|
|
16
|
+
- @ai-sdk/openai-compatible@3.0.0-canary.37
|
|
17
|
+
- @ai-sdk/provider-utils@5.0.0-canary.31
|
|
18
|
+
- @ai-sdk/provider@4.0.0-canary.15
|
|
19
|
+
|
|
20
|
+
## 3.0.0-beta.36
|
|
21
|
+
|
|
22
|
+
### Patch Changes
|
|
23
|
+
|
|
24
|
+
- Updated dependencies [e59c955]
|
|
25
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.36
|
|
26
|
+
|
|
27
|
+
## 3.0.0-beta.35
|
|
28
|
+
|
|
29
|
+
### Major Changes
|
|
30
|
+
|
|
31
|
+
- 04e9009: chore: make provider implementations code patterns more consistent, including renaming certain exported symbols
|
|
32
|
+
|
|
33
|
+
For all externally exported symbols that were renamed, the old names continue to work via deprecated aliases.
|
|
34
|
+
|
|
35
|
+
### Patch Changes
|
|
36
|
+
|
|
37
|
+
- Updated dependencies [08d2129]
|
|
38
|
+
- Updated dependencies [04e9009]
|
|
39
|
+
- @ai-sdk/provider-utils@5.0.0-beta.30
|
|
40
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.35
|
|
41
|
+
|
|
42
|
+
## 3.0.0-beta.34
|
|
43
|
+
|
|
44
|
+
### Patch Changes
|
|
45
|
+
|
|
46
|
+
- 258c093: chore: ensure consistent import handling and avoid import duplicates or cycles
|
|
47
|
+
- Updated dependencies [9bd6512]
|
|
48
|
+
- Updated dependencies [258c093]
|
|
49
|
+
- Updated dependencies [b6783da]
|
|
50
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.34
|
|
51
|
+
- @ai-sdk/provider-utils@5.0.0-beta.29
|
|
52
|
+
- @ai-sdk/provider@4.0.0-beta.14
|
|
53
|
+
|
|
54
|
+
## 3.0.0-beta.33
|
|
55
|
+
|
|
56
|
+
### Patch Changes
|
|
57
|
+
|
|
58
|
+
- 9f0e36c: trigger release for all packages after provenance setup
|
|
59
|
+
- Updated dependencies [9f0e36c]
|
|
60
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.33
|
|
61
|
+
- @ai-sdk/provider@4.0.0-beta.13
|
|
62
|
+
- @ai-sdk/provider-utils@5.0.0-beta.28
|
|
63
|
+
|
|
64
|
+
## 3.0.0-beta.32
|
|
65
|
+
|
|
66
|
+
### Patch Changes
|
|
67
|
+
|
|
68
|
+
- Updated dependencies [ab81968]
|
|
69
|
+
- Updated dependencies [785fe16]
|
|
70
|
+
- Updated dependencies [67df0a0]
|
|
71
|
+
- Updated dependencies [befb78c]
|
|
72
|
+
- Updated dependencies [0458559]
|
|
73
|
+
- Updated dependencies [58a2ad7]
|
|
74
|
+
- Updated dependencies [5852c0a]
|
|
75
|
+
- Updated dependencies [fc92055]
|
|
76
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.32
|
|
77
|
+
- @ai-sdk/provider-utils@5.0.0-beta.27
|
|
78
|
+
|
|
79
|
+
## 3.0.0-beta.31
|
|
80
|
+
|
|
81
|
+
### Patch Changes
|
|
82
|
+
|
|
83
|
+
- Updated dependencies [2e98477]
|
|
84
|
+
- Updated dependencies [bfb756d]
|
|
85
|
+
- @ai-sdk/provider-utils@5.0.0-beta.26
|
|
86
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.31
|
|
87
|
+
|
|
88
|
+
## 3.0.0-beta.30
|
|
89
|
+
|
|
90
|
+
### Patch Changes
|
|
91
|
+
|
|
92
|
+
- Updated dependencies [eea8d98]
|
|
93
|
+
- @ai-sdk/provider-utils@5.0.0-beta.25
|
|
94
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.30
|
|
95
|
+
|
|
96
|
+
## 3.0.0-beta.29
|
|
97
|
+
|
|
98
|
+
### Patch Changes
|
|
99
|
+
|
|
100
|
+
- Updated dependencies [f807e45]
|
|
101
|
+
- @ai-sdk/provider-utils@5.0.0-beta.24
|
|
102
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.29
|
|
103
|
+
|
|
104
|
+
## 3.0.0-beta.28
|
|
105
|
+
|
|
106
|
+
### Patch Changes
|
|
107
|
+
|
|
108
|
+
- Updated dependencies [350ea38]
|
|
109
|
+
- @ai-sdk/provider-utils@5.0.0-beta.23
|
|
110
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.28
|
|
111
|
+
|
|
112
|
+
## 3.0.0-beta.27
|
|
113
|
+
|
|
114
|
+
### Patch Changes
|
|
115
|
+
|
|
116
|
+
- Updated dependencies [083947b]
|
|
117
|
+
- @ai-sdk/provider-utils@5.0.0-beta.22
|
|
118
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.27
|
|
119
|
+
|
|
120
|
+
## 3.0.0-beta.26
|
|
121
|
+
|
|
122
|
+
### Patch Changes
|
|
123
|
+
|
|
124
|
+
- Updated dependencies [add1126]
|
|
125
|
+
- @ai-sdk/provider-utils@5.0.0-beta.21
|
|
126
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.26
|
|
127
|
+
|
|
128
|
+
## 3.0.0-beta.25
|
|
129
|
+
|
|
130
|
+
### Patch Changes
|
|
131
|
+
|
|
132
|
+
- b3976a2: Add workflow serialization support to all provider models.
|
|
133
|
+
|
|
134
|
+
**`@ai-sdk/provider-utils`:** New `serializeModel()` helper that extracts only serializable properties from a model instance, filtering out functions and objects containing functions. Third-party provider authors can use this to add workflow support to their own models.
|
|
135
|
+
|
|
136
|
+
**All providers:** `headers` is now optional in provider config types. This is non-breaking — existing code that passes `headers` continues to work. Custom provider implementations that construct model configs manually can now omit `headers`, which is useful when models are deserialized from a workflow step boundary where auth is provided separately.
|
|
137
|
+
|
|
138
|
+
All provider model classes now include `WORKFLOW_SERIALIZE` and `WORKFLOW_DESERIALIZE` static methods, enabling them to cross workflow step boundaries without serialization errors.
|
|
139
|
+
|
|
140
|
+
- Updated dependencies [b3976a2]
|
|
141
|
+
- Updated dependencies [ff5eba1]
|
|
142
|
+
- @ai-sdk/provider-utils@5.0.0-beta.20
|
|
143
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.25
|
|
144
|
+
- @ai-sdk/provider@4.0.0-beta.12
|
|
145
|
+
|
|
146
|
+
## 3.0.0-beta.24
|
|
147
|
+
|
|
148
|
+
### Major Changes
|
|
149
|
+
|
|
150
|
+
- ef992f8: Remove CommonJS exports from all packages. All packages are now ESM-only (`"type": "module"`). Consumers using `require()` must switch to ESM `import` syntax.
|
|
151
|
+
|
|
152
|
+
### Patch Changes
|
|
153
|
+
|
|
154
|
+
- Updated dependencies [ef992f8]
|
|
155
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.24
|
|
156
|
+
- @ai-sdk/provider@4.0.0-beta.11
|
|
157
|
+
- @ai-sdk/provider-utils@5.0.0-beta.19
|
|
158
|
+
|
|
159
|
+
## 3.0.0-beta.23
|
|
160
|
+
|
|
161
|
+
### Patch Changes
|
|
162
|
+
|
|
163
|
+
- 90e2d8a: chore: fix unused vars not being flagged by our lint tooling
|
|
164
|
+
- Updated dependencies [90e2d8a]
|
|
165
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.23
|
|
166
|
+
- @ai-sdk/provider-utils@5.0.0-beta.18
|
|
167
|
+
|
|
168
|
+
## 3.0.0-beta.22
|
|
169
|
+
|
|
170
|
+
### Patch Changes
|
|
171
|
+
|
|
172
|
+
- Updated dependencies [3ae1786]
|
|
173
|
+
- @ai-sdk/provider-utils@5.0.0-beta.17
|
|
174
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.22
|
|
175
|
+
|
|
176
|
+
## 3.0.0-beta.21
|
|
177
|
+
|
|
178
|
+
### Patch Changes
|
|
179
|
+
|
|
180
|
+
- Updated dependencies [176466a]
|
|
181
|
+
- @ai-sdk/provider@4.0.0-beta.10
|
|
182
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.21
|
|
183
|
+
- @ai-sdk/provider-utils@5.0.0-beta.16
|
|
184
|
+
|
|
185
|
+
## 3.0.0-beta.20
|
|
186
|
+
|
|
187
|
+
### Patch Changes
|
|
188
|
+
|
|
189
|
+
- Updated dependencies [e311194]
|
|
190
|
+
- @ai-sdk/provider@4.0.0-beta.9
|
|
191
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.20
|
|
192
|
+
- @ai-sdk/provider-utils@5.0.0-beta.15
|
|
193
|
+
|
|
194
|
+
## 3.0.0-beta.19
|
|
195
|
+
|
|
196
|
+
### Patch Changes
|
|
197
|
+
|
|
198
|
+
- Updated dependencies [34bd95d]
|
|
199
|
+
- Updated dependencies [008271d]
|
|
200
|
+
- @ai-sdk/provider@4.0.0-beta.8
|
|
201
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.19
|
|
202
|
+
- @ai-sdk/provider-utils@5.0.0-beta.14
|
|
203
|
+
|
|
204
|
+
## 3.0.0-beta.18
|
|
205
|
+
|
|
206
|
+
### Patch Changes
|
|
207
|
+
|
|
208
|
+
- Updated dependencies [b0c2869]
|
|
209
|
+
- Updated dependencies [7e26e81]
|
|
210
|
+
- @ai-sdk/provider-utils@5.0.0-beta.13
|
|
211
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.18
|
|
212
|
+
|
|
213
|
+
## 3.0.0-beta.17
|
|
214
|
+
|
|
215
|
+
### Patch Changes
|
|
216
|
+
|
|
217
|
+
- Updated dependencies [816ff67]
|
|
218
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.17
|
|
219
|
+
|
|
220
|
+
## 3.0.0-beta.16
|
|
221
|
+
|
|
222
|
+
### Patch Changes
|
|
223
|
+
|
|
224
|
+
- Updated dependencies [46d1149]
|
|
225
|
+
- @ai-sdk/provider-utils@5.0.0-beta.12
|
|
226
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.16
|
|
227
|
+
|
|
228
|
+
## 3.0.0-beta.15
|
|
229
|
+
|
|
230
|
+
### Patch Changes
|
|
231
|
+
|
|
232
|
+
- Updated dependencies [6fd51c0]
|
|
233
|
+
- @ai-sdk/provider-utils@5.0.0-beta.11
|
|
234
|
+
- @ai-sdk/provider@4.0.0-beta.7
|
|
235
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.15
|
|
236
|
+
|
|
237
|
+
## 3.0.0-beta.14
|
|
238
|
+
|
|
239
|
+
### Patch Changes
|
|
240
|
+
|
|
241
|
+
- Updated dependencies [c29a26f]
|
|
242
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.14
|
|
243
|
+
- @ai-sdk/provider-utils@5.0.0-beta.10
|
|
244
|
+
- @ai-sdk/provider@4.0.0-beta.6
|
|
245
|
+
|
|
246
|
+
## 3.0.0-beta.13
|
|
247
|
+
|
|
248
|
+
### Patch Changes
|
|
249
|
+
|
|
250
|
+
- 38fc777: Add AI Gateway hint to provider READMEs
|
|
251
|
+
- Updated dependencies [38fc777]
|
|
252
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.13
|
|
253
|
+
|
|
254
|
+
## 3.0.0-beta.12
|
|
255
|
+
|
|
256
|
+
### Patch Changes
|
|
257
|
+
|
|
258
|
+
- Updated dependencies [2e17091]
|
|
259
|
+
- @ai-sdk/provider-utils@5.0.0-beta.9
|
|
260
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.12
|
|
261
|
+
|
|
262
|
+
## 3.0.0-beta.11
|
|
263
|
+
|
|
264
|
+
### Patch Changes
|
|
265
|
+
|
|
266
|
+
- Updated dependencies [986c6fd]
|
|
267
|
+
- Updated dependencies [493295c]
|
|
268
|
+
- @ai-sdk/provider-utils@5.0.0-beta.8
|
|
269
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.11
|
|
270
|
+
|
|
271
|
+
## 3.0.0-beta.10
|
|
272
|
+
|
|
273
|
+
### Patch Changes
|
|
274
|
+
|
|
275
|
+
- Updated dependencies [1f509d4]
|
|
276
|
+
- @ai-sdk/provider-utils@5.0.0-beta.7
|
|
277
|
+
- @ai-sdk/provider@4.0.0-beta.5
|
|
278
|
+
- @ai-sdk/openai-compatible@3.0.0-beta.10
|
|
279
|
+
|
|
3
280
|
## 3.0.0-beta.9
|
|
4
281
|
|
|
5
282
|
### Patch Changes
|
package/README.md
CHANGED
|
@@ -2,6 +2,8 @@
|
|
|
2
2
|
|
|
3
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
4
|
|
|
5
|
+
> **Deploying to Vercel?** With Vercel's AI Gateway you can access Moonshot AI (and hundreds of models from other providers) — no additional packages, API keys, or extra cost. [Get started with AI Gateway](https://vercel.com/ai-gateway).
|
|
6
|
+
|
|
5
7
|
## Setup
|
|
6
8
|
|
|
7
9
|
The Moonshot AI provider is available in the `@ai-sdk/moonshotai` module. You can install it with
|
package/dist/index.d.ts
CHANGED
|
@@ -56,4 +56,6 @@ interface MoonshotAIProvider extends ProviderV4 {
|
|
|
56
56
|
declare function createMoonshotAI(options?: MoonshotAIProviderSettings): MoonshotAIProvider;
|
|
57
57
|
declare const moonshotai: MoonshotAIProvider;
|
|
58
58
|
|
|
59
|
-
|
|
59
|
+
declare const VERSION: string;
|
|
60
|
+
|
|
61
|
+
export { type MoonshotAIChatModelId, type MoonshotAILanguageModelOptions, type MoonshotAIProvider, type MoonshotAILanguageModelOptions as MoonshotAIProviderOptions, type MoonshotAIProviderSettings, VERSION, createMoonshotAI, moonshotai };
|
package/dist/index.js
CHANGED
|
@@ -1,37 +1,21 @@
|
|
|
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
1
|
// src/moonshotai-provider.ts
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
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";
|
|
32
11
|
|
|
33
12
|
// src/moonshotai-chat-language-model.ts
|
|
34
|
-
|
|
13
|
+
import { OpenAICompatibleChatLanguageModel } from "@ai-sdk/openai-compatible";
|
|
14
|
+
import {
|
|
15
|
+
serializeModelOptions,
|
|
16
|
+
WORKFLOW_SERIALIZE,
|
|
17
|
+
WORKFLOW_DESERIALIZE
|
|
18
|
+
} from "@ai-sdk/provider-utils";
|
|
35
19
|
|
|
36
20
|
// src/convert-moonshotai-chat-usage.ts
|
|
37
21
|
function convertMoonshotAIChatUsage(usage) {
|
|
@@ -73,7 +57,16 @@ function convertMoonshotAIChatUsage(usage) {
|
|
|
73
57
|
}
|
|
74
58
|
|
|
75
59
|
// src/moonshotai-chat-language-model.ts
|
|
76
|
-
var MoonshotAIChatLanguageModel = class extends
|
|
60
|
+
var MoonshotAIChatLanguageModel = class _MoonshotAIChatLanguageModel extends OpenAICompatibleChatLanguageModel {
|
|
61
|
+
static [WORKFLOW_SERIALIZE](model) {
|
|
62
|
+
return serializeModelOptions({
|
|
63
|
+
modelId: model.modelId,
|
|
64
|
+
config: model.config
|
|
65
|
+
});
|
|
66
|
+
}
|
|
67
|
+
static [WORKFLOW_DESERIALIZE](options) {
|
|
68
|
+
return new _MoonshotAIChatLanguageModel(options.modelId, options.config);
|
|
69
|
+
}
|
|
77
70
|
constructor(modelId, config) {
|
|
78
71
|
super(modelId, config);
|
|
79
72
|
}
|
|
@@ -109,13 +102,13 @@ var MoonshotAIChatLanguageModel = class extends import_openai_compatible.OpenAIC
|
|
|
109
102
|
};
|
|
110
103
|
|
|
111
104
|
// src/version.ts
|
|
112
|
-
var VERSION = "3.0.0-
|
|
105
|
+
var VERSION = "3.0.0-canary.38";
|
|
113
106
|
|
|
114
107
|
// src/moonshotai-provider.ts
|
|
115
|
-
var moonshotaiErrorSchema =
|
|
116
|
-
error:
|
|
117
|
-
message:
|
|
118
|
-
type:
|
|
108
|
+
var moonshotaiErrorSchema = z.object({
|
|
109
|
+
error: z.object({
|
|
110
|
+
message: z.string(),
|
|
111
|
+
type: z.string().nullish()
|
|
119
112
|
})
|
|
120
113
|
});
|
|
121
114
|
var moonshotaiErrorStructure = {
|
|
@@ -125,10 +118,10 @@ var moonshotaiErrorStructure = {
|
|
|
125
118
|
var defaultBaseURL = "https://api.moonshot.ai/v1";
|
|
126
119
|
function createMoonshotAI(options = {}) {
|
|
127
120
|
var _a;
|
|
128
|
-
const baseURL =
|
|
129
|
-
const getHeaders = () =>
|
|
121
|
+
const baseURL = withoutTrailingSlash((_a = options.baseURL) != null ? _a : defaultBaseURL);
|
|
122
|
+
const getHeaders = () => withUserAgentSuffix(
|
|
130
123
|
{
|
|
131
|
-
Authorization: `Bearer ${
|
|
124
|
+
Authorization: `Bearer ${loadApiKey({
|
|
132
125
|
apiKey: options.apiKey,
|
|
133
126
|
environmentVariableName: "MOONSHOT_API_KEY",
|
|
134
127
|
description: "Moonshot API key"
|
|
@@ -174,17 +167,17 @@ function createMoonshotAI(options = {}) {
|
|
|
174
167
|
provider.chatModel = createChatModel;
|
|
175
168
|
provider.languageModel = createChatModel;
|
|
176
169
|
provider.embeddingModel = (modelId) => {
|
|
177
|
-
throw new
|
|
170
|
+
throw new NoSuchModelError({ modelId, modelType: "embeddingModel" });
|
|
178
171
|
};
|
|
179
172
|
provider.imageModel = (modelId) => {
|
|
180
|
-
throw new
|
|
173
|
+
throw new NoSuchModelError({ modelId, modelType: "imageModel" });
|
|
181
174
|
};
|
|
182
175
|
return provider;
|
|
183
176
|
}
|
|
184
177
|
var moonshotai = createMoonshotAI();
|
|
185
|
-
|
|
186
|
-
|
|
178
|
+
export {
|
|
179
|
+
VERSION,
|
|
187
180
|
createMoonshotAI,
|
|
188
181
|
moonshotai
|
|
189
|
-
}
|
|
182
|
+
};
|
|
190
183
|
//# sourceMappingURL=index.js.map
|
package/dist/index.js.map
CHANGED
|
@@ -1 +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":[]}
|
|
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 type { ProviderErrorStructure } from '@ai-sdk/openai-compatible';\nimport {\n NoSuchModelError,\n type LanguageModelV4,\n type ProviderV4,\n} from '@ai-sdk/provider';\nimport {\n loadApiKey,\n withoutTrailingSlash,\n withUserAgentSuffix,\n type FetchFunction,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport { MoonshotAIChatLanguageModel } from './moonshotai-chat-language-model';\nimport type { 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 type { OpenAICompatibleChatConfig } from '@ai-sdk/openai-compatible/internal';\nimport {\n serializeModelOptions,\n WORKFLOW_SERIALIZE,\n WORKFLOW_DESERIALIZE,\n} from '@ai-sdk/provider-utils';\nimport type {\n LanguageModelV4CallOptions,\n LanguageModelV4GenerateResult,\n LanguageModelV4StreamPart,\n LanguageModelV4StreamResult,\n} from '@ai-sdk/provider';\nimport { convertMoonshotAIChatUsage } from './convert-moonshotai-chat-usage';\nimport type { MoonshotAIChatModelId } from './moonshotai-chat-options';\n\nexport class MoonshotAIChatLanguageModel extends OpenAICompatibleChatLanguageModel {\n static [WORKFLOW_SERIALIZE](model: MoonshotAIChatLanguageModel) {\n return serializeModelOptions({\n modelId: model.modelId,\n config: model.config,\n });\n }\n\n static [WORKFLOW_DESERIALIZE](options: {\n modelId: MoonshotAIChatModelId;\n config: OpenAICompatibleChatConfig;\n }) {\n return new MoonshotAIChatLanguageModel(options.modelId, options.config);\n }\n\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 type { 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,EACE;AAAA,OAGK;AACP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AACP,SAAS,SAAS;;;ACZlB,SAAS,yCAAyC;AAElD;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;;;ACJA,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;;;AD1CO,IAAM,8BAAN,MAAM,qCAAoC,kCAAkC;AAAA,EACjF,QAAQ,kBAAkB,EAAE,OAAoC;AAC9D,WAAO,sBAAsB;AAAA,MAC3B,SAAS,MAAM;AAAA,MACf,QAAQ,MAAM;AAAA,IAChB,CAAC;AAAA,EACH;AAAA,EAEA,QAAQ,oBAAoB,EAAE,SAG3B;AACD,WAAO,IAAI,6BAA4B,QAAQ,SAAS,QAAQ,MAAM;AAAA,EACxE;AAAA,EAEA,YACE,SACA,QACA;AACA,UAAM,SAAS,MAAM;AAAA,EACvB;AAAA,EAEA,MAAM,WACJ,SACwC;AAxC5C;AAyCI,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;;;AE5EO,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
CHANGED
|
@@ -1,10 +1,10 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@ai-sdk/moonshotai",
|
|
3
|
-
"version": "3.0.0-
|
|
3
|
+
"version": "3.0.0-canary.38",
|
|
4
|
+
"type": "module",
|
|
4
5
|
"license": "Apache-2.0",
|
|
5
6
|
"sideEffects": false,
|
|
6
7
|
"main": "./dist/index.js",
|
|
7
|
-
"module": "./dist/index.mjs",
|
|
8
8
|
"types": "./dist/index.d.ts",
|
|
9
9
|
"files": [
|
|
10
10
|
"dist/**/*",
|
|
@@ -20,21 +20,21 @@
|
|
|
20
20
|
"./package.json": "./package.json",
|
|
21
21
|
".": {
|
|
22
22
|
"types": "./dist/index.d.ts",
|
|
23
|
-
"import": "./dist/index.
|
|
24
|
-
"
|
|
23
|
+
"import": "./dist/index.js",
|
|
24
|
+
"default": "./dist/index.js"
|
|
25
25
|
}
|
|
26
26
|
},
|
|
27
27
|
"dependencies": {
|
|
28
|
-
"@ai-sdk/openai-compatible": "3.0.0-
|
|
29
|
-
"@ai-sdk/provider": "4.0.0-
|
|
30
|
-
"@ai-sdk/provider-utils": "5.0.0-
|
|
28
|
+
"@ai-sdk/openai-compatible": "3.0.0-canary.38",
|
|
29
|
+
"@ai-sdk/provider": "4.0.0-canary.15",
|
|
30
|
+
"@ai-sdk/provider-utils": "5.0.0-canary.31"
|
|
31
31
|
},
|
|
32
32
|
"devDependencies": {
|
|
33
33
|
"@types/node": "20.17.24",
|
|
34
34
|
"tsup": "^8",
|
|
35
35
|
"typescript": "5.8.3",
|
|
36
36
|
"zod": "3.25.76",
|
|
37
|
-
"@ai-sdk/test-server": "2.0.0-
|
|
37
|
+
"@ai-sdk/test-server": "2.0.0-canary.4",
|
|
38
38
|
"@vercel/ai-tsconfig": "0.0.0"
|
|
39
39
|
},
|
|
40
40
|
"peerDependencies": {
|
|
@@ -44,12 +44,14 @@
|
|
|
44
44
|
"node": ">=18"
|
|
45
45
|
},
|
|
46
46
|
"publishConfig": {
|
|
47
|
-
"access": "public"
|
|
47
|
+
"access": "public",
|
|
48
|
+
"provenance": true
|
|
48
49
|
},
|
|
49
50
|
"homepage": "https://ai-sdk.dev/docs",
|
|
50
51
|
"repository": {
|
|
51
52
|
"type": "git",
|
|
52
|
-
"url": "
|
|
53
|
+
"url": "https://github.com/vercel/ai",
|
|
54
|
+
"directory": "packages/moonshotai"
|
|
53
55
|
},
|
|
54
56
|
"bugs": {
|
|
55
57
|
"url": "https://github.com/vercel/ai/issues"
|
package/src/index.ts
CHANGED
|
@@ -1,15 +1,34 @@
|
|
|
1
1
|
import { OpenAICompatibleChatLanguageModel } from '@ai-sdk/openai-compatible';
|
|
2
|
-
import { OpenAICompatibleChatConfig } from '@ai-sdk/openai-compatible/internal';
|
|
2
|
+
import type { OpenAICompatibleChatConfig } from '@ai-sdk/openai-compatible/internal';
|
|
3
3
|
import {
|
|
4
|
+
serializeModelOptions,
|
|
5
|
+
WORKFLOW_SERIALIZE,
|
|
6
|
+
WORKFLOW_DESERIALIZE,
|
|
7
|
+
} from '@ai-sdk/provider-utils';
|
|
8
|
+
import type {
|
|
4
9
|
LanguageModelV4CallOptions,
|
|
5
10
|
LanguageModelV4GenerateResult,
|
|
6
11
|
LanguageModelV4StreamPart,
|
|
7
12
|
LanguageModelV4StreamResult,
|
|
8
13
|
} from '@ai-sdk/provider';
|
|
9
14
|
import { convertMoonshotAIChatUsage } from './convert-moonshotai-chat-usage';
|
|
10
|
-
import { MoonshotAIChatModelId } from './moonshotai-chat-options';
|
|
15
|
+
import type { MoonshotAIChatModelId } from './moonshotai-chat-options';
|
|
11
16
|
|
|
12
17
|
export class MoonshotAIChatLanguageModel extends OpenAICompatibleChatLanguageModel {
|
|
18
|
+
static [WORKFLOW_SERIALIZE](model: MoonshotAIChatLanguageModel) {
|
|
19
|
+
return serializeModelOptions({
|
|
20
|
+
modelId: model.modelId,
|
|
21
|
+
config: model.config,
|
|
22
|
+
});
|
|
23
|
+
}
|
|
24
|
+
|
|
25
|
+
static [WORKFLOW_DESERIALIZE](options: {
|
|
26
|
+
modelId: MoonshotAIChatModelId;
|
|
27
|
+
config: OpenAICompatibleChatConfig;
|
|
28
|
+
}) {
|
|
29
|
+
return new MoonshotAIChatLanguageModel(options.modelId, options.config);
|
|
30
|
+
}
|
|
31
|
+
|
|
13
32
|
constructor(
|
|
14
33
|
modelId: MoonshotAIChatModelId,
|
|
15
34
|
config: OpenAICompatibleChatConfig,
|
|
@@ -1,18 +1,18 @@
|
|
|
1
|
-
import { ProviderErrorStructure } from '@ai-sdk/openai-compatible';
|
|
1
|
+
import type { ProviderErrorStructure } from '@ai-sdk/openai-compatible';
|
|
2
2
|
import {
|
|
3
|
-
LanguageModelV4,
|
|
4
3
|
NoSuchModelError,
|
|
5
|
-
|
|
4
|
+
type LanguageModelV4,
|
|
5
|
+
type ProviderV4,
|
|
6
6
|
} from '@ai-sdk/provider';
|
|
7
7
|
import {
|
|
8
|
-
FetchFunction,
|
|
9
8
|
loadApiKey,
|
|
10
9
|
withoutTrailingSlash,
|
|
11
10
|
withUserAgentSuffix,
|
|
11
|
+
type FetchFunction,
|
|
12
12
|
} from '@ai-sdk/provider-utils';
|
|
13
13
|
import { z } from 'zod/v4';
|
|
14
14
|
import { MoonshotAIChatLanguageModel } from './moonshotai-chat-language-model';
|
|
15
|
-
import { MoonshotAIChatModelId } from './moonshotai-chat-options';
|
|
15
|
+
import type { MoonshotAIChatModelId } from './moonshotai-chat-options';
|
|
16
16
|
import { VERSION } from './version';
|
|
17
17
|
|
|
18
18
|
export type MoonshotAIErrorData = z.infer<typeof moonshotaiErrorSchema>;
|
package/dist/index.d.mts
DELETED
|
@@ -1,59 +0,0 @@
|
|
|
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.mjs
DELETED
|
@@ -1,168 +0,0 @@
|
|
|
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 = "3.0.0-beta.9";
|
|
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
|
package/dist/index.mjs.map
DELETED
|
@@ -1 +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 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":[]}
|