@opensearch-project/agent-health 0.3.0 → 0.5.2

package/lib/dist/services/connectors/kiro/KiroConnector.js

@@ -0,0 +1,158 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Kiro CLI Connector
+ *
+ * Specialized subprocess connector for `kiro-cli chat --no-interactive`.
+ *
+ * Why this exists
+ * ---------------
+ * The base `SubprocessConnector` parses Kiro's stdout as a stream of plain
+ * text lines (assistant chunks). That works for live UI streaming but loses
+ * all evidence of which tools the agent actually invoked. The judge then has
+ * nothing to evidence-check against — a polished hallucinated narrative
+ * scores the same as a tool-grounded investigation.
+ *
+ * Kiro DOES emit structured tool events, but on stderr in a line-oriented
+ * format:
+ *
+ *     [tool] Running: <command_or_invocation>
+ *     [tool] status: <Completed|Failed|Cancelled|...>
+ *
+ * This connector intercepts those stderr lines and converts them into
+ * proper `action` + `tool_result` trajectory steps, so:
+ *   - The judge can see what tools ran (and whether they succeeded).
+ *   - SDK matchers like `expect(traj).to.haveCalledTool('glob')` actually
+ *     have something to assert against.
+ *   - A future "RCA + Grounded" evaluator can require that any factual
+ *     claim be backed by at least one `tool_result` step.
+ *
+ * Stdout streaming behaviour (line-by-line `assistant` chunks plus a final
+ * consolidated `response`) is inherited unchanged from the base class.
+ */
+import { ToolCallStatus } from '../../../types/index.js';
+import { SubprocessConnector } from '../../../services/connectors/subprocess/SubprocessConnector.js';
+/**
+ * Default Kiro configuration.
+ *
+ * `--agent-engine v2` is required so slash commands like `/<slash-command> ...`
+ * route through the agent registry. The default (v1) non-interactive parser
+ * treats `/<word>` as an unknown subcommand and exits early.
+ *
+ * `--no-interactive` is required for headless invocation but suppresses
+ * sqlite session persistence, which is why we rely on stderr `[tool]`
+ * markers rather than the on-disk conversations_v2 table.
+ */
+const KIRO_DEFAULT_CONFIG = {
+    command: 'kiro-cli',
+    args: [
+        'chat',
+        '--agent-engine', 'v2',
+        '--no-interactive',
+        '--trust-all-tools',
+    ],
+    env: {},
+    inputMode: 'arg',
+    outputParser: 'streaming',
+    timeout: 600000, // 10 minutes
+};
+export class KiroConnector extends SubprocessConnector {
+    constructor(config) {
+        super({ ...KIRO_DEFAULT_CONFIG, ...config });
+        this.type = 'kiro';
+        this.name = 'Kiro CLI';
+        this.traceContext = { propagateEnv: true, serviceName: 'kiro-agent' };
+        /** Carry-over for a partial last line on stderr stream */
+        this.stderrLineBuffer = '';
+        /** Tool name from the last unmatched `[tool] Running:` line, used to
+         *  attach the same toolName to the corresponding `[tool] status:` row. */
+        this.pendingToolName = null;
+    }
+    /**
+     * Convert kiro-cli's stderr `[tool]` markers into trajectory steps.
+     *
+     * Each chunk may straddle line boundaries, so we accumulate into
+     * `stderrLineBuffer` and only consume complete lines (the trailing
+     * partial line is held until the next chunk or `onBeforeStreamEnd`).
+     */
+    parseStderrChunk(chunk, trajectory, onProgress) {
+        this.stderrLineBuffer += chunk;
+        const lines = this.stderrLineBuffer.split('\n');
+        this.stderrLineBuffer = lines.pop() || '';
+        for (const raw of lines) {
+            this.processStderrLine(raw, trajectory, onProgress);
+        }
+    }
+    /**
+     * Drain any held partial line from the stderr buffer and reset
+     * per-run state, then defer to the base class to flush the
+     * consolidated stdout response step.
+     */
+    onBeforeStreamEnd(trajectory, onProgress) {
+        if (this.stderrLineBuffer.trim()) {
+            this.processStderrLine(this.stderrLineBuffer, trajectory, onProgress);
+        }
+        this.stderrLineBuffer = '';
+        this.pendingToolName = null;
+        // Base class emits the consolidated `response` step from buffered stdout.
+        super.onBeforeStreamEnd(trajectory, onProgress);
+    }
+    /**
+     * Parse a single stderr line. Recognised forms:
+     *
+     *   [tool] Running: <command>          -> emit `action` step
+     *   [tool] status: Completed           -> emit `tool_result` (SUCCESS)
+     *   [tool] status: Failed              -> emit `tool_result` (FAILURE)
+     *   [tool] status: <anything else>     -> emit `tool_result` (FAILURE)
+     *
+     * Other stderr lines (AWS SDK warnings, telemetry info, etc.) are ignored.
+     */
+    processStderrLine(raw, trajectory, onProgress) {
+        // Drop NUL/control bytes that occasionally slip into stderr
+        const line = raw.replace(/[\x00-\x08\x0b-\x1f\x7f]/g, '').trim();
+        if (!line.startsWith('[tool]'))
+            return;
+        const m = line.match(/^\[tool\]\s+(Running|status):\s*(.*)$/i);
+        if (!m)
+            return;
+        const kind = m[1].toLowerCase();
+        const payload = m[2].trim();
+        if (kind === 'running') {
+            // Best-effort tool name: first whitespace-separated token. For
+            // shell tool calls this is the command (e.g. `wc`, `find`); for
+            // first-party tools (e.g. `glob`, `use_subagent`) it's the tool
+            // name itself. We keep the full payload as `command` in toolArgs
+            // so SDK matchers and judges can pattern-match on it.
+            const toolName = payload.split(/\s+/)[0] || 'unknown';
+            const toolArgs = { command: payload };
+            this.pendingToolName = toolName;
+            const step = this.createStep('action', JSON.stringify(toolArgs), {
+                toolName,
+                toolArgs,
+            });
+            trajectory.push(step);
+            onProgress?.(step);
+            return;
+        }
+        if (kind === 'status') {
+            const success = /^(Completed|Success)/i.test(payload);
+            const step = this.createStep('tool_result', `status: ${payload}`, {
+                status: success ? ToolCallStatus.SUCCESS : ToolCallStatus.FAILURE,
+                toolName: this.pendingToolName ?? undefined,
+            });
+            trajectory.push(step);
+            onProgress?.(step);
+            this.pendingToolName = null;
+        }
+    }
+    async healthCheck(endpoint, auth) {
+        return super.healthCheck(endpoint || 'kiro-cli', auth);
+    }
+}
+/**
+ * Default instance for convenience.
+ */
+export const kiroConnector = new KiroConnector();
+//# sourceMappingURL=KiroConnector.js.map

