salmon-loop 0.2.16 → 0.3.1

package/dist/core/protocols/a2a/sdk/executor.js

@@ -10,6 +10,7 @@ export function createA2AInteractionExecutor(deps) {
     const submittedPublished = new Set();
     // Prevents duplicate terminal status events when publishTaskStatus is called multiple times
     const terminalPublished = new Set();
+    const artifactSignaturesByTaskId = new Map();
     return {
         async execute(requestContext, executionEventBus) {
             const capability = parseA2ASkillFlowMode(deps.capabilityResolver?.(requestContext.userMessage)) ?? 'autopilot';
@@ -89,24 +90,9 @@ export function createA2AInteractionExecutor(deps) {
         // ALWAYS publish "submitted" first if not yet published
         if (!submittedPublished.has(taskId)) {
             submittedPublished.add(taskId);
-            const submittedStatus = {
-                state: 'submitted',
-                timestamp: envelope.createdAt ?? new Date().toISOString(),
-                message: undefined,
-            };
-            const submittedUpdate = {
-                kind: 'status-update',
-                taskId: envelope.id,
-                contextId: metadata.contextId,
-                status: submittedStatus,
-                final: false,
-                metadata: { attempt: envelope.attempt },
-            };
-            eventBus.publish(submittedUpdate);
-            // Save as 'running' state (not completed) to keep task cancelable during grace period.
-            // SDK rejects cancellation if task is already in terminal state in the store.
-            const submittedEnvelope = { ...envelope, state: 'running' };
-            const snapshot = buildTaskSnapshot(submittedEnvelope, metadata);
+            const submittedEnvelope = { ...envelope, state: 'accepted' };
+            const snapshot = buildTaskSnapshot(submittedEnvelope, metadata, submittedEnvelope.createdAt ?? new Date().toISOString());
+            eventBus.publish(snapshot);
             await store.save(snapshot);
             // If the current state is still "submitted", we're done
             if (currentState === 'submitted') {
@@ -124,10 +110,11 @@ export function createA2AInteractionExecutor(deps) {
             if (cancelledTaskIds.has(taskId)) {
                 terminalPublished.add(taskId);
                 const canceledEnvelope = { ...envelope, state: 'cancelled' };
-                const snapshot = buildTaskSnapshot(canceledEnvelope, metadata);
+                const recordedAt = new Date().toISOString();
+                const snapshot = buildTaskSnapshot(canceledEnvelope, metadata, recordedAt);
                 await store.save(snapshot);
                 const status = {
-                    ...buildTaskStatus(envelope, metadata.contextId),
+                    ...buildTaskStatus(envelope, metadata.contextId, recordedAt),
                     state: 'canceled',
                 };
                 const update = {
@@ -149,13 +136,14 @@ export function createA2AInteractionExecutor(deps) {
             const wasCancelled = cancelledTaskIds.has(taskId) || taskAfterGrace?.status.state === 'canceled';
             if (wasCancelled) {
                 terminalPublished.add(taskId);
+                const recordedAt = new Date().toISOString();
                 if (taskAfterGrace?.status.state !== 'canceled') {
                     const canceledEnvelope = { ...envelope, state: 'cancelled' };
-                    const snapshot = buildTaskSnapshot(canceledEnvelope, metadata);
+                    const snapshot = buildTaskSnapshot(canceledEnvelope, metadata, recordedAt);
                     await store.save(snapshot);
                 }
                 const status = {
-                    ...buildTaskStatus(envelope, metadata.contextId),
+                    ...buildTaskStatus(envelope, metadata.contextId, recordedAt),
                     state: 'canceled',
                 };
                 const update = {
@@ -171,9 +159,11 @@ export function createA2AInteractionExecutor(deps) {
             }
             // No cancellation detected, safe to publish "completed"
             terminalPublished.add(taskId);
-            const snapshot = buildTaskSnapshot(envelope, metadata);
+            const recordedAt = new Date().toISOString();
+            const snapshot = buildTaskSnapshot(envelope, metadata, recordedAt);
             await store.save(snapshot);
-            const status = buildTaskStatus(envelope, metadata.contextId);
+            publishArtifactUpdates(envelope, metadata, eventBus);
+            const status = buildTaskStatus(envelope, metadata.contextId, recordedAt);
             const update = {
                 kind: 'status-update',
                 taskId: envelope.id,
@@ -186,9 +176,11 @@ export function createA2AInteractionExecutor(deps) {
             return true;
         }
         // For all other states (working, failed, canceled, etc.), save and publish immediately
-        const snapshot = buildTaskSnapshot(envelope, metadata);
+        const recordedAt = new Date().toISOString();
+        const snapshot = buildTaskSnapshot(envelope, metadata, recordedAt);
         await store.save(snapshot);
-        const status = buildTaskStatus(envelope, metadata.contextId);
+        publishArtifactUpdates(envelope, metadata, eventBus);
+        const status = buildTaskStatus(envelope, metadata.contextId, recordedAt);
         const isFinal = isTerminalState(status.state);
         if (isFinal) {
             terminalPublished.add(taskId);
@@ -215,6 +207,7 @@ export function createA2AInteractionExecutor(deps) {
         cancelledTaskIds.delete(taskId);
         submittedPublished.delete(taskId);
         terminalPublished.delete(taskId);
+        artifactSignaturesByTaskId.delete(taskId);
     }
     function extractInstruction(message) {
         const textParts = message.parts
@@ -247,10 +240,10 @@ export function createA2AInteractionExecutor(deps) {
     function isTerminalState(state) {
         return (state === 'completed' || state === 'failed' || state === 'canceled' || state === 'rejected');
     }
-    function buildTaskStatus(envelope, contextId) {
+    function buildTaskStatus(envelope, contextId, timestamp = envelope.createdAt ?? new Date().toISOString()) {
         return {
             state: mapState(envelope.state),
-            timestamp: envelope.createdAt ?? new Date().toISOString(),
+            timestamp,
             message: buildStatusMessage(envelope, contextId),
         };
     }
@@ -271,7 +264,7 @@ export function createA2AInteractionExecutor(deps) {
             ],
         };
     }
-    function buildTaskSnapshot(envelope, metadata) {
+    function buildTaskSnapshot(envelope, metadata, timestamp) {
         return {
             id: envelope.id,
             kind: 'task',
@@ -282,22 +275,98 @@ export function createA2AInteractionExecutor(deps) {
                 attempt: envelope.attempt,
             },
             artifacts: convertArtifacts(envelope.artifacts),
-            status: buildTaskStatus(envelope, metadata.contextId),
+            status: buildTaskStatus(envelope, metadata.contextId, timestamp),
         };
     }
+    function publishArtifactUpdates(envelope, metadata, eventBus) {
+        if (!envelope.artifacts || envelope.artifacts.length === 0) {
+            return;
+        }
+        let knownSignatures = artifactSignaturesByTaskId.get(envelope.id);
+        if (!knownSignatures) {
+            knownSignatures = new Map();
+            artifactSignaturesByTaskId.set(envelope.id, knownSignatures);
+        }
+        for (const artifact of envelope.artifacts) {
+            const signature = artifactSignature(artifact);
+            if (knownSignatures.get(artifact.id) === signature) {
+                continue;
+            }
+            knownSignatures.set(artifact.id, signature);
+            const update = {
+                kind: 'artifact-update',
+                taskId: envelope.id,
+                contextId: metadata.contextId,
+                artifact: {
+                    artifactId: artifact.id,
+                    name: artifact.name,
+                    parts: [artifactToPart(artifact)],
+                },
+                append: false,
+                lastChunk: true,
+                metadata: { attempt: envelope.attempt },
+            };
+            eventBus.publish(update);
+        }
+    }
     function convertArtifacts(artifacts) {
         if (!artifacts || artifacts.length === 0)
             return undefined;
         return artifacts.map((artifact) => ({
             artifactId: artifact.id,
             name: artifact.name,
-            parts: [
-                {
-                    kind: 'text',
-                    text: artifact.content ?? artifact.name ?? '',
-                },
-            ],
+            parts: [artifactToPart(artifact)],
         }));
     }
+    function artifactToPart(artifact) {
+        if (artifact.url) {
+            return {
+                kind: 'file',
+                file: {
+                    name: artifact.name,
+                    mimeType: artifact.mimeType,
+                    uri: artifact.url,
+                },
+            };
+        }
+        if (artifact.handle) {
+            return {
+                kind: 'file',
+                file: {
+                    name: artifact.name,
+                    mimeType: artifact.mimeType,
+                    bytes: artifact.handle,
+                },
+            };
+        }
+        if (artifact.kind === 'data' && artifact.content) {
+            try {
+                return {
+                    kind: 'data',
+                    data: JSON.parse(artifact.content),
+                };
+            }
+            catch {
+                // Fall through to text
+            }
+        }
+        return {
+            kind: 'text',
+            text: artifact.content ?? artifact.name ?? '',
+        };
+    }
+    function artifactSignature(artifact) {
+        return JSON.stringify({
+            id: artifact.id,
+            name: artifact.name,
+            kind: artifact.kind,
+            mimeType: artifact.mimeType,
+            content: artifact.content,
+            delivery: artifact.delivery,
+            handle: artifact.handle,
+            url: artifact.url,
+            expiresAt: artifact.expiresAt,
+        });
+    }
 }
 //# sourceMappingURL=executor.js.map