Class PgBackend

src/compiled/cognition/traces.ts:121–158

}

class PgBackend implements SpanBackend {
  async write(span: CognitionSpan): Promise<void> {
    // Lazy import — keeps tests that don't exercise PG free of pg.
    // eslint-disable-next-line @typescript-eslint/no-require-imports
    const { query } = require("../db") as { query: (sql: string, args: unknown[]) => Promise<unknown> };
    await query(
      `INSERT INTO cognition_traces (
        span_id, trace_id, parent_span_id, workflow, step, kind, prompt, model, provider,
        input_redacted, output_redacted, validate_result, confidence,
        prompt_tokens, completion_tokens, cost_usd, latency_ms, cache_hit,
        status, error_code, metadata, started_at, finished_at
      ) VALUES (
        $1, $2, $3, $4, $5, $6, $7, $8, $9,
        $10, $11, $12, $13, $14, $15, $16, $17, $18,
        $19, $20, $21, to_timestamp($22 / 1000.0), to_timestamp($23 / 1000.0)
      )
      ON CONFLICT (span_id) DO NOTHING`,
      [
        span.span_id, span.trace_id, span.parent_span_id,
        span.workflow, span.step, span.kind, span.prompt, span.model, span.provider,
        JSON.stringify(span.input_redacted), JSON.stringify(span.output_redacted),
        span.validate_result, span.confidence,
        span.prompt_tokens, span.completion_tokens, span.cost_usd, span.latency_ms, span.cache_hit,
        span.status, span.error_code, JSON.stringify(span.metadata),
        span.started_at, span.finished_at,
      ],
    );
  }
  async loadTrace(trace_id: string): Promise<CognitionSpan[]> {
    // eslint-disable-next-line @typescript-eslint/no-require-imports
    const { query } = require("../db") as { query: (sql: string, args: unknown[]) => Promise<Record<string, unknown>[]> };
    const rows = await query(
      "SELECT * FROM cognition_traces WHERE trace_id = $1 ORDER BY started_at",
      [trace_id],
    );
    return rows.map(rowToSpan);
  }
}

class NoopBackend implements SpanBackend {
  async write(_span: CognitionSpan): Promise<void> { /* drop */ }