package/lib/dist/services/connectors/kiro/KiroConnector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"KiroConnector.js","sourceRoot":"","sources":["../../../../../services/connectors/kiro/KiroConnector.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAEzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,sDAAsD,CAAC;AAO3F;;;;;;;;;;GAUG;AACH,MAAM,mBAAmB,GAA8B;IACrD,OAAO,EAAE,UAAU;IACnB,IAAI,EAAE;QACJ,MAAM;QACN,gBAAgB,EAAE,IAAI;QACtB,kBAAkB;QAClB,mBAAmB;KACpB;IACD,GAAG,EAAE,EAAE;IACP,SAAS,EAAE,KAAK;IAChB,YAAY,EAAE,WAAW;IACzB,OAAO,EAAE,MAAM,EAAE,aAAa;CAC/B,CAAC;AAEF,MAAM,OAAO,aAAc,SAAQ,mBAAmB;IAapD,YAAY,MAAkC;QAC5C,KAAK,CAAC,EAAE,GAAG,mBAAmB,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QAb7B,SAAI,GAAG,MAAe,CAAC;QACvB,SAAI,GAAG,UAAU,CAAC;QAE3B,iBAAY,GAAG,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;QAE1E,0DAA0D;QAClD,qBAAgB,GAAG,EAAE,CAAC;QAE9B;kFAC0E;QAClE,oBAAe,GAAkB,IAAI,CAAC;IAI9C,CAAC;IAED;;;;;;OAMG;IACgB,gBAAgB,CACjC,KAAa,EACb,UAA4B,EAC5B,UAAsC;QAEtC,IAAI,CAAC,gBAAgB,IAAI,KAAK,CAAC;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QAE1C,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,iBAAiB,CAAC,GAAG,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED;;;;OAIG;IACgB,iBAAiB,CAClC,UAA4B,EAC5B,UAAsC;QAEtC,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,EAAE,CAAC;YACjC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,gBAAgB,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAE5B,0EAA0E;QAC1E,KAAK,CAAC,iBAAiB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IAClD,CAAC;IAED;;;;;;;;;OASG;IACK,iBAAiB,CACvB,GAAW,EACX,UAA4B,EAC5B,UAAsC;QAEtC,4DAA4D;QAC5D,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACjE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,OAAO;QAEvC,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC/D,IAAI,CAAC,CAAC;YAAE,OAAO;QAEf,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE5B,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,+DAA+D;YAC/D,gEAAgE;YAChE,gEAAgE;YAChE,iEAAiE;YACjE,sDAAsD;YACtD,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;YACtD,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;YACtC,IAAI,CAAC,eAAe,GAAG,QAAQ,CAAC;YAEhC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE;gBAC/D,QAAQ;gBACR,QAAQ;aACT,CAAC,CAAC;YACH,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;YACtB,MAAM,OAAO,GAAG,uBAAuB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACtD,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAC1B,aAAa,EACb,WAAW,OAAO,EAAE,EACpB;gBACE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO;gBACjE,QAAQ,EAAE,IAAI,CAAC,eAAe,IAAI,SAAS;aAC5C,CACF,CAAC;YACF,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;YACnB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;QAC9B,CAAC;IACH,CAAC;IAEQ,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,IAAmB;QAC9D,OAAO,KAAK,CAAC,WAAW,CAAC,QAAQ,IAAI,UAAU,EAAE,IAAI,CAAC,CAAC;IACzD,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC"}

