@artinet/sdk 0.2.0

package/dist/transport/streaming/stream.js

@@ -0,0 +1,128 @@
+import { updateState } from "../../server/lib/state.js";
+import { FAILED_UPDATE, INTERNAL_ERROR } from "../../utils/common/errors.js";
+import { isArtifactUpdate, isTaskStatusUpdate, } from "../../utils/common/utils.js";
+import { logError } from "../../utils/logging/log.js";
+import { FINAL_STATES } from "../../utils/common/constants.js";
+/**
+ * Sets up a Server-Sent Events stream with appropriate headers
+ * and initializes tracking for the stream.
+ *
+ * @param res The Express Response object
+ * @param taskId The task ID to associate with this stream
+ * @param reqId The request ID for acknowledgment
+ * @param initialStatus Optional initial status to send
+ * @param addStreamMethod Function to add stream to tracking
+ * @param sendSseMethod Function to send SSE events
+ */
+export function setupSseStream(res, taskId, initialStatus, addStreamMethod) {
+    // Set up SSE stream headers
+    res.setHeader("Content-Type", "text/event-stream");
+    res.setHeader("Cache-Control", "no-cache");
+    res.setHeader("Connection", "keep-alive");
+    // Track this stream for potential cancellation if tracking method provided
+    if (addStreamMethod) {
+        addStreamMethod(taskId, res);
+    }
+    // Send initial status if provided
+    if (initialStatus) {
+        sendSSEEvent(res, initialStatus);
+    }
+}
+/**
+ * Sends a Server-Sent Event with task data.
+ * @param res The Express Response object
+ * @param reqId The request ID
+ * @param eventData The event data to send
+ */
+export function sendSSEEvent(res, update) {
+    if (!res.writable) {
+        return;
+    }
+    const response = {
+        jsonrpc: "2.0",
+        id: update.id,
+        result: update,
+    };
+    res.write(`data: ${JSON.stringify(response)}\n\n`);
+}
+/**
+ * Sends a Server-Sent Event with error data.
+ * @param res The Express Response object
+ * @param reqId The request ID
+ * @param error The error to send
+ */
+export function sendSSEError(res, error) {
+    if (!res.writable) {
+        return;
+    }
+    const response = {
+        jsonrpc: "2.0",
+        error: error,
+    };
+    res.write(`data: ${JSON.stringify(response)}\n\n`);
+}
+/**
+ * Processes a task stream, handling generator yields and sending SSE events.
+ * @param req The request object
+ * @param res The response object
+ * @param taskId The task ID
+ * @param context The task context
+ * @param initialData The initial task data
+ */
+export async function processTaskStream(taskStore, taskHandler, res, taskId, context, initialData, onCancel, onEnd) {
+    let currentData = initialData;
+    const generator = taskHandler(context);
+    try {
+        for await (const yieldValue of generator) {
+            if (context.isCancelled()) {
+                await onCancel(currentData, res);
+                return;
+            }
+            currentData = await updateState(taskStore, currentData, yieldValue);
+            context.task = currentData.task;
+            if (isTaskStatusUpdate(yieldValue)) {
+                sendSSEEvent(res, {
+                    id: taskId,
+                    status: currentData.task.status,
+                    final: FINAL_STATES.includes(currentData.task.status.state),
+                });
+            }
+            else if (isArtifactUpdate(yieldValue)) {
+                const artifactIndex = currentData.task.artifacts?.findIndex((a) => a.name && a.name === yieldValue.name) ??
+                    yieldValue.index ??
+                    -1;
+                if (artifactIndex >= 0 &&
+                    currentData.task.artifacts &&
+                    artifactIndex < currentData.task.artifacts.length) {
+                    sendSSEEvent(res, {
+                        id: taskId,
+                        artifact: currentData.task.artifacts[artifactIndex],
+                        final: FINAL_STATES.includes(currentData.task.status.state),
+                    });
+                }
+            }
+        }
+    }
+    catch (error) {
+        try {
+            const failedUpdate = FAILED_UPDATE(error instanceof Error ? error.message : String(error));
+            currentData = await updateState(taskStore, currentData, failedUpdate);
+            sendSSEEvent(res, {
+                id: taskId,
+                status: currentData.task.status,
+            });
+        }
+        catch (saveError) {
+            logError("A2AServer", "Failed to save error state for streaming task", saveError, { taskId, originalError: error });
+        }
+        sendSSEError(res, INTERNAL_ERROR(`Task execution error: ${error instanceof Error ? error.message : String(error)}`));
+    }
+    finally {
+        onEnd(taskId, res);
+        if (!res.writableEnded) {
+            res.write("event: close\ndata: {}\n\n");
+            res.end();
+        }
+    }
+}
+//# sourceMappingURL=stream.js.map

