@lobu/gateway 3.0.9 → 3.0.12

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 (210) hide show
  1. package/dist/api/platform.d.ts.map +1 -1
  2. package/dist/api/platform.js +7 -26
  3. package/dist/api/platform.js.map +1 -1
  4. package/dist/auth/mcp/proxy.d.ts +14 -0
  5. package/dist/auth/mcp/proxy.d.ts.map +1 -1
  6. package/dist/auth/mcp/proxy.js +149 -13
  7. package/dist/auth/mcp/proxy.js.map +1 -1
  8. package/dist/cli/gateway.d.ts.map +1 -1
  9. package/dist/cli/gateway.js +29 -0
  10. package/dist/cli/gateway.js.map +1 -1
  11. package/dist/connections/chat-instance-manager.d.ts.map +1 -1
  12. package/dist/connections/chat-instance-manager.js +2 -1
  13. package/dist/connections/chat-instance-manager.js.map +1 -1
  14. package/dist/connections/interaction-bridge.d.ts +9 -2
  15. package/dist/connections/interaction-bridge.d.ts.map +1 -1
  16. package/dist/connections/interaction-bridge.js +121 -261
  17. package/dist/connections/interaction-bridge.js.map +1 -1
  18. package/dist/interactions.d.ts +9 -43
  19. package/dist/interactions.d.ts.map +1 -1
  20. package/dist/interactions.js +10 -52
  21. package/dist/interactions.js.map +1 -1
  22. package/dist/routes/public/agent.d.ts +4 -0
  23. package/dist/routes/public/agent.d.ts.map +1 -1
  24. package/dist/routes/public/agent.js +21 -0
  25. package/dist/routes/public/agent.js.map +1 -1
  26. package/dist/services/core-services.d.ts.map +1 -1
  27. package/dist/services/core-services.js +4 -0
  28. package/dist/services/core-services.js.map +1 -1
  29. package/package.json +9 -9
  30. package/src/__tests__/agent-config-routes.test.ts +0 -254
  31. package/src/__tests__/agent-history-routes.test.ts +0 -72
  32. package/src/__tests__/agent-routes.test.ts +0 -68
  33. package/src/__tests__/agent-schedules-routes.test.ts +0 -59
  34. package/src/__tests__/agent-settings-store.test.ts +0 -323
  35. package/src/__tests__/bedrock-model-catalog.test.ts +0 -40
  36. package/src/__tests__/bedrock-openai-service.test.ts +0 -157
  37. package/src/__tests__/bedrock-provider-module.test.ts +0 -56
  38. package/src/__tests__/chat-instance-manager-slack.test.ts +0 -204
  39. package/src/__tests__/chat-response-bridge.test.ts +0 -131
  40. package/src/__tests__/config-memory-plugins.test.ts +0 -92
  41. package/src/__tests__/config-request-store.test.ts +0 -127
  42. package/src/__tests__/connection-routes.test.ts +0 -144
  43. package/src/__tests__/core-services-store-selection.test.ts +0 -92
  44. package/src/__tests__/docker-deployment.test.ts +0 -1211
  45. package/src/__tests__/embedded-deployment.test.ts +0 -342
  46. package/src/__tests__/grant-store.test.ts +0 -148
  47. package/src/__tests__/http-proxy.test.ts +0 -281
  48. package/src/__tests__/instruction-service.test.ts +0 -37
  49. package/src/__tests__/link-buttons.test.ts +0 -112
  50. package/src/__tests__/lobu.test.ts +0 -32
  51. package/src/__tests__/mcp-config-service.test.ts +0 -347
  52. package/src/__tests__/mcp-proxy.test.ts +0 -694
  53. package/src/__tests__/message-handler-bridge.test.ts +0 -17
  54. package/src/__tests__/model-selection.test.ts +0 -172
  55. package/src/__tests__/oauth-templates.test.ts +0 -39
  56. package/src/__tests__/platform-adapter-slack-send.test.ts +0 -114
  57. package/src/__tests__/platform-helpers-model-resolution.test.ts +0 -253
  58. package/src/__tests__/provider-inheritance.test.ts +0 -212
  59. package/src/__tests__/routes/cli-auth.test.ts +0 -337
  60. package/src/__tests__/routes/interactions.test.ts +0 -121
  61. package/src/__tests__/secret-proxy.test.ts +0 -85
  62. package/src/__tests__/session-manager.test.ts +0 -572
  63. package/src/__tests__/setup.ts +0 -133
  64. package/src/__tests__/skill-and-mcp-registry.test.ts +0 -203
  65. package/src/__tests__/slack-routes.test.ts +0 -161
  66. package/src/__tests__/system-config-resolver.test.ts +0 -75
  67. package/src/__tests__/system-message-limiter.test.ts +0 -89
  68. package/src/__tests__/system-skills-service.test.ts +0 -362
  69. package/src/__tests__/transcription-service.test.ts +0 -222
  70. package/src/__tests__/utils/rate-limiter.test.ts +0 -102
  71. package/src/__tests__/worker-connection-manager.test.ts +0 -497
  72. package/src/__tests__/worker-job-router.test.ts +0 -722
  73. package/src/api/index.ts +0 -1
  74. package/src/api/platform.ts +0 -292
  75. package/src/api/response-renderer.ts +0 -157
  76. package/src/auth/agent-metadata-store.ts +0 -168
  77. package/src/auth/api-auth-middleware.ts +0 -69
  78. package/src/auth/api-key-provider-module.ts +0 -213
  79. package/src/auth/base-provider-module.ts +0 -201
  80. package/src/auth/bedrock/provider-module.ts +0 -110
  81. package/src/auth/chatgpt/chatgpt-oauth-module.ts +0 -185
  82. package/src/auth/chatgpt/device-code-client.ts +0 -218
  83. package/src/auth/chatgpt/index.ts +0 -1
  84. package/src/auth/claude/oauth-module.ts +0 -280
  85. package/src/auth/cli/token-service.ts +0 -249
  86. package/src/auth/external/client.ts +0 -560
  87. package/src/auth/external/device-code-client.ts +0 -235
  88. package/src/auth/mcp/config-service.ts +0 -420
  89. package/src/auth/mcp/proxy.ts +0 -1086
  90. package/src/auth/mcp/string-substitution.ts +0 -17
  91. package/src/auth/mcp/tool-cache.ts +0 -90
  92. package/src/auth/oauth/base-client.ts +0 -267
  93. package/src/auth/oauth/client.ts +0 -153
  94. package/src/auth/oauth/credentials.ts +0 -7
  95. package/src/auth/oauth/providers.ts +0 -69
  96. package/src/auth/oauth/state-store.ts +0 -150
  97. package/src/auth/oauth-templates.ts +0 -179
  98. package/src/auth/provider-catalog.ts +0 -220
  99. package/src/auth/provider-model-options.ts +0 -41
  100. package/src/auth/settings/agent-settings-store.ts +0 -565
  101. package/src/auth/settings/auth-profiles-manager.ts +0 -216
  102. package/src/auth/settings/index.ts +0 -12
  103. package/src/auth/settings/model-preference-store.ts +0 -52
  104. package/src/auth/settings/model-selection.ts +0 -135
  105. package/src/auth/settings/resolved-settings-view.ts +0 -298
  106. package/src/auth/settings/template-utils.ts +0 -44
  107. package/src/auth/settings/token-service.ts +0 -88
  108. package/src/auth/system-env-store.ts +0 -98
  109. package/src/auth/user-agents-store.ts +0 -68
  110. package/src/channels/binding-service.ts +0 -214
  111. package/src/channels/index.ts +0 -4
  112. package/src/cli/gateway.ts +0 -1312
  113. package/src/cli/index.ts +0 -74
  114. package/src/commands/built-in-commands.ts +0 -80
  115. package/src/commands/command-dispatcher.ts +0 -94
  116. package/src/commands/command-reply-adapters.ts +0 -27
  117. package/src/config/file-loader.ts +0 -618
  118. package/src/config/index.ts +0 -588
  119. package/src/config/network-allowlist.ts +0 -71
  120. package/src/connections/chat-instance-manager.ts +0 -1284
  121. package/src/connections/chat-response-bridge.ts +0 -618
  122. package/src/connections/index.ts +0 -7
  123. package/src/connections/interaction-bridge.ts +0 -831
  124. package/src/connections/message-handler-bridge.ts +0 -440
  125. package/src/connections/platform-auth-methods.ts +0 -15
  126. package/src/connections/types.ts +0 -84
  127. package/src/gateway/connection-manager.ts +0 -291
  128. package/src/gateway/index.ts +0 -698
  129. package/src/gateway/job-router.ts +0 -201
  130. package/src/gateway-main.ts +0 -200
  131. package/src/index.ts +0 -41
  132. package/src/infrastructure/queue/index.ts +0 -12
  133. package/src/infrastructure/queue/queue-producer.ts +0 -148
  134. package/src/infrastructure/queue/redis-queue.ts +0 -361
  135. package/src/infrastructure/queue/types.ts +0 -133
  136. package/src/infrastructure/redis/system-message-limiter.ts +0 -94
  137. package/src/interactions/config-request-store.ts +0 -198
  138. package/src/interactions.ts +0 -363
  139. package/src/lobu.ts +0 -311
  140. package/src/metrics/prometheus.ts +0 -159
  141. package/src/modules/module-system.ts +0 -179
  142. package/src/orchestration/base-deployment-manager.ts +0 -900
  143. package/src/orchestration/deployment-utils.ts +0 -98
  144. package/src/orchestration/impl/docker-deployment.ts +0 -620
  145. package/src/orchestration/impl/embedded-deployment.ts +0 -268
  146. package/src/orchestration/impl/index.ts +0 -8
  147. package/src/orchestration/impl/k8s/deployment.ts +0 -1061
  148. package/src/orchestration/impl/k8s/helpers.ts +0 -610
  149. package/src/orchestration/impl/k8s/index.ts +0 -1
  150. package/src/orchestration/index.ts +0 -333
  151. package/src/orchestration/message-consumer.ts +0 -584
  152. package/src/orchestration/scheduled-wakeup.ts +0 -704
  153. package/src/permissions/approval-policy.ts +0 -36
  154. package/src/permissions/grant-store.ts +0 -219
  155. package/src/platform/file-handler.ts +0 -66
  156. package/src/platform/link-buttons.ts +0 -57
  157. package/src/platform/renderer-utils.ts +0 -44
  158. package/src/platform/response-renderer.ts +0 -84
  159. package/src/platform/unified-thread-consumer.ts +0 -194
  160. package/src/platform.ts +0 -318
  161. package/src/proxy/http-proxy.ts +0 -752
  162. package/src/proxy/proxy-manager.ts +0 -81
  163. package/src/proxy/secret-proxy.ts +0 -402
  164. package/src/proxy/token-refresh-job.ts +0 -143
  165. package/src/routes/internal/audio.ts +0 -141
  166. package/src/routes/internal/device-auth.ts +0 -652
  167. package/src/routes/internal/files.ts +0 -226
  168. package/src/routes/internal/history.ts +0 -69
  169. package/src/routes/internal/images.ts +0 -127
  170. package/src/routes/internal/interactions.ts +0 -84
  171. package/src/routes/internal/middleware.ts +0 -23
  172. package/src/routes/internal/schedule.ts +0 -226
  173. package/src/routes/internal/types.ts +0 -22
  174. package/src/routes/openapi-auto.ts +0 -239
  175. package/src/routes/public/agent-access.ts +0 -23
  176. package/src/routes/public/agent-config.ts +0 -675
  177. package/src/routes/public/agent-history.ts +0 -422
  178. package/src/routes/public/agent-schedules.ts +0 -296
  179. package/src/routes/public/agent.ts +0 -1086
  180. package/src/routes/public/agents.ts +0 -373
  181. package/src/routes/public/channels.ts +0 -191
  182. package/src/routes/public/cli-auth.ts +0 -896
  183. package/src/routes/public/connections.ts +0 -574
  184. package/src/routes/public/landing.ts +0 -16
  185. package/src/routes/public/oauth.ts +0 -147
  186. package/src/routes/public/settings-auth.ts +0 -104
  187. package/src/routes/public/slack.ts +0 -173
  188. package/src/routes/shared/agent-ownership.ts +0 -101
  189. package/src/routes/shared/token-verifier.ts +0 -34
  190. package/src/services/bedrock-model-catalog.ts +0 -217
  191. package/src/services/bedrock-openai-service.ts +0 -658
  192. package/src/services/core-services.ts +0 -1072
  193. package/src/services/image-generation-service.ts +0 -257
  194. package/src/services/instruction-service.ts +0 -318
  195. package/src/services/mcp-registry.ts +0 -94
  196. package/src/services/platform-helpers.ts +0 -287
  197. package/src/services/session-manager.ts +0 -262
  198. package/src/services/settings-resolver.ts +0 -74
  199. package/src/services/system-config-resolver.ts +0 -89
  200. package/src/services/system-skills-service.ts +0 -229
  201. package/src/services/transcription-service.ts +0 -684
  202. package/src/session.ts +0 -110
  203. package/src/spaces/index.ts +0 -1
  204. package/src/spaces/space-resolver.ts +0 -17
  205. package/src/stores/in-memory-agent-store.ts +0 -403
  206. package/src/stores/redis-agent-store.ts +0 -279
  207. package/src/utils/public-url.ts +0 -44
  208. package/src/utils/rate-limiter.ts +0 -94
  209. package/tsconfig.json +0 -33
  210. package/tsconfig.tsbuildinfo +0 -1
