@opensearch-project/agent-health 0.4.0 → 0.6.0

package/lib/dist/services/connectors/openai-compatible/OpenAICompatibleConnector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenAICompatibleConnector.d.ts","sourceRoot":"","sources":["../../../../../services/connectors/openai-compatible/OpenAICompatibleConnector.ts"],"names":[],"mappings":"AAKA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAkB,MAAM,SAAS,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,MAAM,0CAA0C,CAAC;AACzE,OAAO,KAAK,EACV,aAAa,EACb,gBAAgB,EAChB,iBAAiB,EACjB,yBAAyB,EACzB,yBAAyB,EAC1B,MAAM,6BAA6B,CAAC;AA0CrC;;GAEG;AACH,qBAAa,yBAA0B,SAAQ,aAAa;IAC1D,QAAQ,CAAC,IAAI,EAAG,mBAAmB,CAAU;IAC7C,QAAQ,CAAC,IAAI,uBAAuB;IAE3B,YAAY;;MAA6B;IAClD,QAAQ,CAAC,iBAAiB,SAAS;IAEnC;;OAEG;IACH,YAAY,CAAC,OAAO,EAAE,gBAAgB,GAAG,GAAG;IAwC5C;;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;IAwC7B;;OAEG;IACH,aAAa,CAAC,IAAI,EAAE,GAAG,GAAG,cAAc,EAAE;CAwC3C;AAED;;GAEG;AACH,eAAO,MAAM,yBAAyB,2BAAkC,CAAC"}

package/lib/dist/services/connectors/openai-compatible/OpenAICompatibleConnector.js

@@ -0,0 +1,133 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseConnector } from '../../../services/connectors/base/BaseConnector.js';
+/**
+ * OpenAI-compatible connector
+ */
+export class OpenAICompatibleConnector extends BaseConnector {
+    constructor() {
+        super(...arguments);
+        this.type = 'openai-compatible';
+        this.name = 'OpenAI-compatible';
+        this.traceContext = { propagateHeader: true };
+        this.supportsStreaming = false;
+    }
+    /**
+     * Build OpenAI Chat Completion payload from test case
+     */
+    buildPayload(request) {
+        const messages = [];
+        // Add context as system message if present
+        if (request.testCase.context && request.testCase.context.length > 0) {
+            const contextText = request.testCase.context
+                .map((c) => typeof c === 'string' ? c : JSON.stringify(c))
+                .join('\n');
+            messages.push({
+                role: 'system',
+                content: contextText,
+            });
+        }
+        // Add the prompt as user message
+        messages.push({
+            role: 'user',
+            content: request.testCase.initialPrompt,
+        });
+        const payload = {
+            model: request.modelId,
+            messages,
+        };
+        // Add tools if the test case defines them
+        if (request.testCase.tools && request.testCase.tools.length > 0) {
+            payload.tools = request.testCase.tools.map((tool) => ({
+                type: 'function',
+                function: {
+                    name: tool.name,
+                    description: tool.description || '',
+                    parameters: tool.parameters || {},
+                },
+            }));
+        }
+        return payload;
+    }
+    /**
+     * Execute OpenAI-compatible Chat Completion request
+     */
+    async execute(endpoint, request, auth, onProgress, onRawEvent) {
+        const payload = request.payload || this.buildPayload(request);
+        const headers = this.buildAuthHeaders(auth);
+        this.injectTraceparentHeaders(headers);
+        this.debug('Executing OpenAI-compatible request');
+        this.debug('Endpoint:', endpoint);
+        this.debug('Model:', payload.model);
+        const response = await fetch(endpoint, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                ...headers,
+            },
+            body: JSON.stringify(payload),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`OpenAI-compatible request failed: ${response.status} - ${errorText}`);
+        }
+        const data = await response.json();
+        onRawEvent?.(data);
+        const trajectory = this.parseResponse(data);
+        trajectory.forEach(step => onProgress?.(step));
+        return {
+            trajectory,
+            runId: data.id || null,
+            rawEvents: [data],
+            metadata: {
+                model: data.model,
+                usage: data.usage,
+                finishReason: data.choices?.[0]?.finish_reason,
+            },
+        };
+    }
+    /**
+     * Parse OpenAI Chat Completion response into trajectory steps
+     */
+    parseResponse(data) {
+        const steps = [];
+        const choice = data.choices?.[0];
+        if (!choice) {
+            steps.push(this.createStep('response', JSON.stringify(data, null, 2)));
+            return steps;
+        }
+        const message = choice.message;
+        // Handle tool calls
+        if (message.tool_calls && message.tool_calls.length > 0) {
+            for (const toolCall of message.tool_calls) {
+                let toolArgs;
+                try {
+                    toolArgs = JSON.parse(toolCall.function.arguments);
+                }
+                catch {
+                    toolArgs = toolCall.function.arguments;
+                }
+                steps.push(this.createStep('action', `Calling ${toolCall.function.name}...`, {
+                    toolName: toolCall.function.name,
+                    toolArgs,
+                }));
+            }
+        }
+        // Handle content response
+        if (message.content) {
+            steps.push(this.createStep('response', message.content));
+        }
+        // If no content and no tool calls, create a generic response
+        if (steps.length === 0) {
+            steps.push(this.createStep('response', '(empty response)'));
+        }
+        return steps;
+    }
+}
+/**
+ * Default instance for convenience
+ */
+export const openaiCompatibleConnector = new OpenAICompatibleConnector();
+//# sourceMappingURL=OpenAICompatibleConnector.js.map