package/lib/dist/services/connectors/langgraph/LangGraphConnector.d.ts

@@ -0,0 +1,36 @@
+/**
+ * LangGraph REST Connector
+ * Handles non-streaming REST calls to LangGraph agents
+ *
+ * For AG-UI streaming LangGraph agents, use the existing agui-streaming connector.
+ * This connector is for LangGraph instances exposed via direct REST API.
+ */
+import type { TrajectoryStep } from '../../../types/index.js';
+import { BaseConnector } from '../../../services/connectors/base/BaseConnector.js';
+import type { ConnectorAuth, ConnectorRequest, ConnectorResponse, ConnectorProgressCallback, ConnectorRawEventCallback } from '../../../services/connectors/types.js';
+/**
+ * LangGraph REST Connector
+ *
+ * Communicates with LangGraph agents via the LangGraph REST API.
+ * Uses the /invoke endpoint for synchronous execution.
+ * For streaming LangGraph agents, use the agui-streaming connector instead.
+ *
+ * Configuration:
+ * - endpoint: Base URL of the LangGraph server (e.g., "http://localhost:8123")
+ * - connectorConfig.graphId: Graph/assistant ID (defaults to "agent")
+ * - connectorConfig.threadId: Optional thread ID for multi-turn
+ */
+export declare class LangGraphConnector extends BaseConnector {
+    readonly type: "langgraph";
+    readonly name = "LangGraph (REST)";
+    traceContext: {
+        propagateHeader: boolean;
+    };
+    readonly supportsStreaming = false;
+    buildPayload(request: ConnectorRequest): any;
+    execute(endpoint: string, request: ConnectorRequest, auth: ConnectorAuth, onProgress?: ConnectorProgressCallback, onRawEvent?: ConnectorRawEventCallback): Promise<ConnectorResponse>;
+    parseResponse(data: any): TrajectoryStep[];
+    healthCheck(endpoint: string, auth: ConnectorAuth): Promise<boolean>;
+}
+export declare const langgraphConnector: LangGraphConnector;
+//# sourceMappingURL=LangGraphConnector.d.ts.map

package/lib/dist/services/connectors/langgraph/LangGraphConnector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"LangGraphConnector.d.ts","sourceRoot":"","sources":["../../../../../services/connectors/langgraph/LangGraphConnector.ts"],"names":[],"mappings":"AAKA;;;;;;GAMG;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;AAErC;;;;;;;;;;;GAWG;AACH,qBAAa,kBAAmB,SAAQ,aAAa;IACnD,QAAQ,CAAC,IAAI,EAAG,WAAW,CAAU;IACrC,QAAQ,CAAC,IAAI,sBAAsB;IAE1B,YAAY;;MAA6B;IAClD,QAAQ,CAAC,iBAAiB,SAAS;IAEnC,YAAY,CAAC,OAAO,EAAE,gBAAgB,GAAG,GAAG;IAoBtC,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;IAgD7B,aAAa,CAAC,IAAI,EAAE,GAAG,GAAG,cAAc,EAAE;IA2EpC,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;CAoB3E;AAED,eAAO,MAAM,kBAAkB,oBAA2B,CAAC"}

package/lib/dist/services/connectors/langgraph/LangGraphConnector.js

