@auxiora/providers 1.0.0
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/LICENSE +191 -0
- package/dist/anthropic.d.ts +82 -0
- package/dist/anthropic.d.ts.map +1 -0
- package/dist/anthropic.js +618 -0
- package/dist/anthropic.js.map +1 -0
- package/dist/claude-code-tools.d.ts +29 -0
- package/dist/claude-code-tools.d.ts.map +1 -0
- package/dist/claude-code-tools.js +221 -0
- package/dist/claude-code-tools.js.map +1 -0
- package/dist/claude-oauth.d.ts +86 -0
- package/dist/claude-oauth.d.ts.map +1 -0
- package/dist/claude-oauth.js +318 -0
- package/dist/claude-oauth.js.map +1 -0
- package/dist/cohere.d.ts +18 -0
- package/dist/cohere.d.ts.map +1 -0
- package/dist/cohere.js +163 -0
- package/dist/cohere.js.map +1 -0
- package/dist/deepseek.d.ts +18 -0
- package/dist/deepseek.d.ts.map +1 -0
- package/dist/deepseek.js +164 -0
- package/dist/deepseek.js.map +1 -0
- package/dist/factory.d.ts +19 -0
- package/dist/factory.d.ts.map +1 -0
- package/dist/factory.js +108 -0
- package/dist/factory.js.map +1 -0
- package/dist/google.d.ts +18 -0
- package/dist/google.d.ts.map +1 -0
- package/dist/google.js +141 -0
- package/dist/google.js.map +1 -0
- package/dist/groq.d.ts +18 -0
- package/dist/groq.d.ts.map +1 -0
- package/dist/groq.js +186 -0
- package/dist/groq.js.map +1 -0
- package/dist/index.d.ts +15 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +14 -0
- package/dist/index.js.map +1 -0
- package/dist/ollama.d.ts +18 -0
- package/dist/ollama.d.ts.map +1 -0
- package/dist/ollama.js +141 -0
- package/dist/ollama.js.map +1 -0
- package/dist/openai-compatible.d.ts +20 -0
- package/dist/openai-compatible.d.ts.map +1 -0
- package/dist/openai-compatible.js +112 -0
- package/dist/openai-compatible.js.map +1 -0
- package/dist/openai.d.ts +20 -0
- package/dist/openai.d.ts.map +1 -0
- package/dist/openai.js +259 -0
- package/dist/openai.js.map +1 -0
- package/dist/replicate.d.ts +20 -0
- package/dist/replicate.d.ts.map +1 -0
- package/dist/replicate.js +186 -0
- package/dist/replicate.js.map +1 -0
- package/dist/thinking-levels.d.ts +16 -0
- package/dist/thinking-levels.d.ts.map +1 -0
- package/dist/thinking-levels.js +34 -0
- package/dist/thinking-levels.js.map +1 -0
- package/dist/types.d.ts +157 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/xai.d.ts +18 -0
- package/dist/xai.d.ts.map +1 -0
- package/dist/xai.js +164 -0
- package/dist/xai.js.map +1 -0
- package/package.json +30 -0
- package/src/anthropic.ts +691 -0
- package/src/claude-code-tools.ts +233 -0
- package/src/claude-oauth.ts +410 -0
- package/src/cohere.ts +242 -0
- package/src/deepseek.ts +241 -0
- package/src/factory.ts +142 -0
- package/src/google.ts +176 -0
- package/src/groq.ts +263 -0
- package/src/index.ts +44 -0
- package/src/ollama.ts +194 -0
- package/src/openai-compatible.ts +154 -0
- package/src/openai.ts +307 -0
- package/src/replicate.ts +247 -0
- package/src/thinking-levels.ts +37 -0
- package/src/types.ts +171 -0
- package/src/xai.ts +241 -0
- package/tests/adapters.test.ts +185 -0
- package/tests/claude-oauth.test.ts +45 -0
- package/tests/new-providers.test.ts +732 -0
- package/tests/thinking-levels.test.ts +82 -0
- package/tsconfig.json +8 -0
- package/tsconfig.tsbuildinfo +1 -0
|
@@ -0,0 +1,112 @@
|
|
|
1
|
+
import OpenAI from 'openai';
|
|
2
|
+
const DEFAULT_MAX_TOKENS = 4096;
|
|
3
|
+
export class OpenAICompatibleProvider {
|
|
4
|
+
name;
|
|
5
|
+
metadata;
|
|
6
|
+
client;
|
|
7
|
+
defaultModel;
|
|
8
|
+
defaultMaxTokens;
|
|
9
|
+
constructor(options) {
|
|
10
|
+
this.name = options.name || 'openai-compatible';
|
|
11
|
+
this.defaultModel = options.model;
|
|
12
|
+
this.defaultMaxTokens = options.maxTokens || DEFAULT_MAX_TOKENS;
|
|
13
|
+
this.client = new OpenAI({
|
|
14
|
+
apiKey: options.apiKey || 'not-needed',
|
|
15
|
+
baseURL: options.baseUrl,
|
|
16
|
+
});
|
|
17
|
+
this.metadata = {
|
|
18
|
+
name: this.name,
|
|
19
|
+
displayName: options.name ? `${options.name} (OpenAI-compatible)` : 'OpenAI-compatible',
|
|
20
|
+
models: {
|
|
21
|
+
[this.defaultModel]: {
|
|
22
|
+
maxContextTokens: 128000,
|
|
23
|
+
supportsVision: false,
|
|
24
|
+
supportsTools: true,
|
|
25
|
+
supportsStreaming: true,
|
|
26
|
+
supportsImageGen: false,
|
|
27
|
+
costPer1kInput: 0,
|
|
28
|
+
costPer1kOutput: 0,
|
|
29
|
+
strengths: ['code', 'reasoning'],
|
|
30
|
+
isLocal: true,
|
|
31
|
+
},
|
|
32
|
+
},
|
|
33
|
+
isAvailable: async () => {
|
|
34
|
+
try {
|
|
35
|
+
await this.client.models.list();
|
|
36
|
+
return true;
|
|
37
|
+
}
|
|
38
|
+
catch {
|
|
39
|
+
return false;
|
|
40
|
+
}
|
|
41
|
+
},
|
|
42
|
+
};
|
|
43
|
+
}
|
|
44
|
+
async complete(messages, options) {
|
|
45
|
+
const openaiMessages = this.prepareMessages(messages, options);
|
|
46
|
+
const response = await this.client.chat.completions.create({
|
|
47
|
+
model: options?.model || this.defaultModel,
|
|
48
|
+
max_tokens: options?.maxTokens || this.defaultMaxTokens,
|
|
49
|
+
messages: openaiMessages,
|
|
50
|
+
temperature: options?.temperature,
|
|
51
|
+
});
|
|
52
|
+
const choice = response.choices[0];
|
|
53
|
+
const content = choice?.message?.content || '';
|
|
54
|
+
return {
|
|
55
|
+
content,
|
|
56
|
+
usage: {
|
|
57
|
+
inputTokens: response.usage?.prompt_tokens || 0,
|
|
58
|
+
outputTokens: response.usage?.completion_tokens || 0,
|
|
59
|
+
},
|
|
60
|
+
model: response.model,
|
|
61
|
+
finishReason: choice?.finish_reason || 'unknown',
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
async *stream(messages, options) {
|
|
65
|
+
const openaiMessages = this.prepareMessages(messages, options);
|
|
66
|
+
try {
|
|
67
|
+
const stream = await this.client.chat.completions.create({
|
|
68
|
+
model: options?.model || this.defaultModel,
|
|
69
|
+
max_tokens: options?.maxTokens || this.defaultMaxTokens,
|
|
70
|
+
messages: openaiMessages,
|
|
71
|
+
temperature: options?.temperature,
|
|
72
|
+
stream: true,
|
|
73
|
+
stream_options: { include_usage: true },
|
|
74
|
+
});
|
|
75
|
+
let inputTokens = 0;
|
|
76
|
+
let outputTokens = 0;
|
|
77
|
+
for await (const chunk of stream) {
|
|
78
|
+
const delta = chunk.choices[0]?.delta;
|
|
79
|
+
if (delta?.content) {
|
|
80
|
+
yield { type: 'text', content: delta.content };
|
|
81
|
+
}
|
|
82
|
+
if (chunk.usage) {
|
|
83
|
+
inputTokens = chunk.usage.prompt_tokens || 0;
|
|
84
|
+
outputTokens = chunk.usage.completion_tokens || 0;
|
|
85
|
+
}
|
|
86
|
+
if (chunk.choices[0]?.finish_reason) {
|
|
87
|
+
yield {
|
|
88
|
+
type: 'done',
|
|
89
|
+
usage: { inputTokens, outputTokens },
|
|
90
|
+
};
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
catch (error) {
|
|
95
|
+
yield {
|
|
96
|
+
type: 'error',
|
|
97
|
+
error: error instanceof Error ? error.message : 'Unknown error',
|
|
98
|
+
};
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
prepareMessages(messages, options) {
|
|
102
|
+
const openaiMessages = [];
|
|
103
|
+
if (options?.systemPrompt) {
|
|
104
|
+
openaiMessages.push({ role: 'system', content: options.systemPrompt });
|
|
105
|
+
}
|
|
106
|
+
for (const msg of messages) {
|
|
107
|
+
openaiMessages.push({ role: msg.role, content: msg.content });
|
|
108
|
+
}
|
|
109
|
+
return openaiMessages;
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
//# sourceMappingURL=openai-compatible.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-compatible.js","sourceRoot":"","sources":["../src/openai-compatible.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAU5B,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAUhC,MAAM,OAAO,wBAAwB;IACnC,IAAI,CAAS;IACb,QAAQ,CAAmB;IACnB,MAAM,CAAS;IACf,YAAY,CAAS;IACrB,gBAAgB,CAAS;IAEjC,YAAY,OAAwC;QAClD,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,mBAAmB,CAAC;QAChD,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC;QAClC,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,SAAS,IAAI,kBAAkB,CAAC;QAEhE,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,YAAY;YACtC,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,GAAG;YACd,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,sBAAsB,CAAC,CAAC,CAAC,mBAAmB;YACvF,MAAM,EAAE;gBACN,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE;oBACnB,gBAAgB,EAAE,MAAM;oBACxB,cAAc,EAAE,KAAK;oBACrB,aAAa,EAAE,IAAI;oBACnB,iBAAiB,EAAE,IAAI;oBACvB,gBAAgB,EAAE,KAAK;oBACvB,cAAc,EAAE,CAAC;oBACjB,eAAe,EAAE,CAAC;oBAClB,SAAS,EAAE,CAAC,MAAM,EAAE,WAAW,CAAC;oBAChC,OAAO,EAAE,IAAI;iBACd;aACF;YACD,WAAW,EAAE,KAAK,IAAI,EAAE;gBACtB,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;oBAChC,OAAO,IAAI,CAAC;gBACd,CAAC;gBAAC,MAAM,CAAC;oBACP,OAAO,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,QAAuB,EACvB,OAA2B;QAE3B,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAE/D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACzD,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY;YAC1C,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,gBAAgB;YACvD,QAAQ,EAAE,cAAc;YACxB,WAAW,EAAE,OAAO,EAAE,WAAW;SAClC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAE/C,OAAO;YACL,OAAO;YACP,KAAK,EAAE;gBACL,WAAW,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;gBAC/C,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;aACrD;YACD,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,YAAY,EAAE,MAAM,EAAE,aAAa,IAAI,SAAS;SACjD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,QAAuB,EACvB,OAA2B;QAE3B,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAE/D,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACvD,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY;gBAC1C,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,gBAAgB;gBACvD,QAAQ,EAAE,cAAc;gBACxB,WAAW,EAAE,OAAO,EAAE,WAAW;gBACjC,MAAM,EAAE,IAAI;gBACZ,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;aACxC,CAAC,CAAC;YAEH,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,IAAI,YAAY,GAAG,CAAC,CAAC;YAErB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;gBAEtC,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;oBACnB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;gBACjD,CAAC;gBAED,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBAChB,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;oBAC7C,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC;gBACpD,CAAC;gBAED,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC;oBACpC,MAAM;wBACJ,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE;qBACrC,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM;gBACJ,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,eAAe,CACrB,QAAuB,EACvB,OAA2B;QAE3B,MAAM,cAAc,GAAwC,EAAE,CAAC;QAE/D,IAAI,OAAO,EAAE,YAAY,EAAE,CAAC;YAC1B,cAAc,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;QACzE,CAAC;QAED,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,cAAc,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAChE,CAAC;QAED,OAAO,cAAc,CAAC;IACxB,CAAC;CACF"}
|
package/dist/openai.d.ts
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { Provider, ProviderMetadata, ChatMessage, CompletionOptions, CompletionResult, StreamChunk } from './types.js';
|
|
2
|
+
export interface OpenAIProviderOptions {
|
|
3
|
+
apiKey: string;
|
|
4
|
+
model?: string;
|
|
5
|
+
maxTokens?: number;
|
|
6
|
+
baseURL?: string;
|
|
7
|
+
}
|
|
8
|
+
export declare class OpenAIProvider implements Provider {
|
|
9
|
+
name: string;
|
|
10
|
+
metadata: ProviderMetadata;
|
|
11
|
+
private client;
|
|
12
|
+
private defaultModel;
|
|
13
|
+
private defaultMaxTokens;
|
|
14
|
+
constructor(options: OpenAIProviderOptions);
|
|
15
|
+
complete(messages: ChatMessage[], options?: CompletionOptions): Promise<CompletionResult>;
|
|
16
|
+
stream(messages: ChatMessage[], options?: CompletionOptions): AsyncGenerator<StreamChunk, void, unknown>;
|
|
17
|
+
private transformTools;
|
|
18
|
+
private prepareMessages;
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=openai.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../src/openai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,QAAQ,EACR,gBAAgB,EAChB,WAAW,EACX,iBAAiB,EACjB,gBAAgB,EAChB,WAAW,EAEZ,MAAM,YAAY,CAAC;AAMpB,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,cAAe,YAAW,QAAQ;IAC7C,IAAI,SAAY;IAChB,QAAQ,EAAE,gBAAgB,CA8ExB;IACF,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,gBAAgB,CAAS;gBAErB,OAAO,EAAE,qBAAqB;IASpC,QAAQ,CACZ,QAAQ,EAAE,WAAW,EAAE,EACvB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,gBAAgB,CAAC;IA2DrB,MAAM,CACX,QAAQ,EAAE,WAAW,EAAE,EACvB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,OAAO,CAAC;IA0F7C,OAAO,CAAC,cAAc;IAWtB,OAAO,CAAC,eAAe;CAwBxB"}
|
package/dist/openai.js
ADDED
|
@@ -0,0 +1,259 @@
|
|
|
1
|
+
import OpenAI from 'openai';
|
|
2
|
+
import { getOpenAIReasoningEffort, isOpenAIReasoningModel } from './thinking-levels.js';
|
|
3
|
+
const DEFAULT_MODEL = 'gpt-5.2';
|
|
4
|
+
const DEFAULT_MAX_TOKENS = 4096;
|
|
5
|
+
export class OpenAIProvider {
|
|
6
|
+
name = 'openai';
|
|
7
|
+
metadata = {
|
|
8
|
+
name: 'openai',
|
|
9
|
+
displayName: 'OpenAI GPT',
|
|
10
|
+
models: {
|
|
11
|
+
'gpt-5.2': {
|
|
12
|
+
maxContextTokens: 1048576,
|
|
13
|
+
supportsVision: true,
|
|
14
|
+
supportsTools: true,
|
|
15
|
+
supportsStreaming: true,
|
|
16
|
+
supportsImageGen: false,
|
|
17
|
+
costPer1kInput: 0.003,
|
|
18
|
+
costPer1kOutput: 0.012,
|
|
19
|
+
strengths: ['reasoning', 'code', 'vision', 'creative', 'agentic'],
|
|
20
|
+
isLocal: false,
|
|
21
|
+
},
|
|
22
|
+
'gpt-5.2-pro': {
|
|
23
|
+
maxContextTokens: 1048576,
|
|
24
|
+
supportsVision: true,
|
|
25
|
+
supportsTools: true,
|
|
26
|
+
supportsStreaming: true,
|
|
27
|
+
supportsImageGen: false,
|
|
28
|
+
costPer1kInput: 0.01,
|
|
29
|
+
costPer1kOutput: 0.04,
|
|
30
|
+
strengths: ['reasoning', 'code', 'precision', 'agentic'],
|
|
31
|
+
isLocal: false,
|
|
32
|
+
},
|
|
33
|
+
'gpt-5': {
|
|
34
|
+
maxContextTokens: 1048576,
|
|
35
|
+
supportsVision: true,
|
|
36
|
+
supportsTools: true,
|
|
37
|
+
supportsStreaming: true,
|
|
38
|
+
supportsImageGen: false,
|
|
39
|
+
costPer1kInput: 0.0025,
|
|
40
|
+
costPer1kOutput: 0.01,
|
|
41
|
+
strengths: ['reasoning', 'code', 'agentic'],
|
|
42
|
+
isLocal: false,
|
|
43
|
+
},
|
|
44
|
+
'gpt-5-mini': {
|
|
45
|
+
maxContextTokens: 1048576,
|
|
46
|
+
supportsVision: true,
|
|
47
|
+
supportsTools: true,
|
|
48
|
+
supportsStreaming: true,
|
|
49
|
+
supportsImageGen: false,
|
|
50
|
+
costPer1kInput: 0.0004,
|
|
51
|
+
costPer1kOutput: 0.0016,
|
|
52
|
+
strengths: ['fast', 'code', 'cost-efficient'],
|
|
53
|
+
isLocal: false,
|
|
54
|
+
},
|
|
55
|
+
'gpt-5-nano': {
|
|
56
|
+
maxContextTokens: 1048576,
|
|
57
|
+
supportsVision: true,
|
|
58
|
+
supportsTools: true,
|
|
59
|
+
supportsStreaming: true,
|
|
60
|
+
supportsImageGen: false,
|
|
61
|
+
costPer1kInput: 0.0001,
|
|
62
|
+
costPer1kOutput: 0.0004,
|
|
63
|
+
strengths: ['fast', 'cost-efficient'],
|
|
64
|
+
isLocal: false,
|
|
65
|
+
},
|
|
66
|
+
'gpt-4.1': {
|
|
67
|
+
maxContextTokens: 1048576,
|
|
68
|
+
supportsVision: true,
|
|
69
|
+
supportsTools: true,
|
|
70
|
+
supportsStreaming: true,
|
|
71
|
+
supportsImageGen: false,
|
|
72
|
+
costPer1kInput: 0.002,
|
|
73
|
+
costPer1kOutput: 0.008,
|
|
74
|
+
strengths: ['code', 'instruction-following', 'long-context'],
|
|
75
|
+
isLocal: false,
|
|
76
|
+
},
|
|
77
|
+
},
|
|
78
|
+
isAvailable: async () => {
|
|
79
|
+
try {
|
|
80
|
+
return this.client !== undefined;
|
|
81
|
+
}
|
|
82
|
+
catch {
|
|
83
|
+
return false;
|
|
84
|
+
}
|
|
85
|
+
},
|
|
86
|
+
};
|
|
87
|
+
client;
|
|
88
|
+
defaultModel;
|
|
89
|
+
defaultMaxTokens;
|
|
90
|
+
constructor(options) {
|
|
91
|
+
this.client = new OpenAI({
|
|
92
|
+
apiKey: options.apiKey,
|
|
93
|
+
baseURL: options.baseURL,
|
|
94
|
+
});
|
|
95
|
+
this.defaultModel = options.model || DEFAULT_MODEL;
|
|
96
|
+
this.defaultMaxTokens = options.maxTokens || DEFAULT_MAX_TOKENS;
|
|
97
|
+
}
|
|
98
|
+
async complete(messages, options) {
|
|
99
|
+
const openaiMessages = this.prepareMessages(messages, options);
|
|
100
|
+
const model = options?.model || this.defaultModel;
|
|
101
|
+
const createParams = {
|
|
102
|
+
model,
|
|
103
|
+
max_tokens: options?.maxTokens || this.defaultMaxTokens,
|
|
104
|
+
messages: openaiMessages,
|
|
105
|
+
};
|
|
106
|
+
// Add tools if provided
|
|
107
|
+
if (options?.tools && options.tools.length > 0) {
|
|
108
|
+
createParams.tools = this.transformTools(options.tools);
|
|
109
|
+
}
|
|
110
|
+
// Add reasoning_effort for o-series models
|
|
111
|
+
if (options?.thinkingLevel && isOpenAIReasoningModel(model)) {
|
|
112
|
+
const effort = getOpenAIReasoningEffort(options.thinkingLevel);
|
|
113
|
+
if (effort) {
|
|
114
|
+
createParams.reasoning_effort = effort;
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
const response = await this.client.chat.completions.create(createParams);
|
|
118
|
+
const choice = response.choices[0];
|
|
119
|
+
const message = choice?.message;
|
|
120
|
+
const content = message?.content || '';
|
|
121
|
+
// Handle tool calls in the response
|
|
122
|
+
const toolCalls = message?.tool_calls;
|
|
123
|
+
if (toolCalls && toolCalls.length > 0) {
|
|
124
|
+
return {
|
|
125
|
+
content,
|
|
126
|
+
toolUse: toolCalls.map((tc) => ({
|
|
127
|
+
id: tc.id,
|
|
128
|
+
name: tc.function.name,
|
|
129
|
+
input: JSON.parse(tc.function.arguments || '{}'),
|
|
130
|
+
})),
|
|
131
|
+
usage: {
|
|
132
|
+
inputTokens: response.usage?.prompt_tokens || 0,
|
|
133
|
+
outputTokens: response.usage?.completion_tokens || 0,
|
|
134
|
+
},
|
|
135
|
+
model: response.model,
|
|
136
|
+
finishReason: 'tool_use',
|
|
137
|
+
};
|
|
138
|
+
}
|
|
139
|
+
return {
|
|
140
|
+
content,
|
|
141
|
+
usage: {
|
|
142
|
+
inputTokens: response.usage?.prompt_tokens || 0,
|
|
143
|
+
outputTokens: response.usage?.completion_tokens || 0,
|
|
144
|
+
},
|
|
145
|
+
model: response.model,
|
|
146
|
+
finishReason: choice?.finish_reason || 'unknown',
|
|
147
|
+
};
|
|
148
|
+
}
|
|
149
|
+
async *stream(messages, options) {
|
|
150
|
+
const openaiMessages = this.prepareMessages(messages, options);
|
|
151
|
+
try {
|
|
152
|
+
const model = options?.model || this.defaultModel;
|
|
153
|
+
const createParams = {
|
|
154
|
+
model,
|
|
155
|
+
max_tokens: options?.maxTokens || this.defaultMaxTokens,
|
|
156
|
+
messages: openaiMessages,
|
|
157
|
+
stream: true,
|
|
158
|
+
stream_options: { include_usage: true },
|
|
159
|
+
};
|
|
160
|
+
// Add tools if provided
|
|
161
|
+
if (options?.tools && options.tools.length > 0) {
|
|
162
|
+
createParams.tools = this.transformTools(options.tools);
|
|
163
|
+
}
|
|
164
|
+
// Add reasoning_effort for o-series models
|
|
165
|
+
if (options?.thinkingLevel && isOpenAIReasoningModel(model)) {
|
|
166
|
+
const effort = getOpenAIReasoningEffort(options.thinkingLevel);
|
|
167
|
+
if (effort) {
|
|
168
|
+
createParams.reasoning_effort = effort;
|
|
169
|
+
}
|
|
170
|
+
}
|
|
171
|
+
const stream = await this.client.chat.completions.create(createParams);
|
|
172
|
+
let inputTokens = 0;
|
|
173
|
+
let outputTokens = 0;
|
|
174
|
+
// Track streaming tool calls: index -> { id, name, arguments }
|
|
175
|
+
const toolCallAccumulators = new Map();
|
|
176
|
+
for await (const chunk of stream) {
|
|
177
|
+
const choice = chunk.choices[0];
|
|
178
|
+
const delta = choice?.delta;
|
|
179
|
+
if (delta?.content) {
|
|
180
|
+
yield { type: 'text', content: delta.content };
|
|
181
|
+
}
|
|
182
|
+
// Handle streaming tool_calls deltas
|
|
183
|
+
if (delta?.tool_calls) {
|
|
184
|
+
for (const tc of delta.tool_calls) {
|
|
185
|
+
const idx = tc.index;
|
|
186
|
+
if (!toolCallAccumulators.has(idx)) {
|
|
187
|
+
toolCallAccumulators.set(idx, { id: '', name: '', arguments: '' });
|
|
188
|
+
}
|
|
189
|
+
const acc = toolCallAccumulators.get(idx);
|
|
190
|
+
if (tc.id)
|
|
191
|
+
acc.id = tc.id;
|
|
192
|
+
if (tc.function?.name)
|
|
193
|
+
acc.name = tc.function.name;
|
|
194
|
+
if (tc.function?.arguments)
|
|
195
|
+
acc.arguments += tc.function.arguments;
|
|
196
|
+
}
|
|
197
|
+
}
|
|
198
|
+
if (chunk.usage) {
|
|
199
|
+
inputTokens = chunk.usage.prompt_tokens || 0;
|
|
200
|
+
outputTokens = chunk.usage.completion_tokens || 0;
|
|
201
|
+
}
|
|
202
|
+
if (choice?.finish_reason) {
|
|
203
|
+
// Yield accumulated tool calls before done
|
|
204
|
+
if (choice.finish_reason === 'tool_calls' && toolCallAccumulators.size > 0) {
|
|
205
|
+
for (const [, acc] of toolCallAccumulators) {
|
|
206
|
+
yield {
|
|
207
|
+
type: 'tool_use',
|
|
208
|
+
toolUse: {
|
|
209
|
+
id: acc.id,
|
|
210
|
+
name: acc.name,
|
|
211
|
+
input: JSON.parse(acc.arguments || '{}'),
|
|
212
|
+
},
|
|
213
|
+
};
|
|
214
|
+
}
|
|
215
|
+
}
|
|
216
|
+
yield {
|
|
217
|
+
type: 'done',
|
|
218
|
+
usage: { inputTokens, outputTokens },
|
|
219
|
+
};
|
|
220
|
+
}
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
catch (error) {
|
|
224
|
+
yield {
|
|
225
|
+
type: 'error',
|
|
226
|
+
error: error instanceof Error ? error.message : 'Unknown error',
|
|
227
|
+
};
|
|
228
|
+
}
|
|
229
|
+
}
|
|
230
|
+
transformTools(tools) {
|
|
231
|
+
return tools.map((tool) => ({
|
|
232
|
+
type: 'function',
|
|
233
|
+
function: {
|
|
234
|
+
name: tool.name,
|
|
235
|
+
description: tool.description,
|
|
236
|
+
parameters: tool.input_schema,
|
|
237
|
+
},
|
|
238
|
+
}));
|
|
239
|
+
}
|
|
240
|
+
prepareMessages(messages, options) {
|
|
241
|
+
const openaiMessages = [];
|
|
242
|
+
// Add system prompt if provided
|
|
243
|
+
if (options?.systemPrompt) {
|
|
244
|
+
openaiMessages.push({
|
|
245
|
+
role: 'system',
|
|
246
|
+
content: options.systemPrompt,
|
|
247
|
+
});
|
|
248
|
+
}
|
|
249
|
+
// Convert messages
|
|
250
|
+
for (const msg of messages) {
|
|
251
|
+
openaiMessages.push({
|
|
252
|
+
role: msg.role,
|
|
253
|
+
content: msg.content,
|
|
254
|
+
});
|
|
255
|
+
}
|
|
256
|
+
return openaiMessages;
|
|
257
|
+
}
|
|
258
|
+
}
|
|
259
|
+
//# sourceMappingURL=openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../src/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAU5B,OAAO,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAExF,MAAM,aAAa,GAAG,SAAS,CAAC;AAChC,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAShC,MAAM,OAAO,cAAc;IACzB,IAAI,GAAG,QAAQ,CAAC;IAChB,QAAQ,GAAqB;QAC3B,IAAI,EAAE,QAAQ;QACd,WAAW,EAAE,YAAY;QACzB,MAAM,EAAE;YACN,SAAS,EAAE;gBACT,gBAAgB,EAAE,OAAO;gBACzB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,IAAI;gBACnB,iBAAiB,EAAE,IAAI;gBACvB,gBAAgB,EAAE,KAAK;gBACvB,cAAc,EAAE,KAAK;gBACrB,eAAe,EAAE,KAAK;gBACtB,SAAS,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,CAAC;gBACjE,OAAO,EAAE,KAAK;aACf;YACD,aAAa,EAAE;gBACb,gBAAgB,EAAE,OAAO;gBACzB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,IAAI;gBACnB,iBAAiB,EAAE,IAAI;gBACvB,gBAAgB,EAAE,KAAK;gBACvB,cAAc,EAAE,IAAI;gBACpB,eAAe,EAAE,IAAI;gBACrB,SAAS,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,CAAC;gBACxD,OAAO,EAAE,KAAK;aACf;YACD,OAAO,EAAE;gBACP,gBAAgB,EAAE,OAAO;gBACzB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,IAAI;gBACnB,iBAAiB,EAAE,IAAI;gBACvB,gBAAgB,EAAE,KAAK;gBACvB,cAAc,EAAE,MAAM;gBACtB,eAAe,EAAE,IAAI;gBACrB,SAAS,EAAE,CAAC,WAAW,EAAE,MAAM,EAAE,SAAS,CAAC;gBAC3C,OAAO,EAAE,KAAK;aACf;YACD,YAAY,EAAE;gBACZ,gBAAgB,EAAE,OAAO;gBACzB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,IAAI;gBACnB,iBAAiB,EAAE,IAAI;gBACvB,gBAAgB,EAAE,KAAK;gBACvB,cAAc,EAAE,MAAM;gBACtB,eAAe,EAAE,MAAM;gBACvB,SAAS,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,gBAAgB,CAAC;gBAC7C,OAAO,EAAE,KAAK;aACf;YACD,YAAY,EAAE;gBACZ,gBAAgB,EAAE,OAAO;gBACzB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,IAAI;gBACnB,iBAAiB,EAAE,IAAI;gBACvB,gBAAgB,EAAE,KAAK;gBACvB,cAAc,EAAE,MAAM;gBACtB,eAAe,EAAE,MAAM;gBACvB,SAAS,EAAE,CAAC,MAAM,EAAE,gBAAgB,CAAC;gBACrC,OAAO,EAAE,KAAK;aACf;YACD,SAAS,EAAE;gBACT,gBAAgB,EAAE,OAAO;gBACzB,cAAc,EAAE,IAAI;gBACpB,aAAa,EAAE,IAAI;gBACnB,iBAAiB,EAAE,IAAI;gBACvB,gBAAgB,EAAE,KAAK;gBACvB,cAAc,EAAE,KAAK;gBACrB,eAAe,EAAE,KAAK;gBACtB,SAAS,EAAE,CAAC,MAAM,EAAE,uBAAuB,EAAE,cAAc,CAAC;gBAC5D,OAAO,EAAE,KAAK;aACf;SACF;QACD,WAAW,EAAE,KAAK,IAAI,EAAE;YACtB,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC;YACnC,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;KACF,CAAC;IACM,MAAM,CAAS;IACf,YAAY,CAAS;IACrB,gBAAgB,CAAS;IAEjC,YAAY,OAA8B;QACxC,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,KAAK,IAAI,aAAa,CAAC;QACnD,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,SAAS,IAAI,kBAAkB,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,QAAuB,EACvB,OAA2B;QAE3B,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAE/D,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QAClD,MAAM,YAAY,GAAsC;YACtD,KAAK;YACL,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,gBAAgB;YACvD,QAAQ,EAAE,cAAc;SACzB,CAAC;QAEF,wBAAwB;QACxB,IAAI,OAAO,EAAE,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/C,YAAY,CAAC,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC1D,CAAC;QAED,2CAA2C;QAC3C,IAAI,OAAO,EAAE,aAAa,IAAI,sBAAsB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC5D,MAAM,MAAM,GAAG,wBAAwB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAC/D,IAAI,MAAM,EAAE,CAAC;gBACV,YAAoB,CAAC,gBAAgB,GAAG,MAAM,CAAC;YAClD,CAAC;QACH,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAEzE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC;QAChC,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAEvC,oCAAoC;QACpC,MAAM,SAAS,GAAG,OAAO,EAAE,UAAU,CAAC;QACtC,IAAI,SAAS,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,OAAO;gBACL,OAAO;gBACP,OAAO,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;oBAC9B,EAAE,EAAE,EAAE,CAAC,EAAE;oBACT,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI;oBACtB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC;iBACjD,CAAC,CAAC;gBACH,KAAK,EAAE;oBACL,WAAW,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;oBAC/C,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;iBACrD;gBACD,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,YAAY,EAAE,UAAU;aACzB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO;YACP,KAAK,EAAE;gBACL,WAAW,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;gBAC/C,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;aACrD;YACD,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,YAAY,EAAE,MAAM,EAAE,aAAa,IAAI,SAAS;SACjD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,QAAuB,EACvB,OAA2B;QAE3B,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAE/D,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;YAClD,MAAM,YAAY,GAAsC;gBACtD,KAAK;gBACL,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC,gBAAgB;gBACvD,QAAQ,EAAE,cAAc;gBACxB,MAAM,EAAE,IAAI;gBACZ,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;aACxC,CAAC;YAEF,wBAAwB;YACxB,IAAI,OAAO,EAAE,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/C,YAAY,CAAC,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC1D,CAAC;YAED,2CAA2C;YAC3C,IAAI,OAAO,EAAE,aAAa,IAAI,sBAAsB,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5D,MAAM,MAAM,GAAG,wBAAwB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;gBAC/D,IAAI,MAAM,EAAE,CAAC;oBACV,YAAoB,CAAC,gBAAgB,GAAG,MAAM,CAAC;gBAClD,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YAEvE,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,IAAI,YAAY,GAAG,CAAC,CAAC;YAErB,+DAA+D;YAC/D,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAA2D,CAAC;YAEhG,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAChC,MAAM,KAAK,GAAG,MAAM,EAAE,KAAK,CAAC;gBAE5B,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;oBACnB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;gBACjD,CAAC;gBAED,qCAAqC;gBACrC,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;oBACtB,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;wBAClC,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC;wBACrB,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;4BACnC,oBAAoB,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;wBACrE,CAAC;wBACD,MAAM,GAAG,GAAG,oBAAoB,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;wBAC3C,IAAI,EAAE,CAAC,EAAE;4BAAE,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;wBAC1B,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI;4BAAE,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;wBACnD,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS;4BAAE,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;oBACrE,CAAC;gBACH,CAAC;gBAED,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBAChB,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;oBAC7C,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC;gBACpD,CAAC;gBAED,IAAI,MAAM,EAAE,aAAa,EAAE,CAAC;oBAC1B,2CAA2C;oBAC3C,IAAI,MAAM,CAAC,aAAa,KAAK,YAAY,IAAI,oBAAoB,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;wBAC3E,KAAK,MAAM,CAAC,EAAE,GAAG,CAAC,IAAI,oBAAoB,EAAE,CAAC;4BAC3C,MAAM;gCACJ,IAAI,EAAE,UAAU;gCAChB,OAAO,EAAE;oCACP,EAAE,EAAE,GAAG,CAAC,EAAE;oCACV,IAAI,EAAE,GAAG,CAAC,IAAI;oCACd,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,IAAI,IAAI,CAAC;iCACzC;6BACF,CAAC;wBACJ,CAAC;oBACH,CAAC;oBAED,MAAM;wBACJ,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE;qBACrC,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM;gBACJ,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,cAAc,CAAC,KAAuB;QAC5C,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,YAAyC;aAC3D;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAEO,eAAe,CACrB,QAAuB,EACvB,OAA2B;QAE3B,MAAM,cAAc,GAAwC,EAAE,CAAC;QAE/D,gCAAgC;QAChC,IAAI,OAAO,EAAE,YAAY,EAAE,CAAC;YAC1B,cAAc,CAAC,IAAI,CAAC;gBAClB,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,OAAO,CAAC,YAAY;aAC9B,CAAC,CAAC;QACL,CAAC;QAED,mBAAmB;QACnB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,cAAc,CAAC,IAAI,CAAC;gBAClB,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,OAAO,EAAE,GAAG,CAAC,OAAO;aACrB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,cAAc,CAAC;IACxB,CAAC;CACF"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { Provider, ProviderMetadata, ChatMessage, CompletionOptions, CompletionResult, StreamChunk } from './types.js';
|
|
2
|
+
export interface ReplicateProviderOptions {
|
|
3
|
+
apiToken: string;
|
|
4
|
+
model?: string;
|
|
5
|
+
pollInterval?: number;
|
|
6
|
+
}
|
|
7
|
+
export declare class ReplicateProvider implements Provider {
|
|
8
|
+
name: string;
|
|
9
|
+
metadata: ProviderMetadata;
|
|
10
|
+
private apiToken;
|
|
11
|
+
private defaultModel;
|
|
12
|
+
private pollInterval;
|
|
13
|
+
constructor(options: ReplicateProviderOptions);
|
|
14
|
+
complete(messages: ChatMessage[], options?: CompletionOptions): Promise<CompletionResult>;
|
|
15
|
+
stream(messages: ChatMessage[], options?: CompletionOptions): AsyncGenerator<StreamChunk, void, unknown>;
|
|
16
|
+
private createPrediction;
|
|
17
|
+
private pollPrediction;
|
|
18
|
+
private formatPrompt;
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=replicate.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"replicate.d.ts","sourceRoot":"","sources":["../src/replicate.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,QAAQ,EACR,gBAAgB,EAChB,WAAW,EACX,iBAAiB,EACjB,gBAAgB,EAChB,WAAW,EACZ,MAAM,YAAY,CAAC;AAMpB,MAAM,WAAW,wBAAwB;IACvC,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAYD,qBAAa,iBAAkB,YAAW,QAAQ;IAChD,IAAI,SAAe;IACnB,QAAQ,EAAE,gBAAgB,CAqCxB;IAEF,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,YAAY,CAAS;gBAEjB,OAAO,EAAE,wBAAwB;IAMvC,QAAQ,CACZ,QAAQ,EAAE,WAAW,EAAE,EACvB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,gBAAgB,CAAC;IA+BrB,MAAM,CACX,QAAQ,EAAE,WAAW,EAAE,EACvB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,OAAO,CAAC;YAyE/B,gBAAgB;YAqBhB,cAAc;IAoB5B,OAAO,CAAC,YAAY;CAerB"}
|
|
@@ -0,0 +1,186 @@
|
|
|
1
|
+
const DEFAULT_MODEL = 'meta/meta-llama-3-70b-instruct';
|
|
2
|
+
const DEFAULT_POLL_INTERVAL = 1000;
|
|
3
|
+
const BASE_URL = 'https://api.replicate.com/v1';
|
|
4
|
+
export class ReplicateProvider {
|
|
5
|
+
name = 'replicate';
|
|
6
|
+
metadata = {
|
|
7
|
+
name: 'replicate',
|
|
8
|
+
displayName: 'Replicate',
|
|
9
|
+
models: {
|
|
10
|
+
'meta/meta-llama-3-70b-instruct': {
|
|
11
|
+
maxContextTokens: 8192,
|
|
12
|
+
supportsVision: false,
|
|
13
|
+
supportsTools: false,
|
|
14
|
+
supportsStreaming: false,
|
|
15
|
+
supportsImageGen: false,
|
|
16
|
+
costPer1kInput: 0.00065,
|
|
17
|
+
costPer1kOutput: 0.00275,
|
|
18
|
+
strengths: ['reasoning', 'code'],
|
|
19
|
+
isLocal: false,
|
|
20
|
+
},
|
|
21
|
+
'stability-ai/sdxl': {
|
|
22
|
+
maxContextTokens: 0,
|
|
23
|
+
supportsVision: false,
|
|
24
|
+
supportsTools: false,
|
|
25
|
+
supportsStreaming: false,
|
|
26
|
+
supportsImageGen: true,
|
|
27
|
+
costPer1kInput: 0,
|
|
28
|
+
costPer1kOutput: 0,
|
|
29
|
+
strengths: ['image-generation'],
|
|
30
|
+
isLocal: false,
|
|
31
|
+
},
|
|
32
|
+
},
|
|
33
|
+
isAvailable: async () => {
|
|
34
|
+
try {
|
|
35
|
+
const response = await fetch(`${BASE_URL}/models`, {
|
|
36
|
+
headers: { Authorization: `Bearer ${this.apiToken}` },
|
|
37
|
+
});
|
|
38
|
+
return response.ok;
|
|
39
|
+
}
|
|
40
|
+
catch {
|
|
41
|
+
return false;
|
|
42
|
+
}
|
|
43
|
+
},
|
|
44
|
+
};
|
|
45
|
+
apiToken;
|
|
46
|
+
defaultModel;
|
|
47
|
+
pollInterval;
|
|
48
|
+
constructor(options) {
|
|
49
|
+
this.apiToken = options.apiToken;
|
|
50
|
+
this.defaultModel = options.model || DEFAULT_MODEL;
|
|
51
|
+
this.pollInterval = options.pollInterval || DEFAULT_POLL_INTERVAL;
|
|
52
|
+
}
|
|
53
|
+
async complete(messages, options) {
|
|
54
|
+
const model = options?.model || this.defaultModel;
|
|
55
|
+
const prompt = this.formatPrompt(messages, options);
|
|
56
|
+
const prediction = await this.createPrediction(model, {
|
|
57
|
+
prompt,
|
|
58
|
+
max_tokens: options?.maxTokens || 4096,
|
|
59
|
+
temperature: options?.temperature,
|
|
60
|
+
});
|
|
61
|
+
const result = await this.pollPrediction(prediction.id);
|
|
62
|
+
if (result.status === 'failed') {
|
|
63
|
+
throw new Error(`Replicate prediction failed: ${result.error || 'unknown error'}`);
|
|
64
|
+
}
|
|
65
|
+
const output = Array.isArray(result.output)
|
|
66
|
+
? result.output.join('')
|
|
67
|
+
: result.output || '';
|
|
68
|
+
return {
|
|
69
|
+
content: output,
|
|
70
|
+
usage: {
|
|
71
|
+
inputTokens: 0,
|
|
72
|
+
outputTokens: 0,
|
|
73
|
+
},
|
|
74
|
+
model,
|
|
75
|
+
finishReason: 'stop',
|
|
76
|
+
};
|
|
77
|
+
}
|
|
78
|
+
async *stream(messages, options) {
|
|
79
|
+
try {
|
|
80
|
+
const model = options?.model || this.defaultModel;
|
|
81
|
+
const prompt = this.formatPrompt(messages, options);
|
|
82
|
+
const response = await fetch(`${BASE_URL}/models/${model}/predictions`, {
|
|
83
|
+
method: 'POST',
|
|
84
|
+
headers: {
|
|
85
|
+
'Content-Type': 'application/json',
|
|
86
|
+
Authorization: `Bearer ${this.apiToken}`,
|
|
87
|
+
Prefer: 'respond-async',
|
|
88
|
+
},
|
|
89
|
+
body: JSON.stringify({
|
|
90
|
+
input: {
|
|
91
|
+
prompt,
|
|
92
|
+
max_tokens: options?.maxTokens || 4096,
|
|
93
|
+
temperature: options?.temperature,
|
|
94
|
+
},
|
|
95
|
+
stream: true,
|
|
96
|
+
}),
|
|
97
|
+
});
|
|
98
|
+
if (!response.ok) {
|
|
99
|
+
throw new Error(`Replicate API error: ${response.status} ${response.statusText}`);
|
|
100
|
+
}
|
|
101
|
+
const prediction = (await response.json());
|
|
102
|
+
const streamUrl = `${BASE_URL}/predictions/${prediction.id}/stream`;
|
|
103
|
+
const streamResponse = await fetch(streamUrl, {
|
|
104
|
+
headers: { Authorization: `Bearer ${this.apiToken}`, Accept: 'text/event-stream' },
|
|
105
|
+
});
|
|
106
|
+
if (!streamResponse.ok || !streamResponse.body) {
|
|
107
|
+
const result = await this.pollPrediction(prediction.id);
|
|
108
|
+
const output = Array.isArray(result.output)
|
|
109
|
+
? result.output.join('')
|
|
110
|
+
: result.output || '';
|
|
111
|
+
yield { type: 'text', content: output };
|
|
112
|
+
yield { type: 'done', usage: { inputTokens: 0, outputTokens: 0 } };
|
|
113
|
+
return;
|
|
114
|
+
}
|
|
115
|
+
const reader = streamResponse.body.getReader();
|
|
116
|
+
const decoder = new TextDecoder();
|
|
117
|
+
let buffer = '';
|
|
118
|
+
while (true) {
|
|
119
|
+
const { done, value } = await reader.read();
|
|
120
|
+
if (done)
|
|
121
|
+
break;
|
|
122
|
+
buffer += decoder.decode(value, { stream: true });
|
|
123
|
+
const lines = buffer.split('\n');
|
|
124
|
+
buffer = lines.pop() || '';
|
|
125
|
+
for (const line of lines) {
|
|
126
|
+
const trimmed = line.trim();
|
|
127
|
+
if (!trimmed || !trimmed.startsWith('data: '))
|
|
128
|
+
continue;
|
|
129
|
+
const data = trimmed.slice(6);
|
|
130
|
+
if (data === '[DONE]')
|
|
131
|
+
continue;
|
|
132
|
+
yield { type: 'text', content: data };
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
yield { type: 'done', usage: { inputTokens: 0, outputTokens: 0 } };
|
|
136
|
+
}
|
|
137
|
+
catch (error) {
|
|
138
|
+
yield {
|
|
139
|
+
type: 'error',
|
|
140
|
+
error: error instanceof Error ? error.message : 'Unknown error',
|
|
141
|
+
};
|
|
142
|
+
}
|
|
143
|
+
}
|
|
144
|
+
async createPrediction(model, input) {
|
|
145
|
+
const response = await fetch(`${BASE_URL}/models/${model}/predictions`, {
|
|
146
|
+
method: 'POST',
|
|
147
|
+
headers: {
|
|
148
|
+
'Content-Type': 'application/json',
|
|
149
|
+
Authorization: `Bearer ${this.apiToken}`,
|
|
150
|
+
Prefer: 'respond-async',
|
|
151
|
+
},
|
|
152
|
+
body: JSON.stringify({ input }),
|
|
153
|
+
});
|
|
154
|
+
if (!response.ok) {
|
|
155
|
+
throw new Error(`Replicate API error: ${response.status} ${response.statusText}`);
|
|
156
|
+
}
|
|
157
|
+
return (await response.json());
|
|
158
|
+
}
|
|
159
|
+
async pollPrediction(id) {
|
|
160
|
+
while (true) {
|
|
161
|
+
const response = await fetch(`${BASE_URL}/predictions/${id}`, {
|
|
162
|
+
headers: { Authorization: `Bearer ${this.apiToken}` },
|
|
163
|
+
});
|
|
164
|
+
if (!response.ok) {
|
|
165
|
+
throw new Error(`Replicate API error: ${response.status} ${response.statusText}`);
|
|
166
|
+
}
|
|
167
|
+
const prediction = (await response.json());
|
|
168
|
+
if (prediction.status === 'succeeded' || prediction.status === 'failed' || prediction.status === 'canceled') {
|
|
169
|
+
return prediction;
|
|
170
|
+
}
|
|
171
|
+
await new Promise((resolve) => setTimeout(resolve, this.pollInterval));
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
formatPrompt(messages, options) {
|
|
175
|
+
const parts = [];
|
|
176
|
+
if (options?.systemPrompt) {
|
|
177
|
+
parts.push(`<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\n${options.systemPrompt}<|eot_id|>`);
|
|
178
|
+
}
|
|
179
|
+
for (const msg of messages) {
|
|
180
|
+
parts.push(`<|start_header_id|>${msg.role}<|end_header_id|>\n\n${msg.content}<|eot_id|>`);
|
|
181
|
+
}
|
|
182
|
+
parts.push('<|start_header_id|>assistant<|end_header_id|>\n\n');
|
|
183
|
+
return parts.join('');
|
|
184
|
+
}
|
|
185
|
+
}
|
|
186
|
+
//# sourceMappingURL=replicate.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"replicate.js","sourceRoot":"","sources":["../src/replicate.ts"],"names":[],"mappings":"AASA,MAAM,aAAa,GAAG,gCAAgC,CAAC;AACvD,MAAM,qBAAqB,GAAG,IAAI,CAAC;AACnC,MAAM,QAAQ,GAAG,8BAA8B,CAAC;AAkBhD,MAAM,OAAO,iBAAiB;IAC5B,IAAI,GAAG,WAAW,CAAC;IACnB,QAAQ,GAAqB;QAC3B,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,WAAW;QACxB,MAAM,EAAE;YACN,gCAAgC,EAAE;gBAChC,gBAAgB,EAAE,IAAI;gBACtB,cAAc,EAAE,KAAK;gBACrB,aAAa,EAAE,KAAK;gBACpB,iBAAiB,EAAE,KAAK;gBACxB,gBAAgB,EAAE,KAAK;gBACvB,cAAc,EAAE,OAAO;gBACvB,eAAe,EAAE,OAAO;gBACxB,SAAS,EAAE,CAAC,WAAW,EAAE,MAAM,CAAC;gBAChC,OAAO,EAAE,KAAK;aACf;YACD,mBAAmB,EAAE;gBACnB,gBAAgB,EAAE,CAAC;gBACnB,cAAc,EAAE,KAAK;gBACrB,aAAa,EAAE,KAAK;gBACpB,iBAAiB,EAAE,KAAK;gBACxB,gBAAgB,EAAE,IAAI;gBACtB,cAAc,EAAE,CAAC;gBACjB,eAAe,EAAE,CAAC;gBAClB,SAAS,EAAE,CAAC,kBAAkB,CAAC;gBAC/B,OAAO,EAAE,KAAK;aACf;SACF;QACD,WAAW,EAAE,KAAK,IAAI,EAAE;YACtB,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,QAAQ,SAAS,EAAE;oBACjD,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,QAAQ,EAAE,EAAE;iBACtD,CAAC,CAAC;gBACH,OAAO,QAAQ,CAAC,EAAE,CAAC;YACrB,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;KACF,CAAC;IAEM,QAAQ,CAAS;IACjB,YAAY,CAAS;IACrB,YAAY,CAAS;IAE7B,YAAY,OAAiC;QAC3C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,KAAK,IAAI,aAAa,CAAC;QACnD,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,qBAAqB,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,QAAuB,EACvB,OAA2B;QAE3B,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QAClD,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAEpD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE;YACpD,MAAM;YACN,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;YACtC,WAAW,EAAE,OAAO,EAAE,WAAW;SAClC,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAExD,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CAAC,gCAAgC,MAAM,CAAC,KAAK,IAAI,eAAe,EAAE,CAAC,CAAC;QACrF,CAAC;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;YACzC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,CAAC,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;QAExB,OAAO;YACL,OAAO,EAAE,MAAM;YACf,KAAK,EAAE;gBACL,WAAW,EAAE,CAAC;gBACd,YAAY,EAAE,CAAC;aAChB;YACD,KAAK;YACL,YAAY,EAAE,MAAM;SACrB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,QAAuB,EACvB,OAA2B;QAE3B,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;YAClD,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAEpD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,QAAQ,WAAW,KAAK,cAAc,EAAE;gBACtE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,QAAQ,EAAE;oBACxC,MAAM,EAAE,eAAe;iBACxB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE;wBACL,MAAM;wBACN,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;wBACtC,WAAW,EAAE,OAAO,EAAE,WAAW;qBAClC;oBACD,MAAM,EAAE,IAAI;iBACb,CAAC;aACH,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACpF,CAAC;YAED,MAAM,UAAU,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAwB,CAAC;YAClE,MAAM,SAAS,GAAG,GAAG,QAAQ,gBAAgB,UAAU,CAAC,EAAE,SAAS,CAAC;YAEpE,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE;gBAC5C,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,mBAAmB,EAAE;aACnF,CAAC,CAAC;YAEH,IAAI,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;gBAC/C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;gBACxD,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;oBACzC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;oBACxB,CAAC,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;gBACxB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;gBACxC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;gBACnE,OAAO;YACT,CAAC;YAED,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;oBAC5B,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;wBAAE,SAAS;oBACxD,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAC9B,IAAI,IAAI,KAAK,QAAQ;wBAAE,SAAS;oBAChC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;gBACxC,CAAC;YACH,CAAC;YAED,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QACrE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM;gBACJ,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,KAAa,EACb,KAA8B;QAE9B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,QAAQ,WAAW,KAAK,cAAc,EAAE;YACtE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,QAAQ,EAAE;gBACxC,MAAM,EAAE,eAAe;aACxB;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,CAAC;SAChC,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACpF,CAAC;QAED,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAwB,CAAC;IACxD,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,EAAU;QACrC,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,QAAQ,gBAAgB,EAAE,EAAE,EAAE;gBAC5D,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,IAAI,CAAC,QAAQ,EAAE,EAAE;aACtD,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACpF,CAAC;YAED,MAAM,UAAU,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAwB,CAAC;YAElE,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW,IAAI,UAAU,CAAC,MAAM,KAAK,QAAQ,IAAI,UAAU,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;gBAC5G,OAAO,UAAU,CAAC;YACpB,CAAC;YAED,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,QAAuB,EAAE,OAA2B;QACvE,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,IAAI,OAAO,EAAE,YAAY,EAAE,CAAC;YAC1B,KAAK,CAAC,IAAI,CAAC,kEAAkE,OAAO,CAAC,YAAY,YAAY,CAAC,CAAC;QACjH,CAAC;QAED,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,sBAAsB,GAAG,CAAC,IAAI,wBAAwB,GAAG,CAAC,OAAO,YAAY,CAAC,CAAC;QAC5F,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,mDAAmD,CAAC,CAAC;QAEhE,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxB,CAAC;CACF"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { ThinkingLevel } from './types.js';
|
|
2
|
+
/**
|
|
3
|
+
* Map ThinkingLevel to Anthropic budget_tokens for the thinking block.
|
|
4
|
+
* Returns undefined for 'off' (no thinking).
|
|
5
|
+
*/
|
|
6
|
+
export declare function getAnthropicThinkingBudget(level: ThinkingLevel): number | undefined;
|
|
7
|
+
/**
|
|
8
|
+
* Map ThinkingLevel to OpenAI reasoning_effort for o-series models.
|
|
9
|
+
* Returns undefined for 'off'.
|
|
10
|
+
*/
|
|
11
|
+
export declare function getOpenAIReasoningEffort(level: ThinkingLevel): 'low' | 'medium' | 'high' | undefined;
|
|
12
|
+
/**
|
|
13
|
+
* Check if an OpenAI model supports reasoning_effort (o-series).
|
|
14
|
+
*/
|
|
15
|
+
export declare function isOpenAIReasoningModel(model: string): boolean;
|
|
16
|
+
//# sourceMappingURL=thinking-levels.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"thinking-levels.d.ts","sourceRoot":"","sources":["../src/thinking-levels.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAEhD;;;GAGG;AACH,wBAAgB,0BAA0B,CAAC,KAAK,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS,CAQnF;AAED;;;GAGG;AACH,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,aAAa,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,GAAG,SAAS,CAQpG;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAG7D"}
|