package/lib/dist/services/connectors/openai-compatible/OpenAICompatibleConnector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"OpenAICompatibleConnector.js","sourceRoot":"","sources":["../../../../../services/connectors/openai-compatible/OpenAICompatibleConnector.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAeH,OAAO,EAAE,aAAa,EAAE,MAAM,0CAA0C,CAAC;AAiDzE;;GAEG;AACH,MAAM,OAAO,yBAA0B,SAAQ,aAAa;IAA5D;;QACW,SAAI,GAAG,mBAA4B,CAAC;QACpC,SAAI,GAAG,mBAAmB,CAAC;QAE3B,iBAAY,GAAG,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC;QACzC,sBAAiB,GAAG,KAAK,CAAC;IAyIrC,CAAC;IAvIC;;OAEG;IACH,YAAY,CAAC,OAAyB;QACpC,MAAM,QAAQ,GAAkB,EAAE,CAAC;QAEnC,2CAA2C;QAC3C,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpE,MAAM,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO;iBACzC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;iBAC9D,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,WAAW;aACrB,CAAC,CAAC;QACL,CAAC;QAED,iCAAiC;QACjC,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,aAAa;SACxC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAQ;YACnB,KAAK,EAAE,OAAO,CAAC,OAAO;YACtB,QAAQ;SACT,CAAC;QAEF,0CAA0C;QAC1C,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChE,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC;gBACzD,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACR,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,EAAE;oBACnC,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE;iBAClC;aACF,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,QAAgB,EAChB,OAAyB,EACzB,IAAmB,EACnB,UAAsC,EACtC,UAAsC;QAEtC,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,IAAI,CAAC,KAAK,CAAC,qCAAqC,CAAC,CAAC;QAClD,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QAEpC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;YACrC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,OAAO;aACX;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;SAC9B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,qCAAqC,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,IAAI,GAA2B,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC3D,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;QAEnB,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QAC5C,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;QAE/C,OAAO;YACL,UAAU;YACV,KAAK,EAAE,IAAI,CAAC,EAAE,IAAI,IAAI;YACtB,SAAS,EAAE,CAAC,IAAI,CAAC;YACjB,QAAQ,EAAE;gBACR,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa;aAC/C;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,IAAS;QACrB,MAAM,KAAK,GAAqB,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAEjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACvE,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAE/B,oBAAoB;QACpB,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBAC1C,IAAI,QAAa,CAAC;gBAClB,IAAI,CAAC;oBACH,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBACrD,CAAC;gBAAC,MAAM,CAAC;oBACP,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;gBACzC,CAAC;gBAED,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,WAAW,QAAQ,CAAC,QAAQ,CAAC,IAAI,KAAK,EAAE;oBAC3E,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;oBAChC,QAAQ;iBACT,CAAC,CAAC,CAAC;YACN,CAAC;QACH,CAAC;QAED,0BAA0B;QAC1B,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;QAC3D,CAAC;QAED,6DAA6D;QAC7D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,yBAAyB,EAAE,CAAC"}

package/lib/dist/services/connectors/pi/PiConnector.d.ts

