@opensearch-project/agent-health 0.3.0 → 0.5.2

package/lib/dist/services/agent/sseStream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sseStream.d.ts","sourceRoot":"","sources":["../../../../services/agent/sseStream.ts"],"names":[],"mappings":"AAKA;;;GAGG;AAEH,OAAO,EAAE,SAAS,EAAiB,MAAM,cAAc,CAAC;AAGxD,MAAM,WAAW,gBAAgB;IAC/B,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,IAAI,CAAC,EAAE,GAAG,CAAC;IACX,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,IAAI,CAAC;IACpC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACjC,UAAU,CAAC,EAAE,MAAM,IAAI,CAAC;IACxB,uEAAuE;IACvE,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,mFAAmF;IACnF,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,qBAAa,SAAS;IACpB,OAAO,CAAC,eAAe,CAAgC;IAEvD;;OAEG;IACG,OAAO,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,IAAI,CAAC;IAgHvD;;;OAGG;YACW,aAAa;IAgG3B;;OAEG;IACH,KAAK,IAAI,IAAI;CAGd;AAED;;;GAGG;AACH,wBAAsB,gBAAgB,CACpC,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,GAAG,EACZ,OAAO,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,IAAI,EACnC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAChC,OAAO,CAAC,EAAE;IAAE,aAAa,CAAC,EAAE,MAAM,CAAA;CAAE,GACnC,OAAO,CAAC,IAAI,CAAC,CAkBf"}

package/lib/dist/services/agent/sseStream.js

