AxTraceLogger

Defined in: https://github.com/ax-llm/ax/blob/a8847bd2906efff202fde10d776fddd20fd2ff57/src/ax/trace/logger.ts#L68

AxTraceLogger wraps an AxGen instance to automatically log all forward() calls.

Example

const gen = ax(`query -> response`);
const storage = new AxMemoryStorage();

const tracedGen = new AxTraceLogger(gen, {
  agentId: 'my-agent',
  storage,
});

// Use exactly like AxGen
const result = await tracedGen.forward(ai, { query: 'Hello' });

// Traces are automatically saved to storage
const traces = await storage.getTraces('my-agent');

Type Parameters

Type Parameter
`IN` extends `AxGenIn`
`OUT` extends `AxGenOut`

Constructors

Constructor

new AxTraceLogger<IN, OUT>(gen: AxGen<IN, OUT>, options: AxTraceLoggerOptions): AxTraceLogger<IN, OUT>;

Defined in: https://github.com/ax-llm/ax/blob/a8847bd2906efff202fde10d776fddd20fd2ff57/src/ax/trace/logger.ts#L78

Parameters

Parameter	Type
`gen`	`AxGen`<`IN`, `OUT`>
`options`	`AxTraceLoggerOptions`

Returns

AxTraceLogger<IN, OUT>

Methods

clone()

clone(newGen?: AxGen<IN, OUT>): AxTraceLogger<IN, OUT>;

Defined in: https://github.com/ax-llm/ax/blob/a8847bd2906efff202fde10d776fddd20fd2ff57/src/ax/trace/logger.ts#L246

Clone the logger with a new underlying AxGen.

Parameters

Parameter	Type
`newGen?`	`AxGen`<`IN`, `OUT`>

Returns

AxTraceLogger<IN, OUT>

forward()

forward(
   ai: AxAIService, 
   values: IN | AxMessage<IN>[], 
options?: Readonly<AxProgramForwardOptions<string>>): Promise<OUT>;

Defined in: https://github.com/ax-llm/ax/blob/a8847bd2906efff202fde10d776fddd20fd2ff57/src/ax/trace/logger.ts#L151

Forward call to the underlying AxGen with trace logging.

Parameters

Parameter	Type
`ai`	`AxAIService`
`values`	`IN` \| `AxMessage`<`IN`>[]
`options?`	`Readonly`<`AxProgramForwardOptions`<`string`>>

Returns

Promise<OUT>

getGen()

getGen(): AxGen<IN, OUT>;

Defined in: https://github.com/ax-llm/ax/blob/a8847bd2906efff202fde10d776fddd20fd2ff57/src/ax/trace/logger.ts#L218

Get the underlying AxGen instance.

Returns

AxGen<IN, OUT>

getName()

getName(): string;

Defined in: https://github.com/ax-llm/ax/blob/a8847bd2906efff202fde10d776fddd20fd2ff57/src/ax/trace/logger.ts#L225

Get the agent ID.

Returns

string

getStorage()

getStorage(): AxStorage;

Defined in: https://github.com/ax-llm/ax/blob/a8847bd2906efff202fde10d776fddd20fd2ff57/src/ax/trace/logger.ts#L232

Get the storage backend.

Returns

AxStorage

setMetadata()

setMetadata(metadata: Record<string, unknown>): void;

Defined in: https://github.com/ax-llm/ax/blob/a8847bd2906efff202fde10d776fddd20fd2ff57/src/ax/trace/logger.ts#L239

Update the metadata for future traces.

Parameters

Parameter	Type
`metadata`	`Record`<`string`, `unknown`>

Returns

void

streamingForward()

streamingForward(
   ai: AxAIService, 
   values: IN | AxMessage<IN>[], 
options?: Readonly<AxProgramForwardOptions<string>>): AxGenStreamingOut<OUT>;

Defined in: https://github.com/ax-llm/ax/blob/a8847bd2906efff202fde10d776fddd20fd2ff57/src/ax/trace/logger.ts#L94

Streaming forward call to the underlying AxGen with trace logging.

Parameters

Parameter	Type
`ai`	`AxAIService`
`values`	`IN` \| `AxMessage`<`IN`>[]
`options?`	`Readonly`<`AxProgramForwardOptions`<`string`>>

Returns

AxGenStreamingOut<OUT>