package/dist/transport/streaming/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../src/transport/streaming/stream.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAIxD,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC7E,OAAO,EACL,gBAAgB,EAChB,kBAAkB,GACnB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAE/D;;;;;;;;;;GAUG;AACH,MAAM,UAAU,cAAc,CAC5B,GAAa,EACb,MAAc,EACd,aAAwB,EACxB,eAAyD;IAEzD,4BAA4B;IAC5B,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;IACnD,GAAG,CAAC,SAAS,CAAC,eAAe,EAAE,UAAU,CAAC,CAAC;IAC3C,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;IAE1C,2EAA2E;IAC3E,IAAI,eAAe,EAAE,CAAC;QACpB,eAAe,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,kCAAkC;IAClC,IAAI,aAAa,EAAE,CAAC;QAClB,YAAY,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;IACnC,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,YAAY,CAAC,GAAa,EAAE,MAAiB;IAC3D,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAClB,OAAO;IACT,CAAC;IAED,MAAM,QAAQ,GAA+B;QAC3C,OAAO,EAAE,KAAK;QACd,EAAE,EAAE,MAAM,CAAC,EAAE;QACb,MAAM,EAAE,MAAM;KACf,CAAC;IAEF,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AACrD,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,YAAY,CAC1B,GAAa,EACb,KAA6B;IAE7B,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAClB,OAAO;IACT,CAAC;IAED,MAAM,QAAQ,GAA+B;QAC3C,OAAO,EAAE,KAAK;QACd,KAAK,EAAE,KAAK;KACb,CAAC;IAEF,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AACrD,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,SAAoB,EACpB,WAAwB,EACxB,GAAa,EACb,MAAc,EACd,OAAoB,EACpB,WAA2B,EAC3B,QAAgE,EAChE,KAAuD;IAEvD,IAAI,WAAW,GAAG,WAAW,CAAC;IAE9B,MAAM,SAAS,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IAEvC,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,SAAS,EAAE,CAAC;YACzC,IAAI,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;gBAC1B,MAAM,QAAQ,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;gBACjC,OAAO;YACT,CAAC;YAED,WAAW,GAAG,MAAM,WAAW,CAAC,SAAS,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC;YAEpE,OAAO,CAAC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC;YAEhC,IAAI,kBAAkB,CAAC,UAAU,CAAC,EAAE,CAAC;gBACnC,YAAY,CAAC,GAAG,EAAE;oBAChB,EAAE,EAAE,MAAM;oBACV,MAAM,EAAE,WAAW,CAAC,IAAI,CAAC,MAAM;oBAC/B,KAAK,EAAE,YAAY,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC5D,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,gBAAgB,CAAC,UAAU,CAAC,EAAE,CAAC;gBACxC,MAAM,aAAa,GACjB,WAAW,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CACnC,CAAC,CAAW,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,CACtD;oBACD,UAAU,CAAC,KAAK;oBAChB,CAAC,CAAC,CAAC;gBAEL,IACE,aAAa,IAAI,CAAC;oBAClB,WAAW,CAAC,IAAI,CAAC,SAAS;oBAC1B,aAAa,GAAG,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EACjD,CAAC;oBACD,YAAY,CAAC,GAAG,EAAE;wBAChB,EAAE,EAAE,MAAM;wBACV,QAAQ,EAAE,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;wBACnD,KAAK,EAAE,YAAY,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC5D,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,aAAa,CAChC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD,CAAC;YAEF,WAAW,GAAG,MAAM,WAAW,CAAC,SAAS,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;YAEtE,YAAY,CAAC,GAAG,EAAE;gBAChB,EAAE,EAAE,MAAM;gBACV,MAAM,EAAE,WAAW,CAAC,IAAI,CAAC,MAAM;aAChC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,SAAS,EAAE,CAAC;YACnB,QAAQ,CACN,WAAW,EACX,+CAA+C,EAC/C,SAAS,EACT,EAAE,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,CACjC,CAAC;QACJ,CAAC;QAED,YAAY,CACV,GAAG,EACH,cAAc,CACZ,yBACE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CACvD,EAAE,CACH,CACF,CAAC;IACJ,CAAC;YAAS,CAAC;QACT,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACnB,IAAI,CAAC,GAAG,CAAC,aAAa,EAAE,CAAC;YACvB,GAAG,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;YACxC,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/types/extended-schema.d.ts