package/src/lobu.ts DELETED
@@ -1,311 +0,0 @@
1
- import type { Server } from "node:http";
2
- import { createLogger } from "@lobu/core";
3
- import { ApiPlatform } from "./api";
4
- import { createGatewayApp, startGatewayServer } from "./cli/gateway";
5
- import { buildGatewayConfig, type GatewayConfig } from "./config";
6
- import { ChatInstanceManager, ChatResponseBridge } from "./connections";
7
- import { Gateway } from "./gateway-main";
8
- import { InMemoryAgentStore } from "./stores/in-memory-agent-store";
9
-
10
- const logger = createLogger("lobu");
11
-
12
- // ── Public Config Types ────────────────────────────────────────────────────
13
-
14
- export interface LobuAgentConfig {
15
- id: string;
16
- name?: string;
17
- description?: string;
18
- identity?: string;
19
- soul?: string;
20
- user?: string;
21
- providers?: Array<{ id: string; model?: string; key?: string }>;
22
- connections?: Array<{ type: string; [key: string]: string }>;
23
- skills?: string[];
24
- network?: { allowed?: string[]; denied?: string[] };
25
- nixPackages?: string[];
26
- }
27
-
28
- export interface LobuConfig {
29
- redis: string;
30
- memory?: string;
31
- agents?: LobuAgentConfig[];
32
- port?: number;
33
- deploymentMode?: "embedded" | "docker";
34
- /** Public URL of the gateway (used for OAuth callbacks). Defaults to http://localhost:{port}. */
35
- publicUrl?: string;
36
- /** Admin password for API auth. Auto-generated if not provided. */
37
- adminPassword?: string;
38
- }
39
-
40
- // ── Lobu Facade ────────────────────────────────────────────────────────────
41
-
42
- export class Lobu {
43
- private gateway: Gateway;
44
- private gatewayConfig: GatewayConfig;
45
- private agentConfigs: LobuAgentConfig[];
46
- private httpServer: Server | null = null;
47
- private chatInstanceManager: ChatInstanceManager | null = null;
48
- private initialized = false;
49
- private port: number;
50
-
51
- constructor(config: LobuConfig) {
52
- this.agentConfigs = config.agents ?? [];
53
- this.port = config.port ?? 8080;
54
-
55
- if (config.memory) {
56
- process.env.MEMORY_URL = config.memory;
57
- }
58
-
59
- // Set ADMIN_PASSWORD in env so gateway.ts picks it up.
60
- // Auto-generate one if not provided.
61
- if (config.adminPassword) {
62
- process.env.ADMIN_PASSWORD = config.adminPassword;
63
- } else if (!process.env.ADMIN_PASSWORD) {
64
- const crypto = require("node:crypto");
65
- process.env.ADMIN_PASSWORD = crypto.randomBytes(16).toString("base64url");
66
- }
67
-
68
- const defaultPublicUrl = `http://localhost:${this.port}`;
69
-
70
- // Convert LobuConfig -> GatewayConfig via buildGatewayConfig overrides
71
- this.gatewayConfig = buildGatewayConfig({
72
- queues: { connectionString: config.redis },
73
- orchestration: {
74
- deploymentMode: config.deploymentMode ?? "embedded",
75
- },
76
- mcp: {
77
- publicGatewayUrl: config.publicUrl ?? defaultPublicUrl,
78
- },
79
- });
80
-
81
- // Build InMemoryAgentStore pre-populated with agents from config
82
- const store = new InMemoryAgentStore();
83
- this.gateway = new Gateway(this.gatewayConfig, {
84
- configStore: store,
85
- connectionStore: store,
86
- accessStore: store,
87
- });
88
-
89
- // Register API platform (always enabled)
90
- this.gateway.registerPlatform(new ApiPlatform());
91
- }
92
-
93
- /**
94
- * Initialize services, populate agents, and seed connections without starting
95
- * an HTTP server. Call this when embedding Lobu in an existing server, then
96
- * use `getApp()` to get the Hono handler.
97
- *
98
- * Safe to call multiple times — initialization runs only once.
99
- */
100
- async initialize(): Promise<void> {
101
- if (this.initialized) return;
102
-
103
- // Start the gateway (initializes CoreServices, platforms, consumer)
104
- await this.gateway.start();
105
-
106
- const coreServices = this.gateway.getCoreServices();
107
-
108
- // Populate agents from config into the InMemoryAgentStore
109
- await this.populateAgents();
110
-
111
- // Initialize Chat SDK connection manager for platform connections
112
- this.chatInstanceManager = new ChatInstanceManager();
113
- try {
114
- await this.chatInstanceManager.initialize(coreServices);
115
-
116
- // Register chat platform adapters
117
- for (const adapter of this.chatInstanceManager.createPlatformAdapters()) {
118
- this.gateway.registerPlatform(adapter);
119
- }
120
-
121
- // Seed connections from agent configs
122
- await this.seedConnections();
123
-
124
- // Wire ChatResponseBridge into unified thread consumer
125
- const unifiedConsumer = this.gateway.getUnifiedConsumer();
126
- if (unifiedConsumer) {
127
- const chatResponseBridge = new ChatResponseBridge(
128
- this.chatInstanceManager
129
- );
130
- unifiedConsumer.setChatResponseBridge(chatResponseBridge);
131
- }
132
- } catch (error) {
133
- logger.warn(
134
- { error: String(error) },
135
- "ChatInstanceManager initialization failed — connections feature disabled"
136
- );
137
- }
138
-
139
- this.initialized = true;
140
- logger.info("Lobu gateway initialized");
141
- }
142
-
143
- /**
144
- * Start the gateway and an HTTP server. Shorthand for `initialize()` + bind
145
- * to a port. Use `initialize()` + `getApp()` instead when mounting Lobu in
146
- * an existing framework.
147
- */
148
- async start(): Promise<void> {
149
- await this.initialize();
150
-
151
- const app = this.getApp();
152
- this.httpServer = startGatewayServer(app, this.port);
153
-
154
- logger.info(`Lobu gateway running on port ${this.port}`);
155
- }
156
-
157
- /**
158
- * Stop the gateway and HTTP server.
159
- */
160
- async stop(): Promise<void> {
161
- if (this.chatInstanceManager) {
162
- await this.chatInstanceManager.shutdown();
163
- }
164
- await this.gateway.stop();
165
- if (this.httpServer) {
166
- this.httpServer.close();
167
- this.httpServer = null;
168
- }
169
- logger.info("Lobu gateway stopped");
170
- }
171
-
172
- /**
173
- * Get the Hono app for mounting in another server (no HTTP server started).
174
- * Call this instead of start() when embedding in an existing HTTP server.
175
- */
176
- getApp() {
177
- const coreServices = this.gateway.getCoreServices();
178
- return createGatewayApp({
179
- secretProxy: coreServices.getSecretProxy(),
180
- workerGateway: coreServices.getWorkerGateway(),
181
- mcpProxy: coreServices.getMcpProxy(),
182
- interactionService: coreServices.getInteractionService(),
183
- platformRegistry: this.gateway.getPlatformRegistry(),
184
- coreServices,
185
- chatInstanceManager: this.chatInstanceManager,
186
- });
187
- }
188
-
189
- // ── Private Helpers ──────────────────────────────────────────────────────
190
-
191
- /**
192
- * Populate InMemoryAgentStore + seed credentials from LobuAgentConfig[].
193
- * Mirrors what CoreServices.populateStoreFromFiles() does for file-loaded agents.
194
- */
195
- private async populateAgents(): Promise<void> {
196
- if (this.agentConfigs.length === 0) return;
197
-
198
- const coreServices = this.gateway.getCoreServices();
199
- const configStore = coreServices.getConfigStore();
200
- if (!configStore) return;
201
-
202
- const store = configStore as InMemoryAgentStore;
203
- const authProfilesManager = coreServices.getAuthProfilesManager();
204
-
205
- for (const agent of this.agentConfigs) {
206
- // Save metadata
207
- await store.saveMetadata(agent.id, {
208
- agentId: agent.id,
209
- name: agent.name ?? agent.id,
210
- description: agent.description,
211
- owner: { platform: "system", userId: "sdk" },
212
- createdAt: Date.now(),
213
- });
214
-
215
- // Build settings (same shape as FileLoadedAgent.settings)
216
- const settings: Record<string, any> = {};
217
-
218
- if (agent.identity) settings.identityMd = agent.identity;
219
- if (agent.soul) settings.soulMd = agent.soul;
220
- if (agent.user) settings.userMd = agent.user;
221
-
222
- if (agent.providers?.length) {
223
- settings.installedProviders = agent.providers.map((p) => ({
224
- providerId: p.id,
225
- installedAt: Date.now(),
226
- }));
227
- settings.modelSelection = { mode: "auto" };
228
- const providerModelPreferences = Object.fromEntries(
229
- agent.providers
230
- .filter((p) => !!p.model?.trim())
231
- .map((p) => [p.id, p.model!.trim()])
232
- );
233
- if (Object.keys(providerModelPreferences).length > 0) {
234
- settings.providerModelPreferences = providerModelPreferences;
235
- }
236
- }
237
-
238
- if (agent.network) {
239
- settings.networkConfig = {
240
- allowedDomains: agent.network.allowed,
241
- deniedDomains: agent.network.denied,
242
- };
243
- }
244
-
245
- if (agent.nixPackages?.length) {
246
- settings.nixConfig = { packages: agent.nixPackages };
247
- }
248
-
249
- await store.saveSettings(agent.id, {
250
- ...settings,
251
- updatedAt: Date.now(),
252
- } as any);
253
-
254
- // Seed provider credentials
255
- if (authProfilesManager && agent.providers) {
256
- for (const provider of agent.providers) {
257
- if (!provider.key) continue;
258
- await authProfilesManager.upsertProfile({
259
- agentId: agent.id,
260
- provider: provider.id,
261
- credential: provider.key,
262
- authType: "api-key",
263
- label: `${provider.id} (from SDK config)`,
264
- makePrimary: true,
265
- });
266
- }
267
- }
268
- }
269
-
270
- logger.info(
271
- `Populated ${this.agentConfigs.length} agent(s) from SDK config`
272
- );
273
- }
274
-
275
- /**
276
- * Seed platform connections from agent configs.
277
- * Mirrors the file-loaded agent connection seeding in gateway.ts startGateway().
278
- */
279
- private async seedConnections(): Promise<void> {
280
- if (!this.chatInstanceManager) return;
281
-
282
- for (const agent of this.agentConfigs) {
283
- if (!agent.connections?.length) continue;
284
-
285
- for (const conn of agent.connections) {
286
- const { type, ...configFields } = conn;
287
-
288
- const existing = await this.chatInstanceManager.listConnections({
289
- platform: type,
290
- templateAgentId: agent.id,
291
- });
292
- if (existing.length > 0) continue;
293
-
294
- try {
295
- await this.chatInstanceManager.addConnection(
296
- type,
297
- agent.id,
298
- { platform: type as any, ...configFields },
299
- { allowGroups: true }
300
- );
301
- logger.debug(`Created ${type} connection for agent "${agent.id}"`);
302
- } catch (err) {
303
- logger.error(
304
- `Failed to create ${type} connection for agent "${agent.id}"`,
305
- { error: err instanceof Error ? err.message : String(err) }
306
- );
307
- }
308
- }
309
- }
310
- }
311
- }
@@ -1,159 +0,0 @@
1
- /**
2
- * Simple Prometheus metrics exporter (no external dependencies)
3
- * Exposes basic gateway metrics in Prometheus text format
4
- */
5
-
6
- import { createLogger } from "@lobu/core";
7
-
8
- const logger = createLogger("metrics");
9
-
10
- interface MetricValue {
11
- value: number;
12
- labels: Record<string, string>;
13
- }
14
-
15
- interface Metric {
16
- name: string;
17
- help: string;
18
- type: "counter" | "gauge" | "histogram";
19
- values: MetricValue[];
20
- }
21
-
22
- const metrics: Map<string, Metric> = new Map();
23
-
24
- function initializeMetrics() {
25
- registerMetric(
26
- "lobu_worker_deployments_total",
27
- "Total number of worker deployments created",
28
- "counter"
29
- );
30
- registerMetric(
31
- "lobu_worker_deployments_failed_total",
32
- "Total number of failed worker deployments",
33
- "counter"
34
- );
35
- registerMetric(
36
- "lobu_worker_deployments_active",
37
- "Current number of active worker deployments",
38
- "gauge"
39
- );
40
- registerMetric(
41
- "lobu_messages_received_total",
42
- "Total number of messages received",
43
- "counter"
44
- );
45
- registerMetric(
46
- "lobu_messages_processed_total",
47
- "Total number of messages processed",
48
- "counter"
49
- );
50
- registerMetric("lobu_queue_length", "Current message queue length", "gauge");
51
- registerMetric(
52
- "lobu_pvc_created_total",
53
- "Total number of PVCs created",
54
- "counter"
55
- );
56
- registerMetric(
57
- "lobu_pvc_deleted_total",
58
- "Total number of PVCs deleted",
59
- "counter"
60
- );
61
- registerMetric(
62
- "lobu_pvc_cleanup_failed_total",
63
- "Total number of failed PVC cleanup operations",
64
- "counter"
65
- );
66
- registerMetric(
67
- "lobu_redis_connection_errors_total",
68
- "Total number of Redis connection errors",
69
- "counter"
70
- );
71
- registerMetric(
72
- "lobu_proxy_requests_total",
73
- "Total number of HTTP proxy requests",
74
- "counter"
75
- );
76
- registerMetric(
77
- "lobu_proxy_requests_blocked_total",
78
- "Total number of blocked proxy requests",
79
- "counter"
80
- );
81
- registerMetric(
82
- "lobu_process_start_time_seconds",
83
- "Start time of the process since unix epoch in seconds",
84
- "gauge"
85
- );
86
-
87
- setGauge("lobu_process_start_time_seconds", Math.floor(Date.now() / 1000));
88
- logger.info("Prometheus metrics initialized");
89
- }
90
-
91
- function registerMetric(
92
- name: string,
93
- help: string,
94
- type: "counter" | "gauge" | "histogram"
95
- ) {
96
- metrics.set(name, { name, help, type, values: [] });
97
- }
98
-
99
- function setGauge(
100
- name: string,
101
- value: number,
102
- labels: Record<string, string> = {}
103
- ) {
104
- const metric = metrics.get(name);
105
- if (!metric || metric.type !== "gauge") {
106
- logger.warn(`Gauge metric ${name} not found`);
107
- return;
108
- }
109
-
110
- const labelKey = JSON.stringify(labels);
111
- const existing = metric.values.find(
112
- (entry) => JSON.stringify(entry.labels) === labelKey
113
- );
114
- if (existing) {
115
- existing.value = value;
116
- } else {
117
- metric.values.push({ value, labels });
118
- }
119
- }
120
-
121
- export function getMetricsText(): string {
122
- const lines: string[] = [];
123
-
124
- for (const metric of metrics.values()) {
125
- lines.push(`# HELP ${metric.name} ${metric.help}`);
126
- lines.push(`# TYPE ${metric.name} ${metric.type}`);
127
-
128
- if (metric.values.length === 0) {
129
- lines.push(`${metric.name} 0`);
130
- continue;
131
- }
132
-
133
- for (const { value, labels } of metric.values) {
134
- const labelStr = Object.entries(labels)
135
- .map(([key, labelValue]) => `${key}="${labelValue}"`)
136
- .join(",");
137
- if (labelStr) {
138
- lines.push(`${metric.name}{${labelStr}} ${value}`);
139
- } else {
140
- lines.push(`${metric.name} ${value}`);
141
- }
142
- }
143
- }
144
-
145
- const memUsage = process.memoryUsage();
146
- lines.push("# HELP nodejs_heap_size_bytes Node.js heap size in bytes");
147
- lines.push("# TYPE nodejs_heap_size_bytes gauge");
148
- lines.push(`nodejs_heap_size_bytes{type="used"} ${memUsage.heapUsed}`);
149
- lines.push(`nodejs_heap_size_bytes{type="total"} ${memUsage.heapTotal}`);
150
- lines.push(
151
- "# HELP nodejs_external_memory_bytes Node.js external memory in bytes"
152
- );
153
- lines.push("# TYPE nodejs_external_memory_bytes gauge");
154
- lines.push(`nodejs_external_memory_bytes ${memUsage.external}`);
155
-
156
- return `${lines.join("\n")}\n`;
157
- }
158
-
159
- initializeMetrics();
@@ -1,179 +0,0 @@
1
- import type { CliBackendConfig } from "@lobu/core";
2
- import {
3
- type ActionButton,
4
- type ModuleInterface,
5
- type ModuleSessionContext,
6
- moduleRegistry,
7
- type WorkerContext,
8
- type WorkerModule,
9
- } from "@lobu/core";
10
-
11
- export interface ModelOption {
12
- value: string;
13
- label: string;
14
- }
15
-
16
- export interface OrchestratorModule<TModuleData = unknown>
17
- extends ModuleInterface<TModuleData> {
18
- buildEnvVars(
19
- agentId: string,
20
- baseEnv: Record<string, string>
21
- ): Promise<Record<string, string>>;
22
- getContainerAddress(): string;
23
- }
24
-
25
- export interface ProviderUpstreamConfig {
26
- slug: string;
27
- upstreamBaseUrl: string;
28
- }
29
-
30
- export interface ModelProviderModule extends OrchestratorModule {
31
- providerId: string;
32
- providerDisplayName: string;
33
- providerIconUrl?: string;
34
- authType?: "oauth" | "device-code" | "api-key";
35
- supportedAuthTypes?: ("oauth" | "device-code" | "api-key")[];
36
- apiKeyInstructions?: string;
37
- apiKeyPlaceholder?: string;
38
- catalogDescription?: string;
39
- catalogVisible?: boolean;
40
- getSecretEnvVarNames(): string[];
41
- getCredentialEnvVarName(): string;
42
- getUpstreamConfig?(): ProviderUpstreamConfig | null;
43
- hasCredentials(agentId: string): Promise<boolean>;
44
- hasSystemKey(): boolean;
45
- getProxyBaseUrlMappings(
46
- proxyUrl: string,
47
- agentId?: string
48
- ): Record<string, string>;
49
- injectSystemKeyFallback(
50
- envVars: Record<string, string>
51
- ): Record<string, string>;
52
- getApp?(): any;
53
- getModelOptions?(agentId: string, userId: string): Promise<ModelOption[]>;
54
- getCliBackendConfig?(): CliBackendConfig | null;
55
- buildCredentialPlaceholder?(agentId: string): Promise<string> | string;
56
- startDeviceCode?(agentId: string): Promise<{
57
- userCode: string;
58
- deviceAuthId: string;
59
- interval: number;
60
- verificationUrl?: string;
61
- }>;
62
- pollDeviceCode?(
63
- agentId: string,
64
- payload: { deviceAuthId: string; userCode: string }
65
- ): Promise<{
66
- status: "pending" | "success";
67
- error?: string;
68
- accountId?: string;
69
- }>;
70
- }
71
-
72
- export interface DispatcherContext<TModuleData = unknown> {
73
- userId: string;
74
- channelId: string;
75
- threadTs: string;
76
- platformClient?: unknown;
77
- moduleData: TModuleData;
78
- }
79
-
80
- export interface DispatcherModule<TModuleData = unknown>
81
- extends ModuleInterface<TModuleData> {
82
- generateActionButtons(
83
- context: DispatcherContext<TModuleData>
84
- ): Promise<ActionButton[]>;
85
- handleAction(
86
- actionId: string,
87
- userId: string,
88
- agentId: string,
89
- context: any
90
- ): Promise<boolean>;
91
- }
92
-
93
- export interface DispatcherModuleSource {
94
- getDispatcherModules(): DispatcherModule[];
95
- }
96
-
97
- export abstract class BaseModule<TModuleData = unknown>
98
- implements
99
- WorkerModule<TModuleData>,
100
- DispatcherModule<TModuleData>,
101
- OrchestratorModule<TModuleData>
102
- {
103
- abstract name: string;
104
- abstract isEnabled(): boolean;
105
-
106
- async init(): Promise<void> {
107
- // no-op
108
- }
109
-
110
- registerEndpoints(_app: any): void {
111
- // no-op
112
- }
113
-
114
- async initWorkspace(_config: any): Promise<void> {
115
- // no-op
116
- }
117
-
118
- async onSessionStart(
119
- context: ModuleSessionContext
120
- ): Promise<ModuleSessionContext> {
121
- return context;
122
- }
123
-
124
- async onSessionEnd(_context: ModuleSessionContext): Promise<ActionButton[]> {
125
- return [];
126
- }
127
-
128
- async onBeforeResponse(_context: WorkerContext): Promise<TModuleData | null> {
129
- return null;
130
- }
131
-
132
- async buildEnvVars(
133
- _agentId: string,
134
- baseEnv: Record<string, string>
135
- ): Promise<Record<string, string>> {
136
- return baseEnv;
137
- }
138
-
139
- async getModelOptions(
140
- _agentId: string,
141
- _userId: string
142
- ): Promise<ModelOption[]> {
143
- return [];
144
- }
145
-
146
- getContainerAddress(): string {
147
- return "";
148
- }
149
-
150
- async generateActionButtons(
151
- _context: DispatcherContext<TModuleData>
152
- ): Promise<ActionButton[]> {
153
- return [];
154
- }
155
-
156
- async handleAction(
157
- _actionId: string,
158
- _userId: string,
159
- _agentId: string,
160
- _context: any
161
- ): Promise<boolean> {
162
- return false;
163
- }
164
- }
165
-
166
- export function getOrchestratorModules(): OrchestratorModule[] {
167
- return moduleRegistry
168
- .getModules()
169
- .filter((m): m is OrchestratorModule => "buildEnvVars" in m);
170
- }
171
-
172
- export function getModelProviderModules(): ModelProviderModule[] {
173
- return moduleRegistry
174
- .getModules()
175
- .filter(
176
- (m): m is ModelProviderModule =>
177
- "providerId" in m && "getSecretEnvVarNames" in m
178
- );
179
- }