@lobu/gateway 3.0.9 → 3.0.13

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 (212) 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/gateway/index.js +1 -1
  19. package/dist/gateway/index.js.map +1 -1
  20. package/dist/interactions.d.ts +9 -43
  21. package/dist/interactions.d.ts.map +1 -1
  22. package/dist/interactions.js +10 -52
  23. package/dist/interactions.js.map +1 -1
  24. package/dist/routes/public/agent.d.ts +4 -0
  25. package/dist/routes/public/agent.d.ts.map +1 -1
  26. package/dist/routes/public/agent.js +21 -0
  27. package/dist/routes/public/agent.js.map +1 -1
  28. package/dist/services/core-services.d.ts.map +1 -1
  29. package/dist/services/core-services.js +4 -0
  30. package/dist/services/core-services.js.map +1 -1
  31. package/package.json +9 -9
  32. package/src/__tests__/agent-config-routes.test.ts +0 -254
  33. package/src/__tests__/agent-history-routes.test.ts +0 -72
  34. package/src/__tests__/agent-routes.test.ts +0 -68
  35. package/src/__tests__/agent-schedules-routes.test.ts +0 -59
  36. package/src/__tests__/agent-settings-store.test.ts +0 -323
  37. package/src/__tests__/bedrock-model-catalog.test.ts +0 -40
  38. package/src/__tests__/bedrock-openai-service.test.ts +0 -157
  39. package/src/__tests__/bedrock-provider-module.test.ts +0 -56
  40. package/src/__tests__/chat-instance-manager-slack.test.ts +0 -204
  41. package/src/__tests__/chat-response-bridge.test.ts +0 -131
  42. package/src/__tests__/config-memory-plugins.test.ts +0 -92
  43. package/src/__tests__/config-request-store.test.ts +0 -127
  44. package/src/__tests__/connection-routes.test.ts +0 -144
  45. package/src/__tests__/core-services-store-selection.test.ts +0 -92
  46. package/src/__tests__/docker-deployment.test.ts +0 -1211
  47. package/src/__tests__/embedded-deployment.test.ts +0 -342
  48. package/src/__tests__/grant-store.test.ts +0 -148
  49. package/src/__tests__/http-proxy.test.ts +0 -281
  50. package/src/__tests__/instruction-service.test.ts +0 -37
  51. package/src/__tests__/link-buttons.test.ts +0 -112
  52. package/src/__tests__/lobu.test.ts +0 -32
  53. package/src/__tests__/mcp-config-service.test.ts +0 -347
  54. package/src/__tests__/mcp-proxy.test.ts +0 -694
  55. package/src/__tests__/message-handler-bridge.test.ts +0 -17
  56. package/src/__tests__/model-selection.test.ts +0 -172
  57. package/src/__tests__/oauth-templates.test.ts +0 -39
  58. package/src/__tests__/platform-adapter-slack-send.test.ts +0 -114
  59. package/src/__tests__/platform-helpers-model-resolution.test.ts +0 -253
  60. package/src/__tests__/provider-inheritance.test.ts +0 -212
  61. package/src/__tests__/routes/cli-auth.test.ts +0 -337
  62. package/src/__tests__/routes/interactions.test.ts +0 -121
  63. package/src/__tests__/secret-proxy.test.ts +0 -85
  64. package/src/__tests__/session-manager.test.ts +0 -572
  65. package/src/__tests__/setup.ts +0 -133
  66. package/src/__tests__/skill-and-mcp-registry.test.ts +0 -203
  67. package/src/__tests__/slack-routes.test.ts +0 -161
  68. package/src/__tests__/system-config-resolver.test.ts +0 -75
  69. package/src/__tests__/system-message-limiter.test.ts +0 -89
  70. package/src/__tests__/system-skills-service.test.ts +0 -362
  71. package/src/__tests__/transcription-service.test.ts +0 -222
  72. package/src/__tests__/utils/rate-limiter.test.ts +0 -102
  73. package/src/__tests__/worker-connection-manager.test.ts +0 -497
  74. package/src/__tests__/worker-job-router.test.ts +0 -722
  75. package/src/api/index.ts +0 -1
  76. package/src/api/platform.ts +0 -292
  77. package/src/api/response-renderer.ts +0 -157
  78. package/src/auth/agent-metadata-store.ts +0 -168
  79. package/src/auth/api-auth-middleware.ts +0 -69
  80. package/src/auth/api-key-provider-module.ts +0 -213
  81. package/src/auth/base-provider-module.ts +0 -201
  82. package/src/auth/bedrock/provider-module.ts +0 -110
  83. package/src/auth/chatgpt/chatgpt-oauth-module.ts +0 -185
  84. package/src/auth/chatgpt/device-code-client.ts +0 -218
  85. package/src/auth/chatgpt/index.ts +0 -1
  86. package/src/auth/claude/oauth-module.ts +0 -280
  87. package/src/auth/cli/token-service.ts +0 -249
  88. package/src/auth/external/client.ts +0 -560
  89. package/src/auth/external/device-code-client.ts +0 -235
  90. package/src/auth/mcp/config-service.ts +0 -420
  91. package/src/auth/mcp/proxy.ts +0 -1086
  92. package/src/auth/mcp/string-substitution.ts +0 -17
  93. package/src/auth/mcp/tool-cache.ts +0 -90
  94. package/src/auth/oauth/base-client.ts +0 -267
  95. package/src/auth/oauth/client.ts +0 -153
  96. package/src/auth/oauth/credentials.ts +0 -7
  97. package/src/auth/oauth/providers.ts +0 -69
  98. package/src/auth/oauth/state-store.ts +0 -150
  99. package/src/auth/oauth-templates.ts +0 -179
  100. package/src/auth/provider-catalog.ts +0 -220
  101. package/src/auth/provider-model-options.ts +0 -41
  102. package/src/auth/settings/agent-settings-store.ts +0 -565
  103. package/src/auth/settings/auth-profiles-manager.ts +0 -216
  104. package/src/auth/settings/index.ts +0 -12
  105. package/src/auth/settings/model-preference-store.ts +0 -52
  106. package/src/auth/settings/model-selection.ts +0 -135
  107. package/src/auth/settings/resolved-settings-view.ts +0 -298
  108. package/src/auth/settings/template-utils.ts +0 -44
  109. package/src/auth/settings/token-service.ts +0 -88
  110. package/src/auth/system-env-store.ts +0 -98
  111. package/src/auth/user-agents-store.ts +0 -68
  112. package/src/channels/binding-service.ts +0 -214
  113. package/src/channels/index.ts +0 -4
  114. package/src/cli/gateway.ts +0 -1312
  115. package/src/cli/index.ts +0 -74
  116. package/src/commands/built-in-commands.ts +0 -80
  117. package/src/commands/command-dispatcher.ts +0 -94
  118. package/src/commands/command-reply-adapters.ts +0 -27
  119. package/src/config/file-loader.ts +0 -618
  120. package/src/config/index.ts +0 -588
  121. package/src/config/network-allowlist.ts +0 -71
  122. package/src/connections/chat-instance-manager.ts +0 -1284
  123. package/src/connections/chat-response-bridge.ts +0 -618
  124. package/src/connections/index.ts +0 -7
  125. package/src/connections/interaction-bridge.ts +0 -831
  126. package/src/connections/message-handler-bridge.ts +0 -440
  127. package/src/connections/platform-auth-methods.ts +0 -15
  128. package/src/connections/types.ts +0 -84
  129. package/src/gateway/connection-manager.ts +0 -291
  130. package/src/gateway/index.ts +0 -698
  131. package/src/gateway/job-router.ts +0 -201
  132. package/src/gateway-main.ts +0 -200
  133. package/src/index.ts +0 -41
  134. package/src/infrastructure/queue/index.ts +0 -12
  135. package/src/infrastructure/queue/queue-producer.ts +0 -148
  136. package/src/infrastructure/queue/redis-queue.ts +0 -361
  137. package/src/infrastructure/queue/types.ts +0 -133
  138. package/src/infrastructure/redis/system-message-limiter.ts +0 -94
  139. package/src/interactions/config-request-store.ts +0 -198
  140. package/src/interactions.ts +0 -363
  141. package/src/lobu.ts +0 -311
  142. package/src/metrics/prometheus.ts +0 -159
  143. package/src/modules/module-system.ts +0 -179
  144. package/src/orchestration/base-deployment-manager.ts +0 -900
  145. package/src/orchestration/deployment-utils.ts +0 -98
  146. package/src/orchestration/impl/docker-deployment.ts +0 -620
  147. package/src/orchestration/impl/embedded-deployment.ts +0 -268
  148. package/src/orchestration/impl/index.ts +0 -8
  149. package/src/orchestration/impl/k8s/deployment.ts +0 -1061
  150. package/src/orchestration/impl/k8s/helpers.ts +0 -610
  151. package/src/orchestration/impl/k8s/index.ts +0 -1
  152. package/src/orchestration/index.ts +0 -333
  153. package/src/orchestration/message-consumer.ts +0 -584
  154. package/src/orchestration/scheduled-wakeup.ts +0 -704
  155. package/src/permissions/approval-policy.ts +0 -36
  156. package/src/permissions/grant-store.ts +0 -219
  157. package/src/platform/file-handler.ts +0 -66
  158. package/src/platform/link-buttons.ts +0 -57
  159. package/src/platform/renderer-utils.ts +0 -44
  160. package/src/platform/response-renderer.ts +0 -84
  161. package/src/platform/unified-thread-consumer.ts +0 -194
  162. package/src/platform.ts +0 -318
  163. package/src/proxy/http-proxy.ts +0 -752
  164. package/src/proxy/proxy-manager.ts +0 -81
  165. package/src/proxy/secret-proxy.ts +0 -402
  166. package/src/proxy/token-refresh-job.ts +0 -143
  167. package/src/routes/internal/audio.ts +0 -141
  168. package/src/routes/internal/device-auth.ts +0 -652
  169. package/src/routes/internal/files.ts +0 -226
  170. package/src/routes/internal/history.ts +0 -69
  171. package/src/routes/internal/images.ts +0 -127
  172. package/src/routes/internal/interactions.ts +0 -84
  173. package/src/routes/internal/middleware.ts +0 -23
  174. package/src/routes/internal/schedule.ts +0 -226
  175. package/src/routes/internal/types.ts +0 -22
  176. package/src/routes/openapi-auto.ts +0 -239
  177. package/src/routes/public/agent-access.ts +0 -23
  178. package/src/routes/public/agent-config.ts +0 -675
  179. package/src/routes/public/agent-history.ts +0 -422
  180. package/src/routes/public/agent-schedules.ts +0 -296
  181. package/src/routes/public/agent.ts +0 -1086
  182. package/src/routes/public/agents.ts +0 -373
  183. package/src/routes/public/channels.ts +0 -191
  184. package/src/routes/public/cli-auth.ts +0 -896
  185. package/src/routes/public/connections.ts +0 -574
  186. package/src/routes/public/landing.ts +0 -16
  187. package/src/routes/public/oauth.ts +0 -147
  188. package/src/routes/public/settings-auth.ts +0 -104
  189. package/src/routes/public/slack.ts +0 -173
  190. package/src/routes/shared/agent-ownership.ts +0 -101
  191. package/src/routes/shared/token-verifier.ts +0 -34
  192. package/src/services/bedrock-model-catalog.ts +0 -217
  193. package/src/services/bedrock-openai-service.ts +0 -658
  194. package/src/services/core-services.ts +0 -1072
  195. package/src/services/image-generation-service.ts +0 -257
  196. package/src/services/instruction-service.ts +0 -318
  197. package/src/services/mcp-registry.ts +0 -94
  198. package/src/services/platform-helpers.ts +0 -287
  199. package/src/services/session-manager.ts +0 -262
  200. package/src/services/settings-resolver.ts +0 -74
  201. package/src/services/system-config-resolver.ts +0 -89
  202. package/src/services/system-skills-service.ts +0 -229
  203. package/src/services/transcription-service.ts +0 -684
  204. package/src/session.ts +0 -110
  205. package/src/spaces/index.ts +0 -1
  206. package/src/spaces/space-resolver.ts +0 -17
  207. package/src/stores/in-memory-agent-store.ts +0 -403
  208. package/src/stores/redis-agent-store.ts +0 -279
  209. package/src/utils/public-url.ts +0 -44
  210. package/src/utils/rate-limiter.ts +0 -94
  211. package/tsconfig.json +0 -33
  212. 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
- }