elasticdash-test 0.1.18-alpha-23 → 0.1.18-alpha-24

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "elasticdash-test",
-  "version": "0.1.18-alpha-23",
+  "version": "0.1.18-alpha-24",
   "description": "AI-native test runner for ElasticDash workflow testing",
   "type": "module",
   "bin": {

package/src/interceptors/workflow-ai.ts

@@ -64,15 +64,36 @@ function extractUsage(output: unknown): UsageInfo | undefined {
  * HTTP payload sent to the LLM (messages, model, parameters).
  * Also returns captured usage when the app's output doesn't include it.
  */
-function enrichFromLLMCapture(input: unknown, appUsage: UsageInfo | undefined): { input: unknown; usage: UsageInfo | undefined } {
+function enrichFromLLMCapture(
+  input: unknown,
+  appUsage: UsageInfo | undefined,
+  fallbackModel?: string,
+  fallbackProvider?: string,
+): { input: unknown; usage: UsageInfo | undefined } {
   const captured = consumeCapturedLLMRequest()
-  if (!captured) return { input, usage: appUsage }
-  const enrichedInput = (input && typeof input === 'object')
-    ? { ...(input as Record<string, unknown>), llmRequest: captured.body, promptSnippet: captured.promptSnippet }
-    : { originalInput: input, llmRequest: captured.body, promptSnippet: captured.promptSnippet }
-  // Prefer app-level usage if available, fall back to HTTP-level usage
-  const usage = appUsage ?? captured.usage
-  return { input: enrichedInput, usage }
+  if (captured) {
+    const enrichedInput = (input && typeof input === 'object')
+      ? { ...(input as Record<string, unknown>), llmRequest: captured.body, promptSnippet: captured.promptSnippet }
+      : { originalInput: input, llmRequest: captured.body, promptSnippet: captured.promptSnippet }
+    const usage = appUsage ?? captured.usage
+    return { input: enrichedInput, usage }
+  }
+
+  // Interceptor capture failed (e.g. OpenAI SDK uses native fetch not globalThis.fetch).
+  // Fall back to model/provider from wrapAI options or function arguments.
+  const resolvedModel = fallbackModel
+    || (input && typeof input === 'object' ? (input as Record<string, unknown>).model as string | undefined : undefined)
+  const resolvedProvider = fallbackProvider
+    || (input && typeof input === 'object' ? (input as Record<string, unknown>).provider as string | undefined : undefined)
+
+  if (resolvedModel || resolvedProvider) {
+    const enrichedInput = (input && typeof input === 'object')
+      ? { ...(input as Record<string, unknown>), llmRequest: { model: resolvedModel, provider: resolvedProvider } }
+      : { originalInput: input, llmRequest: { model: resolvedModel, provider: resolvedProvider } }
+    return { input: enrichedInput, usage: appUsage }
+  }
+
+  return { input, usage: appUsage }
 }
 
 function isReadableStream(v: unknown): v is ReadableStream<Uint8Array> {
@@ -165,7 +186,7 @@ export function wrapAI<Args extends unknown[], R>(
       const input = args.length === 1 ? args[0] : args
       try {
         const output = await callFn(...args)
-        const enriched = enrichFromLLMCapture(input, extractUsage(output))
+        const enriched = enrichFromLLMCapture(input, extractUsage(output), _options?.model, _options?.provider)
 
         if (isReadableStream(output)) {
           const [streamForCaller, streamForRecorder] = output.tee()
@@ -196,7 +217,7 @@ export function wrapAI<Args extends unknown[], R>(
         pushTelemetryEvent(event)
         return output
       } catch (e) {
-        const enriched = enrichFromLLMCapture(input, undefined)
+        const enriched = enrichFromLLMCapture(input, undefined, _options?.model, _options?.provider)
         const durationMs = rawDateNow() - start
         pushTelemetryEvent({
           id, type: 'ai', name: modelName, input: enriched.input,
@@ -327,7 +348,7 @@ export function wrapAI<Args extends unknown[], R>(
 
     try {
       const output = await callFn(...httpEffectiveArgs)
-      const enriched = enrichFromLLMCapture(input, extractUsage(output))
+      const enriched = enrichFromLLMCapture(input, extractUsage(output), _options?.model, _options?.provider)
 
       if (isReadableStream(output)) {
         const [streamForCaller, streamForRecorder] = output.tee()
@@ -358,7 +379,7 @@ export function wrapAI<Args extends unknown[], R>(
       pushTelemetryEvent(event)
       return output
     } catch (e) {
-      const enriched = enrichFromLLMCapture(input, undefined)
+      const enriched = enrichFromLLMCapture(input, undefined, _options?.model, _options?.provider)
       const durationMs = rawDateNow() - start
       pushTelemetryEvent({
         id, type: 'ai', name: modelName, input: enriched.input,