@@ -0,0 +1,175 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseConnector } from '../../../services/connectors/base/BaseConnector.js';
+/**
+ * LangGraph REST Connector
+ *
+ * Communicates with LangGraph agents via the LangGraph REST API.
+ * Uses the /invoke endpoint for synchronous execution.
+ * For streaming LangGraph agents, use the agui-streaming connector instead.
+ *
+ * Configuration:
+ * - endpoint: Base URL of the LangGraph server (e.g., "http://localhost:8123")
+ * - connectorConfig.graphId: Graph/assistant ID (defaults to "agent")
+ * - connectorConfig.threadId: Optional thread ID for multi-turn
+ */
+export class LangGraphConnector extends BaseConnector {
+    constructor() {
+        super(...arguments);
+        this.type = 'langgraph';
+        this.name = 'LangGraph (REST)';
+        this.traceContext = { propagateHeader: true };
+        this.supportsStreaming = false;
+    }
+    buildPayload(request) {
+        const config = request.connectorConfig || {};
+        return {
+            input: {
+                messages: [
+                    {
+                        role: 'user',
+                        content: request.testCase.initialPrompt,
+                    },
+                ],
+            },
+            config: {
+                configurable: {
+                    ...(request.modelId && { model: request.modelId }),
+                    ...config.configurable,
+                },
+            },
+        };
+    }
+    async execute(endpoint, request, auth, onProgress, onRawEvent) {
+        const payload = request.payload || this.buildPayload(request);
+        const headers = this.buildAuthHeaders(auth);
+        this.injectTraceparentHeaders(headers);
+        const config = request.connectorConfig || {};
+        const graphId = config.graphId || 'agent';
+        // Build the invoke URL
+        const baseUrl = endpoint.replace(/\/+$/, '');
+        const threadId = config.threadId || request.threadId;
+        const invokeUrl = threadId
+            ? `${baseUrl}/threads/${threadId}/runs/wait`
+            : `${baseUrl}/assistants/${graphId}/invoke`;
+        this.debug('Executing LangGraph request');
+        this.debug('URL:', invokeUrl);
+        const response = await fetch(invokeUrl, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                ...headers,
+            },
+            body: JSON.stringify(payload),
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            throw new Error(`LangGraph 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.run_id || data.thread_id || threadId || null,
+            rawEvents: [data],
+            metadata: {
+                graphId,
+                threadId: data.thread_id || threadId,
+            },
+        };
+    }
+    parseResponse(data) {
+        const steps = [];
+        // LangGraph invoke response format: { output: { messages: [...] } }
+        // Or thread-based: { values: { messages: [...] } }
+        const messages = data.output?.messages || data.values?.messages || data.messages || [];
+        for (const msg of messages) {
+            const role = msg.type || msg.role;
+            const content = typeof msg.content === 'string'
+                ? msg.content
+                : Array.isArray(msg.content)
+                    ? msg.content.map((c) => c.text || JSON.stringify(c)).join('\n')
+                    : JSON.stringify(msg.content);
+            if (role === 'ai' || role === 'assistant') {
+                // Check for tool calls in the AI message
+                if (msg.tool_calls && Array.isArray(msg.tool_calls) && msg.tool_calls.length > 0) {
+                    for (const toolCall of msg.tool_calls) {
+                        steps.push(this.createStep('action', `Calling ${toolCall.name}...`, {
+                            toolName: toolCall.name,
+                            toolArgs: toolCall.args,
+                        }));
+                    }
+                }
+                // AI reasoning or intermediate response
+                if (content && !msg.tool_calls?.length) {
+                    steps.push(this.createStep('assistant', content));
+                }
+            }
+            else if (role === 'tool') {
+                steps.push(this.createStep('tool_result', content, {
+                    toolName: msg.name,
+                    status: 'SUCCESS',
+                }));
+            }
+        }
+        // The last AI message without tool calls is the final response
+        const lastAssistant = [...steps].reverse().find(s => s.type === 'assistant');
+        if (lastAssistant) {
+            lastAssistant.type = 'response';
+        }
+        // Handle intermediate_steps format (older LangGraph versions)
+        if (data.intermediate_steps && Array.isArray(data.intermediate_steps)) {
+            for (const [action, observation] of data.intermediate_steps) {
+                if (action?.tool) {
+                    steps.push(this.createStep('action', `Calling ${action.tool}...`, {
+                        toolName: action.tool,
+                        toolArgs: action.tool_input,
+                    }));
+                }
+                if (observation !== undefined) {
+                    const obsContent = typeof observation === 'string' ? observation : JSON.stringify(observation);
+                    steps.push(this.createStep('tool_result', obsContent, {
+                        status: 'SUCCESS',
+                    }));
+                }
+            }
+        }
+        // Handle direct output field
+        if (data.output && typeof data.output === 'string' && steps.length === 0) {
+            steps.push(this.createStep('response', data.output));
+        }
+        // Fallback
+        if (steps.length === 0 && data) {
+            steps.push(this.createStep('response', JSON.stringify(data, null, 2)));
+        }
+        return steps;
+    }
+    async healthCheck(endpoint, auth) {
+        try {
+            const headers = this.buildAuthHeaders(auth);
+            const baseUrl = endpoint.replace(/\/+$/, '');
+            const response = await fetch(`${baseUrl}/ok`, {
+                method: 'GET',
+                headers,
+            });
+            return response.ok;
+        }
+        catch {
+            // Try root endpoint as fallback
+            try {
+                const headers = this.buildAuthHeaders(auth);
+                const response = await fetch(endpoint, { method: 'GET', headers });
+                return response.ok;
+            }
+            catch {
+                return false;
+            }
+        }
+    }
+}
+export const langgraphConnector = new LangGraphConnector();
+//# sourceMappingURL=LangGraphConnector.js.map

package/lib/dist/services/connectors/langgraph/LangGraphConnector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LangGraphConnector.js","sourceRoot":"","sources":["../../../../../services/connectors/langgraph/LangGraphConnector.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAWH,OAAO,EAAE,aAAa,EAAE,MAAM,0CAA0C,CAAC;AASzE;;;;;;;;;;;GAWG;AACH,MAAM,OAAO,kBAAmB,SAAQ,aAAa;IAArD;;QACW,SAAI,GAAG,WAAoB,CAAC;QAC5B,SAAI,GAAG,kBAAkB,CAAC;QAE1B,iBAAY,GAAG,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC;QACzC,sBAAiB,GAAG,KAAK,CAAC;IA2KrC,CAAC;IAzKC,YAAY,CAAC,OAAyB;QACpC,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,IAAI,EAAE,CAAC;QAC7C,OAAO;YACL,KAAK,EAAE;gBACL,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,aAAa;qBACxC;iBACF;aACF;YACD,MAAM,EAAE;gBACN,YAAY,EAAE;oBACZ,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;oBAClD,GAAG,MAAM,CAAC,YAAY;iBACvB;aACF;SACF,CAAC;IACJ,CAAC;IAED,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,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,IAAI,EAAE,CAAC;QAC7C,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,OAAO,CAAC;QAE1C,uBAAuB;QACvB,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAC7C,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC;QACrD,MAAM,SAAS,GAAG,QAAQ;YACxB,CAAC,CAAC,GAAG,OAAO,YAAY,QAAQ,YAAY;YAC5C,CAAC,CAAC,GAAG,OAAO,eAAe,OAAO,SAAS,CAAC;QAE9C,IAAI,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;QAC1C,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAE9B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,SAAS,EAAE;YACtC,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,6BAA6B,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;QACjF,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,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,MAAM,IAAI,IAAI,CAAC,SAAS,IAAI,QAAQ,IAAI,IAAI;YACxD,SAAS,EAAE,CAAC,IAAI,CAAC;YACjB,QAAQ,EAAE;gBACR,OAAO;gBACP,QAAQ,EAAE,IAAI,CAAC,SAAS,IAAI,QAAQ;aACrC;SACF,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,IAAS;QACrB,MAAM,KAAK,GAAqB,EAAE,CAAC;QAEnC,oEAAoE;QACpE,mDAAmD;QACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,EAAE,QAAQ,IAAI,IAAI,CAAC,MAAM,EAAE,QAAQ,IAAI,IAAI,CAAC,QAAQ,IAAI,EAAE,CAAC;QAEvF,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,CAAC;YAClC,MAAM,OAAO,GAAG,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ;gBAC7C,CAAC,CAAC,GAAG,CAAC,OAAO;gBACb,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;oBAC1B,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;oBACrE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAElC,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC1C,yCAAyC;gBACzC,IAAI,GAAG,CAAC,UAAU,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACjF,KAAK,MAAM,QAAQ,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;wBACtC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,WAAW,QAAQ,CAAC,IAAI,KAAK,EAAE;4BAClE,QAAQ,EAAE,QAAQ,CAAC,IAAI;4BACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;yBACxB,CAAC,CAAC,CAAC;oBACN,CAAC;gBACH,CAAC;gBAED,wCAAwC;gBACxC,IAAI,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;oBACvC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;iBAAM,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC3B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,OAAO,EAAE;oBACjD,QAAQ,EAAE,GAAG,CAAC,IAAI;oBAClB,MAAM,EAAE,SAA2B;iBACpC,CAAC,CAAC,CAAC;YACN,CAAC;QACH,CAAC;QAED,+DAA+D;QAC/D,MAAM,aAAa,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QAC7E,IAAI,aAAa,EAAE,CAAC;YAClB,aAAa,CAAC,IAAI,GAAG,UAAU,CAAC;QAClC,CAAC;QAED,8DAA8D;QAC9D,IAAI,IAAI,CAAC,kBAAkB,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACtE,KAAK,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC5D,IAAI,MAAM,EAAE,IAAI,EAAE,CAAC;oBACjB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,WAAW,MAAM,CAAC,IAAI,KAAK,EAAE;wBAChE,QAAQ,EAAE,MAAM,CAAC,IAAI;wBACrB,QAAQ,EAAE,MAAM,CAAC,UAAU;qBAC5B,CAAC,CAAC,CAAC;gBACN,CAAC;gBACD,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;oBAC9B,MAAM,UAAU,GAAG,OAAO,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;oBAC/F,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE,UAAU,EAAE;wBACpD,MAAM,EAAE,SAA2B;qBACpC,CAAC,CAAC,CAAC;gBACN,CAAC;YACH,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,IAAI,IAAI,CAAC,MAAM,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACvD,CAAC;QAED,WAAW;QACX,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,IAAmB;QACrD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,KAAK,EAAE;gBAC5C,MAAM,EAAE,KAAK;gBACb,OAAO;aACR,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,EAAE,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACP,gCAAgC;YAChC,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;gBAC5C,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;gBACnE,OAAO,QAAQ,CAAC,EAAE,CAAC;YACrB,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,IAAI,kBAAkB,EAAE,CAAC"}

package/lib/dist/services/connectors/mock/MockConnector.d.ts

@@ -0,0 +1,37 @@
+/**
+ * Mock Connector
+ * Generates demo/test trajectory without calling a real agent
+ */
+import type { TrajectoryStep } from '../../../types/index.js';
+import { BaseConnector } from '../../../services/connectors/base/BaseConnector.js';
+import type { ConnectorAuth, ConnectorRequest, ConnectorResponse, ConnectorProgressCallback, ConnectorRawEventCallback } from '../../../services/connectors/types.js';
+/**
+ * Mock Connector for demo and testing
+ * Generates a realistic trajectory without calling a real agent
+ */
+export declare class MockConnector extends BaseConnector {
+    readonly type: "mock";
+    readonly name = "Demo Agent (Mock)";
+    readonly supportsStreaming = true;
+    /**
+     * Build payload - not used for mock but required by interface
+     */
+    buildPayload(request: ConnectorRequest): any;
+    /**
+     * Execute mock request - generates realistic trajectory with delays
+     */
+    execute(endpoint: string, request: ConnectorRequest, auth: ConnectorAuth, onProgress?: ConnectorProgressCallback, onRawEvent?: ConnectorRawEventCallback): Promise<ConnectorResponse>;
+    /**
+     * Parse raw events - for mock, just extract steps from our custom format
+     */
+    parseResponse(rawEvents: any[]): TrajectoryStep[];
+    /**
+     * Health check - mock is always "healthy"
+     */
+    healthCheck(endpoint: string, auth: ConnectorAuth): Promise<boolean>;
+}
+/**
+ * Default instance for convenience
+ */
+export declare const mockConnector: MockConnector;
+//# sourceMappingURL=MockConnector.d.ts.map

package/lib/dist/services/connectors/mock/MockConnector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MockConnector.d.ts","sourceRoot":"","sources":["../../../../../services/connectors/mock/MockConnector.ts"],"names":[],"mappings":"AAKA;;;GAGG;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;AAErC;;;GAGG;AACH,qBAAa,aAAc,SAAQ,aAAa;IAC9C,QAAQ,CAAC,IAAI,EAAG,MAAM,CAAU;IAChC,QAAQ,CAAC,IAAI,uBAAuB;IACpC,QAAQ,CAAC,iBAAiB,QAAQ;IAElC;;OAEG;IACH,YAAY,CAAC,OAAO,EAAE,gBAAgB,GAAG,GAAG;IAO5C;;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;IAsG7B;;OAEG;IACH,aAAa,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,cAAc,EAAE;IAMjD;;OAEG;IACG,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;CAG3E;AAED;;GAEG;AACH,eAAO,MAAM,aAAa,eAAsB,CAAC"}

package/lib/dist/services/connectors/mock/MockConnector.js

@@ -0,0 +1,120 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseConnector } from '../../../services/connectors/base/BaseConnector.js';
+/**
+ * Mock Connector for demo and testing
+ * Generates a realistic trajectory without calling a real agent
+ */
+export class MockConnector extends BaseConnector {
+    constructor() {
+        super(...arguments);
+        this.type = 'mock';
+        this.name = 'Demo Agent (Mock)';
+        this.supportsStreaming = true;
+    }
+    /**
+     * Build payload - not used for mock but required by interface
+     */
+    buildPayload(request) {
+        return {
+            question: request.testCase.initialPrompt,
+            context: request.testCase.context,
+        };
+    }
+    /**
+     * Execute mock request - generates realistic trajectory with delays
+     */
+    async execute(endpoint, request, auth, onProgress, onRawEvent) {
+        const trajectory = [];
+        const rawEvents = [];
+        const runId = `mock-run-${Date.now()}`;
+        const sleep = (ms) => new Promise(r => setTimeout(r, ms));
+        this.debug('Generating mock trajectory for:', request.testCase.name);
+        // Helper to emit step
+        const emitStep = (step) => {
+            trajectory.push(step);
+            onProgress?.(step);
+            // Also emit as "raw event" for consistency
+            rawEvents.push({ type: 'MOCK_STEP', step });
+            onRawEvent?.({ type: 'MOCK_STEP', step });
+        };
+        // Step 1: Initial thinking/analysis
+        await sleep(100);
+        emitStep(this.createStep('assistant', 'I need to investigate this issue. Let me start by checking the cluster health and then drill down into specific metrics.'));
+        // Step 2: First tool call
+        await sleep(300);
+        emitStep(this.createStep('action', 'Calling opensearch_cluster_health...', {
+            toolName: 'opensearch_cluster_health',
+            toolArgs: { local: true },
+        }));
+        // Step 3: First tool result
+        await sleep(500);
+        emitStep(this.createStep('tool_result', JSON.stringify({ status: 'yellow', number_of_nodes: 3, unassigned_shards: 0 }, null, 2), {
+            status: 'SUCCESS',
+            latencyMs: 500,
+        }));
+        // Step 4: Analysis after first tool
+        await sleep(300);
+        emitStep(this.createStep('assistant', 'The cluster is in yellow state. Let me check the node stats to identify which node might be causing issues.'));
+        // Step 5: Second tool call
+        await sleep(300);
+        emitStep(this.createStep('action', 'Calling opensearch_nodes_stats...', {
+            toolName: 'opensearch_nodes_stats',
+            toolArgs: { metric: 'jvm,os' },
+        }));
+        // Step 6: Second tool result
+        await sleep(600);
+        emitStep(this.createStep('tool_result', 'Node-1: CPU 12%, JVM Heap 45%\nNode-2: CPU 15%, JVM Heap 52%\nNode-3: CPU 98%, JVM Heap 89% (Data Node)', {
+            status: 'SUCCESS',
+            latencyMs: 600,
+        }));
+        // Step 7: Final response with RCA
+        await sleep(400);
+        emitStep(this.createStep('response', `## Root Cause Analysis Complete
+
+**Finding:** High CPU utilization detected on Node-3 (98% CPU, 89% JVM Heap)
+
+**Root Cause:** Node-3 is experiencing resource exhaustion, likely due to:
+1. Heavy indexing or search operations
+2. Garbage collection pressure from high heap usage
+3. Possible hot spot in shard distribution
+
+**Recommendations:**
+1. Check hot threads on Node-3 using \`_nodes/Node-3/hot_threads\`
+2. Review shard distribution and consider rebalancing
+3. Monitor GC logs for long pauses
+4. Consider scaling horizontally if load persists`));
+        this.debug('Mock trajectory completed. Steps:', trajectory.length);
+        return {
+            trajectory,
+            runId,
+            rawEvents,
+            metadata: {
+                mock: true,
+                testCaseId: request.testCase.id,
+                testCaseName: request.testCase.name,
+            },
+        };
+    }
+    /**
+     * Parse raw events - for mock, just extract steps from our custom format
+     */
+    parseResponse(rawEvents) {
+        return rawEvents
+            .filter(e => e.type === 'MOCK_STEP' && e.step)
+            .map(e => e.step);
+    }
+    /**
+     * Health check - mock is always "healthy"
+     */
+    async healthCheck(endpoint, auth) {
+        return true;
+    }
+}
+/**
+ * Default instance for convenience
+ */
+export const mockConnector = new MockConnector();
+//# sourceMappingURL=MockConnector.js.map

