@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 +212 -0
- package/LICENSE +13 -0
- package/README.md +62 -0
- package/dist/index.d.mts +59 -0
- package/dist/index.d.ts +59 -0
- package/dist/index.js +190 -0
- package/dist/index.js.map +1 -0
- package/dist/index.mjs +168 -0
- package/dist/index.mjs.map +1 -0
- package/package.json +71 -0
- package/src/convert-moonshotai-chat-usage.ts +59 -0
- package/src/index.ts +11 -0
- package/src/moonshotai-chat-language-model.ts +60 -0
- package/src/moonshotai-chat-options.ts +28 -0
- package/src/moonshotai-provider.ts +151 -0
- package/src/version.ts +3 -0
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.
|
package/dist/index.d.mts
ADDED
|
@@ -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.d.ts
ADDED
|
@@ -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