Axiom

import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http';
import { NodeSDK } from '@opentelemetry/sdk-node';
import { createAxiomExporterOptions } from '@core-ai/axiom';

function getRequiredEnv(name: 'AXIOM_TOKEN' | 'AXIOM_DATASET'): string {
  const value = process.env[name];
  if (!value) {
    throw new Error(`Missing required environment variable: ${name}`);
  }
  return value;
}

export const sdk = new NodeSDK({
  traceExporter: new OTLPTraceExporter(
    createAxiomExporterOptions({
      token: getRequiredEnv('AXIOM_TOKEN'),
      dataset: getRequiredEnv('AXIOM_DATASET'),
    })
  ),
});

sdk.start();

type AxiomExporterConfig = {
  token: string;
  dataset: string;
  endpoint?: string;
};

import { generate, wrapChatModel } from '@core-ai/core-ai';
import { createAxiomMiddleware } from '@core-ai/axiom';

const tracedModel = wrapChatModel({
  model,
  middleware: createAxiomMiddleware(),
});

const result = await generate({
  model: tracedModel,
  messages: [{ role: 'user', content: 'Explain quantum computing.' }],
});

import { embed, wrapEmbeddingModel } from '@core-ai/core-ai';
import { createAxiomEmbeddingMiddleware } from '@core-ai/axiom';

const tracedModel = wrapEmbeddingModel({
  model: embeddingModel,
  middleware: createAxiomEmbeddingMiddleware(),
});

const result = await embed({
  model: tracedModel,
  input: 'Sample text for embedding',
});

import { generateImage, wrapImageModel } from '@core-ai/core-ai';
import { createAxiomImageMiddleware } from '@core-ai/axiom';

const tracedModel = wrapImageModel({
  model: imageModel,
  middleware: createAxiomImageMiddleware(),
});

const result = await generateImage({
  model: tracedModel,
  prompt: 'A mountain landscape at sunset',
});

type AxiomMiddlewareOptions = {
  recordContent?: boolean;
  tracerName?: string;
};

const middleware = createAxiomMiddleware({ recordContent: true });

const middleware = createAxiomMiddleware({ tracerName: 'my-app' });

['your-dataset']
| where ['attributes.gen_ai.operation.name'] == "chat"