@ank1015/providers 0.0.36 → 0.0.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/dist/debug-conversation.d.ts +2 -0
- package/dist/debug-conversation.d.ts.map +1 -0
- package/dist/debug-conversation.js +98 -0
- package/dist/debug-conversation.js.map +1 -0
- package/dist/debug-minimal-test.d.ts +2 -0
- package/dist/debug-minimal-test.d.ts.map +1 -0
- package/dist/debug-minimal-test.js +47 -0
- package/dist/debug-minimal-test.js.map +1 -0
- package/dist/debug-sdk-tool.d.ts +2 -0
- package/dist/debug-sdk-tool.d.ts.map +1 -0
- package/dist/debug-sdk-tool.js +99 -0
- package/dist/debug-sdk-tool.js.map +1 -0
- package/dist/debug-sdk.d.ts +2 -0
- package/dist/debug-sdk.d.ts.map +1 -0
- package/dist/debug-sdk.js +59 -0
- package/dist/debug-sdk.js.map +1 -0
- package/dist/debug-session-flow.d.ts +2 -0
- package/dist/debug-session-flow.d.ts.map +1 -0
- package/dist/debug-session-flow.js +134 -0
- package/dist/debug-session-flow.js.map +1 -0
- package/dist/debug-session-hack.d.ts +2 -0
- package/dist/debug-session-hack.d.ts.map +1 -0
- package/dist/debug-session-hack.js +143 -0
- package/dist/debug-session-hack.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +2 -0
- package/dist/index.js.map +1 -1
- package/dist/llm.d.ts.map +1 -1
- package/dist/llm.js +19 -1
- package/dist/llm.js.map +1 -1
- package/dist/models.generated.d.ts +72 -0
- package/dist/models.generated.d.ts.map +1 -1
- package/dist/models.generated.js +72 -0
- package/dist/models.generated.js.map +1 -1
- package/dist/providers/anthropic-agents-sdk/complete.d.ts +3 -0
- package/dist/providers/anthropic-agents-sdk/complete.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/complete.js +246 -0
- package/dist/providers/anthropic-agents-sdk/complete.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/index.d.ts +4 -0
- package/dist/providers/anthropic-agents-sdk/index.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/index.js +4 -0
- package/dist/providers/anthropic-agents-sdk/index.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/session.d.ts +37 -0
- package/dist/providers/anthropic-agents-sdk/session.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/session.js +188 -0
- package/dist/providers/anthropic-agents-sdk/session.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/stream.d.ts +3 -0
- package/dist/providers/anthropic-agents-sdk/stream.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/stream.js +402 -0
- package/dist/providers/anthropic-agents-sdk/stream.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/types.d.ts +22 -0
- package/dist/providers/anthropic-agents-sdk/types.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/types.js +3 -0
- package/dist/providers/anthropic-agents-sdk/types.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/utils.d.ts +39 -0
- package/dist/providers/anthropic-agents-sdk/utils.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/utils.js +251 -0
- package/dist/providers/anthropic-agents-sdk/utils.js.map +1 -0
- package/dist/providers/cerebras/complete.d.ts +3 -0
- package/dist/providers/cerebras/complete.d.ts.map +1 -0
- package/dist/providers/cerebras/complete.js +58 -0
- package/dist/providers/cerebras/complete.js.map +1 -0
- package/dist/providers/cerebras/index.d.ts +5 -0
- package/dist/providers/cerebras/index.d.ts.map +1 -0
- package/dist/providers/cerebras/index.js +4 -0
- package/dist/providers/cerebras/index.js.map +1 -0
- package/dist/providers/cerebras/stream.d.ts +3 -0
- package/dist/providers/cerebras/stream.d.ts.map +1 -0
- package/dist/providers/cerebras/stream.js +321 -0
- package/dist/providers/cerebras/stream.js.map +1 -0
- package/dist/providers/cerebras/types.d.ts +14 -0
- package/dist/providers/cerebras/types.d.ts.map +1 -0
- package/dist/providers/cerebras/types.js +2 -0
- package/dist/providers/cerebras/types.js.map +1 -0
- package/dist/providers/cerebras/utils.d.ts +19 -0
- package/dist/providers/cerebras/utils.d.ts.map +1 -0
- package/dist/providers/cerebras/utils.js +274 -0
- package/dist/providers/cerebras/utils.js.map +1 -0
- package/dist/providers/google/utils.d.ts.map +1 -1
- package/dist/providers/google/utils.js +2 -0
- package/dist/providers/google/utils.js.map +1 -1
- package/dist/providers/google-interactions/complete.d.ts +3 -0
- package/dist/providers/google-interactions/complete.d.ts.map +1 -0
- package/dist/providers/google-interactions/complete.js +63 -0
- package/dist/providers/google-interactions/complete.js.map +1 -0
- package/dist/providers/google-interactions/index.d.ts +5 -0
- package/dist/providers/google-interactions/index.d.ts.map +1 -0
- package/dist/providers/google-interactions/index.js +4 -0
- package/dist/providers/google-interactions/index.js.map +1 -0
- package/dist/providers/google-interactions/poll.d.ts +59 -0
- package/dist/providers/google-interactions/poll.d.ts.map +1 -0
- package/dist/providers/google-interactions/poll.js +161 -0
- package/dist/providers/google-interactions/poll.js.map +1 -0
- package/dist/providers/google-interactions/stream.d.ts +3 -0
- package/dist/providers/google-interactions/stream.d.ts.map +1 -0
- package/dist/providers/google-interactions/stream.js +242 -0
- package/dist/providers/google-interactions/stream.js.map +1 -0
- package/dist/providers/google-interactions/types.d.ts +10 -0
- package/dist/providers/google-interactions/types.d.ts.map +1 -0
- package/dist/providers/google-interactions/types.js +2 -0
- package/dist/providers/google-interactions/types.js.map +1 -0
- package/dist/providers/google-interactions/utils.d.ts +61 -0
- package/dist/providers/google-interactions/utils.d.ts.map +1 -0
- package/dist/providers/google-interactions/utils.js +252 -0
- package/dist/providers/google-interactions/utils.js.map +1 -0
- package/dist/providers/zai/complete.d.ts +3 -0
- package/dist/providers/zai/complete.d.ts.map +1 -0
- package/dist/providers/zai/complete.js +58 -0
- package/dist/providers/zai/complete.js.map +1 -0
- package/dist/providers/zai/index.d.ts +5 -0
- package/dist/providers/zai/index.d.ts.map +1 -0
- package/dist/providers/zai/index.js +4 -0
- package/dist/providers/zai/index.js.map +1 -0
- package/dist/providers/zai/stream.d.ts +3 -0
- package/dist/providers/zai/stream.d.ts.map +1 -0
- package/dist/providers/zai/stream.js +321 -0
- package/dist/providers/zai/stream.js.map +1 -0
- package/dist/providers/zai/types.d.ts +13 -0
- package/dist/providers/zai/types.d.ts.map +1 -0
- package/dist/providers/zai/types.js +2 -0
- package/dist/providers/zai/types.js.map +1 -0
- package/dist/providers/zai/utils.d.ts +19 -0
- package/dist/providers/zai/utils.d.ts.map +1 -0
- package/dist/providers/zai/utils.js +253 -0
- package/dist/providers/zai/utils.js.map +1 -0
- package/dist/test-sdk-direct.d.ts +2 -0
- package/dist/test-sdk-direct.d.ts.map +1 -0
- package/dist/test-sdk-direct.js +41 -0
- package/dist/test-sdk-direct.js.map +1 -0
- package/dist/test.d.ts +2 -0
- package/dist/test.d.ts.map +1 -0
- package/dist/test.js +213 -0
- package/dist/test.js.map +1 -0
- package/dist/types.d.ts +7 -1
- package/dist/types.d.ts.map +1 -1
- package/dist/types.js +1 -1
- package/dist/types.js.map +1 -1
- package/dist/utils/oauth/google-antigravity.d.ts +25 -0
- package/dist/utils/oauth/google-antigravity.d.ts.map +1 -0
- package/dist/utils/oauth/google-antigravity.js +337 -0
- package/dist/utils/oauth/google-antigravity.js.map +1 -0
- package/dist/utils/oauth/pkce.d.ts +13 -0
- package/dist/utils/oauth/pkce.d.ts.map +1 -0
- package/dist/utils/oauth/pkce.js +31 -0
- package/dist/utils/oauth/pkce.js.map +1 -0
- package/package.json +1 -1
|
@@ -0,0 +1,251 @@
|
|
|
1
|
+
import { createSdkMcpServer, tool } from "@anthropic-ai/claude-agent-sdk";
|
|
2
|
+
import { z } from "zod";
|
|
3
|
+
/**
|
|
4
|
+
* Creates an MCP server with tools converted from our Tool[] format.
|
|
5
|
+
* The SDK uses MCP protocol for custom tools.
|
|
6
|
+
*/
|
|
7
|
+
export function createToolsServer(tools) {
|
|
8
|
+
if (!tools || tools.length === 0) {
|
|
9
|
+
return undefined;
|
|
10
|
+
}
|
|
11
|
+
const mcpTools = tools.map((t) => {
|
|
12
|
+
// Convert TypeBox schema to Zod schema for MCP
|
|
13
|
+
// TypeBox generates JSON Schema, we need to convert to Zod
|
|
14
|
+
const jsonSchema = t.parameters;
|
|
15
|
+
const zodSchema = jsonSchemaToZod(jsonSchema);
|
|
16
|
+
return tool(t.name, t.description, zodSchema, async (args) => {
|
|
17
|
+
// This handler won't actually be called during inference
|
|
18
|
+
// The SDK returns tool calls to us, and we handle them externally
|
|
19
|
+
// This is just a placeholder to satisfy the MCP tool definition
|
|
20
|
+
return {
|
|
21
|
+
content: [{ type: "text", text: JSON.stringify(args) }]
|
|
22
|
+
};
|
|
23
|
+
});
|
|
24
|
+
});
|
|
25
|
+
return createSdkMcpServer({
|
|
26
|
+
name: "custom-tools",
|
|
27
|
+
version: "1.0.0",
|
|
28
|
+
tools: mcpTools
|
|
29
|
+
});
|
|
30
|
+
}
|
|
31
|
+
/**
|
|
32
|
+
* Converts JSON Schema (from TypeBox) to Zod schema for MCP tools.
|
|
33
|
+
* This is a simplified converter that handles common cases.
|
|
34
|
+
*/
|
|
35
|
+
function jsonSchemaToZod(schema) {
|
|
36
|
+
const zodSchema = {};
|
|
37
|
+
const properties = schema.properties || {};
|
|
38
|
+
const required = schema.required || [];
|
|
39
|
+
for (const [key, prop] of Object.entries(properties)) {
|
|
40
|
+
let zodType;
|
|
41
|
+
switch (prop.type) {
|
|
42
|
+
case 'string':
|
|
43
|
+
zodType = z.string().describe(prop.description || '');
|
|
44
|
+
break;
|
|
45
|
+
case 'number':
|
|
46
|
+
case 'integer':
|
|
47
|
+
zodType = z.number().describe(prop.description || '');
|
|
48
|
+
break;
|
|
49
|
+
case 'boolean':
|
|
50
|
+
zodType = z.boolean().describe(prop.description || '');
|
|
51
|
+
break;
|
|
52
|
+
case 'array':
|
|
53
|
+
if (prop.items) {
|
|
54
|
+
const itemType = getZodType(prop.items);
|
|
55
|
+
zodType = z.array(itemType).describe(prop.description || '');
|
|
56
|
+
}
|
|
57
|
+
else {
|
|
58
|
+
zodType = z.array(z.unknown()).describe(prop.description || '');
|
|
59
|
+
}
|
|
60
|
+
break;
|
|
61
|
+
case 'object':
|
|
62
|
+
if (prop.properties) {
|
|
63
|
+
zodType = z.object(jsonSchemaToZod(prop)).describe(prop.description || '');
|
|
64
|
+
}
|
|
65
|
+
else {
|
|
66
|
+
zodType = z.record(z.string(), z.unknown()).describe(prop.description || '');
|
|
67
|
+
}
|
|
68
|
+
break;
|
|
69
|
+
default:
|
|
70
|
+
zodType = z.unknown().describe(prop.description || '');
|
|
71
|
+
}
|
|
72
|
+
// Make optional if not in required array
|
|
73
|
+
if (!required.includes(key)) {
|
|
74
|
+
zodType = zodType.optional();
|
|
75
|
+
}
|
|
76
|
+
zodSchema[key] = zodType;
|
|
77
|
+
}
|
|
78
|
+
return zodSchema;
|
|
79
|
+
}
|
|
80
|
+
function getZodType(prop) {
|
|
81
|
+
switch (prop.type) {
|
|
82
|
+
case 'string':
|
|
83
|
+
return z.string();
|
|
84
|
+
case 'number':
|
|
85
|
+
case 'integer':
|
|
86
|
+
return z.number();
|
|
87
|
+
case 'boolean':
|
|
88
|
+
return z.boolean();
|
|
89
|
+
case 'object':
|
|
90
|
+
if (prop.properties) {
|
|
91
|
+
return z.object(jsonSchemaToZod(prop));
|
|
92
|
+
}
|
|
93
|
+
return z.record(z.string(), z.unknown());
|
|
94
|
+
case 'array':
|
|
95
|
+
if (prop.items) {
|
|
96
|
+
return z.array(getZodType(prop.items));
|
|
97
|
+
}
|
|
98
|
+
return z.array(z.unknown());
|
|
99
|
+
default:
|
|
100
|
+
return z.unknown();
|
|
101
|
+
}
|
|
102
|
+
}
|
|
103
|
+
/**
|
|
104
|
+
* Strips the MCP prefix from tool names.
|
|
105
|
+
* SDK returns tool names like "mcp__custom-tools__calculate",
|
|
106
|
+
* but the Conversation class expects "calculate".
|
|
107
|
+
*/
|
|
108
|
+
export function stripMcpToolPrefix(name) {
|
|
109
|
+
const prefix = 'mcp__custom-tools__';
|
|
110
|
+
if (name.startsWith(prefix)) {
|
|
111
|
+
return name.slice(prefix.length);
|
|
112
|
+
}
|
|
113
|
+
return name;
|
|
114
|
+
}
|
|
115
|
+
/**
|
|
116
|
+
* Converts a BetaMessage to our AssistantResponse format.
|
|
117
|
+
*/
|
|
118
|
+
export function getResponseAssistantResponse(message) {
|
|
119
|
+
const assistantResponse = [];
|
|
120
|
+
if (message.content) {
|
|
121
|
+
for (const block of message.content) {
|
|
122
|
+
if (block.type === 'thinking') {
|
|
123
|
+
const thinkingBlock = block;
|
|
124
|
+
assistantResponse.push({
|
|
125
|
+
type: 'thinking',
|
|
126
|
+
thinkingText: thinkingBlock.thinking
|
|
127
|
+
});
|
|
128
|
+
}
|
|
129
|
+
else if (block.type === 'text') {
|
|
130
|
+
const textBlock = block;
|
|
131
|
+
assistantResponse.push({
|
|
132
|
+
type: 'response',
|
|
133
|
+
content: [{
|
|
134
|
+
type: 'text',
|
|
135
|
+
content: textBlock.text
|
|
136
|
+
}]
|
|
137
|
+
});
|
|
138
|
+
}
|
|
139
|
+
else if (block.type === 'tool_use') {
|
|
140
|
+
const toolBlock = block;
|
|
141
|
+
assistantResponse.push({
|
|
142
|
+
type: 'toolCall',
|
|
143
|
+
toolCallId: toolBlock.id,
|
|
144
|
+
// Strip MCP prefix so Conversation class can find the tool
|
|
145
|
+
name: stripMcpToolPrefix(toolBlock.name),
|
|
146
|
+
arguments: toolBlock.input
|
|
147
|
+
});
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
return assistantResponse;
|
|
152
|
+
}
|
|
153
|
+
/**
|
|
154
|
+
* Converts SDK usage to our Usage format.
|
|
155
|
+
*/
|
|
156
|
+
export function getResponseUsage(sdkUsage, totalCostUsd, model) {
|
|
157
|
+
const usage = {
|
|
158
|
+
input: sdkUsage.input_tokens,
|
|
159
|
+
output: sdkUsage.output_tokens,
|
|
160
|
+
cacheRead: sdkUsage.cache_read_input_tokens || 0,
|
|
161
|
+
cacheWrite: sdkUsage.cache_creation_input_tokens || 0,
|
|
162
|
+
totalTokens: sdkUsage.input_tokens + sdkUsage.output_tokens +
|
|
163
|
+
(sdkUsage.cache_read_input_tokens || 0) +
|
|
164
|
+
(sdkUsage.cache_creation_input_tokens || 0),
|
|
165
|
+
cost: {
|
|
166
|
+
input: 0,
|
|
167
|
+
output: 0,
|
|
168
|
+
cacheRead: 0,
|
|
169
|
+
cacheWrite: 0,
|
|
170
|
+
total: totalCostUsd
|
|
171
|
+
}
|
|
172
|
+
};
|
|
173
|
+
// Calculate individual costs based on model pricing if available
|
|
174
|
+
if (model.cost) {
|
|
175
|
+
usage.cost.input = (usage.input / 1000000) * model.cost.input;
|
|
176
|
+
usage.cost.output = (usage.output / 1000000) * model.cost.output;
|
|
177
|
+
usage.cost.cacheRead = (usage.cacheRead / 1000000) * model.cost.cacheRead;
|
|
178
|
+
usage.cost.cacheWrite = (usage.cacheWrite / 1000000) * model.cost.cacheWrite;
|
|
179
|
+
}
|
|
180
|
+
return usage;
|
|
181
|
+
}
|
|
182
|
+
/**
|
|
183
|
+
* Maps SDK stop reason to our StopReason type.
|
|
184
|
+
*/
|
|
185
|
+
export function mapStopReason(stopReason) {
|
|
186
|
+
switch (stopReason) {
|
|
187
|
+
case "end_turn":
|
|
188
|
+
return "stop";
|
|
189
|
+
case "max_tokens":
|
|
190
|
+
return "length";
|
|
191
|
+
case "tool_use":
|
|
192
|
+
return "toolUse";
|
|
193
|
+
case "stop_sequence":
|
|
194
|
+
return "stop";
|
|
195
|
+
default:
|
|
196
|
+
return "stop";
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
/**
|
|
200
|
+
* Maps SDK result subtype to our StopReason type.
|
|
201
|
+
*/
|
|
202
|
+
export function mapResultSubtypeToStopReason(subtype) {
|
|
203
|
+
switch (subtype) {
|
|
204
|
+
case "success":
|
|
205
|
+
return "stop";
|
|
206
|
+
case "error_during_execution":
|
|
207
|
+
return "error";
|
|
208
|
+
case "error_max_turns":
|
|
209
|
+
return "length";
|
|
210
|
+
case "error_max_budget_usd":
|
|
211
|
+
return "error";
|
|
212
|
+
case "error_max_structured_output_retries":
|
|
213
|
+
return "error";
|
|
214
|
+
default:
|
|
215
|
+
return "error";
|
|
216
|
+
}
|
|
217
|
+
}
|
|
218
|
+
/**
|
|
219
|
+
* Creates a mock BetaMessage for error cases.
|
|
220
|
+
*/
|
|
221
|
+
export function getMockBetaMessage() {
|
|
222
|
+
return {
|
|
223
|
+
id: "msg_mock",
|
|
224
|
+
type: "message",
|
|
225
|
+
role: "assistant",
|
|
226
|
+
content: [],
|
|
227
|
+
model: "claude-sonnet-4-5-20250514",
|
|
228
|
+
stop_reason: "end_turn",
|
|
229
|
+
stop_sequence: null,
|
|
230
|
+
usage: {
|
|
231
|
+
input_tokens: 0,
|
|
232
|
+
output_tokens: 0,
|
|
233
|
+
cache_creation_input_tokens: 0,
|
|
234
|
+
cache_read_input_tokens: 0,
|
|
235
|
+
}
|
|
236
|
+
};
|
|
237
|
+
}
|
|
238
|
+
/**
|
|
239
|
+
* Creates empty usage for error cases.
|
|
240
|
+
*/
|
|
241
|
+
export function getEmptyUsage() {
|
|
242
|
+
return {
|
|
243
|
+
input: 0,
|
|
244
|
+
output: 0,
|
|
245
|
+
cacheRead: 0,
|
|
246
|
+
cacheWrite: 0,
|
|
247
|
+
totalTokens: 0,
|
|
248
|
+
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
|
|
249
|
+
};
|
|
250
|
+
}
|
|
251
|
+
//# sourceMappingURL=utils.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/providers/anthropic-agents-sdk/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,MAAM,gCAAgC,CAAC;AAC1E,OAAO,EAAE,CAAC,EAAmB,MAAM,KAAK,CAAC;AAUzC;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,KAAa;IAC9C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAClC,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAChC,+CAA+C;QAC/C,2DAA2D;QAC3D,MAAM,UAAU,GAAG,CAAC,CAAC,UAAiB,CAAC;QACvC,MAAM,SAAS,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;QAE9C,OAAO,IAAI,CACV,CAAC,CAAC,IAAI,EACN,CAAC,CAAC,WAAW,EACb,SAAgB,EAChB,KAAK,EAAE,IAAI,EAAE,EAAE;YACd,yDAAyD;YACzD,kEAAkE;YAClE,gEAAgE;YAChE,OAAO;gBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;aAChE,CAAC;QACH,CAAC,CACD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,kBAAkB,CAAC;QACzB,IAAI,EAAE,cAAc;QACpB,OAAO,EAAE,OAAO;QAChB,KAAK,EAAE,QAAe;KACtB,CAAC,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAS,eAAe,CAAC,MAAW;IACnC,MAAM,SAAS,GAA+B,EAAE,CAAC;IACjD,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;IAC3C,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;IAEvC,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAoB,EAAE,CAAC;QACzE,IAAI,OAAmB,CAAC;QAExB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,QAAQ;gBACZ,OAAO,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;gBACtD,MAAM;YACP,KAAK,QAAQ,CAAC;YACd,KAAK,SAAS;gBACb,OAAO,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;gBACtD,MAAM;YACP,KAAK,SAAS;gBACb,OAAO,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;gBACvD,MAAM;YACP,KAAK,OAAO;gBACX,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBAChB,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACxC,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;gBAC9D,CAAC;qBAAM,CAAC;oBACP,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;gBACjE,CAAC;gBACD,MAAM;YACP,KAAK,QAAQ;gBACZ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACrB,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;gBAC5E,CAAC;qBAAM,CAAC;oBACP,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;gBAC9E,CAAC;gBACD,MAAM;YACP;gBACC,OAAO,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;QACzD,CAAC;QAED,yCAAyC;QACzC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7B,OAAO,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QAC9B,CAAC;QAED,SAAS,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,SAAS,UAAU,CAAC,IAAS;IAC5B,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,QAAQ;YACZ,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QACnB,KAAK,QAAQ,CAAC;QACd,KAAK,SAAS;YACb,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QACnB,KAAK,SAAS;YACb,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;QACpB,KAAK,QAAQ;YACZ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrB,OAAO,CAAC,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;YACxC,CAAC;YACD,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC1C,KAAK,OAAO;YACX,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBAChB,OAAO,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACxC,CAAC;YACD,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7B;YACC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;IACrB,CAAC;AACF,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAAC,IAAY;IAC9C,MAAM,MAAM,GAAG,qBAAqB,CAAC;IACrC,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,4BAA4B,CAAC,OAAoB;IAChE,MAAM,iBAAiB,GAAsB,EAAE,CAAC;IAEhD,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACrB,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACrC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC/B,MAAM,aAAa,GAAG,KAA0B,CAAC;gBACjD,iBAAiB,CAAC,IAAI,CAAC;oBACtB,IAAI,EAAE,UAAU;oBAChB,YAAY,EAAE,aAAa,CAAC,QAAQ;iBACpC,CAAC,CAAC;YACJ,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAClC,MAAM,SAAS,GAAG,KAAsB,CAAC;gBACzC,iBAAiB,CAAC,IAAI,CAAC;oBACtB,IAAI,EAAE,UAAU;oBAChB,OAAO,EAAE,CAAC;4BACT,IAAI,EAAE,MAAM;4BACZ,OAAO,EAAE,SAAS,CAAC,IAAI;yBACvB,CAAC;iBACF,CAAC,CAAC;YACJ,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACtC,MAAM,SAAS,GAAG,KAAyB,CAAC;gBAC5C,iBAAiB,CAAC,IAAI,CAAC;oBACtB,IAAI,EAAE,UAAU;oBAChB,UAAU,EAAE,SAAS,CAAC,EAAE;oBACxB,2DAA2D;oBAC3D,IAAI,EAAE,kBAAkB,CAAC,SAAS,CAAC,IAAI,CAAC;oBACxC,SAAS,EAAE,SAAS,CAAC,KAA4B;iBACjD,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,iBAAiB,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAC/B,QAA0B,EAC1B,YAAoB,EACpB,KAAoC;IAEpC,MAAM,KAAK,GAAU;QACpB,KAAK,EAAE,QAAQ,CAAC,YAAY;QAC5B,MAAM,EAAE,QAAQ,CAAC,aAAa;QAC9B,SAAS,EAAE,QAAQ,CAAC,uBAAuB,IAAI,CAAC;QAChD,UAAU,EAAE,QAAQ,CAAC,2BAA2B,IAAI,CAAC;QACrD,WAAW,EAAE,QAAQ,CAAC,YAAY,GAAG,QAAQ,CAAC,aAAa;YAC1D,CAAC,QAAQ,CAAC,uBAAuB,IAAI,CAAC,CAAC;YACvC,CAAC,QAAQ,CAAC,2BAA2B,IAAI,CAAC,CAAC;QAC5C,IAAI,EAAE;YACL,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,KAAK,EAAE,YAAY;SACnB;KACD,CAAC;IAEF,iEAAiE;IACjE,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;QAChB,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,KAAK,CAAC,KAAK,GAAG,OAAS,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;QAChE,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,KAAK,CAAC,MAAM,GAAG,OAAS,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;QACnE,KAAK,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,KAAK,CAAC,SAAS,GAAG,OAAS,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;QAC5E,KAAK,CAAC,IAAI,CAAC,UAAU,GAAG,CAAC,KAAK,CAAC,UAAU,GAAG,OAAS,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC;IAChF,CAAC;IAED,OAAO,KAAK,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,UAAyB;IACtD,QAAQ,UAAU,EAAE,CAAC;QACpB,KAAK,UAAU;YACd,OAAO,MAAM,CAAC;QACf,KAAK,YAAY;YAChB,OAAO,QAAQ,CAAC;QACjB,KAAK,UAAU;YACd,OAAO,SAAS,CAAC;QAClB,KAAK,eAAe;YACnB,OAAO,MAAM,CAAC;QACf;YACC,OAAO,MAAM,CAAC;IAChB,CAAC;AACF,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,4BAA4B,CAAC,OAAoC;IAChF,QAAQ,OAAO,EAAE,CAAC;QACjB,KAAK,SAAS;YACb,OAAO,MAAM,CAAC;QACf,KAAK,wBAAwB;YAC5B,OAAO,OAAO,CAAC;QAChB,KAAK,iBAAiB;YACrB,OAAO,QAAQ,CAAC;QACjB,KAAK,sBAAsB;YAC1B,OAAO,OAAO,CAAC;QAChB,KAAK,qCAAqC;YACzC,OAAO,OAAO,CAAC;QAChB;YACC,OAAO,OAAO,CAAC;IACjB,CAAC;AACF,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB;IACjC,OAAO;QACN,EAAE,EAAE,UAAU;QACd,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,EAAE;QACX,KAAK,EAAE,4BAA4B;QACnC,WAAW,EAAE,UAAU;QACvB,aAAa,EAAE,IAAI;QACnB,KAAK,EAAE;YACN,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,2BAA2B,EAAE,CAAC;YAC9B,uBAAuB,EAAE,CAAC;SAC1B;KACyB,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa;IAC5B,OAAO;QACN,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;KACpE,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"complete.d.ts","sourceRoot":"","sources":["../../../src/providers/cerebras/complete.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAA4B,MAAM,gBAAgB,CAAC;AAM5E,eAAO,MAAM,gBAAgB,EAAE,gBAAgB,CAAC,UAAU,CAoEzD,CAAC"}
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
import { createClient, buildParams, getResponseAssistantResponse, getResponseUsage, getAssistantStopReason } from "./utils.js";
|
|
2
|
+
export const completeCerebras = async (model, context, options, id) => {
|
|
3
|
+
const startTimestamp = Date.now();
|
|
4
|
+
const client = createClient(model, options?.apiKey);
|
|
5
|
+
const params = buildParams(model, context, options);
|
|
6
|
+
try {
|
|
7
|
+
const response = await client.chat.completions.create(params, { signal: options?.signal });
|
|
8
|
+
// Cache processed content for performance and consistency
|
|
9
|
+
const content = getResponseAssistantResponse(response);
|
|
10
|
+
const usage = getResponseUsage(response, model);
|
|
11
|
+
let stopReason = getAssistantStopReason(response);
|
|
12
|
+
// Ensure stopReason is toolUse if we have tool calls
|
|
13
|
+
const hasToolCall = content.some(c => c.type === 'toolCall');
|
|
14
|
+
if (hasToolCall && stopReason === 'stop') {
|
|
15
|
+
stopReason = 'toolUse';
|
|
16
|
+
}
|
|
17
|
+
return {
|
|
18
|
+
role: "assistant",
|
|
19
|
+
message: response,
|
|
20
|
+
id,
|
|
21
|
+
api: model.api,
|
|
22
|
+
model,
|
|
23
|
+
timestamp: Date.now(),
|
|
24
|
+
duration: Date.now() - startTimestamp,
|
|
25
|
+
stopReason,
|
|
26
|
+
content,
|
|
27
|
+
usage
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
catch (error) {
|
|
31
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
32
|
+
const isAborted = options.signal?.aborted;
|
|
33
|
+
const stopReason = isAborted ? "aborted" : "error";
|
|
34
|
+
// Return error response with empty content and zero usage
|
|
35
|
+
const emptyUsage = {
|
|
36
|
+
input: 0,
|
|
37
|
+
output: 0,
|
|
38
|
+
cacheRead: 0,
|
|
39
|
+
cacheWrite: 0,
|
|
40
|
+
totalTokens: 0,
|
|
41
|
+
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
|
|
42
|
+
};
|
|
43
|
+
return {
|
|
44
|
+
role: "assistant",
|
|
45
|
+
message: {}, // Empty response object for error case
|
|
46
|
+
id,
|
|
47
|
+
api: model.api,
|
|
48
|
+
model,
|
|
49
|
+
errorMessage,
|
|
50
|
+
timestamp: Date.now(),
|
|
51
|
+
duration: Date.now() - startTimestamp,
|
|
52
|
+
stopReason,
|
|
53
|
+
content: [],
|
|
54
|
+
usage: emptyUsage
|
|
55
|
+
};
|
|
56
|
+
}
|
|
57
|
+
};
|
|
58
|
+
//# sourceMappingURL=complete.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"complete.js","sourceRoot":"","sources":["../../../src/providers/cerebras/complete.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,4BAA4B,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAG/H,MAAM,CAAC,MAAM,gBAAgB,GAAiC,KAAK,EAClE,KAAwB,EACxB,OAAO,EACP,OAAgC,EAChC,EAAU,EACT,EAAE;IACH,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAClC,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACpD,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAEpD,IAAI,CAAC;QACJ,MAAM,QAAQ,GAAmB,MAAM,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CACpE,MAAM,EACN,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAC3B,CAAC;QAEF,0DAA0D;QAC1D,MAAM,OAAO,GAAG,4BAA4B,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAChD,IAAI,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QAElD,qDAAqD;QACrD,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QAC7D,IAAI,WAAW,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;YAC1C,UAAU,GAAG,SAAS,CAAC;QACxB,CAAC;QAED,OAAO;YACN,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,QAAQ;YACjB,EAAE;YACF,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK;YACL,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;YACrC,UAAU;YACV,OAAO;YACP,KAAK;SACL,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC;QAC1C,MAAM,UAAU,GAAe,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;QAE/D,0DAA0D;QAC1D,MAAM,UAAU,GAAU;YACzB,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;YACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;SACpE,CAAC;QAEF,OAAO;YACN,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAoB,EAAE,uCAAuC;YACtE,EAAE;YACF,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK;YACL,YAAY;YACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;YACrC,UAAU;YACV,OAAO,EAAE,EAAE;YACX,KAAK,EAAE,UAAU;SACjB,CAAC;IACH,CAAC;AACF,CAAC,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export { completeCerebras } from "./complete.js";
|
|
2
|
+
export { streamCerebras } from "./stream.js";
|
|
3
|
+
export { CerebrasProviderOptions, CerebrasReasoningFormat, CerebrasReasoningEffort } from "./types.js";
|
|
4
|
+
export { getMockCerebrasMessage } from "./utils.js";
|
|
5
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/cerebras/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,YAAY,CAAC;AACvG,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/cerebras/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../../src/providers/cerebras/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAA0I,cAAc,EAAe,MAAM,gBAAgB,CAAA;AAoBpM,eAAO,MAAM,cAAc,EAAE,cAAc,CAAC,UAAU,CA4WrD,CAAA"}
|