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.
Files changed (138) hide show
  1. package/README.md +55 -0
  2. package/dist/clerk/index.d.ts +62 -0
  3. package/dist/clerk/index.d.ts.map +1 -0
  4. package/dist/clerk/index.js +186 -0
  5. package/dist/clerk/index.js.map +1 -0
  6. package/dist/cli.d.ts +3 -0
  7. package/dist/cli.d.ts.map +1 -0
  8. package/dist/cli.js +42 -0
  9. package/dist/cli.js.map +1 -0
  10. package/dist/commands/configure.d.ts +2 -0
  11. package/dist/commands/configure.d.ts.map +1 -0
  12. package/dist/commands/configure.js +252 -0
  13. package/dist/commands/configure.js.map +1 -0
  14. package/dist/commands/init.d.ts +6 -0
  15. package/dist/commands/init.d.ts.map +1 -0
  16. package/dist/commands/init.js +151 -0
  17. package/dist/commands/init.js.map +1 -0
  18. package/dist/commands/login.d.ts +6 -0
  19. package/dist/commands/login.d.ts.map +1 -0
  20. package/dist/commands/login.js +170 -0
  21. package/dist/commands/login.js.map +1 -0
  22. package/dist/commands/start.d.ts +8 -0
  23. package/dist/commands/start.d.ts.map +1 -0
  24. package/dist/commands/start.js +179 -0
  25. package/dist/commands/start.js.map +1 -0
  26. package/dist/commands/status.d.ts +2 -0
  27. package/dist/commands/status.d.ts.map +1 -0
  28. package/dist/commands/status.js +55 -0
  29. package/dist/commands/status.js.map +1 -0
  30. package/dist/config.d.ts +46 -0
  31. package/dist/config.d.ts.map +1 -0
  32. package/dist/config.js +113 -0
  33. package/dist/config.js.map +1 -0
  34. package/dist/index.d.ts +24 -0
  35. package/dist/index.d.ts.map +1 -0
  36. package/dist/index.js +48 -0
  37. package/dist/index.js.map +1 -0
  38. package/dist/mcp/index.d.ts +9 -0
  39. package/dist/mcp/index.d.ts.map +1 -0
  40. package/dist/mcp/index.js +15 -0
  41. package/dist/mcp/index.js.map +1 -0
  42. package/dist/mcp/manager.d.ts +83 -0
  43. package/dist/mcp/manager.d.ts.map +1 -0
  44. package/dist/mcp/manager.js +338 -0
  45. package/dist/mcp/manager.js.map +1 -0
  46. package/dist/mcp/registry.d.ts +32 -0
  47. package/dist/mcp/registry.d.ts.map +1 -0
  48. package/dist/mcp/registry.js +139 -0
  49. package/dist/mcp/registry.js.map +1 -0
  50. package/dist/message-loop.d.ts +18 -0
  51. package/dist/message-loop.d.ts.map +1 -0
  52. package/dist/message-loop.js +165 -0
  53. package/dist/message-loop.js.map +1 -0
  54. package/dist/mqtt-client.d.ts +67 -0
  55. package/dist/mqtt-client.d.ts.map +1 -0
  56. package/dist/mqtt-client.js +148 -0
  57. package/dist/mqtt-client.js.map +1 -0
  58. package/dist/providers/anthropic.d.ts +10 -0
  59. package/dist/providers/anthropic.d.ts.map +1 -0
  60. package/dist/providers/anthropic.js +183 -0
  61. package/dist/providers/anthropic.js.map +1 -0
  62. package/dist/providers/google.d.ts +10 -0
  63. package/dist/providers/google.d.ts.map +1 -0
  64. package/dist/providers/google.js +161 -0
  65. package/dist/providers/google.js.map +1 -0
  66. package/dist/providers/index.d.ts +42 -0
  67. package/dist/providers/index.d.ts.map +1 -0
  68. package/dist/providers/index.js +19 -0
  69. package/dist/providers/index.js.map +1 -0
  70. package/dist/providers/openai.d.ts +10 -0
  71. package/dist/providers/openai.d.ts.map +1 -0
  72. package/dist/providers/openai.js +173 -0
  73. package/dist/providers/openai.js.map +1 -0
  74. package/dist/providers/retry.d.ts +6 -0
  75. package/dist/providers/retry.d.ts.map +1 -0
  76. package/dist/providers/retry.js +38 -0
  77. package/dist/providers/retry.js.map +1 -0
  78. package/dist/subagent/index.d.ts +8 -0
  79. package/dist/subagent/index.d.ts.map +1 -0
  80. package/dist/subagent/index.js +14 -0
  81. package/dist/subagent/index.js.map +1 -0
  82. package/dist/subagent/orchestrator.d.ts +67 -0
  83. package/dist/subagent/orchestrator.d.ts.map +1 -0
  84. package/dist/subagent/orchestrator.js +152 -0
  85. package/dist/subagent/orchestrator.js.map +1 -0
  86. package/dist/subagent/queue.d.ts +66 -0
  87. package/dist/subagent/queue.d.ts.map +1 -0
  88. package/dist/subagent/queue.js +298 -0
  89. package/dist/subagent/queue.js.map +1 -0
  90. package/dist/subagent/types.d.ts +76 -0
  91. package/dist/subagent/types.d.ts.map +1 -0
  92. package/dist/subagent/types.js +14 -0
  93. package/dist/subagent/types.js.map +1 -0
  94. package/dist/tools/edit-file.d.ts +3 -0
  95. package/dist/tools/edit-file.d.ts.map +1 -0
  96. package/dist/tools/edit-file.js +112 -0
  97. package/dist/tools/edit-file.js.map +1 -0
  98. package/dist/tools/git-tools.d.ts +5 -0
  99. package/dist/tools/git-tools.d.ts.map +1 -0
  100. package/dist/tools/git-tools.js +144 -0
  101. package/dist/tools/git-tools.js.map +1 -0
  102. package/dist/tools/index.d.ts +40 -0
  103. package/dist/tools/index.d.ts.map +1 -0
  104. package/dist/tools/index.js +126 -0
  105. package/dist/tools/index.js.map +1 -0
  106. package/dist/tools/installer.d.ts +41 -0
  107. package/dist/tools/installer.d.ts.map +1 -0
  108. package/dist/tools/installer.js +227 -0
  109. package/dist/tools/installer.js.map +1 -0
  110. package/dist/tools/list-directory.d.ts +3 -0
  111. package/dist/tools/list-directory.d.ts.map +1 -0
  112. package/dist/tools/list-directory.js +107 -0
  113. package/dist/tools/list-directory.js.map +1 -0
  114. package/dist/tools/read-file.d.ts +3 -0
  115. package/dist/tools/read-file.d.ts.map +1 -0
  116. package/dist/tools/read-file.js +89 -0
  117. package/dist/tools/read-file.js.map +1 -0
  118. package/dist/tools/run-command.d.ts +3 -0
  119. package/dist/tools/run-command.d.ts.map +1 -0
  120. package/dist/tools/run-command.js +86 -0
  121. package/dist/tools/run-command.js.map +1 -0
  122. package/dist/tools/sandbox.d.ts +17 -0
  123. package/dist/tools/sandbox.d.ts.map +1 -0
  124. package/dist/tools/sandbox.js +78 -0
  125. package/dist/tools/sandbox.js.map +1 -0
  126. package/dist/tools/write-file.d.ts +3 -0
  127. package/dist/tools/write-file.d.ts.map +1 -0
  128. package/dist/tools/write-file.js +88 -0
  129. package/dist/tools/write-file.js.map +1 -0
  130. package/dist/types.d.ts +67 -0
  131. package/dist/types.d.ts.map +1 -0
  132. package/dist/types.js +6 -0
  133. package/dist/types.js.map +1 -0
  134. package/dist/verification/index.d.ts +17 -0
  135. package/dist/verification/index.d.ts.map +1 -0
  136. package/dist/verification/index.js +224 -0
  137. package/dist/verification/index.js.map +1 -0
  138. 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"}