@lelemondev/sdk 0.7.0 → 0.7.1

package/dist/index.d.mts

@@ -28,7 +28,7 @@ interface ObserveOptions {
     /** Tags for filtering */
     tags?: string[];
 }
-type SpanType = 'llm' | 'tool' | 'retrieval' | 'custom';
+type SpanType = 'llm' | 'tool' | 'retrieval' | 'embedding' | 'guardrail' | 'rerank' | 'custom';
 interface CaptureSpanOptions {
     /** Span type */
     type: SpanType;
@@ -106,6 +106,120 @@ declare function observe<T>(client: T, options?: ObserveOptions): T;
  */
 declare function createObserve(defaultOptions: ObserveOptions): <T>(client: T, options?: ObserveOptions) => T;
 
+/**
+ * Trace Context Module
+ *
+ * Provides AsyncLocalStorage-based context for grouping spans under a parent trace.
+ * This enables hierarchical tracing without modifying the simple observe() API.
+ *
+ * @example
+ * ```typescript
+ * import { trace, span } from '@lelemondev/sdk';
+ *
+ * await trace('sales-agent', async () => {
+ *   await client.send(new ConverseCommand({...})); // Span 1
+ *   await client.send(new ConverseCommand({...})); // Span 2
+ *   span({ type: 'retrieval', name: 'pinecone', output: { count: 5 } });
+ * });
+ * ```
+ */
+interface TraceContext {
+    /** Unique trace ID (shared by all spans in this trace) */
+    traceId: string;
+    /** Root span ID (the agent/workflow span) */
+    rootSpanId: string;
+    /** Current span ID (for nesting - LLM calls become children of this) */
+    currentSpanId: string;
+    /** Parent span ID (for nested trace() calls) */
+    parentSpanId?: string;
+    /** Trace name */
+    name: string;
+    /** Start time in ms */
+    startTime: number;
+    /** Input data */
+    input?: unknown;
+    /** Trace metadata */
+    metadata?: Record<string, unknown>;
+    /** Trace tags */
+    tags?: string[];
+}
+interface TraceOptions {
+    /** Name for the trace (e.g., 'sales-agent', 'rag-query') */
+    name: string;
+    /** Input data for the trace */
+    input?: unknown;
+    /** Custom metadata */
+    metadata?: Record<string, unknown>;
+    /** Tags for filtering */
+    tags?: string[];
+}
+interface SpanOptions {
+    /** Span type */
+    type: 'retrieval' | 'embedding' | 'tool' | 'guardrail' | 'rerank' | 'custom';
+    /** Span name (e.g., 'pinecone-search', 'cohere-rerank') */
+    name: string;
+    /** Input data */
+    input?: unknown;
+    /** Output data */
+    output?: unknown;
+    /** Duration in milliseconds (optional, will be set automatically if not provided) */
+    durationMs?: number;
+    /** Status */
+    status?: 'success' | 'error';
+    /** Error message if status is 'error' */
+    errorMessage?: string;
+    /** Custom metadata */
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Get the current trace context, if any
+ */
+declare function getTraceContext(): TraceContext | undefined;
+/**
+ * Execute a function within a trace context.
+ * All LLM calls made within this function will be grouped under the same trace.
+ *
+ * @example Simple usage (just name)
+ * ```typescript
+ * await trace('sales-agent', async () => {
+ *   await client.send(new ConverseCommand({...}));
+ *   await client.send(new ConverseCommand({...}));
+ * });
+ * ```
+ *
+ * @example With options
+ * ```typescript
+ * await trace({ name: 'rag-query', input: question, tags: ['production'] }, async () => {
+ *   const docs = await search(question);
+ *   span({ type: 'retrieval', name: 'pinecone', output: { count: docs.length } });
+ *   return client.send(new ConverseCommand({...}));
+ * });
+ * ```
+ */
+declare function trace<T>(nameOrOptions: string | TraceOptions, fn: () => Promise<T>): Promise<T>;
+/**
+ * Manually capture a span for non-LLM operations (retrieval, embedding, etc.)
+ * Must be called within a trace() block.
+ *
+ * @example
+ * ```typescript
+ * await trace('rag-query', async () => {
+ *   const t0 = Date.now();
+ *   const docs = await pinecone.query({ vector, topK: 5 });
+ *   span({
+ *     type: 'retrieval',
+ *     name: 'pinecone-search',
+ *     input: { topK: 5 },
+ *     output: { count: docs.length },
+ *     durationMs: Date.now() - t0,
+ *   });
+ *
+ *   return client.send(new ConverseCommand({...}));
+ * });
+ * ```
+ */
+declare function span(options: SpanOptions): void;
+
 /**
  * Capture Module
  *
@@ -139,4 +253,4 @@ declare function createObserve(defaultOptions: ObserveOptions): <T>(client: T, o
  */
 declare function captureSpan(options: CaptureSpanOptions): void;
 
-export { type CaptureSpanOptions, type LelemonConfig, type ObserveOptions, type ProviderName, type SpanType, captureSpan, createObserve, flush, init, isEnabled, observe };
+export { type CaptureSpanOptions, type LelemonConfig, type ObserveOptions, type ProviderName, type SpanOptions, type SpanType, type TraceOptions, captureSpan, createObserve, flush, getTraceContext, init, isEnabled, observe, span, trace };

package/dist/index.d.ts

@@ -28,7 +28,7 @@ interface ObserveOptions {
     /** Tags for filtering */
     tags?: string[];
 }
-type SpanType = 'llm' | 'tool' | 'retrieval' | 'custom';
+type SpanType = 'llm' | 'tool' | 'retrieval' | 'embedding' | 'guardrail' | 'rerank' | 'custom';
 interface CaptureSpanOptions {
     /** Span type */
     type: SpanType;
@@ -106,6 +106,120 @@ declare function observe<T>(client: T, options?: ObserveOptions): T;
  */
 declare function createObserve(defaultOptions: ObserveOptions): <T>(client: T, options?: ObserveOptions) => T;
 
+/**
+ * Trace Context Module
+ *
+ * Provides AsyncLocalStorage-based context for grouping spans under a parent trace.
+ * This enables hierarchical tracing without modifying the simple observe() API.
+ *
+ * @example
+ * ```typescript
+ * import { trace, span } from '@lelemondev/sdk';
+ *
+ * await trace('sales-agent', async () => {
+ *   await client.send(new ConverseCommand({...})); // Span 1
+ *   await client.send(new ConverseCommand({...})); // Span 2
+ *   span({ type: 'retrieval', name: 'pinecone', output: { count: 5 } });
+ * });
+ * ```
+ */
+interface TraceContext {
+    /** Unique trace ID (shared by all spans in this trace) */
+    traceId: string;
+    /** Root span ID (the agent/workflow span) */
+    rootSpanId: string;
+    /** Current span ID (for nesting - LLM calls become children of this) */
+    currentSpanId: string;
+    /** Parent span ID (for nested trace() calls) */
+    parentSpanId?: string;
+    /** Trace name */
+    name: string;
+    /** Start time in ms */
+    startTime: number;
+    /** Input data */
+    input?: unknown;
+    /** Trace metadata */
+    metadata?: Record<string, unknown>;
+    /** Trace tags */
+    tags?: string[];
+}
+interface TraceOptions {
+    /** Name for the trace (e.g., 'sales-agent', 'rag-query') */
+    name: string;
+    /** Input data for the trace */
+    input?: unknown;
+    /** Custom metadata */
+    metadata?: Record<string, unknown>;
+    /** Tags for filtering */
+    tags?: string[];
+}
+interface SpanOptions {
+    /** Span type */
+    type: 'retrieval' | 'embedding' | 'tool' | 'guardrail' | 'rerank' | 'custom';
+    /** Span name (e.g., 'pinecone-search', 'cohere-rerank') */
+    name: string;
+    /** Input data */
+    input?: unknown;
+    /** Output data */
+    output?: unknown;
+    /** Duration in milliseconds (optional, will be set automatically if not provided) */
+    durationMs?: number;
+    /** Status */
+    status?: 'success' | 'error';
+    /** Error message if status is 'error' */
+    errorMessage?: string;
+    /** Custom metadata */
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Get the current trace context, if any
+ */
+declare function getTraceContext(): TraceContext | undefined;
+/**
+ * Execute a function within a trace context.
+ * All LLM calls made within this function will be grouped under the same trace.
+ *
+ * @example Simple usage (just name)
+ * ```typescript
+ * await trace('sales-agent', async () => {
+ *   await client.send(new ConverseCommand({...}));
+ *   await client.send(new ConverseCommand({...}));
+ * });
+ * ```
+ *
+ * @example With options
+ * ```typescript
+ * await trace({ name: 'rag-query', input: question, tags: ['production'] }, async () => {
+ *   const docs = await search(question);
+ *   span({ type: 'retrieval', name: 'pinecone', output: { count: docs.length } });
+ *   return client.send(new ConverseCommand({...}));
+ * });
+ * ```
+ */
+declare function trace<T>(nameOrOptions: string | TraceOptions, fn: () => Promise<T>): Promise<T>;
+/**
+ * Manually capture a span for non-LLM operations (retrieval, embedding, etc.)
+ * Must be called within a trace() block.
+ *
+ * @example
+ * ```typescript
+ * await trace('rag-query', async () => {
+ *   const t0 = Date.now();
+ *   const docs = await pinecone.query({ vector, topK: 5 });
+ *   span({
+ *     type: 'retrieval',
+ *     name: 'pinecone-search',
+ *     input: { topK: 5 },
+ *     output: { count: docs.length },
+ *     durationMs: Date.now() - t0,
+ *   });
+ *
+ *   return client.send(new ConverseCommand({...}));
+ * });
+ * ```
+ */
+declare function span(options: SpanOptions): void;
+
 /**
  * Capture Module
  *
@@ -139,4 +253,4 @@ declare function createObserve(defaultOptions: ObserveOptions): <T>(client: T, o
  */
 declare function captureSpan(options: CaptureSpanOptions): void;
 
-export { type CaptureSpanOptions, type LelemonConfig, type ObserveOptions, type ProviderName, type SpanType, captureSpan, createObserve, flush, init, isEnabled, observe };
+export { type CaptureSpanOptions, type LelemonConfig, type ObserveOptions, type ProviderName, type SpanOptions, type SpanType, type TraceOptions, captureSpan, createObserve, flush, getTraceContext, init, isEnabled, observe, span, trace };