@@ -0,0 +1,223 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Server-Sent Events (SSE) Stream Consumer
+ * Handles streaming connections to the agent endpoint
+ */
+import { AGUIEventType } from '../../types/agui.js';
+import { debug, isDebugEnabled } from '../../lib/debug.js';
+export class SSEClient {
+    constructor() {
+        this.abortController = null;
+    }
+    /**
+     * Start consuming SSE stream from the agent endpoint
+     */
+    async consume(options) {
+        const { url, method = 'POST', headers = {}, body, onEvent, onError, onComplete, completeOnRunEnd = false, idleTimeoutMs = 120000, // 2 minute idle timeout by default (LLM agents can be slow)
+         } = options;
+        this.abortController = new AbortController();
+        debug('SSE', 'Connecting to', url);
+        debug('SSE', 'Method:', method);
+        debug('SSE', 'Headers:', headers);
+        debug('SSE', 'Payload:', body ? JSON.stringify(body, null, 2).substring(0, 500) : 'none');
+        debug('SSE', 'Timeout:', idleTimeoutMs, 'ms');
+        try {
+            const requestConfig = {
+                method,
+                headers: {
+                    'Content-Type': 'application/json',
+                    Accept: 'text/event-stream',
+                    ...headers,
+                },
+                body: body ? JSON.stringify(body) : undefined,
+                signal: this.abortController.signal,
+            };
+            debug('SSE', 'Request config:', JSON.stringify(requestConfig, null, 2).substring(0, 500));
+            const response = await fetch(url, requestConfig);
+            debug('SSE', 'Response received:', response.status, response.statusText);
+            if (!response.ok) {
+                // Try to get error response body for debugging
+                let errorBody = '';
+                try {
+                    errorBody = await response.text();
+                    debug('SSE', 'Error response body:', errorBody.substring(0, 500));
+                }
+                catch {
+                    debug('SSE', 'Could not read error response body');
+                }
+                throw new Error(`HTTP ${response.status}: ${response.statusText}${errorBody ? ` - ${errorBody}` : ''}`);
+            }
+            if (!response.body) {
+                throw new Error('Response body is null');
+            }
+            console.info('[SSE] Connected to agent endpoint, streaming events...');
+            debug('SSE', 'Response status:', response.status);
+            debug('SSE', 'Content-Type:', response.headers.get('content-type'));
+            // Process the stream with idle timeout support
+            const completionReason = await this.processStream(response.body, onEvent, completeOnRunEnd, idleTimeoutMs);
+            console.info(`[SSE] Stream completed: ${completionReason}`);
+            debug('SSE', `Stream completed: ${completionReason}`);
+            onComplete?.();
+        }
+        catch (error) {
+            if (error instanceof Error) {
+                if (error.name === 'AbortError') {
+                    // AbortError after completeOnRunEnd is expected, call onComplete
+                    debug('SSE', 'Stream aborted (expected after run completion)');
+                    onComplete?.();
+                }
+                else {
+                    console.error('[SSE] Stream error:', error.message);
+                    console.error('[SSE] Debug mode is:', isDebugEnabled() ? 'ENABLED ✅' : 'DISABLED ❌');
+                    // Detailed debug logging with diagnostics
+                    const errorDetails = {
+                        name: error.name,
+                        message: error.message,
+                        stack: error.stack,
+                        cause: error.cause,
+                        url,
+                        method,
+                        headers: headers,
+                    };
+                    debug('SSE', 'Error details:', errorDetails);
+                    // Provide diagnostic hints based on error type
+                    if (error.message.includes('fetch failed')) {
+                        debug('SSE', '💡 Diagnostic: "fetch failed" typically means:');
+                        debug('SSE', '   - Connection refused (endpoint not running)');
+                        debug('SSE', '   - DNS resolution failed (invalid hostname)');
+                        debug('SSE', '   - Network unreachable (firewall/VPN issues)');
+                        debug('SSE', '   - SSL/TLS certificate issues (self-signed cert)');
+                        debug('SSE', `   Check if ${url} is accessible`);
+                    }
+                    else if (error.message.includes('timeout')) {
+                        debug('SSE', '💡 Diagnostic: Request timed out - endpoint may be slow or unresponsive');
+                    }
+                    else if (error.message.includes('ENOTFOUND')) {
+                        debug('SSE', '💡 Diagnostic: DNS lookup failed - hostname not found');
+                    }
+                    else if (error.message.includes('ECONNREFUSED')) {
+                        debug('SSE', '💡 Diagnostic: Connection refused - service not listening on this port');
+                    }
+                    onError?.(error);
+                }
+            }
+            else {
+                console.error('[SSE] Unknown error:', error);
+                debug('SSE', 'Unknown error details:', error);
+                onError?.(new Error('Unknown error occurred'));
+            }
+        }
+    }
+    /**
+     * Process the ReadableStream and parse SSE events
+     * @returns Reason for stream completion
+     */
+    async processStream(stream, onEvent, completeOnRunEnd = false, idleTimeoutMs = 120000) {
+        const reader = stream.getReader();
+        const decoder = new TextDecoder();
+        let buffer = '';
+        let lastEventTime = Date.now();
+        let eventCount = 0;
+        let idleCheckInterval = null;
+        // Set up idle timeout checker
+        const idleTimeoutPromise = new Promise((resolve) => {
+            idleCheckInterval = setInterval(() => {
+                const idleTime = Date.now() - lastEventTime;
+                if (eventCount > 0 && idleTime > idleTimeoutMs) {
+                    debug('SSE', `Idle timeout: no events for ${idleTime}ms (threshold: ${idleTimeoutMs}ms)`);
+                    this.abort();
+                    resolve('idle_timeout');
+                }
+            }, 1000); // Check every second
+        });
+        try {
+            // Race between stream processing and idle timeout
+            const streamPromise = (async () => {
+                while (true) {
+                    const { done, value } = await reader.read();
+                    if (done) {
+                        return 'connection_closed';
+                    }
+                    // Update last event time on any data received
+                    lastEventTime = Date.now();
+                    // Decode chunk and add to buffer
+                    buffer += decoder.decode(value, { stream: true });
+                    // Process complete SSE messages
+                    const lines = buffer.split('\n');
+                    // Keep the last incomplete line in the buffer
+                    buffer = lines.pop() || '';
+                    for (const line of lines) {
+                        if (line.startsWith('data: ')) {
+                            const data = line.slice(6); // Remove "data: " prefix
+                            if (data.trim()) {
+                                debug('SSE', 'Raw event:', data.substring(0, 200) + (data.length > 200 ? '...' : ''));
+                                try {
+                                    const event = JSON.parse(data);
+                                    debug('SSE', 'Parsed event:', event.type);
+                                    eventCount++;
+                                    // Log progress milestone every 10 events
+                                    if (eventCount % 10 === 0) {
+                                        console.info(`[SSE] Processed ${eventCount} events...`);
+                                    }
+                                    onEvent(event);
+                                    // Check for run-ending events when completeOnRunEnd is enabled
+                                    if (completeOnRunEnd && (event.type === AGUIEventType.RUN_FINISHED ||
+                                        event.type === AGUIEventType.RUN_ERROR)) {
+                                        debug('SSE', `Received ${event.type}, completing stream`);
+                                        this.abort();
+                                        return `event:${event.type}`;
+                                    }
+                                }
+                                catch (parseError) {
+                                    console.error('[SSE] Parse error:', parseError);
+                                    debug('SSE', 'Failed data:', data);
+                                }
+                            }
+                        }
+                        // Ignore other SSE fields (id:, event:, retry:, comments)
+                    }
+                }
+            })();
+            const reason = await Promise.race([streamPromise, idleTimeoutPromise]);
+            return reason;
+        }
+        finally {
+            if (idleCheckInterval) {
+                clearInterval(idleCheckInterval);
+            }
+            reader.releaseLock();
+        }
+    }
+    /**
+     * Abort the current stream connection
+     */
+    abort() {
+        this.abortController?.abort();
+    }
+}
+/**
+ * Convenience function for simple SSE consumption
+ * Completes when RUN_FINISHED/RUN_ERROR event is received, or when connection closes
+ */
+export async function consumeSSEStream(url, payload, onEvent, headers, options) {
+    const client = new SSEClient();
+    return new Promise((resolve, reject) => {
+        client.consume({
+            url,
+            method: 'POST',
+            headers,
+            body: payload,
+            onEvent,
+            onError: (error) => reject(error),
+            onComplete: () => resolve(),
+            // Enable auto-completion on RUN_FINISHED/RUN_ERROR events
+            // This prevents hanging when the agent doesn't close the connection
+            completeOnRunEnd: true,
+            idleTimeoutMs: options?.idleTimeoutMs,
+        });
+    });
+}
+//# sourceMappingURL=sseStream.js.map

