@bandito-ai/sdk 0.1.7

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/client.ts","../src/engine.ts","../src/models.ts","../src/config.ts","../src/http.ts","../src/store.ts","../src/worker.ts"],"sourcesContent":["/**\n * Bandito SDK — contextual bandit optimization for LLM selection.\n *\n * Recommended (explicit client):\n *   import { BanditoClient } from 'bandito';\n *\n *   const client = new BanditoClient({ apiKey: 'bnd_...' });\n *   await client.connect();\n *   const result = client.pull('my-chatbot', { query: userMessage });\n *   // ... call LLM with result.model, result.prompt ...\n *   client.update(result, { response: response.text });\n *   await client.close();\n *\n * Module-level singleton (convenience):\n *   import { connect, pull, update, close } from 'bandito';\n *\n *   await connect({ apiKey: 'bnd_...' });\n *   const result = pull('my-chatbot', { query: userMessage });\n *   update(result, { response: response.text });\n *   await close();\n */\n\nexport { BanditoClient, type ClientOptions, type PullOptions, type UpdateOptions } from \"./client.js\";\nexport { type Arm, type PullResult } from \"./models.js\";\n\nimport { BanditoClient, type ClientOptions, type PullOptions, type UpdateOptions } from \"./client.js\";\nimport type { PullResult } from \"./models.js\";\n\nlet _client: BanditoClient | null = null;\n\nfunction getClient(): BanditoClient {\n  if (!_client) {\n    throw new Error(\"Not connected — call connect() first\");\n  }\n  return _client;\n}\n\n/** Connect to the Bandito cloud and hydrate local state. */\nexport async function connect(options: ClientOptions = {}): Promise<void> {\n  if (_client) {\n    await _client.close();\n  }\n  _client = new BanditoClient(options);\n  await _client.connect();\n}\n\n/** Local Thompson Sampling decision. <1ms, no network. */\nexport function pull(banditName: string, options?: PullOptions): PullResult {\n  return getClient().pull(banditName, options);\n}\n\n/** Record an LLM call outcome (writes to SQLite, fire-and-forget flush). */\nexport function update(pullResult: PullResult, options?: UpdateOptions): void {\n  getClient().update(pullResult, options);\n}\n\n/** Send a human grade for an existing event. */\nexport async function grade(eventId: string, gradeValue: number): Promise<void> {\n  await getClient().grade(eventId, gradeValue);\n}\n\n/** Explicit state refresh from cloud. */\nexport async function sync(): Promise<void> {\n  await getClient().sync();\n}\n\n/** Shut down: flush events, close connections. */\nexport async function close(): Promise<void> {\n  if (_client) {\n    await _client.close();\n    _client = null;\n  }\n}\n","/**\n * BanditoClient — main orchestrator for the JS/TS SDK.\n *\n * Mirrors the Python SDK's sync-first design:\n * - pull() is synchronous (WASM math, <1ms)\n * - connect(), grade(), sync(), close() are async (HTTP I/O)\n * - update() is synchronous (SQLite write + fire-and-forget flush)\n */\n\nimport { randomUUID } from \"node:crypto\";\nimport * as fs from \"node:fs\";\nimport * as path from \"node:path\";\nimport * as os from \"node:os\";\nimport { performance } from \"node:perf_hooks\";\n\nimport { initWasm, createEngine, updateEngine, type BanditEngine, type EnginePullResult } from \"./engine.js\";\nimport {\n  type Arm,\n  type PullResult,\n  type BanditCache,\n  type ArmWire,\n  createArm,\n  createPullResult,\n} from \"./models.js\";\nimport { loadConfig, DEFAULT_BASE_URL } from \"./config.js\";\nimport { BanditoHTTP } from \"./http.js\";\nimport { EventStore, type EventPayload } from \"./store.js\";\nimport { prepareCloudPayload } from \"./worker.js\";\n\nconst DEFAULT_STORE_PATH = path.join(os.homedir(), \".bandito\", \"events.db\");\nconst MAX_EVENT_RETRIES = 5;\n\nexport interface ClientOptions {\n  apiKey?: string;\n  baseUrl?: string;\n  storePath?: string;\n  dataStorage?: string;\n}\n\nexport interface PullOptions {\n  query?: string;\n  exclude?: number[];\n}\n\nexport interface UpdateOptions {\n  queryText?: string;\n  response?: string | Record<string, unknown>;\n  reward?: number;\n  cost?: number;\n  latency?: number;\n  inputTokens?: number;\n  outputTokens?: number;\n  segment?: Record<string, string>;\n  failed?: boolean;\n}\n\nexport class BanditoClient {\n  private apiKey: string | undefined;\n  private baseUrl: string | undefined;\n  private storePath: string | undefined;\n  private dataStorageArg: string | undefined;\n  private dataStorage: string;\n\n  private http: BanditoHTTP | null = null;\n  private store: EventStore | null = null;\n  private engines: Map<string, BanditEngine> = new Map();\n  private bandits: Map<string, BanditCache> = new Map();\n  private connected = false;\n  private flushInterval: ReturnType<typeof setInterval> | null = null;\n  private flushInProgress = false;\n  private deadUuids: Set<string> = new Set();\n  private retryCounts: Map<string, number> = new Map();\n\n  constructor(options: ClientOptions = {}) {\n    this.apiKey = options.apiKey;\n    this.baseUrl = options.baseUrl;\n    this.storePath = options.storePath;\n    this.dataStorageArg = options.dataStorage;\n    this.dataStorage = options.dataStorage ?? \"local\";\n  }\n\n  /**\n   * Bootstrap: authenticate and hydrate in-memory state from cloud.\n   *\n   * Resolves config from: constructor args → env vars → ~/.bandito/config.toml.\n   * Initializes WASM, creates HTTP client, SQLite store, fetches full state.\n   */\n  async connect(): Promise<void> {\n    // Tear down previous connection if reconnecting\n    if (this.connected) {\n      await this.close();\n    }\n\n    // Init WASM (loads .wasm binary once)\n    await initWasm();\n\n    // Resolve config\n    const config = loadConfig();\n    const apiKey = this.apiKey ?? config.apiKey;\n    if (!apiKey) {\n      throw new Error(\n        \"apiKey required — pass it to constructor, set BANDITO_API_KEY, \" +\n          \"or run `bandito signup`\",\n      );\n    }\n\n    const baseUrl = this.baseUrl ?? config.baseUrl;\n    if (!this.dataStorageArg) {\n      this.dataStorage = config.dataStorage;\n    }\n\n    this.http = new BanditoHTTP(baseUrl, apiKey);\n    const storePath = this.storePath ?? DEFAULT_STORE_PATH;\n    if (storePath !== \":memory:\") {\n      const dir = path.dirname(storePath);\n      if (!fs.existsSync(dir)) {\n        fs.mkdirSync(dir, { recursive: true });\n      }\n    }\n    this.store = new EventStore(storePath);\n\n    // Bootstrap: fetch state, hydrate cache, flush pending\n    try {\n      const data = await this.http.connect();\n      this.applySync(data);\n\n      // Reset retry state\n      this.deadUuids.clear();\n      this.retryCounts.clear();\n\n      // Flush pending events from previous crash\n      await this.flushPending();\n\n      // Start periodic flush (every 30s)\n      this.flushInterval = setInterval(() => {\n        this.flushPending().catch(() => {});\n      }, 30_000);\n\n      this.connected = true;\n    } catch (err) {\n      this.store?.close();\n      this.store = null;\n      this.http = null;\n      throw err;\n    }\n  }\n\n  /**\n   * Local Thompson Sampling decision. Synchronous, <1ms, no network.\n   */\n  pull(banditName: string, options: PullOptions = {}): PullResult {\n    this.ensureConnected();\n\n    const cache = this.bandits.get(banditName);\n    if (!cache) {\n      const available = [...this.bandits.keys()];\n      throw new Error(\n        `Unknown bandit '${banditName}'. Available: [${available.join(\", \")}]`,\n      );\n    }\n\n    if (cache.arms.length === 0) {\n      throw new Error(`Bandit '${banditName}' has no active arms`);\n    }\n\n    const engine = this.engines.get(banditName)!;\n    const queryLength = options.query?.length ?? undefined;\n    const excludeIds = options.exclude ? Int32Array.from(options.exclude) : undefined;\n\n    const resultJson = engine.pull(queryLength, excludeIds);\n    const raw: EnginePullResult = JSON.parse(resultJson);\n\n    // Look up the winning arm from our cached active arms\n    const winnerArm = cache.arms.find((a) => a.armId === raw.arm_id);\n    if (!winnerArm) {\n      throw new Error(\n        `Engine selected arm ${raw.arm_id} but it's not in active arm cache for \"${banditName}\". ` +\n          \"This is likely a bug — please report it at https://github.com/bandito-ai/bandito/issues\",\n      );\n    }\n\n    return createPullResult({\n      arm: winnerArm,\n      eventId: randomUUID(),\n      banditId: cache.banditId,\n      banditName,\n      scores: raw.scores,\n      pullTime: performance.now(),\n    });\n  }\n\n  /**\n   * Record an LLM call outcome. Writes to SQLite first (crash-safe),\n   * then fires off a non-blocking flush to cloud.\n   */\n  update(pullResult: PullResult, options: UpdateOptions = {}): void {\n    this.ensureConnected();\n\n    let reward = options.reward;\n    if (options.failed && reward == null) {\n      reward = 0.0;\n    }\n\n    // Auto-calculate latency from pull timestamp\n    let latency = options.latency;\n    if (latency == null && pullResult._pullTime > 0) {\n      latency = performance.now() - pullResult._pullTime;\n    }\n\n    // Build event payload (snake_case for wire format)\n    const event: EventPayload = {\n      local_event_uuid: pullResult.eventId,\n      bandit_id: pullResult.banditId,\n      arm_id: pullResult.arm.armId,\n      model_name: pullResult.arm.modelName,\n      model_provider: pullResult.arm.modelProvider,\n    };\n\n    if (options.queryText != null) {\n      (event as Record<string, unknown>).query_text = options.queryText;\n    }\n    if (options.response != null) {\n      (event as Record<string, unknown>).response =\n        typeof options.response === \"string\"\n          ? { response: options.response }\n          : options.response;\n    }\n    if (reward != null) {\n      (event as Record<string, unknown>).early_reward = reward;\n    }\n    if (options.cost != null) {\n      (event as Record<string, unknown>).cost = options.cost;\n    }\n    if (latency != null) {\n      (event as Record<string, unknown>).latency = latency;\n    }\n    if (options.inputTokens != null) {\n      (event as Record<string, unknown>).input_tokens = options.inputTokens;\n    }\n    if (options.outputTokens != null) {\n      (event as Record<string, unknown>).output_tokens = options.outputTokens;\n    }\n    if (options.segment != null) {\n      (event as Record<string, unknown>).segment = options.segment;\n    }\n    if (options.failed) {\n      (event as Record<string, unknown>).run_error = true;\n    }\n\n    // Write to SQLite WAL first — survives crashes\n    this.store!.push(event);\n\n    // Fire-and-forget flush (errors logged inside flushPending)\n    this.flushPending().catch(() => {});\n  }\n\n  /**\n   * Send a human grade for an existing event. Async (HTTP).\n   */\n  async grade(eventId: string, grade: number): Promise<void> {\n    this.ensureConnected();\n    await this.http!.submitGrade(eventId, grade);\n  }\n\n  /**\n   * Explicit state refresh from cloud.\n   */\n  async sync(): Promise<void> {\n    this.ensureConnected();\n    const data = await this.http!.heartbeat();\n\n    const prevBandits = new Map(this.bandits);\n    const prevEngines = new Map(this.engines);\n    try {\n      this.applySync(data);\n    } catch (err) {\n      // Rollback on malformed response — keep last-known-good state\n      this.bandits = prevBandits;\n      this.engines = prevEngines;\n      console.warn(\"[bandito] Sync response malformed — keeping last-known-good state\", err);\n    }\n  }\n\n  /**\n   * Shut down: clear interval, flush remaining events, close connections.\n   */\n  async close(): Promise<void> {\n    if (this.flushInterval) {\n      clearInterval(this.flushInterval);\n      this.flushInterval = null;\n    }\n\n    // Final flush\n    if (this.store && this.http) {\n      await this.flushPending();\n    }\n\n    this.store?.close();\n    this.store = null;\n    this.http = null;\n    this.engines.clear();\n    this.bandits.clear();\n    this.connected = false;\n  }\n\n  // --- Internal ---\n\n  private ensureConnected(): void {\n    if (!this.connected) {\n      throw new Error(\"Not connected — call connect() first\");\n    }\n  }\n\n  private applySync(data: Record<string, unknown>): void {\n    const banditsData = (data.bandits ?? []) as Record<string, unknown>[];\n\n    const newBandits = new Map<string, BanditCache>();\n    const newEngines = new Map<string, BanditEngine>();\n\n    for (const b of banditsData) {\n      const arms = (b.arms ?? []) as ArmWire[];\n      if (arms.length === 0) continue;\n\n      const activeArms: Arm[] = arms\n        .filter((a) => a.is_active)\n        .map((a) => createArm(a));\n\n      const name = b.name as string;\n      const cache: BanditCache = {\n        banditId: Number(b.bandit_id),\n        name,\n        arms: activeArms,\n        armWire: arms,\n        optimizationMode: (b.optimization_mode as string) ?? \"base\",\n        avgLatencyLastN: b.avg_latency_last_n as number | null,\n        budget: b.budget as number | null,\n        totalCost: b.total_cost as number | null,\n      };\n\n      newBandits.set(name, cache);\n\n      // Reuse existing engine (preserves RNG state) or create new one\n      const existingEngine = this.engines.get(name);\n      const banditJson = JSON.stringify(b);\n      if (existingEngine) {\n        updateEngine(existingEngine, banditJson);\n        newEngines.set(name, existingEngine);\n      } else {\n        newEngines.set(name, createEngine(banditJson));\n      }\n    }\n\n    this.bandits = newBandits;\n    this.engines = newEngines;\n  }\n\n  private async flushPending(): Promise<void> {\n    if (this.flushInProgress || !this.store || !this.http) return;\n    this.flushInProgress = true;\n\n    try {\n      const pending = this.store.pending();\n      if (pending.length === 0) return;\n\n      // Filter out dead events\n      const alive = this.deadUuids.size > 0\n        ? pending.filter((e) => !this.deadUuids.has(e.local_event_uuid))\n        : pending;\n      if (alive.length === 0) return;\n\n      // Prepare payload (strip metadata/text as configured)\n      const payload = prepareCloudPayload(\n        alive as Record<string, unknown>[],\n        this.dataStorage !== \"local\",\n      );\n\n      const result = await this.http.ingestEvents(payload);\n\n      // Parse per-event errors\n      const errors = (result.errors ?? []) as {\n        local_event_uuid?: string;\n        reason?: string;\n      }[];\n      const erroredUuids = new Set(\n        errors.map((e) => e.local_event_uuid).filter(Boolean) as string[],\n      );\n\n      // Update retry counts\n      for (const uid of erroredUuids) {\n        const count = (this.retryCounts.get(uid) ?? 0) + 1;\n        this.retryCounts.set(uid, count);\n        if (count >= MAX_EVENT_RETRIES) {\n          this.deadUuids.add(uid);\n        }\n      }\n\n      // Mark accepted events as flushed\n      const flushedUuids = alive\n        .map((e) => e.local_event_uuid)\n        .filter((uid) => !erroredUuids.has(uid));\n      if (flushedUuids.length > 0 && this.store) {\n        this.store.markFlushed(flushedUuids);\n      }\n    } catch (err) {\n      // Flush failure is non-fatal — events stay pending for next attempt\n      console.warn(\"[bandito] Event flush failed — will retry\", err);\n    } finally {\n      this.flushInProgress = false;\n    }\n  }\n}\n","/**\n * Thin wrapper around the WASM engine import.\n *\n * Handles async WASM initialization (loading .wasm binary happens once)\n * and re-exports the BanditEngine constructor for the client.\n */\n\nimport type { BanditEngine as WasmBanditEngine } from \"../wasm/bandito_engine\";\n\nlet wasmModule: typeof import(\"../wasm/bandito_engine\") | null = null;\n\n/**\n * Initialize the WASM module. Must be called before creating BanditEngine instances.\n * Safe to call multiple times — only loads once.\n */\nexport async function initWasm(): Promise<void> {\n  if (wasmModule) return;\n  wasmModule = await import(\"../wasm/bandito_engine\");\n}\n\n/**\n * Create a BanditEngine from a sync response JSON string.\n * Requires initWasm() to have been called first.\n */\nexport function createEngine(banditJson: string): WasmBanditEngine {\n  if (!wasmModule) {\n    throw new Error(\"WASM not initialized — call initWasm() first\");\n  }\n  return new wasmModule.BanditEngine(banditJson);\n}\n\n/**\n * Update an existing BanditEngine with new sync response data.\n * Preserves RNG state (avoids the \"always picks same arm\" bug).\n */\nexport function updateEngine(engine: WasmBanditEngine, banditJson: string): void {\n  engine.updateFromSync(banditJson);\n}\n\nexport type { WasmBanditEngine as BanditEngine };\n\n/**\n * Pull result parsed from engine JSON output.\n */\nexport interface EnginePullResult {\n  arm_index: number;\n  arm_id: number;\n  scores: Record<number, number>;\n}\n","/**\n * SDK types: Arm, PullResult, and internal cache structures.\n */\n\n/** An arm returned to the user after pull(). Immutable. */\nexport interface Arm {\n  readonly armId: number;\n  readonly modelName: string;\n  readonly modelProvider: string;\n  readonly systemPrompt: string;\n  readonly isPromptTemplated: boolean;\n  /** Convenience alias for modelName. */\n  readonly model: string;\n  /** Convenience alias for systemPrompt. */\n  readonly prompt: string;\n}\n\n/** Returned by pull(), passed to update(). Immutable. */\nexport interface PullResult {\n  readonly arm: Arm;\n  readonly eventId: string;\n  readonly banditId: number;\n  readonly banditName: string;\n  readonly scores: Readonly<Record<number, number>>;\n  /** Convenience reach-through to arm.modelName. */\n  readonly model: string;\n  /** Convenience reach-through to arm.systemPrompt. */\n  readonly prompt: string;\n  /** @internal perf timestamp */\n  readonly _pullTime: number;\n}\n\n/** Create a frozen Arm from raw wire data. */\nexport function createArm(data: {\n  arm_id: number;\n  model_name: string;\n  model_provider: string;\n  system_prompt: string;\n  is_prompt_templated?: boolean;\n}): Arm {\n  const arm: Arm = {\n    armId: data.arm_id,\n    modelName: data.model_name,\n    modelProvider: data.model_provider,\n    systemPrompt: data.system_prompt,\n    isPromptTemplated: data.is_prompt_templated ?? false,\n    get model() {\n      return this.modelName;\n    },\n    get prompt() {\n      return this.systemPrompt;\n    },\n  };\n  return Object.freeze(arm);\n}\n\n/** Create a frozen PullResult. */\nexport function createPullResult(data: {\n  arm: Arm;\n  eventId: string;\n  banditId: number;\n  banditName: string;\n  scores: Record<number, number>;\n  pullTime: number;\n}): PullResult {\n  const result: PullResult = {\n    arm: data.arm,\n    eventId: data.eventId,\n    banditId: data.banditId,\n    banditName: data.banditName,\n    scores: Object.freeze({ ...data.scores }),\n    get model() {\n      return this.arm.modelName;\n    },\n    get prompt() {\n      return this.arm.systemPrompt;\n    },\n    _pullTime: data.pullTime,\n  };\n  return Object.freeze(result);\n}\n\n/** Raw arm data from sync response (snake_case wire format). */\nexport interface ArmWire {\n  arm_id: number;\n  model_name: string;\n  model_provider: string;\n  system_prompt: string;\n  is_prompt_templated: boolean;\n  is_active: boolean;\n  avg_latency_last_n: number | null;\n}\n\n/** Internal mutable cache for a bandit's state. */\nexport interface BanditCache {\n  banditId: number;\n  name: string;\n  arms: Arm[]; // active only\n  armWire: ArmWire[]; // all arms (for engine JSON)\n  optimizationMode: string;\n  avgLatencyLastN: number | null;\n  budget: number | null;\n  totalCost: number | null;\n}\n","/**\n * Config loader — reads ~/.bandito/config.toml and env vars.\n *\n * Resolution order: constructor args → env vars → TOML → defaults.\n * Same file as the Python SDK uses.\n */\n\nimport * as fs from \"node:fs\";\nimport * as path from \"node:path\";\nimport * as os from \"node:os\";\nimport { parse as parseToml } from \"smol-toml\";\n\nexport const DEFAULT_BASE_URL = \"https://bandito-api.onrender.com\";\nconst CONFIG_DIR = path.join(os.homedir(), \".bandito\");\nconst CONFIG_FILE = path.join(CONFIG_DIR, \"config.toml\");\n\nexport interface BanditoConfig {\n  apiKey: string | null;\n  baseUrl: string;\n  dataStorage: string; // \"local\" or \"cloud\"\n}\n\n/**\n * Load config from TOML file + env var overrides.\n */\nexport function loadConfig(): BanditoConfig {\n  const config: BanditoConfig = {\n    apiKey: null,\n    baseUrl: DEFAULT_BASE_URL,\n    dataStorage: \"local\",\n  };\n\n  // TOML file first\n  if (fs.existsSync(CONFIG_FILE)) {\n    try {\n      const content = fs.readFileSync(CONFIG_FILE, \"utf-8\");\n      const data = parseToml(content);\n      if (data.api_key) config.apiKey = data.api_key;\n      if (data.base_url) config.baseUrl = data.base_url;\n      if (data.data_storage) config.dataStorage = data.data_storage;\n    } catch {\n      // Failed to parse TOML — fall back to env vars\n    }\n  }\n\n  // Env vars override\n  const envKey = process.env.BANDITO_API_KEY;\n  if (envKey) config.apiKey = envKey;\n\n  const envUrl = process.env.BANDITO_BASE_URL;\n  if (envUrl) config.baseUrl = envUrl;\n\n  const envStorage = process.env.BANDITO_DATA_STORAGE;\n  if (envStorage) config.dataStorage = envStorage;\n\n  return config;\n}\n","/**\n * HTTP transport — fetch-based client for cloud API.\n *\n * Retry config: 3 attempts, exponential backoff (0.5s, 1s, 2s),\n * retries only 5xx and network errors. Never retries 4xx.\n */\n\nconst MAX_RETRIES = 3;\nconst RETRY_BACKOFF_BASE = 500; // ms — 500, 1000, 2000\n\nfunction sleep(ms: number): Promise<void> {\n  return new Promise((resolve) => setTimeout(resolve, ms));\n}\n\nfunction isRetryable(status: number): boolean {\n  return status >= 500;\n}\n\nexport class BanditoHTTP {\n  private baseUrl: string;\n  private apiKey: string;\n  private timeout: number;\n\n  constructor(baseUrl: string, apiKey: string, timeout: number = 10_000) {\n    this.baseUrl = `${baseUrl.replace(/\\/$/, \"\")}/api/v1`;\n    this.apiKey = apiKey;\n    this.timeout = timeout;\n  }\n\n  private async request(\n    method: string,\n    path: string,\n    body?: unknown,\n  ): Promise<Record<string, unknown>> {\n    let lastError: Error | null = null;\n\n    for (let attempt = 0; attempt < MAX_RETRIES; attempt++) {\n      const controller = new AbortController();\n      const timer = setTimeout(() => controller.abort(), this.timeout);\n\n      try {\n        const resp = await fetch(`${this.baseUrl}${path}`, {\n          method,\n          headers: {\n            \"X-API-Key\": this.apiKey,\n            \"Content-Type\": \"application/json\",\n          },\n          body: body != null ? JSON.stringify(body) : undefined,\n          signal: controller.signal,\n        });\n\n        clearTimeout(timer);\n\n        if (!resp.ok) {\n          const text = await resp.text().catch(() => \"\");\n          if (!isRetryable(resp.status) || attempt === MAX_RETRIES - 1) {\n            throw new Error(\n              `HTTP ${resp.status} on ${method} ${path}: ${text}`,\n            );\n          }\n          // Retryable server error — fall through to retry\n          lastError = new Error(\n            `HTTP ${resp.status} on ${method} ${path}: ${text}`,\n          );\n        } else {\n          return (await resp.json()) as Record<string, unknown>;\n        }\n      } catch (err) {\n        clearTimeout(timer);\n        lastError = err as Error;\n\n        // AbortError = timeout, TypeError = network error (in fetch)\n        const isNetworkOrTimeout =\n          (err as Error).name === \"AbortError\" ||\n          (err as Error).name === \"TypeError\";\n        if (!isNetworkOrTimeout && attempt < MAX_RETRIES - 1) {\n          // Non-retryable (4xx already handled above)\n          throw err;\n        }\n        if (attempt === MAX_RETRIES - 1) {\n          throw err;\n        }\n      }\n\n      // Exponential backoff\n      const delay = RETRY_BACKOFF_BASE * 2 ** attempt;\n      await sleep(delay);\n    }\n\n    throw lastError!;\n  }\n\n  /** POST /sync/connect — SDK bootstrap. */\n  async connect(): Promise<Record<string, unknown>> {\n    return this.request(\"POST\", \"/sync/connect\");\n  }\n\n  /** POST /sync/heartbeat — periodic state refresh. */\n  async heartbeat(): Promise<Record<string, unknown>> {\n    return this.request(\"POST\", \"/sync/heartbeat\", {});\n  }\n\n  /** POST /events — batch event ingestion. */\n  async ingestEvents(\n    events: Record<string, unknown>[],\n  ): Promise<Record<string, unknown>> {\n    return this.request(\"POST\", \"/events\", { events });\n  }\n\n  /** PATCH /events/{uuid}/grade — submit human grade. */\n  async submitGrade(\n    eventUuid: string,\n    grade: number,\n  ): Promise<Record<string, unknown>> {\n    return this.request(\"PATCH\", `/events/${eventUuid}/grade`, {\n      grade,\n      is_graded: true,\n    });\n  }\n}\n","/**\n * SQLite WAL durability layer — crash-safe event storage.\n *\n * Events are written here immediately after update(). Background flush\n * sends them to cloud. If the process crashes, pending events survive\n * and are retried on next connect().\n */\n\nimport Database from \"better-sqlite3\";\n\nconst SCHEMA = `\nCREATE TABLE IF NOT EXISTS events (\n    local_event_uuid TEXT PRIMARY KEY,\n    bandit_id        INTEGER NOT NULL,\n    arm_id           INTEGER NOT NULL,\n    payload          TEXT NOT NULL,\n    status           TEXT NOT NULL DEFAULT 'pending',\n    created_at       REAL NOT NULL,\n    human_reward     REAL,\n    graded_at        REAL\n);\nCREATE INDEX IF NOT EXISTS idx_events_status ON events(status);\n`;\n\nconst MIGRATION_GRADING = [\n  \"ALTER TABLE events ADD COLUMN human_reward REAL\",\n  \"ALTER TABLE events ADD COLUMN graded_at REAL\",\n];\n\nexport interface EventPayload {\n  local_event_uuid: string;\n  bandit_id: number;\n  arm_id: number;\n  [key: string]: unknown;\n}\n\nexport class EventStore {\n  private db: Database.Database;\n  private pushStmt: Database.Statement;\n  private pendingStmt: Database.Statement;\n\n  constructor(dbPath: string = \":memory:\") {\n    this.db = new Database(dbPath);\n    this.db.pragma(\"journal_mode = WAL\");\n    this.db.pragma(\"busy_timeout = 5000\");\n    this.db.pragma(\"synchronous = NORMAL\");\n    this.db.exec(SCHEMA);\n    this.migrate();\n\n    // Pre-compile frequently-used statements\n    this.pushStmt = this.db.prepare(\n      `INSERT OR IGNORE INTO events\n       (local_event_uuid, bandit_id, arm_id, payload, status, created_at)\n       VALUES (?, ?, ?, ?, 'pending', ?)`,\n    );\n    this.pendingStmt = this.db.prepare(\n      `SELECT payload FROM events WHERE status = 'pending'\n       ORDER BY created_at ASC LIMIT ?`,\n    );\n  }\n\n  /** Insert a pending event. */\n  push(event: EventPayload): void {\n    this.pushStmt.run(\n      event.local_event_uuid,\n      event.bandit_id,\n      event.arm_id,\n      JSON.stringify(event),\n      Date.now() / 1000,\n    );\n  }\n\n  /** Return up to `limit` pending events (oldest first). */\n  pending(limit: number = 50): EventPayload[] {\n    const rows = this.pendingStmt.all(limit) as { payload: string }[];\n    return rows.map((row) => JSON.parse(row.payload));\n  }\n\n  /** Mark events as successfully flushed to cloud. */\n  markFlushed(uuids: string[]): void {\n    if (uuids.length === 0) return;\n    const placeholders = uuids.map(() => \"?\").join(\",\");\n    this.db\n      .prepare(\n        `UPDATE events SET status = 'flushed'\n         WHERE local_event_uuid IN (${placeholders})`,\n      )\n      .run(...uuids);\n  }\n\n  /** Record a human grade locally. */\n  markGraded(uuid: string, reward: number): void {\n    this.db\n      .prepare(\n        `UPDATE events SET human_reward = ?, graded_at = ?\n         WHERE local_event_uuid = ?`,\n      )\n      .run(reward, Date.now() / 1000, uuid);\n  }\n\n  /** Close the database connection. */\n  close(): void {\n    this.db.close();\n  }\n\n  private migrate(): void {\n    for (const stmt of MIGRATION_GRADING) {\n      try {\n        this.db.exec(stmt);\n      } catch {\n        // Column already exists\n      }\n    }\n  }\n}\n","/**\n * Payload utilities for cloud event ingestion.\n */\n\nconst TEXT_FIELDS = [\"query_text\", \"response\"] as const;\nconst METADATA_FIELDS = [\"model_name\", \"model_provider\"] as const;\n\n/**\n * Return shallow copies of events ready for cloud ingest.\n *\n * Always strips model_name/model_provider (only needed in local SQLite for TUI).\n * Strips query_text/response when includeText is false (dataStorage=\"local\").\n */\nexport function prepareCloudPayload(\n  events: Record<string, unknown>[],\n  includeText: boolean,\n): Record<string, unknown>[] {\n  return events.map((event) => {\n    const copy = { ...event };\n    for (const field of METADATA_FIELDS) {\n      delete copy[field];\n    }\n    if (!includeText) {\n      for (const field of TEXT_FIELDS) {\n        delete copy[field];\n      }\n    }\n    return copy;\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACSA,yBAA2B;AAC3B,IAAAA,MAAoB;AACpB,IAAAC,QAAsB;AACtB,IAAAC,MAAoB;AACpB,6BAA4B;;;ACJ5B,IAAI,aAA6D;AAMjE,eAAsB,WAA0B;AAC9C,MAAI,WAAY;AAChB,eAAa,MAAM,OAAO,wBAAwB;AACpD;AAMO,SAAS,aAAa,YAAsC;AACjE,MAAI,CAAC,YAAY;AACf,UAAM,IAAI,MAAM,mDAA8C;AAAA,EAChE;AACA,SAAO,IAAI,WAAW,aAAa,UAAU;AAC/C;AAMO,SAAS,aAAa,QAA0B,YAA0B;AAC/E,SAAO,eAAe,UAAU;AAClC;;;ACJO,SAAS,UAAU,MAMlB;AACN,QAAM,MAAW;AAAA,IACf,OAAO,KAAK;AAAA,IACZ,WAAW,KAAK;AAAA,IAChB,eAAe,KAAK;AAAA,IACpB,cAAc,KAAK;AAAA,IACnB,mBAAmB,KAAK,uBAAuB;AAAA,IAC/C,IAAI,QAAQ;AACV,aAAO,KAAK;AAAA,IACd;AAAA,IACA,IAAI,SAAS;AACX,aAAO,KAAK;AAAA,IACd;AAAA,EACF;AACA,SAAO,OAAO,OAAO,GAAG;AAC1B;AAGO,SAAS,iBAAiB,MAOlB;AACb,QAAM,SAAqB;AAAA,IACzB,KAAK,KAAK;AAAA,IACV,SAAS,KAAK;AAAA,IACd,UAAU,KAAK;AAAA,IACf,YAAY,KAAK;AAAA,IACjB,QAAQ,OAAO,OAAO,EAAE,GAAG,KAAK,OAAO,CAAC;AAAA,IACxC,IAAI,QAAQ;AACV,aAAO,KAAK,IAAI;AAAA,IAClB;AAAA,IACA,IAAI,SAAS;AACX,aAAO,KAAK,IAAI;AAAA,IAClB;AAAA,IACA,WAAW,KAAK;AAAA,EAClB;AACA,SAAO,OAAO,OAAO,MAAM;AAC7B;;;ACzEA,SAAoB;AACpB,WAAsB;AACtB,SAAoB;AACpB,uBAAmC;AAE5B,IAAM,mBAAmB;AAChC,IAAM,aAAkB,UAAQ,WAAQ,GAAG,UAAU;AACrD,IAAM,cAAmB,UAAK,YAAY,aAAa;AAWhD,SAAS,aAA4B;AAC1C,QAAM,SAAwB;AAAA,IAC5B,QAAQ;AAAA,IACR,SAAS;AAAA,IACT,aAAa;AAAA,EACf;AAGA,MAAO,cAAW,WAAW,GAAG;AAC9B,QAAI;AACF,YAAM,UAAa,gBAAa,aAAa,OAAO;AACpD,YAAM,WAAO,iBAAAC,OAAU,OAAO;AAC9B,UAAI,KAAK,QAAS,QAAO,SAAS,KAAK;AACvC,UAAI,KAAK,SAAU,QAAO,UAAU,KAAK;AACzC,UAAI,KAAK,aAAc,QAAO,cAAc,KAAK;AAAA,IACnD,QAAQ;AAAA,IAER;AAAA,EACF;AAGA,QAAM,SAAS,QAAQ,IAAI;AAC3B,MAAI,OAAQ,QAAO,SAAS;AAE5B,QAAM,SAAS,QAAQ,IAAI;AAC3B,MAAI,OAAQ,QAAO,UAAU;AAE7B,QAAM,aAAa,QAAQ,IAAI;AAC/B,MAAI,WAAY,QAAO,cAAc;AAErC,SAAO;AACT;;;ACjDA,IAAM,cAAc;AACpB,IAAM,qBAAqB;AAE3B,SAAS,MAAM,IAA2B;AACxC,SAAO,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,EAAE,CAAC;AACzD;AAEA,SAAS,YAAY,QAAyB;AAC5C,SAAO,UAAU;AACnB;AAEO,IAAM,cAAN,MAAkB;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EAER,YAAY,SAAiB,QAAgB,UAAkB,KAAQ;AACrE,SAAK,UAAU,GAAG,QAAQ,QAAQ,OAAO,EAAE,CAAC;AAC5C,SAAK,SAAS;AACd,SAAK,UAAU;AAAA,EACjB;AAAA,EAEA,MAAc,QACZ,QACAC,OACA,MACkC;AAClC,QAAI,YAA0B;AAE9B,aAAS,UAAU,GAAG,UAAU,aAAa,WAAW;AACtD,YAAM,aAAa,IAAI,gBAAgB;AACvC,YAAM,QAAQ,WAAW,MAAM,WAAW,MAAM,GAAG,KAAK,OAAO;AAE/D,UAAI;AACF,cAAM,OAAO,MAAM,MAAM,GAAG,KAAK,OAAO,GAAGA,KAAI,IAAI;AAAA,UACjD;AAAA,UACA,SAAS;AAAA,YACP,aAAa,KAAK;AAAA,YAClB,gBAAgB;AAAA,UAClB;AAAA,UACA,MAAM,QAAQ,OAAO,KAAK,UAAU,IAAI,IAAI;AAAA,UAC5C,QAAQ,WAAW;AAAA,QACrB,CAAC;AAED,qBAAa,KAAK;AAElB,YAAI,CAAC,KAAK,IAAI;AACZ,gBAAM,OAAO,MAAM,KAAK,KAAK,EAAE,MAAM,MAAM,EAAE;AAC7C,cAAI,CAAC,YAAY,KAAK,MAAM,KAAK,YAAY,cAAc,GAAG;AAC5D,kBAAM,IAAI;AAAA,cACR,QAAQ,KAAK,MAAM,OAAO,MAAM,IAAIA,KAAI,KAAK,IAAI;AAAA,YACnD;AAAA,UACF;AAEA,sBAAY,IAAI;AAAA,YACd,QAAQ,KAAK,MAAM,OAAO,MAAM,IAAIA,KAAI,KAAK,IAAI;AAAA,UACnD;AAAA,QACF,OAAO;AACL,iBAAQ,MAAM,KAAK,KAAK;AAAA,QAC1B;AAAA,MACF,SAAS,KAAK;AACZ,qBAAa,KAAK;AAClB,oBAAY;AAGZ,cAAM,qBACH,IAAc,SAAS,gBACvB,IAAc,SAAS;AAC1B,YAAI,CAAC,sBAAsB,UAAU,cAAc,GAAG;AAEpD,gBAAM;AAAA,QACR;AACA,YAAI,YAAY,cAAc,GAAG;AAC/B,gBAAM;AAAA,QACR;AAAA,MACF;AAGA,YAAM,QAAQ,qBAAqB,KAAK;AACxC,YAAM,MAAM,KAAK;AAAA,IACnB;AAEA,UAAM;AAAA,EACR;AAAA;AAAA,EAGA,MAAM,UAA4C;AAChD,WAAO,KAAK,QAAQ,QAAQ,eAAe;AAAA,EAC7C;AAAA;AAAA,EAGA,MAAM,YAA8C;AAClD,WAAO,KAAK,QAAQ,QAAQ,mBAAmB,CAAC,CAAC;AAAA,EACnD;AAAA;AAAA,EAGA,MAAM,aACJ,QACkC;AAClC,WAAO,KAAK,QAAQ,QAAQ,WAAW,EAAE,OAAO,CAAC;AAAA,EACnD;AAAA;AAAA,EAGA,MAAM,YACJ,WACAC,QACkC;AAClC,WAAO,KAAK,QAAQ,SAAS,WAAW,SAAS,UAAU;AAAA,MACzD,OAAAA;AAAA,MACA,WAAW;AAAA,IACb,CAAC;AAAA,EACH;AACF;;;AC/GA,4BAAqB;AAErB,IAAM,SAAS;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAcf,IAAM,oBAAoB;AAAA,EACxB;AAAA,EACA;AACF;AASO,IAAM,aAAN,MAAiB;AAAA,EACd;AAAA,EACA;AAAA,EACA;AAAA,EAER,YAAY,SAAiB,YAAY;AACvC,SAAK,KAAK,IAAI,sBAAAC,QAAS,MAAM;AAC7B,SAAK,GAAG,OAAO,oBAAoB;AACnC,SAAK,GAAG,OAAO,qBAAqB;AACpC,SAAK,GAAG,OAAO,sBAAsB;AACrC,SAAK,GAAG,KAAK,MAAM;AACnB,SAAK,QAAQ;AAGb,SAAK,WAAW,KAAK,GAAG;AAAA,MACtB;AAAA;AAAA;AAAA,IAGF;AACA,SAAK,cAAc,KAAK,GAAG;AAAA,MACzB;AAAA;AAAA,IAEF;AAAA,EACF;AAAA;AAAA,EAGA,KAAK,OAA2B;AAC9B,SAAK,SAAS;AAAA,MACZ,MAAM;AAAA,MACN,MAAM;AAAA,MACN,MAAM;AAAA,MACN,KAAK,UAAU,KAAK;AAAA,MACpB,KAAK,IAAI,IAAI;AAAA,IACf;AAAA,EACF;AAAA;AAAA,EAGA,QAAQ,QAAgB,IAAoB;AAC1C,UAAM,OAAO,KAAK,YAAY,IAAI,KAAK;AACvC,WAAO,KAAK,IAAI,CAAC,QAAQ,KAAK,MAAM,IAAI,OAAO,CAAC;AAAA,EAClD;AAAA;AAAA,EAGA,YAAY,OAAuB;AACjC,QAAI,MAAM,WAAW,EAAG;AACxB,UAAM,eAAe,MAAM,IAAI,MAAM,GAAG,EAAE,KAAK,GAAG;AAClD,SAAK,GACF;AAAA,MACC;AAAA,sCAC8B,YAAY;AAAA,IAC5C,EACC,IAAI,GAAG,KAAK;AAAA,EACjB;AAAA;AAAA,EAGA,WAAW,MAAc,QAAsB;AAC7C,SAAK,GACF;AAAA,MACC;AAAA;AAAA,IAEF,EACC,IAAI,QAAQ,KAAK,IAAI,IAAI,KAAM,IAAI;AAAA,EACxC;AAAA;AAAA,EAGA,QAAc;AACZ,SAAK,GAAG,MAAM;AAAA,EAChB;AAAA,EAEQ,UAAgB;AACtB,eAAW,QAAQ,mBAAmB;AACpC,UAAI;AACF,aAAK,GAAG,KAAK,IAAI;AAAA,MACnB,QAAQ;AAAA,MAER;AAAA,IACF;AAAA,EACF;AACF;;;AC9GA,IAAM,cAAc,CAAC,cAAc,UAAU;AAC7C,IAAM,kBAAkB,CAAC,cAAc,gBAAgB;AAQhD,SAAS,oBACd,QACA,aAC2B;AAC3B,SAAO,OAAO,IAAI,CAAC,UAAU;AAC3B,UAAM,OAAO,EAAE,GAAG,MAAM;AACxB,eAAW,SAAS,iBAAiB;AACnC,aAAO,KAAK,KAAK;AAAA,IACnB;AACA,QAAI,CAAC,aAAa;AAChB,iBAAW,SAAS,aAAa;AAC/B,eAAO,KAAK,KAAK;AAAA,MACnB;AAAA,IACF;AACA,WAAO;AAAA,EACT,CAAC;AACH;;;ANAA,IAAM,qBAA0B,WAAQ,YAAQ,GAAG,YAAY,WAAW;AAC1E,IAAM,oBAAoB;AA0BnB,IAAM,gBAAN,MAAoB;AAAA,EACjB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAEA,OAA2B;AAAA,EAC3B,QAA2B;AAAA,EAC3B,UAAqC,oBAAI,IAAI;AAAA,EAC7C,UAAoC,oBAAI,IAAI;AAAA,EAC5C,YAAY;AAAA,EACZ,gBAAuD;AAAA,EACvD,kBAAkB;AAAA,EAClB,YAAyB,oBAAI,IAAI;AAAA,EACjC,cAAmC,oBAAI,IAAI;AAAA,EAEnD,YAAY,UAAyB,CAAC,GAAG;AACvC,SAAK,SAAS,QAAQ;AACtB,SAAK,UAAU,QAAQ;AACvB,SAAK,YAAY,QAAQ;AACzB,SAAK,iBAAiB,QAAQ;AAC9B,SAAK,cAAc,QAAQ,eAAe;AAAA,EAC5C;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,UAAyB;AAE7B,QAAI,KAAK,WAAW;AAClB,YAAM,KAAK,MAAM;AAAA,IACnB;AAGA,UAAM,SAAS;AAGf,UAAM,SAAS,WAAW;AAC1B,UAAM,SAAS,KAAK,UAAU,OAAO;AACrC,QAAI,CAAC,QAAQ;AACX,YAAM,IAAI;AAAA,QACR;AAAA,MAEF;AAAA,IACF;AAEA,UAAM,UAAU,KAAK,WAAW,OAAO;AACvC,QAAI,CAAC,KAAK,gBAAgB;AACxB,WAAK,cAAc,OAAO;AAAA,IAC5B;AAEA,SAAK,OAAO,IAAI,YAAY,SAAS,MAAM;AAC3C,UAAM,YAAY,KAAK,aAAa;AACpC,QAAI,cAAc,YAAY;AAC5B,YAAM,MAAW,cAAQ,SAAS;AAClC,UAAI,CAAI,eAAW,GAAG,GAAG;AACvB,QAAG,cAAU,KAAK,EAAE,WAAW,KAAK,CAAC;AAAA,MACvC;AAAA,IACF;AACA,SAAK,QAAQ,IAAI,WAAW,SAAS;AAGrC,QAAI;AACF,YAAM,OAAO,MAAM,KAAK,KAAK,QAAQ;AACrC,WAAK,UAAU,IAAI;AAGnB,WAAK,UAAU,MAAM;AACrB,WAAK,YAAY,MAAM;AAGvB,YAAM,KAAK,aAAa;AAGxB,WAAK,gBAAgB,YAAY,MAAM;AACrC,aAAK,aAAa,EAAE,MAAM,MAAM;AAAA,QAAC,CAAC;AAAA,MACpC,GAAG,GAAM;AAET,WAAK,YAAY;AAAA,IACnB,SAAS,KAAK;AACZ,WAAK,OAAO,MAAM;AAClB,WAAK,QAAQ;AACb,WAAK,OAAO;AACZ,YAAM;AAAA,IACR;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,KAAK,YAAoB,UAAuB,CAAC,GAAe;AAC9D,SAAK,gBAAgB;AAErB,UAAM,QAAQ,KAAK,QAAQ,IAAI,UAAU;AACzC,QAAI,CAAC,OAAO;AACV,YAAM,YAAY,CAAC,GAAG,KAAK,QAAQ,KAAK,CAAC;AACzC,YAAM,IAAI;AAAA,QACR,mBAAmB,UAAU,kBAAkB,UAAU,KAAK,IAAI,CAAC;AAAA,MACrE;AAAA,IACF;AAEA,QAAI,MAAM,KAAK,WAAW,GAAG;AAC3B,YAAM,IAAI,MAAM,WAAW,UAAU,sBAAsB;AAAA,IAC7D;AAEA,UAAM,SAAS,KAAK,QAAQ,IAAI,UAAU;AAC1C,UAAM,cAAc,QAAQ,OAAO,UAAU;AAC7C,UAAM,aAAa,QAAQ,UAAU,WAAW,KAAK,QAAQ,OAAO,IAAI;AAExE,UAAM,aAAa,OAAO,KAAK,aAAa,UAAU;AACtD,UAAM,MAAwB,KAAK,MAAM,UAAU;AAGnD,UAAM,YAAY,MAAM,KAAK,KAAK,CAAC,MAAM,EAAE,UAAU,IAAI,MAAM;AAC/D,QAAI,CAAC,WAAW;AACd,YAAM,IAAI;AAAA,QACR,uBAAuB,IAAI,MAAM,0CAA0C,UAAU;AAAA,MAEvF;AAAA,IACF;AAEA,WAAO,iBAAiB;AAAA,MACtB,KAAK;AAAA,MACL,aAAS,+BAAW;AAAA,MACpB,UAAU,MAAM;AAAA,MAChB;AAAA,MACA,QAAQ,IAAI;AAAA,MACZ,UAAU,mCAAY,IAAI;AAAA,IAC5B,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,OAAO,YAAwB,UAAyB,CAAC,GAAS;AAChE,SAAK,gBAAgB;AAErB,QAAI,SAAS,QAAQ;AACrB,QAAI,QAAQ,UAAU,UAAU,MAAM;AACpC,eAAS;AAAA,IACX;AAGA,QAAI,UAAU,QAAQ;AACtB,QAAI,WAAW,QAAQ,WAAW,YAAY,GAAG;AAC/C,gBAAU,mCAAY,IAAI,IAAI,WAAW;AAAA,IAC3C;AAGA,UAAM,QAAsB;AAAA,MAC1B,kBAAkB,WAAW;AAAA,MAC7B,WAAW,WAAW;AAAA,MACtB,QAAQ,WAAW,IAAI;AAAA,MACvB,YAAY,WAAW,IAAI;AAAA,MAC3B,gBAAgB,WAAW,IAAI;AAAA,IACjC;AAEA,QAAI,QAAQ,aAAa,MAAM;AAC7B,MAAC,MAAkC,aAAa,QAAQ;AAAA,IAC1D;AACA,QAAI,QAAQ,YAAY,MAAM;AAC5B,MAAC,MAAkC,WACjC,OAAO,QAAQ,aAAa,WACxB,EAAE,UAAU,QAAQ,SAAS,IAC7B,QAAQ;AAAA,IAChB;AACA,QAAI,UAAU,MAAM;AAClB,MAAC,MAAkC,eAAe;AAAA,IACpD;AACA,QAAI,QAAQ,QAAQ,MAAM;AACxB,MAAC,MAAkC,OAAO,QAAQ;AAAA,IACpD;AACA,QAAI,WAAW,MAAM;AACnB,MAAC,MAAkC,UAAU;AAAA,IAC/C;AACA,QAAI,QAAQ,eAAe,MAAM;AAC/B,MAAC,MAAkC,eAAe,QAAQ;AAAA,IAC5D;AACA,QAAI,QAAQ,gBAAgB,MAAM;AAChC,MAAC,MAAkC,gBAAgB,QAAQ;AAAA,IAC7D;AACA,QAAI,QAAQ,WAAW,MAAM;AAC3B,MAAC,MAAkC,UAAU,QAAQ;AAAA,IACvD;AACA,QAAI,QAAQ,QAAQ;AAClB,MAAC,MAAkC,YAAY;AAAA,IACjD;AAGA,SAAK,MAAO,KAAK,KAAK;AAGtB,SAAK,aAAa,EAAE,MAAM,MAAM;AAAA,IAAC,CAAC;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,MAAM,SAAiBC,QAA8B;AACzD,SAAK,gBAAgB;AACrB,UAAM,KAAK,KAAM,YAAY,SAASA,MAAK;AAAA,EAC7C;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,OAAsB;AAC1B,SAAK,gBAAgB;AACrB,UAAM,OAAO,MAAM,KAAK,KAAM,UAAU;AAExC,UAAM,cAAc,IAAI,IAAI,KAAK,OAAO;AACxC,UAAM,cAAc,IAAI,IAAI,KAAK,OAAO;AACxC,QAAI;AACF,WAAK,UAAU,IAAI;AAAA,IACrB,SAAS,KAAK;AAEZ,WAAK,UAAU;AACf,WAAK,UAAU;AACf,cAAQ,KAAK,0EAAqE,GAAG;AAAA,IACvF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,QAAuB;AAC3B,QAAI,KAAK,eAAe;AACtB,oBAAc,KAAK,aAAa;AAChC,WAAK,gBAAgB;AAAA,IACvB;AAGA,QAAI,KAAK,SAAS,KAAK,MAAM;AAC3B,YAAM,KAAK,aAAa;AAAA,IAC1B;AAEA,SAAK,OAAO,MAAM;AAClB,SAAK,QAAQ;AACb,SAAK,OAAO;AACZ,SAAK,QAAQ,MAAM;AACnB,SAAK,QAAQ,MAAM;AACnB,SAAK,YAAY;AAAA,EACnB;AAAA;AAAA,EAIQ,kBAAwB;AAC9B,QAAI,CAAC,KAAK,WAAW;AACnB,YAAM,IAAI,MAAM,2CAAsC;AAAA,IACxD;AAAA,EACF;AAAA,EAEQ,UAAU,MAAqC;AACrD,UAAM,cAAe,KAAK,WAAW,CAAC;AAEtC,UAAM,aAAa,oBAAI,IAAyB;AAChD,UAAM,aAAa,oBAAI,IAA0B;AAEjD,eAAW,KAAK,aAAa;AAC3B,YAAM,OAAQ,EAAE,QAAQ,CAAC;AACzB,UAAI,KAAK,WAAW,EAAG;AAEvB,YAAM,aAAoB,KACvB,OAAO,CAAC,MAAM,EAAE,SAAS,EACzB,IAAI,CAAC,MAAM,UAAU,CAAC,CAAC;AAE1B,YAAM,OAAO,EAAE;AACf,YAAM,QAAqB;AAAA,QACzB,UAAU,OAAO,EAAE,SAAS;AAAA,QAC5B;AAAA,QACA,MAAM;AAAA,QACN,SAAS;AAAA,QACT,kBAAmB,EAAE,qBAAgC;AAAA,QACrD,iBAAiB,EAAE;AAAA,QACnB,QAAQ,EAAE;AAAA,QACV,WAAW,EAAE;AAAA,MACf;AAEA,iBAAW,IAAI,MAAM,KAAK;AAG1B,YAAM,iBAAiB,KAAK,QAAQ,IAAI,IAAI;AAC5C,YAAM,aAAa,KAAK,UAAU,CAAC;AACnC,UAAI,gBAAgB;AAClB,qBAAa,gBAAgB,UAAU;AACvC,mBAAW,IAAI,MAAM,cAAc;AAAA,MACrC,OAAO;AACL,mBAAW,IAAI,MAAM,aAAa,UAAU,CAAC;AAAA,MAC/C;AAAA,IACF;AAEA,SAAK,UAAU;AACf,SAAK,UAAU;AAAA,EACjB;AAAA,EAEA,MAAc,eAA8B;AAC1C,QAAI,KAAK,mBAAmB,CAAC,KAAK,SAAS,CAAC,KAAK,KAAM;AACvD,SAAK,kBAAkB;AAEvB,QAAI;AACF,YAAM,UAAU,KAAK,MAAM,QAAQ;AACnC,UAAI,QAAQ,WAAW,EAAG;AAG1B,YAAM,QAAQ,KAAK,UAAU,OAAO,IAChC,QAAQ,OAAO,CAAC,MAAM,CAAC,KAAK,UAAU,IAAI,EAAE,gBAAgB,CAAC,IAC7D;AACJ,UAAI,MAAM,WAAW,EAAG;AAGxB,YAAM,UAAU;AAAA,QACd;AAAA,QACA,KAAK,gBAAgB;AAAA,MACvB;AAEA,YAAM,SAAS,MAAM,KAAK,KAAK,aAAa,OAAO;AAGnD,YAAM,SAAU,OAAO,UAAU,CAAC;AAIlC,YAAM,eAAe,IAAI;AAAA,QACvB,OAAO,IAAI,CAAC,MAAM,EAAE,gBAAgB,EAAE,OAAO,OAAO;AAAA,MACtD;AAGA,iBAAW,OAAO,cAAc;AAC9B,cAAM,SAAS,KAAK,YAAY,IAAI,GAAG,KAAK,KAAK;AACjD,aAAK,YAAY,IAAI,KAAK,KAAK;AAC/B,YAAI,SAAS,mBAAmB;AAC9B,eAAK,UAAU,IAAI,GAAG;AAAA,QACxB;AAAA,MACF;AAGA,YAAM,eAAe,MAClB,IAAI,CAAC,MAAM,EAAE,gBAAgB,EAC7B,OAAO,CAAC,QAAQ,CAAC,aAAa,IAAI,GAAG,CAAC;AACzC,UAAI,aAAa,SAAS,KAAK,KAAK,OAAO;AACzC,aAAK,MAAM,YAAY,YAAY;AAAA,MACrC;AAAA,IACF,SAAS,KAAK;AAEZ,cAAQ,KAAK,kDAA6C,GAAG;AAAA,IAC/D,UAAE;AACA,WAAK,kBAAkB;AAAA,IACzB;AAAA,EACF;AACF;;;AD9XA,IAAI,UAAgC;AAEpC,SAAS,YAA2B;AAClC,MAAI,CAAC,SAAS;AACZ,UAAM,IAAI,MAAM,2CAAsC;AAAA,EACxD;AACA,SAAO;AACT;AAGA,eAAsB,QAAQ,UAAyB,CAAC,GAAkB;AACxE,MAAI,SAAS;AACX,UAAM,QAAQ,MAAM;AAAA,EACtB;AACA,YAAU,IAAI,cAAc,OAAO;AACnC,QAAM,QAAQ,QAAQ;AACxB;AAGO,SAAS,KAAK,YAAoB,SAAmC;AAC1E,SAAO,UAAU,EAAE,KAAK,YAAY,OAAO;AAC7C;AAGO,SAAS,OAAO,YAAwB,SAA+B;AAC5E,YAAU,EAAE,OAAO,YAAY,OAAO;AACxC;AAGA,eAAsB,MAAM,SAAiB,YAAmC;AAC9E,QAAM,UAAU,EAAE,MAAM,SAAS,UAAU;AAC7C;AAGA,eAAsB,OAAsB;AAC1C,QAAM,UAAU,EAAE,KAAK;AACzB;AAGA,eAAsB,QAAuB;AAC3C,MAAI,SAAS;AACX,UAAM,QAAQ,MAAM;AACpB,cAAU;AAAA,EACZ;AACF;","names":["fs","path","os","parseToml","path","grade","Database","grade"]}

