cursor-recursive-rag 0.2.0-alpha.2 → 0.2.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/README.md +179 -203
- package/dist/adapters/llm/anthropic.d.ts +27 -0
- package/dist/adapters/llm/anthropic.d.ts.map +1 -0
- package/dist/adapters/llm/anthropic.js +287 -0
- package/dist/adapters/llm/anthropic.js.map +1 -0
- package/dist/adapters/llm/base.d.ts +62 -0
- package/dist/adapters/llm/base.d.ts.map +1 -0
- package/dist/adapters/llm/base.js +140 -0
- package/dist/adapters/llm/base.js.map +1 -0
- package/dist/adapters/llm/deepseek.d.ts +24 -0
- package/dist/adapters/llm/deepseek.d.ts.map +1 -0
- package/dist/adapters/llm/deepseek.js +228 -0
- package/dist/adapters/llm/deepseek.js.map +1 -0
- package/dist/adapters/llm/groq.d.ts +25 -0
- package/dist/adapters/llm/groq.d.ts.map +1 -0
- package/dist/adapters/llm/groq.js +265 -0
- package/dist/adapters/llm/groq.js.map +1 -0
- package/dist/adapters/llm/index.d.ts +62 -0
- package/dist/adapters/llm/index.d.ts.map +1 -0
- package/dist/adapters/llm/index.js +380 -0
- package/dist/adapters/llm/index.js.map +1 -0
- package/dist/adapters/llm/ollama.d.ts +23 -0
- package/dist/adapters/llm/ollama.d.ts.map +1 -0
- package/dist/adapters/llm/ollama.js +261 -0
- package/dist/adapters/llm/ollama.js.map +1 -0
- package/dist/adapters/llm/openai.d.ts +22 -0
- package/dist/adapters/llm/openai.d.ts.map +1 -0
- package/dist/adapters/llm/openai.js +232 -0
- package/dist/adapters/llm/openai.js.map +1 -0
- package/dist/adapters/llm/openrouter.d.ts +27 -0
- package/dist/adapters/llm/openrouter.d.ts.map +1 -0
- package/dist/adapters/llm/openrouter.js +305 -0
- package/dist/adapters/llm/openrouter.js.map +1 -0
- package/dist/adapters/vector/index.d.ts.map +1 -1
- package/dist/adapters/vector/index.js +8 -0
- package/dist/adapters/vector/index.js.map +1 -1
- package/dist/adapters/vector/redis-native.d.ts +35 -0
- package/dist/adapters/vector/redis-native.d.ts.map +1 -0
- package/dist/adapters/vector/redis-native.js +170 -0
- package/dist/adapters/vector/redis-native.js.map +1 -0
- package/dist/cli/commands/chat.d.ts +4 -0
- package/dist/cli/commands/chat.d.ts.map +1 -0
- package/dist/cli/commands/chat.js +374 -0
- package/dist/cli/commands/chat.js.map +1 -0
- package/dist/cli/commands/maintenance.d.ts +4 -0
- package/dist/cli/commands/maintenance.d.ts.map +1 -0
- package/dist/cli/commands/maintenance.js +237 -0
- package/dist/cli/commands/maintenance.js.map +1 -0
- package/dist/cli/commands/rules.d.ts +9 -0
- package/dist/cli/commands/rules.d.ts.map +1 -0
- package/dist/cli/commands/rules.js +639 -0
- package/dist/cli/commands/rules.js.map +1 -0
- package/dist/cli/commands/setup.js +5 -4
- package/dist/cli/commands/setup.js.map +1 -1
- package/dist/cli/index.js +6 -0
- package/dist/cli/index.js.map +1 -1
- package/dist/config/memoryConfig.d.ts +427 -0
- package/dist/config/memoryConfig.d.ts.map +1 -0
- package/dist/config/memoryConfig.js +258 -0
- package/dist/config/memoryConfig.js.map +1 -0
- package/dist/config/rulesConfig.d.ts +486 -0
- package/dist/config/rulesConfig.d.ts.map +1 -0
- package/dist/config/rulesConfig.js +345 -0
- package/dist/config/rulesConfig.js.map +1 -0
- package/dist/dashboard/coreTools.d.ts +14 -0
- package/dist/dashboard/coreTools.d.ts.map +1 -0
- package/dist/dashboard/coreTools.js +413 -0
- package/dist/dashboard/coreTools.js.map +1 -0
- package/dist/dashboard/public/index.html +1982 -13
- package/dist/dashboard/server.d.ts +1 -8
- package/dist/dashboard/server.d.ts.map +1 -1
- package/dist/dashboard/server.js +846 -13
- package/dist/dashboard/server.js.map +1 -1
- package/dist/dashboard/toolRegistry.d.ts +192 -0
- package/dist/dashboard/toolRegistry.d.ts.map +1 -0
- package/dist/dashboard/toolRegistry.js +322 -0
- package/dist/dashboard/toolRegistry.js.map +1 -0
- package/dist/proxy/index.d.ts +1 -1
- package/dist/proxy/index.d.ts.map +1 -1
- package/dist/proxy/index.js +9 -6
- package/dist/proxy/index.js.map +1 -1
- package/dist/server/index.js +21 -0
- package/dist/server/index.js.map +1 -1
- package/dist/server/tools/crawl.d.ts.map +1 -1
- package/dist/server/tools/crawl.js +8 -0
- package/dist/server/tools/crawl.js.map +1 -1
- package/dist/server/tools/index.d.ts.map +1 -1
- package/dist/server/tools/index.js +19 -1
- package/dist/server/tools/index.js.map +1 -1
- package/dist/server/tools/ingest.d.ts.map +1 -1
- package/dist/server/tools/ingest.js +5 -0
- package/dist/server/tools/ingest.js.map +1 -1
- package/dist/server/tools/memory.d.ts +250 -0
- package/dist/server/tools/memory.d.ts.map +1 -0
- package/dist/server/tools/memory.js +472 -0
- package/dist/server/tools/memory.js.map +1 -0
- package/dist/server/tools/recursive-query.d.ts.map +1 -1
- package/dist/server/tools/recursive-query.js +6 -0
- package/dist/server/tools/recursive-query.js.map +1 -1
- package/dist/server/tools/search.d.ts.map +1 -1
- package/dist/server/tools/search.js +6 -0
- package/dist/server/tools/search.js.map +1 -1
- package/dist/services/activity-log.d.ts +10 -0
- package/dist/services/activity-log.d.ts.map +1 -0
- package/dist/services/activity-log.js +53 -0
- package/dist/services/activity-log.js.map +1 -0
- package/dist/services/categoryManager.d.ts +110 -0
- package/dist/services/categoryManager.d.ts.map +1 -0
- package/dist/services/categoryManager.js +549 -0
- package/dist/services/categoryManager.js.map +1 -0
- package/dist/services/contextEnvironment.d.ts +206 -0
- package/dist/services/contextEnvironment.d.ts.map +1 -0
- package/dist/services/contextEnvironment.js +481 -0
- package/dist/services/contextEnvironment.js.map +1 -0
- package/dist/services/conversationProcessor.d.ts +99 -0
- package/dist/services/conversationProcessor.d.ts.map +1 -0
- package/dist/services/conversationProcessor.js +311 -0
- package/dist/services/conversationProcessor.js.map +1 -0
- package/dist/services/cursorChatReader.d.ts +129 -0
- package/dist/services/cursorChatReader.d.ts.map +1 -0
- package/dist/services/cursorChatReader.js +419 -0
- package/dist/services/cursorChatReader.js.map +1 -0
- package/dist/services/decayCalculator.d.ts +85 -0
- package/dist/services/decayCalculator.d.ts.map +1 -0
- package/dist/services/decayCalculator.js +182 -0
- package/dist/services/decayCalculator.js.map +1 -0
- package/dist/services/enhancedVectorStore.d.ts +102 -0
- package/dist/services/enhancedVectorStore.d.ts.map +1 -0
- package/dist/services/enhancedVectorStore.js +245 -0
- package/dist/services/enhancedVectorStore.js.map +1 -0
- package/dist/services/hybridScorer.d.ts +120 -0
- package/dist/services/hybridScorer.d.ts.map +1 -0
- package/dist/services/hybridScorer.js +334 -0
- package/dist/services/hybridScorer.js.map +1 -0
- package/dist/services/knowledgeExtractor.d.ts +45 -0
- package/dist/services/knowledgeExtractor.d.ts.map +1 -0
- package/dist/services/knowledgeExtractor.js +436 -0
- package/dist/services/knowledgeExtractor.js.map +1 -0
- package/dist/services/knowledgeStorage.d.ts +102 -0
- package/dist/services/knowledgeStorage.d.ts.map +1 -0
- package/dist/services/knowledgeStorage.js +383 -0
- package/dist/services/knowledgeStorage.js.map +1 -0
- package/dist/services/maintenanceScheduler.d.ts +89 -0
- package/dist/services/maintenanceScheduler.d.ts.map +1 -0
- package/dist/services/maintenanceScheduler.js +479 -0
- package/dist/services/maintenanceScheduler.js.map +1 -0
- package/dist/services/memoryMetadataStore.d.ts +62 -0
- package/dist/services/memoryMetadataStore.d.ts.map +1 -0
- package/dist/services/memoryMetadataStore.js +570 -0
- package/dist/services/memoryMetadataStore.js.map +1 -0
- package/dist/services/recursiveRetrieval.d.ts +122 -0
- package/dist/services/recursiveRetrieval.d.ts.map +1 -0
- package/dist/services/recursiveRetrieval.js +443 -0
- package/dist/services/recursiveRetrieval.js.map +1 -0
- package/dist/services/relationshipGraph.d.ts +77 -0
- package/dist/services/relationshipGraph.d.ts.map +1 -0
- package/dist/services/relationshipGraph.js +411 -0
- package/dist/services/relationshipGraph.js.map +1 -0
- package/dist/services/rlmSafeguards.d.ts +273 -0
- package/dist/services/rlmSafeguards.d.ts.map +1 -0
- package/dist/services/rlmSafeguards.js +705 -0
- package/dist/services/rlmSafeguards.js.map +1 -0
- package/dist/services/rulesAnalyzer.d.ts +119 -0
- package/dist/services/rulesAnalyzer.d.ts.map +1 -0
- package/dist/services/rulesAnalyzer.js +768 -0
- package/dist/services/rulesAnalyzer.js.map +1 -0
- package/dist/services/rulesMerger.d.ts +75 -0
- package/dist/services/rulesMerger.d.ts.map +1 -0
- package/dist/services/rulesMerger.js +404 -0
- package/dist/services/rulesMerger.js.map +1 -0
- package/dist/services/rulesParser.d.ts +127 -0
- package/dist/services/rulesParser.d.ts.map +1 -0
- package/dist/services/rulesParser.js +594 -0
- package/dist/services/rulesParser.js.map +1 -0
- package/dist/services/smartChunker.d.ts +110 -0
- package/dist/services/smartChunker.d.ts.map +1 -0
- package/dist/services/smartChunker.js +520 -0
- package/dist/services/smartChunker.js.map +1 -0
- package/dist/types/categories.d.ts +105 -0
- package/dist/types/categories.d.ts.map +1 -0
- package/dist/types/categories.js +108 -0
- package/dist/types/categories.js.map +1 -0
- package/dist/types/extractedKnowledge.d.ts +233 -0
- package/dist/types/extractedKnowledge.d.ts.map +1 -0
- package/dist/types/extractedKnowledge.js +56 -0
- package/dist/types/extractedKnowledge.js.map +1 -0
- package/dist/types/index.d.ts +9 -2
- package/dist/types/index.d.ts.map +1 -1
- package/dist/types/index.js +12 -1
- package/dist/types/index.js.map +1 -1
- package/dist/types/llmProvider.d.ts +282 -0
- package/dist/types/llmProvider.d.ts.map +1 -0
- package/dist/types/llmProvider.js +48 -0
- package/dist/types/llmProvider.js.map +1 -0
- package/dist/types/memory.d.ts +227 -0
- package/dist/types/memory.d.ts.map +1 -0
- package/dist/types/memory.js +76 -0
- package/dist/types/memory.js.map +1 -0
- package/dist/types/relationships.d.ts +167 -0
- package/dist/types/relationships.d.ts.map +1 -0
- package/dist/types/relationships.js +106 -0
- package/dist/types/relationships.js.map +1 -0
- package/dist/types/rulesOptimizer.d.ts +345 -0
- package/dist/types/rulesOptimizer.d.ts.map +1 -0
- package/dist/types/rulesOptimizer.js +22 -0
- package/dist/types/rulesOptimizer.js.map +1 -0
- package/docs/cursor-recursive-rag-memory-spec.md +4569 -0
- package/docs/cursor-recursive-rag-tasks.md +1355 -0
- package/package.json +6 -3
- package/restart-rag.sh +16 -0
|
@@ -0,0 +1,228 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* DeepSeek LLM Provider
|
|
3
|
+
*
|
|
4
|
+
* Supports: deepseek-chat, deepseek-coder
|
|
5
|
+
* Uses OpenAI-compatible API
|
|
6
|
+
*/
|
|
7
|
+
import { BaseLLMProvider } from './base.js';
|
|
8
|
+
import { LLMError } from '../../types/llmProvider.js';
|
|
9
|
+
const DEEPSEEK_MODELS = {
|
|
10
|
+
'deepseek-chat': {
|
|
11
|
+
contextLength: 128000,
|
|
12
|
+
supportsVision: false,
|
|
13
|
+
supportsFunctions: true,
|
|
14
|
+
supportsJsonMode: true,
|
|
15
|
+
supportsStreaming: true,
|
|
16
|
+
maxOutputTokens: 8192,
|
|
17
|
+
inputCostPer1M: 0.14,
|
|
18
|
+
outputCostPer1M: 0.28,
|
|
19
|
+
},
|
|
20
|
+
'deepseek-coder': {
|
|
21
|
+
contextLength: 128000,
|
|
22
|
+
supportsVision: false,
|
|
23
|
+
supportsFunctions: true,
|
|
24
|
+
supportsJsonMode: true,
|
|
25
|
+
supportsStreaming: true,
|
|
26
|
+
maxOutputTokens: 8192,
|
|
27
|
+
inputCostPer1M: 0.14,
|
|
28
|
+
outputCostPer1M: 0.28,
|
|
29
|
+
},
|
|
30
|
+
'deepseek-reasoner': {
|
|
31
|
+
contextLength: 64000,
|
|
32
|
+
supportsVision: false,
|
|
33
|
+
supportsFunctions: false,
|
|
34
|
+
supportsJsonMode: false,
|
|
35
|
+
supportsStreaming: true,
|
|
36
|
+
maxOutputTokens: 8192,
|
|
37
|
+
inputCostPer1M: 0.55,
|
|
38
|
+
outputCostPer1M: 2.19,
|
|
39
|
+
},
|
|
40
|
+
};
|
|
41
|
+
export class DeepSeekProvider extends BaseLLMProvider {
|
|
42
|
+
type = 'deepseek';
|
|
43
|
+
name = 'DeepSeek';
|
|
44
|
+
apiKey;
|
|
45
|
+
baseUrl;
|
|
46
|
+
model;
|
|
47
|
+
constructor(config) {
|
|
48
|
+
super(config);
|
|
49
|
+
this.apiKey = config.apiKey;
|
|
50
|
+
this.baseUrl = config.baseUrl ?? 'https://api.deepseek.com';
|
|
51
|
+
this.model = config.defaultModel ?? 'deepseek-chat';
|
|
52
|
+
}
|
|
53
|
+
async isAvailable() {
|
|
54
|
+
try {
|
|
55
|
+
const response = await this.fetchWithTimeout(`${this.baseUrl}/v1/models`, {
|
|
56
|
+
method: 'GET',
|
|
57
|
+
headers: { 'Authorization': `Bearer ${this.apiKey}` },
|
|
58
|
+
}, 5000);
|
|
59
|
+
return response.ok;
|
|
60
|
+
}
|
|
61
|
+
catch {
|
|
62
|
+
return false;
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
async listModels() {
|
|
66
|
+
return Object.entries(DEEPSEEK_MODELS).map(([id, capabilities]) => ({
|
|
67
|
+
id,
|
|
68
|
+
name: id.replace('deepseek-', 'DeepSeek ').replace(/^\w/, c => c.toUpperCase()),
|
|
69
|
+
provider: 'deepseek',
|
|
70
|
+
capabilities,
|
|
71
|
+
}));
|
|
72
|
+
}
|
|
73
|
+
getModelCapabilities(model) {
|
|
74
|
+
return DEEPSEEK_MODELS[model] ?? null;
|
|
75
|
+
}
|
|
76
|
+
async chat(messages, options) {
|
|
77
|
+
const model = options?.headers?.['x-model'] ?? this.model;
|
|
78
|
+
const capabilities = this.getModelCapabilities(model);
|
|
79
|
+
const startTime = Date.now();
|
|
80
|
+
return this.withRetry(async () => {
|
|
81
|
+
try {
|
|
82
|
+
const body = {
|
|
83
|
+
model,
|
|
84
|
+
messages: messages.map(m => ({ role: m.role, content: m.content })),
|
|
85
|
+
max_tokens: options?.maxTokens ?? capabilities?.maxOutputTokens ?? 4096,
|
|
86
|
+
};
|
|
87
|
+
if (options?.temperature !== undefined) {
|
|
88
|
+
body.temperature = options.temperature;
|
|
89
|
+
}
|
|
90
|
+
if (options?.topP !== undefined) {
|
|
91
|
+
body.top_p = options.topP;
|
|
92
|
+
}
|
|
93
|
+
if (options?.stop) {
|
|
94
|
+
body.stop = options.stop;
|
|
95
|
+
}
|
|
96
|
+
if (options?.jsonMode && capabilities?.supportsJsonMode) {
|
|
97
|
+
body.response_format = { type: 'json_object' };
|
|
98
|
+
}
|
|
99
|
+
// Handle streaming
|
|
100
|
+
if (options?.onStream && capabilities?.supportsStreaming) {
|
|
101
|
+
return this.handleStream(body, options.onStream, startTime, model, capabilities);
|
|
102
|
+
}
|
|
103
|
+
const response = await this.fetchWithTimeout(`${this.baseUrl}/v1/chat/completions`, {
|
|
104
|
+
method: 'POST',
|
|
105
|
+
headers: {
|
|
106
|
+
'Content-Type': 'application/json',
|
|
107
|
+
'Authorization': `Bearer ${this.apiKey}`,
|
|
108
|
+
},
|
|
109
|
+
body: JSON.stringify(body),
|
|
110
|
+
}, options?.timeout);
|
|
111
|
+
if (!response.ok) {
|
|
112
|
+
throw await this.parseErrorResponse(response);
|
|
113
|
+
}
|
|
114
|
+
const data = await response.json();
|
|
115
|
+
const latencyMs = Date.now() - startTime;
|
|
116
|
+
const usage = {
|
|
117
|
+
promptTokens: data.usage.prompt_tokens,
|
|
118
|
+
completionTokens: data.usage.completion_tokens,
|
|
119
|
+
totalTokens: data.usage.total_tokens,
|
|
120
|
+
estimatedCost: this.calculateCost({
|
|
121
|
+
promptTokens: data.usage.prompt_tokens,
|
|
122
|
+
completionTokens: data.usage.completion_tokens,
|
|
123
|
+
}, capabilities?.inputCostPer1M, capabilities?.outputCostPer1M),
|
|
124
|
+
};
|
|
125
|
+
this.trackUsage(usage);
|
|
126
|
+
return {
|
|
127
|
+
content: data.choices[0]?.message?.content ?? '',
|
|
128
|
+
model: data.model,
|
|
129
|
+
usage,
|
|
130
|
+
latencyMs,
|
|
131
|
+
streamed: false,
|
|
132
|
+
finishReason: data.choices[0]?.finish_reason ?? undefined,
|
|
133
|
+
};
|
|
134
|
+
}
|
|
135
|
+
catch (error) {
|
|
136
|
+
if (error instanceof LLMError)
|
|
137
|
+
throw error;
|
|
138
|
+
throw this.handleError(error);
|
|
139
|
+
}
|
|
140
|
+
}, `DeepSeek chat completion`);
|
|
141
|
+
}
|
|
142
|
+
async handleStream(body, onStream, startTime, model, capabilities) {
|
|
143
|
+
const response = await this.fetchWithTimeout(`${this.baseUrl}/v1/chat/completions`, {
|
|
144
|
+
method: 'POST',
|
|
145
|
+
headers: {
|
|
146
|
+
'Content-Type': 'application/json',
|
|
147
|
+
'Authorization': `Bearer ${this.apiKey}`,
|
|
148
|
+
},
|
|
149
|
+
body: JSON.stringify({ ...body, stream: true }),
|
|
150
|
+
});
|
|
151
|
+
if (!response.ok) {
|
|
152
|
+
throw await this.parseErrorResponse(response);
|
|
153
|
+
}
|
|
154
|
+
let content = '';
|
|
155
|
+
let finishReason;
|
|
156
|
+
const reader = response.body?.getReader();
|
|
157
|
+
if (!reader) {
|
|
158
|
+
throw new LLMError('No response body', 'INVALID_RESPONSE', 'deepseek', false);
|
|
159
|
+
}
|
|
160
|
+
const decoder = new TextDecoder();
|
|
161
|
+
let buffer = '';
|
|
162
|
+
try {
|
|
163
|
+
while (true) {
|
|
164
|
+
const { done, value } = await reader.read();
|
|
165
|
+
if (done)
|
|
166
|
+
break;
|
|
167
|
+
buffer += decoder.decode(value, { stream: true });
|
|
168
|
+
const lines = buffer.split('\n');
|
|
169
|
+
buffer = lines.pop() ?? '';
|
|
170
|
+
for (const line of lines) {
|
|
171
|
+
if (!line.startsWith('data: '))
|
|
172
|
+
continue;
|
|
173
|
+
const jsonStr = line.slice(6).trim();
|
|
174
|
+
if (!jsonStr || jsonStr === '[DONE]')
|
|
175
|
+
continue;
|
|
176
|
+
try {
|
|
177
|
+
const chunk = JSON.parse(jsonStr);
|
|
178
|
+
const delta = chunk.choices?.[0]?.delta?.content;
|
|
179
|
+
if (delta) {
|
|
180
|
+
content += delta;
|
|
181
|
+
onStream(delta);
|
|
182
|
+
}
|
|
183
|
+
if (chunk.choices?.[0]?.finish_reason) {
|
|
184
|
+
finishReason = chunk.choices[0].finish_reason;
|
|
185
|
+
}
|
|
186
|
+
}
|
|
187
|
+
catch {
|
|
188
|
+
// Ignore malformed JSON
|
|
189
|
+
}
|
|
190
|
+
}
|
|
191
|
+
}
|
|
192
|
+
}
|
|
193
|
+
finally {
|
|
194
|
+
reader.releaseLock();
|
|
195
|
+
}
|
|
196
|
+
const latencyMs = Date.now() - startTime;
|
|
197
|
+
return {
|
|
198
|
+
content,
|
|
199
|
+
model,
|
|
200
|
+
latencyMs,
|
|
201
|
+
streamed: true,
|
|
202
|
+
finishReason,
|
|
203
|
+
};
|
|
204
|
+
}
|
|
205
|
+
async parseErrorResponse(response) {
|
|
206
|
+
try {
|
|
207
|
+
const data = await response.json();
|
|
208
|
+
const message = data.error?.message ?? 'Unknown error';
|
|
209
|
+
if (response.status === 401) {
|
|
210
|
+
return new LLMError('DeepSeek authentication failed', 'AUTHENTICATION_FAILED', 'deepseek', false);
|
|
211
|
+
}
|
|
212
|
+
if (response.status === 429) {
|
|
213
|
+
return new LLMError('DeepSeek rate limit exceeded', 'RATE_LIMITED', 'deepseek', true);
|
|
214
|
+
}
|
|
215
|
+
return new LLMError(`DeepSeek API error: ${message}`, 'UNKNOWN', 'deepseek', response.status >= 500);
|
|
216
|
+
}
|
|
217
|
+
catch {
|
|
218
|
+
return new LLMError(`DeepSeek API error (${response.status})`, 'UNKNOWN', 'deepseek', response.status >= 500);
|
|
219
|
+
}
|
|
220
|
+
}
|
|
221
|
+
handleError(error) {
|
|
222
|
+
if (error instanceof Error && error.name === 'AbortError') {
|
|
223
|
+
return new LLMError('DeepSeek request timed out', 'TIMEOUT', 'deepseek', true, error);
|
|
224
|
+
}
|
|
225
|
+
return new LLMError(`DeepSeek error: ${error instanceof Error ? error.message : String(error)}`, 'UNKNOWN', 'deepseek', false, error instanceof Error ? error : undefined);
|
|
226
|
+
}
|
|
227
|
+
}
|
|
228
|
+
//# sourceMappingURL=deepseek.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"deepseek.js","sourceRoot":"","sources":["../../../src/adapters/llm/deepseek.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAS5C,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAEtD,MAAM,eAAe,GAAsC;IACzD,eAAe,EAAE;QACf,aAAa,EAAE,MAAM;QACrB,cAAc,EAAE,KAAK;QACrB,iBAAiB,EAAE,IAAI;QACvB,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,IAAI;QACvB,eAAe,EAAE,IAAI;QACrB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;KACtB;IACD,gBAAgB,EAAE;QAChB,aAAa,EAAE,MAAM;QACrB,cAAc,EAAE,KAAK;QACrB,iBAAiB,EAAE,IAAI;QACvB,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,IAAI;QACvB,eAAe,EAAE,IAAI;QACrB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;KACtB;IACD,mBAAmB,EAAE;QACnB,aAAa,EAAE,KAAK;QACpB,cAAc,EAAE,KAAK;QACrB,iBAAiB,EAAE,KAAK;QACxB,gBAAgB,EAAE,KAAK;QACvB,iBAAiB,EAAE,IAAI;QACvB,eAAe,EAAE,IAAI;QACrB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;KACtB;CACF,CAAC;AAmBF,MAAM,OAAO,gBAAiB,SAAQ,eAAe;IAC1C,IAAI,GAAG,UAAmB,CAAC;IAC3B,IAAI,GAAG,UAAU,CAAC;IAEnB,MAAM,CAAS;IACf,OAAO,CAAS;IAChB,KAAK,CAAS;IAEtB,YAAY,MAA8B;QACxC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,0BAA0B,CAAC;QAC5D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,YAAY,IAAI,eAAe,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC1C,GAAG,IAAI,CAAC,OAAO,YAAY,EAC3B;gBACE,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE,EAAE,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,EAAE;aACtD,EACD,IAAI,CACL,CAAC;YACF,OAAO,QAAQ,CAAC,EAAE,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE,CAAC,CAAC;YAClE,EAAE;YACF,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/E,QAAQ,EAAE,UAAmB;YAC7B,YAAY;SACb,CAAC,CAAC,CAAC;IACN,CAAC;IAED,oBAAoB,CAAC,KAAa;QAChC,OAAO,eAAe,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,IAAI,CACR,QAAsB,EACtB,OAA8B;QAE9B,MAAM,KAAK,GAAG,OAAO,EAAE,OAAO,EAAE,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC;QAC1D,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE;YAC/B,IAAI,CAAC;gBACH,MAAM,IAAI,GAA4B;oBACpC,KAAK;oBACL,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;oBACnE,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,YAAY,EAAE,eAAe,IAAI,IAAI;iBACxE,CAAC;gBAEF,IAAI,OAAO,EAAE,WAAW,KAAK,SAAS,EAAE,CAAC;oBACvC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;gBACzC,CAAC;gBACD,IAAI,OAAO,EAAE,IAAI,KAAK,SAAS,EAAE,CAAC;oBAChC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC;gBAC5B,CAAC;gBACD,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;oBAClB,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;gBAC3B,CAAC;gBACD,IAAI,OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE,gBAAgB,EAAE,CAAC;oBACxD,IAAI,CAAC,eAAe,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;gBACjD,CAAC;gBAED,mBAAmB;gBACnB,IAAI,OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE,iBAAiB,EAAE,CAAC;oBACzD,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;gBACnF,CAAC;gBAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC1C,GAAG,IAAI,CAAC,OAAO,sBAAsB,EACrC;oBACE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE;wBACP,cAAc,EAAE,kBAAkB;wBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;qBACzC;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;iBAC3B,EACD,OAAO,EAAE,OAAO,CACjB,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,MAAM,MAAM,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;gBAChD,CAAC;gBAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAsB,CAAC;gBACvD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBAEzC,MAAM,KAAK,GAAG;oBACZ,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;oBACtC,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;oBAC9C,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;oBACpC,aAAa,EAAE,IAAI,CAAC,aAAa,CAC/B;wBACE,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;wBACtC,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;qBAC/C,EACD,YAAY,EAAE,cAAc,EAC5B,YAAY,EAAE,eAAe,CAC9B;iBACF,CAAC;gBAEF,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAEvB,OAAO;oBACL,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE;oBAChD,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,KAAK;oBACL,SAAS;oBACT,QAAQ,EAAE,KAAK;oBACf,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,SAAS;iBAC1D,CAAC;YAEJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,QAAQ;oBAAE,MAAM,KAAK,CAAC;gBAC3C,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAChC,CAAC;QACH,CAAC,EAAE,0BAA0B,CAAC,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,IAA6B,EAC7B,QAAiC,EACjC,SAAiB,EACjB,KAAa,EACb,YAAsC;QAEtC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC1C,GAAG,IAAI,CAAC,OAAO,sBAAsB,EACrC;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;aACzC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SAChD,CACF,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,MAAM,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,YAAgC,CAAC;QAErC,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,QAAQ,CAAC,kBAAkB,EAAE,kBAAkB,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;QAChF,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,IAAI,CAAC;YACH,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,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;wBAAE,SAAS;oBACzC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACrC,IAAI,CAAC,OAAO,IAAI,OAAO,KAAK,QAAQ;wBAAE,SAAS;oBAE/C,IAAI,CAAC;wBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAClC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC;wBACjD,IAAI,KAAK,EAAE,CAAC;4BACV,OAAO,IAAI,KAAK,CAAC;4BACjB,QAAQ,CAAC,KAAK,CAAC,CAAC;wBAClB,CAAC;wBACD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC;4BACtC,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;wBAChD,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,wBAAwB;oBAC1B,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAEzC,OAAO;YACL,OAAO;YACP,KAAK;YACL,SAAS;YACT,QAAQ,EAAE,IAAI;YACd,YAAY;SACb,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,QAAkB;QACjD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAsC,CAAC;YACvE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,eAAe,CAAC;YAEvD,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,OAAO,IAAI,QAAQ,CAAC,gCAAgC,EAAE,uBAAuB,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;YACpG,CAAC;YACD,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,OAAO,IAAI,QAAQ,CAAC,8BAA8B,EAAE,cAAc,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;YACxF,CAAC;YAED,OAAO,IAAI,QAAQ,CAAC,uBAAuB,OAAO,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC;QACvG,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,QAAQ,CAAC,uBAAuB,QAAQ,CAAC,MAAM,GAAG,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC;QAChH,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,KAAc;QAChC,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC1D,OAAO,IAAI,QAAQ,CAAC,4BAA4B,EAAE,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QACxF,CAAC;QACD,OAAO,IAAI,QAAQ,CACjB,mBAAmB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAC3E,SAAS,EACT,UAAU,EACV,KAAK,EACL,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAC3C,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Groq LLM Provider
|
|
3
|
+
*
|
|
4
|
+
* Fast inference for open-source models.
|
|
5
|
+
* Supports: LLaMA, Mixtral, Gemma models
|
|
6
|
+
*/
|
|
7
|
+
import { BaseLLMProvider } from './base.js';
|
|
8
|
+
import type { LLMMessage, LLMCompletionOptions, LLMResponse, ModelInfo, ModelCapabilities, GroqProviderConfig } from '../../types/llmProvider.js';
|
|
9
|
+
export declare class GroqProvider extends BaseLLMProvider {
|
|
10
|
+
readonly type: "groq";
|
|
11
|
+
readonly name = "Groq";
|
|
12
|
+
private apiKey;
|
|
13
|
+
private baseUrl;
|
|
14
|
+
private model;
|
|
15
|
+
constructor(config: GroqProviderConfig);
|
|
16
|
+
isAvailable(): Promise<boolean>;
|
|
17
|
+
listModels(): Promise<ModelInfo[]>;
|
|
18
|
+
getModelCapabilities(model: string): ModelCapabilities | null;
|
|
19
|
+
chat(messages: LLMMessage[], options?: LLMCompletionOptions): Promise<LLMResponse>;
|
|
20
|
+
private handleStream;
|
|
21
|
+
private parseErrorResponse;
|
|
22
|
+
private handleError;
|
|
23
|
+
private formatModelName;
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=groq.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"groq.d.ts","sourceRoot":"","sources":["../../../src/adapters/llm/groq.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAC5C,OAAO,KAAK,EACV,UAAU,EACV,oBAAoB,EACpB,WAAW,EACX,SAAS,EACT,iBAAiB,EACjB,kBAAkB,EACnB,MAAM,4BAA4B,CAAC;AAmFpC,qBAAa,YAAa,SAAQ,eAAe;IAC/C,QAAQ,CAAC,IAAI,EAAG,MAAM,CAAU;IAChC,QAAQ,CAAC,IAAI,UAAU;IAEvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,EAAE,kBAAkB;IAOhC,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;IAgB/B,UAAU,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IASxC,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,iBAAiB,GAAG,IAAI;IAIvD,IAAI,CACR,QAAQ,EAAE,UAAU,EAAE,EACtB,OAAO,CAAC,EAAE,oBAAoB,GAC7B,OAAO,CAAC,WAAW,CAAC;YAmFT,YAAY;YA8EZ,kBAAkB;IAkBhC,OAAO,CAAC,WAAW;IAanB,OAAO,CAAC,eAAe;CAOxB"}
|
|
@@ -0,0 +1,265 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Groq LLM Provider
|
|
3
|
+
*
|
|
4
|
+
* Fast inference for open-source models.
|
|
5
|
+
* Supports: LLaMA, Mixtral, Gemma models
|
|
6
|
+
*/
|
|
7
|
+
import { BaseLLMProvider } from './base.js';
|
|
8
|
+
import { LLMError } from '../../types/llmProvider.js';
|
|
9
|
+
const GROQ_MODELS = {
|
|
10
|
+
'llama-3.3-70b-versatile': {
|
|
11
|
+
contextLength: 128000,
|
|
12
|
+
supportsVision: false,
|
|
13
|
+
supportsFunctions: true,
|
|
14
|
+
supportsJsonMode: true,
|
|
15
|
+
supportsStreaming: true,
|
|
16
|
+
maxOutputTokens: 32768,
|
|
17
|
+
inputCostPer1M: 0.59,
|
|
18
|
+
outputCostPer1M: 0.79,
|
|
19
|
+
},
|
|
20
|
+
'llama-3.1-70b-versatile': {
|
|
21
|
+
contextLength: 128000,
|
|
22
|
+
supportsVision: false,
|
|
23
|
+
supportsFunctions: true,
|
|
24
|
+
supportsJsonMode: true,
|
|
25
|
+
supportsStreaming: true,
|
|
26
|
+
maxOutputTokens: 8000,
|
|
27
|
+
inputCostPer1M: 0.59,
|
|
28
|
+
outputCostPer1M: 0.79,
|
|
29
|
+
},
|
|
30
|
+
'llama-3.1-8b-instant': {
|
|
31
|
+
contextLength: 128000,
|
|
32
|
+
supportsVision: false,
|
|
33
|
+
supportsFunctions: true,
|
|
34
|
+
supportsJsonMode: true,
|
|
35
|
+
supportsStreaming: true,
|
|
36
|
+
maxOutputTokens: 8000,
|
|
37
|
+
inputCostPer1M: 0.05,
|
|
38
|
+
outputCostPer1M: 0.08,
|
|
39
|
+
},
|
|
40
|
+
'mixtral-8x7b-32768': {
|
|
41
|
+
contextLength: 32768,
|
|
42
|
+
supportsVision: false,
|
|
43
|
+
supportsFunctions: true,
|
|
44
|
+
supportsJsonMode: true,
|
|
45
|
+
supportsStreaming: true,
|
|
46
|
+
maxOutputTokens: 32768,
|
|
47
|
+
inputCostPer1M: 0.24,
|
|
48
|
+
outputCostPer1M: 0.24,
|
|
49
|
+
},
|
|
50
|
+
'gemma2-9b-it': {
|
|
51
|
+
contextLength: 8192,
|
|
52
|
+
supportsVision: false,
|
|
53
|
+
supportsFunctions: true,
|
|
54
|
+
supportsJsonMode: true,
|
|
55
|
+
supportsStreaming: true,
|
|
56
|
+
maxOutputTokens: 8192,
|
|
57
|
+
inputCostPer1M: 0.20,
|
|
58
|
+
outputCostPer1M: 0.20,
|
|
59
|
+
},
|
|
60
|
+
'llama-3.2-90b-vision-preview': {
|
|
61
|
+
contextLength: 128000,
|
|
62
|
+
supportsVision: true,
|
|
63
|
+
supportsFunctions: true,
|
|
64
|
+
supportsJsonMode: true,
|
|
65
|
+
supportsStreaming: true,
|
|
66
|
+
maxOutputTokens: 8000,
|
|
67
|
+
inputCostPer1M: 0.90,
|
|
68
|
+
outputCostPer1M: 0.90,
|
|
69
|
+
},
|
|
70
|
+
};
|
|
71
|
+
export class GroqProvider extends BaseLLMProvider {
|
|
72
|
+
type = 'groq';
|
|
73
|
+
name = 'Groq';
|
|
74
|
+
apiKey;
|
|
75
|
+
baseUrl;
|
|
76
|
+
model;
|
|
77
|
+
constructor(config) {
|
|
78
|
+
super(config);
|
|
79
|
+
this.apiKey = config.apiKey;
|
|
80
|
+
this.baseUrl = config.baseUrl ?? 'https://api.groq.com/openai';
|
|
81
|
+
this.model = config.defaultModel ?? 'llama-3.3-70b-versatile';
|
|
82
|
+
}
|
|
83
|
+
async isAvailable() {
|
|
84
|
+
try {
|
|
85
|
+
const response = await this.fetchWithTimeout(`${this.baseUrl}/v1/models`, {
|
|
86
|
+
method: 'GET',
|
|
87
|
+
headers: { 'Authorization': `Bearer ${this.apiKey}` },
|
|
88
|
+
}, 5000);
|
|
89
|
+
return response.ok;
|
|
90
|
+
}
|
|
91
|
+
catch {
|
|
92
|
+
return false;
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
async listModels() {
|
|
96
|
+
return Object.entries(GROQ_MODELS).map(([id, capabilities]) => ({
|
|
97
|
+
id,
|
|
98
|
+
name: this.formatModelName(id),
|
|
99
|
+
provider: 'groq',
|
|
100
|
+
capabilities,
|
|
101
|
+
}));
|
|
102
|
+
}
|
|
103
|
+
getModelCapabilities(model) {
|
|
104
|
+
return GROQ_MODELS[model] ?? null;
|
|
105
|
+
}
|
|
106
|
+
async chat(messages, options) {
|
|
107
|
+
const model = options?.headers?.['x-model'] ?? this.model;
|
|
108
|
+
const capabilities = this.getModelCapabilities(model);
|
|
109
|
+
const startTime = Date.now();
|
|
110
|
+
return this.withRetry(async () => {
|
|
111
|
+
try {
|
|
112
|
+
const body = {
|
|
113
|
+
model,
|
|
114
|
+
messages: messages.map(m => ({ role: m.role, content: m.content })),
|
|
115
|
+
max_tokens: options?.maxTokens ?? capabilities?.maxOutputTokens ?? 4096,
|
|
116
|
+
};
|
|
117
|
+
if (options?.temperature !== undefined) {
|
|
118
|
+
body.temperature = options.temperature;
|
|
119
|
+
}
|
|
120
|
+
if (options?.topP !== undefined) {
|
|
121
|
+
body.top_p = options.topP;
|
|
122
|
+
}
|
|
123
|
+
if (options?.stop) {
|
|
124
|
+
body.stop = options.stop;
|
|
125
|
+
}
|
|
126
|
+
if (options?.jsonMode && capabilities?.supportsJsonMode) {
|
|
127
|
+
body.response_format = { type: 'json_object' };
|
|
128
|
+
}
|
|
129
|
+
// Handle streaming
|
|
130
|
+
if (options?.onStream && capabilities?.supportsStreaming) {
|
|
131
|
+
return this.handleStream(body, options.onStream, startTime, model, capabilities);
|
|
132
|
+
}
|
|
133
|
+
const response = await this.fetchWithTimeout(`${this.baseUrl}/v1/chat/completions`, {
|
|
134
|
+
method: 'POST',
|
|
135
|
+
headers: {
|
|
136
|
+
'Content-Type': 'application/json',
|
|
137
|
+
'Authorization': `Bearer ${this.apiKey}`,
|
|
138
|
+
},
|
|
139
|
+
body: JSON.stringify(body),
|
|
140
|
+
}, options?.timeout);
|
|
141
|
+
if (!response.ok) {
|
|
142
|
+
throw await this.parseErrorResponse(response);
|
|
143
|
+
}
|
|
144
|
+
const data = await response.json();
|
|
145
|
+
const latencyMs = Date.now() - startTime;
|
|
146
|
+
const usage = {
|
|
147
|
+
promptTokens: data.usage.prompt_tokens,
|
|
148
|
+
completionTokens: data.usage.completion_tokens,
|
|
149
|
+
totalTokens: data.usage.total_tokens,
|
|
150
|
+
estimatedCost: this.calculateCost({
|
|
151
|
+
promptTokens: data.usage.prompt_tokens,
|
|
152
|
+
completionTokens: data.usage.completion_tokens,
|
|
153
|
+
}, capabilities?.inputCostPer1M, capabilities?.outputCostPer1M),
|
|
154
|
+
};
|
|
155
|
+
this.trackUsage(usage);
|
|
156
|
+
return {
|
|
157
|
+
content: data.choices[0]?.message?.content ?? '',
|
|
158
|
+
model: data.model,
|
|
159
|
+
usage,
|
|
160
|
+
latencyMs,
|
|
161
|
+
streamed: false,
|
|
162
|
+
finishReason: data.choices[0]?.finish_reason ?? undefined,
|
|
163
|
+
};
|
|
164
|
+
}
|
|
165
|
+
catch (error) {
|
|
166
|
+
if (error instanceof LLMError)
|
|
167
|
+
throw error;
|
|
168
|
+
throw this.handleError(error);
|
|
169
|
+
}
|
|
170
|
+
}, `Groq chat completion`);
|
|
171
|
+
}
|
|
172
|
+
async handleStream(body, onStream, startTime, model, capabilities) {
|
|
173
|
+
const response = await this.fetchWithTimeout(`${this.baseUrl}/v1/chat/completions`, {
|
|
174
|
+
method: 'POST',
|
|
175
|
+
headers: {
|
|
176
|
+
'Content-Type': 'application/json',
|
|
177
|
+
'Authorization': `Bearer ${this.apiKey}`,
|
|
178
|
+
},
|
|
179
|
+
body: JSON.stringify({ ...body, stream: true }),
|
|
180
|
+
});
|
|
181
|
+
if (!response.ok) {
|
|
182
|
+
throw await this.parseErrorResponse(response);
|
|
183
|
+
}
|
|
184
|
+
let content = '';
|
|
185
|
+
let finishReason;
|
|
186
|
+
const reader = response.body?.getReader();
|
|
187
|
+
if (!reader) {
|
|
188
|
+
throw new LLMError('No response body', 'INVALID_RESPONSE', 'groq', false);
|
|
189
|
+
}
|
|
190
|
+
const decoder = new TextDecoder();
|
|
191
|
+
let buffer = '';
|
|
192
|
+
try {
|
|
193
|
+
while (true) {
|
|
194
|
+
const { done, value } = await reader.read();
|
|
195
|
+
if (done)
|
|
196
|
+
break;
|
|
197
|
+
buffer += decoder.decode(value, { stream: true });
|
|
198
|
+
const lines = buffer.split('\n');
|
|
199
|
+
buffer = lines.pop() ?? '';
|
|
200
|
+
for (const line of lines) {
|
|
201
|
+
if (!line.startsWith('data: '))
|
|
202
|
+
continue;
|
|
203
|
+
const jsonStr = line.slice(6).trim();
|
|
204
|
+
if (!jsonStr || jsonStr === '[DONE]')
|
|
205
|
+
continue;
|
|
206
|
+
try {
|
|
207
|
+
const chunk = JSON.parse(jsonStr);
|
|
208
|
+
const delta = chunk.choices?.[0]?.delta?.content;
|
|
209
|
+
if (delta) {
|
|
210
|
+
content += delta;
|
|
211
|
+
onStream(delta);
|
|
212
|
+
}
|
|
213
|
+
if (chunk.choices?.[0]?.finish_reason) {
|
|
214
|
+
finishReason = chunk.choices[0].finish_reason;
|
|
215
|
+
}
|
|
216
|
+
}
|
|
217
|
+
catch {
|
|
218
|
+
// Ignore malformed JSON
|
|
219
|
+
}
|
|
220
|
+
}
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
finally {
|
|
224
|
+
reader.releaseLock();
|
|
225
|
+
}
|
|
226
|
+
const latencyMs = Date.now() - startTime;
|
|
227
|
+
return {
|
|
228
|
+
content,
|
|
229
|
+
model,
|
|
230
|
+
latencyMs,
|
|
231
|
+
streamed: true,
|
|
232
|
+
finishReason,
|
|
233
|
+
};
|
|
234
|
+
}
|
|
235
|
+
async parseErrorResponse(response) {
|
|
236
|
+
try {
|
|
237
|
+
const data = await response.json();
|
|
238
|
+
const message = data.error?.message ?? 'Unknown error';
|
|
239
|
+
if (response.status === 401) {
|
|
240
|
+
return new LLMError('Groq authentication failed', 'AUTHENTICATION_FAILED', 'groq', false);
|
|
241
|
+
}
|
|
242
|
+
if (response.status === 429) {
|
|
243
|
+
return new LLMError('Groq rate limit exceeded', 'RATE_LIMITED', 'groq', true);
|
|
244
|
+
}
|
|
245
|
+
return new LLMError(`Groq API error: ${message}`, 'UNKNOWN', 'groq', response.status >= 500);
|
|
246
|
+
}
|
|
247
|
+
catch {
|
|
248
|
+
return new LLMError(`Groq API error (${response.status})`, 'UNKNOWN', 'groq', response.status >= 500);
|
|
249
|
+
}
|
|
250
|
+
}
|
|
251
|
+
handleError(error) {
|
|
252
|
+
if (error instanceof Error && error.name === 'AbortError') {
|
|
253
|
+
return new LLMError('Groq request timed out', 'TIMEOUT', 'groq', true, error);
|
|
254
|
+
}
|
|
255
|
+
return new LLMError(`Groq error: ${error instanceof Error ? error.message : String(error)}`, 'UNKNOWN', 'groq', false, error instanceof Error ? error : undefined);
|
|
256
|
+
}
|
|
257
|
+
formatModelName(modelId) {
|
|
258
|
+
return modelId
|
|
259
|
+
.replace(/-/g, ' ')
|
|
260
|
+
.replace(/\b\w/g, c => c.toUpperCase())
|
|
261
|
+
.replace('Llama', 'LLaMA')
|
|
262
|
+
.replace('It', 'Instruct');
|
|
263
|
+
}
|
|
264
|
+
}
|
|
265
|
+
//# sourceMappingURL=groq.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"groq.js","sourceRoot":"","sources":["../../../src/adapters/llm/groq.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAS5C,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAEtD,MAAM,WAAW,GAAsC;IACrD,yBAAyB,EAAE;QACzB,aAAa,EAAE,MAAM;QACrB,cAAc,EAAE,KAAK;QACrB,iBAAiB,EAAE,IAAI;QACvB,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,IAAI;QACvB,eAAe,EAAE,KAAK;QACtB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;KACtB;IACD,yBAAyB,EAAE;QACzB,aAAa,EAAE,MAAM;QACrB,cAAc,EAAE,KAAK;QACrB,iBAAiB,EAAE,IAAI;QACvB,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,IAAI;QACvB,eAAe,EAAE,IAAI;QACrB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;KACtB;IACD,sBAAsB,EAAE;QACtB,aAAa,EAAE,MAAM;QACrB,cAAc,EAAE,KAAK;QACrB,iBAAiB,EAAE,IAAI;QACvB,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,IAAI;QACvB,eAAe,EAAE,IAAI;QACrB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;KACtB;IACD,oBAAoB,EAAE;QACpB,aAAa,EAAE,KAAK;QACpB,cAAc,EAAE,KAAK;QACrB,iBAAiB,EAAE,IAAI;QACvB,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,IAAI;QACvB,eAAe,EAAE,KAAK;QACtB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;KACtB;IACD,cAAc,EAAE;QACd,aAAa,EAAE,IAAI;QACnB,cAAc,EAAE,KAAK;QACrB,iBAAiB,EAAE,IAAI;QACvB,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,IAAI;QACvB,eAAe,EAAE,IAAI;QACrB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;KACtB;IACD,8BAA8B,EAAE;QAC9B,aAAa,EAAE,MAAM;QACrB,cAAc,EAAE,IAAI;QACpB,iBAAiB,EAAE,IAAI;QACvB,gBAAgB,EAAE,IAAI;QACtB,iBAAiB,EAAE,IAAI;QACvB,eAAe,EAAE,IAAI;QACrB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;KACtB;CACF,CAAC;AAmBF,MAAM,OAAO,YAAa,SAAQ,eAAe;IACtC,IAAI,GAAG,MAAe,CAAC;IACvB,IAAI,GAAG,MAAM,CAAC;IAEf,MAAM,CAAS;IACf,OAAO,CAAS;IAChB,KAAK,CAAS;IAEtB,YAAY,MAA0B;QACpC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,6BAA6B,CAAC;QAC/D,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,YAAY,IAAI,yBAAyB,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC1C,GAAG,IAAI,CAAC,OAAO,YAAY,EAC3B;gBACE,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE,EAAE,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,EAAE;aACtD,EACD,IAAI,CACL,CAAC;YACF,OAAO,QAAQ,CAAC,EAAE,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9D,EAAE;YACF,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YAC9B,QAAQ,EAAE,MAAe;YACzB,YAAY;SACb,CAAC,CAAC,CAAC;IACN,CAAC;IAED,oBAAoB,CAAC,KAAa;QAChC,OAAO,WAAW,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,IAAI,CACR,QAAsB,EACtB,OAA8B;QAE9B,MAAM,KAAK,GAAG,OAAO,EAAE,OAAO,EAAE,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC;QAC1D,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,EAAE;YAC/B,IAAI,CAAC;gBACH,MAAM,IAAI,GAA4B;oBACpC,KAAK;oBACL,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;oBACnE,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,YAAY,EAAE,eAAe,IAAI,IAAI;iBACxE,CAAC;gBAEF,IAAI,OAAO,EAAE,WAAW,KAAK,SAAS,EAAE,CAAC;oBACvC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;gBACzC,CAAC;gBACD,IAAI,OAAO,EAAE,IAAI,KAAK,SAAS,EAAE,CAAC;oBAChC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC;gBAC5B,CAAC;gBACD,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;oBAClB,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;gBAC3B,CAAC;gBACD,IAAI,OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE,gBAAgB,EAAE,CAAC;oBACxD,IAAI,CAAC,eAAe,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;gBACjD,CAAC;gBAED,mBAAmB;gBACnB,IAAI,OAAO,EAAE,QAAQ,IAAI,YAAY,EAAE,iBAAiB,EAAE,CAAC;oBACzD,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;gBACnF,CAAC;gBAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC1C,GAAG,IAAI,CAAC,OAAO,sBAAsB,EACrC;oBACE,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE;wBACP,cAAc,EAAE,kBAAkB;wBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;qBACzC;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;iBAC3B,EACD,OAAO,EAAE,OAAO,CACjB,CAAC;gBAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,MAAM,MAAM,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;gBAChD,CAAC;gBAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAkB,CAAC;gBACnD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;gBAEzC,MAAM,KAAK,GAAG;oBACZ,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;oBACtC,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;oBAC9C,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;oBACpC,aAAa,EAAE,IAAI,CAAC,aAAa,CAC/B;wBACE,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;wBACtC,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;qBAC/C,EACD,YAAY,EAAE,cAAc,EAC5B,YAAY,EAAE,eAAe,CAC9B;iBACF,CAAC;gBAEF,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAEvB,OAAO;oBACL,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE;oBAChD,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,KAAK;oBACL,SAAS;oBACT,QAAQ,EAAE,KAAK;oBACf,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,IAAI,SAAS;iBAC1D,CAAC;YAEJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,QAAQ;oBAAE,MAAM,KAAK,CAAC;gBAC3C,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAChC,CAAC;QACH,CAAC,EAAE,sBAAsB,CAAC,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,IAA6B,EAC7B,QAAiC,EACjC,SAAiB,EACjB,KAAa,EACb,YAAsC;QAEtC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAC1C,GAAG,IAAI,CAAC,OAAO,sBAAsB,EACrC;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;aACzC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SAChD,CACF,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,MAAM,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,YAAgC,CAAC;QAErC,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,QAAQ,CAAC,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,IAAI,CAAC;YACH,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,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;wBAAE,SAAS;oBACzC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACrC,IAAI,CAAC,OAAO,IAAI,OAAO,KAAK,QAAQ;wBAAE,SAAS;oBAE/C,IAAI,CAAC;wBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAClC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC;wBACjD,IAAI,KAAK,EAAE,CAAC;4BACV,OAAO,IAAI,KAAK,CAAC;4BACjB,QAAQ,CAAC,KAAK,CAAC,CAAC;wBAClB,CAAC;wBACD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,CAAC;4BACtC,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;wBAChD,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,wBAAwB;oBAC1B,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAEzC,OAAO;YACL,OAAO;YACP,KAAK;YACL,SAAS;YACT,QAAQ,EAAE,IAAI;YACd,YAAY;SACb,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,QAAkB;QACjD,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAsC,CAAC;YACvE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,eAAe,CAAC;YAEvD,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,OAAO,IAAI,QAAQ,CAAC,4BAA4B,EAAE,uBAAuB,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;YAC5F,CAAC;YACD,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,OAAO,IAAI,QAAQ,CAAC,0BAA0B,EAAE,cAAc,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;YAChF,CAAC;YAED,OAAO,IAAI,QAAQ,CAAC,mBAAmB,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC;QAC/F,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,QAAQ,CAAC,mBAAmB,QAAQ,CAAC,MAAM,GAAG,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC;QACxG,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,KAAc;QAChC,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC1D,OAAO,IAAI,QAAQ,CAAC,wBAAwB,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,IAAI,QAAQ,CACjB,eAAe,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACvE,SAAS,EACT,MAAM,EACN,KAAK,EACL,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAC3C,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,OAAe;QACrC,OAAO,OAAO;aACX,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC;aAClB,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;aACtC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC;aACzB,OAAO,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAC/B,CAAC;CACF"}
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* LLM Provider Factory
|
|
3
|
+
*
|
|
4
|
+
* Creates and manages LLM providers with auto-detection, fallback chains,
|
|
5
|
+
* and configuration from environment variables and config files.
|
|
6
|
+
*/
|
|
7
|
+
import type { LLMProvider, LLMProviderType, LLMProviderConfig, LLMProviderFactoryConfig, LLMMessage, LLMCompletionOptions, LLMResponse } from '../../types/llmProvider.js';
|
|
8
|
+
export { OpenAIProvider } from './openai.js';
|
|
9
|
+
export { AnthropicProvider } from './anthropic.js';
|
|
10
|
+
export { DeepSeekProvider } from './deepseek.js';
|
|
11
|
+
export { GroqProvider } from './groq.js';
|
|
12
|
+
export { OllamaProvider } from './ollama.js';
|
|
13
|
+
export { OpenRouterProvider } from './openrouter.js';
|
|
14
|
+
export { BaseLLMProvider } from './base.js';
|
|
15
|
+
/**
|
|
16
|
+
* Create a provider from configuration
|
|
17
|
+
*/
|
|
18
|
+
export declare function createProvider(config: LLMProviderConfig): LLMProvider;
|
|
19
|
+
/**
|
|
20
|
+
* LLM Provider Manager with fallback support
|
|
21
|
+
*/
|
|
22
|
+
export declare class LLMProviderManager implements LLMProvider {
|
|
23
|
+
readonly type: LLMProviderType;
|
|
24
|
+
readonly name: string;
|
|
25
|
+
private providers;
|
|
26
|
+
private primaryProvider;
|
|
27
|
+
private totalTokens;
|
|
28
|
+
private totalCost;
|
|
29
|
+
constructor(configs: LLMProviderConfig[]);
|
|
30
|
+
isAvailable(): Promise<boolean>;
|
|
31
|
+
listModels(): Promise<import("../../types/llmProvider.js").ModelInfo[]>;
|
|
32
|
+
getModelCapabilities(model: string): import("../../types/llmProvider.js").ModelCapabilities | null;
|
|
33
|
+
complete(prompt: string, options?: LLMCompletionOptions): Promise<LLMResponse>;
|
|
34
|
+
chat(messages: LLMMessage[], options?: LLMCompletionOptions): Promise<LLMResponse>;
|
|
35
|
+
getTotalTokensUsed(): number;
|
|
36
|
+
getTotalCost(): number;
|
|
37
|
+
resetUsageTracking(): void;
|
|
38
|
+
/**
|
|
39
|
+
* Get the primary provider
|
|
40
|
+
*/
|
|
41
|
+
getPrimaryProvider(): LLMProvider;
|
|
42
|
+
/**
|
|
43
|
+
* Get all configured providers
|
|
44
|
+
*/
|
|
45
|
+
getAllProviders(): LLMProvider[];
|
|
46
|
+
/**
|
|
47
|
+
* Switch primary provider
|
|
48
|
+
*/
|
|
49
|
+
setPrimaryProvider(providerType: LLMProviderType): boolean;
|
|
50
|
+
}
|
|
51
|
+
export declare function getLLMProvider(config?: LLMProviderFactoryConfig): LLMProviderManager;
|
|
52
|
+
/**
|
|
53
|
+
* Create a single provider directly (without fallbacks)
|
|
54
|
+
*/
|
|
55
|
+
export declare function createSingleProvider(providerType: LLMProviderType, config?: Partial<LLMProviderConfig>): LLMProvider;
|
|
56
|
+
/**
|
|
57
|
+
* Check which providers are available
|
|
58
|
+
*/
|
|
59
|
+
export declare function checkAvailableProviders(): Promise<Record<LLMProviderType, boolean>>;
|
|
60
|
+
export type { LLMProvider, LLMProviderType, LLMProviderConfig, LLMConfigFile, LLMMessage, LLMCompletionOptions, LLMResponse, ModelInfo, ModelCapabilities, TokenUsage, } from '../../types/llmProvider.js';
|
|
61
|
+
export { LLMError, LLM_ENV_VARS, DEFAULT_MODELS } from '../../types/llmProvider.js';
|
|
62
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/adapters/llm/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH,OAAO,KAAK,EACV,WAAW,EACX,eAAe,EACf,iBAAiB,EAEjB,wBAAwB,EACxB,UAAU,EACV,oBAAoB,EACpB,WAAW,EACZ,MAAM,4BAA4B,CAAC;AAWpC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAE5C;;GAEG;AACH,wBAAgB,cAAc,CAAC,MAAM,EAAE,iBAAiB,GAAG,WAAW,CAqBrE;AA4JD;;GAEG;AACH,qBAAa,kBAAmB,YAAW,WAAW;IACpD,QAAQ,CAAC,IAAI,EAAE,eAAe,CAAC;IAC/B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAEtB,OAAO,CAAC,SAAS,CAAqB;IACtC,OAAO,CAAC,eAAe,CAAc;IACrC,OAAO,CAAC,WAAW,CAAa;IAChC,OAAO,CAAC,SAAS,CAAa;gBAElB,OAAO,EAAE,iBAAiB,EAAE;IAWlC,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;IAS/B,UAAU;IAIhB,oBAAoB,CAAC,KAAK,EAAE,MAAM;IAI5B,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO,CAAC,WAAW,CAAC;IAI9E,IAAI,CAAC,QAAQ,EAAE,UAAU,EAAE,EAAE,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO,CAAC,WAAW,CAAC;IAuCxF,kBAAkB,IAAI,MAAM;IAI5B,YAAY,IAAI,MAAM;IAItB,kBAAkB,IAAI,IAAI;IAQ1B;;OAEG;IACH,kBAAkB,IAAI,WAAW;IAIjC;;OAEG;IACH,eAAe,IAAI,WAAW,EAAE;IAIhC;;OAEG;IACH,kBAAkB,CAAC,YAAY,EAAE,eAAe,GAAG,OAAO;CAQ3D;AAOD,wBAAgB,cAAc,CAAC,MAAM,CAAC,EAAE,wBAAwB,GAAG,kBAAkB,CAuDpF;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,YAAY,EAAE,eAAe,EAC7B,MAAM,CAAC,EAAE,OAAO,CAAC,iBAAiB,CAAC,GAClC,WAAW,CAgBb;AAED;;GAEG;AACH,wBAAsB,uBAAuB,IAAI,OAAO,CAAC,MAAM,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,CAsBzF;AAGD,YAAY,EACV,WAAW,EACX,eAAe,EACf,iBAAiB,EACjB,aAAa,EACb,UAAU,EACV,oBAAoB,EACpB,WAAW,EACX,SAAS,EACT,iBAAiB,EACjB,UAAU,GACX,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC"}
|