Function loadTraceData

src/assertions/index.ts:180–224

(traceId: string)

}

async function loadTraceData(traceId: string): Promise<TraceData | null> {
  const traceStore = getTraceStore();
  const maxAttempts = Math.min(
    MAX_TRACE_FETCH_MAX_ATTEMPTS,
    Math.max(1, getEnvInt('PROMPTFOO_TRACE_FETCH_MAX_ATTEMPTS', DEFAULT_TRACE_FETCH_MAX_ATTEMPTS)),
  );
  const retryDelayMs = Math.min(
    MAX_TRACE_FETCH_RETRY_DELAY_MS,
    Math.max(
      0,
      getEnvInt('PROMPTFOO_TRACE_FETCH_RETRY_DELAY_MS', DEFAULT_TRACE_FETCH_RETRY_DELAY_MS),
    ),
  );
  const stablePolls = Math.min(
    MAX_TRACE_FETCH_STABLE_POLLS,
    Math.max(1, getEnvInt('PROMPTFOO_TRACE_FETCH_STABLE_POLLS', DEFAULT_TRACE_FETCH_STABLE_POLLS)),
  );

  let lastSpanCount = -1;
  let stableObservations = 0;
  let latestTrace: TraceData | null = null;

  for (let attempt = 0; attempt < maxAttempts; attempt++) {
    latestTrace = await traceStore.getTrace(traceId, { sanitizeAttributes: false });

    const spanCount = latestTrace?.spans?.length ?? 0;
    if (spanCount > 0) {
      stableObservations = spanCount === lastSpanCount ? stableObservations + 1 : 1;
      lastSpanCount = spanCount;

      if (stableObservations >= stablePolls || attempt === maxAttempts - 1) {
        return latestTrace;
      }
    } else {
      stableObservations = 0;
      lastSpanCount = spanCount;
    }

    if (attempt < maxAttempts - 1) {
      await sleep(retryDelayMs);
    }
  }

  return latestTrace;
}

const ASSERTION_HANDLERS: Record<
  BaseAssertionTypes,