@@ -0,0 +1,87 @@
+/**
+ * Pi Connector
+ * Subprocess connector for pi.dev CLI — a minimal terminal coding harness.
+ * Spawns `pi` with --print mode and parses JSON output.
+ */
+import type { TrajectoryStep } from '../../../types/index.js';
+import { SubprocessConnector } from '../../../services/connectors/subprocess/SubprocessConnector.js';
+import type { ConnectorAuth, ConnectorRequest, ConnectorResponse, ConnectorProgressCallback, ConnectorRawEventCallback, SubprocessConfig } from '../../../services/connectors/types.js';
+/**
+ * Configuration options for Pi connector
+ * Passed via agent.connectorConfig in agent-health.config.ts
+ */
+export interface PiConnectorConfig {
+    env?: Record<string, string>;
+    /** Path to pi package to load (e.g., ./observio-sample-agent/pi-package) */
+    packagePath?: string;
+    /** Pi model override (e.g., 'claude-sonnet-4-5') */
+    model?: string;
+    /** Working directory for pi */
+    workingDir?: string;
+    /** Timeout in ms (default: 600000 = 10 minutes) */
+    timeout?: number;
+    /** Additional CLI args to pass to pi */
+    additionalArgs?: string[];
+}
+/**
+ * Pi CLI Connector
+ * Invokes pi.dev as a subprocess for agent evaluation
+ */
+export declare class PiConnector extends SubprocessConnector {
+    readonly type: "pi";
+    readonly name = "Pi (pi.dev)";
+    traceContext: {
+        propagateEnv: boolean;
+        serviceName: string;
+    };
+    private piOutputBuffer;
+    private piThinkingBuffer;
+    private piTextBuffer;
+    constructor(config?: Partial<SubprocessConfig>);
+    /**
+     * Build prompt for Pi
+     */
+    buildPayload(request: ConnectorRequest): string;
+    /**
+     * Parse Pi streaming output (JSON format)
+     * Pi's --mode json produces NDJSON lines
+     */
+    protected parseStreamingOutput(chunk: string, trajectory: TrajectoryStep[], onProgress?: ConnectorProgressCallback): void;
+    /**
+     * Parse a single JSON event from Pi output.
+     *
+     * Pi's --mode json NDJSON format uses these event types:
+     *  - session, agent_start, agent_end — lifecycle (ignored)
+     *  - turn_start, turn_end — turn boundaries
+     *  - message_start / message_end — full message with content blocks
+     *  - message_update — streaming deltas with assistantMessageEvent
+     */
+    private parsePiEvent;
+    /**
+     * Parse final output for Pi
+     */
+    parseResponse(data: {
+        stdout: string;
+        stderr: string;
+        exitCode: number;
+    }): TrajectoryStep[];
+    /**
+     * Flush remaining buffers when the subprocess stream ends.
+     */
+    protected onBeforeStreamEnd(trajectory: TrajectoryStep[], onProgress?: ConnectorProgressCallback): void;
+    /**
+     * Reset state and apply connectorConfig
+     */
+    execute(endpoint: string, request: ConnectorRequest, auth: ConnectorAuth, onProgress?: ConnectorProgressCallback, onRawEvent?: ConnectorRawEventCallback): Promise<ConnectorResponse>;
+    /**
+     * Health check - verify pi command exists
+     */
+    healthCheck(endpoint: string, auth: ConnectorAuth): Promise<boolean>;
+}
+/** Singleton instance for registry */
+export declare const piConnector: PiConnector;
+/**
+ * Create a Pi connector with the Agent Health package pre-configured
+ */
+export declare function createAgentHealthPiConnector(packagePath?: string): PiConnector;
+//# sourceMappingURL=PiConnector.d.ts.map

package/lib/dist/services/connectors/pi/PiConnector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PiConnector.d.ts","sourceRoot":"","sources":["../../../../../services/connectors/pi/PiConnector.ts"],"names":[],"mappings":"AAKA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAE9C,OAAO,EAAE,mBAAmB,EAAE,MAAM,sDAAsD,CAAC;AAC3F,OAAO,KAAK,EACV,aAAa,EACb,gBAAgB,EAChB,iBAAiB,EACjB,yBAAyB,EACzB,yBAAyB,EACzB,gBAAgB,EACjB,MAAM,6BAA6B,CAAC;AAErC;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7B,4EAA4E;IAC5E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,oDAAoD;IACpD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,+BAA+B;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,mDAAmD;IACnD,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,wCAAwC;IACxC,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;CAC3B;AAcD;;;GAGG;AACH,qBAAa,WAAY,SAAQ,mBAAmB;IAClD,QAAQ,CAAC,IAAI,EAAG,IAAI,CAAU;IAC9B,SAAkB,IAAI,iBAAiB;IAE9B,YAAY;;;MAAmD;IAExE,OAAO,CAAC,cAAc,CAAM;IAC5B,OAAO,CAAC,gBAAgB,CAAM;IAC9B,OAAO,CAAC,YAAY,CAAM;gBAEd,MAAM,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC;IAI9C;;OAEG;IACM,YAAY,CAAC,OAAO,EAAE,gBAAgB,GAAG,MAAM;IAoBxD;;;OAGG;cACgB,oBAAoB,CACrC,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,cAAc,EAAE,EAC5B,UAAU,CAAC,EAAE,yBAAyB,GACrC,IAAI;IA6BP;;;;;;;;OAQG;IACH,OAAO,CAAC,YAAY;IAiDpB;;OAEG;IACM,aAAa,CAAC,IAAI,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,cAAc,EAAE;IAiBpG;;OAEG;cACgB,iBAAiB,CAClC,UAAU,EAAE,cAAc,EAAE,EAC5B,UAAU,CAAC,EAAE,yBAAyB,GACrC,IAAI;IAgCP;;OAEG;IACY,OAAO,CACpB,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;IAmE7B;;OAEG;IACY,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;CAGpF;AAED,sCAAsC;AACtC,eAAO,MAAM,WAAW,aAAoB,CAAC;AAE7C;;GAEG;AACH,wBAAgB,4BAA4B,CAAC,WAAW,CAAC,EAAE,MAAM,GAAG,WAAW,CAQ9E"}