package/dist/index.mjs

@@ -0,0 +1,597 @@
+// src/client.ts
+import { randomUUID } from "crypto";
+import * as fs2 from "fs";
+import * as path2 from "path";
+import * as os2 from "os";
+import { performance } from "perf_hooks";
+
+// src/engine.ts
+var wasmModule = null;
+async function initWasm() {
+  if (wasmModule) return;
+  wasmModule = await import("../wasm/bandito_engine");
+}
+function createEngine(banditJson) {
+  if (!wasmModule) {
+    throw new Error("WASM not initialized \u2014 call initWasm() first");
+  }
+  return new wasmModule.BanditEngine(banditJson);
+}
+function updateEngine(engine, banditJson) {
+  engine.updateFromSync(banditJson);
+}
+
+// src/models.ts
+function createArm(data) {
+  const arm = {
+    armId: data.arm_id,
+    modelName: data.model_name,
+    modelProvider: data.model_provider,
+    systemPrompt: data.system_prompt,
+    isPromptTemplated: data.is_prompt_templated ?? false,
+    get model() {
+      return this.modelName;
+    },
+    get prompt() {
+      return this.systemPrompt;
+    }
+  };
+  return Object.freeze(arm);
+}
+function createPullResult(data) {
+  const result = {
+    arm: data.arm,
+    eventId: data.eventId,
+    banditId: data.banditId,
+    banditName: data.banditName,
+    scores: Object.freeze({ ...data.scores }),
+    get model() {
+      return this.arm.modelName;
+    },
+    get prompt() {
+      return this.arm.systemPrompt;
+    },
+    _pullTime: data.pullTime
+  };
+  return Object.freeze(result);
+}
+
+// src/config.ts
+import * as fs from "fs";
+import * as path from "path";
+import * as os from "os";
+import { parse as parseToml } from "smol-toml";
+var DEFAULT_BASE_URL = "https://bandito-api.onrender.com";
+var CONFIG_DIR = path.join(os.homedir(), ".bandito");
+var CONFIG_FILE = path.join(CONFIG_DIR, "config.toml");
+function loadConfig() {
+  const config = {
+    apiKey: null,
+    baseUrl: DEFAULT_BASE_URL,
+    dataStorage: "local"
+  };
+  if (fs.existsSync(CONFIG_FILE)) {
+    try {
+      const content = fs.readFileSync(CONFIG_FILE, "utf-8");
+      const data = parseToml(content);
+      if (data.api_key) config.apiKey = data.api_key;
+      if (data.base_url) config.baseUrl = data.base_url;
+      if (data.data_storage) config.dataStorage = data.data_storage;
+    } catch {
+    }
+  }
+  const envKey = process.env.BANDITO_API_KEY;
+  if (envKey) config.apiKey = envKey;
+  const envUrl = process.env.BANDITO_BASE_URL;
+  if (envUrl) config.baseUrl = envUrl;
+  const envStorage = process.env.BANDITO_DATA_STORAGE;
+  if (envStorage) config.dataStorage = envStorage;
+  return config;
+}
+
+// src/http.ts
+var MAX_RETRIES = 3;
+var RETRY_BACKOFF_BASE = 500;
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function isRetryable(status) {
+  return status >= 500;
+}
+var BanditoHTTP = class {
+  baseUrl;
+  apiKey;
+  timeout;
+  constructor(baseUrl, apiKey, timeout = 1e4) {
+    this.baseUrl = `${baseUrl.replace(/\/$/, "")}/api/v1`;
+    this.apiKey = apiKey;
+    this.timeout = timeout;
+  }
+  async request(method, path3, body) {
+    let lastError = null;
+    for (let attempt = 0; attempt < MAX_RETRIES; attempt++) {
+      const controller = new AbortController();
+      const timer = setTimeout(() => controller.abort(), this.timeout);
+      try {
+        const resp = await fetch(`${this.baseUrl}${path3}`, {
+          method,
+          headers: {
+            "X-API-Key": this.apiKey,
+            "Content-Type": "application/json"
+          },
+          body: body != null ? JSON.stringify(body) : void 0,
+          signal: controller.signal
+        });
+        clearTimeout(timer);
+        if (!resp.ok) {
+          const text = await resp.text().catch(() => "");
+          if (!isRetryable(resp.status) || attempt === MAX_RETRIES - 1) {
+            throw new Error(
+              `HTTP ${resp.status} on ${method} ${path3}: ${text}`
+            );
+          }
+          lastError = new Error(
+            `HTTP ${resp.status} on ${method} ${path3}: ${text}`
+          );
+        } else {
+          return await resp.json();
+        }
+      } catch (err) {
+        clearTimeout(timer);
+        lastError = err;
+        const isNetworkOrTimeout = err.name === "AbortError" || err.name === "TypeError";
+        if (!isNetworkOrTimeout && attempt < MAX_RETRIES - 1) {
+          throw err;
+        }
+        if (attempt === MAX_RETRIES - 1) {
+          throw err;
+        }
+      }
+      const delay = RETRY_BACKOFF_BASE * 2 ** attempt;
+      await sleep(delay);
+    }
+    throw lastError;
+  }
+  /** POST /sync/connect — SDK bootstrap. */
+  async connect() {
+    return this.request("POST", "/sync/connect");
+  }
+  /** POST /sync/heartbeat — periodic state refresh. */
+  async heartbeat() {
+    return this.request("POST", "/sync/heartbeat", {});
+  }
+  /** POST /events — batch event ingestion. */
+  async ingestEvents(events) {
+    return this.request("POST", "/events", { events });
+  }
+  /** PATCH /events/{uuid}/grade — submit human grade. */
+  async submitGrade(eventUuid, grade2) {
+    return this.request("PATCH", `/events/${eventUuid}/grade`, {
+      grade: grade2,
+      is_graded: true
+    });
+  }
+};
+
+// src/store.ts
+import Database from "better-sqlite3";
+var SCHEMA = `
+CREATE TABLE IF NOT EXISTS events (
+    local_event_uuid TEXT PRIMARY KEY,
+    bandit_id        INTEGER NOT NULL,
+    arm_id           INTEGER NOT NULL,
+    payload          TEXT NOT NULL,
+    status           TEXT NOT NULL DEFAULT 'pending',
+    created_at       REAL NOT NULL,
+    human_reward     REAL,
+    graded_at        REAL
+);
+CREATE INDEX IF NOT EXISTS idx_events_status ON events(status);
+`;
+var MIGRATION_GRADING = [
+  "ALTER TABLE events ADD COLUMN human_reward REAL",
+  "ALTER TABLE events ADD COLUMN graded_at REAL"
+];
+var EventStore = class {
+  db;
+  pushStmt;
+  pendingStmt;
+  constructor(dbPath = ":memory:") {
+    this.db = new Database(dbPath);
+    this.db.pragma("journal_mode = WAL");
+    this.db.pragma("busy_timeout = 5000");
+    this.db.pragma("synchronous = NORMAL");
+    this.db.exec(SCHEMA);
+    this.migrate();
+    this.pushStmt = this.db.prepare(
+      `INSERT OR IGNORE INTO events
+       (local_event_uuid, bandit_id, arm_id, payload, status, created_at)
+       VALUES (?, ?, ?, ?, 'pending', ?)`
+    );
+    this.pendingStmt = this.db.prepare(
+      `SELECT payload FROM events WHERE status = 'pending'
+       ORDER BY created_at ASC LIMIT ?`
+    );
+  }
+  /** Insert a pending event. */
+  push(event) {
+    this.pushStmt.run(
+      event.local_event_uuid,
+      event.bandit_id,
+      event.arm_id,
+      JSON.stringify(event),
+      Date.now() / 1e3
+    );
+  }
+  /** Return up to `limit` pending events (oldest first). */
+  pending(limit = 50) {
+    const rows = this.pendingStmt.all(limit);
+    return rows.map((row) => JSON.parse(row.payload));
+  }
+  /** Mark events as successfully flushed to cloud. */
+  markFlushed(uuids) {
+    if (uuids.length === 0) return;
+    const placeholders = uuids.map(() => "?").join(",");
+    this.db.prepare(
+      `UPDATE events SET status = 'flushed'
+         WHERE local_event_uuid IN (${placeholders})`
+    ).run(...uuids);
+  }
+  /** Record a human grade locally. */
+  markGraded(uuid, reward) {
+    this.db.prepare(
+      `UPDATE events SET human_reward = ?, graded_at = ?
+         WHERE local_event_uuid = ?`
+    ).run(reward, Date.now() / 1e3, uuid);
+  }
+  /** Close the database connection. */
+  close() {
+    this.db.close();
+  }
+  migrate() {
+    for (const stmt of MIGRATION_GRADING) {
+      try {
+        this.db.exec(stmt);
+      } catch {
+      }
+    }
+  }
+};
+
+// src/worker.ts
+var TEXT_FIELDS = ["query_text", "response"];
+var METADATA_FIELDS = ["model_name", "model_provider"];
+function prepareCloudPayload(events, includeText) {
+  return events.map((event) => {
+    const copy = { ...event };
+    for (const field of METADATA_FIELDS) {
+      delete copy[field];
+    }
+    if (!includeText) {
+      for (const field of TEXT_FIELDS) {
+        delete copy[field];
+      }
+    }
+    return copy;
+  });
+}
+
+// src/client.ts
+var DEFAULT_STORE_PATH = path2.join(os2.homedir(), ".bandito", "events.db");
+var MAX_EVENT_RETRIES = 5;
+var BanditoClient = class {
+  apiKey;
+  baseUrl;
+  storePath;
+  dataStorageArg;
+  dataStorage;
+  http = null;
+  store = null;
+  engines = /* @__PURE__ */ new Map();
+  bandits = /* @__PURE__ */ new Map();
+  connected = false;
+  flushInterval = null;
+  flushInProgress = false;
+  deadUuids = /* @__PURE__ */ new Set();
+  retryCounts = /* @__PURE__ */ new Map();
+  constructor(options = {}) {
+    this.apiKey = options.apiKey;
+    this.baseUrl = options.baseUrl;
+    this.storePath = options.storePath;
+    this.dataStorageArg = options.dataStorage;
+    this.dataStorage = options.dataStorage ?? "local";
+  }
+  /**
+   * Bootstrap: authenticate and hydrate in-memory state from cloud.
+   *
+   * Resolves config from: constructor args → env vars → ~/.bandito/config.toml.
+   * Initializes WASM, creates HTTP client, SQLite store, fetches full state.
+   */
+  async connect() {
+    if (this.connected) {
+      await this.close();
+    }
+    await initWasm();
+    const config = loadConfig();
+    const apiKey = this.apiKey ?? config.apiKey;
+    if (!apiKey) {
+      throw new Error(
+        "apiKey required \u2014 pass it to constructor, set BANDITO_API_KEY, or run `bandito signup`"
+      );
+    }
+    const baseUrl = this.baseUrl ?? config.baseUrl;
+    if (!this.dataStorageArg) {
+      this.dataStorage = config.dataStorage;
+    }
+    this.http = new BanditoHTTP(baseUrl, apiKey);
+    const storePath = this.storePath ?? DEFAULT_STORE_PATH;
+    if (storePath !== ":memory:") {
+      const dir = path2.dirname(storePath);
+      if (!fs2.existsSync(dir)) {
+        fs2.mkdirSync(dir, { recursive: true });
+      }
+    }
+    this.store = new EventStore(storePath);
+    try {
+      const data = await this.http.connect();
+      this.applySync(data);
+      this.deadUuids.clear();
+      this.retryCounts.clear();
+      await this.flushPending();
+      this.flushInterval = setInterval(() => {
+        this.flushPending().catch(() => {
+        });
+      }, 3e4);
+      this.connected = true;
+    } catch (err) {
+      this.store?.close();
+      this.store = null;
+      this.http = null;
+      throw err;
+    }
+  }
+  /**
+   * Local Thompson Sampling decision. Synchronous, <1ms, no network.
+   */
+  pull(banditName, options = {}) {
+    this.ensureConnected();
+    const cache = this.bandits.get(banditName);
+    if (!cache) {
+      const available = [...this.bandits.keys()];
+      throw new Error(
+        `Unknown bandit '${banditName}'. Available: [${available.join(", ")}]`
+      );
+    }
+    if (cache.arms.length === 0) {
+      throw new Error(`Bandit '${banditName}' has no active arms`);
+    }
+    const engine = this.engines.get(banditName);
+    const queryLength = options.query?.length ?? void 0;
+    const excludeIds = options.exclude ? Int32Array.from(options.exclude) : void 0;
+    const resultJson = engine.pull(queryLength, excludeIds);
+    const raw = JSON.parse(resultJson);
+    const winnerArm = cache.arms.find((a) => a.armId === raw.arm_id);
+    if (!winnerArm) {
+      throw new Error(
+        `Engine selected arm ${raw.arm_id} but it's not in active arm cache for "${banditName}". This is likely a bug \u2014 please report it at https://github.com/bandito-ai/bandito/issues`
+      );
+    }
+    return createPullResult({
+      arm: winnerArm,
+      eventId: randomUUID(),
+      banditId: cache.banditId,
+      banditName,
+      scores: raw.scores,
+      pullTime: performance.now()
+    });
+  }
+  /**
+   * Record an LLM call outcome. Writes to SQLite first (crash-safe),
+   * then fires off a non-blocking flush to cloud.
+   */
+  update(pullResult, options = {}) {
+    this.ensureConnected();
+    let reward = options.reward;
+    if (options.failed && reward == null) {
+      reward = 0;
+    }
+    let latency = options.latency;
+    if (latency == null && pullResult._pullTime > 0) {
+      latency = performance.now() - pullResult._pullTime;
+    }
+    const event = {
+      local_event_uuid: pullResult.eventId,
+      bandit_id: pullResult.banditId,
+      arm_id: pullResult.arm.armId,
+      model_name: pullResult.arm.modelName,
+      model_provider: pullResult.arm.modelProvider
+    };
+    if (options.queryText != null) {
+      event.query_text = options.queryText;
+    }
+    if (options.response != null) {
+      event.response = typeof options.response === "string" ? { response: options.response } : options.response;
+    }
+    if (reward != null) {
+      event.early_reward = reward;
+    }
+    if (options.cost != null) {
+      event.cost = options.cost;
+    }
+    if (latency != null) {
+      event.latency = latency;
+    }
+    if (options.inputTokens != null) {
+      event.input_tokens = options.inputTokens;
+    }
+    if (options.outputTokens != null) {
+      event.output_tokens = options.outputTokens;
+    }
+    if (options.segment != null) {
+      event.segment = options.segment;
+    }
+    if (options.failed) {
+      event.run_error = true;
+    }
+    this.store.push(event);
+    this.flushPending().catch(() => {
+    });
+  }
+  /**
+   * Send a human grade for an existing event. Async (HTTP).
+   */
+  async grade(eventId, grade2) {
+    this.ensureConnected();
+    await this.http.submitGrade(eventId, grade2);
+  }
+  /**
+   * Explicit state refresh from cloud.
+   */
+  async sync() {
+    this.ensureConnected();
+    const data = await this.http.heartbeat();
+    const prevBandits = new Map(this.bandits);
+    const prevEngines = new Map(this.engines);
+    try {
+      this.applySync(data);
+    } catch (err) {
+      this.bandits = prevBandits;
+      this.engines = prevEngines;
+      console.warn("[bandito] Sync response malformed \u2014 keeping last-known-good state", err);
+    }
+  }
+  /**
+   * Shut down: clear interval, flush remaining events, close connections.
+   */
+  async close() {
+    if (this.flushInterval) {
+      clearInterval(this.flushInterval);
+      this.flushInterval = null;
+    }
+    if (this.store && this.http) {
+      await this.flushPending();
+    }
+    this.store?.close();
+    this.store = null;
+    this.http = null;
+    this.engines.clear();
+    this.bandits.clear();
+    this.connected = false;
+  }
+  // --- Internal ---
+  ensureConnected() {
+    if (!this.connected) {
+      throw new Error("Not connected \u2014 call connect() first");
+    }
+  }
+  applySync(data) {
+    const banditsData = data.bandits ?? [];
+    const newBandits = /* @__PURE__ */ new Map();
+    const newEngines = /* @__PURE__ */ new Map();
+    for (const b of banditsData) {
+      const arms = b.arms ?? [];
+      if (arms.length === 0) continue;
+      const activeArms = arms.filter((a) => a.is_active).map((a) => createArm(a));
+      const name = b.name;
+      const cache = {
+        banditId: Number(b.bandit_id),
+        name,
+        arms: activeArms,
+        armWire: arms,
+        optimizationMode: b.optimization_mode ?? "base",
+        avgLatencyLastN: b.avg_latency_last_n,
+        budget: b.budget,
+        totalCost: b.total_cost
+      };
+      newBandits.set(name, cache);
+      const existingEngine = this.engines.get(name);
+      const banditJson = JSON.stringify(b);
+      if (existingEngine) {
+        updateEngine(existingEngine, banditJson);
+        newEngines.set(name, existingEngine);
+      } else {
+        newEngines.set(name, createEngine(banditJson));
+      }
+    }
+    this.bandits = newBandits;
+    this.engines = newEngines;
+  }
+  async flushPending() {
+    if (this.flushInProgress || !this.store || !this.http) return;
+    this.flushInProgress = true;
+    try {
+      const pending = this.store.pending();
+      if (pending.length === 0) return;
+      const alive = this.deadUuids.size > 0 ? pending.filter((e) => !this.deadUuids.has(e.local_event_uuid)) : pending;
+      if (alive.length === 0) return;
+      const payload = prepareCloudPayload(
+        alive,
+        this.dataStorage !== "local"
+      );
+      const result = await this.http.ingestEvents(payload);
+      const errors = result.errors ?? [];
+      const erroredUuids = new Set(
+        errors.map((e) => e.local_event_uuid).filter(Boolean)
+      );
+      for (const uid of erroredUuids) {
+        const count = (this.retryCounts.get(uid) ?? 0) + 1;
+        this.retryCounts.set(uid, count);
+        if (count >= MAX_EVENT_RETRIES) {
+          this.deadUuids.add(uid);
+        }
+      }
+      const flushedUuids = alive.map((e) => e.local_event_uuid).filter((uid) => !erroredUuids.has(uid));
+      if (flushedUuids.length > 0 && this.store) {
+        this.store.markFlushed(flushedUuids);
+      }
+    } catch (err) {
+      console.warn("[bandito] Event flush failed \u2014 will retry", err);
+    } finally {
+      this.flushInProgress = false;
+    }
+  }
+};
+
+// src/index.ts
+var _client = null;
+function getClient() {
+  if (!_client) {
+    throw new Error("Not connected \u2014 call connect() first");
+  }
+  return _client;
+}
+async function connect(options = {}) {
+  if (_client) {
+    await _client.close();
+  }
+  _client = new BanditoClient(options);
+  await _client.connect();
+}
+function pull(banditName, options) {
+  return getClient().pull(banditName, options);
+}
+function update(pullResult, options) {
+  getClient().update(pullResult, options);
+}
+async function grade(eventId, gradeValue) {
+  await getClient().grade(eventId, gradeValue);
+}
+async function sync() {
+  await getClient().sync();
+}
+async function close() {
+  if (_client) {
+    await _client.close();
+    _client = null;
+  }
+}
+export {
+  BanditoClient,
+  close,
+  connect,
+  grade,
+  pull,
+  sync,
+  update
+};
+//# sourceMappingURL=index.mjs.map