@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.
Files changed (88) hide show
  1. package/LICENSE +191 -0
  2. package/dist/anthropic.d.ts +82 -0
  3. package/dist/anthropic.d.ts.map +1 -0
  4. package/dist/anthropic.js +618 -0
  5. package/dist/anthropic.js.map +1 -0
  6. package/dist/claude-code-tools.d.ts +29 -0
  7. package/dist/claude-code-tools.d.ts.map +1 -0
  8. package/dist/claude-code-tools.js +221 -0
  9. package/dist/claude-code-tools.js.map +1 -0
  10. package/dist/claude-oauth.d.ts +86 -0
  11. package/dist/claude-oauth.d.ts.map +1 -0
  12. package/dist/claude-oauth.js +318 -0
  13. package/dist/claude-oauth.js.map +1 -0
  14. package/dist/cohere.d.ts +18 -0
  15. package/dist/cohere.d.ts.map +1 -0
  16. package/dist/cohere.js +163 -0
  17. package/dist/cohere.js.map +1 -0
  18. package/dist/deepseek.d.ts +18 -0
  19. package/dist/deepseek.d.ts.map +1 -0
  20. package/dist/deepseek.js +164 -0
  21. package/dist/deepseek.js.map +1 -0
  22. package/dist/factory.d.ts +19 -0
  23. package/dist/factory.d.ts.map +1 -0
  24. package/dist/factory.js +108 -0
  25. package/dist/factory.js.map +1 -0
  26. package/dist/google.d.ts +18 -0
  27. package/dist/google.d.ts.map +1 -0
  28. package/dist/google.js +141 -0
  29. package/dist/google.js.map +1 -0
  30. package/dist/groq.d.ts +18 -0
  31. package/dist/groq.d.ts.map +1 -0
  32. package/dist/groq.js +186 -0
  33. package/dist/groq.js.map +1 -0
  34. package/dist/index.d.ts +15 -0
  35. package/dist/index.d.ts.map +1 -0
  36. package/dist/index.js +14 -0
  37. package/dist/index.js.map +1 -0
  38. package/dist/ollama.d.ts +18 -0
  39. package/dist/ollama.d.ts.map +1 -0
  40. package/dist/ollama.js +141 -0
  41. package/dist/ollama.js.map +1 -0
  42. package/dist/openai-compatible.d.ts +20 -0
  43. package/dist/openai-compatible.d.ts.map +1 -0
  44. package/dist/openai-compatible.js +112 -0
  45. package/dist/openai-compatible.js.map +1 -0
  46. package/dist/openai.d.ts +20 -0
  47. package/dist/openai.d.ts.map +1 -0
  48. package/dist/openai.js +259 -0
  49. package/dist/openai.js.map +1 -0
  50. package/dist/replicate.d.ts +20 -0
  51. package/dist/replicate.d.ts.map +1 -0
  52. package/dist/replicate.js +186 -0
  53. package/dist/replicate.js.map +1 -0
  54. package/dist/thinking-levels.d.ts +16 -0
  55. package/dist/thinking-levels.d.ts.map +1 -0
  56. package/dist/thinking-levels.js +34 -0
  57. package/dist/thinking-levels.js.map +1 -0
  58. package/dist/types.d.ts +157 -0
  59. package/dist/types.d.ts.map +1 -0
  60. package/dist/types.js +2 -0
  61. package/dist/types.js.map +1 -0
  62. package/dist/xai.d.ts +18 -0
  63. package/dist/xai.d.ts.map +1 -0
  64. package/dist/xai.js +164 -0
  65. package/dist/xai.js.map +1 -0
  66. package/package.json +30 -0
  67. package/src/anthropic.ts +691 -0
  68. package/src/claude-code-tools.ts +233 -0
  69. package/src/claude-oauth.ts +410 -0
  70. package/src/cohere.ts +242 -0
  71. package/src/deepseek.ts +241 -0
  72. package/src/factory.ts +142 -0
  73. package/src/google.ts +176 -0
  74. package/src/groq.ts +263 -0
  75. package/src/index.ts +44 -0
  76. package/src/ollama.ts +194 -0
  77. package/src/openai-compatible.ts +154 -0
  78. package/src/openai.ts +307 -0
  79. package/src/replicate.ts +247 -0
  80. package/src/thinking-levels.ts +37 -0
  81. package/src/types.ts +171 -0
  82. package/src/xai.ts +241 -0
  83. package/tests/adapters.test.ts +185 -0
  84. package/tests/claude-oauth.test.ts +45 -0
  85. package/tests/new-providers.test.ts +732 -0
  86. package/tests/thinking-levels.test.ts +82 -0
  87. package/tsconfig.json +8 -0
  88. 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"}
@@ -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"}