package/lib/dist/services/agent/sseStream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sseStream.js","sourceRoot":"","sources":["../../../../services/agent/sseStream.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;GAGG;AAEH,OAAO,EAAa,aAAa,EAAE,MAAM,cAAc,CAAC;AACxD,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAiBpD,MAAM,OAAO,SAAS;IAAtB;QACU,oBAAe,GAA2B,IAAI,CAAC;IA+NzD,CAAC;IA7NC;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,OAAyB;QACrC,MAAM,EACJ,GAAG,EACH,MAAM,GAAG,MAAM,EACf,OAAO,GAAG,EAAE,EACZ,IAAI,EACJ,OAAO,EACP,OAAO,EACP,UAAU,EACV,gBAAgB,GAAG,KAAK,EACxB,aAAa,GAAG,MAAM,EAAE,4DAA4D;UACrF,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAE7C,KAAK,CAAC,KAAK,EAAE,eAAe,EAAE,GAAG,CAAC,CAAC;QACnC,KAAK,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QAChC,KAAK,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAClC,KAAK,CAAC,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QAC1F,KAAK,CAAC,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;QAE9C,IAAI,CAAC;YACH,MAAM,aAAa,GAAG;gBACpB,MAAM;gBACN,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,MAAM,EAAE,mBAAmB;oBAC3B,GAAG,OAAO;iBACX;gBACD,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;gBAC7C,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM;aACpC,CAAC;YACF,KAAK,CAAC,KAAK,EAAE,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;YAE1F,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;YAEjD,KAAK,CAAC,KAAK,EAAE,oBAAoB,EAAE,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;YAEzE,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,+CAA+C;gBAC/C,IAAI,SAAS,GAAG,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAClC,KAAK,CAAC,KAAK,EAAE,sBAAsB,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;gBACpE,CAAC;gBAAC,MAAM,CAAC;oBACP,KAAK,CAAC,KAAK,EAAE,oCAAoC,CAAC,CAAC;gBACrD,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,GAAG,SAAS,CAAC,CAAC,CAAC,MAAM,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC1G,CAAC;YAED,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAC3C,CAAC;YAED,OAAO,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACvE,KAAK,CAAC,KAAK,EAAE,kBAAkB,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;YAClD,KAAK,CAAC,KAAK,EAAE,eAAe,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC;YAEpE,+CAA+C;YAC/C,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,EAAE,gBAAgB,EAAE,aAAa,CAAC,CAAC;YAE3G,OAAO,CAAC,IAAI,CAAC,2BAA2B,gBAAgB,EAAE,CAAC,CAAC;YAC5D,KAAK,CAAC,KAAK,EAAE,qBAAqB,gBAAgB,EAAE,CAAC,CAAC;YACtD,UAAU,EAAE,EAAE,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAChC,iEAAiE;oBACjE,KAAK,CAAC,KAAK,EAAE,gDAAgD,CAAC,CAAC;oBAC/D,UAAU,EAAE,EAAE,CAAC;gBACjB,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;oBACpD,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC;oBAErF,0CAA0C;oBAC1C,MAAM,YAAY,GAAG;wBACnB,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,OAAO,EAAE,KAAK,CAAC,OAAO;wBACtB,KAAK,EAAE,KAAK,CAAC,KAAK;wBAClB,KAAK,EAAG,KAAa,CAAC,KAAK;wBAC3B,GAAG;wBACH,MAAM;wBACN,OAAO,EAAE,OAAO;qBACjB,CAAC;oBACF,KAAK,CAAC,KAAK,EAAE,gBAAgB,EAAE,YAAY,CAAC,CAAC;oBAE7C,+CAA+C;oBAC/C,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;wBAC3C,KAAK,CAAC,KAAK,EAAE,gDAAgD,CAAC,CAAC;wBAC/D,KAAK,CAAC,KAAK,EAAE,gDAAgD,CAAC,CAAC;wBAC/D,KAAK,CAAC,KAAK,EAAE,+CAA+C,CAAC,CAAC;wBAC9D,KAAK,CAAC,KAAK,EAAE,gDAAgD,CAAC,CAAC;wBAC/D,KAAK,CAAC,KAAK,EAAE,oDAAoD,CAAC,CAAC;wBACnE,KAAK,CAAC,KAAK,EAAE,eAAe,GAAG,gBAAgB,CAAC,CAAC;oBACnD,CAAC;yBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;wBAC7C,KAAK,CAAC,KAAK,EAAE,yEAAyE,CAAC,CAAC;oBAC1F,CAAC;yBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;wBAC/C,KAAK,CAAC,KAAK,EAAE,uDAAuD,CAAC,CAAC;oBACxE,CAAC;yBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;wBAClD,KAAK,CAAC,KAAK,EAAE,wEAAwE,CAAC,CAAC;oBACzF,CAAC;oBAED,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;gBAC7C,KAAK,CAAC,KAAK,EAAE,wBAAwB,EAAE,KAAK,CAAC,CAAC;gBAC9C,OAAO,EAAE,CAAC,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,aAAa,CACzB,MAAkC,EAClC,OAAmC,EACnC,mBAA4B,KAAK,EACjC,gBAAwB,MAAM;QAE9B,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,iBAAiB,GAA0C,IAAI,CAAC;QAEpE,8BAA8B;QAC9B,MAAM,kBAAkB,GAAG,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,EAAE;YACzD,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;gBACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,CAAC;gBAC5C,IAAI,UAAU,GAAG,CAAC,IAAI,QAAQ,GAAG,aAAa,EAAE,CAAC;oBAC/C,KAAK,CAAC,KAAK,EAAE,+BAA+B,QAAQ,kBAAkB,aAAa,KAAK,CAAC,CAAC;oBAC1F,IAAI,CAAC,KAAK,EAAE,CAAC;oBACb,OAAO,CAAC,cAAc,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC,qBAAqB;QACjC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,kDAAkD;YAClD,MAAM,aAAa,GAAG,CAAC,KAAK,IAAqB,EAAE;gBACjD,OAAO,IAAI,EAAE,CAAC;oBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;oBAE5C,IAAI,IAAI,EAAE,CAAC;wBACT,OAAO,mBAAmB,CAAC;oBAC7B,CAAC;oBAED,8CAA8C;oBAC9C,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAE3B,iCAAiC;oBACjC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBAElD,gCAAgC;oBAChC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAEjC,8CAA8C;oBAC9C,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;oBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACzB,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;4BAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,yBAAyB;4BAErD,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;gCAChB,KAAK,CAAC,KAAK,EAAE,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gCAEtF,IAAI,CAAC;oCACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAc,CAAC;oCAC5C,KAAK,CAAC,KAAK,EAAE,eAAe,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;oCAC1C,UAAU,EAAE,CAAC;oCAEb,yCAAyC;oCACzC,IAAI,UAAU,GAAG,EAAE,KAAK,CAAC,EAAE,CAAC;wCAC1B,OAAO,CAAC,IAAI,CAAC,mBAAmB,UAAU,YAAY,CAAC,CAAC;oCAC1D,CAAC;oCAED,OAAO,CAAC,KAAK,CAAC,CAAC;oCAEf,+DAA+D;oCAC/D,IAAI,gBAAgB,IAAI,CACtB,KAAK,CAAC,IAAI,KAAK,aAAa,CAAC,YAAY;wCACzC,KAAK,CAAC,IAAI,KAAK,aAAa,CAAC,SAAS,CACvC,EAAE,CAAC;wCACF,KAAK,CAAC,KAAK,EAAE,YAAY,KAAK,CAAC,IAAI,qBAAqB,CAAC,CAAC;wCAC1D,IAAI,CAAC,KAAK,EAAE,CAAC;wCACb,OAAO,SAAS,KAAK,CAAC,IAAI,EAAE,CAAC;oCAC/B,CAAC;gCACH,CAAC;gCAAC,OAAO,UAAU,EAAE,CAAC;oCACpB,OAAO,CAAC,KAAK,CAAC,oBAAoB,EAAE,UAAU,CAAC,CAAC;oCAChD,KAAK,CAAC,KAAK,EAAE,cAAc,EAAE,IAAI,CAAC,CAAC;gCACrC,CAAC;4BACH,CAAC;wBACH,CAAC;wBACD,0DAA0D;oBAC5D,CAAC;gBACH,CAAC;YACH,CAAC,CAAC,EAAE,CAAC;YAEL,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC,CAAC;YACvE,OAAO,MAAM,CAAC;QAChB,CAAC;gBAAS,CAAC;YACT,IAAI,iBAAiB,EAAE,CAAC;gBACtB,aAAa,CAAC,iBAAiB,CAAC,CAAC;YACnC,CAAC;YACD,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;IAChC,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,GAAW,EACX,OAAY,EACZ,OAAmC,EACnC,OAAgC,EAChC,OAAoC;IAEpC,MAAM,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;IAE/B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,CAAC,OAAO,CAAC;YACb,GAAG;YACH,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,OAAO;YACb,OAAO;YACP,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;YACjC,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE;YAC3B,0DAA0D;YAC1D,oEAAoE;YACpE,gBAAgB,EAAE,IAAI;YACtB,aAAa,EAAE,OAAO,EAAE,aAAa;SACtC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}