package/lib/dist/services/connectors/mock/MockConnector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MockConnector.js","sourceRoot":"","sources":["../../../../../services/connectors/mock/MockConnector.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAQH,OAAO,EAAE,aAAa,EAAE,MAAM,0CAA0C,CAAC;AASzE;;;GAGG;AACH,MAAM,OAAO,aAAc,SAAQ,aAAa;IAAhD;;QACW,SAAI,GAAG,MAAe,CAAC;QACvB,SAAI,GAAG,mBAAmB,CAAC;QAC3B,sBAAiB,GAAG,IAAI,CAAC;IA0IpC,CAAC;IAxIC;;OAEG;IACH,YAAY,CAAC,OAAyB;QACpC,OAAO;YACL,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,aAAa;YACxC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,OAAO;SAClC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,QAAgB,EAChB,OAAyB,EACzB,IAAmB,EACnB,UAAsC,EACtC,UAAsC;QAEtC,MAAM,UAAU,GAAqB,EAAE,CAAC;QACxC,MAAM,SAAS,GAAU,EAAE,CAAC;QAC5B,MAAM,KAAK,GAAG,YAAY,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,CAAC,EAAU,EAAE,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAElE,IAAI,CAAC,KAAK,CAAC,iCAAiC,EAAE,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAErE,sBAAsB;QACtB,MAAM,QAAQ,GAAG,CAAC,IAAoB,EAAE,EAAE;YACxC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC;YACnB,2CAA2C;YAC3C,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5C,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC;QAEF,oCAAoC;QACpC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,EAClC,0HAA0H,CAC3H,CAAC,CAAC;QAEH,0BAA0B;QAC1B,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,EAC/B,sCAAsC,EACtC;YACE,QAAQ,EAAE,2BAA2B;YACrC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;SAC1B,CACF,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,EACpC,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,EACvF;YACE,MAAM,EAAE,SAA2B;YACnC,SAAS,EAAE,GAAG;SACf,CACF,CAAC,CAAC;QAEH,oCAAoC;QACpC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,WAAW,EAClC,6GAA6G,CAC9G,CAAC,CAAC;QAEH,2BAA2B;QAC3B,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,EAC/B,mCAAmC,EACnC;YACE,QAAQ,EAAE,wBAAwB;YAClC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE;SAC/B,CACF,CAAC,CAAC;QAEH,6BAA6B;QAC7B,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,EACpC,yGAAyG,EACzG;YACE,MAAM,EAAE,SAA2B;YACnC,SAAS,EAAE,GAAG;SACf,CACF,CAAC,CAAC;QAEH,kCAAkC;QAClC,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;QACjB,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,EACjC;;;;;;;;;;;;;kDAa4C,CAC7C,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,mCAAmC,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;QAEnE,OAAO;YACL,UAAU;YACV,KAAK;YACL,SAAS;YACT,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI;gBACV,UAAU,EAAE,OAAO,CAAC,QAAQ,CAAC,EAAE;gBAC/B,YAAY,EAAE,OAAO,CAAC,QAAQ,CAAC,IAAI;aACpC;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,SAAgB;QAC5B,OAAO,SAAS;aACb,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,IAAI,CAAC;aAC7C,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,IAAmB;QACrD,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC"}

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