package/lib/dist/services/connectors/pi/PiConnector.js

@@ -0,0 +1,274 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ToolCallStatus } from '../../../types/index.js';
+import { SubprocessConnector } from '../../../services/connectors/subprocess/SubprocessConnector.js';
+/**
+ * Default Pi configuration
+ */
+const PI_DEFAULT_CONFIG = {
+    command: 'pi',
+    args: ['--print', '--mode', 'json'],
+    env: {},
+    inputMode: 'stdin',
+    outputParser: 'streaming',
+    timeout: 600000, // 10 minutes
+};
+/**
+ * Pi CLI Connector
+ * Invokes pi.dev as a subprocess for agent evaluation
+ */
+export class PiConnector extends SubprocessConnector {
+    constructor(config) {
+        super({ ...PI_DEFAULT_CONFIG, ...config });
+        this.type = 'pi';
+        this.name = 'Pi (pi.dev)';
+        this.traceContext = { propagateEnv: true, serviceName: 'pi-agent' };
+        this.piOutputBuffer = '';
+        this.piThinkingBuffer = '';
+        this.piTextBuffer = '';
+    }
+    /**
+     * Build prompt for Pi
+     */
+    buildPayload(request) {
+        const parts = [];
+        // Add system context if available
+        if (request.testCase.context && request.testCase.context.length > 0) {
+            parts.push('## Context');
+            for (const ctx of request.testCase.context) {
+                parts.push(`**${ctx.description}:**`);
+                parts.push(ctx.value);
+                parts.push('');
+            }
+        }
+        // Add the main prompt
+        parts.push('## Task');
+        parts.push(request.testCase.initialPrompt);
+        return parts.join('\n');
+    }
+    /**
+     * Parse Pi streaming output (JSON format)
+     * Pi's --mode json produces NDJSON lines
+     */
+    parseStreamingOutput(chunk, trajectory, onProgress) {
+        this.piOutputBuffer += chunk;
+        // Parse complete JSON lines (NDJSON format)
+        const lines = this.piOutputBuffer.split('\n');
+        this.piOutputBuffer = lines.pop() || ''; // Keep incomplete line
+        for (const line of lines) {
+            const trimmed = line.trim();
+            if (!trimmed)
+                continue;
+            try {
+                const event = JSON.parse(trimmed);
+                const steps = this.parsePiEvent(event);
+                for (const step of steps) {
+                    trajectory.push(step);
+                    onProgress?.(step);
+                }
+            }
+            catch {
+                // Not JSON, treat as plain text
+                if (trimmed) {
+                    const step = this.createStep('assistant', trimmed);
+                    trajectory.push(step);
+                    onProgress?.(step);
+                }
+            }
+        }
+    }
+    /**
+     * Parse a single JSON event from Pi output.
+     *
+     * Pi's --mode json NDJSON format uses these event types:
+     *  - session, agent_start, agent_end — lifecycle (ignored)
+     *  - turn_start, turn_end — turn boundaries
+     *  - message_start / message_end — full message with content blocks
+     *  - message_update — streaming deltas with assistantMessageEvent
+     */
+    parsePiEvent(event) {
+        const steps = [];
+        // message_end contains the full final message with all content blocks
+        if (event.type === 'message_end' && event.message?.role === 'assistant') {
+            const content = event.message.content;
+            if (Array.isArray(content)) {
+                for (const block of content) {
+                    if (block.type === 'thinking' && block.thinking) {
+                        steps.push(this.createStep('thinking', block.thinking));
+                    }
+                    else if (block.type === 'text' && block.text) {
+                        steps.push(this.createStep('assistant', block.text));
+                    }
+                    else if (block.type === 'tool_use') {
+                        steps.push(this.createStep('action', JSON.stringify(block.input || {}), {
+                            toolName: block.name,
+                            toolArgs: block.input,
+                        }));
+                    }
+                }
+            }
+        }
+        else if (event.type === 'message_update') {
+            // Streaming deltas — accumulate into buffers
+            const assistantEvent = event.assistantMessageEvent;
+            if (assistantEvent?.type === 'text_delta' && assistantEvent.delta) {
+                this.piTextBuffer += assistantEvent.delta;
+            }
+            else if (assistantEvent?.type === 'thinking_delta' && assistantEvent.delta) {
+                this.piThinkingBuffer += assistantEvent.delta;
+            }
+        }
+        else if (event.type === 'tool_result') {
+            const content = event.content || event.output || JSON.stringify(event);
+            steps.push(this.createStep('tool_result', typeof content === 'string' ? content : JSON.stringify(content), { status: event.is_error ? ToolCallStatus.FAILURE : ToolCallStatus.SUCCESS }));
+        }
+        else if (event.type === 'agent_end') {
+            // Flush any remaining buffers on agent_end
+            if (this.piThinkingBuffer) {
+                steps.push(this.createStep('thinking', this.piThinkingBuffer));
+                this.piThinkingBuffer = '';
+            }
+            if (this.piTextBuffer) {
+                steps.push(this.createStep('response', this.piTextBuffer));
+                this.piTextBuffer = '';
+            }
+        }
+        return steps;
+    }
+    /**
+     * Parse final output for Pi
+     */
+    parseResponse(data) {
+        const steps = [];
+        const response = data.stdout.trim();
+        if (response) {
+            steps.push(this.createStep('response', response));
+        }
+        if (data.exitCode !== 0 && data.stderr.trim()) {
+            steps.push(this.createStep('tool_result', `Error: ${data.stderr.trim()}`, {
+                status: ToolCallStatus.FAILURE,
+            }));
+        }
+        return steps;
+    }
+    /**
+     * Flush remaining buffers when the subprocess stream ends.
+     */
+    onBeforeStreamEnd(trajectory, onProgress) {
+        if (this.piOutputBuffer.trim()) {
+            try {
+                const event = JSON.parse(this.piOutputBuffer.trim());
+                const steps = this.parsePiEvent(event);
+                for (const step of steps) {
+                    trajectory.push(step);
+                    onProgress?.(step);
+                }
+            }
+            catch {
+                const step = this.createStep('assistant', this.piOutputBuffer.trim());
+                trajectory.push(step);
+                onProgress?.(step);
+            }
+            this.piOutputBuffer = '';
+        }
+        if (this.piThinkingBuffer) {
+            const step = this.createStep('thinking', this.piThinkingBuffer);
+            trajectory.push(step);
+            onProgress?.(step);
+            this.piThinkingBuffer = '';
+        }
+        if (this.piTextBuffer) {
+            const step = this.createStep('response', this.piTextBuffer);
+            trajectory.push(step);
+            onProgress?.(step);
+            this.piTextBuffer = '';
+        }
+    }
+    /**
+     * Reset state and apply connectorConfig
+     */
+    async execute(endpoint, request, auth, onProgress, onRawEvent) {
+        this.piOutputBuffer = '';
+        this.piThinkingBuffer = '';
+        this.piTextBuffer = '';
+        // Save original config
+        const originalArgs = this.config.args ? [...this.config.args] : [];
+        const originalEnv = this.config.env ? structuredClone(this.config.env) : {};
+        const originalTimeout = this.config.timeout;
+        const originalWorkingDir = this.config.workingDir;
+        // Apply connectorConfig
+        const piConfig = request.connectorConfig;
+        if (piConfig) {
+            if (piConfig.env) {
+                this.config.env = { ...this.config.env, ...piConfig.env };
+            }
+            if (piConfig.timeout !== undefined) {
+                this.config.timeout = piConfig.timeout;
+            }
+            if (piConfig.workingDir) {
+                this.config.workingDir = piConfig.workingDir;
+            }
+            // Build additional args from config
+            const extraArgs = [];
+            if (piConfig.packagePath) {
+                // Pi uses --skill and --extension to load package components
+                extraArgs.push('--skill', `${piConfig.packagePath}/skills/*`);
+                extraArgs.push('--extension', `${piConfig.packagePath}/extensions/agent-health.ts`);
+                extraArgs.push('--append-system-prompt', `${piConfig.packagePath}/prompts/agent-health.md`);
+            }
+            if (piConfig.model) {
+                extraArgs.push('--model', piConfig.model);
+            }
+            if (piConfig.additionalArgs) {
+                extraArgs.push(...piConfig.additionalArgs);
+            }
+            if (extraArgs.length > 0) {
+                this.config.args = [...(this.config.args || []), ...extraArgs];
+            }
+        }
+        // Pass --model flag from request if specified
+        if (request.modelId) {
+            this.config.args = [...(this.config.args || []), '--model', request.modelId];
+        }
+        // Inherit AWS credentials
+        if (process.env.AWS_PROFILE) {
+            this.config.env = { ...this.config.env, AWS_PROFILE: process.env.AWS_PROFILE };
+        }
+        if (process.env.AWS_REGION) {
+            this.config.env = { ...this.config.env, AWS_REGION: process.env.AWS_REGION };
+        }
+        try {
+            return await super.execute(endpoint, request, auth, onProgress, onRawEvent);
+        }
+        finally {
+            // Restore config
+            this.config.args = originalArgs;
+            this.config.env = originalEnv;
+            this.config.timeout = originalTimeout;
+            this.config.workingDir = originalWorkingDir;
+        }
+    }
+    /**
+     * Health check - verify pi command exists
+     */
+    async healthCheck(endpoint, auth) {
+        return super.healthCheck(endpoint || 'pi', auth);
+    }
+}
+/** Singleton instance for registry */
+export const piConnector = new PiConnector();
+/**
+ * Create a Pi connector with the Agent Health package pre-configured
+ */
+export function createAgentHealthPiConnector(packagePath) {
+    const args = ['--print', '--mode', 'json'];
+    if (packagePath) {
+        args.push('--skill', `${packagePath}/skills/*`);
+        args.push('--extension', `${packagePath}/extensions/agent-health.ts`);
+        args.push('--append-system-prompt', `${packagePath}/prompts/agent-health.md`);
+    }
+    return new PiConnector({ args });
+}
+//# sourceMappingURL=PiConnector.js.map