package/lib/dist/services/connectors/agui/AGUIStreamingConnector.d.ts

@@ -0,0 +1,44 @@
+/**
+ * AG-UI Streaming Connector
+ * Handles communication with agents using the AG-UI protocol over SSE
+ */
+import type { TrajectoryStep } from '../../../types/index.js';
+import type { AGUIEvent } from '../../../types/agui.js';
+import { BaseConnector } from '../../../services/connectors/base/BaseConnector.js';
+import type { ConnectorAuth, ConnectorRequest, ConnectorResponse, ConnectorProgressCallback, ConnectorRawEventCallback } from '../../../services/connectors/types.js';
+import { AgentRequestPayload } from '../../../services/agent/payloadBuilder.js';
+/**
+ * AG-UI Streaming Connector
+ * Uses Server-Sent Events (SSE) to stream agent responses in AG-UI format
+ */
+export declare class AGUIStreamingConnector extends BaseConnector {
+    readonly type: "agui-streaming";
+    readonly name = "AG-UI Streaming";
+    traceContext: {
+        propagateHeader: boolean;
+    };
+    readonly supportsStreaming = true;
+    /**
+     * Build AG-UI payload from standard request
+     */
+    buildPayload(request: ConnectorRequest): AgentRequestPayload;
+    /**
+     * Execute the request using SSE streaming
+     */
+    execute(endpoint: string, request: ConnectorRequest, auth: ConnectorAuth, onProgress?: ConnectorProgressCallback, onRawEvent?: ConnectorRawEventCallback): Promise<ConnectorResponse>;
+    /**
+     * Parse raw AG-UI events into trajectory steps
+     * Used for re-processing stored raw events
+     */
+    parseResponse(rawEvents: AGUIEvent[]): TrajectoryStep[];
+    /**
+     * Health check for AG-UI endpoint
+     * Tries to connect without sending a full request
+     */
+    healthCheck(endpoint: string, auth: ConnectorAuth): Promise<boolean>;
+}
+/**
+ * Default instance for convenience
+ */
+export declare const aguiStreamingConnector: AGUIStreamingConnector;
+//# sourceMappingURL=AGUIStreamingConnector.d.ts.map

package/lib/dist/services/connectors/agui/AGUIStreamingConnector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AGUIStreamingConnector.d.ts","sourceRoot":"","sources":["../../../../../services/connectors/agui/AGUIStreamingConnector.ts"],"names":[],"mappings":"AAKA;;;GAGG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAC9C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,0CAA0C,CAAC;AACzE,OAAO,KAAK,EACV,aAAa,EACb,gBAAgB,EAChB,iBAAiB,EACjB,yBAAyB,EACzB,yBAAyB,EAC1B,MAAM,6BAA6B,CAAC;AAErC,OAAO,EAAqB,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AAGzF;;;GAGG;AACH,qBAAa,sBAAuB,SAAQ,aAAa;IACvD,QAAQ,CAAC,IAAI,EAAG,gBAAgB,CAAU;IAC1C,QAAQ,CAAC,IAAI,qBAAqB;IAEzB,YAAY;;MAA6B;IAClD,QAAQ,CAAC,iBAAiB,QAAQ;IAElC;;OAEG;IACH,YAAY,CAAC,OAAO,EAAE,gBAAgB,GAAG,mBAAmB;IAS5D;;OAEG;IACG,OAAO,CACX,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,gBAAgB,EACzB,IAAI,EAAE,aAAa,EACnB,UAAU,CAAC,EAAE,yBAAyB,EACtC,UAAU,CAAC,EAAE,yBAAyB,GACrC,OAAO,CAAC,iBAAiB,CAAC;IA2C7B;;;OAGG;IACH,aAAa,CAAC,SAAS,EAAE,SAAS,EAAE,GAAG,cAAc,EAAE;IAIvD;;;OAGG;IACG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;CAgB3E;AAED;;GAEG;AACH,eAAO,MAAM,sBAAsB,wBAA+B,CAAC"}

