@od-oneapp/ai-platform 0.1.0

package/dist/auto-resume-BuFRNvAX.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-resume-BuFRNvAX.mjs","names":[],"sources":["../src/streaming/core/protocol.ts","../src/streaming/core/multi-step.ts","../src/streaming/core/auto-resume.ts"],"sourcesContent":["/**\n * @fileoverview Streaming protocol utilities for artifact streaming.\n * @module @od-oneapp/ai-platform/streaming\n */\n\nimport type { ArtifactStreamWriter, CustomUIDataTypes } from './types';\n\n/**\n * Generic data stream writer interface.\n * Compatible with AI SDK's data stream patterns.\n */\nexport interface DataStreamWriter {\n  write(data: unknown): void;\n}\n\n/**\n * Creates an artifact stream writer for custom data streaming.\n *\n * @param writer - Data stream writer with write method\n * @returns Artifact stream writer with typed methods\n *\n * @example\n * ```ts\n * // With createUIMessageStream\n * const stream = createUIMessageStream({\n *   execute: async ({ writer }) => {\n *     const artifactWriter = createArtifactStreamWriter({\n *       write: (data) => writer.write(JSON.stringify(data)),\n *     });\n *     artifactWriter.writeId('artifact-123');\n *     artifactWriter.writeTextDelta('Hello world');\n *     artifactWriter.writeFinish(true);\n *   },\n * });\n * ```\n */\nexport function createArtifactStreamWriter(writer: DataStreamWriter): ArtifactStreamWriter {\n  return {\n    writeId(id: string) {\n      writer.write({ type: 'id', data: id });\n    },\n\n    writeKind(kind: string) {\n      writer.write({ type: 'kind', data: kind });\n    },\n\n    writeTitle(title: string) {\n      writer.write({ type: 'title', data: title });\n    },\n\n    writeStatus(status: CustomUIDataTypes['status']) {\n      writer.write({ type: 'status', data: status });\n    },\n\n    writeTextDelta(text: string) {\n      writer.write({ type: 'textDelta', data: text });\n    },\n\n    writeCodeDelta(code: string, language?: string) {\n      writer.write({ type: 'codeDelta', data: { code, language } });\n    },\n\n    writeProgress(current: number, total: number, message?: string) {\n      writer.write({ type: 'progress', data: { current, total, message } });\n    },\n\n    writeFinish(success: boolean, metadata?: Record<string, unknown>) {\n      writer.write({ type: 'finish', data: { success, metadata } });\n    },\n\n    writeError(message: string, code?: string, recoverable?: boolean) {\n      writer.write({ type: 'error', data: { message, code, recoverable } });\n    },\n\n    clear() {\n      writer.write({ type: 'clear', data: null });\n    },\n  };\n}\n\n/**\n * Helper to stream artifact creation with standard lifecycle.\n *\n * @param writer - Data stream writer\n * @param config - Artifact configuration\n * @param generator - Async generator that yields content\n *\n * @example\n * ```ts\n * await streamArtifact(writer, {\n *   id: 'doc-123',\n *   kind: 'text',\n *   title: 'My Document',\n * }, async function* () {\n *   yield 'Hello ';\n *   yield 'World!';\n * });\n * ```\n */\nexport async function streamArtifact(\n  dataWriter: DataStreamWriter,\n  config: {\n    id: string;\n    kind: string;\n    title: string;\n  },\n  generator: () => AsyncGenerator<string, void, unknown>,\n): Promise<void> {\n  const writer = createArtifactStreamWriter(dataWriter);\n\n  try {\n    // Initialize artifact\n    writer.writeId(config.id);\n    writer.writeKind(config.kind);\n    writer.writeTitle(config.title);\n    writer.writeStatus('streaming');\n\n    // Stream content\n    for await (const chunk of generator()) {\n      writer.writeTextDelta(chunk);\n    }\n\n    // Complete\n    writer.writeStatus('complete');\n    writer.writeFinish(true);\n  } catch (error) {\n    writer.writeStatus('error');\n    writer.writeError(\n      error instanceof Error ? error.message : 'Unknown error',\n      'STREAM_ERROR',\n      false,\n    );\n    writer.writeFinish(false);\n    throw error;\n  }\n}\n\n/**\n * Creates a buffered stream writer that batches writes for efficiency.\n *\n * @param writer - Data stream writer\n * @param options - Buffer options\n * @returns Buffered writer with flush method\n */\nexport function createBufferedStreamWriter(\n  writer: DataStreamWriter,\n  options: {\n    /** Buffer size before auto-flush */\n    bufferSize?: number;\n    /** Time in ms before auto-flush */\n    flushInterval?: number;\n  } = {},\n) {\n  const { bufferSize = 10, flushInterval = 100 } = options;\n  const buffer: Array<{ type: string; data: unknown }> = [];\n  let flushTimer: ReturnType<typeof setTimeout> | null = null;\n\n  function flush() {\n    if (buffer.length === 0) return;\n\n    for (const item of buffer) {\n      writer.write(item);\n    }\n    buffer.length = 0;\n\n    if (flushTimer) {\n      clearTimeout(flushTimer);\n      flushTimer = null;\n    }\n  }\n\n  function scheduleFlush() {\n    if (flushTimer) return;\n    flushTimer = setTimeout(flush, flushInterval);\n  }\n\n  function write(type: string, data: unknown) {\n    buffer.push({ type, data });\n\n    if (buffer.length >= bufferSize) {\n      flush();\n    } else {\n      scheduleFlush();\n    }\n  }\n\n  return {\n    write,\n    flush,\n    get bufferLength() {\n      return buffer.length;\n    },\n  };\n}\n","/**\n * Multi-Step Streaming Module\n *\n * SDK v6 patterns for streaming multi-step agent executions:\n * - createUIMessageStream with sendFinish: false\n * - Step-by-step UI updates\n * - Nested stream forwarding\n *\n * @module streaming/multi-step\n */\n\nimport { createUIMessageStream, createUIMessageStreamResponse } from '../../shared';\n\nimport type {\n  SDKStreamTextResult,\n  SDKUIMessage,\n  SDKUIMessageStreamWriter,\n  SDKToolSet,\n} from '../../shared';\n\n// -----------------------------------------------------------------------------\n// Types\n// -----------------------------------------------------------------------------\n\n/** Multi-step stream event */\nexport interface MultiStepStreamEvent {\n  type:\n    | 'step-start'\n    | 'step-text'\n    | 'step-tool-call'\n    | 'step-tool-result'\n    | 'step-end'\n    | 'finish'\n    | 'error';\n  stepIndex?: number;\n  stepName?: string;\n  content?: string;\n  toolCall?: {\n    id: string;\n    name: string;\n    input: unknown;\n  };\n  toolResult?: {\n    id: string;\n    name: string;\n    result: unknown;\n  };\n  error?: Error;\n}\n\n/** Step definition for multi-step execution */\nexport interface StepDefinition<TContext = unknown> {\n  /** Step name */\n  name: string;\n  /** Execute the step */\n  execute: (context: TContext, writer: SDKUIMessageStreamWriter<SDKUIMessage>) => Promise<TContext>;\n  /** Optional: check if step should run */\n  shouldRun?: (context: TContext) => boolean;\n}\n\n/** Multi-step stream options */\nexport interface MultiStepStreamOptions<TContext> {\n  /** Initial context */\n  initialContext: TContext;\n  /** Steps to execute */\n  steps: StepDefinition<TContext>[];\n  /** Error handler */\n  onError?: (error: Error, stepIndex: number) => void;\n  /** Step completion handler */\n  onStepComplete?: (stepIndex: number, context: TContext) => void;\n}\n\n// -----------------------------------------------------------------------------\n// Multi-Step Stream Writer\n// -----------------------------------------------------------------------------\n\ntype MultiStepStreamWriter = SDKUIMessageStreamWriter<SDKUIMessage>;\ntype MultiStepStreamResult = SDKStreamTextResult<SDKToolSet, never>;\n\n/**\n * Enhanced writer for multi-step streams.\n * Wraps UIMessageStreamWriter with step-aware methods.\n *\n * Note: SDK v6 uses typed message chunks. This wrapper uses\n * the `data-{name}` format for custom data parts to maintain\n * compatibility with the typed stream protocol.\n */\nexport class MultiStepWriter {\n  private currentStep = 0;\n  private stepStarted = false;\n\n  constructor(\n    private readonly writer: MultiStepStreamWriter,\n    private readonly options?: { sendFinish?: boolean },\n  ) {}\n\n  /**\n   * Start a new step.\n   * Uses data-step-event format for SDK v6 compatibility.\n   */\n  startStep(name: string): void {\n    if (name.trim().length === 0) {\n      throw new Error('Step name is required');\n    }\n    this.stepStarted = true;\n    // Use data-${name} format for custom data in SDK v6\n    this.writer.write({\n      type: 'data-step-event',\n      data: { event: 'step-start', stepIndex: this.currentStep, stepName: name },\n    });\n  }\n\n  /**\n   * End the current step.\n   */\n  endStep(): void {\n    this.writer.write({\n      type: 'data-step-event',\n      data: { event: 'step-end', stepIndex: this.currentStep },\n    });\n    this.currentStep++;\n    this.stepStarted = false;\n  }\n\n  /**\n   * Write text delta within current step.\n   * Uses the standard text-delta format.\n   */\n  writeText(text: string): void {\n    this.writer.write({\n      type: 'text-delta',\n      id: `step-${this.currentStep}`,\n      delta: text,\n    });\n  }\n\n  /**\n   * Write tool call within current step.\n   */\n  writeToolCall(id: string, name: string, input: unknown): void {\n    this.writer.write({\n      type: 'data-step-event',\n      data: {\n        event: 'step-tool-call',\n        stepIndex: this.currentStep,\n        toolCall: { id, name, input },\n      },\n    });\n  }\n\n  /**\n   * Write tool result within current step.\n   */\n  writeToolResult(id: string, name: string, result: unknown): void {\n    this.writer.write({\n      type: 'data-step-event',\n      data: {\n        event: 'step-tool-result',\n        stepIndex: this.currentStep,\n        toolResult: { id, name, result },\n      },\n    });\n  }\n\n  /**\n   * Forward a streamText result to this writer.\n   * Uses merge() to combine streams properly in SDK v6.\n   */\n  async forwardStream(\n    stream: MultiStepStreamResult,\n    options?: { sendFinish?: boolean },\n  ): Promise<void> {\n    void options;\n    // SDK v6 uses toUIMessageStream() which returns a ReadableStream\n    // Use merge() to forward the stream\n    const uiStream = stream.toUIMessageStream?.<SDKUIMessage>();\n    if (uiStream) {\n      this.writer.merge(uiStream);\n    }\n  }\n\n  /**\n   * Write error.\n   */\n  writeError(error: Error): void {\n    this.writer.write({\n      type: 'error',\n      errorText: error.message,\n    });\n  }\n\n  /**\n   * Finish the stream.\n   */\n  finish(): void {\n    if (this.options?.sendFinish !== false) {\n      this.writer.write({\n        type: 'finish',\n        finishReason: 'stop',\n      });\n    }\n  }\n\n  /** Get current step index */\n  get stepIndex(): number {\n    return this.currentStep;\n  }\n\n  /** Get underlying writer */\n  get raw(): MultiStepStreamWriter {\n    return this.writer;\n  }\n}\n\n// -----------------------------------------------------------------------------\n// Multi-Step Stream Factory\n// -----------------------------------------------------------------------------\n\n/**\n * Create a multi-step streaming response.\n *\n * @example\n * ```typescript\n * // API Route\n * export async function POST(request: Request) {\n *   const { messages } = await request.json();\n *\n *   return createMultiStepStreamResponse({\n *     initialContext: { messages, results: [] },\n *     steps: [\n *       {\n *         name: 'analyze',\n *         execute: async (ctx, writer) => {\n *           writer.startStep('Analyzing query...');\n *           const result = await streamText({\n *             model,\n *             prompt: 'Analyze: ' + ctx.messages[0].content,\n *           });\n *           await writer.forwardStream(result, { sendFinish: false });\n *           writer.endStep();\n *           return { ...ctx, analysis: await result.text };\n *         },\n *       },\n *       {\n *         name: 'respond',\n *         execute: async (ctx, writer) => {\n *           writer.startStep('Generating response...');\n *           const result = await streamText({\n *             model,\n *             prompt: `Based on: ${ctx.analysis}\\nRespond to user.`,\n *           });\n *           await writer.forwardStream(result);\n *           writer.endStep();\n *           return ctx;\n *         },\n *       },\n *     ],\n *   });\n * }\n * ```\n */\nexport function createMultiStepStreamResponse<TContext>(\n  options: MultiStepStreamOptions<TContext>,\n): Response {\n  const stream = createUIMessageStream({\n    execute: async ({ writer }) => {\n      await executeMultiStepStream(writer, options);\n    },\n  });\n\n  return createUIMessageStreamResponse({ stream });\n}\n\n/**\n * Create a multi-step stream (raw stream, not Response).\n */\nexport function createMultiStepStream<TContext>(\n  options: MultiStepStreamOptions<TContext>,\n): ReadableStream {\n  return createUIMessageStream({\n    execute: async ({ writer }) => {\n      await executeMultiStepStream(writer, options);\n    },\n  });\n}\n\n/**\n * Execute multi-step stream with provided writer.\n */\nexport async function executeMultiStepStream<TContext>(\n  rawWriter: MultiStepStreamWriter,\n  options: MultiStepStreamOptions<TContext>,\n): Promise<TContext> {\n  const { initialContext, steps, onError, onStepComplete } = options;\n  const writer = new MultiStepWriter(rawWriter);\n\n  let context = initialContext;\n\n  for (let i = 0; i < steps.length; i++) {\n    const step = steps[i];\n\n    // Check if step should run\n    if (!step || (step.shouldRun && !step.shouldRun(context))) {\n      continue;\n    }\n\n    try {\n      // Execute step\n      context = await step.execute(context, rawWriter);\n\n      // Notify completion\n      onStepComplete?.(i, context);\n    } catch (error) {\n      const err = error instanceof Error ? error : new Error(String(error));\n      writer.writeError(err);\n      onError?.(err, i);\n      throw err;\n    }\n  }\n\n  writer.finish();\n  return context;\n}\n\n// -----------------------------------------------------------------------------\n// Step Helpers\n// -----------------------------------------------------------------------------\n\n/**\n * Create a streaming text step.\n */\nexport function createTextStep<TContext>(config: {\n  name: string;\n  getPrompt: (context: TContext) => string;\n  streamText: (prompt: string) => Promise<MultiStepStreamResult>;\n  updateContext: (context: TContext, text: string) => TContext;\n  shouldRun?: (context: TContext) => boolean;\n}): StepDefinition<TContext> {\n  return {\n    name: config.name,\n    shouldRun: config.shouldRun,\n    execute: async (context, writer) => {\n      const msWriter = new MultiStepWriter(writer);\n      msWriter.startStep(config.name);\n\n      const prompt = config.getPrompt(context);\n      const result = await config.streamText(prompt);\n\n      await msWriter.forwardStream(result, { sendFinish: false });\n      msWriter.endStep();\n\n      const text = await result.text;\n      return config.updateContext(context, text);\n    },\n  };\n}\n\n/**\n * Create a tool execution step.\n */\nexport function createToolStep<TContext, TResult>(config: {\n  name: string;\n  toolName: string;\n  getInput: (context: TContext) => unknown;\n  executeTool: (input: unknown) => Promise<TResult>;\n  updateContext: (context: TContext, result: TResult) => TContext;\n  shouldRun?: (context: TContext) => boolean;\n}): StepDefinition<TContext> {\n  return {\n    name: config.name,\n    shouldRun: config.shouldRun,\n    execute: async (context, writer) => {\n      const msWriter = new MultiStepWriter(writer);\n      msWriter.startStep(config.name);\n\n      const input = config.getInput(context);\n      const toolCallId = `call_${Date.now()}`;\n\n      msWriter.writeToolCall(toolCallId, config.toolName, input);\n\n      const result = await config.executeTool(input);\n\n      msWriter.writeToolResult(toolCallId, config.toolName, result);\n      msWriter.endStep();\n\n      return config.updateContext(context, result);\n    },\n  };\n}\n\n/**\n * Create a conditional step that branches based on context.\n */\nexport function createConditionalStep<TContext>(config: {\n  name: string;\n  condition: (context: TContext) => boolean;\n  ifTrue: StepDefinition<TContext>;\n  ifFalse?: StepDefinition<TContext>;\n}): StepDefinition<TContext> {\n  return {\n    name: config.name,\n    execute: async (context, writer) => {\n      const step = config.condition(context) ? config.ifTrue : config.ifFalse;\n\n      if (step) {\n        return step.execute(context, writer);\n      }\n\n      return context;\n    },\n  };\n}\n\n/**\n * Create a parallel step that runs multiple steps concurrently.\n * Note: Results are merged into context in order of completion.\n */\nexport function createParallelStep<TContext>(config: {\n  name: string;\n  steps: StepDefinition<TContext>[];\n  mergeResults: (context: TContext, results: TContext[]) => TContext;\n}): StepDefinition<TContext> {\n  return {\n    name: config.name,\n    execute: async (context, writer) => {\n      const msWriter = new MultiStepWriter(writer);\n      msWriter.startStep(config.name);\n\n      // Execute all steps in parallel\n      const results = await Promise.all(config.steps.map(step => step.execute(context, writer)));\n\n      msWriter.endStep();\n\n      return config.mergeResults(context, results);\n    },\n  };\n}\n\n// -----------------------------------------------------------------------------\n// Utilities\n// -----------------------------------------------------------------------------\n\n/**\n * Create a simple multi-step pipeline.\n */\nexport function createStepPipeline<TContext>(\n  ...steps: StepDefinition<TContext>[]\n): StepDefinition<TContext>[] {\n  return steps;\n}\n\n/**\n * Chain steps sequentially.\n */\nexport function chainSteps<TContext>(\n  name: string,\n  steps: StepDefinition<TContext>[],\n): StepDefinition<TContext> {\n  return {\n    name,\n    execute: async (context, writer) => {\n      let current = context;\n      for (const step of steps) {\n        if (!step.shouldRun || step.shouldRun(current)) {\n          current = await step.execute(current, writer);\n        }\n      }\n      return current;\n    },\n  };\n}\n","/**\n * @fileoverview Auto-resume helpers for interrupted streams.\n */\n\nexport interface ResumeMessage {\n  id?: string;\n  role: 'system' | 'user' | 'assistant' | 'tool';\n  content: string;\n  createdAt?: Date;\n  metadata?: Record<string, unknown>;\n}\n\nexport interface PendingToolCall {\n  toolCallId: string;\n  toolName: string;\n  args?: Record<string, unknown>;\n  createdAt?: Date;\n}\n\nexport interface StreamResumeState {\n  streamId: string;\n  messages: ResumeMessage[];\n  pendingToolCalls: PendingToolCall[];\n  lastUpdatedAt: Date;\n  metadata?: Record<string, unknown>;\n}\n\nexport interface ResumeStorage {\n  get: (streamId: string) => Promise<StreamResumeState | null>;\n  set: (streamId: string, state: StreamResumeState, ttlMs?: number) => Promise<void>;\n  delete: (streamId: string) => Promise<void>;\n}\n\nexport interface AutoResumeConfig {\n  streamId: string;\n  storage: ResumeStorage;\n  ttlMs?: number;\n}\n\nexport interface AutoResumeHandler {\n  saveState: (\n    state: Omit<StreamResumeState, 'streamId' | 'lastUpdatedAt'>,\n  ) => Promise<StreamResumeState>;\n  loadState: () => Promise<StreamResumeState | null>;\n  clearState: () => Promise<void>;\n  detect: (\n    messages: ResumeMessage[],\n    pendingToolCalls?: PendingToolCall[],\n  ) => ResumeDetectionResult;\n}\n\nexport interface ResumeDetectionResult {\n  shouldResume: boolean;\n  reason: 'pending-tool-call' | 'last-message-user' | 'no-resume';\n  state?: StreamResumeState;\n}\n\nexport interface ResumeStreamPart {\n  type: 'data-appendMessage';\n  data: string;\n  transient?: boolean;\n}\n\nconst safeStringify = (value: unknown): string => {\n  const seen = new WeakSet<object>();\n\n  return JSON.stringify(value, (key, current) => {\n    void key;\n    if (typeof current === 'bigint') {\n      return current.toString();\n    }\n\n    if (current instanceof Date) {\n      return current.toISOString();\n    }\n\n    if (typeof current === 'object' && current !== null) {\n      if (seen.has(current)) {\n        return '[Circular]';\n      }\n      seen.add(current);\n    }\n\n    return current;\n  });\n};\n\ntype StoredState = {\n  state: StreamResumeState;\n  expiresAt?: number;\n};\n\nexport function createInMemoryStorage(): ResumeStorage {\n  const store = new Map<string, StoredState>();\n\n  return {\n    async get(streamId: string): Promise<StreamResumeState | null> {\n      const entry = store.get(streamId);\n      if (!entry) return null;\n\n      if (entry.expiresAt && Date.now() > entry.expiresAt) {\n        store.delete(streamId);\n        return null;\n      }\n\n      return entry.state;\n    },\n\n    async set(streamId: string, state: StreamResumeState, ttlMs?: number): Promise<void> {\n      store.set(streamId, {\n        state,\n        expiresAt: typeof ttlMs === 'number' ? Date.now() + ttlMs : undefined,\n      });\n    },\n\n    async delete(streamId: string): Promise<void> {\n      store.delete(streamId);\n    },\n  };\n}\n\nexport function createExternalStorage(storage: ResumeStorage): ResumeStorage {\n  return storage;\n}\n\nexport function detectResumeState(options: {\n  streamId: string;\n  messages: ResumeMessage[];\n  pendingToolCalls?: PendingToolCall[];\n  metadata?: Record<string, unknown>;\n}): ResumeDetectionResult {\n  const pendingToolCalls = options.pendingToolCalls ?? [];\n  const lastMessage = options.messages.at(-1);\n\n  if (pendingToolCalls.length > 0) {\n    return {\n      shouldResume: true,\n      reason: 'pending-tool-call',\n      state: {\n        streamId: options.streamId,\n        messages: options.messages,\n        pendingToolCalls,\n        lastUpdatedAt: new Date(),\n        metadata: options.metadata,\n      },\n    };\n  }\n\n  if (lastMessage?.role === 'user') {\n    return {\n      shouldResume: true,\n      reason: 'last-message-user',\n      state: {\n        streamId: options.streamId,\n        messages: options.messages,\n        pendingToolCalls,\n        lastUpdatedAt: new Date(),\n        metadata: options.metadata,\n      },\n    };\n  }\n\n  return {\n    shouldResume: false,\n    reason: 'no-resume',\n  };\n}\n\nexport function restoreDataStreamMessages(messages: ResumeMessage[]): ResumeStreamPart[] {\n  return messages.map(message => ({\n    type: 'data-appendMessage',\n    data: safeStringify(message),\n    transient: true,\n  }));\n}\n\nexport function buildContinuationPrompt(\n  messages: ResumeMessage[],\n  pendingToolCalls: PendingToolCall[] = [],\n): string {\n  const lastUser = [...messages].reverse().find(message => message.role === 'user');\n  const lastAssistant = [...messages].reverse().find(message => message.role === 'assistant');\n\n  const lines: string[] = [\n    'Continue the conversation from the last point without repeating prior content.',\n  ];\n\n  if (pendingToolCalls.length > 0) {\n    const pending = pendingToolCalls\n      .map(call => `${call.toolName} (${call.toolCallId})`)\n      .join(', ');\n    lines.push(`Pending tool calls: ${pending}.`);\n  }\n\n  if (lastUser) {\n    lines.push(`Last user message: ${lastUser.content}`);\n  }\n\n  if (lastAssistant) {\n    lines.push(`Last assistant message: ${lastAssistant.content}`);\n  }\n\n  return lines.join('\\n');\n}\n\nexport function createStreamStateTracker(options: {\n  streamId: string;\n  messages?: ResumeMessage[];\n  pendingToolCalls?: PendingToolCall[];\n  metadata?: Record<string, unknown>;\n}) {\n  const messages = [...(options.messages ?? [])];\n  const pendingToolCalls = [...(options.pendingToolCalls ?? [])];\n  let { metadata } = options;\n\n  return {\n    addMessage(message: ResumeMessage) {\n      messages.push(message);\n    },\n    addPendingToolCall(toolCall: PendingToolCall) {\n      pendingToolCalls.push(toolCall);\n    },\n    resolveToolCall(toolCallId: string) {\n      const index = pendingToolCalls.findIndex(call => call.toolCallId === toolCallId);\n      if (index >= 0) {\n        pendingToolCalls.splice(index, 1);\n      }\n    },\n    setMetadata(nextMetadata: Record<string, unknown>) {\n      metadata = nextMetadata;\n    },\n    getState(): StreamResumeState {\n      return {\n        streamId: options.streamId,\n        messages: [...messages],\n        pendingToolCalls: [...pendingToolCalls],\n        lastUpdatedAt: new Date(),\n        metadata,\n      };\n    },\n    reset() {\n      messages.length = 0;\n      pendingToolCalls.length = 0;\n      metadata = undefined;\n    },\n  };\n}\n\nexport function createAutoResume(config: AutoResumeConfig): AutoResumeHandler {\n  return {\n    async saveState(state) {\n      const nextState: StreamResumeState = {\n        streamId: config.streamId,\n        messages: state.messages,\n        pendingToolCalls: state.pendingToolCalls,\n        lastUpdatedAt: new Date(),\n        metadata: state.metadata,\n      };\n      await config.storage.set(config.streamId, nextState, config.ttlMs);\n      return nextState;\n    },\n    async loadState() {\n      return config.storage.get(config.streamId);\n    },\n    async clearState() {\n      await config.storage.delete(config.streamId);\n    },\n    detect(messages, pendingToolCalls) {\n      return detectResumeState({\n        streamId: config.streamId,\n        messages,\n        pendingToolCalls,\n      });\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAoCA,SAAgB,2BAA2B,QAAgD;AACzF,QAAO;EACL,QAAQ,IAAY;AAClB,UAAO,MAAM;IAAE,MAAM;IAAM,MAAM;IAAI,CAAC;;EAGxC,UAAU,MAAc;AACtB,UAAO,MAAM;IAAE,MAAM;IAAQ,MAAM;IAAM,CAAC;;EAG5C,WAAW,OAAe;AACxB,UAAO,MAAM;IAAE,MAAM;IAAS,MAAM;IAAO,CAAC;;EAG9C,YAAY,QAAqC;AAC/C,UAAO,MAAM;IAAE,MAAM;IAAU,MAAM;IAAQ,CAAC;;EAGhD,eAAe,MAAc;AAC3B,UAAO,MAAM;IAAE,MAAM;IAAa,MAAM;IAAM,CAAC;;EAGjD,eAAe,MAAc,UAAmB;AAC9C,UAAO,MAAM;IAAE,MAAM;IAAa,MAAM;KAAE;KAAM;KAAU;IAAE,CAAC;;EAG/D,cAAc,SAAiB,OAAe,SAAkB;AAC9D,UAAO,MAAM;IAAE,MAAM;IAAY,MAAM;KAAE;KAAS;KAAO;KAAS;IAAE,CAAC;;EAGvE,YAAY,SAAkB,UAAoC;AAChE,UAAO,MAAM;IAAE,MAAM;IAAU,MAAM;KAAE;KAAS;KAAU;IAAE,CAAC;;EAG/D,WAAW,SAAiB,MAAe,aAAuB;AAChE,UAAO,MAAM;IAAE,MAAM;IAAS,MAAM;KAAE;KAAS;KAAM;KAAa;IAAE,CAAC;;EAGvE,QAAQ;AACN,UAAO,MAAM;IAAE,MAAM;IAAS,MAAM;IAAM,CAAC;;EAE9C;;;;;;;;;;;;;;;;;;;;;AAsBH,eAAsB,eACpB,YACA,QAKA,WACe;CACf,MAAM,SAAS,2BAA2B,WAAW;AAErD,KAAI;AAEF,SAAO,QAAQ,OAAO,GAAG;AACzB,SAAO,UAAU,OAAO,KAAK;AAC7B,SAAO,WAAW,OAAO,MAAM;AAC/B,SAAO,YAAY,YAAY;AAG/B,aAAW,MAAM,SAAS,WAAW,CACnC,QAAO,eAAe,MAAM;AAI9B,SAAO,YAAY,WAAW;AAC9B,SAAO,YAAY,KAAK;UACjB,OAAO;AACd,SAAO,YAAY,QAAQ;AAC3B,SAAO,WACL,iBAAiB,QAAQ,MAAM,UAAU,iBACzC,gBACA,MACD;AACD,SAAO,YAAY,MAAM;AACzB,QAAM;;;;;;;;;;AAWV,SAAgB,2BACd,QACA,UAKI,EAAE,EACN;CACA,MAAM,EAAE,aAAa,IAAI,gBAAgB,QAAQ;CACjD,MAAM,SAAiD,EAAE;CACzD,IAAI,aAAmD;CAEvD,SAAS,QAAQ;AACf,MAAI,OAAO,WAAW,EAAG;AAEzB,OAAK,MAAM,QAAQ,OACjB,QAAO,MAAM,KAAK;AAEpB,SAAO,SAAS;AAEhB,MAAI,YAAY;AACd,gBAAa,WAAW;AACxB,gBAAa;;;CAIjB,SAAS,gBAAgB;AACvB,MAAI,WAAY;AAChB,eAAa,WAAW,OAAO,cAAc;;CAG/C,SAAS,MAAM,MAAc,MAAe;AAC1C,SAAO,KAAK;GAAE;GAAM;GAAM,CAAC;AAE3B,MAAI,OAAO,UAAU,WACnB,QAAO;MAEP,gBAAe;;AAInB,QAAO;EACL;EACA;EACA,IAAI,eAAe;AACjB,UAAO,OAAO;;EAEjB;;;;;;;;;;;;;;;;;;;;;;;ACzGH,IAAa,kBAAb,MAA6B;CAC3B,AAAQ,cAAc;CACtB,AAAQ,cAAc;CAEtB,YACE,AAAiB,QACjB,AAAiB,SACjB;EAFiB;EACA;;;;;;CAOnB,UAAU,MAAoB;AAC5B,MAAI,KAAK,MAAM,CAAC,WAAW,EACzB,OAAM,IAAI,MAAM,wBAAwB;AAE1C,OAAK,cAAc;AAEnB,OAAK,OAAO,MAAM;GAChB,MAAM;GACN,MAAM;IAAE,OAAO;IAAc,WAAW,KAAK;IAAa,UAAU;IAAM;GAC3E,CAAC;;;;;CAMJ,UAAgB;AACd,OAAK,OAAO,MAAM;GAChB,MAAM;GACN,MAAM;IAAE,OAAO;IAAY,WAAW,KAAK;IAAa;GACzD,CAAC;AACF,OAAK;AACL,OAAK,cAAc;;;;;;CAOrB,UAAU,MAAoB;AAC5B,OAAK,OAAO,MAAM;GAChB,MAAM;GACN,IAAI,QAAQ,KAAK;GACjB,OAAO;GACR,CAAC;;;;;CAMJ,cAAc,IAAY,MAAc,OAAsB;AAC5D,OAAK,OAAO,MAAM;GAChB,MAAM;GACN,MAAM;IACJ,OAAO;IACP,WAAW,KAAK;IAChB,UAAU;KAAE;KAAI;KAAM;KAAO;IAC9B;GACF,CAAC;;;;;CAMJ,gBAAgB,IAAY,MAAc,QAAuB;AAC/D,OAAK,OAAO,MAAM;GAChB,MAAM;GACN,MAAM;IACJ,OAAO;IACP,WAAW,KAAK;IAChB,YAAY;KAAE;KAAI;KAAM;KAAQ;IACjC;GACF,CAAC;;;;;;CAOJ,MAAM,cACJ,QACA,SACe;EAIf,MAAM,WAAW,OAAO,qBAAmC;AAC3D,MAAI,SACF,MAAK,OAAO,MAAM,SAAS;;;;;CAO/B,WAAW,OAAoB;AAC7B,OAAK,OAAO,MAAM;GAChB,MAAM;GACN,WAAW,MAAM;GAClB,CAAC;;;;;CAMJ,SAAe;AACb,MAAI,KAAK,SAAS,eAAe,MAC/B,MAAK,OAAO,MAAM;GAChB,MAAM;GACN,cAAc;GACf,CAAC;;;CAKN,IAAI,YAAoB;AACtB,SAAO,KAAK;;;CAId,IAAI,MAA6B;AAC/B,SAAO,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmDhB,SAAgB,8BACd,SACU;AAOV,QAAO,8BAA8B,EAAE,QANxB,sBAAsB,EACnC,SAAS,OAAO,EAAE,aAAa;AAC7B,QAAM,uBAAuB,QAAQ,QAAQ;IAEhD,CAAC,EAE6C,CAAC;;;;;AAMlD,SAAgB,sBACd,SACgB;AAChB,QAAO,sBAAsB,EAC3B,SAAS,OAAO,EAAE,aAAa;AAC7B,QAAM,uBAAuB,QAAQ,QAAQ;IAEhD,CAAC;;;;;AAMJ,eAAsB,uBACpB,WACA,SACmB;CACnB,MAAM,EAAE,gBAAgB,OAAO,SAAS,mBAAmB;CAC3D,MAAM,SAAS,IAAI,gBAAgB,UAAU;CAE7C,IAAI,UAAU;AAEd,MAAK,IAAI,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;EACrC,MAAM,OAAO,MAAM;AAGnB,MAAI,CAAC,QAAS,KAAK,aAAa,CAAC,KAAK,UAAU,QAAQ,CACtD;AAGF,MAAI;AAEF,aAAU,MAAM,KAAK,QAAQ,SAAS,UAAU;AAGhD,oBAAiB,GAAG,QAAQ;WACrB,OAAO;GACd,MAAM,MAAM,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,MAAM,CAAC;AACrE,UAAO,WAAW,IAAI;AACtB,aAAU,KAAK,EAAE;AACjB,SAAM;;;AAIV,QAAO,QAAQ;AACf,QAAO;;;;;AAUT,SAAgB,eAAyB,QAMZ;AAC3B,QAAO;EACL,MAAM,OAAO;EACb,WAAW,OAAO;EAClB,SAAS,OAAO,SAAS,WAAW;GAClC,MAAM,WAAW,IAAI,gBAAgB,OAAO;AAC5C,YAAS,UAAU,OAAO,KAAK;GAE/B,MAAM,SAAS,OAAO,UAAU,QAAQ;GACxC,MAAM,SAAS,MAAM,OAAO,WAAW,OAAO;AAE9C,SAAM,SAAS,cAAc,QAAQ,EAAE,YAAY,OAAO,CAAC;AAC3D,YAAS,SAAS;GAElB,MAAM,OAAO,MAAM,OAAO;AAC1B,UAAO,OAAO,cAAc,SAAS,KAAK;;EAE7C;;;;;AAMH,SAAgB,eAAkC,QAOrB;AAC3B,QAAO;EACL,MAAM,OAAO;EACb,WAAW,OAAO;EAClB,SAAS,OAAO,SAAS,WAAW;GAClC,MAAM,WAAW,IAAI,gBAAgB,OAAO;AAC5C,YAAS,UAAU,OAAO,KAAK;GAE/B,MAAM,QAAQ,OAAO,SAAS,QAAQ;GACtC,MAAM,aAAa,QAAQ,KAAK,KAAK;AAErC,YAAS,cAAc,YAAY,OAAO,UAAU,MAAM;GAE1D,MAAM,SAAS,MAAM,OAAO,YAAY,MAAM;AAE9C,YAAS,gBAAgB,YAAY,OAAO,UAAU,OAAO;AAC7D,YAAS,SAAS;AAElB,UAAO,OAAO,cAAc,SAAS,OAAO;;EAE/C;;;;;AAMH,SAAgB,sBAAgC,QAKnB;AAC3B,QAAO;EACL,MAAM,OAAO;EACb,SAAS,OAAO,SAAS,WAAW;GAClC,MAAM,OAAO,OAAO,UAAU,QAAQ,GAAG,OAAO,SAAS,OAAO;AAEhE,OAAI,KACF,QAAO,KAAK,QAAQ,SAAS,OAAO;AAGtC,UAAO;;EAEV;;;;;;AAOH,SAAgB,mBAA6B,QAIhB;AAC3B,QAAO;EACL,MAAM,OAAO;EACb,SAAS,OAAO,SAAS,WAAW;GAClC,MAAM,WAAW,IAAI,gBAAgB,OAAO;AAC5C,YAAS,UAAU,OAAO,KAAK;GAG/B,MAAM,UAAU,MAAM,QAAQ,IAAI,OAAO,MAAM,KAAI,SAAQ,KAAK,QAAQ,SAAS,OAAO,CAAC,CAAC;AAE1F,YAAS,SAAS;AAElB,UAAO,OAAO,aAAa,SAAS,QAAQ;;EAE/C;;;;;AAUH,SAAgB,mBACd,GAAG,OACyB;AAC5B,QAAO;;;;;AAMT,SAAgB,WACd,MACA,OAC0B;AAC1B,QAAO;EACL;EACA,SAAS,OAAO,SAAS,WAAW;GAClC,IAAI,UAAU;AACd,QAAK,MAAM,QAAQ,MACjB,KAAI,CAAC,KAAK,aAAa,KAAK,UAAU,QAAQ,CAC5C,WAAU,MAAM,KAAK,QAAQ,SAAS,OAAO;AAGjD,UAAO;;EAEV;;;;;ACtZH,MAAM,iBAAiB,UAA2B;CAChD,MAAM,uBAAO,IAAI,SAAiB;AAElC,QAAO,KAAK,UAAU,QAAQ,KAAK,YAAY;AAE7C,MAAI,OAAO,YAAY,SACrB,QAAO,QAAQ,UAAU;AAG3B,MAAI,mBAAmB,KACrB,QAAO,QAAQ,aAAa;AAG9B,MAAI,OAAO,YAAY,YAAY,YAAY,MAAM;AACnD,OAAI,KAAK,IAAI,QAAQ,CACnB,QAAO;AAET,QAAK,IAAI,QAAQ;;AAGnB,SAAO;GACP;;AAQJ,SAAgB,wBAAuC;CACrD,MAAM,wBAAQ,IAAI,KAA0B;AAE5C,QAAO;EACL,MAAM,IAAI,UAAqD;GAC7D,MAAM,QAAQ,MAAM,IAAI,SAAS;AACjC,OAAI,CAAC,MAAO,QAAO;AAEnB,OAAI,MAAM,aAAa,KAAK,KAAK,GAAG,MAAM,WAAW;AACnD,UAAM,OAAO,SAAS;AACtB,WAAO;;AAGT,UAAO,MAAM;;EAGf,MAAM,IAAI,UAAkB,OAA0B,OAA+B;AACnF,SAAM,IAAI,UAAU;IAClB;IACA,WAAW,OAAO,UAAU,WAAW,KAAK,KAAK,GAAG,QAAQ;IAC7D,CAAC;;EAGJ,MAAM,OAAO,UAAiC;AAC5C,SAAM,OAAO,SAAS;;EAEzB;;AAGH,SAAgB,sBAAsB,SAAuC;AAC3E,QAAO;;AAGT,SAAgB,kBAAkB,SAKR;CACxB,MAAM,mBAAmB,QAAQ,oBAAoB,EAAE;CACvD,MAAM,cAAc,QAAQ,SAAS,GAAG,GAAG;AAE3C,KAAI,iBAAiB,SAAS,EAC5B,QAAO;EACL,cAAc;EACd,QAAQ;EACR,OAAO;GACL,UAAU,QAAQ;GAClB,UAAU,QAAQ;GAClB;GACA,+BAAe,IAAI,MAAM;GACzB,UAAU,QAAQ;GACnB;EACF;AAGH,KAAI,aAAa,SAAS,OACxB,QAAO;EACL,cAAc;EACd,QAAQ;EACR,OAAO;GACL,UAAU,QAAQ;GAClB,UAAU,QAAQ;GAClB;GACA,+BAAe,IAAI,MAAM;GACzB,UAAU,QAAQ;GACnB;EACF;AAGH,QAAO;EACL,cAAc;EACd,QAAQ;EACT;;AAGH,SAAgB,0BAA0B,UAA+C;AACvF,QAAO,SAAS,KAAI,aAAY;EAC9B,MAAM;EACN,MAAM,cAAc,QAAQ;EAC5B,WAAW;EACZ,EAAE;;AAGL,SAAgB,wBACd,UACA,mBAAsC,EAAE,EAChC;CACR,MAAM,WAAW,CAAC,GAAG,SAAS,CAAC,SAAS,CAAC,MAAK,YAAW,QAAQ,SAAS,OAAO;CACjF,MAAM,gBAAgB,CAAC,GAAG,SAAS,CAAC,SAAS,CAAC,MAAK,YAAW,QAAQ,SAAS,YAAY;CAE3F,MAAM,QAAkB,CACtB,iFACD;AAED,KAAI,iBAAiB,SAAS,GAAG;EAC/B,MAAM,UAAU,iBACb,KAAI,SAAQ,GAAG,KAAK,SAAS,IAAI,KAAK,WAAW,GAAG,CACpD,KAAK,KAAK;AACb,QAAM,KAAK,uBAAuB,QAAQ,GAAG;;AAG/C,KAAI,SACF,OAAM,KAAK,sBAAsB,SAAS,UAAU;AAGtD,KAAI,cACF,OAAM,KAAK,2BAA2B,cAAc,UAAU;AAGhE,QAAO,MAAM,KAAK,KAAK;;AAGzB,SAAgB,yBAAyB,SAKtC;CACD,MAAM,WAAW,CAAC,GAAI,QAAQ,YAAY,EAAE,CAAE;CAC9C,MAAM,mBAAmB,CAAC,GAAI,QAAQ,oBAAoB,EAAE,CAAE;CAC9D,IAAI,EAAE,aAAa;AAEnB,QAAO;EACL,WAAW,SAAwB;AACjC,YAAS,KAAK,QAAQ;;EAExB,mBAAmB,UAA2B;AAC5C,oBAAiB,KAAK,SAAS;;EAEjC,gBAAgB,YAAoB;GAClC,MAAM,QAAQ,iBAAiB,WAAU,SAAQ,KAAK,eAAe,WAAW;AAChF,OAAI,SAAS,EACX,kBAAiB,OAAO,OAAO,EAAE;;EAGrC,YAAY,cAAuC;AACjD,cAAW;;EAEb,WAA8B;AAC5B,UAAO;IACL,UAAU,QAAQ;IAClB,UAAU,CAAC,GAAG,SAAS;IACvB,kBAAkB,CAAC,GAAG,iBAAiB;IACvC,+BAAe,IAAI,MAAM;IACzB;IACD;;EAEH,QAAQ;AACN,YAAS,SAAS;AAClB,oBAAiB,SAAS;AAC1B,cAAW;;EAEd;;AAGH,SAAgB,iBAAiB,QAA6C;AAC5E,QAAO;EACL,MAAM,UAAU,OAAO;GACrB,MAAM,YAA+B;IACnC,UAAU,OAAO;IACjB,UAAU,MAAM;IAChB,kBAAkB,MAAM;IACxB,+BAAe,IAAI,MAAM;IACzB,UAAU,MAAM;IACjB;AACD,SAAM,OAAO,QAAQ,IAAI,OAAO,UAAU,WAAW,OAAO,MAAM;AAClE,UAAO;;EAET,MAAM,YAAY;AAChB,UAAO,OAAO,QAAQ,IAAI,OAAO,SAAS;;EAE5C,MAAM,aAAa;AACjB,SAAM,OAAO,QAAQ,OAAO,OAAO,SAAS;;EAE9C,OAAO,UAAU,kBAAkB;AACjC,UAAO,kBAAkB;IACvB,UAAU,OAAO;IACjB;IACA;IACD,CAAC;;EAEL"}

package/dist/budget-guard-C83KCH9V.d.mts

@@ -0,0 +1,52 @@
+import { Ur as SDKLanguageModelV3Middleware } from "./ai-runtime-DIwOEc6g.mjs";
+import { n as Model } from "./registry-DVPWzkXR.mjs";
+
+//#region src/observability/cost/types.d.ts
+interface UsageEvent {
+  model: Model;
+  inputTokens: number;
+  outputTokens: number;
+  cost: number;
+  timestamp: Date;
+  metadata: CostMetadata;
+}
+interface CostMetadata {
+  userId?: string;
+  teamId?: string;
+  feature?: string;
+  requestId?: string;
+  tags?: Record<string, string>;
+}
+interface BudgetConfig {
+  maxDailyCost: number;
+  maxMonthlyCost: number;
+  warningThreshold?: number;
+  maxRequestCost?: number;
+}
+interface UsageSummary {
+  daily: number;
+  monthly: number;
+  requestCount: number;
+}
+type UsageCallback = (event: UsageEvent) => Promise<void>;
+declare class BudgetExceededError extends Error {
+  readonly budgetType: 'daily' | 'monthly' | 'request';
+  readonly current: number;
+  readonly limit: number;
+  constructor(message: string, budgetType: 'daily' | 'monthly' | 'request', current: number, limit: number);
+}
+//#endregion
+//#region src/observability/cost/tracking/usage-tracker.d.ts
+declare function calculateCost(model: Model, usage?: {
+  inputTokens?: unknown;
+  outputTokens?: unknown;
+}): number;
+declare function extractMetadata(params: unknown): CostMetadata;
+declare function createUsageTracker(callback: UsageCallback): SDKLanguageModelV3Middleware;
+//#endregion
+//#region src/observability/cost/tracking/budget-guard.d.ts
+type GetCurrentUsage = () => Promise<UsageSummary>;
+declare function createBudgetGuard(config: BudgetConfig, getCurrentUsage: GetCurrentUsage): SDKLanguageModelV3Middleware;
+//#endregion
+export { extractMetadata as a, CostMetadata as c, UsageSummary as d, createUsageTracker as i, UsageCallback as l, createBudgetGuard as n, BudgetConfig as o, calculateCost as r, BudgetExceededError as s, GetCurrentUsage as t, UsageEvent as u };
+//# sourceMappingURL=budget-guard-C83KCH9V.d.mts.map

package/dist/budget-guard-C83KCH9V.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-guard-C83KCH9V.d.mts","names":[],"sources":["../src/observability/cost/types.ts","../src/observability/cost/tracking/usage-tracker.ts","../src/observability/cost/tracking/budget-guard.ts"],"mappings":";;;;UASiB,UAAA;EAEf,KAAA,EAAO,KAAA;EAEP,WAAA;EAEA,YAAA;EAEA,IAAA;EAEA,SAAA,EAAW,IAAA;EAEX,QAAA,EAAU,YAAA;AAAA;AAAA,UAMK,YAAA;EAEf,MAAA;EAEA,MAAA;EAEA,OAAA;EAEA,SAAA;EAEA,IAAA,GAAO,MAAA;AAAA;AAAA,UAMQ,YAAA;EAEf,YAAA;EAEA,cAAA;EAEA,gBAAA;EAEA,cAAA;AAAA;AAAA,UAMe,YAAA;EAEf,KAAA;EAEA,OAAA;EAEA,YAAA;AAAA;AAAA,KAMU,aAAA,IAAiB,KAAA,EAAO,UAAA,KAAe,OAAA;AAAA,cAKtC,mBAAA,SAA4B,KAAA;EAAA,SAGrB,UAAA;EAAA,SACA,OAAA;EAAA,SACA,KAAA;cAHhB,OAAA,UACgB,UAAA,mCACA,OAAA,UACA,KAAA;AAAA;;;iBCvDX,aAAA,CACP,KAAA,EAAO,KAAA,EACP,KAAA;EAAU,WAAA;EAAuB,YAAA;AAAA;AAAA,iBAgB1B,eAAA,CAAgB,MAAA,YAAkB,YAAA;AAAA,iBA4B3B,kBAAA,CAAmB,QAAA,EAAU,aAAA,GAAgB,4BAAA;;;KCtDjD,eAAA,SAAwB,OAAA,CAAQ,YAAA;AAAA,iBA4B5B,iBAAA,CACd,MAAA,EAAQ,YAAA,EACR,eAAA,EAAiB,eAAA,GAChB,4BAAA"}

package/dist/budget-guard-d_b5rq4u.mjs

@@ -0,0 +1,158 @@
+import { i as getModelMetadata, n as Model } from "./registry-CsD3iTIx.mjs";
+import { logError, logWarn } from "@od-oneapp/shared/logs";
+
+//#region src/observability/cost/types.ts
+/**
+* Budget exceeded error
+*/
+var BudgetExceededError = class extends Error {
+	constructor(message, budgetType, current, limit) {
+		super(message);
+		this.budgetType = budgetType;
+		this.current = current;
+		this.limit = limit;
+		this.name = "BudgetExceededError";
+	}
+};
+
+//#endregion
+//#region src/observability/cost/tracking/usage-tracker.ts
+/**
+* Usage tracking middleware for AI SDK
+*/
+/**
+* Calculate cost based on token usage and model pricing
+*/
+const getUsageTotal = (value) => {
+	if (typeof value === "number") return value;
+	if (value && typeof value === "object" && "total" in value) {
+		const { total } = value;
+		return typeof total === "number" ? total : 0;
+	}
+	return 0;
+};
+function calculateCost(model, usage) {
+	if (!usage) return 0;
+	const metadata = getModelMetadata(model);
+	const inputTokens = getUsageTotal(usage.inputTokens);
+	const outputTokens = getUsageTotal(usage.outputTokens);
+	return inputTokens / 1e3 * metadata.costPer1kInput + outputTokens / 1e3 * metadata.costPer1kOutput;
+}
+/**
+* Extract cost metadata from request parameters
+*/
+function extractMetadata(params) {
+	const headers = params.headers;
+	return {
+		userId: headers?.["x-user-id"] ?? void 0,
+		teamId: headers?.["x-team-id"] ?? void 0,
+		feature: headers?.["x-feature"] ?? void 0,
+		requestId: headers?.["x-request-id"] ?? void 0
+	};
+}
+/**
+* Create usage tracking middleware
+*
+* @example
+* ```ts
+* const tracker = createUsageTracker(async (event) => {
+*   await db.insert(usageTable).values(event);
+*   console.log(`Cost: $${event.cost.toFixed(4)}`);
+* });
+*
+* const model = wrapLanguageModel({
+*   model: createLanguageModel(Model.SONNET),
+*   middleware: tracker,
+* });
+* ```
+*/
+function createUsageTracker(callback) {
+	return {
+		specificationVersion: "v3",
+		wrapGenerate: async ({ doGenerate, params }) => {
+			const result = await doGenerate();
+			const rawParams = params;
+			const modelId = (typeof rawParams.model === "string" ? rawParams.model : void 0) ?? (typeof rawParams.modelId === "string" ? rawParams.modelId : void 0);
+			const model = Object.values(Model).find((entry) => entry === modelId) ?? Model.SONNET;
+			const event = {
+				model,
+				inputTokens: getUsageTotal(result.usage?.inputTokens),
+				outputTokens: getUsageTotal(result.usage?.outputTokens),
+				cost: calculateCost(model, result.usage),
+				timestamp: /* @__PURE__ */ new Date(),
+				metadata: extractMetadata(params)
+			};
+			(async () => {
+				try {
+					await callback(event);
+				} catch (err) {
+					logError("[UsageTracker] Callback error", { error: err instanceof Error ? err : new Error(String(err)) });
+				}
+			})();
+			return result;
+		},
+		wrapStream: async ({ doStream }) => {
+			const { stream, ...rest } = await doStream();
+			return {
+				stream,
+				...rest
+			};
+		}
+	};
+}
+
+//#endregion
+//#region src/observability/cost/tracking/budget-guard.ts
+/**
+* Budget guard middleware for cost control
+*/
+/**
+* Create budget guard middleware that enforces spending limits
+*
+* @example
+* ```ts
+* const budgetGuard = createBudgetGuard(
+*   {
+*     maxDailyCost: 100,
+*     maxMonthlyCost: 2000,
+*     warningThreshold: 0.8,
+*   },
+*   async () => {
+*     const [daily, monthly] = await Promise.all([
+*       db.query.dailyUsage(),
+*       db.query.monthlyUsage(),
+*     ]);
+*     return { daily, monthly, requestCount: 0 };
+*   }
+* );
+*
+* const model = wrapLanguageModel({
+*   model: createLanguageModel(Model.SONNET),
+*   middleware: budgetGuard,
+* });
+* ```
+*/
+function createBudgetGuard(config, getCurrentUsage) {
+	const { maxDailyCost, maxMonthlyCost, warningThreshold = .8 } = config;
+	return {
+		specificationVersion: "v3",
+		wrapGenerate: async ({ doGenerate }) => {
+			const usage = await getCurrentUsage();
+			if (usage.daily >= maxDailyCost) throw new BudgetExceededError(`Daily budget exceeded: $${usage.daily.toFixed(2)} / $${maxDailyCost}`, "daily", usage.daily, maxDailyCost);
+			if (usage.monthly >= maxMonthlyCost) throw new BudgetExceededError(`Monthly budget exceeded: $${usage.monthly.toFixed(2)} / $${maxMonthlyCost}`, "monthly", usage.monthly, maxMonthlyCost);
+			if (usage.daily >= maxDailyCost * warningThreshold) logWarn(`[BudgetGuard] Daily budget warning: ${(usage.daily / maxDailyCost * 100).toFixed(1)}% used`);
+			if (usage.monthly >= maxMonthlyCost * warningThreshold) logWarn(`[BudgetGuard] Monthly budget warning: ${(usage.monthly / maxMonthlyCost * 100).toFixed(1)}% used`);
+			return doGenerate();
+		},
+		wrapStream: async ({ doStream }) => {
+			const usage = await getCurrentUsage();
+			if (usage.daily >= maxDailyCost) throw new BudgetExceededError(`Daily budget exceeded`, "daily", usage.daily, maxDailyCost);
+			if (usage.monthly >= maxMonthlyCost) throw new BudgetExceededError(`Monthly budget exceeded`, "monthly", usage.monthly, maxMonthlyCost);
+			return doStream();
+		}
+	};
+}
+
+//#endregion
+export { BudgetExceededError as a, extractMetadata as i, calculateCost as n, createUsageTracker as r, createBudgetGuard as t };
+//# sourceMappingURL=budget-guard-d_b5rq4u.mjs.map

package/dist/budget-guard-d_b5rq4u.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-guard-d_b5rq4u.mjs","names":[],"sources":["../src/observability/cost/types.ts","../src/observability/cost/tracking/usage-tracker.ts","../src/observability/cost/tracking/budget-guard.ts"],"sourcesContent":["/**\n * Cost tracking and attribution types\n */\n\nimport type { Model } from '../../models/registry';\n\n/**\n * Token usage from a single AI request\n */\nexport interface UsageEvent {\n  /** Semantic model name */\n  model: Model;\n  /** Input/prompt tokens */\n  inputTokens: number;\n  /** Output/completion tokens */\n  outputTokens: number;\n  /** Calculated cost in USD */\n  cost: number;\n  /** When the request occurred */\n  timestamp: Date;\n  /** Attribution metadata */\n  metadata: CostMetadata;\n}\n\n/**\n * Cost attribution metadata\n */\nexport interface CostMetadata {\n  /** User ID for per-user tracking */\n  userId?: string;\n  /** Team ID for team-level aggregation */\n  teamId?: string;\n  /** Feature/product area for attribution */\n  feature?: string;\n  /** Request ID for correlation */\n  requestId?: string;\n  /** Additional custom tags */\n  tags?: Record<string, string>;\n}\n\n/**\n * Budget configuration for cost controls\n */\nexport interface BudgetConfig {\n  /** Maximum daily spend in USD */\n  maxDailyCost: number;\n  /** Maximum monthly spend in USD */\n  maxMonthlyCost: number;\n  /** Warning threshold (0-1), e.g., 0.8 = 80% */\n  warningThreshold?: number;\n  /** Per-request cost limit */\n  maxRequestCost?: number;\n}\n\n/**\n * Current usage summary\n */\nexport interface UsageSummary {\n  /** Daily spend so far */\n  daily: number;\n  /** Monthly spend so far */\n  monthly: number;\n  /** Request count */\n  requestCount: number;\n}\n\n/**\n * Usage callback for tracking\n */\nexport type UsageCallback = (event: UsageEvent) => Promise<void>;\n\n/**\n * Budget exceeded error\n */\nexport class BudgetExceededError extends Error {\n  constructor(\n    message: string,\n    public readonly budgetType: 'daily' | 'monthly' | 'request',\n    public readonly current: number,\n    public readonly limit: number,\n  ) {\n    super(message);\n    this.name = 'BudgetExceededError';\n  }\n}\n","/**\n * Usage tracking middleware for AI SDK\n */\n\nimport { logError } from '@repo/shared/logs';\n\nimport { Model, getModelMetadata } from '../../../models/registry';\n\nimport type { WrapGenerateArgs, WrapStreamArgs } from '../../../sdk/middleware/model-middleware';\nimport type { SDKLanguageModelV3Middleware } from '../../../shared';\nimport type { UsageEvent, UsageCallback, CostMetadata } from '../types';\n\n/**\n * Calculate cost based on token usage and model pricing\n */\nconst getUsageTotal = (value: unknown): number => {\n  if (typeof value === 'number') return value;\n  if (value && typeof value === 'object' && 'total' in value) {\n    const { total } = value as { total?: unknown };\n    return typeof total === 'number' ? total : 0;\n  }\n  return 0;\n};\n\nfunction calculateCost(\n  model: Model,\n  usage?: { inputTokens?: unknown; outputTokens?: unknown },\n): number {\n  if (!usage) return 0;\n\n  const metadata = getModelMetadata(model);\n  const inputTokens = getUsageTotal(usage.inputTokens);\n  const outputTokens = getUsageTotal(usage.outputTokens);\n  const inputCost = (inputTokens / 1000) * metadata.costPer1kInput;\n  const outputCost = (outputTokens / 1000) * metadata.costPer1kOutput;\n\n  return inputCost + outputCost;\n}\n\n/**\n * Extract cost metadata from request parameters\n */\nfunction extractMetadata(params: unknown): CostMetadata {\n  const p = params as Record<string, unknown>;\n  const headers = p.headers as Record<string, string> | undefined;\n\n  return {\n    userId: headers?.['x-user-id'] ?? undefined,\n    teamId: headers?.['x-team-id'] ?? undefined,\n    feature: headers?.['x-feature'] ?? undefined,\n    requestId: headers?.['x-request-id'] ?? undefined,\n  };\n}\n\n/**\n * Create usage tracking middleware\n *\n * @example\n * ```ts\n * const tracker = createUsageTracker(async (event) => {\n *   await db.insert(usageTable).values(event);\n *   console.log(`Cost: $${event.cost.toFixed(4)}`);\n * });\n *\n * const model = wrapLanguageModel({\n *   model: createLanguageModel(Model.SONNET),\n *   middleware: tracker,\n * });\n * ```\n */\nexport function createUsageTracker(callback: UsageCallback): SDKLanguageModelV3Middleware {\n  return {\n    specificationVersion: 'v3',\n    wrapGenerate: async ({ doGenerate, params }: WrapGenerateArgs) => {\n      const result = await doGenerate();\n\n      // Try to determine model from params\n      const rawParams = params as Record<string, unknown>;\n      const modelId =\n        (typeof rawParams.model === 'string' ? rawParams.model : undefined) ??\n        (typeof rawParams.modelId === 'string' ? rawParams.modelId : undefined);\n      const model = Object.values(Model).find(entry => entry === modelId) ?? Model.SONNET;\n\n      const event: UsageEvent = {\n        model,\n        inputTokens: getUsageTotal(result.usage?.inputTokens),\n        outputTokens: getUsageTotal(result.usage?.outputTokens),\n        cost: calculateCost(model, result.usage),\n        timestamp: new Date(),\n        metadata: extractMetadata(params),\n      };\n\n      // Fire callback async (don't block response)\n      void (async () => {\n        try {\n          await callback(event);\n        } catch (err) {\n          logError('[UsageTracker] Callback error', {\n            error: err instanceof Error ? err : new Error(String(err)),\n          });\n        }\n      })();\n\n      return result;\n    },\n\n    wrapStream: async ({ doStream }: WrapStreamArgs) => {\n      const { stream, ...rest } = await doStream();\n\n      // For streaming, we track usage from the final chunk\n      // The AI SDK accumulates usage across chunks\n      return {\n        stream,\n        ...rest,\n      };\n    },\n  };\n}\n\nexport { calculateCost, extractMetadata };\n","/**\n * Budget guard middleware for cost control\n */\n\nimport { logWarn } from '@repo/shared/logs';\n\nimport { BudgetExceededError } from '../types';\n\nimport type { WrapGenerateArgs, WrapStreamArgs } from '../../../sdk/middleware/model-middleware';\nimport type { SDKLanguageModelV3Middleware } from '../../../shared';\nimport type { BudgetConfig, UsageSummary } from '../types';\n\n\n/**\n * Function to get current usage - implement with your storage\n */\nexport type GetCurrentUsage = () => Promise<UsageSummary>;\n\n/**\n * Create budget guard middleware that enforces spending limits\n *\n * @example\n * ```ts\n * const budgetGuard = createBudgetGuard(\n *   {\n *     maxDailyCost: 100,\n *     maxMonthlyCost: 2000,\n *     warningThreshold: 0.8,\n *   },\n *   async () => {\n *     const [daily, monthly] = await Promise.all([\n *       db.query.dailyUsage(),\n *       db.query.monthlyUsage(),\n *     ]);\n *     return { daily, monthly, requestCount: 0 };\n *   }\n * );\n *\n * const model = wrapLanguageModel({\n *   model: createLanguageModel(Model.SONNET),\n *   middleware: budgetGuard,\n * });\n * ```\n */\nexport function createBudgetGuard(\n  config: BudgetConfig,\n  getCurrentUsage: GetCurrentUsage,\n): SDKLanguageModelV3Middleware {\n  const { maxDailyCost, maxMonthlyCost, warningThreshold = 0.8 } = config;\n\n  return {\n    specificationVersion: 'v3',\n    wrapGenerate: async ({ doGenerate }: WrapGenerateArgs) => {\n      const usage = await getCurrentUsage();\n\n      // Check daily limit\n      if (usage.daily >= maxDailyCost) {\n        throw new BudgetExceededError(\n          `Daily budget exceeded: $${usage.daily.toFixed(2)} / $${maxDailyCost}`,\n          'daily',\n          usage.daily,\n          maxDailyCost,\n        );\n      }\n\n      // Check monthly limit\n      if (usage.monthly >= maxMonthlyCost) {\n        throw new BudgetExceededError(\n          `Monthly budget exceeded: $${usage.monthly.toFixed(2)} / $${maxMonthlyCost}`,\n          'monthly',\n          usage.monthly,\n          maxMonthlyCost,\n        );\n      }\n\n      // Log warning if approaching limits\n      if (usage.daily >= maxDailyCost * warningThreshold) {\n        logWarn(\n          `[BudgetGuard] Daily budget warning: ${((usage.daily / maxDailyCost) * 100).toFixed(1)}% used`,\n        );\n      }\n      if (usage.monthly >= maxMonthlyCost * warningThreshold) {\n        logWarn(\n          `[BudgetGuard] Monthly budget warning: ${((usage.monthly / maxMonthlyCost) * 100).toFixed(1)}% used`,\n        );\n      }\n\n      return doGenerate();\n    },\n\n    wrapStream: async ({ doStream }: WrapStreamArgs) => {\n      const usage = await getCurrentUsage();\n\n      if (usage.daily >= maxDailyCost) {\n        throw new BudgetExceededError(`Daily budget exceeded`, 'daily', usage.daily, maxDailyCost);\n      }\n\n      if (usage.monthly >= maxMonthlyCost) {\n        throw new BudgetExceededError(\n          `Monthly budget exceeded`,\n          'monthly',\n          usage.monthly,\n          maxMonthlyCost,\n        );\n      }\n\n      return doStream();\n    },\n  };\n}\n"],"mappings":";;;;;;;AA0EA,IAAa,sBAAb,cAAyC,MAAM;CAC7C,YACE,SACA,AAAgB,YAChB,AAAgB,SAChB,AAAgB,OAChB;AACA,QAAM,QAAQ;EAJE;EACA;EACA;AAGhB,OAAK,OAAO;;;;;;;;;;;;ACnEhB,MAAM,iBAAiB,UAA2B;AAChD,KAAI,OAAO,UAAU,SAAU,QAAO;AACtC,KAAI,SAAS,OAAO,UAAU,YAAY,WAAW,OAAO;EAC1D,MAAM,EAAE,UAAU;AAClB,SAAO,OAAO,UAAU,WAAW,QAAQ;;AAE7C,QAAO;;AAGT,SAAS,cACP,OACA,OACQ;AACR,KAAI,CAAC,MAAO,QAAO;CAEnB,MAAM,WAAW,iBAAiB,MAAM;CACxC,MAAM,cAAc,cAAc,MAAM,YAAY;CACpD,MAAM,eAAe,cAAc,MAAM,aAAa;AAItD,QAHmB,cAAc,MAAQ,SAAS,iBAC9B,eAAe,MAAQ,SAAS;;;;;AAQtD,SAAS,gBAAgB,QAA+B;CAEtD,MAAM,UADI,OACQ;AAElB,QAAO;EACL,QAAQ,UAAU,gBAAgB;EAClC,QAAQ,UAAU,gBAAgB;EAClC,SAAS,UAAU,gBAAgB;EACnC,WAAW,UAAU,mBAAmB;EACzC;;;;;;;;;;;;;;;;;;AAmBH,SAAgB,mBAAmB,UAAuD;AACxF,QAAO;EACL,sBAAsB;EACtB,cAAc,OAAO,EAAE,YAAY,aAA+B;GAChE,MAAM,SAAS,MAAM,YAAY;GAGjC,MAAM,YAAY;GAClB,MAAM,WACH,OAAO,UAAU,UAAU,WAAW,UAAU,QAAQ,YACxD,OAAO,UAAU,YAAY,WAAW,UAAU,UAAU;GAC/D,MAAM,QAAQ,OAAO,OAAO,MAAM,CAAC,MAAK,UAAS,UAAU,QAAQ,IAAI,MAAM;GAE7E,MAAM,QAAoB;IACxB;IACA,aAAa,cAAc,OAAO,OAAO,YAAY;IACrD,cAAc,cAAc,OAAO,OAAO,aAAa;IACvD,MAAM,cAAc,OAAO,OAAO,MAAM;IACxC,2BAAW,IAAI,MAAM;IACrB,UAAU,gBAAgB,OAAO;IAClC;AAGD,IAAM,YAAY;AAChB,QAAI;AACF,WAAM,SAAS,MAAM;aACd,KAAK;AACZ,cAAS,iCAAiC,EACxC,OAAO,eAAe,QAAQ,MAAM,IAAI,MAAM,OAAO,IAAI,CAAC,EAC3D,CAAC;;OAEF;AAEJ,UAAO;;EAGT,YAAY,OAAO,EAAE,eAA+B;GAClD,MAAM,EAAE,QAAQ,GAAG,SAAS,MAAM,UAAU;AAI5C,UAAO;IACL;IACA,GAAG;IACJ;;EAEJ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;ACxEH,SAAgB,kBACd,QACA,iBAC8B;CAC9B,MAAM,EAAE,cAAc,gBAAgB,mBAAmB,OAAQ;AAEjE,QAAO;EACL,sBAAsB;EACtB,cAAc,OAAO,EAAE,iBAAmC;GACxD,MAAM,QAAQ,MAAM,iBAAiB;AAGrC,OAAI,MAAM,SAAS,aACjB,OAAM,IAAI,oBACR,2BAA2B,MAAM,MAAM,QAAQ,EAAE,CAAC,MAAM,gBACxD,SACA,MAAM,OACN,aACD;AAIH,OAAI,MAAM,WAAW,eACnB,OAAM,IAAI,oBACR,6BAA6B,MAAM,QAAQ,QAAQ,EAAE,CAAC,MAAM,kBAC5D,WACA,MAAM,SACN,eACD;AAIH,OAAI,MAAM,SAAS,eAAe,iBAChC,SACE,wCAAyC,MAAM,QAAQ,eAAgB,KAAK,QAAQ,EAAE,CAAC,QACxF;AAEH,OAAI,MAAM,WAAW,iBAAiB,iBACpC,SACE,0CAA2C,MAAM,UAAU,iBAAkB,KAAK,QAAQ,EAAE,CAAC,QAC9F;AAGH,UAAO,YAAY;;EAGrB,YAAY,OAAO,EAAE,eAA+B;GAClD,MAAM,QAAQ,MAAM,iBAAiB;AAErC,OAAI,MAAM,SAAS,aACjB,OAAM,IAAI,oBAAoB,yBAAyB,SAAS,MAAM,OAAO,aAAa;AAG5F,OAAI,MAAM,WAAW,eACnB,OAAM,IAAI,oBACR,2BACA,WACA,MAAM,SACN,eACD;AAGH,UAAO,UAAU;;EAEpB"}

package/dist/budget-guard-gyhJS00s.mjs

@@ -0,0 +1,234 @@
+//#region src/streaming/control/budget-guard.ts
+/**
+* Default cost table for common models.
+* Prices are per 1K tokens in USD.
+*
+* @remarks
+* Prices are approximate and may change. Update this table
+* or provide custom entries via BudgetGuardConfig.costTable.
+*/
+const DEFAULT_COST_TABLE = {
+	"gpt-4o": {
+		input: .0025,
+		output: .01
+	},
+	"gpt-4o-mini": {
+		input: 15e-5,
+		output: 6e-4
+	},
+	"gpt-4-turbo": {
+		input: .01,
+		output: .03
+	},
+	"gpt-4": {
+		input: .03,
+		output: .06
+	},
+	"gpt-3.5-turbo": {
+		input: 5e-4,
+		output: .0015
+	},
+	o1: {
+		input: .015,
+		output: .06
+	},
+	"o1-mini": {
+		input: .003,
+		output: .012
+	},
+	"o3-mini": {
+		input: .0011,
+		output: .0044
+	},
+	"claude-3-5-sonnet-20241022": {
+		input: .003,
+		output: .015
+	},
+	"claude-3-5-sonnet": {
+		input: .003,
+		output: .015
+	},
+	"claude-sonnet-4-20250514": {
+		input: .003,
+		output: .015
+	},
+	"claude-3-5-haiku-20241022": {
+		input: 8e-4,
+		output: .004
+	},
+	"claude-3-5-haiku": {
+		input: 8e-4,
+		output: .004
+	},
+	"claude-3-opus": {
+		input: .015,
+		output: .075
+	},
+	"claude-3-sonnet": {
+		input: .003,
+		output: .015
+	},
+	"claude-3-haiku": {
+		input: 25e-5,
+		output: .00125
+	},
+	"gemini-1.5-pro": {
+		input: .00125,
+		output: .005
+	},
+	"gemini-1.5-flash": {
+		input: 75e-6,
+		output: 3e-4
+	},
+	"gemini-2.0-flash": {
+		input: 1e-4,
+		output: 4e-4
+	}
+};
+/**
+* Calculates the cost in USD for a given token usage and model.
+*
+* @param usage - Token usage with prompt and completion tokens.
+* @param modelId - Model identifier to look up in cost table.
+* @param costTable - Optional custom cost table (merged with defaults).
+* @returns Cost in USD, or 0 if model is not in cost table.
+*
+* @example
+* ```ts
+* const usage = { promptTokens: 1000, completionTokens: 500, totalTokens: 1500 };
+*
+* // Using default cost table
+* const cost = calculateCost(usage, 'gpt-4o');
+* // cost = (1000/1000 * 0.0025) + (500/1000 * 0.01) = 0.0025 + 0.005 = 0.0075
+*
+* // Using custom cost table
+* const customCost = calculateCost(usage, 'custom-model', {
+*   'custom-model': { input: 0.001, output: 0.002 }
+* });
+* ```
+*/
+function calculateCost(usage, modelId, costTable) {
+	const entry = {
+		...DEFAULT_COST_TABLE,
+		...costTable
+	}[modelId];
+	if (!entry) return 0;
+	return usage.promptTokens / 1e3 * entry.input + usage.completionTokens / 1e3 * entry.output;
+}
+/**
+* Creates a budget checker function from configuration.
+*
+* @param config - Budget guard configuration with limits.
+* @returns Function that checks if budget is exceeded for given usage.
+*
+* @example
+* ```ts
+* const checkBudget = createBudgetChecker({
+*   maxTokens: 10_000,
+*   maxCost: 1.00,
+*   onBudgetExceeded: (usage, limit) => {
+*     console.error(`Budget exceeded: ${limit.type}`);
+*   }
+* });
+*
+* const result = checkBudget(
+*   { promptTokens: 5000, completionTokens: 6000, totalTokens: 11000 },
+*   'gpt-4o'
+* );
+* // result.exceeded === true
+* // result.limit === { type: 'tokens', limit: 10000, actual: 11000 }
+* ```
+*/
+function createBudgetChecker(config) {
+	return (usage, modelId) => {
+		if (config.maxTokens !== void 0 && usage.totalTokens > config.maxTokens) {
+			const limit = {
+				type: "tokens",
+				limit: config.maxTokens,
+				actual: usage.totalTokens
+			};
+			config.onBudgetExceeded?.(usage, limit);
+			return {
+				exceeded: true,
+				limit
+			};
+		}
+		if (config.maxCost !== void 0) {
+			const cost = calculateCost(usage, modelId, config.costTable);
+			if (cost > config.maxCost) {
+				const limit = {
+					type: "cost",
+					limit: config.maxCost,
+					actual: cost
+				};
+				config.onBudgetExceeded?.(usage, limit);
+				return {
+					exceeded: true,
+					limit
+				};
+			}
+		}
+		return { exceeded: false };
+	};
+}
+/**
+* Creates an onFinish handler for AI SDK that checks budget limits.
+*
+* @param config - Budget guard configuration.
+* @param modelId - Model identifier for cost calculation.
+* @returns onFinish handler function for AI SDK.
+*
+* @example
+* ```ts
+* import { streamText } from 'ai';
+*
+* const result = await streamText({
+*   model: myModel,
+*   prompt: 'Hello',
+*   onFinish: createBudgetGuardedOnFinish(
+*     { maxTokens: 50_000, maxCost: 5.00 },
+*     'gpt-4o'
+*   )
+* });
+* ```
+*/
+function createBudgetGuardedOnFinish(config, modelId) {
+	const checkBudget = createBudgetChecker(config);
+	return (result) => {
+		if (!result.usage) return;
+		checkBudget({
+			promptTokens: result.usage.promptTokens,
+			completionTokens: result.usage.completionTokens,
+			totalTokens: result.usage.totalTokens ?? result.usage.promptTokens + result.usage.completionTokens
+		}, modelId);
+	};
+}
+/**
+* Estimates cost for a request before execution.
+* Useful for pre-flight budget checks.
+*
+* @param estimatedPromptTokens - Estimated number of prompt tokens.
+* @param estimatedCompletionTokens - Estimated number of completion tokens.
+* @param modelId - Model identifier for cost lookup.
+* @param costTable - Optional custom cost table.
+* @returns Estimated cost in USD.
+*
+* @example
+* ```ts
+* const estimatedCost = estimateCost(2000, 1000, 'gpt-4o');
+* if (estimatedCost > budget) {
+*   throw new Error('Estimated cost exceeds budget');
+* }
+* ```
+*/
+function estimateCost(estimatedPromptTokens, estimatedCompletionTokens, modelId, costTable) {
+	return calculateCost({
+		promptTokens: estimatedPromptTokens,
+		completionTokens: estimatedCompletionTokens,
+		totalTokens: estimatedPromptTokens + estimatedCompletionTokens
+	}, modelId, costTable);
+}
+
+//#endregion
+export { estimateCost as a, createBudgetGuardedOnFinish as i, calculateCost as n, createBudgetChecker as r, DEFAULT_COST_TABLE as t };
+//# sourceMappingURL=budget-guard-gyhJS00s.mjs.map

package/dist/budget-guard-gyhJS00s.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-guard-gyhJS00s.mjs","names":[],"sources":["../src/streaming/control/budget-guard.ts"],"sourcesContent":["/**\n * @fileoverview Budget guard utilities for token and cost limits.\n * @module @od-oneapp/ai-platform/streaming-control\n *\n * Provides functions for calculating costs and checking budget limits:\n * - Cost calculation based on token usage and model pricing\n * - Budget checking for token and cost limits\n * - OnFinish handlers for AI SDK integration\n *\n * @example\n * ```ts\n * import {\n *   calculateCost,\n *   createBudgetChecker,\n *   createBudgetGuardedOnFinish,\n *   DEFAULT_COST_TABLE\n * } from '@od-oneapp/ai-platform/streaming-control';\n *\n * const cost = calculateCost(\n *   { promptTokens: 1000, completionTokens: 500, totalTokens: 1500 },\n *   'gpt-4o'\n * );\n * ```\n */\n\nimport {\n  type BudgetCheckResult,\n  type BudgetGuardConfig,\n  type CostEntry,\n  type TokenUsage,\n} from './types';\n\n/**\n * Default cost table for common models.\n * Prices are per 1K tokens in USD.\n *\n * @remarks\n * Prices are approximate and may change. Update this table\n * or provide custom entries via BudgetGuardConfig.costTable.\n */\nexport const DEFAULT_COST_TABLE: Record<string, CostEntry> = {\n  // OpenAI models\n  'gpt-4o': { input: 0.0025, output: 0.01 },\n  'gpt-4o-mini': { input: 0.00015, output: 0.0006 },\n  'gpt-4-turbo': { input: 0.01, output: 0.03 },\n  'gpt-4': { input: 0.03, output: 0.06 },\n  'gpt-3.5-turbo': { input: 0.0005, output: 0.0015 },\n  o1: { input: 0.015, output: 0.06 },\n  'o1-mini': { input: 0.003, output: 0.012 },\n  'o3-mini': { input: 0.0011, output: 0.0044 },\n  // Anthropic models\n  'claude-3-5-sonnet-20241022': { input: 0.003, output: 0.015 },\n  'claude-3-5-sonnet': { input: 0.003, output: 0.015 },\n  'claude-sonnet-4-20250514': { input: 0.003, output: 0.015 },\n  'claude-3-5-haiku-20241022': { input: 0.0008, output: 0.004 },\n  'claude-3-5-haiku': { input: 0.0008, output: 0.004 },\n  'claude-3-opus': { input: 0.015, output: 0.075 },\n  'claude-3-sonnet': { input: 0.003, output: 0.015 },\n  'claude-3-haiku': { input: 0.00025, output: 0.00125 },\n  // Google models\n  'gemini-1.5-pro': { input: 0.00125, output: 0.005 },\n  'gemini-1.5-flash': { input: 0.000075, output: 0.0003 },\n  'gemini-2.0-flash': { input: 0.0001, output: 0.0004 },\n};\n\n/**\n * Calculates the cost in USD for a given token usage and model.\n *\n * @param usage - Token usage with prompt and completion tokens.\n * @param modelId - Model identifier to look up in cost table.\n * @param costTable - Optional custom cost table (merged with defaults).\n * @returns Cost in USD, or 0 if model is not in cost table.\n *\n * @example\n * ```ts\n * const usage = { promptTokens: 1000, completionTokens: 500, totalTokens: 1500 };\n *\n * // Using default cost table\n * const cost = calculateCost(usage, 'gpt-4o');\n * // cost = (1000/1000 * 0.0025) + (500/1000 * 0.01) = 0.0025 + 0.005 = 0.0075\n *\n * // Using custom cost table\n * const customCost = calculateCost(usage, 'custom-model', {\n *   'custom-model': { input: 0.001, output: 0.002 }\n * });\n * ```\n */\nexport function calculateCost(\n  usage: TokenUsage,\n  modelId: string,\n  costTable?: Record<string, CostEntry>,\n): number {\n  const mergedTable = { ...DEFAULT_COST_TABLE, ...costTable };\n  const entry = mergedTable[modelId];\n\n  if (!entry) {\n    return 0;\n  }\n\n  const inputCost = (usage.promptTokens / 1000) * entry.input;\n  const outputCost = (usage.completionTokens / 1000) * entry.output;\n\n  return inputCost + outputCost;\n}\n\n/**\n * Creates a budget checker function from configuration.\n *\n * @param config - Budget guard configuration with limits.\n * @returns Function that checks if budget is exceeded for given usage.\n *\n * @example\n * ```ts\n * const checkBudget = createBudgetChecker({\n *   maxTokens: 10_000,\n *   maxCost: 1.00,\n *   onBudgetExceeded: (usage, limit) => {\n *     console.error(`Budget exceeded: ${limit.type}`);\n *   }\n * });\n *\n * const result = checkBudget(\n *   { promptTokens: 5000, completionTokens: 6000, totalTokens: 11000 },\n *   'gpt-4o'\n * );\n * // result.exceeded === true\n * // result.limit === { type: 'tokens', limit: 10000, actual: 11000 }\n * ```\n */\nexport function createBudgetChecker(\n  config: BudgetGuardConfig,\n): (usage: TokenUsage, modelId: string) => BudgetCheckResult {\n  return (usage: TokenUsage, modelId: string): BudgetCheckResult => {\n    // Check token limit first\n    if (config.maxTokens !== undefined && usage.totalTokens > config.maxTokens) {\n      const limit = {\n        type: 'tokens' as const,\n        limit: config.maxTokens,\n        actual: usage.totalTokens,\n      };\n\n      config.onBudgetExceeded?.(usage, limit);\n\n      return {\n        exceeded: true,\n        limit,\n      };\n    }\n\n    // Check cost limit\n    if (config.maxCost !== undefined) {\n      const cost = calculateCost(usage, modelId, config.costTable);\n\n      if (cost > config.maxCost) {\n        const limit = {\n          type: 'cost' as const,\n          limit: config.maxCost,\n          actual: cost,\n        };\n\n        config.onBudgetExceeded?.(usage, limit);\n\n        return {\n          exceeded: true,\n          limit,\n        };\n      }\n    }\n\n    return { exceeded: false };\n  };\n}\n\n/**\n * Result type for AI SDK onFinish callback.\n * Matches the shape expected by Vercel AI SDK v6.\n */\nexport interface OnFinishResult {\n  /** Token usage from the model response. */\n  usage?: {\n    promptTokens: number;\n    completionTokens: number;\n    totalTokens?: number;\n  };\n}\n\n/**\n * Creates an onFinish handler for AI SDK that checks budget limits.\n *\n * @param config - Budget guard configuration.\n * @param modelId - Model identifier for cost calculation.\n * @returns onFinish handler function for AI SDK.\n *\n * @example\n * ```ts\n * import { streamText } from 'ai';\n *\n * const result = await streamText({\n *   model: myModel,\n *   prompt: 'Hello',\n *   onFinish: createBudgetGuardedOnFinish(\n *     { maxTokens: 50_000, maxCost: 5.00 },\n *     'gpt-4o'\n *   )\n * });\n * ```\n */\nexport function createBudgetGuardedOnFinish(\n  config: BudgetGuardConfig,\n  modelId: string,\n): (result: OnFinishResult) => void {\n  const checkBudget = createBudgetChecker(config);\n\n  return (result: OnFinishResult): void => {\n    if (!result.usage) {\n      return;\n    }\n\n    const usage: TokenUsage = {\n      promptTokens: result.usage.promptTokens,\n      completionTokens: result.usage.completionTokens,\n      totalTokens:\n        result.usage.totalTokens ?? result.usage.promptTokens + result.usage.completionTokens,\n    };\n\n    // Check budget (callback will be invoked if exceeded)\n    checkBudget(usage, modelId);\n  };\n}\n\n/**\n * Estimates cost for a request before execution.\n * Useful for pre-flight budget checks.\n *\n * @param estimatedPromptTokens - Estimated number of prompt tokens.\n * @param estimatedCompletionTokens - Estimated number of completion tokens.\n * @param modelId - Model identifier for cost lookup.\n * @param costTable - Optional custom cost table.\n * @returns Estimated cost in USD.\n *\n * @example\n * ```ts\n * const estimatedCost = estimateCost(2000, 1000, 'gpt-4o');\n * if (estimatedCost > budget) {\n *   throw new Error('Estimated cost exceeds budget');\n * }\n * ```\n */\nexport function estimateCost(\n  estimatedPromptTokens: number,\n  estimatedCompletionTokens: number,\n  modelId: string,\n  costTable?: Record<string, CostEntry>,\n): number {\n  return calculateCost(\n    {\n      promptTokens: estimatedPromptTokens,\n      completionTokens: estimatedCompletionTokens,\n      totalTokens: estimatedPromptTokens + estimatedCompletionTokens,\n    },\n    modelId,\n    costTable,\n  );\n}\n"],"mappings":";;;;;;;;;AAwCA,MAAa,qBAAgD;CAE3D,UAAU;EAAE,OAAO;EAAQ,QAAQ;EAAM;CACzC,eAAe;EAAE,OAAO;EAAS,QAAQ;EAAQ;CACjD,eAAe;EAAE,OAAO;EAAM,QAAQ;EAAM;CAC5C,SAAS;EAAE,OAAO;EAAM,QAAQ;EAAM;CACtC,iBAAiB;EAAE,OAAO;EAAQ,QAAQ;EAAQ;CAClD,IAAI;EAAE,OAAO;EAAO,QAAQ;EAAM;CAClC,WAAW;EAAE,OAAO;EAAO,QAAQ;EAAO;CAC1C,WAAW;EAAE,OAAO;EAAQ,QAAQ;EAAQ;CAE5C,8BAA8B;EAAE,OAAO;EAAO,QAAQ;EAAO;CAC7D,qBAAqB;EAAE,OAAO;EAAO,QAAQ;EAAO;CACpD,4BAA4B;EAAE,OAAO;EAAO,QAAQ;EAAO;CAC3D,6BAA6B;EAAE,OAAO;EAAQ,QAAQ;EAAO;CAC7D,oBAAoB;EAAE,OAAO;EAAQ,QAAQ;EAAO;CACpD,iBAAiB;EAAE,OAAO;EAAO,QAAQ;EAAO;CAChD,mBAAmB;EAAE,OAAO;EAAO,QAAQ;EAAO;CAClD,kBAAkB;EAAE,OAAO;EAAS,QAAQ;EAAS;CAErD,kBAAkB;EAAE,OAAO;EAAS,QAAQ;EAAO;CACnD,oBAAoB;EAAE,OAAO;EAAU,QAAQ;EAAQ;CACvD,oBAAoB;EAAE,OAAO;EAAQ,QAAQ;EAAQ;CACtD;;;;;;;;;;;;;;;;;;;;;;;AAwBD,SAAgB,cACd,OACA,SACA,WACQ;CAER,MAAM,QADc;EAAE,GAAG;EAAoB,GAAG;EAAW,CACjC;AAE1B,KAAI,CAAC,MACH,QAAO;AAMT,QAHmB,MAAM,eAAe,MAAQ,MAAM,QAClC,MAAM,mBAAmB,MAAQ,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;AA6B7D,SAAgB,oBACd,QAC2D;AAC3D,SAAQ,OAAmB,YAAuC;AAEhE,MAAI,OAAO,cAAc,UAAa,MAAM,cAAc,OAAO,WAAW;GAC1E,MAAM,QAAQ;IACZ,MAAM;IACN,OAAO,OAAO;IACd,QAAQ,MAAM;IACf;AAED,UAAO,mBAAmB,OAAO,MAAM;AAEvC,UAAO;IACL,UAAU;IACV;IACD;;AAIH,MAAI,OAAO,YAAY,QAAW;GAChC,MAAM,OAAO,cAAc,OAAO,SAAS,OAAO,UAAU;AAE5D,OAAI,OAAO,OAAO,SAAS;IACzB,MAAM,QAAQ;KACZ,MAAM;KACN,OAAO,OAAO;KACd,QAAQ;KACT;AAED,WAAO,mBAAmB,OAAO,MAAM;AAEvC,WAAO;KACL,UAAU;KACV;KACD;;;AAIL,SAAO,EAAE,UAAU,OAAO;;;;;;;;;;;;;;;;;;;;;;;;AAsC9B,SAAgB,4BACd,QACA,SACkC;CAClC,MAAM,cAAc,oBAAoB,OAAO;AAE/C,SAAQ,WAAiC;AACvC,MAAI,CAAC,OAAO,MACV;AAWF,cAR0B;GACxB,cAAc,OAAO,MAAM;GAC3B,kBAAkB,OAAO,MAAM;GAC/B,aACE,OAAO,MAAM,eAAe,OAAO,MAAM,eAAe,OAAO,MAAM;GACxE,EAGkB,QAAQ;;;;;;;;;;;;;;;;;;;;;AAsB/B,SAAgB,aACd,uBACA,2BACA,SACA,WACQ;AACR,QAAO,cACL;EACE,cAAc;EACd,kBAAkB;EAClB,aAAa,wBAAwB;EACtC,EACD,SACA,UACD"}