@@ -0,0 +1,52 @@
+/**
+ * This file contains Artinet-specific schema definitions separate from the A2A protocol.
+ * These are extensions to the standard A2A protocol defined by Google:
+ * @see https://github.com/google/A2A/blob/main/samples/js/src/schema.ts
+ */
+import type { JSONRPCMessage, JSONRPCError, TaskStatusUpdateEvent, TaskArtifactUpdateEvent, TaskStatus, Artifact } from "./schema.js";
+/**
+ * Improved JSON-RPC Response types using discriminated unions
+ * These provide better type safety than the standard optional field approach
+ */
+/**
+ * Represents a successful JSON-RPC response with a result.
+ */
+export interface JSONRPCSuccessResponse<R = any> extends JSONRPCMessage {
+    /**
+     * The result of the method invocation.
+     */
+    result: R;
+    /**
+     * In a success response, error must never be present.
+     */
+    error?: never;
+}
+/**
+ * Represents an error JSON-RPC response.
+ */
+export interface JSONRPCErrorResponse<E = any> extends JSONRPCMessage {
+    /**
+     * The error object.
+     */
+    error: JSONRPCError<E>;
+    /**
+     * In an error response, result must never be present.
+     */
+    result?: never;
+}
+/**
+ * Combined JSON-RPC response type as a discriminated union.
+ * This ensures a response is either a success with a result, or an error.
+ */
+export type ExtendedJSONRPCResponse<R = any, E = any> = JSONRPCSuccessResponse<R> | JSONRPCErrorResponse<E>;
+export type TaskEvent = TaskStatusUpdateEvent | TaskArtifactUpdateEvent;
+/**
+ * Represents the possible types of updates a TaskHandler can yield.
+ * Either a partial TaskStatus (without the server-managed timestamp)
+ * or a complete Artifact object.
+ */
+export type TaskYieldUpdate = Omit<TaskStatus, "timestamp"> | Artifact;
+export type ExtendedTaskStatusUpdate = Omit<TaskStatus, "timestamp">;
+export * from "./schema.js";
+export type { A2AError as A2AErrorType } from "./schema.js";
+//# sourceMappingURL=extended-schema.d.ts.map

package/dist/types/extended-schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"extended-schema.d.ts","sourceRoot":"","sources":["../../src/types/extended-schema.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EACV,cAAc,EACd,YAAY,EACZ,qBAAqB,EACrB,uBAAuB,EACvB,UAAU,EACV,QAAQ,EACT,MAAM,aAAa,CAAC;AAErB;;;GAGG;AAEH;;GAEG;AACH,MAAM,WAAW,sBAAsB,CAAC,CAAC,GAAG,GAAG,CAAE,SAAQ,cAAc;IACrE;;OAEG;IACH,MAAM,EAAE,CAAC,CAAC;IAEV;;OAEG;IACH,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB,CAAC,CAAC,GAAG,GAAG,CAAE,SAAQ,cAAc;IACnE;;OAEG;IACH,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC;IAEvB;;OAEG;IACH,MAAM,CAAC,EAAE,KAAK,CAAC;CAChB;AAED;;;GAGG;AACH,MAAM,MAAM,uBAAuB,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC,GAAG,GAAG,IAChD,sBAAsB,CAAC,CAAC,CAAC,GACzB,oBAAoB,CAAC,CAAC,CAAC,CAAC;AAE5B,MAAM,MAAM,SAAS,GAAG,qBAAqB,GAAG,uBAAuB,CAAC;AACxE;;;;GAIG;AACH,MAAM,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC;AAEvE,MAAM,MAAM,wBAAwB,GAAG,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;AAErE,cAAc,aAAa,CAAC;AAC5B,YAAY,EAAE,QAAQ,IAAI,YAAY,EAAE,MAAM,aAAa,CAAC"}

package/dist/types/extended-schema.js

@@ -0,0 +1,7 @@
+/**
+ * This file contains Artinet-specific schema definitions separate from the A2A protocol.
+ * These are extensions to the standard A2A protocol defined by Google:
+ * @see https://github.com/google/A2A/blob/main/samples/js/src/schema.ts
+ */
+export * from "./schema.js";
+//# sourceMappingURL=extended-schema.js.map

package/dist/types/extended-schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extended-schema.js","sourceRoot":"","sources":["../../src/types/extended-schema.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgEH,cAAc,aAAa,CAAC"}

package/dist/types/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./extended-schema.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/types/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,cAAc,sBAAsB,CAAC"}

package/dist/types/index.js

@@ -0,0 +1,2 @@
+export * from "./extended-schema.js";
+//# sourceMappingURL=index.js.map

package/dist/types/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,cAAc,sBAAsB,CAAC"}