@@ -0,0 +1,42 @@
+/**
+ * OpenAI-compatible Connector
+ *
+ * Connector for any OpenAI-compatible API endpoint.
+ * Formats requests as OpenAI Chat Completions and parses the response.
+ *
+ * Works with:
+ * - OpenAI API directly
+ * - Azure OpenAI
+ * - Any OpenAI-compatible endpoint (vLLM, Ollama, LiteLLM proxy, etc.)
+ */
+import type { TrajectoryStep } from '../../../types/index.js';
+import { BaseConnector } from '../../../services/connectors/base/BaseConnector.js';
+import type { ConnectorAuth, ConnectorRequest, ConnectorResponse, ConnectorProgressCallback, ConnectorRawEventCallback } from '../../../services/connectors/types.js';
+/**
+ * OpenAI-compatible connector
+ */
+export declare class OpenAICompatibleConnector extends BaseConnector {
+    readonly type: "openai-compatible";
+    readonly name = "OpenAI-compatible";
+    traceContext: {
+        propagateHeader: boolean;
+    };
+    readonly supportsStreaming = false;
+    /**
+     * Build OpenAI Chat Completion payload from test case
+     */
+    buildPayload(request: ConnectorRequest): any;
+    /**
+     * Execute OpenAI-compatible Chat Completion request
+     */
+    execute(endpoint: string, request: ConnectorRequest, auth: ConnectorAuth, onProgress?: ConnectorProgressCallback, onRawEvent?: ConnectorRawEventCallback): Promise<ConnectorResponse>;
+    /**
+     * Parse OpenAI Chat Completion response into trajectory steps
+     */
+    parseResponse(data: any): TrajectoryStep[];
+}
+/**
+ * Default instance for convenience
+ */
+export declare const openaiCompatibleConnector: OpenAICompatibleConnector;
+//# sourceMappingURL=OpenAICompatibleConnector.d.ts.map

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"}