package/lib/dist/services/connectors/agui/AGUIStreamingConnector.js

@@ -0,0 +1,95 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseConnector } from '../../../services/connectors/base/BaseConnector.js';
+import { consumeSSEStream } from '../../../services/agent/sseStream.js';
+import { buildAgentPayload } from '../../../services/agent/payloadBuilder.js';
+import { AGUIToTrajectoryConverter, computeTrajectoryFromRawEvents } from '../../../services/agent/aguiConverter.js';
+/**
+ * AG-UI Streaming Connector
+ * Uses Server-Sent Events (SSE) to stream agent responses in AG-UI format
+ */
+export class AGUIStreamingConnector extends BaseConnector {
+    constructor() {
+        super(...arguments);
+        this.type = 'agui-streaming';
+        this.name = 'AG-UI Streaming';
+        this.traceContext = { propagateHeader: true };
+        this.supportsStreaming = true;
+    }
+    /**
+     * Build AG-UI payload from standard request
+     */
+    buildPayload(request) {
+        return buildAgentPayload(request.testCase, request.modelId, request.threadId, request.runId);
+    }
+    /**
+     * Execute the request using SSE streaming
+     */
+    async execute(endpoint, request, auth, onProgress, onRawEvent) {
+        // Use pre-built payload from hook if available, otherwise build fresh
+        const hasPrebuiltPayload = !!request.payload;
+        const payload = request.payload || this.buildPayload(request);
+        const headers = this.buildAuthHeaders(auth);
+        this.injectTraceparentHeaders(headers);
+        const trajectory = [];
+        const rawEvents = [];
+        const converter = new AGUIToTrajectoryConverter();
+        this.debug('Executing AG-UI streaming request');
+        await consumeSSEStream(endpoint, payload, (event) => {
+            // Capture raw event for debugging
+            rawEvents.push(event);
+            onRawEvent?.(event);
+            // Convert to trajectory steps
+            const steps = converter.processEvent(event);
+            steps.forEach(step => {
+                trajectory.push(step);
+                onProgress?.(step);
+            });
+        }, headers);
+        const runId = converter.getRunId();
+        this.debug('Stream completed. RunId:', runId, 'Steps:', trajectory.length);
+        return {
+            trajectory,
+            runId,
+            rawEvents,
+            metadata: {
+                threadId: converter.getThreadId(),
+            },
+        };
+    }
+    /**
+     * Parse raw AG-UI events into trajectory steps
+     * Used for re-processing stored raw events
+     */
+    parseResponse(rawEvents) {
+        return computeTrajectoryFromRawEvents(rawEvents);
+    }
+    /**
+     * Health check for AG-UI endpoint
+     * Tries to connect without sending a full request
+     */
+    async healthCheck(endpoint, auth) {
+        try {
+            const headers = this.buildAuthHeaders(auth);
+            // For AG-UI endpoints, we can't really do a health check without
+            // making a full request, so just check if the endpoint is reachable
+            const response = await fetch(endpoint, {
+                method: 'OPTIONS',
+                headers,
+            });
+            // Accept any response that isn't a network error
+            return true;
+        }
+        catch (error) {
+            this.error('Health check failed:', error);
+            return false;
+        }
+    }
+}
+/**
+ * Default instance for convenience
+ */
+export const aguiStreamingConnector = new AGUIStreamingConnector();
+//# sourceMappingURL=AGUIStreamingConnector.js.map

package/lib/dist/services/connectors/agui/AGUIStreamingConnector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AGUIStreamingConnector.js","sourceRoot":"","sources":["../../../../../services/connectors/agui/AGUIStreamingConnector.ts"],"names":[],"mappings":"AAAA;;;GAGG;AASH,OAAO,EAAE,aAAa,EAAE,MAAM,0CAA0C,CAAC;AAQzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAuB,MAAM,iCAAiC,CAAC;AACzF,OAAO,EAAE,yBAAyB,EAAE,8BAA8B,EAAE,MAAM,gCAAgC,CAAC;AAE3G;;;GAGG;AACH,MAAM,OAAO,sBAAuB,SAAQ,aAAa;IAAzD;;QACW,SAAI,GAAG,gBAAyB,CAAC;QACjC,SAAI,GAAG,iBAAiB,CAAC;QAEzB,iBAAY,GAAG,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC;QACzC,sBAAiB,GAAG,IAAI,CAAC;IA8FpC,CAAC;IA5FC;;OAEG;IACH,YAAY,CAAC,OAAyB;QACpC,OAAO,iBAAiB,CACtB,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,OAAO,EACf,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,KAAK,CACd,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,QAAgB,EAChB,OAAyB,EACzB,IAAmB,EACnB,UAAsC,EACtC,UAAsC;QAEtC,sEAAsE;QACtE,MAAM,kBAAkB,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QAC7C,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;QACvC,MAAM,UAAU,GAAqB,EAAE,CAAC;QACxC,MAAM,SAAS,GAAgB,EAAE,CAAC;QAClC,MAAM,SAAS,GAAG,IAAI,yBAAyB,EAAE,CAAC;QAElD,IAAI,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAEhD,MAAM,gBAAgB,CACpB,QAAQ,EACR,OAAO,EACP,CAAC,KAAgB,EAAE,EAAE;YACnB,kCAAkC;YAClC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACtB,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC;YAEpB,8BAA8B;YAC9B,MAAM,KAAK,GAAG,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YAC5C,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;gBACnB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC,CAAC,CAAC;QACL,CAAC,EACD,OAAO,CACR,CAAC;QAEF,MAAM,KAAK,GAAG,SAAS,CAAC,QAAQ,EAAE,CAAC;QACnC,IAAI,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;QAE3E,OAAO;YACL,UAAU;YACV,KAAK;YACL,SAAS;YACT,QAAQ,EAAE;gBACR,QAAQ,EAAE,SAAS,CAAC,WAAW,EAAE;aAClC;SACF,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,aAAa,CAAC,SAAsB;QAClC,OAAO,8BAA8B,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,IAAmB;QACrD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAC5C,iEAAiE;YACjE,oEAAoE;YACpE,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;gBACrC,MAAM,EAAE,SAAS;gBACjB,OAAO;aACR,CAAC,CAAC;YACH,iDAAiD;YACjD,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;YAC1C,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,IAAI,sBAAsB,EAAE,CAAC"}

