mcp-rubber-duck 1.1.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/.dockerignore +19 -0
- package/.env.desktop.example +145 -0
- package/.env.example +45 -0
- package/.env.pi.example +106 -0
- package/.env.template +165 -0
- package/.eslintrc.json +40 -0
- package/.github/ISSUE_TEMPLATE/bug_report.md +65 -0
- package/.github/ISSUE_TEMPLATE/feature_request.md +58 -0
- package/.github/ISSUE_TEMPLATE/question.md +67 -0
- package/.github/pull_request_template.md +111 -0
- package/.github/workflows/docker-build.yml +138 -0
- package/.github/workflows/release.yml +182 -0
- package/.github/workflows/security.yml +141 -0
- package/.github/workflows/semantic-release.yml +89 -0
- package/.prettierrc +10 -0
- package/.releaserc.json +66 -0
- package/CHANGELOG.md +95 -0
- package/CONTRIBUTING.md +242 -0
- package/Dockerfile +62 -0
- package/LICENSE +21 -0
- package/README.md +803 -0
- package/audit-ci.json +8 -0
- package/config/claude_desktop.json +14 -0
- package/config/config.example.json +91 -0
- package/dist/config/config.d.ts +51 -0
- package/dist/config/config.d.ts.map +1 -0
- package/dist/config/config.js +301 -0
- package/dist/config/config.js.map +1 -0
- package/dist/config/types.d.ts +356 -0
- package/dist/config/types.d.ts.map +1 -0
- package/dist/config/types.js +41 -0
- package/dist/config/types.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +109 -0
- package/dist/index.js.map +1 -0
- package/dist/providers/duck-provider-enhanced.d.ts +29 -0
- package/dist/providers/duck-provider-enhanced.d.ts.map +1 -0
- package/dist/providers/duck-provider-enhanced.js +230 -0
- package/dist/providers/duck-provider-enhanced.js.map +1 -0
- package/dist/providers/enhanced-manager.d.ts +54 -0
- package/dist/providers/enhanced-manager.d.ts.map +1 -0
- package/dist/providers/enhanced-manager.js +217 -0
- package/dist/providers/enhanced-manager.js.map +1 -0
- package/dist/providers/manager.d.ts +28 -0
- package/dist/providers/manager.d.ts.map +1 -0
- package/dist/providers/manager.js +204 -0
- package/dist/providers/manager.js.map +1 -0
- package/dist/providers/provider.d.ts +29 -0
- package/dist/providers/provider.d.ts.map +1 -0
- package/dist/providers/provider.js +179 -0
- package/dist/providers/provider.js.map +1 -0
- package/dist/providers/types.d.ts +69 -0
- package/dist/providers/types.d.ts.map +1 -0
- package/dist/providers/types.js +2 -0
- package/dist/providers/types.js.map +1 -0
- package/dist/server.d.ts +24 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +501 -0
- package/dist/server.js.map +1 -0
- package/dist/services/approval.d.ts +44 -0
- package/dist/services/approval.d.ts.map +1 -0
- package/dist/services/approval.js +159 -0
- package/dist/services/approval.js.map +1 -0
- package/dist/services/cache.d.ts +21 -0
- package/dist/services/cache.d.ts.map +1 -0
- package/dist/services/cache.js +63 -0
- package/dist/services/cache.js.map +1 -0
- package/dist/services/conversation.d.ts +24 -0
- package/dist/services/conversation.d.ts.map +1 -0
- package/dist/services/conversation.js +108 -0
- package/dist/services/conversation.js.map +1 -0
- package/dist/services/function-bridge.d.ts +41 -0
- package/dist/services/function-bridge.d.ts.map +1 -0
- package/dist/services/function-bridge.js +259 -0
- package/dist/services/function-bridge.js.map +1 -0
- package/dist/services/health.d.ts +17 -0
- package/dist/services/health.d.ts.map +1 -0
- package/dist/services/health.js +77 -0
- package/dist/services/health.js.map +1 -0
- package/dist/services/mcp-client-manager.d.ts +49 -0
- package/dist/services/mcp-client-manager.d.ts.map +1 -0
- package/dist/services/mcp-client-manager.js +279 -0
- package/dist/services/mcp-client-manager.js.map +1 -0
- package/dist/tools/approve-mcp-request.d.ts +9 -0
- package/dist/tools/approve-mcp-request.d.ts.map +1 -0
- package/dist/tools/approve-mcp-request.js +111 -0
- package/dist/tools/approve-mcp-request.js.map +1 -0
- package/dist/tools/ask-duck.d.ts +9 -0
- package/dist/tools/ask-duck.d.ts.map +1 -0
- package/dist/tools/ask-duck.js +43 -0
- package/dist/tools/ask-duck.js.map +1 -0
- package/dist/tools/chat-duck.d.ts +9 -0
- package/dist/tools/chat-duck.d.ts.map +1 -0
- package/dist/tools/chat-duck.js +57 -0
- package/dist/tools/chat-duck.js.map +1 -0
- package/dist/tools/clear-conversations.d.ts +8 -0
- package/dist/tools/clear-conversations.d.ts.map +1 -0
- package/dist/tools/clear-conversations.js +17 -0
- package/dist/tools/clear-conversations.js.map +1 -0
- package/dist/tools/compare-ducks.d.ts +8 -0
- package/dist/tools/compare-ducks.d.ts.map +1 -0
- package/dist/tools/compare-ducks.js +49 -0
- package/dist/tools/compare-ducks.js.map +1 -0
- package/dist/tools/duck-council.d.ts +8 -0
- package/dist/tools/duck-council.d.ts.map +1 -0
- package/dist/tools/duck-council.js +69 -0
- package/dist/tools/duck-council.js.map +1 -0
- package/dist/tools/get-pending-approvals.d.ts +15 -0
- package/dist/tools/get-pending-approvals.d.ts.map +1 -0
- package/dist/tools/get-pending-approvals.js +74 -0
- package/dist/tools/get-pending-approvals.js.map +1 -0
- package/dist/tools/list-ducks.d.ts +9 -0
- package/dist/tools/list-ducks.d.ts.map +1 -0
- package/dist/tools/list-ducks.js +47 -0
- package/dist/tools/list-ducks.js.map +1 -0
- package/dist/tools/list-models.d.ts +8 -0
- package/dist/tools/list-models.d.ts.map +1 -0
- package/dist/tools/list-models.js +72 -0
- package/dist/tools/list-models.js.map +1 -0
- package/dist/tools/mcp-status.d.ts +17 -0
- package/dist/tools/mcp-status.d.ts.map +1 -0
- package/dist/tools/mcp-status.js +100 -0
- package/dist/tools/mcp-status.js.map +1 -0
- package/dist/utils/ascii-art.d.ts +19 -0
- package/dist/utils/ascii-art.d.ts.map +1 -0
- package/dist/utils/ascii-art.js +73 -0
- package/dist/utils/ascii-art.js.map +1 -0
- package/dist/utils/logger.d.ts +3 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +86 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/safe-logger.d.ts +23 -0
- package/dist/utils/safe-logger.d.ts.map +1 -0
- package/dist/utils/safe-logger.js +145 -0
- package/dist/utils/safe-logger.js.map +1 -0
- package/docker-compose.yml +161 -0
- package/jest.config.js +26 -0
- package/package.json +65 -0
- package/scripts/build-multiarch.sh +290 -0
- package/scripts/deploy-raspbian.sh +410 -0
- package/scripts/deploy.sh +322 -0
- package/scripts/gh-deploy.sh +343 -0
- package/scripts/setup-docker-raspbian.sh +530 -0
- package/server.json +8 -0
- package/src/config/config.ts +357 -0
- package/src/config/types.ts +89 -0
- package/src/index.ts +114 -0
- package/src/providers/duck-provider-enhanced.ts +294 -0
- package/src/providers/enhanced-manager.ts +290 -0
- package/src/providers/manager.ts +257 -0
- package/src/providers/provider.ts +207 -0
- package/src/providers/types.ts +78 -0
- package/src/server.ts +603 -0
- package/src/services/approval.ts +225 -0
- package/src/services/cache.ts +79 -0
- package/src/services/conversation.ts +146 -0
- package/src/services/function-bridge.ts +329 -0
- package/src/services/health.ts +107 -0
- package/src/services/mcp-client-manager.ts +362 -0
- package/src/tools/approve-mcp-request.ts +126 -0
- package/src/tools/ask-duck.ts +74 -0
- package/src/tools/chat-duck.ts +82 -0
- package/src/tools/clear-conversations.ts +24 -0
- package/src/tools/compare-ducks.ts +67 -0
- package/src/tools/duck-council.ts +88 -0
- package/src/tools/get-pending-approvals.ts +90 -0
- package/src/tools/list-ducks.ts +65 -0
- package/src/tools/list-models.ts +101 -0
- package/src/tools/mcp-status.ts +117 -0
- package/src/utils/ascii-art.ts +85 -0
- package/src/utils/logger.ts +116 -0
- package/src/utils/safe-logger.ts +165 -0
- package/systemd/mcp-rubber-duck-with-ollama.service +55 -0
- package/systemd/mcp-rubber-duck.service +58 -0
- package/test-functionality.js +147 -0
- package/test-mcp-interface.js +221 -0
- package/tests/ascii-art.test.ts +36 -0
- package/tests/config.test.ts +239 -0
- package/tests/conversation.test.ts +308 -0
- package/tests/mcp-bridge.test.ts +291 -0
- package/tests/providers.test.ts +269 -0
- package/tests/tools/clear-conversations.test.ts +163 -0
- package/tsconfig.json +26 -0
|
@@ -0,0 +1,230 @@
|
|
|
1
|
+
import { DuckProvider } from './provider.js';
|
|
2
|
+
import { logger } from '../utils/logger.js';
|
|
3
|
+
import { SafeLogger } from '../utils/safe-logger.js';
|
|
4
|
+
export class EnhancedDuckProvider extends DuckProvider {
|
|
5
|
+
functionBridge;
|
|
6
|
+
mcpEnabled;
|
|
7
|
+
constructor(name, nickname, options, functionBridge, mcpEnabled = true) {
|
|
8
|
+
super(name, nickname, options);
|
|
9
|
+
this.functionBridge = functionBridge;
|
|
10
|
+
this.mcpEnabled = mcpEnabled;
|
|
11
|
+
}
|
|
12
|
+
async chat(options) {
|
|
13
|
+
try {
|
|
14
|
+
// If MCP is enabled, add function definitions
|
|
15
|
+
if (this.mcpEnabled) {
|
|
16
|
+
const functions = await this.functionBridge.getFunctionDefinitions();
|
|
17
|
+
if (functions.length > 0) {
|
|
18
|
+
options.tools = functions;
|
|
19
|
+
options.toolChoice = 'auto';
|
|
20
|
+
logger.debug(`Added ${functions.length} MCP functions for ${this.nickname}`);
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
// Prepare messages for function calling
|
|
24
|
+
const messages = this.prepareMessages(options.messages, options.systemPrompt);
|
|
25
|
+
const modelToUse = options.model || this.options.model;
|
|
26
|
+
const baseParams = {
|
|
27
|
+
model: modelToUse,
|
|
28
|
+
messages: messages,
|
|
29
|
+
stream: false,
|
|
30
|
+
};
|
|
31
|
+
// Add temperature if model supports it
|
|
32
|
+
if (this.supportsTemperature(modelToUse)) {
|
|
33
|
+
baseParams.temperature = options.temperature ?? this.options.temperature ?? 0.7;
|
|
34
|
+
}
|
|
35
|
+
// Add tools if available
|
|
36
|
+
if (options.tools && options.tools.length > 0) {
|
|
37
|
+
baseParams.tools = options.tools.map(tool => ({
|
|
38
|
+
type: 'function',
|
|
39
|
+
function: {
|
|
40
|
+
name: tool.name,
|
|
41
|
+
description: tool.description,
|
|
42
|
+
parameters: tool.parameters,
|
|
43
|
+
},
|
|
44
|
+
}));
|
|
45
|
+
baseParams.tool_choice = options.toolChoice || 'auto';
|
|
46
|
+
}
|
|
47
|
+
// First API call
|
|
48
|
+
const response = await this.createChatCompletion(baseParams);
|
|
49
|
+
const choice = response.choices[0];
|
|
50
|
+
// Check if the model wants to call functions
|
|
51
|
+
if (choice.message?.tool_calls && choice.message.tool_calls.length > 0) {
|
|
52
|
+
return await this.handleToolCalls(choice.message.tool_calls, messages, baseParams, modelToUse);
|
|
53
|
+
}
|
|
54
|
+
// No tool calls, return regular response
|
|
55
|
+
return {
|
|
56
|
+
content: choice.message?.content || '',
|
|
57
|
+
usage: response.usage ? {
|
|
58
|
+
promptTokens: response.usage.prompt_tokens,
|
|
59
|
+
completionTokens: response.usage.completion_tokens,
|
|
60
|
+
totalTokens: response.usage.total_tokens,
|
|
61
|
+
} : undefined,
|
|
62
|
+
model: modelToUse,
|
|
63
|
+
finishReason: choice.finish_reason || undefined,
|
|
64
|
+
};
|
|
65
|
+
}
|
|
66
|
+
catch (error) {
|
|
67
|
+
logger.error(`Enhanced provider ${this.name} chat error:`, error);
|
|
68
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
69
|
+
throw new Error(`Duck ${this.nickname} couldn't respond: ${errorMessage}`);
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
async handleToolCalls(toolCalls, messages, baseParams, modelToUse) {
|
|
73
|
+
const pendingApprovals = [];
|
|
74
|
+
const toolMessages = [];
|
|
75
|
+
let hasExecutedTools = false;
|
|
76
|
+
// Add the assistant message with tool calls
|
|
77
|
+
const assistantMessage = {
|
|
78
|
+
role: 'assistant',
|
|
79
|
+
content: null,
|
|
80
|
+
tool_calls: toolCalls,
|
|
81
|
+
};
|
|
82
|
+
messages.push(assistantMessage);
|
|
83
|
+
// Process each tool call
|
|
84
|
+
for (const toolCall of toolCalls) {
|
|
85
|
+
try {
|
|
86
|
+
const functionName = toolCall.function.name;
|
|
87
|
+
const args = JSON.parse(toolCall.function.arguments);
|
|
88
|
+
logger.info(`${this.nickname} wants to call function: ${functionName}`);
|
|
89
|
+
SafeLogger.debug(`Function call arguments for ${functionName}:`, args);
|
|
90
|
+
const result = await this.functionBridge.handleFunctionCall(this.nickname, functionName, args);
|
|
91
|
+
if (result.needsApproval && result.approvalId) {
|
|
92
|
+
// Function needs approval
|
|
93
|
+
pendingApprovals.push({
|
|
94
|
+
id: result.approvalId,
|
|
95
|
+
message: result.message || `Approval needed for ${functionName}`,
|
|
96
|
+
});
|
|
97
|
+
// Add a tool message indicating approval is needed
|
|
98
|
+
toolMessages.push({
|
|
99
|
+
role: 'tool',
|
|
100
|
+
tool_call_id: toolCall.id,
|
|
101
|
+
content: JSON.stringify({
|
|
102
|
+
status: 'approval_needed',
|
|
103
|
+
approval_id: result.approvalId,
|
|
104
|
+
message: result.message,
|
|
105
|
+
}),
|
|
106
|
+
});
|
|
107
|
+
}
|
|
108
|
+
else if (result.success && result.data) {
|
|
109
|
+
// Function executed successfully
|
|
110
|
+
hasExecutedTools = true;
|
|
111
|
+
toolMessages.push({
|
|
112
|
+
role: 'tool',
|
|
113
|
+
tool_call_id: toolCall.id,
|
|
114
|
+
content: typeof result.data === 'string'
|
|
115
|
+
? result.data
|
|
116
|
+
: JSON.stringify(result.data),
|
|
117
|
+
});
|
|
118
|
+
}
|
|
119
|
+
else {
|
|
120
|
+
// Function failed
|
|
121
|
+
toolMessages.push({
|
|
122
|
+
role: 'tool',
|
|
123
|
+
tool_call_id: toolCall.id,
|
|
124
|
+
content: JSON.stringify({
|
|
125
|
+
error: result.error || 'Unknown error',
|
|
126
|
+
}),
|
|
127
|
+
});
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
catch (error) {
|
|
131
|
+
logger.error(`Error processing tool call ${toolCall.id}:`, error);
|
|
132
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
133
|
+
toolMessages.push({
|
|
134
|
+
role: 'tool',
|
|
135
|
+
tool_call_id: toolCall.id,
|
|
136
|
+
content: JSON.stringify({
|
|
137
|
+
error: `Tool execution failed: ${errorMessage}`,
|
|
138
|
+
}),
|
|
139
|
+
});
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
// If we have pending approvals, return them without calling the model again
|
|
143
|
+
if (pendingApprovals.length > 0) {
|
|
144
|
+
const approvalMessage = pendingApprovals.length === 1
|
|
145
|
+
? pendingApprovals[0].message
|
|
146
|
+
: `Multiple approvals needed: ${pendingApprovals.map(a => a.id).join(', ')}`;
|
|
147
|
+
return {
|
|
148
|
+
content: `⏳ ${approvalMessage}`,
|
|
149
|
+
model: modelToUse,
|
|
150
|
+
pendingApprovals,
|
|
151
|
+
finishReason: 'tool_calls',
|
|
152
|
+
};
|
|
153
|
+
}
|
|
154
|
+
// Add tool messages and call model again for final response
|
|
155
|
+
messages.push(...toolMessages);
|
|
156
|
+
// Remove tools from the follow-up call to get a natural language response
|
|
157
|
+
const followUpParams = {
|
|
158
|
+
...baseParams,
|
|
159
|
+
messages,
|
|
160
|
+
};
|
|
161
|
+
delete followUpParams.tools;
|
|
162
|
+
delete followUpParams.tool_choice;
|
|
163
|
+
const finalResponse = await this.createChatCompletion(followUpParams);
|
|
164
|
+
const finalChoice = finalResponse.choices[0];
|
|
165
|
+
return {
|
|
166
|
+
content: finalChoice.message?.content || '',
|
|
167
|
+
usage: finalResponse.usage ? {
|
|
168
|
+
promptTokens: finalResponse.usage.prompt_tokens,
|
|
169
|
+
completionTokens: finalResponse.usage.completion_tokens,
|
|
170
|
+
totalTokens: finalResponse.usage.total_tokens,
|
|
171
|
+
} : undefined,
|
|
172
|
+
model: modelToUse,
|
|
173
|
+
finishReason: finalChoice.finish_reason || undefined,
|
|
174
|
+
mcpResults: hasExecutedTools ? toolMessages : undefined,
|
|
175
|
+
};
|
|
176
|
+
}
|
|
177
|
+
// Method to retry with approval
|
|
178
|
+
async retryWithApproval(approvalId, originalMessages, options) {
|
|
179
|
+
// Add approval ID to the tool arguments
|
|
180
|
+
if (options.tools) {
|
|
181
|
+
options.tools = options.tools.map(tool => ({
|
|
182
|
+
...tool,
|
|
183
|
+
parameters: {
|
|
184
|
+
...tool.parameters,
|
|
185
|
+
properties: {
|
|
186
|
+
...tool.parameters.properties,
|
|
187
|
+
_approval_id: {
|
|
188
|
+
type: 'string',
|
|
189
|
+
default: approvalId,
|
|
190
|
+
},
|
|
191
|
+
},
|
|
192
|
+
},
|
|
193
|
+
}));
|
|
194
|
+
}
|
|
195
|
+
// Retry the chat with the approval ID
|
|
196
|
+
return this.chat({
|
|
197
|
+
...options,
|
|
198
|
+
messages: originalMessages,
|
|
199
|
+
});
|
|
200
|
+
}
|
|
201
|
+
// Check if MCP functions are available
|
|
202
|
+
async getMCPFunctionCount() {
|
|
203
|
+
if (!this.mcpEnabled) {
|
|
204
|
+
return 0;
|
|
205
|
+
}
|
|
206
|
+
try {
|
|
207
|
+
const functions = await this.functionBridge.getFunctionDefinitions();
|
|
208
|
+
return functions.length;
|
|
209
|
+
}
|
|
210
|
+
catch (error) {
|
|
211
|
+
return 0;
|
|
212
|
+
}
|
|
213
|
+
}
|
|
214
|
+
// Get MCP bridge statistics
|
|
215
|
+
getMCPStats() {
|
|
216
|
+
if (!this.mcpEnabled) {
|
|
217
|
+
return null;
|
|
218
|
+
}
|
|
219
|
+
return this.functionBridge.getStats();
|
|
220
|
+
}
|
|
221
|
+
// Enable/disable MCP for this provider
|
|
222
|
+
setMCPEnabled(enabled) {
|
|
223
|
+
this.mcpEnabled = enabled;
|
|
224
|
+
logger.info(`MCP ${enabled ? 'enabled' : 'disabled'} for ${this.nickname}`);
|
|
225
|
+
}
|
|
226
|
+
isMCPEnabled() {
|
|
227
|
+
return this.mcpEnabled;
|
|
228
|
+
}
|
|
229
|
+
}
|
|
230
|
+
//# sourceMappingURL=duck-provider-enhanced.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"duck-provider-enhanced.js","sourceRoot":"","sources":["../../src/providers/duck-provider-enhanced.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAI7C,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAUrD,MAAM,OAAO,oBAAqB,SAAQ,YAAY;IAC5C,cAAc,CAAiB;IAC/B,UAAU,CAAU;IAE5B,YACE,IAAY,EACZ,QAAgB,EAChB,OAAwB,EACxB,cAA8B,EAC9B,aAAsB,IAAI;QAE1B,KAAK,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC/B,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,IAAI,CAAC;YACH,8CAA8C;YAC9C,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,sBAAsB,EAAE,CAAC;gBACrE,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzB,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC;oBAC1B,OAAO,CAAC,UAAU,GAAG,MAAM,CAAC;oBAC5B,MAAM,CAAC,KAAK,CAAC,SAAS,SAAS,CAAC,MAAM,sBAAsB,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC/E,CAAC;YACH,CAAC;YAED,wCAAwC;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;YAC9E,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;YAEvD,MAAM,UAAU,GAA8B;gBAC5C,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,QAA2B;gBACrC,MAAM,EAAE,KAAK;aACd,CAAC;YAEF,uCAAuC;YACvC,IAAI,IAAI,CAAC,mBAAmB,CAAC,UAAU,CAAC,EAAE,CAAC;gBACzC,UAAU,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,GAAG,CAAC;YAClF,CAAC;YAED,yBAAyB;YACzB,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9C,UAAU,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAC5C,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACR,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,WAAW,EAAE,IAAI,CAAC,WAAW;wBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;qBAC5B;iBACF,CAAC,CAAC,CAAC;gBACJ,UAAU,CAAC,WAAW,GAAG,OAAO,CAAC,UAAU,IAAI,MAAM,CAAC;YACxD,CAAC;YAED,iBAAiB;YACjB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAEnC,6CAA6C;YAC7C,IAAI,MAAM,CAAC,OAAO,EAAE,UAAU,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvE,OAAO,MAAM,IAAI,CAAC,eAAe,CAC/B,MAAM,CAAC,OAAO,CAAC,UAAU,EACzB,QAA2B,EAC3B,UAAU,EACV,UAAU,CACX,CAAC;YACJ,CAAC;YAED,yCAAyC;YACzC,OAAO;gBACL,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE;gBACtC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;oBACtB,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;oBAC1C,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,iBAAiB;oBAClD,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;iBACzC,CAAC,CAAC,CAAC,SAAS;gBACb,KAAK,EAAE,UAAU;gBACjB,YAAY,EAAE,MAAM,CAAC,aAAa,IAAI,SAAS;aAChD,CAAC;QAEJ,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,IAAI,cAAc,EAAE,KAAK,CAAC,CAAC;YAClE,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,QAAQ,sBAAsB,YAAY,EAAE,CAAC,CAAC;QAC7E,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,SAA2B,EAC3B,QAAyB,EACzB,UAAqC,EACrC,UAAkB;QAElB,MAAM,gBAAgB,GAAsC,EAAE,CAAC;QAC/D,MAAM,YAAY,GAAoB,EAAE,CAAC;QACzC,IAAI,gBAAgB,GAAG,KAAK,CAAC;QAE7B,4CAA4C;QAC5C,MAAM,gBAAgB,GAAkB;YACtC,IAAI,EAAE,WAAoB;YAC1B,OAAO,EAAE,IAAI;YACb,UAAU,EAAE,SAAS;SACtB,CAAC;QACF,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAEhC,yBAAyB;QACzB,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,MAAM,YAAY,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAC5C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAA4B,CAAC;gBAEhF,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,QAAQ,4BAA4B,YAAY,EAAE,CAAC,CAAC;gBACxE,UAAU,CAAC,KAAK,CAAC,+BAA+B,YAAY,GAAG,EAAE,IAAI,CAAC,CAAC;gBAEvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,kBAAkB,CACzD,IAAI,CAAC,QAAQ,EACb,YAAY,EACZ,IAAI,CACL,CAAC;gBAEF,IAAI,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;oBAC9C,0BAA0B;oBAC1B,gBAAgB,CAAC,IAAI,CAAC;wBACpB,EAAE,EAAE,MAAM,CAAC,UAAU;wBACrB,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,uBAAuB,YAAY,EAAE;qBACjE,CAAC,CAAC;oBAEH,mDAAmD;oBACnD,YAAY,CAAC,IAAI,CAAC;wBAChB,IAAI,EAAE,MAAM;wBACZ,YAAY,EAAE,QAAQ,CAAC,EAAE;wBACzB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;4BACtB,MAAM,EAAE,iBAAiB;4BACzB,WAAW,EAAE,MAAM,CAAC,UAAU;4BAC9B,OAAO,EAAE,MAAM,CAAC,OAAO;yBACxB,CAAC;qBACH,CAAC,CAAC;gBAEL,CAAC;qBAAM,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;oBACzC,iCAAiC;oBACjC,gBAAgB,GAAG,IAAI,CAAC;oBACxB,YAAY,CAAC,IAAI,CAAC;wBAChB,IAAI,EAAE,MAAM;wBACZ,YAAY,EAAE,QAAQ,CAAC,EAAE;wBACzB,OAAO,EAAE,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ;4BACtC,CAAC,CAAC,MAAM,CAAC,IAAI;4BACb,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC;qBAChC,CAAC,CAAC;gBAEL,CAAC;qBAAM,CAAC;oBACN,kBAAkB;oBAClB,YAAY,CAAC,IAAI,CAAC;wBAChB,IAAI,EAAE,MAAM;wBACZ,YAAY,EAAE,QAAQ,CAAC,EAAE;wBACzB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;4BACtB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,eAAe;yBACvC,CAAC;qBACH,CAAC,CAAC;gBACL,CAAC;YAEH,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,MAAM,CAAC,KAAK,CAAC,8BAA8B,QAAQ,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;gBAClE,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC5E,YAAY,CAAC,IAAI,CAAC;oBAChB,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,QAAQ,CAAC,EAAE;oBACzB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC;wBACtB,KAAK,EAAE,0BAA0B,YAAY,EAAE;qBAChD,CAAC;iBACH,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,4EAA4E;QAC5E,IAAI,gBAAgB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,MAAM,eAAe,GAAG,gBAAgB,CAAC,MAAM,KAAK,CAAC;gBACnD,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,OAAO;gBAC7B,CAAC,CAAC,8BAA8B,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAE/E,OAAO;gBACL,OAAO,EAAE,KAAK,eAAe,EAAE;gBAC/B,KAAK,EAAE,UAAU;gBACjB,gBAAgB;gBAChB,YAAY,EAAE,YAAY;aAC3B,CAAC;QACJ,CAAC;QAED,4DAA4D;QAC5D,QAAQ,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;QAE/B,0EAA0E;QAC1E,MAAM,cAAc,GAAG;YACrB,GAAG,UAAU;YACb,QAAQ;SACT,CAAC;QACF,OAAO,cAAc,CAAC,KAAK,CAAC;QAC5B,OAAO,cAAc,CAAC,WAAW,CAAC;QAElC,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC;QACtE,MAAM,WAAW,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAE7C,OAAO;YACL,OAAO,EAAE,WAAW,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE;YAC3C,KAAK,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC3B,YAAY,EAAE,aAAa,CAAC,KAAK,CAAC,aAAa;gBAC/C,gBAAgB,EAAE,aAAa,CAAC,KAAK,CAAC,iBAAiB;gBACvD,WAAW,EAAE,aAAa,CAAC,KAAK,CAAC,YAAY;aAC9C,CAAC,CAAC,CAAC,SAAS;YACb,KAAK,EAAE,UAAU;YACjB,YAAY,EAAE,WAAW,CAAC,aAAa,IAAI,SAAS;YACpD,UAAU,EAAE,gBAAgB,CAAC,CAAC,CAAE,YAAuC,CAAC,CAAC,CAAC,SAAS;SACpF,CAAC;IACJ,CAAC;IAED,gCAAgC;IAChC,KAAK,CAAC,iBAAiB,CACrB,UAAkB,EAClB,gBAAuC,EACvC,OAAoB;QAEpB,wCAAwC;QACxC,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACzC,GAAG,IAAI;gBACP,UAAU,EAAE;oBACV,GAAG,IAAI,CAAC,UAAU;oBAClB,UAAU,EAAE;wBACV,GAAI,IAAI,CAAC,UAAU,CAAC,UAAsC;wBAC1D,YAAY,EAAE;4BACZ,IAAI,EAAE,QAAQ;4BACd,OAAO,EAAE,UAAU;yBACpB;qBACF;iBACF;aACF,CAAC,CAAC,CAAC;QACN,CAAC;QAED,sCAAsC;QACtC,OAAO,IAAI,CAAC,IAAI,CAAC;YACf,GAAG,OAAO;YACV,QAAQ,EAAE,gBAAgB;SAC3B,CAAC,CAAC;IACL,CAAC;IAED,uCAAuC;IACvC,KAAK,CAAC,mBAAmB;QACvB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,OAAO,CAAC,CAAC;QACX,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,sBAAsB,EAAE,CAAC;YACrE,OAAO,SAAS,CAAC,MAAM,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IAED,4BAA4B;IAC5B,WAAW;QACT,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC;IACxC,CAAC;IAED,uCAAuC;IACvC,aAAa,CAAC,OAAgB;QAC5B,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC;QAC1B,MAAM,CAAC,IAAI,CAAC,OAAO,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;CACF"}
|
|
@@ -0,0 +1,54 @@
|
|
|
1
|
+
import { EnhancedDuckProvider } from './duck-provider-enhanced.js';
|
|
2
|
+
import { ProviderManager } from './manager.js';
|
|
3
|
+
import { ConfigManager } from '../config/config.js';
|
|
4
|
+
import { FunctionBridge } from '../services/function-bridge.js';
|
|
5
|
+
import { DuckResponse } from '../config/types.js';
|
|
6
|
+
import { ChatOptions, MCPResult } from './types.js';
|
|
7
|
+
export declare class EnhancedProviderManager extends ProviderManager {
|
|
8
|
+
private enhancedProviders;
|
|
9
|
+
private functionBridge?;
|
|
10
|
+
private mcpEnabled;
|
|
11
|
+
constructor(configManager: ConfigManager, functionBridge?: FunctionBridge);
|
|
12
|
+
private initializeEnhancedProviders;
|
|
13
|
+
getEnhancedProvider(name?: string): EnhancedDuckProvider;
|
|
14
|
+
askDuckWithMCP(providerName: string | undefined, prompt: string, options?: Partial<ChatOptions>): Promise<DuckResponse & {
|
|
15
|
+
pendingApprovals?: {
|
|
16
|
+
id: string;
|
|
17
|
+
message: string;
|
|
18
|
+
}[];
|
|
19
|
+
mcpResults?: MCPResult[];
|
|
20
|
+
}>;
|
|
21
|
+
private askDuckWithMCPFailover;
|
|
22
|
+
compareDucksWithMCP(prompt: string, providerNames?: string[], options?: Partial<ChatOptions>): Promise<Array<DuckResponse & {
|
|
23
|
+
pendingApprovals?: {
|
|
24
|
+
id: string;
|
|
25
|
+
message: string;
|
|
26
|
+
}[];
|
|
27
|
+
mcpResults?: MCPResult[];
|
|
28
|
+
}>>;
|
|
29
|
+
duckCouncilWithMCP(prompt: string, options?: Partial<ChatOptions>): Promise<Array<DuckResponse & {
|
|
30
|
+
pendingApprovals?: {
|
|
31
|
+
id: string;
|
|
32
|
+
message: string;
|
|
33
|
+
}[];
|
|
34
|
+
mcpResults?: MCPResult[];
|
|
35
|
+
}>>;
|
|
36
|
+
retryWithApproval(approvalId: string, providerName: string | undefined, prompt: string, options?: Partial<ChatOptions>): Promise<DuckResponse & {
|
|
37
|
+
pendingApprovals?: {
|
|
38
|
+
id: string;
|
|
39
|
+
message: string;
|
|
40
|
+
}[];
|
|
41
|
+
mcpResults?: MCPResult[];
|
|
42
|
+
}>;
|
|
43
|
+
getAllEnhancedProviders(): Array<{
|
|
44
|
+
name: string;
|
|
45
|
+
info: ReturnType<EnhancedDuckProvider['getInfo']>;
|
|
46
|
+
mcpEnabled: boolean;
|
|
47
|
+
mcpStats?: ReturnType<EnhancedDuckProvider['getMCPStats']>;
|
|
48
|
+
functionCount?: number;
|
|
49
|
+
}>;
|
|
50
|
+
isMCPEnabled(): boolean;
|
|
51
|
+
setMCPEnabled(enabled: boolean): void;
|
|
52
|
+
getMCPFunctionCount(providerName?: string): Promise<number>;
|
|
53
|
+
}
|
|
54
|
+
//# sourceMappingURL=enhanced-manager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"enhanced-manager.d.ts","sourceRoot":"","sources":["../../src/providers/enhanced-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,gCAAgC,CAAC;AAChE,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAGpD,qBAAa,uBAAwB,SAAQ,eAAe;IAC1D,OAAO,CAAC,iBAAiB,CAAgD;IACzE,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,UAAU,CAAkB;gBAExB,aAAa,EAAE,aAAa,EAAE,cAAc,CAAC,EAAE,cAAc;IAWzE,OAAO,CAAC,2BAA2B;IAqCnC,mBAAmB,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,oBAAoB;IAoBlD,cAAc,CAClB,YAAY,EAAE,MAAM,GAAG,SAAS,EAChC,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAC7B,OAAO,CAAC,YAAY,GAAG;QAAE,gBAAgB,CAAC,EAAE;YAAE,EAAE,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAAC,UAAU,CAAC,EAAE,SAAS,EAAE,CAAA;KAAE,CAAC;YA4C/F,sBAAsB;IAsB9B,mBAAmB,CACvB,MAAM,EAAE,MAAM,EACd,aAAa,CAAC,EAAE,MAAM,EAAE,EACxB,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAC7B,OAAO,CAAC,KAAK,CAAC,YAAY,GAAG;QAAE,gBAAgB,CAAC,EAAE;YAAE,EAAE,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAAC,UAAU,CAAC,EAAE,SAAS,EAAE,CAAA;KAAE,CAAC,CAAC;IAmC9G,kBAAkB,CACtB,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAC7B,OAAO,CAAC,KAAK,CAAC,YAAY,GAAG;QAAE,gBAAgB,CAAC,EAAE;YAAE,EAAE,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAAC,UAAU,CAAC,EAAE,SAAS,EAAE,CAAA;KAAE,CAAC,CAAC;IAK9G,iBAAiB,CACrB,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,MAAM,GAAG,SAAS,EAChC,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAC7B,OAAO,CAAC,YAAY,GAAG;QAAE,gBAAgB,CAAC,EAAE;YAAE,EAAE,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,EAAE,CAAC;QAAC,UAAU,CAAC,EAAE,SAAS,EAAE,CAAA;KAAE,CAAC;IA2C7G,uBAAuB,IAAI,KAAK,CAAC;QAC/B,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,UAAU,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC,CAAC;QAClD,UAAU,EAAE,OAAO,CAAC;QACpB,QAAQ,CAAC,EAAE,UAAU,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC,CAAC;QAC3D,aAAa,CAAC,EAAE,MAAM,CAAC;KACxB,CAAC;IAWF,YAAY,IAAI,OAAO;IAKvB,aAAa,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAS/B,mBAAmB,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAYlE"}
|
|
@@ -0,0 +1,217 @@
|
|
|
1
|
+
import { EnhancedDuckProvider } from './duck-provider-enhanced.js';
|
|
2
|
+
import { ProviderManager } from './manager.js';
|
|
3
|
+
import { logger } from '../utils/logger.js';
|
|
4
|
+
export class EnhancedProviderManager extends ProviderManager {
|
|
5
|
+
enhancedProviders = new Map();
|
|
6
|
+
functionBridge;
|
|
7
|
+
mcpEnabled = false;
|
|
8
|
+
constructor(configManager, functionBridge) {
|
|
9
|
+
super(configManager);
|
|
10
|
+
this.functionBridge = functionBridge;
|
|
11
|
+
this.mcpEnabled = !!functionBridge &&
|
|
12
|
+
(configManager.getConfig().mcp_bridge?.enabled || false);
|
|
13
|
+
if (this.mcpEnabled) {
|
|
14
|
+
this.initializeEnhancedProviders();
|
|
15
|
+
}
|
|
16
|
+
}
|
|
17
|
+
initializeEnhancedProviders() {
|
|
18
|
+
if (!this.functionBridge) {
|
|
19
|
+
logger.warn('Function bridge not available, skipping enhanced providers');
|
|
20
|
+
return;
|
|
21
|
+
}
|
|
22
|
+
const config = this.configManager.getConfig();
|
|
23
|
+
const allProviders = config.providers;
|
|
24
|
+
for (const [name, providerConfig] of Object.entries(allProviders)) {
|
|
25
|
+
try {
|
|
26
|
+
// Create enhanced provider if MCP is enabled
|
|
27
|
+
const enhancedProvider = new EnhancedDuckProvider(name, providerConfig.nickname, {
|
|
28
|
+
apiKey: providerConfig.api_key,
|
|
29
|
+
baseURL: providerConfig.base_url,
|
|
30
|
+
model: providerConfig.default_model,
|
|
31
|
+
availableModels: providerConfig.models,
|
|
32
|
+
temperature: providerConfig.temperature,
|
|
33
|
+
timeout: providerConfig.timeout,
|
|
34
|
+
maxRetries: providerConfig.max_retries,
|
|
35
|
+
systemPrompt: providerConfig.system_prompt,
|
|
36
|
+
}, this.functionBridge, this.mcpEnabled);
|
|
37
|
+
this.enhancedProviders.set(name, enhancedProvider);
|
|
38
|
+
logger.info(`Initialized enhanced provider: ${name} (${providerConfig.nickname}) with MCP support`);
|
|
39
|
+
}
|
|
40
|
+
catch (error) {
|
|
41
|
+
logger.error(`Failed to initialize enhanced provider ${name}:`, error);
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
getEnhancedProvider(name) {
|
|
46
|
+
if (!this.mcpEnabled) {
|
|
47
|
+
throw new Error('MCP bridge is not enabled');
|
|
48
|
+
}
|
|
49
|
+
const providerName = name || this.configManager.getConfig().default_provider;
|
|
50
|
+
if (!providerName) {
|
|
51
|
+
throw new Error('No provider specified and no default provider configured');
|
|
52
|
+
}
|
|
53
|
+
const provider = this.enhancedProviders.get(providerName);
|
|
54
|
+
if (!provider) {
|
|
55
|
+
throw new Error(`Enhanced duck "${providerName}" not found in the pond`);
|
|
56
|
+
}
|
|
57
|
+
return provider;
|
|
58
|
+
}
|
|
59
|
+
async askDuckWithMCP(providerName, prompt, options) {
|
|
60
|
+
if (!this.mcpEnabled) {
|
|
61
|
+
// Fall back to regular provider
|
|
62
|
+
return this.askDuck(providerName, prompt, options);
|
|
63
|
+
}
|
|
64
|
+
const provider = this.getEnhancedProvider(providerName);
|
|
65
|
+
const startTime = Date.now();
|
|
66
|
+
try {
|
|
67
|
+
const response = await provider.chat({
|
|
68
|
+
messages: [{ role: 'user', content: prompt, timestamp: new Date() }],
|
|
69
|
+
...options,
|
|
70
|
+
});
|
|
71
|
+
return {
|
|
72
|
+
provider: provider.name,
|
|
73
|
+
nickname: provider.nickname,
|
|
74
|
+
model: response.model,
|
|
75
|
+
content: response.content,
|
|
76
|
+
usage: response.usage ? {
|
|
77
|
+
prompt_tokens: response.usage.promptTokens,
|
|
78
|
+
completion_tokens: response.usage.completionTokens,
|
|
79
|
+
total_tokens: response.usage.totalTokens,
|
|
80
|
+
promptTokens: response.usage.promptTokens,
|
|
81
|
+
completionTokens: response.usage.completionTokens,
|
|
82
|
+
totalTokens: response.usage.totalTokens,
|
|
83
|
+
} : undefined,
|
|
84
|
+
latency: Date.now() - startTime,
|
|
85
|
+
cached: false,
|
|
86
|
+
pendingApprovals: response.pendingApprovals,
|
|
87
|
+
mcpResults: response.mcpResults,
|
|
88
|
+
};
|
|
89
|
+
}
|
|
90
|
+
catch (error) {
|
|
91
|
+
// Try failover if enabled
|
|
92
|
+
if (this.configManager.getConfig().enable_failover && providerName === undefined) {
|
|
93
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
94
|
+
logger.warn(`Primary enhanced provider failed, attempting failover: ${errorMessage}`);
|
|
95
|
+
return this.askDuckWithMCPFailover(prompt, options, provider.name);
|
|
96
|
+
}
|
|
97
|
+
throw error;
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
async askDuckWithMCPFailover(prompt, options, failedProvider) {
|
|
101
|
+
const availableProviders = Array.from(this.enhancedProviders.keys()).filter(name => name !== failedProvider);
|
|
102
|
+
for (const providerName of availableProviders) {
|
|
103
|
+
try {
|
|
104
|
+
logger.info(`Trying enhanced failover to ${providerName}...`);
|
|
105
|
+
return await this.askDuckWithMCP(providerName, prompt, options);
|
|
106
|
+
}
|
|
107
|
+
catch (error) {
|
|
108
|
+
logger.warn(`Enhanced failover to ${providerName} failed:`, error);
|
|
109
|
+
continue;
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
throw new Error('All enhanced ducks have flown away! No providers available.');
|
|
113
|
+
}
|
|
114
|
+
async compareDucksWithMCP(prompt, providerNames, options) {
|
|
115
|
+
if (!this.mcpEnabled) {
|
|
116
|
+
// Fall back to regular comparison
|
|
117
|
+
return this.compareDucks(prompt, providerNames, options);
|
|
118
|
+
}
|
|
119
|
+
const providersToUse = providerNames
|
|
120
|
+
? providerNames.map(name => this.enhancedProviders.get(name)).filter(Boolean)
|
|
121
|
+
: Array.from(this.enhancedProviders.values());
|
|
122
|
+
if (providersToUse.length === 0) {
|
|
123
|
+
throw new Error('No valid enhanced providers specified');
|
|
124
|
+
}
|
|
125
|
+
const promises = providersToUse.map(provider => provider ? this.askDuckWithMCP(provider.name, prompt, options).catch(error => ({
|
|
126
|
+
provider: provider.name,
|
|
127
|
+
nickname: provider.nickname,
|
|
128
|
+
model: '',
|
|
129
|
+
content: `Error: ${error instanceof Error ? error.message : String(error)}`,
|
|
130
|
+
latency: 0,
|
|
131
|
+
cached: false,
|
|
132
|
+
})) : Promise.resolve({
|
|
133
|
+
provider: 'unknown',
|
|
134
|
+
nickname: 'Unknown',
|
|
135
|
+
model: '',
|
|
136
|
+
content: 'Error: Invalid provider',
|
|
137
|
+
latency: 0,
|
|
138
|
+
cached: false,
|
|
139
|
+
}));
|
|
140
|
+
return Promise.all(promises);
|
|
141
|
+
}
|
|
142
|
+
async duckCouncilWithMCP(prompt, options) {
|
|
143
|
+
return this.compareDucksWithMCP(prompt, undefined, options);
|
|
144
|
+
}
|
|
145
|
+
// Method to retry with approval
|
|
146
|
+
async retryWithApproval(approvalId, providerName, prompt, options) {
|
|
147
|
+
if (!this.mcpEnabled) {
|
|
148
|
+
throw new Error('MCP bridge is not enabled');
|
|
149
|
+
}
|
|
150
|
+
const provider = this.getEnhancedProvider(providerName);
|
|
151
|
+
const startTime = Date.now();
|
|
152
|
+
try {
|
|
153
|
+
const response = await provider.retryWithApproval(approvalId, [{ role: 'user', content: prompt, timestamp: new Date() }], {
|
|
154
|
+
messages: [{ role: 'user', content: prompt, timestamp: new Date() }],
|
|
155
|
+
...options,
|
|
156
|
+
});
|
|
157
|
+
return {
|
|
158
|
+
provider: provider.name,
|
|
159
|
+
nickname: provider.nickname,
|
|
160
|
+
model: response.model,
|
|
161
|
+
content: response.content,
|
|
162
|
+
usage: response.usage ? {
|
|
163
|
+
prompt_tokens: response.usage.promptTokens,
|
|
164
|
+
completion_tokens: response.usage.completionTokens,
|
|
165
|
+
total_tokens: response.usage.totalTokens,
|
|
166
|
+
promptTokens: response.usage.promptTokens,
|
|
167
|
+
completionTokens: response.usage.completionTokens,
|
|
168
|
+
totalTokens: response.usage.totalTokens,
|
|
169
|
+
} : undefined,
|
|
170
|
+
latency: Date.now() - startTime,
|
|
171
|
+
cached: false,
|
|
172
|
+
pendingApprovals: response.pendingApprovals,
|
|
173
|
+
mcpResults: response.mcpResults,
|
|
174
|
+
};
|
|
175
|
+
}
|
|
176
|
+
catch (error) {
|
|
177
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
178
|
+
throw new Error(`Failed to retry with approval: ${errorMessage}`);
|
|
179
|
+
}
|
|
180
|
+
}
|
|
181
|
+
// Get enhanced provider statistics
|
|
182
|
+
getAllEnhancedProviders() {
|
|
183
|
+
return Array.from(this.enhancedProviders.entries()).map(([name, provider]) => ({
|
|
184
|
+
name,
|
|
185
|
+
info: provider.getInfo(),
|
|
186
|
+
mcpEnabled: provider.isMCPEnabled(),
|
|
187
|
+
mcpStats: provider.getMCPStats(),
|
|
188
|
+
functionCount: 0, // Will be populated when functions are loaded
|
|
189
|
+
}));
|
|
190
|
+
}
|
|
191
|
+
// Check if MCP is enabled
|
|
192
|
+
isMCPEnabled() {
|
|
193
|
+
return this.mcpEnabled;
|
|
194
|
+
}
|
|
195
|
+
// Enable/disable MCP for all providers
|
|
196
|
+
setMCPEnabled(enabled) {
|
|
197
|
+
this.mcpEnabled = enabled;
|
|
198
|
+
for (const provider of this.enhancedProviders.values()) {
|
|
199
|
+
provider.setMCPEnabled(enabled);
|
|
200
|
+
}
|
|
201
|
+
logger.info(`MCP ${enabled ? 'enabled' : 'disabled'} for all providers`);
|
|
202
|
+
}
|
|
203
|
+
// Get MCP function count for a provider
|
|
204
|
+
async getMCPFunctionCount(providerName) {
|
|
205
|
+
if (!this.mcpEnabled) {
|
|
206
|
+
return 0;
|
|
207
|
+
}
|
|
208
|
+
try {
|
|
209
|
+
const provider = this.getEnhancedProvider(providerName);
|
|
210
|
+
return await provider.getMCPFunctionCount();
|
|
211
|
+
}
|
|
212
|
+
catch (error) {
|
|
213
|
+
return 0;
|
|
214
|
+
}
|
|
215
|
+
}
|
|
216
|
+
}
|
|
217
|
+
//# sourceMappingURL=enhanced-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"enhanced-manager.js","sourceRoot":"","sources":["../../src/providers/enhanced-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAK/C,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,MAAM,OAAO,uBAAwB,SAAQ,eAAe;IAClD,iBAAiB,GAAsC,IAAI,GAAG,EAAE,CAAC;IACjE,cAAc,CAAkB;IAChC,UAAU,GAAY,KAAK,CAAC;IAEpC,YAAY,aAA4B,EAAE,cAA+B;QACvE,KAAK,CAAC,aAAa,CAAC,CAAC;QACrB,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC,cAAc;YAChC,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC,UAAU,EAAE,OAAO,IAAI,KAAK,CAAC,CAAC;QAE3D,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,2BAA2B;QACjC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,4DAA4D,CAAC,CAAC;YAC1E,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC;QAC9C,MAAM,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC;QAEtC,KAAK,MAAM,CAAC,IAAI,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;YAClE,IAAI,CAAC;gBACH,6CAA6C;gBAC7C,MAAM,gBAAgB,GAAG,IAAI,oBAAoB,CAC/C,IAAI,EACJ,cAAc,CAAC,QAAQ,EACvB;oBACE,MAAM,EAAE,cAAc,CAAC,OAAO;oBAC9B,OAAO,EAAE,cAAc,CAAC,QAAQ;oBAChC,KAAK,EAAE,cAAc,CAAC,aAAa;oBACnC,eAAe,EAAE,cAAc,CAAC,MAAM;oBACtC,WAAW,EAAE,cAAc,CAAC,WAAW;oBACvC,OAAO,EAAE,cAAc,CAAC,OAAO;oBAC/B,UAAU,EAAE,cAAc,CAAC,WAAW;oBACtC,YAAY,EAAE,cAAc,CAAC,aAAa;iBAC3C,EACD,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,UAAU,CAChB,CAAC;gBAEF,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC;gBACnD,MAAM,CAAC,IAAI,CAAC,kCAAkC,IAAI,KAAK,cAAc,CAAC,QAAQ,oBAAoB,CAAC,CAAC;YACtG,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,KAAK,CAAC,0CAA0C,IAAI,GAAG,EAAE,KAAK,CAAC,CAAC;YACzE,CAAC;QACH,CAAC;IACH,CAAC;IAED,mBAAmB,CAAC,IAAa;QAC/B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,IAAI,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC,gBAAgB,CAAC;QAE7E,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAC9E,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAE1D,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,kBAAkB,YAAY,yBAAyB,CAAC,CAAC;QAC3E,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,YAAgC,EAChC,MAAc,EACd,OAA8B;QAE9B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,gCAAgC;YAChC,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;gBACnC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;gBACpE,GAAG,OAAO;aACX,CAAC,CAAC;YAEH,OAAO;gBACL,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;oBACtB,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;oBAC1C,iBAAiB,EAAE,QAAQ,CAAC,KAAK,CAAC,gBAAgB;oBAClD,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW;oBACxC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;oBACzC,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,gBAAgB;oBACjD,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW;iBACxC,CAAC,CAAC,CAAC,SAAS;gBACb,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAC/B,MAAM,EAAE,KAAK;gBACb,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;gBAC3C,UAAU,EAAE,QAAQ,CAAC,UAAU;aAChC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,0BAA0B;YAC1B,IAAI,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC,eAAe,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBACjF,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC5E,MAAM,CAAC,IAAI,CAAC,0DAA0D,YAAY,EAAE,CAAC,CAAC;gBACtF,OAAO,IAAI,CAAC,sBAAsB,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC;YACrE,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAClC,MAAc,EACd,OAAyC,EACzC,cAAsB;QAEtB,MAAM,kBAAkB,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CACzE,IAAI,CAAC,EAAE,CAAC,IAAI,KAAK,cAAc,CAChC,CAAC;QAEF,KAAK,MAAM,YAAY,IAAI,kBAAkB,EAAE,CAAC;YAC9C,IAAI,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,+BAA+B,YAAY,KAAK,CAAC,CAAC;gBAC9D,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAClE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,wBAAwB,YAAY,UAAU,EAAE,KAAK,CAAC,CAAC;gBACnE,SAAS;YACX,CAAC;QACH,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,6DAA6D,CAAC,CAAC;IACjF,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,MAAc,EACd,aAAwB,EACxB,OAA8B;QAE9B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,kCAAkC;YAClC,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,cAAc,GAAG,aAAa;YAClC,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC;YAC7E,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC;QAEhD,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,QAAQ,GAAG,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAC7C,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC7E,QAAQ,EAAE,QAAQ,CAAC,IAAI;YACvB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,KAAK,EAAE,EAAE;YACT,OAAO,EAAE,UAAU,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;YAC3E,OAAO,EAAE,CAAC;YACV,MAAM,EAAE,KAAK;SACd,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;YACpB,QAAQ,EAAE,SAAS;YACnB,QAAQ,EAAE,SAAS;YACnB,KAAK,EAAE,EAAE;YACT,OAAO,EAAE,yBAAyB;YAClC,OAAO,EAAE,CAAC;YACV,MAAM,EAAE,KAAK;SACd,CAAC,CACH,CAAC;QAEF,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,MAAc,EACd,OAA8B;QAE9B,OAAO,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAED,gCAAgC;IAChC,KAAK,CAAC,iBAAiB,CACrB,UAAkB,EAClB,YAAgC,EAChC,MAAc,EACd,OAA8B;QAE9B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,iBAAiB,CAC/C,UAAU,EACV,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,EAC1D;gBACE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;gBACpE,GAAG,OAAO;aACX,CACF,CAAC;YAEF,OAAO;gBACL,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,KAAK,EAAE,QAAQ,CAAC,KAAK;gBACrB,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;oBACtB,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;oBAC1C,iBAAiB,EAAE,QAAQ,CAAC,KAAK,CAAC,gBAAgB;oBAClD,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW;oBACxC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;oBACzC,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,gBAAgB;oBACjD,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW;iBACxC,CAAC,CAAC,CAAC,SAAS;gBACb,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAC/B,MAAM,EAAE,KAAK;gBACb,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;gBAC3C,UAAU,EAAE,QAAQ,CAAC,UAAU;aAChC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,MAAM,IAAI,KAAK,CAAC,kCAAkC,YAAY,EAAE,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAED,mCAAmC;IACnC,uBAAuB;QAOrB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7E,IAAI;YACJ,IAAI,EAAE,QAAQ,CAAC,OAAO,EAAE;YACxB,UAAU,EAAE,QAAQ,CAAC,YAAY,EAAE;YACnC,QAAQ,EAAE,QAAQ,CAAC,WAAW,EAAE;YAChC,aAAa,EAAE,CAAC,EAAE,8CAA8C;SACjE,CAAC,CAAC,CAAC;IACN,CAAC;IAED,0BAA0B;IAC1B,YAAY;QACV,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,uCAAuC;IACvC,aAAa,CAAC,OAAgB;QAC5B,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC;QAC1B,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,EAAE,CAAC;YACvD,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,OAAO,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,oBAAoB,CAAC,CAAC;IAC3E,CAAC;IAED,wCAAwC;IACxC,KAAK,CAAC,mBAAmB,CAAC,YAAqB;QAC7C,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,OAAO,CAAC,CAAC;QACX,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;YACxD,OAAO,MAAM,QAAQ,CAAC,mBAAmB,EAAE,CAAC;QAC9C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { DuckProvider } from './provider.js';
|
|
2
|
+
import { ConfigManager } from '../config/config.js';
|
|
3
|
+
import { ProviderHealth, DuckResponse } from '../config/types.js';
|
|
4
|
+
import { ChatOptions, ModelInfo } from './types.js';
|
|
5
|
+
export declare class ProviderManager {
|
|
6
|
+
private providers;
|
|
7
|
+
private healthStatus;
|
|
8
|
+
protected configManager: ConfigManager;
|
|
9
|
+
private defaultProvider?;
|
|
10
|
+
constructor(configManager: ConfigManager);
|
|
11
|
+
private initializeProviders;
|
|
12
|
+
checkHealth(providerName?: string): Promise<ProviderHealth[]>;
|
|
13
|
+
askDuck(providerName: string | undefined, prompt: string, options?: Partial<ChatOptions>): Promise<DuckResponse>;
|
|
14
|
+
compareDucks(prompt: string, providerNames?: string[], options?: Partial<ChatOptions>): Promise<DuckResponse[]>;
|
|
15
|
+
duckCouncil(prompt: string, options?: Partial<ChatOptions>): Promise<DuckResponse[]>;
|
|
16
|
+
private askWithFailover;
|
|
17
|
+
getProvider(name?: string): DuckProvider;
|
|
18
|
+
getAllProviders(): Array<{
|
|
19
|
+
name: string;
|
|
20
|
+
info: ReturnType<DuckProvider['getInfo']>;
|
|
21
|
+
health?: ProviderHealth;
|
|
22
|
+
}>;
|
|
23
|
+
getProviderNames(): string[];
|
|
24
|
+
getAvailableModels(providerName: string): Promise<ModelInfo[]>;
|
|
25
|
+
getAllModels(): Promise<Map<string, ModelInfo[]>>;
|
|
26
|
+
validateModel(providerName: string, modelId: string): boolean;
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=manager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/providers/manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClE,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAIpD,qBAAa,eAAe;IAC1B,OAAO,CAAC,SAAS,CAAwC;IACzD,OAAO,CAAC,YAAY,CAA0C;IAC9D,SAAS,CAAC,aAAa,EAAE,aAAa,CAAC;IACvC,OAAO,CAAC,eAAe,CAAC,CAAS;gBAErB,aAAa,EAAE,aAAa;IAKxC,OAAO,CAAC,mBAAmB;IA+BrB,WAAW,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAqC7D,OAAO,CACX,YAAY,EAAE,MAAM,GAAG,SAAS,EAChC,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAC7B,OAAO,CAAC,YAAY,CAAC;IAqClB,YAAY,CAChB,MAAM,EAAE,MAAM,EACd,aAAa,CAAC,EAAE,MAAM,EAAE,EACxB,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAC7B,OAAO,CAAC,YAAY,EAAE,CAAC;IA8BpB,WAAW,CACf,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAC7B,OAAO,CAAC,YAAY,EAAE,CAAC;YAIZ,eAAe;IAsB7B,WAAW,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,YAAY;IAgBxC,eAAe,IAAI,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC;QAAC,MAAM,CAAC,EAAE,cAAc,CAAA;KAAE,CAAC;IAQ9G,gBAAgB,IAAI,MAAM,EAAE;IAItB,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAQ9D,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;IAgBvD,aAAa,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO;CAc9D"}
|