package/lib/dist/services/connectors/pi/PiConnector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PiConnector.js","sourceRoot":"","sources":["../../../../../services/connectors/pi/PiConnector.ts"],"names":[],"mappings":"AAAA;;;GAGG;AASH,OAAO,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,sDAAsD,CAAC;AA4B3F;;GAEG;AACH,MAAM,iBAAiB,GAA8B;IACnD,OAAO,EAAE,IAAI;IACb,IAAI,EAAE,CAAC,SAAS,EAAE,QAAQ,EAAE,MAAM,CAAC;IACnC,GAAG,EAAE,EAAE;IACP,SAAS,EAAE,OAAO;IAClB,YAAY,EAAE,WAAW;IACzB,OAAO,EAAE,MAAM,EAAE,aAAa;CAC/B,CAAC;AAEF;;;GAGG;AACH,MAAM,OAAO,WAAY,SAAQ,mBAAmB;IAUlD,YAAY,MAAkC;QAC5C,KAAK,CAAC,EAAE,GAAG,iBAAiB,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QAVpC,SAAI,GAAG,IAAa,CAAC;QACZ,SAAI,GAAG,aAAa,CAAC;QAE9B,iBAAY,GAAG,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;QAEhE,mBAAc,GAAG,EAAE,CAAC;QACpB,qBAAgB,GAAG,EAAE,CAAC;QACtB,iBAAY,GAAG,EAAE,CAAC;IAI1B,CAAC;IAED;;OAEG;IACM,YAAY,CAAC,OAAyB;QAC7C,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,kCAAkC;QAClC,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpE,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACzB,KAAK,MAAM,GAAG,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBAC3C,KAAK,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,WAAW,KAAK,CAAC,CAAC;gBACtC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACjB,CAAC;QACH,CAAC;QAED,sBAAsB;QACtB,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QAE3C,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;;OAGG;IACgB,oBAAoB,CACrC,KAAa,EACb,UAA4B,EAC5B,UAAsC;QAEtC,IAAI,CAAC,cAAc,IAAI,KAAK,CAAC;QAE7B,4CAA4C;QAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,uBAAuB;QAEhE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAC5B,IAAI,CAAC,OAAO;gBAAE,SAAS;YAEvB,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAClC,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;gBACrB,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,gCAAgC;gBAChC,IAAI,OAAO,EAAE,CAAC;oBACZ,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;oBACnD,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;gBACrB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;;;;OAQG;IACK,YAAY,CAAC,KAAU;QAC7B,MAAM,KAAK,GAAqB,EAAE,CAAC;QAEnC,sEAAsE;QACtE,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,OAAO,EAAE,IAAI,KAAK,WAAW,EAAE,CAAC;YACxE,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;YACtC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC3B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;oBAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;wBAChD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;oBAC1D,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBAC/C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;oBACvD,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACrC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE;4BACtE,QAAQ,EAAE,KAAK,CAAC,IAAI;4BACpB,QAAQ,EAAE,KAAK,CAAC,KAAK;yBACtB,CAAC,CAAC,CAAC;oBACN,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;YAC3C,6CAA6C;YAC7C,MAAM,cAAc,GAAG,KAAK,CAAC,qBAAqB,CAAC;YACnD,IAAI,cAAc,EAAE,IAAI,KAAK,YAAY,IAAI,cAAc,CAAC,KAAK,EAAE,CAAC;gBAClE,IAAI,CAAC,YAAY,IAAI,cAAc,CAAC,KAAK,CAAC;YAC5C,CAAC;iBAAM,IAAI,cAAc,EAAE,IAAI,KAAK,gBAAgB,IAAI,cAAc,CAAC,KAAK,EAAE,CAAC;gBAC7E,IAAI,CAAC,gBAAgB,IAAI,cAAc,CAAC,KAAK,CAAC;YAChD,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACxC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACvE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,EACtC,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAC/D,EAAE,MAAM,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,EAAE,CAC7E,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACtC,2CAA2C;YAC3C,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBAC1B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC;gBAC/D,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;YAC7B,CAAC;YACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;gBAC3D,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;YACzB,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACM,aAAa,CAAC,IAA0D;QAC/E,MAAM,KAAK,GAAqB,EAAE,CAAC;QACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAEpC,IAAI,QAAQ,EAAE,CAAC;YACb,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YAC9C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,EAAE;gBACxE,MAAM,EAAE,cAAc,CAAC,OAAO;aAC/B,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACgB,iBAAiB,CAClC,UAA4B,EAC5B,UAAsC;QAEtC,IAAI,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,EAAE,CAAC;YAC/B,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;gBACrD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBACvC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;gBACrB,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC;gBACtE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC;YACD,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QAC3B,CAAC;QAED,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAChE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;YACnB,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC7B,CAAC;QAED,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;YAC5D,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;YACnB,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QACzB,CAAC;IACH,CAAC;IAED;;OAEG;IACM,KAAK,CAAC,OAAO,CACpB,QAAgB,EAChB,OAAyB,EACzB,IAAmB,EACnB,UAAsC,EACtC,UAAsC;QAEtC,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QAEvB,uBAAuB;QACvB,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACnE,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5E,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QAC5C,MAAM,kBAAkB,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAElD,wBAAwB;QACxB,MAAM,QAAQ,GAAG,OAAO,CAAC,eAAgD,CAAC;QAC1E,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACjB,IAAI,CAAC,MAAM,CAAC,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC;YAC5D,CAAC;YACD,IAAI,QAAQ,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;gBACnC,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;YACzC,CAAC;YACD,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;gBACxB,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,QAAQ,CAAC,UAAU,CAAC;YAC/C,CAAC;YAED,oCAAoC;YACpC,MAAM,SAAS,GAAa,EAAE,CAAC;YAC/B,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;gBACzB,6DAA6D;gBAC7D,SAAS,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,QAAQ,CAAC,WAAW,WAAW,CAAC,CAAC;gBAC9D,SAAS,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,QAAQ,CAAC,WAAW,6BAA6B,CAAC,CAAC;gBACpF,SAAS,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,QAAQ,CAAC,WAAW,0BAA0B,CAAC,CAAC;YAC9F,CAAC;YACD,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;gBACnB,SAAS,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC5C,CAAC;YACD,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;gBAC5B,SAAS,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,cAAc,CAAC,CAAC;YAC7C,CAAC;YACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzB,IAAI,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,GAAG,SAAS,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QAED,8CAA8C;QAC9C,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAC/E,CAAC;QAED,0BAA0B;QAC1B,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;YAC5B,IAAI,CAAC,MAAM,CAAC,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;QACjF,CAAC;QACD,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;QAC/E,CAAC;QAED,IAAI,CAAC;YACH,OAAO,MAAM,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAC9E,CAAC;gBAAS,CAAC;YACT,iBAAiB;YACjB,IAAI,CAAC,MAAM,CAAC,IAAI,GAAG,YAAY,CAAC;YAChC,IAAI,CAAC,MAAM,CAAC,GAAG,GAAG,WAAW,CAAC;YAC9B,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,eAAe,CAAC;YACtC,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,kBAAkB,CAAC;QAC9C,CAAC;IACH,CAAC;IAED;;OAEG;IACM,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,IAAmB;QAC9D,OAAO,KAAK,CAAC,WAAW,CAAC,QAAQ,IAAI,IAAI,EAAE,IAAI,CAAC,CAAC;IACnD,CAAC;CACF;AAED,sCAAsC;AACtC,MAAM,CAAC,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;AAE7C;;GAEG;AACH,MAAM,UAAU,4BAA4B,CAAC,WAAoB;IAC/D,MAAM,IAAI,GAAG,CAAC,SAAS,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC3C,IAAI,WAAW,EAAE,CAAC;QAChB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,WAAW,WAAW,CAAC,CAAC;QAChD,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,WAAW,6BAA6B,CAAC,CAAC;QACtE,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE,GAAG,WAAW,0BAA0B,CAAC,CAAC;IAChF,CAAC;IACD,OAAO,IAAI,WAAW,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;AACnC,CAAC"}

