@centrali-io/centrali-mcp 4.4.4 → 4.4.6

package/dist/tools/compute.js

@@ -8,8 +8,12 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.registerComputeTools = registerComputeTools;
+const axios_1 = __importDefault(require("axios"));
 const zod_1 = require("zod");
 function formatError(error, context) {
     var _a, _b;
@@ -33,7 +37,22 @@ function formatError(error, context) {
     }
     return `Error ${context}: ${error instanceof Error ? error.message : String(error)}`;
 }
-function registerComputeTools(server, sdk) {
+/**
+ * Ensures the SDK has a valid token.
+ */
+function ensureToken(sdk) {
+    return __awaiter(this, void 0, void 0, function* () {
+        let token = sdk.getToken();
+        if (token)
+            return token;
+        try {
+            yield sdk.functions.list({ limit: 1 });
+        }
+        catch ( /* token refresh side effect */_a) { /* token refresh side effect */ }
+        return sdk.getToken();
+    });
+}
+function registerComputeTools(server, sdk, centraliUrl, workspaceId) {
     server.tool("list_functions", "List all compute functions in the workspace. Compute functions are JavaScript code blocks that run server-side.", {
         page: zod_1.z.number().optional().describe("Page number"),
         limit: zod_1.z.number().optional().describe("Results per page"),
@@ -112,7 +131,11 @@ function registerComputeTools(server, sdk) {
                 content: [
                     {
                         type: "text",
-                        text: JSON.stringify({ jobId: result.data, message: "Trigger invoked successfully" }, null, 2),
+                        text: JSON.stringify({
+                            jobId: result.data,
+                            message: "Trigger invoked successfully. Execution is async — the function is running in the background.",
+                            next_step: `To check the result, call get_compute_job_status with jobId='${result.data}'. It returns the job state (queued → running → completed | failed) and the return value or error.`,
+                        }, null, 2),
                     },
                 ],
             };
@@ -129,6 +152,37 @@ function registerComputeTools(server, sdk) {
             };
         }
     }));
+    server.tool("get_compute_job_status", "Check the status of an async compute job by job ID. Returns the current state (queued, running, completed, failed), the return value on success, or the failure reason on error. Use this after invoke_trigger to poll for results.", {
+        jobId: zod_1.z.string().describe("The job ID returned by invoke_trigger"),
+    }, (_a) => __awaiter(this, [_a], void 0, function* ({ jobId }) {
+        try {
+            const token = yield ensureToken(sdk);
+            const url = new URL(centraliUrl);
+            const hostname = url.hostname.startsWith("api.")
+                ? url.hostname
+                : `api.${url.hostname}`;
+            const apiUrl = `${url.protocol}//${hostname}/data/workspace/${workspaceId}/api/v1/jobs/compute/${jobId}`;
+            const result = yield axios_1.default.get(apiUrl, {
+                headers: token ? { Authorization: `Bearer ${token}` } : {},
+            });
+            return {
+                content: [
+                    { type: "text", text: JSON.stringify(result.data, null, 2) },
+                ],
+            };
+        }
+        catch (error) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: formatError(error, `getting job status for '${jobId}'`),
+                    },
+                ],
+                isError: true,
+            };
+        }
+    }));
     // ── Function CRUD tools ──────────────────────────────────────────
     server.tool("get_function", "Get a compute function by ID. Returns the full function definition including code.", {
         functionId: zod_1.z.string().describe("The compute function ID (UUID)"),
@@ -487,6 +541,81 @@ function registerComputeTools(server, sdk) {
             };
         }
     }));
+    // ── Function Runs tools ────────────────────────────────────────────
+    server.tool("get_function_run", "Get a function run by ID. Returns status, output, error, timing, and execution metadata. Use this to check the result of an async trigger invocation or test execution.", {
+        runId: zod_1.z.string().describe("The function run ID (UUID)"),
+    }, (_a) => __awaiter(this, [_a], void 0, function* ({ runId }) {
+        try {
+            const result = yield sdk.runs.get(runId);
+            return {
+                content: [
+                    { type: "text", text: JSON.stringify(result.data, null, 2) },
+                ],
+            };
+        }
+        catch (error) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: formatError(error, `getting function run '${runId}'`),
+                    },
+                ],
+                isError: true,
+            };
+        }
+    }));
+    server.tool("list_function_runs", "List function runs filtered by trigger ID or function ID. Returns execution history with status, timing, and errors. Useful for checking whether a trigger invocation completed and what it returned.", {
+        triggerId: zod_1.z.string().optional().describe("Filter runs by trigger ID (UUID)"),
+        functionId: zod_1.z.string().optional().describe("Filter runs by function ID (UUID)"),
+        status: zod_1.z
+            .enum(["pending", "running", "completed", "failure", "timeout"])
+            .optional()
+            .describe("Filter by run status"),
+        page: zod_1.z.number().optional().describe("Page number"),
+        limit: zod_1.z.number().optional().describe("Results per page"),
+    }, (_a) => __awaiter(this, [_a], void 0, function* ({ triggerId, functionId, status, page, limit }) {
+        try {
+            if (!triggerId && !functionId) {
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: "Error: provide either triggerId or functionId to list runs.",
+                        },
+                    ],
+                    isError: true,
+                };
+            }
+            const options = {};
+            if (status)
+                options.status = status;
+            if (page !== undefined)
+                options.page = page;
+            if (limit !== undefined)
+                options.limit = limit;
+            const opts = Object.keys(options).length > 0 ? options : undefined;
+            const result = triggerId
+                ? yield sdk.runs.listByTrigger(triggerId, opts)
+                : yield sdk.runs.listByFunction(functionId, opts);
+            return {
+                content: [
+                    { type: "text", text: JSON.stringify(result.data, null, 2) },
+                ],
+            };
+        }
+        catch (error) {
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: formatError(error, `listing function runs`),
+                    },
+                ],
+                isError: true,
+            };
+        }
+    }));
     // ── Allowed Domains tools ──────────────────────────────────────────
     server.tool("list_allowed_domains", "List all allowed domains for compute function HTTP requests. Functions can only call external APIs on domains in this allowlist.", {}, () => __awaiter(this, void 0, void 0, function* () {
         try {