package/lib/dist/services/connectors/base/BaseConnector.d.ts

@@ -0,0 +1,81 @@
+import type { TrajectoryStep } from '../../../types/index.js';
+import type { AgentConnector, ConnectorAuth, ConnectorProtocol, ConnectorRequest, ConnectorResponse, ConnectorProgressCallback, ConnectorRawEventCallback, TraceContextStrategy } from '../../../services/connectors/types.js';
+/**
+ * Abstract base class for connectors
+ * Provides common functionality like auth header building and error handling
+ */
+export declare abstract class BaseConnector implements AgentConnector {
+    abstract readonly type: ConnectorProtocol;
+    abstract readonly name: string;
+    abstract readonly supportsStreaming: boolean;
+    /**
+     * Trace-context propagation strategy. Subclasses set defaults; users can
+     * override per-agent via `connectorConfig.traceContext` (merged in execute()).
+     */
+    traceContext?: TraceContextStrategy;
+    /**
+     * Build payload for the agent request
+     * Subclasses must implement this to transform standard format to agent-specific format
+     */
+    abstract buildPayload(request: ConnectorRequest): any;
+    /**
+     * Execute the request
+     * Subclasses must implement the actual execution logic
+     */
+    abstract execute(endpoint: string, request: ConnectorRequest, auth: ConnectorAuth, onProgress?: ConnectorProgressCallback, onRawEvent?: ConnectorRawEventCallback): Promise<ConnectorResponse>;
+    /**
+     * Parse raw response into TrajectoryStep array
+     * Subclasses must implement protocol-specific parsing
+     */
+    abstract parseResponse(rawResponse: any): TrajectoryStep[];
+    /**
+     * Build HTTP headers from auth configuration
+     * @param auth Authentication configuration
+     * @returns Headers object ready for fetch/axios
+     */
+    protected buildAuthHeaders(auth: ConnectorAuth): Record<string, string>;
+    /**
+     * Build environment variables from auth configuration
+     * Used by subprocess connectors
+     */
+    protected buildAuthEnv(auth: ConnectorAuth): Record<string, string>;
+    /**
+     * Generate a unique ID for trajectory steps
+     */
+    protected generateId(): string;
+    /**
+     * Create a trajectory step with common fields
+     */
+    protected createStep(type: TrajectoryStep['type'], content: string, extra?: Partial<TrajectoryStep>): TrajectoryStep;
+    /**
+     * Default health check implementation
+     * Subclasses can override for protocol-specific checks
+     */
+    healthCheck(endpoint: string, auth: ConnectorAuth): Promise<boolean>;
+    /**
+     * Log debug message with connector type prefix
+     */
+    protected debug(message: string, ...args: any[]): void;
+    /**
+     * Log error message with connector type prefix
+     */
+    protected error(message: string, ...args: any[]): void;
+    /**
+     * Build a W3C `TRACEPARENT` env var string from the active OTel span,
+     * if any. Returns an env-fragment object (empty if no active span or
+     * propagation is disabled for this connector).
+     *
+     * Used by subprocess connectors to make the child's OTel SDK adopt the
+     * eval `test_case` span as parent context (Strategy A).
+     */
+    protected buildTraceparentEnv(): Record<string, string>;
+    /**
+     * Inject W3C `traceparent` (and `tracestate`) headers into an outgoing
+     * HTTP request, if a span is active and propagation is enabled.
+     *
+     * Used by REST/SSE connectors so the remote agent's OTel SDK adopts the
+     * eval `test_case` span as parent context (Strategy A).
+     */
+    protected injectTraceparentHeaders(headers: Record<string, string>): void;
+}
+//# sourceMappingURL=BaseConnector.d.ts.map

package/lib/dist/services/connectors/base/BaseConnector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseConnector.d.ts","sourceRoot":"","sources":["../../../../../services/connectors/base/BaseConnector.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAG9C,OAAO,KAAK,EACV,cAAc,EACd,aAAa,EACb,iBAAiB,EACjB,gBAAgB,EAChB,iBAAiB,EACjB,yBAAyB,EACzB,yBAAyB,EACzB,oBAAoB,EACrB,MAAM,6BAA6B,CAAC;AAErC;;;GAGG;AACH,8BAAsB,aAAc,YAAW,cAAc;IAC3D,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,iBAAiB,CAAC;IAC1C,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAC/B,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,EAAE,OAAO,CAAC;IAE7C;;;OAGG;IACH,YAAY,CAAC,EAAE,oBAAoB,CAAC;IAEpC;;;OAGG;IACH,QAAQ,CAAC,YAAY,CAAC,OAAO,EAAE,gBAAgB,GAAG,GAAG;IAErD;;;OAGG;IACH,QAAQ,CAAC,OAAO,CACd,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,gBAAgB,EACzB,IAAI,EAAE,aAAa,EACnB,UAAU,CAAC,EAAE,yBAAyB,EACtC,UAAU,CAAC,EAAE,yBAAyB,GACrC,OAAO,CAAC,iBAAiB,CAAC;IAE7B;;;OAGG;IACH,QAAQ,CAAC,aAAa,CAAC,WAAW,EAAE,GAAG,GAAG,cAAc,EAAE;IAE1D;;;;OAIG;IACH,SAAS,CAAC,gBAAgB,CAAC,IAAI,EAAE,aAAa,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IA6CvE;;;OAGG;IACH,SAAS,CAAC,YAAY,CAAC,IAAI,EAAE,aAAa,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAanE;;OAEG;IACH,SAAS,CAAC,UAAU,IAAI,MAAM;IAI9B;;OAEG;IACH,SAAS,CAAC,UAAU,CAClB,IAAI,EAAE,cAAc,CAAC,MAAM,CAAC,EAC5B,OAAO,EAAE,MAAM,EACf,KAAK,CAAC,EAAE,OAAO,CAAC,cAAc,CAAC,GAC9B,cAAc;IAUjB;;;OAGG;IACG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;IAe1E;;OAEG;IACH,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,GAAG,IAAI;IAItD;;OAEG;IACH,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,GAAG,IAAI;IAItD;;;;;;;OAOG;IACH,SAAS,CAAC,mBAAmB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAwBvD;;;;;;OAMG;IACH,SAAS,CAAC,wBAAwB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI;CAI1E"}