package/lib/dist/services/connectors/registry.d.ts

@@ -0,0 +1,57 @@
+import type { AgentConnector, ConnectorProtocol, ConnectorRegistry, AgentConfigWithConnector } from '../../services/connectors/types.js';
+/**
+ * Implementation of ConnectorRegistry
+ * Manages registration and lookup of connector implementations
+ */
+declare class ConnectorRegistryImpl implements ConnectorRegistry {
+    private connectors;
+    /**
+     * Register a connector implementation
+     * @throws Error if connector with same type is already registered
+     */
+    register(connector: AgentConnector): void;
+    /**
+     * Get a connector by protocol type
+     */
+    get(type: ConnectorProtocol): AgentConnector | undefined;
+    /**
+     * Get all registered connectors
+     */
+    getAll(): AgentConnector[];
+    /**
+     * Check if a connector is registered
+     */
+    has(type: ConnectorProtocol): boolean;
+    /**
+     * Get connector for an agent config
+     * Handles backwards compatibility with legacy configs
+     *
+     * Resolution order:
+     * 1. If endpoint starts with 'mock://', use mock connector
+     * 2. If connectorType is specified, use that
+     * 3. Default to 'agui-streaming'
+     */
+    getForAgent(agent: AgentConfigWithConnector): AgentConnector;
+    /**
+     * Clear all registered connectors (useful for testing)
+     */
+    clear(): void;
+    /**
+     * Get list of registered connector types
+     */
+    getRegisteredTypes(): ConnectorProtocol[];
+}
+/**
+ * Singleton instance of the connector registry
+ */
+export declare const connectorRegistry: ConnectorRegistryImpl;
+/**
+ * Helper function to get connector for an agent
+ */
+export declare function getConnectorForAgent(agent: AgentConfigWithConnector): AgentConnector;
+/**
+ * Helper function to register a connector
+ */
+export declare function registerConnector(connector: AgentConnector): void;
+export {};
+//# sourceMappingURL=registry.d.ts.map

