@zmeel/server 0.3.2 → 0.3.4

package/dist/services/heartbeat.js

@@ -15,6 +15,7 @@ import { costService } from "./costs.js";
 import { trackAgentFirstHeartbeat } from "@zmeel/shared/telemetry";
 import { getTelemetryClient } from "../telemetry.js";
 import { companySkillService } from "./company-skills.js";
+import { injectManagedKeyIfNeeded } from "./managed-api-keys.js";
 import { budgetService } from "./budgets.js";
 import { secretService } from "./secrets.js";
 import { resolveDefaultAgentWorkspaceDir, resolveManagedProjectWorkspaceDir } from "../home-paths.js";
@@ -2181,11 +2182,48 @@ export function heartbeatService(db) {
                         adapterType: agent.adapterType,
                     }, "local agent jwt secret missing or invalid; running without injected ZMEEL_API_KEY");
                 }
+                // Inject managed API key for cloud adapter
+                let configWithKey = injectManagedKeyIfNeeded(agent.adapterType ?? "", runtimeConfig);
+                // Inject employee brain context and file updater for cloud agents
+                if (agent.adapterType === "cloud") {
+                    const { employeeFileService } = await import("./employee-files.js");
+                    const fileService = employeeFileService(db);
+                    const brainContext = await fileService.getFullContext(agent.id);
+                    if (brainContext) {
+                        // Override the system prompt with the full brain context
+                        configWithKey = { ...configWithKey, systemPrompt: brainContext };
+                    }
+                    configWithKey = {
+                        ...configWithKey,
+                        _fileUpdater: async (fileType, content) => {
+                            await fileService.appendToFile(agent.id, fileType, content, "agent");
+                        },
+                    };
+                }
+                // Inject Composio tools for cloud agents in SaaS mode
+                if (agent.adapterType === "cloud") {
+                    const composioApps = configWithKey.composioApps ?? [];
+                    if (composioApps.length > 0) {
+                        try {
+                            const composio = await import("./composio.js");
+                            const entityId = agent.companyId;
+                            const tools = await composio.getToolsForApps(entityId, composioApps);
+                            configWithKey = {
+                                ...configWithKey,
+                                tools,
+                                _toolExecutor: (name, input) => composio.executeToolCall(entityId, name, input),
+                            };
+                        }
+                        catch (err) {
+                            // Composio not configured — continue without tools
+                        }
+                    }
+                }
                 const adapterResult = await adapter.execute({
                     runId: run.id,
                     agent,
                     runtime: runtimeForAdapter,
-                    config: runtimeConfig,
+                    config: configWithKey,
                     context,
                     onLog,
                     onMeta: onAdapterMeta,