package/lib/dist/services/connectors/base/BaseConnector.js

@@ -0,0 +1,170 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { debug as centralDebug } from '../../../lib/debug.js';
+import { trace, context as otelContext, propagation } from '@opentelemetry/api';
+/**
+ * Abstract base class for connectors
+ * Provides common functionality like auth header building and error handling
+ */
+export class BaseConnector {
+    /**
+     * Build HTTP headers from auth configuration
+     * @param auth Authentication configuration
+     * @returns Headers object ready for fetch/axios
+     */
+    buildAuthHeaders(auth) {
+        const headers = {};
+        switch (auth.type) {
+            case 'basic':
+                if (auth.username && auth.password) {
+                    const credentials = Buffer.from(`${auth.username}:${auth.password}`).toString('base64');
+                    headers['Authorization'] = `Basic ${credentials}`;
+                }
+                break;
+            case 'bearer':
+                if (auth.token) {
+                    headers['Authorization'] = `Bearer ${auth.token}`;
+                }
+                break;
+            case 'api-key':
+                if (auth.token) {
+                    // Common API key header patterns
+                    headers['X-API-Key'] = auth.token;
+                    headers['x-api-key'] = auth.token; // Some APIs use lowercase
+                }
+                break;
+            case 'aws-sigv4':
+                // AWS SigV4 signing would be handled separately
+                // This is a placeholder - actual implementation would use aws4 or @aws-sdk/signature-v4
+                console.warn('[BaseConnector] AWS SigV4 auth requires runtime signing');
+                break;
+            case 'none':
+            default:
+                // No auth headers needed
+                break;
+        }
+        // Always apply custom headers (can override auth headers if needed)
+        if (auth.headers) {
+            Object.assign(headers, auth.headers);
+        }
+        return headers;
+    }
+    /**
+     * Build environment variables from auth configuration
+     * Used by subprocess connectors
+     */
+    buildAuthEnv(auth) {
+        const env = {};
+        if (auth.type === 'aws-sigv4') {
+            if (auth.awsRegion)
+                env['AWS_REGION'] = auth.awsRegion;
+            if (auth.awsAccessKeyId)
+                env['AWS_ACCESS_KEY_ID'] = auth.awsAccessKeyId;
+            if (auth.awsSecretAccessKey)
+                env['AWS_SECRET_ACCESS_KEY'] = auth.awsSecretAccessKey;
+            if (auth.awsSessionToken)
+                env['AWS_SESSION_TOKEN'] = auth.awsSessionToken;
+        }
+        return env;
+    }
+    /**
+     * Generate a unique ID for trajectory steps
+     */
+    generateId() {
+        return `${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
+    }
+    /**
+     * Create a trajectory step with common fields
+     */
+    createStep(type, content, extra) {
+        return {
+            id: this.generateId(),
+            timestamp: Date.now(),
+            type,
+            content,
+            ...extra,
+        };
+    }
+    /**
+     * Default health check implementation
+     * Subclasses can override for protocol-specific checks
+     */
+    async healthCheck(endpoint, auth) {
+        try {
+            // Default: try a simple fetch with HEAD method
+            const headers = this.buildAuthHeaders(auth);
+            const response = await fetch(endpoint, {
+                method: 'HEAD',
+                headers,
+            });
+            return response.ok;
+        }
+        catch (error) {
+            console.error(`[${this.type}] Health check failed:`, error);
+            return false;
+        }
+    }
+    /**
+     * Log debug message with connector type prefix
+     */
+    debug(message, ...args) {
+        centralDebug(this.type, message, ...args);
+    }
+    /**
+     * Log error message with connector type prefix
+     */
+    error(message, ...args) {
+        console.error(`[${this.type}] ${message}`, ...args);
+    }
+    /**
+     * Build a W3C `TRACEPARENT` env var string from the active OTel span,
+     * if any. Returns an env-fragment object (empty if no active span or
+     * propagation is disabled for this connector).
+     *
+     * Used by subprocess connectors to make the child's OTel SDK adopt the
+     * eval `test_case` span as parent context (Strategy A).
+     */
+    buildTraceparentEnv() {
+        if (!this.traceContext?.propagateEnv)
+            return {};
+        const span = trace.getSpan(otelContext.active());
+        if (!span)
+            return {};
+        const ctx = span.spanContext();
+        if (!ctx.traceId || !ctx.spanId)
+            return {};
+        // Both trace-id and parent-id must be non-zero per W3C trace context spec
+        // (https://www.w3.org/TR/trace-context/#trace-id). Emitting an all-zero
+        // value would produce an invalid traceparent that downstream SDKs
+        // typically reject silently — the agent's root span won't adopt our
+        // context as parent and we'd get back to two unrelated trace trees.
+        if (ctx.traceId === '0'.repeat(32) || ctx.spanId === '0'.repeat(16))
+            return {};
+        // 00 = W3C version; flags=01 means SAMPLED so the agent's exporter forwards.
+        const flags = (ctx.traceFlags ?? 1).toString(16).padStart(2, '0');
+        const env = {
+            TRACEPARENT: `00-${ctx.traceId}-${ctx.spanId}-${flags}`,
+        };
+        if (ctx.traceState) {
+            const ts = ctx.traceState.serialize();
+            if (ts)
+                env.TRACESTATE = ts;
+        }
+        return env;
+    }
+    /**
+     * Inject W3C `traceparent` (and `tracestate`) headers into an outgoing
+     * HTTP request, if a span is active and propagation is enabled.
+     *
+     * Used by REST/SSE connectors so the remote agent's OTel SDK adopts the
+     * eval `test_case` span as parent context (Strategy A).
+     */
+    injectTraceparentHeaders(headers) {
+        if (!this.traceContext?.propagateHeader)
+            return;
+        propagation.inject(otelContext.active(), headers);
+    }
+}
+//# sourceMappingURL=BaseConnector.js.map

package/lib/dist/services/connectors/base/BaseConnector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseConnector.js","sourceRoot":"","sources":["../../../../../services/connectors/base/BaseConnector.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,KAAK,IAAI,YAAY,EAAE,MAAM,aAAa,CAAC;AACpD,OAAO,EAAE,KAAK,EAAE,OAAO,IAAI,WAAW,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAYhF;;;GAGG;AACH,MAAM,OAAgB,aAAa;IAmCjC;;;;OAIG;IACO,gBAAgB,CAAC,IAAmB;QAC5C,MAAM,OAAO,GAA2B,EAAE,CAAC;QAE3C,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,OAAO;gBACV,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACnC,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBACxF,OAAO,CAAC,eAAe,CAAC,GAAG,SAAS,WAAW,EAAE,CAAC;gBACpD,CAAC;gBACD,MAAM;YAER,KAAK,QAAQ;gBACX,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,KAAK,EAAE,CAAC;gBACpD,CAAC;gBACD,MAAM;YAER,KAAK,SAAS;gBACZ,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBACf,iCAAiC;oBACjC,OAAO,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;oBAClC,OAAO,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,0BAA0B;gBAC/D,CAAC;gBACD,MAAM;YAER,KAAK,WAAW;gBACd,gDAAgD;gBAChD,wFAAwF;gBACxF,OAAO,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;gBACxE,MAAM;YAER,KAAK,MAAM,CAAC;YACZ;gBACE,yBAAyB;gBACzB,MAAM;QACV,CAAC;QAED,oEAAoE;QACpE,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;OAGG;IACO,YAAY,CAAC,IAAmB;QACxC,MAAM,GAAG,GAA2B,EAAE,CAAC;QAEvC,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9B,IAAI,IAAI,CAAC,SAAS;gBAAE,GAAG,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;YACvD,IAAI,IAAI,CAAC,cAAc;gBAAE,GAAG,CAAC,mBAAmB,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC;YACxE,IAAI,IAAI,CAAC,kBAAkB;gBAAE,GAAG,CAAC,uBAAuB,CAAC,GAAG,IAAI,CAAC,kBAAkB,CAAC;YACpF,IAAI,IAAI,CAAC,eAAe;gBAAE,GAAG,CAAC,mBAAmB,CAAC,GAAG,IAAI,CAAC,eAAe,CAAC;QAC5E,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAED;;OAEG;IACO,UAAU;QAClB,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IACvE,CAAC;IAED;;OAEG;IACO,UAAU,CAClB,IAA4B,EAC5B,OAAe,EACf,KAA+B;QAE/B,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,UAAU,EAAE;YACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,IAAI;YACJ,OAAO;YACP,GAAG,KAAK;SACT,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,IAAmB;QACrD,IAAI,CAAC;YACH,+CAA+C;YAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;gBACrC,MAAM,EAAE,MAAM;gBACd,OAAO;aACR,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,EAAE,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,wBAAwB,EAAE,KAAK,CAAC,CAAC;YAC5D,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACO,KAAK,CAAC,OAAe,EAAE,GAAG,IAAW;QAC7C,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACO,KAAK,CAAC,OAAe,EAAE,GAAG,IAAW;QAC7C,OAAO,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IACtD,CAAC;IAED;;;;;;;OAOG;IACO,mBAAmB;QAC3B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,YAAY;YAAE,OAAO,EAAE,CAAC;QAChD,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QACjD,IAAI,CAAC,IAAI;YAAE,OAAO,EAAE,CAAC;QACrB,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAC/B,IAAI,CAAC,GAAG,CAAC,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC;QAC3C,0EAA0E;QAC1E,wEAAwE;QACxE,kEAAkE;QAClE,oEAAoE;QACpE,oEAAoE;QACpE,IAAI,GAAG,CAAC,OAAO,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YAAE,OAAO,EAAE,CAAC;QAC/E,6EAA6E;QAC7E,MAAM,KAAK,GAAG,CAAC,GAAG,CAAC,UAAU,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAClE,MAAM,GAAG,GAA2B;YAClC,WAAW,EAAE,MAAM,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,MAAM,IAAI,KAAK,EAAE;SACxD,CAAC;QACF,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;YACnB,MAAM,EAAE,GAAG,GAAG,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;YACtC,IAAI,EAAE;gBAAE,GAAG,CAAC,UAAU,GAAG,EAAE,CAAC;QAC9B,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED;;;;;;OAMG;IACO,wBAAwB,CAAC,OAA+B;QAChE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,eAAe;YAAE,OAAO;QAChD,WAAW,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;CACF"}