package/lib/dist/services/connectors/registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../../../../services/connectors/registry.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,cAAc,EACd,iBAAiB,EACjB,iBAAiB,EACjB,wBAAwB,EACzB,MAAM,6BAA6B,CAAC;AAOrC;;;GAGG;AACH,cAAM,qBAAsB,YAAW,iBAAiB;IACtD,OAAO,CAAC,UAAU,CAAqD;IAEvE;;;OAGG;IACH,QAAQ,CAAC,SAAS,EAAE,cAAc,GAAG,IAAI;IASzC;;OAEG;IACH,GAAG,CAAC,IAAI,EAAE,iBAAiB,GAAG,cAAc,GAAG,SAAS;IAIxD;;OAEG;IACH,MAAM,IAAI,cAAc,EAAE;IAI1B;;OAEG;IACH,GAAG,CAAC,IAAI,EAAE,iBAAiB,GAAG,OAAO;IAIrC;;;;;;;;OAQG;IACH,WAAW,CAAC,KAAK,EAAE,wBAAwB,GAAG,cAAc;IA+B5D;;OAEG;IACH,KAAK,IAAI,IAAI;IAIb;;OAEG;IACH,kBAAkB,IAAI,iBAAiB,EAAE;CAG1C;AAED;;GAEG;AACH,eAAO,MAAM,iBAAiB,uBAA8B,CAAC;AAE7D;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,wBAAwB,GAAG,cAAc,CAEpF;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,SAAS,EAAE,cAAc,GAAG,IAAI,CAEjE"}