open-managed-agents 0.1.0

package/dist/src/control-plane/events/runtime-helpers.js

@@ -0,0 +1,134 @@
+import { isJsonObject } from "../../types/json.js";
+import { ApiError } from "../errors.js";
+export function leaseExpiresAt(now, ttlMs) {
+    return new Date(Date.parse(now) + ttlMs).toISOString();
+}
+export function isRuntimeLeaseExpired(leaseExpiresAtValue) {
+    return Date.parse(leaseExpiresAtValue) <= Date.now();
+}
+export function runtimeLeaseRetryDelayMs(leaseExpiresAtValue) {
+    return Math.max(0, Date.parse(leaseExpiresAtValue) - Date.now());
+}
+export function isRuntimeTurnClosed(state) {
+    return state === "completed" || state === "terminalized";
+}
+export function actionClosedWithoutResult(action) {
+    return (action?.close_reason === "interrupted" ||
+        action?.close_reason === "timeout" ||
+        action?.close_reason === "terminalized");
+}
+export function isAcknowledgedInFlightAction(action) {
+    return (action?.state === "acknowledged" &&
+        !isRuntimeTurnClosed(action.turn.state) &&
+        action.close_reason === null);
+}
+export function unique(values) {
+    return [...new Set(values)];
+}
+export function runtimeTurnStillOwned(turnId) {
+    return new ApiError(529, "overloaded_error", `Runtime turn ${turnId} is still owned by another worker; retry later`);
+}
+export function textFromContent(content) {
+    const text = content
+        .filter((block) => block.type === "text")
+        .map((block) => block.text)
+        .join("\n")
+        .trim();
+    return text.length > 0 ? text : undefined;
+}
+export function runtimeErrorDraft(error) {
+    const message = "Runtime execution failed";
+    return {
+        type: "session.error",
+        payload: { message },
+    };
+}
+export function isRuntimeCustomToolUseEvent(event) {
+    if (!isObjectRecord(event))
+        return false;
+    return (event.type === "oma.custom_tool_use" &&
+        typeof event.piToolCallId === "string" &&
+        typeof event.name === "string" &&
+        isJsonObject(event.input) &&
+        typeof event.bindCustomToolUseId === "function" &&
+        typeof event.rejectCustomToolUse === "function");
+}
+export function isRuntimeToolPermissionUseEvent(event) {
+    if (!isObjectRecord(event))
+        return false;
+    return (event.type === "oma.tool_permission_use" &&
+        typeof event.piToolCallId === "string" &&
+        typeof event.name === "string" &&
+        isJsonObject(event.input) &&
+        (event.evaluatedPermission === "allow" ||
+            event.evaluatedPermission === "ask" ||
+            event.evaluatedPermission === "deny") &&
+        typeof event.bindToolUseId === "function" &&
+        typeof event.rejectToolUse === "function");
+}
+export function isRuntimeToolPermissionWithModelEndEvent(event) {
+    if (!isObjectRecord(event))
+        return false;
+    return (event.type === "oma.tool_permission_with_model_end" &&
+        "messageEnd" in event &&
+        isRuntimeToolPermissionUseEvent(event.permissionUse) &&
+        Array.isArray(event.suppressedPiToolCallIds) &&
+        event.suppressedPiToolCallIds.every((id) => typeof id === "string"));
+}
+export function isRuntimeMcpToolUseEvent(event) {
+    if (!isObjectRecord(event))
+        return false;
+    return (event.type === "oma.mcp_tool_use" &&
+        typeof event.piToolCallId === "string" &&
+        typeof event.mcpServerName === "string" &&
+        typeof event.name === "string" &&
+        isJsonObject(event.input) &&
+        (event.evaluatedPermission === "allow" ||
+            event.evaluatedPermission === "ask" ||
+            event.evaluatedPermission === "deny") &&
+        typeof event.bindToolUseId === "function" &&
+        typeof event.rejectToolUse === "function");
+}
+export function isRuntimeMcpToolWithModelEndEvent(event) {
+    if (!isObjectRecord(event))
+        return false;
+    return (event.type === "oma.mcp_tool_with_model_end" &&
+        "messageEnd" in event &&
+        isRuntimeMcpToolUseEvent(event.mcpToolUse) &&
+        Array.isArray(event.suppressedPiToolCallIds) &&
+        event.suppressedPiToolCallIds.every((id) => typeof id === "string"));
+}
+export function isRuntimeMcpToolResultEvent(event) {
+    if (!isObjectRecord(event))
+        return false;
+    return (event.type === "oma.mcp_tool_result" &&
+        typeof event.mcpToolUseId === "string" &&
+        Array.isArray(event.content) &&
+        typeof event.isError === "boolean");
+}
+export function isRuntimeMcpConnectionFailedEvent(event) {
+    if (!isObjectRecord(event))
+        return false;
+    return (event.type === "oma.mcp_connection_failed" &&
+        typeof event.mcpServerName === "string" &&
+        typeof event.message === "string" &&
+        (event.retryStatus === "retrying" ||
+            event.retryStatus === "exhausted" ||
+            event.retryStatus === "terminal"));
+}
+export function hasTerminalIdleDraft(drafts) {
+    return drafts.some((draft) => {
+        if (draft.type !== "session.status_idle")
+            return false;
+        const stopReason = draft.payload.stop_reason;
+        if (!isJsonObject(stopReason))
+            return true;
+        return stopReason.type !== "requires_action";
+    });
+}
+export function toError(error) {
+    return error instanceof Error ? error : new Error(String(error));
+}
+function isObjectRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}