funolio-agent 0.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/README.md +55 -0
- package/dist/clerk/index.d.ts +62 -0
- package/dist/clerk/index.d.ts.map +1 -0
- package/dist/clerk/index.js +186 -0
- package/dist/clerk/index.js.map +1 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +42 -0
- package/dist/cli.js.map +1 -0
- package/dist/commands/configure.d.ts +2 -0
- package/dist/commands/configure.d.ts.map +1 -0
- package/dist/commands/configure.js +252 -0
- package/dist/commands/configure.js.map +1 -0
- package/dist/commands/init.d.ts +6 -0
- package/dist/commands/init.d.ts.map +1 -0
- package/dist/commands/init.js +151 -0
- package/dist/commands/init.js.map +1 -0
- package/dist/commands/login.d.ts +6 -0
- package/dist/commands/login.d.ts.map +1 -0
- package/dist/commands/login.js +170 -0
- package/dist/commands/login.js.map +1 -0
- package/dist/commands/start.d.ts +8 -0
- package/dist/commands/start.d.ts.map +1 -0
- package/dist/commands/start.js +179 -0
- package/dist/commands/start.js.map +1 -0
- package/dist/commands/status.d.ts +2 -0
- package/dist/commands/status.d.ts.map +1 -0
- package/dist/commands/status.js +55 -0
- package/dist/commands/status.js.map +1 -0
- package/dist/config.d.ts +46 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +113 -0
- package/dist/config.js.map +1 -0
- package/dist/index.d.ts +24 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +48 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp/index.d.ts +9 -0
- package/dist/mcp/index.d.ts.map +1 -0
- package/dist/mcp/index.js +15 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/manager.d.ts +83 -0
- package/dist/mcp/manager.d.ts.map +1 -0
- package/dist/mcp/manager.js +338 -0
- package/dist/mcp/manager.js.map +1 -0
- package/dist/mcp/registry.d.ts +32 -0
- package/dist/mcp/registry.d.ts.map +1 -0
- package/dist/mcp/registry.js +139 -0
- package/dist/mcp/registry.js.map +1 -0
- package/dist/message-loop.d.ts +18 -0
- package/dist/message-loop.d.ts.map +1 -0
- package/dist/message-loop.js +165 -0
- package/dist/message-loop.js.map +1 -0
- package/dist/mqtt-client.d.ts +67 -0
- package/dist/mqtt-client.d.ts.map +1 -0
- package/dist/mqtt-client.js +148 -0
- package/dist/mqtt-client.js.map +1 -0
- package/dist/providers/anthropic.d.ts +10 -0
- package/dist/providers/anthropic.d.ts.map +1 -0
- package/dist/providers/anthropic.js +183 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/google.d.ts +10 -0
- package/dist/providers/google.d.ts.map +1 -0
- package/dist/providers/google.js +161 -0
- package/dist/providers/google.js.map +1 -0
- package/dist/providers/index.d.ts +42 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +19 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/openai.d.ts +10 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +173 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/retry.d.ts +6 -0
- package/dist/providers/retry.d.ts.map +1 -0
- package/dist/providers/retry.js +38 -0
- package/dist/providers/retry.js.map +1 -0
- package/dist/subagent/index.d.ts +8 -0
- package/dist/subagent/index.d.ts.map +1 -0
- package/dist/subagent/index.js +14 -0
- package/dist/subagent/index.js.map +1 -0
- package/dist/subagent/orchestrator.d.ts +67 -0
- package/dist/subagent/orchestrator.d.ts.map +1 -0
- package/dist/subagent/orchestrator.js +152 -0
- package/dist/subagent/orchestrator.js.map +1 -0
- package/dist/subagent/queue.d.ts +66 -0
- package/dist/subagent/queue.d.ts.map +1 -0
- package/dist/subagent/queue.js +298 -0
- package/dist/subagent/queue.js.map +1 -0
- package/dist/subagent/types.d.ts +76 -0
- package/dist/subagent/types.d.ts.map +1 -0
- package/dist/subagent/types.js +14 -0
- package/dist/subagent/types.js.map +1 -0
- package/dist/tools/edit-file.d.ts +3 -0
- package/dist/tools/edit-file.d.ts.map +1 -0
- package/dist/tools/edit-file.js +112 -0
- package/dist/tools/edit-file.js.map +1 -0
- package/dist/tools/git-tools.d.ts +5 -0
- package/dist/tools/git-tools.d.ts.map +1 -0
- package/dist/tools/git-tools.js +144 -0
- package/dist/tools/git-tools.js.map +1 -0
- package/dist/tools/index.d.ts +40 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +126 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/installer.d.ts +41 -0
- package/dist/tools/installer.d.ts.map +1 -0
- package/dist/tools/installer.js +227 -0
- package/dist/tools/installer.js.map +1 -0
- package/dist/tools/list-directory.d.ts +3 -0
- package/dist/tools/list-directory.d.ts.map +1 -0
- package/dist/tools/list-directory.js +107 -0
- package/dist/tools/list-directory.js.map +1 -0
- package/dist/tools/read-file.d.ts +3 -0
- package/dist/tools/read-file.d.ts.map +1 -0
- package/dist/tools/read-file.js +89 -0
- package/dist/tools/read-file.js.map +1 -0
- package/dist/tools/run-command.d.ts +3 -0
- package/dist/tools/run-command.d.ts.map +1 -0
- package/dist/tools/run-command.js +86 -0
- package/dist/tools/run-command.js.map +1 -0
- package/dist/tools/sandbox.d.ts +17 -0
- package/dist/tools/sandbox.d.ts.map +1 -0
- package/dist/tools/sandbox.js +78 -0
- package/dist/tools/sandbox.js.map +1 -0
- package/dist/tools/write-file.d.ts +3 -0
- package/dist/tools/write-file.d.ts.map +1 -0
- package/dist/tools/write-file.js +88 -0
- package/dist/tools/write-file.js.map +1 -0
- package/dist/types.d.ts +67 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +6 -0
- package/dist/types.js.map +1 -0
- package/dist/verification/index.d.ts +17 -0
- package/dist/verification/index.d.ts.map +1 -0
- package/dist/verification/index.js +224 -0
- package/dist/verification/index.js.map +1 -0
- package/package.json +41 -0
|
@@ -0,0 +1,165 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.MessageLoop = void 0;
|
|
7
|
+
const chalk_1 = __importDefault(require("chalk"));
|
|
8
|
+
const index_1 = require("./providers/index");
|
|
9
|
+
class MessageLoop {
|
|
10
|
+
options;
|
|
11
|
+
llmProvider;
|
|
12
|
+
activeCommandId = null;
|
|
13
|
+
constructor(options) {
|
|
14
|
+
this.options = options;
|
|
15
|
+
this.llmProvider = (0, index_1.createProvider)(options.provider, {
|
|
16
|
+
apiKey: options.apiKey,
|
|
17
|
+
model: options.model,
|
|
18
|
+
});
|
|
19
|
+
}
|
|
20
|
+
async handleCommand(command) {
|
|
21
|
+
if (command.type === 'ping') {
|
|
22
|
+
await this.options.mqttClient.publishResult({
|
|
23
|
+
commandId: command.id,
|
|
24
|
+
type: 'complete',
|
|
25
|
+
content: 'pong',
|
|
26
|
+
timestamp: Date.now(),
|
|
27
|
+
});
|
|
28
|
+
return;
|
|
29
|
+
}
|
|
30
|
+
if (command.type === 'cancel') {
|
|
31
|
+
if (this.activeCommandId === command.id) {
|
|
32
|
+
console.log(chalk_1.default.yellow(`Cancelling command ${command.id}`));
|
|
33
|
+
this.activeCommandId = null;
|
|
34
|
+
}
|
|
35
|
+
return;
|
|
36
|
+
}
|
|
37
|
+
if (!command.prompt) {
|
|
38
|
+
await this.publishError(command.id, 'No prompt provided');
|
|
39
|
+
return;
|
|
40
|
+
}
|
|
41
|
+
// Use model override from command if provided
|
|
42
|
+
const provider = command.model?.provider || this.options.provider;
|
|
43
|
+
const model = command.model?.model || this.options.model;
|
|
44
|
+
const apiKey = command.model?.apiKey || this.options.apiKey;
|
|
45
|
+
// Create provider if overridden
|
|
46
|
+
const llm = (provider !== this.options.provider || model !== this.options.model)
|
|
47
|
+
? (0, index_1.createProvider)(provider, { apiKey, model })
|
|
48
|
+
: this.llmProvider;
|
|
49
|
+
this.activeCommandId = command.id;
|
|
50
|
+
console.log(chalk_1.default.blue(`← Command ${command.id.slice(0, 8)}...`));
|
|
51
|
+
console.log(chalk_1.default.gray(` Prompt: ${command.prompt.slice(0, 100)}...`));
|
|
52
|
+
try {
|
|
53
|
+
// Build messages from context
|
|
54
|
+
const messages = [];
|
|
55
|
+
// Add previous conversation context
|
|
56
|
+
if (command.context?.previousMessages) {
|
|
57
|
+
for (const msg of command.context.previousMessages) {
|
|
58
|
+
messages.push({ role: msg.role, content: msg.content });
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
// Build system prompt with context
|
|
62
|
+
let systemPrompt = 'You are a Funolio AI agent running locally on the user\'s machine. You have access to their project files and can execute code.';
|
|
63
|
+
if (command.context?.facts?.length) {
|
|
64
|
+
systemPrompt += '\n\nRelevant memory/facts:\n' + command.context.facts.map(f => `- ${f}`).join('\n');
|
|
65
|
+
}
|
|
66
|
+
if (command.context?.files?.length) {
|
|
67
|
+
systemPrompt += '\n\nRelevant files: ' + command.context.files.join(', ');
|
|
68
|
+
}
|
|
69
|
+
// Add the current prompt
|
|
70
|
+
messages.push({ role: 'user', content: command.prompt });
|
|
71
|
+
// Agentic loop - keep calling LLM until no more tool calls
|
|
72
|
+
let iteration = 0;
|
|
73
|
+
const MAX_ITERATIONS = 20;
|
|
74
|
+
while (iteration < MAX_ITERATIONS && this.activeCommandId === command.id) {
|
|
75
|
+
iteration++;
|
|
76
|
+
const response = await llm.chat({
|
|
77
|
+
messages,
|
|
78
|
+
system: systemPrompt,
|
|
79
|
+
stream: true,
|
|
80
|
+
onChunk: async (chunk) => {
|
|
81
|
+
if (this.activeCommandId !== command.id)
|
|
82
|
+
return;
|
|
83
|
+
await this.options.mqttClient.publishResult({
|
|
84
|
+
commandId: command.id,
|
|
85
|
+
type: 'chunk',
|
|
86
|
+
content: chunk,
|
|
87
|
+
timestamp: Date.now(),
|
|
88
|
+
});
|
|
89
|
+
},
|
|
90
|
+
});
|
|
91
|
+
// If there are tool calls, publish them and continue the loop
|
|
92
|
+
if (response.toolCalls && response.toolCalls.length > 0) {
|
|
93
|
+
// Add assistant message with tool calls to history
|
|
94
|
+
messages.push({
|
|
95
|
+
role: 'assistant',
|
|
96
|
+
content: response.content || '',
|
|
97
|
+
toolCalls: response.toolCalls,
|
|
98
|
+
});
|
|
99
|
+
for (const toolCall of response.toolCalls) {
|
|
100
|
+
console.log(chalk_1.default.cyan(` 🔧 Tool: ${toolCall.name}(${JSON.stringify(toolCall.arguments).slice(0, 60)}...)`));
|
|
101
|
+
await this.options.mqttClient.publishResult({
|
|
102
|
+
commandId: command.id,
|
|
103
|
+
type: 'tool_call',
|
|
104
|
+
toolCall: {
|
|
105
|
+
id: toolCall.id,
|
|
106
|
+
name: toolCall.name,
|
|
107
|
+
arguments: toolCall.arguments,
|
|
108
|
+
},
|
|
109
|
+
timestamp: Date.now(),
|
|
110
|
+
});
|
|
111
|
+
// TODO: Phase 3D will implement actual tool execution here
|
|
112
|
+
// For now, return a placeholder
|
|
113
|
+
const toolResult = {
|
|
114
|
+
callId: toolCall.id,
|
|
115
|
+
output: `[Tool execution not yet implemented - Phase 3D]`,
|
|
116
|
+
isError: false,
|
|
117
|
+
};
|
|
118
|
+
await this.options.mqttClient.publishResult({
|
|
119
|
+
commandId: command.id,
|
|
120
|
+
type: 'tool_result',
|
|
121
|
+
toolResult,
|
|
122
|
+
timestamp: Date.now(),
|
|
123
|
+
});
|
|
124
|
+
// Add tool result to messages for next iteration
|
|
125
|
+
messages.push({
|
|
126
|
+
role: 'user',
|
|
127
|
+
content: `Tool result for ${toolCall.name}: ${toolResult.output}`,
|
|
128
|
+
});
|
|
129
|
+
}
|
|
130
|
+
// Continue the loop for the next LLM call
|
|
131
|
+
continue;
|
|
132
|
+
}
|
|
133
|
+
// No tool calls - final response
|
|
134
|
+
await this.options.mqttClient.publishResult({
|
|
135
|
+
commandId: command.id,
|
|
136
|
+
type: 'complete',
|
|
137
|
+
content: response.content,
|
|
138
|
+
timestamp: Date.now(),
|
|
139
|
+
});
|
|
140
|
+
console.log(chalk_1.default.green(` ✓ Command complete (${iteration} iteration${iteration > 1 ? 's' : ''})`));
|
|
141
|
+
break;
|
|
142
|
+
}
|
|
143
|
+
if (iteration >= MAX_ITERATIONS) {
|
|
144
|
+
await this.publishError(command.id, `Max iterations (${MAX_ITERATIONS}) reached`);
|
|
145
|
+
}
|
|
146
|
+
}
|
|
147
|
+
catch (error) {
|
|
148
|
+
console.error(chalk_1.default.red(` ✗ Error: ${error.message}`));
|
|
149
|
+
await this.publishError(command.id, error.message);
|
|
150
|
+
}
|
|
151
|
+
finally {
|
|
152
|
+
this.activeCommandId = null;
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
async publishError(commandId, error) {
|
|
156
|
+
await this.options.mqttClient.publishResult({
|
|
157
|
+
commandId,
|
|
158
|
+
type: 'error',
|
|
159
|
+
error,
|
|
160
|
+
timestamp: Date.now(),
|
|
161
|
+
});
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
exports.MessageLoop = MessageLoop;
|
|
165
|
+
//# sourceMappingURL=message-loop.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"message-loop.js","sourceRoot":"","sources":["../src/message-loop.ts"],"names":[],"mappings":";;;;;;AAAA,kDAA0B;AAE1B,6CAAyF;AAWzF,MAAa,WAAW;IACd,OAAO,CAAqB;IAC5B,WAAW,CAAc;IACzB,eAAe,GAAkB,IAAI,CAAC;IAE9C,YAAY,OAA2B;QACrC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,WAAW,GAAG,IAAA,sBAAc,EAAC,OAAO,CAAC,QAAQ,EAAE;YAClD,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAAqB;QACvC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC;gBAC1C,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,MAAM;gBACf,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC9B,IAAI,IAAI,CAAC,eAAe,KAAK,OAAO,CAAC,EAAE,EAAE,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,MAAM,CAAC,sBAAsB,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;gBAC9D,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;YAC9B,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,EAAE,oBAAoB,CAAC,CAAC;YAC1D,OAAO;QACT,CAAC;QAED,8CAA8C;QAC9C,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,EAAE,QAAQ,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;QAClE,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,EAAE,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;QACzD,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;QAE5D,gCAAgC;QAChC,MAAM,GAAG,GAAG,CAAC,QAAQ,KAAK,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;YAC9E,CAAC,CAAC,IAAA,sBAAc,EAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;YAC7C,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;QAErB,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,EAAE,CAAC;QAClC,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,aAAa,OAAO,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QAClE,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,aAAa,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;QAExE,IAAI,CAAC;YACH,8BAA8B;YAC9B,MAAM,QAAQ,GAAiB,EAAE,CAAC;YAElC,oCAAoC;YACpC,IAAI,OAAO,CAAC,OAAO,EAAE,gBAAgB,EAAE,CAAC;gBACtC,KAAK,MAAM,GAAG,IAAI,OAAO,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;oBACnD,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,IAA4B,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;gBAClF,CAAC;YACH,CAAC;YAED,mCAAmC;YACnC,IAAI,YAAY,GAAG,iIAAiI,CAAC;YACrJ,IAAI,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;gBACnC,YAAY,IAAI,8BAA8B,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvG,CAAC;YACD,IAAI,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;gBACnC,YAAY,IAAI,sBAAsB,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5E,CAAC;YAED,yBAAyB;YACzB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAEzD,2DAA2D;YAC3D,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,MAAM,cAAc,GAAG,EAAE,CAAC;YAE1B,OAAO,SAAS,GAAG,cAAc,IAAI,IAAI,CAAC,eAAe,KAAK,OAAO,CAAC,EAAE,EAAE,CAAC;gBACzE,SAAS,EAAE,CAAC;gBAEZ,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC;oBAC9B,QAAQ;oBACR,MAAM,EAAE,YAAY;oBACpB,MAAM,EAAE,IAAI;oBACZ,OAAO,EAAE,KAAK,EAAE,KAAa,EAAE,EAAE;wBAC/B,IAAI,IAAI,CAAC,eAAe,KAAK,OAAO,CAAC,EAAE;4BAAE,OAAO;wBAChD,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC;4BAC1C,SAAS,EAAE,OAAO,CAAC,EAAE;4BACrB,IAAI,EAAE,OAAO;4BACb,OAAO,EAAE,KAAK;4BACd,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;yBACtB,CAAC,CAAC;oBACL,CAAC;iBACF,CAAC,CAAC;gBAEH,8DAA8D;gBAC9D,IAAI,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxD,mDAAmD;oBACnD,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,EAAE;wBAC/B,SAAS,EAAE,QAAQ,CAAC,SAAS;qBAC9B,CAAC,CAAC;oBAEH,KAAK,MAAM,QAAQ,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;wBAC1C,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,cAAc,QAAQ,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;wBAE9G,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC;4BAC1C,SAAS,EAAE,OAAO,CAAC,EAAE;4BACrB,IAAI,EAAE,WAAW;4BACjB,QAAQ,EAAE;gCACR,EAAE,EAAE,QAAQ,CAAC,EAAE;gCACf,IAAI,EAAE,QAAQ,CAAC,IAAI;gCACnB,SAAS,EAAE,QAAQ,CAAC,SAAS;6BAC9B;4BACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;yBACtB,CAAC,CAAC;wBAEH,2DAA2D;wBAC3D,gCAAgC;wBAChC,MAAM,UAAU,GAAG;4BACjB,MAAM,EAAE,QAAQ,CAAC,EAAE;4BACnB,MAAM,EAAE,iDAAiD;4BACzD,OAAO,EAAE,KAAK;yBACf,CAAC;wBAEF,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC;4BAC1C,SAAS,EAAE,OAAO,CAAC,EAAE;4BACrB,IAAI,EAAE,aAAa;4BACnB,UAAU;4BACV,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;yBACtB,CAAC,CAAC;wBAEH,iDAAiD;wBACjD,QAAQ,CAAC,IAAI,CAAC;4BACZ,IAAI,EAAE,MAAM;4BACZ,OAAO,EAAE,mBAAmB,QAAQ,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,EAAE;yBAClE,CAAC,CAAC;oBACL,CAAC;oBAED,0CAA0C;oBAC1C,SAAS;gBACX,CAAC;gBAED,iCAAiC;gBACjC,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC;oBAC1C,SAAS,EAAE,OAAO,CAAC,EAAE;oBACrB,IAAI,EAAE,UAAU;oBAChB,OAAO,EAAE,QAAQ,CAAC,OAAO;oBACzB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC,CAAC;gBAEH,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,yBAAyB,SAAS,aAAa,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;gBACrG,MAAM;YACR,CAAC;YAED,IAAI,SAAS,IAAI,cAAc,EAAE,CAAC;gBAChC,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,EAAE,mBAAmB,cAAc,WAAW,CAAC,CAAC;YACpF,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,cAAc,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACxD,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QACrD,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC9B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,SAAiB,EAAE,KAAa;QACzD,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC;YAC1C,SAAS;YACT,IAAI,EAAE,OAAO;YACb,KAAK;YACL,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;IACL,CAAC;CACF;AAhLD,kCAgLC"}
|
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
export interface MqttClientOptions {
|
|
2
|
+
brokerUrl: string;
|
|
3
|
+
userId: string;
|
|
4
|
+
mqttJwt: string;
|
|
5
|
+
heartbeatIntervalMs: number;
|
|
6
|
+
}
|
|
7
|
+
export interface AgentCommand {
|
|
8
|
+
id: string;
|
|
9
|
+
type: 'prompt' | 'cancel' | 'ping';
|
|
10
|
+
prompt?: string;
|
|
11
|
+
context?: {
|
|
12
|
+
conversationId?: string;
|
|
13
|
+
projectId?: string;
|
|
14
|
+
files?: string[];
|
|
15
|
+
facts?: string[];
|
|
16
|
+
previousMessages?: Array<{
|
|
17
|
+
role: string;
|
|
18
|
+
content: string;
|
|
19
|
+
}>;
|
|
20
|
+
};
|
|
21
|
+
model?: {
|
|
22
|
+
provider?: string;
|
|
23
|
+
model?: string;
|
|
24
|
+
apiKey?: string;
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
export interface AgentResult {
|
|
28
|
+
commandId: string;
|
|
29
|
+
type: 'chunk' | 'tool_call' | 'tool_result' | 'complete' | 'error' | 'heartbeat';
|
|
30
|
+
content?: string;
|
|
31
|
+
toolCall?: {
|
|
32
|
+
id: string;
|
|
33
|
+
name: string;
|
|
34
|
+
arguments: Record<string, any>;
|
|
35
|
+
};
|
|
36
|
+
toolResult?: {
|
|
37
|
+
callId: string;
|
|
38
|
+
output: string;
|
|
39
|
+
isError: boolean;
|
|
40
|
+
};
|
|
41
|
+
error?: string;
|
|
42
|
+
timestamp: number;
|
|
43
|
+
}
|
|
44
|
+
type CommandHandler = (command: AgentCommand) => void;
|
|
45
|
+
export declare class AgentMqttClient {
|
|
46
|
+
private client;
|
|
47
|
+
private options;
|
|
48
|
+
private heartbeatTimer;
|
|
49
|
+
private commandHandler;
|
|
50
|
+
private reconnectCount;
|
|
51
|
+
constructor(options: MqttClientOptions);
|
|
52
|
+
connect(): Promise<void>;
|
|
53
|
+
onCommand(handler: CommandHandler): void;
|
|
54
|
+
publishResult(result: AgentResult): Promise<void>;
|
|
55
|
+
private publishStatus;
|
|
56
|
+
private startHeartbeat;
|
|
57
|
+
private stopHeartbeat;
|
|
58
|
+
publishProviders(providers: {
|
|
59
|
+
id: string;
|
|
60
|
+
model: string;
|
|
61
|
+
label?: string;
|
|
62
|
+
}[]): void;
|
|
63
|
+
disconnect(): Promise<void>;
|
|
64
|
+
get connected(): boolean;
|
|
65
|
+
}
|
|
66
|
+
export {};
|
|
67
|
+
//# sourceMappingURL=mqtt-client.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mqtt-client.d.ts","sourceRoot":"","sources":["../src/mqtt-client.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,QAAQ,GAAG,QAAQ,GAAG,MAAM,CAAC;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE;QACR,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;QACjB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;QACjB,gBAAgB,CAAC,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KAC7D,CAAC;IACF,KAAK,CAAC,EAAE;QACN,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,CAAC;CACH;AAED,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,OAAO,GAAG,WAAW,GAAG,aAAa,GAAG,UAAU,GAAG,OAAO,GAAG,WAAW,CAAC;IACjF,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE;QACT,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAChC,CAAC;IACF,UAAU,CAAC,EAAE;QACX,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,OAAO,CAAC;KAClB,CAAC;IACF,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,KAAK,cAAc,GAAG,CAAC,OAAO,EAAE,YAAY,KAAK,IAAI,CAAC;AAEtD,qBAAa,eAAe;IAC1B,OAAO,CAAC,MAAM,CAA2B;IACzC,OAAO,CAAC,OAAO,CAAoB;IACnC,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,cAAc,CAAK;gBAEf,OAAO,EAAE,iBAAiB;IAIhC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAuE9B,SAAS,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAIlC,aAAa,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBvD,OAAO,CAAC,aAAa;IAYrB,OAAO,CAAC,cAAc;IAMtB,OAAO,CAAC,aAAa;IAOrB,gBAAgB,CAAC,SAAS,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,GAAG,IAAI;IAY5E,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAajC,IAAI,SAAS,IAAI,OAAO,CAEvB;CACF"}
|
|
@@ -0,0 +1,148 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.AgentMqttClient = void 0;
|
|
7
|
+
const mqtt_1 = __importDefault(require("mqtt"));
|
|
8
|
+
const chalk_1 = __importDefault(require("chalk"));
|
|
9
|
+
class AgentMqttClient {
|
|
10
|
+
client = null;
|
|
11
|
+
options;
|
|
12
|
+
heartbeatTimer = null;
|
|
13
|
+
commandHandler = null;
|
|
14
|
+
reconnectCount = 0;
|
|
15
|
+
constructor(options) {
|
|
16
|
+
this.options = options;
|
|
17
|
+
}
|
|
18
|
+
async connect() {
|
|
19
|
+
return new Promise((resolve, reject) => {
|
|
20
|
+
const clientId = `funolio-agent-${this.options.userId}-${Date.now()}`;
|
|
21
|
+
const mqttOptions = {
|
|
22
|
+
clientId,
|
|
23
|
+
username: this.options.userId,
|
|
24
|
+
password: this.options.mqttJwt,
|
|
25
|
+
clean: true,
|
|
26
|
+
keepalive: 60,
|
|
27
|
+
reconnectPeriod: 5000,
|
|
28
|
+
connectTimeout: 10000,
|
|
29
|
+
will: {
|
|
30
|
+
topic: `funolio/${this.options.userId}/agent/status`,
|
|
31
|
+
payload: Buffer.from(JSON.stringify({
|
|
32
|
+
status: 'offline',
|
|
33
|
+
timestamp: Date.now(),
|
|
34
|
+
})),
|
|
35
|
+
qos: 1,
|
|
36
|
+
retain: true,
|
|
37
|
+
},
|
|
38
|
+
};
|
|
39
|
+
this.client = mqtt_1.default.connect(this.options.brokerUrl, mqttOptions);
|
|
40
|
+
this.client.on('connect', () => {
|
|
41
|
+
this.reconnectCount = 0;
|
|
42
|
+
// Subscribe to commands
|
|
43
|
+
const commandTopic = `funolio/${this.options.userId}/agent/commands`;
|
|
44
|
+
this.client.subscribe(commandTopic, { qos: 1 }, (err) => {
|
|
45
|
+
if (err) {
|
|
46
|
+
reject(new Error(`Failed to subscribe: ${err.message}`));
|
|
47
|
+
return;
|
|
48
|
+
}
|
|
49
|
+
// Publish online status
|
|
50
|
+
this.publishStatus('online');
|
|
51
|
+
// Start heartbeat
|
|
52
|
+
this.startHeartbeat();
|
|
53
|
+
resolve();
|
|
54
|
+
});
|
|
55
|
+
});
|
|
56
|
+
this.client.on('message', (_topic, payload) => {
|
|
57
|
+
try {
|
|
58
|
+
const message = JSON.parse(payload.toString());
|
|
59
|
+
if (this.commandHandler) {
|
|
60
|
+
this.commandHandler(message);
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
catch (error) {
|
|
64
|
+
console.error(chalk_1.default.red('Failed to parse command:'), error);
|
|
65
|
+
}
|
|
66
|
+
});
|
|
67
|
+
this.client.on('reconnect', () => {
|
|
68
|
+
this.reconnectCount++;
|
|
69
|
+
console.log(chalk_1.default.yellow(`Reconnecting... (attempt ${this.reconnectCount})`));
|
|
70
|
+
});
|
|
71
|
+
this.client.on('error', (error) => {
|
|
72
|
+
console.error(chalk_1.default.red(`MQTT error: ${error.message}`));
|
|
73
|
+
if (this.reconnectCount === 0) {
|
|
74
|
+
reject(error);
|
|
75
|
+
}
|
|
76
|
+
});
|
|
77
|
+
this.client.on('close', () => {
|
|
78
|
+
this.stopHeartbeat();
|
|
79
|
+
});
|
|
80
|
+
});
|
|
81
|
+
}
|
|
82
|
+
onCommand(handler) {
|
|
83
|
+
this.commandHandler = handler;
|
|
84
|
+
}
|
|
85
|
+
async publishResult(result) {
|
|
86
|
+
if (!this.client?.connected) {
|
|
87
|
+
throw new Error('Not connected to MQTT broker');
|
|
88
|
+
}
|
|
89
|
+
const topic = `funolio/${this.options.userId}/agent/results`;
|
|
90
|
+
const payload = JSON.stringify(result);
|
|
91
|
+
return new Promise((resolve, reject) => {
|
|
92
|
+
this.client.publish(topic, payload, { qos: 1 }, (err) => {
|
|
93
|
+
if (err)
|
|
94
|
+
reject(err);
|
|
95
|
+
else
|
|
96
|
+
resolve();
|
|
97
|
+
});
|
|
98
|
+
});
|
|
99
|
+
}
|
|
100
|
+
publishStatus(status) {
|
|
101
|
+
if (!this.client?.connected)
|
|
102
|
+
return;
|
|
103
|
+
const topic = `funolio/${this.options.userId}/agent/status`;
|
|
104
|
+
const payload = JSON.stringify({
|
|
105
|
+
status,
|
|
106
|
+
timestamp: Date.now(),
|
|
107
|
+
});
|
|
108
|
+
this.client.publish(topic, payload, { qos: 1, retain: true });
|
|
109
|
+
}
|
|
110
|
+
startHeartbeat() {
|
|
111
|
+
this.heartbeatTimer = setInterval(() => {
|
|
112
|
+
this.publishStatus('online');
|
|
113
|
+
}, this.options.heartbeatIntervalMs);
|
|
114
|
+
}
|
|
115
|
+
stopHeartbeat() {
|
|
116
|
+
if (this.heartbeatTimer) {
|
|
117
|
+
clearInterval(this.heartbeatTimer);
|
|
118
|
+
this.heartbeatTimer = null;
|
|
119
|
+
}
|
|
120
|
+
}
|
|
121
|
+
publishProviders(providers) {
|
|
122
|
+
if (!this.client?.connected)
|
|
123
|
+
return;
|
|
124
|
+
const topic = `funolio/${this.options.userId}/agent/providers`;
|
|
125
|
+
const payload = JSON.stringify({
|
|
126
|
+
providers,
|
|
127
|
+
timestamp: Date.now(),
|
|
128
|
+
});
|
|
129
|
+
this.client.publish(topic, payload, { qos: 1, retain: true });
|
|
130
|
+
}
|
|
131
|
+
async disconnect() {
|
|
132
|
+
this.stopHeartbeat();
|
|
133
|
+
this.publishStatus('offline');
|
|
134
|
+
return new Promise((resolve) => {
|
|
135
|
+
if (this.client) {
|
|
136
|
+
this.client.end(false, {}, () => resolve());
|
|
137
|
+
}
|
|
138
|
+
else {
|
|
139
|
+
resolve();
|
|
140
|
+
}
|
|
141
|
+
});
|
|
142
|
+
}
|
|
143
|
+
get connected() {
|
|
144
|
+
return this.client?.connected ?? false;
|
|
145
|
+
}
|
|
146
|
+
}
|
|
147
|
+
exports.AgentMqttClient = AgentMqttClient;
|
|
148
|
+
//# sourceMappingURL=mqtt-client.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mqtt-client.js","sourceRoot":"","sources":["../src/mqtt-client.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAwD;AACxD,kDAA0B;AA+C1B,MAAa,eAAe;IAClB,MAAM,GAAsB,IAAI,CAAC;IACjC,OAAO,CAAoB;IAC3B,cAAc,GAA0B,IAAI,CAAC;IAC7C,cAAc,GAA0B,IAAI,CAAC;IAC7C,cAAc,GAAG,CAAC,CAAC;IAE3B,YAAY,OAA0B;QACpC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,OAAO;QACX,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,QAAQ,GAAG,iBAAiB,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YAEtE,MAAM,WAAW,GAAmB;gBAClC,QAAQ;gBACR,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;gBAC7B,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;gBAC9B,KAAK,EAAE,IAAI;gBACX,SAAS,EAAE,EAAE;gBACb,eAAe,EAAE,IAAI;gBACrB,cAAc,EAAE,KAAK;gBACrB,IAAI,EAAE;oBACJ,KAAK,EAAE,WAAW,IAAI,CAAC,OAAO,CAAC,MAAM,eAAe;oBACpD,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;wBAClC,MAAM,EAAE,SAAS;wBACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qBACtB,CAAC,CAAC;oBACH,GAAG,EAAE,CAAC;oBACN,MAAM,EAAE,IAAI;iBACb;aACF,CAAC;YAEF,IAAI,CAAC,MAAM,GAAG,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YAEhE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;gBAC7B,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;gBACxB,wBAAwB;gBACxB,MAAM,YAAY,GAAG,WAAW,IAAI,CAAC,OAAO,CAAC,MAAM,iBAAiB,CAAC;gBACrE,IAAI,CAAC,MAAO,CAAC,SAAS,CAAC,YAAY,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE;oBACvD,IAAI,GAAG,EAAE,CAAC;wBACR,MAAM,CAAC,IAAI,KAAK,CAAC,wBAAwB,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;wBACzD,OAAO;oBACT,CAAC;oBACD,wBAAwB;oBACxB,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;oBAC7B,kBAAkB;oBAClB,IAAI,CAAC,cAAc,EAAE,CAAC;oBACtB,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;gBAC5C,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAiB,CAAC;oBAC/D,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;wBACxB,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;oBAC/B,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,KAAK,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;gBAC/B,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,MAAM,CAAC,4BAA4B,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC;YAChF,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAChC,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,eAAe,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBACzD,IAAI,IAAI,CAAC,cAAc,KAAK,CAAC,EAAE,CAAC;oBAC9B,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBAC3B,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,OAAuB;QAC/B,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAmB;QACrC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;QAED,MAAM,KAAK,GAAG,WAAW,IAAI,CAAC,OAAO,CAAC,MAAM,gBAAgB,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAEvC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,MAAO,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE;gBACvD,IAAI,GAAG;oBAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;oBAChB,OAAO,EAAE,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,MAAqC;QACzD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS;YAAE,OAAO;QAEpC,MAAM,KAAK,GAAG,WAAW,IAAI,CAAC,OAAO,CAAC,MAAM,eAAe,CAAC;QAC5D,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;YAC7B,MAAM;YACN,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAChE,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YACrC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IACvC,CAAC;IAEO,aAAa;QACnB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,gBAAgB,CAAC,SAA0D;QACzE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,SAAS;YAAE,OAAO;QAEpC,MAAM,KAAK,GAAG,WAAW,IAAI,CAAC,OAAO,CAAC,MAAM,kBAAkB,CAAC;QAC/D,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;YAC7B,SAAS;YACT,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAE9B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACN,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,EAAE,SAAS,IAAI,KAAK,CAAC;IACzC,CAAC;CACF;AA3JD,0CA2JC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import { LLMProvider, LLMProviderConfig, LLMChatOptions, LLMResponse } from './index';
|
|
2
|
+
export declare class AnthropicProvider implements LLMProvider {
|
|
3
|
+
private config;
|
|
4
|
+
private baseUrl;
|
|
5
|
+
constructor(config: LLMProviderConfig);
|
|
6
|
+
chat(options: LLMChatOptions): Promise<LLMResponse>;
|
|
7
|
+
private chatStream;
|
|
8
|
+
private parseResponse;
|
|
9
|
+
}
|
|
10
|
+
//# sourceMappingURL=anthropic.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,cAAc,EAAE,WAAW,EAAe,MAAM,SAAS,CAAC;AAGnG,qBAAa,iBAAkB,YAAW,WAAW;IACnD,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,OAAO,CAAkC;gBAErC,MAAM,EAAE,iBAAiB;IAI/B,IAAI,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,WAAW,CAAC;YAuD3C,UAAU;IA0GxB,OAAO,CAAC,aAAa;CA0BtB"}
|
|
@@ -0,0 +1,183 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.AnthropicProvider = void 0;
|
|
4
|
+
const retry_1 = require("./retry");
|
|
5
|
+
class AnthropicProvider {
|
|
6
|
+
config;
|
|
7
|
+
baseUrl = 'https://api.anthropic.com/v1';
|
|
8
|
+
constructor(config) {
|
|
9
|
+
this.config = config;
|
|
10
|
+
}
|
|
11
|
+
async chat(options) {
|
|
12
|
+
return (0, retry_1.withRetry)(async () => {
|
|
13
|
+
const messages = options.messages.map(m => {
|
|
14
|
+
if (m.role === 'assistant' && m.toolCalls?.length) {
|
|
15
|
+
return {
|
|
16
|
+
role: 'assistant',
|
|
17
|
+
content: [
|
|
18
|
+
...(m.content ? [{ type: 'text', text: m.content }] : []),
|
|
19
|
+
...m.toolCalls.map(tc => ({
|
|
20
|
+
type: 'tool_use',
|
|
21
|
+
id: tc.id,
|
|
22
|
+
name: tc.name,
|
|
23
|
+
input: tc.arguments,
|
|
24
|
+
})),
|
|
25
|
+
],
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
return { role: m.role, content: m.content };
|
|
29
|
+
});
|
|
30
|
+
const body = {
|
|
31
|
+
model: this.config.model,
|
|
32
|
+
max_tokens: 4096,
|
|
33
|
+
messages,
|
|
34
|
+
};
|
|
35
|
+
if (options.system) {
|
|
36
|
+
body.system = options.system;
|
|
37
|
+
}
|
|
38
|
+
if (options.stream && options.onChunk) {
|
|
39
|
+
return this.chatStream(body, options.onChunk);
|
|
40
|
+
}
|
|
41
|
+
const response = await fetch(`${this.baseUrl}/messages`, {
|
|
42
|
+
method: 'POST',
|
|
43
|
+
headers: {
|
|
44
|
+
'Content-Type': 'application/json',
|
|
45
|
+
'x-api-key': this.config.apiKey,
|
|
46
|
+
'anthropic-version': '2023-06-01',
|
|
47
|
+
},
|
|
48
|
+
body: JSON.stringify(body),
|
|
49
|
+
});
|
|
50
|
+
if (!response.ok) {
|
|
51
|
+
const error = new Error(`Anthropic API error: ${response.status}`);
|
|
52
|
+
error.status = response.status;
|
|
53
|
+
throw error;
|
|
54
|
+
}
|
|
55
|
+
const data = await response.json();
|
|
56
|
+
return this.parseResponse(data);
|
|
57
|
+
}, this.config.maxRetries, this.config.baseDelay);
|
|
58
|
+
}
|
|
59
|
+
async chatStream(body, onChunk) {
|
|
60
|
+
body.stream = true;
|
|
61
|
+
const response = await fetch(`${this.baseUrl}/messages`, {
|
|
62
|
+
method: 'POST',
|
|
63
|
+
headers: {
|
|
64
|
+
'Content-Type': 'application/json',
|
|
65
|
+
'x-api-key': this.config.apiKey,
|
|
66
|
+
'anthropic-version': '2023-06-01',
|
|
67
|
+
},
|
|
68
|
+
body: JSON.stringify(body),
|
|
69
|
+
});
|
|
70
|
+
if (!response.ok) {
|
|
71
|
+
const error = new Error(`Anthropic API error: ${response.status}`);
|
|
72
|
+
error.status = response.status;
|
|
73
|
+
throw error;
|
|
74
|
+
}
|
|
75
|
+
const reader = response.body?.getReader();
|
|
76
|
+
if (!reader)
|
|
77
|
+
throw new Error('No response body');
|
|
78
|
+
const decoder = new TextDecoder();
|
|
79
|
+
let fullContent = '';
|
|
80
|
+
const toolCalls = [];
|
|
81
|
+
let currentToolCall = null;
|
|
82
|
+
let toolCallJson = '';
|
|
83
|
+
let inputTokens = 0;
|
|
84
|
+
let outputTokens = 0;
|
|
85
|
+
let buffer = '';
|
|
86
|
+
while (true) {
|
|
87
|
+
const { done, value } = await reader.read();
|
|
88
|
+
if (done)
|
|
89
|
+
break;
|
|
90
|
+
buffer += decoder.decode(value, { stream: true });
|
|
91
|
+
const lines = buffer.split('\n');
|
|
92
|
+
buffer = lines.pop() || '';
|
|
93
|
+
for (const line of lines) {
|
|
94
|
+
if (!line.startsWith('data: '))
|
|
95
|
+
continue;
|
|
96
|
+
const data = line.slice(6).trim();
|
|
97
|
+
if (data === '[DONE]')
|
|
98
|
+
continue;
|
|
99
|
+
try {
|
|
100
|
+
const event = JSON.parse(data);
|
|
101
|
+
switch (event.type) {
|
|
102
|
+
case 'content_block_start':
|
|
103
|
+
if (event.content_block?.type === 'tool_use') {
|
|
104
|
+
currentToolCall = {
|
|
105
|
+
id: event.content_block.id,
|
|
106
|
+
name: event.content_block.name,
|
|
107
|
+
};
|
|
108
|
+
toolCallJson = '';
|
|
109
|
+
}
|
|
110
|
+
break;
|
|
111
|
+
case 'content_block_delta':
|
|
112
|
+
if (event.delta?.type === 'text_delta') {
|
|
113
|
+
fullContent += event.delta.text;
|
|
114
|
+
await onChunk(event.delta.text);
|
|
115
|
+
}
|
|
116
|
+
else if (event.delta?.type === 'input_json_delta') {
|
|
117
|
+
toolCallJson += event.delta.partial_json;
|
|
118
|
+
}
|
|
119
|
+
break;
|
|
120
|
+
case 'content_block_stop':
|
|
121
|
+
if (currentToolCall) {
|
|
122
|
+
try {
|
|
123
|
+
currentToolCall.arguments = JSON.parse(toolCallJson || '{}');
|
|
124
|
+
}
|
|
125
|
+
catch {
|
|
126
|
+
currentToolCall.arguments = {};
|
|
127
|
+
}
|
|
128
|
+
toolCalls.push(currentToolCall);
|
|
129
|
+
currentToolCall = null;
|
|
130
|
+
toolCallJson = '';
|
|
131
|
+
}
|
|
132
|
+
break;
|
|
133
|
+
case 'message_delta':
|
|
134
|
+
if (event.usage) {
|
|
135
|
+
outputTokens = event.usage.output_tokens || 0;
|
|
136
|
+
}
|
|
137
|
+
break;
|
|
138
|
+
case 'message_start':
|
|
139
|
+
if (event.message?.usage) {
|
|
140
|
+
inputTokens = event.message.usage.input_tokens || 0;
|
|
141
|
+
}
|
|
142
|
+
break;
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
catch {
|
|
146
|
+
// skip malformed events
|
|
147
|
+
}
|
|
148
|
+
}
|
|
149
|
+
}
|
|
150
|
+
return {
|
|
151
|
+
content: fullContent,
|
|
152
|
+
toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
153
|
+
usage: { inputTokens, outputTokens },
|
|
154
|
+
};
|
|
155
|
+
}
|
|
156
|
+
parseResponse(data) {
|
|
157
|
+
let content = '';
|
|
158
|
+
const toolCalls = [];
|
|
159
|
+
for (const block of data.content || []) {
|
|
160
|
+
if (block.type === 'text') {
|
|
161
|
+
content += block.text;
|
|
162
|
+
}
|
|
163
|
+
else if (block.type === 'tool_use') {
|
|
164
|
+
toolCalls.push({
|
|
165
|
+
id: block.id,
|
|
166
|
+
name: block.name,
|
|
167
|
+
arguments: block.input,
|
|
168
|
+
});
|
|
169
|
+
}
|
|
170
|
+
}
|
|
171
|
+
return {
|
|
172
|
+
content,
|
|
173
|
+
toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
174
|
+
usage: data.usage ? {
|
|
175
|
+
inputTokens: data.usage.input_tokens,
|
|
176
|
+
outputTokens: data.usage.output_tokens,
|
|
177
|
+
} : undefined,
|
|
178
|
+
finishReason: data.stop_reason,
|
|
179
|
+
};
|
|
180
|
+
}
|
|
181
|
+
}
|
|
182
|
+
exports.AnthropicProvider = AnthropicProvider;
|
|
183
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":";;;AACA,mCAAoC;AAEpC,MAAa,iBAAiB;IACpB,MAAM,CAAoB;IAC1B,OAAO,GAAG,8BAA8B,CAAC;IAEjD,YAAY,MAAyB;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAuB;QAChC,OAAO,IAAA,iBAAS,EAAC,KAAK,IAAI,EAAE;YAC1B,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;gBACxC,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;oBAClD,OAAO;wBACL,IAAI,EAAE,WAAoB;wBAC1B,OAAO,EAAE;4BACP,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;4BAClE,GAAG,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gCACxB,IAAI,EAAE,UAAmB;gCACzB,EAAE,EAAE,EAAE,CAAC,EAAE;gCACT,IAAI,EAAE,EAAE,CAAC,IAAI;gCACb,KAAK,EAAE,EAAE,CAAC,SAAS;6BACpB,CAAC,CAAC;yBACJ;qBACF,CAAC;gBACJ,CAAC;gBACD,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,MAAM,IAAI,GAAwB;gBAChC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBACxB,UAAU,EAAE,IAAI;gBAChB,QAAQ;aACT,CAAC;YAEF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAC/B,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACtC,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;YAChD,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;gBACvD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;oBAC/B,mBAAmB,EAAE,YAAY;iBAClC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,KAAK,GAAQ,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBACxE,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;gBAC/B,MAAM,KAAK,CAAC;YACd,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAS,CAAC;YAC1C,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACpD,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,IAAyB,EAAE,OAAyC;QAC3F,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QAEnB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;YACvD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;gBAC/B,mBAAmB,EAAE,YAAY;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,KAAK,GAAQ,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YACxE,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;YAC/B,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;QAC1C,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QAEjD,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,MAAM,SAAS,GAAkB,EAAE,CAAC;QACpC,IAAI,eAAe,GAAgC,IAAI,CAAC;QACxD,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,IAAI,YAAY,GAAG,CAAC,CAAC;QAErB,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;oBAAE,SAAS;gBACzC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAClC,IAAI,IAAI,KAAK,QAAQ;oBAAE,SAAS;gBAEhC,IAAI,CAAC;oBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAE/B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;wBACnB,KAAK,qBAAqB;4BACxB,IAAI,KAAK,CAAC,aAAa,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;gCAC7C,eAAe,GAAG;oCAChB,EAAE,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE;oCAC1B,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,IAAI;iCAC/B,CAAC;gCACF,YAAY,GAAG,EAAE,CAAC;4BACpB,CAAC;4BACD,MAAM;wBAER,KAAK,qBAAqB;4BACxB,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;gCACvC,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;gCAChC,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;4BAClC,CAAC;iCAAM,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,kBAAkB,EAAE,CAAC;gCACpD,YAAY,IAAI,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;4BAC3C,CAAC;4BACD,MAAM;wBAER,KAAK,oBAAoB;4BACvB,IAAI,eAAe,EAAE,CAAC;gCACpB,IAAI,CAAC;oCACH,eAAe,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,IAAI,CAAC,CAAC;gCAC/D,CAAC;gCAAC,MAAM,CAAC;oCACP,eAAe,CAAC,SAAS,GAAG,EAAE,CAAC;gCACjC,CAAC;gCACD,SAAS,CAAC,IAAI,CAAC,eAA8B,CAAC,CAAC;gCAC/C,eAAe,GAAG,IAAI,CAAC;gCACvB,YAAY,GAAG,EAAE,CAAC;4BACpB,CAAC;4BACD,MAAM;wBAER,KAAK,eAAe;4BAClB,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;gCAChB,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;4BAChD,CAAC;4BACD,MAAM;wBAER,KAAK,eAAe;4BAClB,IAAI,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;gCACzB,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;4BACtD,CAAC;4BACD,MAAM;oBACV,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,wBAAwB;gBAC1B,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE,WAAW;YACpB,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACvD,KAAK,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE;SACrC,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,IAAS;QAC7B,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,MAAM,SAAS,GAAkB,EAAE,CAAC;QAEpC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YACvC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC;YACxB,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACrC,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,KAAK,CAAC,EAAE;oBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,SAAS,EAAE,KAAK,CAAC,KAAK;iBACvB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO;YACP,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACvD,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;gBAClB,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;gBACpC,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;aACvC,CAAC,CAAC,CAAC,SAAS;YACb,YAAY,EAAE,IAAI,CAAC,WAAW;SAC/B,CAAC;IACJ,CAAC;CACF;AAnMD,8CAmMC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import { LLMProvider, LLMProviderConfig, LLMChatOptions, LLMResponse } from './index';
|
|
2
|
+
export declare class GoogleProvider implements LLMProvider {
|
|
3
|
+
private config;
|
|
4
|
+
private baseUrl;
|
|
5
|
+
constructor(config: LLMProviderConfig);
|
|
6
|
+
chat(options: LLMChatOptions): Promise<LLMResponse>;
|
|
7
|
+
private chatStream;
|
|
8
|
+
private parseResponse;
|
|
9
|
+
}
|
|
10
|
+
//# sourceMappingURL=google.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../src/providers/google.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,cAAc,EAAE,WAAW,EAAe,MAAM,SAAS,CAAC;AAGnG,qBAAa,cAAe,YAAW,WAAW;IAChD,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,OAAO,CAAsD;gBAEzD,MAAM,EAAE,iBAAiB;IAI/B,IAAI,CAAC,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,WAAW,CAAC;YA+D3C,UAAU;IAoExB,OAAO,CAAC,aAAa;CA+BtB"}
|