@databricks/appkit 0.42.0 → 0.43.0

package/dist/agents/databricks.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"databricks.d.ts","names":[],"sources":["../../src/agents/databricks.ts"],"mappings":";;;;;;;AAMgB;;;KAgEX,UAAA,IACH,IAAA,EAAM,MAAA,mBACN,MAAA,GAAS,WAAA,KACN,OAAA,CAAQ,cAAA,CAAe,UAAA;;;;;;UAOlB,sBAAA;EACR,WAAA;EACA,YAAA,QAAoB,OAAA,CAAQ,MAAA;EAC5B,QAAA;EACA,SAAA;EAXG;EAaH,eAAA;EAb0B;EAe1B,kBAAA;EAfoC;EAiBpC,qBAAA;AAAA;;;;;;;UASQ,wBAAA;EACR,UAAA,EAAY,UAAA;EACZ,QAAA;EACA,SAAA;EACA,eAAA;EACA,kBAAA;EACA,qBAAA;AAAA;AAAA,KAGG,wBAAA,GACD,sBAAA,GACA,wBAAA;;;;;;;;UAeM,mBAAA;EACR,SAAA;IACE,OAAA,CAAQ,OAAA,EAAS,MAAA,oBAA0B,OAAA;EAAA;AAAA;AAAA,UAIrC,sBAAA;EACR,eAAA,EAAiB,mBAAA;EACjB,YAAA;EACA,QAAA;EACA,SAAA;EACA,eAAA;EACA,kBAAA;EACA,qBAAA;AAAA;AAAA,UAGQ,mBAAA;EACR,QAAA;EACA,SAAA;EACA,eAAA,GAAkB,mBAAA;EAClB,eAAA;EACA,kBAAA;EACA,qBAAA;AAAA;;;;;;;;;;;;;AATqB;;;;;;;;;;;;;AAiGvB;;;;;;;;;;;;;;;;;;cAAa,iBAAA,YAA6B,YAAA;EAAA,QAChC,UAAA;EAAA,QACA,QAAA;EAAA,QACA,SAAA;EAAA,QACA,eAAA;EAAA,QACA,kBAAA;EAAA,QACA,qBAAA;cAEI,OAAA,EAAS,wBAAA;EAAT;;;;;;;;EAAA,OA+CC,mBAAA,CACX,OAAA,EAAS,sBAAA,GACR,OAAA,CAAQ,iBAAA;EAmDT;;;;;;;;;;;;;;;;;AA6ZJ;;;EA7ZI,OAFW,gBAAA,CACX,YAAA,WACA,OAAA,GAAU,mBAAA,GACT,OAAA,CAAQ,iBAAA;EA+BJ,GAAA,CACL,KAAA,EAAO,UAAA,EACP,OAAA,EAAS,eAAA,GACR,cAAA,CAAe,UAAA;EA4XjB;EAAA,QArUc,iBAAA;EAAA,QA8CA,gBAAA;EAAA,QAuKA,gBAAA;EAgHY;;;;;EAAA,QA5EnB,aAAA;EAAA,QAgDA,UAAA;AAAA;;;;;;;;iBA0BM,kBAAA,CACd,IAAA,WACC,KAAA;EAAQ,IAAA;EAAc,IAAA;AAAA"}
+{"version":3,"file":"databricks.d.ts","names":[],"sources":["../../src/agents/databricks.ts"],"mappings":";;;;;;;AAMgB;;;KAiEX,UAAA,IACH,IAAA,EAAM,MAAA,mBACN,MAAA,GAAS,WAAA,KACN,OAAA,CAAQ,cAAA,CAAe,UAAA;;;;;;UAOlB,sBAAA;EACR,WAAA;EACA,YAAA,QAAoB,OAAA,CAAQ,MAAA;EAC5B,QAAA;EACA,SAAA;EAXG;EAaH,eAAA;EAb0B;EAe1B,kBAAA;EAfoC;EAiBpC,qBAAA;AAAA;;;;;;;UASQ,wBAAA;EACR,UAAA,EAAY,UAAA;EACZ,QAAA;EACA,SAAA;EACA,eAAA;EACA,kBAAA;EACA,qBAAA;AAAA;AAAA,KAGG,wBAAA,GACD,sBAAA,GACA,wBAAA;;;;;;;;UAeM,mBAAA;EACR,SAAA;IACE,OAAA,CAAQ,OAAA,EAAS,MAAA,oBAA0B,OAAA;EAAA;AAAA;AAAA,UAIrC,sBAAA;EACR,eAAA,EAAiB,mBAAA;EACjB,YAAA;EACA,QAAA;EACA,SAAA;EACA,eAAA;EACA,kBAAA;EACA,qBAAA;AAAA;AAAA,UAGQ,mBAAA;EACR,QAAA;EACA,SAAA;EACA,eAAA,GAAkB,mBAAA;EAClB,eAAA;EACA,kBAAA;EACA,qBAAA;AAAA;;;;;;;;;;;;;AATqB;;;;;;;;;;;;;AAiGvB;;;;;;;;;;;;;;;;;;cAAa,iBAAA,YAA6B,YAAA;EAAA,QAChC,UAAA;EAAA,QACA,QAAA;EAAA,QACA,SAAA;EAAA,QACA,eAAA;EAAA,QACA,kBAAA;EAAA,QACA,qBAAA;cAEI,OAAA,EAAS,wBAAA;EAAT;;;;;;;;EAAA,OAgDC,mBAAA,CACX,OAAA,EAAS,sBAAA,GACR,OAAA,CAAQ,iBAAA;EAmDT;;;;;;;;;;;;;;;;;AA8ZJ;;;EA9ZI,OAFW,gBAAA,CACX,YAAA,WACA,OAAA,GAAU,mBAAA,GACT,OAAA,CAAQ,iBAAA;EAgCJ,GAAA,CACL,KAAA,EAAO,UAAA,EACP,OAAA,EAAS,eAAA,GACR,cAAA,CAAe,UAAA;EA4XjB;EAAA,QArUc,iBAAA;EAAA,QA8CA,gBAAA;EAAA,QAuKA,gBAAA;EAgHY;;;;;EAAA,QA5EnB,aAAA;EAAA,QAgDA,UAAA;AAAA;;;;;;;;iBA0BM,kBAAA,CACd,IAAA,WACC,KAAA;EAAQ,IAAA;EAAc,IAAA;AAAA"}

package/dist/agents/databricks.js

@@ -1,3 +1,4 @@
+import { APPKIT_USER_AGENT, getClientOptions } from "../context/client-options.js";
 import { stream } from "../connectors/serving/client.js";
 
 //#region src/agents/databricks.ts
@@ -105,6 +106,7 @@ var DatabricksAdapter = class DatabricksAdapter {
 				const response = await fetch(endpointUrl, {
 					method: "POST",
 					headers: {
+						"User-Agent": APPKIT_USER_AGENT,
 						"Content-Type": "application/json",
 						...authHeaders
 					},
@@ -163,7 +165,7 @@ var DatabricksAdapter = class DatabricksAdapter {
 		const resolvedEndpoint = endpointName ?? process.env.DATABRICKS_SERVING_ENDPOINT_NAME;
 		if (!resolvedEndpoint) throw new Error("No endpoint name provided and DATABRICKS_SERVING_ENDPOINT_NAME env var is not set. Pass an endpoint name or set DATABRICKS_SERVING_ENDPOINT_NAME.");
 		let workspaceClient = options?.workspaceClient;
-		if (!workspaceClient) workspaceClient = new (await (import("@databricks/sdk-experimental"))).WorkspaceClient({});
+		if (!workspaceClient) workspaceClient = new (await (import("@databricks/sdk-experimental"))).WorkspaceClient({}, getClientOptions());
 		return DatabricksAdapter.fromServingEndpoint({
 			workspaceClient,
 			endpointName: resolvedEndpoint,

package/dist/agents/databricks.js.map

@@ -1 +1 @@
-{"version":3,"file":"databricks.js","names":["servingStream"],"sources":["../../src/agents/databricks.ts"],"sourcesContent":["import type {\n  AgentAdapter,\n  AgentEvent,\n  AgentInput,\n  AgentRunContext,\n  AgentToolDefinition,\n} from \"shared\";\nimport { stream as servingStream } from \"../connectors/serving/client\";\n\n/** Default cap for a single incomplete SSE line tail (DoS guard). */\nconst DEFAULT_MAX_SSE_LINE_CHARS = 1024 * 1024;\n\n/** Default cap for accumulated assistant text from `delta.content`. */\nconst DEFAULT_MAX_STREAM_TEXT_CHARS = 4 * 1024 * 1024;\n\n/** Default cap for accumulated JSON arguments per streamed tool call index. */\nconst DEFAULT_MAX_TOOL_ARGUMENT_CHARS = 2 * 1024 * 1024;\n\n/** Cap text length before running Python-style tool-call regex (ReDoS guard). */\nconst PYTHON_STYLE_TOOL_PARSE_MAX_INPUT = 64 * 1024;\n\n/** Fallback HTTP timeout when the raw fetch adapter path receives no AbortSignal from the runner. */\nconst RAW_FETCH_DEFAULT_TIMEOUT_MS = 120_000;\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return typeof value === \"object\" && value !== null;\n}\n\nfunction extractLlamaToolJsonSlice(text: string): string | undefined {\n  const start = text.indexOf(\"[{\");\n  if (start < 0) return undefined;\n  const endBracket = text.lastIndexOf(\"}]\");\n  if (endBracket < start) return undefined;\n  return text.slice(start, endBracket + 2);\n}\n\n/** OpenAI SSE payload: `{ choices: [{ delta }] }`. */\nfunction openAiChoicesDelta(parsed: unknown): unknown {\n  if (!isRecord(parsed)) return undefined;\n  const choices = parsed.choices;\n  if (!Array.isArray(choices) || choices.length < 1) return undefined;\n  const first = choices[0];\n  if (!isRecord(first)) return undefined;\n  return first.delta;\n}\n\nfunction isStreamingDeltaToolCall(value: unknown): value is DeltaToolCall {\n  if (!isRecord(value)) return false;\n  return typeof value.index === \"number\";\n}\n\nfunction throwIfExceedsStreamLimit(\n  label: string,\n  currentLength: number,\n  chunk: string,\n  max: number,\n): void {\n  if (currentLength + chunk.length > max) {\n    throw new Error(\n      `DatabricksAdapter: ${label} exceeds configured limit (${max} UTF-16 code units)`,\n    );\n  }\n}\n\n/**\n * Transport shim: given an OpenAI-compatible request body, returns the raw\n * SSE byte stream from the serving endpoint. Injected at construction time so\n * callers can swap in the workspace SDK (factory paths), a bare `fetch`\n * (the raw constructor), or a test fake.\n */\ntype StreamBody = (\n  body: Record<string, unknown>,\n  signal?: AbortSignal,\n) => Promise<ReadableStream<Uint8Array>>;\n\n/**\n * Escape-hatch options: provide an `endpointUrl` + `authenticate()` and the\n * adapter uses a bare `fetch()` to call it. Useful for tests and for pointing\n * the adapter at non-workspace endpoints (reverse proxies, mocks).\n */\ninterface RawFetchAdapterOptions {\n  endpointUrl: string;\n  authenticate: () => Promise<Record<string, string>>;\n  maxSteps?: number;\n  maxTokens?: number;\n  /** Max length of one SSE line (including an incomplete tail in the buffer). */\n  maxSseLineChars?: number;\n  /** Max total length of assistant `delta.content` across the stream. */\n  maxStreamTextChars?: number;\n  /** Max length of streamed `function.arguments` per tool call index. */\n  maxToolArgumentsChars?: number;\n}\n\n/**\n * Preferred options: caller provides the transport function directly.\n * The `fromServingEndpoint` / `fromModelServing` factories use this to route\n * through `connectors/serving/stream`, which centralises URL encoding, auth\n * via the SDK's `apiClient.request`, and any future retries/telemetry.\n */\ninterface StreamBodyAdapterOptions {\n  streamBody: StreamBody;\n  maxSteps?: number;\n  maxTokens?: number;\n  maxSseLineChars?: number;\n  maxStreamTextChars?: number;\n  maxToolArgumentsChars?: number;\n}\n\ntype DatabricksAdapterOptions =\n  | RawFetchAdapterOptions\n  | StreamBodyAdapterOptions;\n\nfunction isStreamBodyOptions(\n  o: DatabricksAdapterOptions,\n): o is StreamBodyAdapterOptions {\n  return \"streamBody\" in o;\n}\n\n/**\n * Duck-typed subset of the Databricks SDK `WorkspaceClient`. Callers of\n * `fromServingEndpoint` and `fromModelServing` pass a real `WorkspaceClient`,\n * but we only need the `apiClient.request` surface — so we declare the minimal\n * interface rather than importing the SDK type directly. This keeps the adapter\n * free of a hard compile-time dependency on `@databricks/sdk-experimental`.\n */\ninterface WorkspaceClientLike {\n  apiClient: {\n    request(options: Record<string, unknown>): Promise<unknown>;\n  };\n}\n\ninterface ServingEndpointOptions {\n  workspaceClient: WorkspaceClientLike;\n  endpointName: string;\n  maxSteps?: number;\n  maxTokens?: number;\n  maxSseLineChars?: number;\n  maxStreamTextChars?: number;\n  maxToolArgumentsChars?: number;\n}\n\ninterface ModelServingOptions {\n  maxSteps?: number;\n  maxTokens?: number;\n  workspaceClient?: WorkspaceClientLike;\n  maxSseLineChars?: number;\n  maxStreamTextChars?: number;\n  maxToolArgumentsChars?: number;\n}\n\ninterface OpenAIMessage {\n  role: \"system\" | \"user\" | \"assistant\" | \"tool\";\n  content: string | null;\n  tool_calls?: OpenAIToolCall[];\n  tool_call_id?: string;\n}\n\ninterface OpenAIToolCall {\n  id: string;\n  type: \"function\";\n  function: { name: string; arguments: string };\n  /**\n   * Opaque Vertex/Gemini \"thought signature\" blob the request must echo\n   * back verbatim on the next turn. Vertex's OpenAI-compat proxy emits\n   * this as `thoughtSignature` (camelCase) at the top level of the\n   * tool_call delta (verified against `gemini-3.1-flash-lite-preview`),\n   * and accepts the same spelling back on outbound. Non-Gemini endpoints\n   * (Claude on Databricks, external OpenAI-compat models, Llama, etc.)\n   * leave this undefined and the serializer omits the key.\n   * See https://docs.cloud.google.com/vertex-ai/generative-ai/docs/thought-signatures\n   */\n  thoughtSignature?: string;\n}\n\ninterface OpenAITool {\n  type: \"function\";\n  function: {\n    name: string;\n    description: string;\n    parameters: unknown;\n  };\n}\n\ninterface DeltaToolCall {\n  index: number;\n  id?: string;\n  type?: string;\n  function?: { name?: string; arguments?: string };\n  /** See {@link OpenAIToolCall.thoughtSignature}. */\n  thoughtSignature?: string;\n}\n\n/**\n * Adapter that talks directly to Databricks Model Serving `/invocations` endpoint.\n *\n * No dependency on the Vercel AI SDK or LangChain. Uses raw `fetch()` to POST\n * OpenAI-compatible payloads and parses the SSE stream itself. Calls\n * `authenticate()` per-request so tokens are always fresh.\n *\n * Handles both structured `tool_calls` responses and text-based tool call\n * fallback parsing for models that output tool calls as text.\n *\n * @example Using the factory (recommended)\n * ```ts\n * import { createApp, createAgent, agents } from \"@databricks/appkit\";\n * import { DatabricksAdapter } from \"@databricks/appkit/beta\";\n * import { WorkspaceClient } from \"@databricks/sdk-experimental\";\n *\n * const adapter = DatabricksAdapter.fromServingEndpoint({\n *   workspaceClient: new WorkspaceClient({}),\n *   endpointName: \"my-endpoint\",\n * });\n *\n * await createApp({\n *   plugins: [\n *     agents({\n *       agents: {\n *         assistant: createAgent({\n *           instructions: \"You are a helpful assistant.\",\n *           model: adapter,\n *         }),\n *       },\n *     }),\n *   ],\n * });\n * ```\n *\n * @example Using the raw constructor\n * ```ts\n * const adapter = new DatabricksAdapter({\n *   endpointUrl: \"https://host/serving-endpoints/my-endpoint/invocations\",\n *   authenticate: async () => ({ Authorization: `Bearer ${token}` }),\n * });\n * ```\n */\nexport class DatabricksAdapter implements AgentAdapter {\n  private streamBody: StreamBody;\n  private maxSteps: number;\n  private maxTokens: number;\n  private maxSseLineChars: number;\n  private maxStreamTextChars: number;\n  private maxToolArgumentsChars: number;\n\n  constructor(options: DatabricksAdapterOptions) {\n    this.maxSteps = options.maxSteps ?? 10;\n    this.maxTokens = options.maxTokens ?? 4096;\n    this.maxSseLineChars =\n      options.maxSseLineChars ?? DEFAULT_MAX_SSE_LINE_CHARS;\n    this.maxStreamTextChars =\n      options.maxStreamTextChars ?? DEFAULT_MAX_STREAM_TEXT_CHARS;\n    this.maxToolArgumentsChars =\n      options.maxToolArgumentsChars ?? DEFAULT_MAX_TOOL_ARGUMENT_CHARS;\n\n    if (isStreamBodyOptions(options)) {\n      this.streamBody = options.streamBody;\n    } else {\n      const { endpointUrl, authenticate } = options;\n      this.streamBody = async (body, signal) => {\n        const fetchSignal =\n          signal ?? AbortSignal.timeout(RAW_FETCH_DEFAULT_TIMEOUT_MS);\n        const authHeaders = await authenticate();\n        const response = await fetch(endpointUrl, {\n          method: \"POST\",\n          headers: {\n            \"Content-Type\": \"application/json\",\n            ...authHeaders,\n          },\n          body: JSON.stringify(body),\n          signal: fetchSignal,\n        });\n        if (!response.ok) {\n          const errorText = await response.text().catch(() => \"Unknown error\");\n          throw new Error(\n            `Databricks API error (${response.status}): ${errorText}`,\n          );\n        }\n        if (!response.body) throw new Error(\"No response body\");\n        return response.body;\n      };\n    }\n  }\n\n  /**\n   * Creates a DatabricksAdapter for a Databricks Model Serving endpoint.\n   *\n   * Routes through the shared `connectors/serving/stream` helper, which\n   * delegates to the SDK's `apiClient.request({ raw: true })`. That gives the\n   * adapter centralised URL encoding + authentication with the rest of the\n   * serving surface — no bespoke `fetch()` + `authenticate()` plumbing.\n   */\n  static async fromServingEndpoint(\n    options: ServingEndpointOptions,\n  ): Promise<DatabricksAdapter> {\n    const {\n      workspaceClient,\n      endpointName,\n      maxSteps,\n      maxTokens,\n      maxSseLineChars,\n      maxStreamTextChars,\n      maxToolArgumentsChars,\n    } = options;\n    return new DatabricksAdapter({\n      streamBody: (body, signal) =>\n        // Cast through the structural shape: the connector types\n        // `workspaceClient` as the SDK's concrete `WorkspaceClient`, but we\n        // only need `apiClient.request`.\n        servingStream(\n          workspaceClient as unknown as Parameters<typeof servingStream>[0],\n          endpointName,\n          body,\n          signal,\n        ),\n      maxSteps,\n      maxTokens,\n      maxSseLineChars,\n      maxStreamTextChars,\n      maxToolArgumentsChars,\n    });\n  }\n\n  /**\n   * Creates a DatabricksAdapter from a Model Serving endpoint name.\n   * Auto-creates a WorkspaceClient internally. Reads the endpoint name\n   * from the argument or the `DATABRICKS_SERVING_ENDPOINT_NAME` env var.\n   *\n   * @example\n   * ```ts\n   * // Reads endpoint from DATABRICKS_SERVING_ENDPOINT_NAME env var\n   * const adapter = await DatabricksAdapter.fromModelServing();\n   *\n   * // Explicit endpoint\n   * const adapter = await DatabricksAdapter.fromModelServing(\"my-endpoint\");\n   *\n   * // With options\n   * const adapter = await DatabricksAdapter.fromModelServing(\"my-endpoint\", {\n   *   maxSteps: 5,\n   *   maxTokens: 2048,\n   * });\n   * ```\n   */\n  static async fromModelServing(\n    endpointName?: string,\n    options?: ModelServingOptions,\n  ): Promise<DatabricksAdapter> {\n    const resolvedEndpoint =\n      endpointName ?? process.env.DATABRICKS_SERVING_ENDPOINT_NAME;\n\n    if (!resolvedEndpoint) {\n      throw new Error(\n        \"No endpoint name provided and DATABRICKS_SERVING_ENDPOINT_NAME env var is not set. \" +\n          \"Pass an endpoint name or set DATABRICKS_SERVING_ENDPOINT_NAME.\",\n      );\n    }\n\n    let workspaceClient: WorkspaceClientLike | undefined =\n      options?.workspaceClient;\n    if (!workspaceClient) {\n      const sdk = await import(\"@databricks/sdk-experimental\");\n      workspaceClient = new sdk.WorkspaceClient(\n        {},\n      ) as unknown as WorkspaceClientLike;\n    }\n\n    return DatabricksAdapter.fromServingEndpoint({\n      workspaceClient,\n      endpointName: resolvedEndpoint,\n      maxSteps: options?.maxSteps,\n      maxTokens: options?.maxTokens,\n      maxSseLineChars: options?.maxSseLineChars,\n      maxStreamTextChars: options?.maxStreamTextChars,\n      maxToolArgumentsChars: options?.maxToolArgumentsChars,\n    });\n  }\n\n  async *run(\n    input: AgentInput,\n    context: AgentRunContext,\n  ): AsyncGenerator<AgentEvent, void, unknown> {\n    // Databricks API requires tool names to match [a-zA-Z0-9_-].\n    // Our tool names use dots (e.g. \"analytics.query\"), so we swap dots\n    // for double-underscores in the wire format and map back on receipt.\n    const nameToWire = new Map<string, string>();\n    const wireToName = new Map<string, string>();\n    for (const tool of input.tools) {\n      const wire = tool.name.replace(/\\./g, \"__\");\n      if (wireToName.has(wire) && wireToName.get(wire) !== tool.name) {\n        throw new Error(\n          `Tool name collision: '${tool.name}' and '${wireToName.get(wire)}' both map to wire name '${wire}'`,\n        );\n      }\n      nameToWire.set(tool.name, wire);\n      wireToName.set(wire, tool.name);\n    }\n\n    const tools = this.buildTools(input.tools, nameToWire);\n    const messages = this.buildMessages(input.messages, nameToWire);\n\n    yield { type: \"status\", status: \"running\" };\n\n    for (let step = 0; step < this.maxSteps; step++) {\n      if (context.signal?.aborted) break;\n\n      const { text, toolCalls } = yield* this.streamCompletion(\n        messages,\n        tools,\n        context,\n      );\n\n      if (toolCalls.length === 0) {\n        const parsed = parseTextToolCalls(text);\n        if (parsed.length > 0) {\n          yield* this.executeToolCalls(parsed, messages, context, nameToWire);\n          continue;\n        }\n        break;\n      }\n\n      messages.push({\n        role: \"assistant\",\n        content: text || null,\n        tool_calls: toolCalls,\n      });\n\n      for (const tc of toolCalls) {\n        const wireName = tc.function.name;\n        const originalName = wireToName.get(wireName) ?? wireName;\n        yield* this.executeSingleTool(tc, originalName, messages, context);\n      }\n    }\n  }\n\n  /** Parse wire arguments, emit tool_call / tool_result, append tool messages. */\n  private async *executeSingleTool(\n    tc: OpenAIToolCall,\n    originalName: string,\n    messages: OpenAIMessage[],\n    context: AgentRunContext,\n  ): AsyncGenerator<AgentEvent, void, unknown> {\n    let args: unknown;\n    try {\n      args = JSON.parse(tc.function.arguments);\n    } catch {\n      args = {};\n    }\n\n    yield { type: \"tool_call\", callId: tc.id, name: originalName, args };\n\n    try {\n      const result = await context.executeTool(originalName, args);\n      const resultStr =\n        typeof result === \"string\" ? result : JSON.stringify(result);\n\n      yield { type: \"tool_result\", callId: tc.id, result };\n\n      messages.push({\n        role: \"tool\",\n        content: resultStr,\n        tool_call_id: tc.id,\n      });\n    } catch (error) {\n      const errMsg =\n        error instanceof Error ? error.message : \"Tool execution failed\";\n\n      yield {\n        type: \"tool_result\",\n        callId: tc.id,\n        result: null,\n        error: errMsg,\n      };\n\n      messages.push({\n        role: \"tool\",\n        content: JSON.stringify({ error: errMsg }),\n        tool_call_id: tc.id,\n      });\n    }\n  }\n\n  private async *streamCompletion(\n    messages: OpenAIMessage[],\n    tools: OpenAITool[],\n    context: AgentRunContext,\n  ): AsyncGenerator<\n    AgentEvent,\n    { text: string; toolCalls: OpenAIToolCall[] },\n    unknown\n  > {\n    const body: Record<string, unknown> = {\n      messages,\n      stream: true,\n      max_tokens: this.maxTokens,\n    };\n\n    if (tools.length > 0) {\n      body.tools = tools;\n    }\n\n    let responseBody: ReadableStream<Uint8Array>;\n    try {\n      responseBody = await this.streamBody(body, context.signal);\n    } catch (err) {\n      const msg = err instanceof Error ? err.message : \"Stream request failed\";\n      yield { type: \"status\", status: \"error\", error: msg };\n      throw err;\n    }\n\n    const reader = responseBody.getReader();\n\n    const decoder = new TextDecoder();\n    let buffer = \"\";\n    let fullText = \"\";\n    const toolCallAccumulator = new Map<\n      number,\n      {\n        id: string;\n        name: string;\n        arguments: string;\n        thoughtSignature?: string;\n      }\n    >();\n\n    try {\n      while (true) {\n        if (context.signal?.aborted) break;\n\n        const { done, value } = await reader.read();\n        if (done) break;\n\n        buffer += decoder.decode(value, { stream: true });\n        const lines = buffer.split(\"\\n\");\n        buffer = lines.pop() ?? \"\";\n\n        if (buffer.length > this.maxSseLineChars) {\n          throw new Error(\n            `DatabricksAdapter: SSE line buffer exceeds configured limit (${this.maxSseLineChars} UTF-16 code units)`,\n          );\n        }\n\n        for (const line of lines) {\n          if (line.length > this.maxSseLineChars) {\n            throw new Error(\n              `DatabricksAdapter: SSE line exceeds configured limit (${this.maxSseLineChars} UTF-16 code units)`,\n            );\n          }\n\n          const trimmed = line.trim();\n          if (!trimmed.startsWith(\"data: \")) continue;\n          const data = trimmed.slice(6);\n          if (data === \"[DONE]\") continue;\n\n          let parsed: unknown;\n          try {\n            parsed = JSON.parse(data);\n          } catch (parseErr) {\n            console.debug(\n              \"[DatabricksAdapter] malformed SSE data line JSON\",\n              { line: `${data.slice(0, 256)}${data.length > 256 ? \"…\" : \"\"}` },\n              parseErr,\n            );\n            continue;\n          }\n\n          const deltaUnknown = openAiChoicesDelta(parsed);\n          if (!isRecord(deltaUnknown)) continue;\n\n          if (typeof deltaUnknown.content === \"string\") {\n            const content = deltaUnknown.content;\n            throwIfExceedsStreamLimit(\n              \"streamed assistant text\",\n              fullText.length,\n              content,\n              this.maxStreamTextChars,\n            );\n            fullText += content;\n            yield { type: \"message_delta\" as const, content };\n          }\n\n          const toolCallsRaw = deltaUnknown.tool_calls;\n          if (!Array.isArray(toolCallsRaw)) continue;\n\n          for (const tc of toolCallsRaw) {\n            if (!isStreamingDeltaToolCall(tc)) continue;\n            const sig = tc.thoughtSignature;\n            const existing = toolCallAccumulator.get(tc.index);\n            if (existing) {\n              if (tc.function?.arguments) {\n                throwIfExceedsStreamLimit(\n                  \"tool call arguments\",\n                  existing.arguments.length,\n                  tc.function.arguments,\n                  this.maxToolArgumentsChars,\n                );\n                existing.arguments += tc.function.arguments;\n              }\n              if (sig && !existing.thoughtSignature) {\n                existing.thoughtSignature = sig;\n              }\n            } else {\n              const initial = tc.function?.arguments ?? \"\";\n              if (initial.length > this.maxToolArgumentsChars) {\n                throw new Error(\n                  `DatabricksAdapter: tool call arguments exceed configured limit (${this.maxToolArgumentsChars} UTF-16 code units)`,\n                );\n              }\n              toolCallAccumulator.set(tc.index, {\n                id: tc.id ?? `call_${tc.index}`,\n                name: tc.function?.name ?? \"\",\n                arguments: initial,\n                ...(sig ? { thoughtSignature: sig } : {}),\n              });\n            }\n          }\n        }\n      }\n    } finally {\n      try {\n        await reader.cancel();\n      } catch (cancelErr) {\n        console.debug(\n          \"[DatabricksAdapter] reader.cancel() failed during teardown\",\n          cancelErr,\n        );\n      }\n      try {\n        reader.releaseLock();\n      } catch (unlockErr) {\n        console.debug(\n          \"[DatabricksAdapter] reader.releaseLock() failed during teardown\",\n          unlockErr,\n        );\n      }\n    }\n\n    const toolCalls: OpenAIToolCall[] = Array.from(\n      toolCallAccumulator.values(),\n    ).map((tc) => ({\n      id: tc.id,\n      type: \"function\" as const,\n      function: { name: tc.name, arguments: tc.arguments || \"{}\" },\n      ...(tc.thoughtSignature ? { thoughtSignature: tc.thoughtSignature } : {}),\n    }));\n\n    return { text: fullText, toolCalls };\n  }\n\n  private async *executeToolCalls(\n    calls: Array<{ name: string; args: unknown }>,\n    messages: OpenAIMessage[],\n    context: AgentRunContext,\n    nameToWire: Map<string, string>,\n  ): AsyncGenerator<AgentEvent, void, unknown> {\n    const wireToolName = (name: string) =>\n      nameToWire.get(name) ?? name.replace(/\\./g, \"__\");\n\n    const toolCallObjs: OpenAIToolCall[] = calls.map((c, i) => ({\n      id: `text_call_${i}`,\n      type: \"function\" as const,\n      function: {\n        name: wireToolName(c.name),\n        arguments: JSON.stringify(c.args),\n      },\n    }));\n\n    messages.push({\n      role: \"assistant\",\n      content: null,\n      tool_calls: toolCallObjs,\n    });\n\n    for (let i = 0; i < toolCallObjs.length; i++) {\n      const tc = toolCallObjs[i];\n      const originalName = calls[i]?.name ?? tc.function.name;\n      yield* this.executeSingleTool(tc, originalName, messages, context);\n    }\n  }\n\n  /**\n   * Maps AppKit {@link AgentInput} messages into OpenAI-compatible wire messages.\n   * Preserves multi-turn tool state (`toolCalls` → `tool_calls`, `toolCallId` →\n   * `tool_call_id`) so resumed threads and hydrated history reach the model.\n   */\n  private buildMessages(\n    messages: AgentInput[\"messages\"],\n    nameToWire: Map<string, string>,\n  ): OpenAIMessage[] {\n    const wireToolName = (name: string) =>\n      nameToWire.get(name) ?? name.replace(/\\./g, \"__\");\n\n    return messages.map((m) => {\n      let content: string | null = m.content;\n      if (\n        m.role === \"assistant\" &&\n        m.toolCalls &&\n        m.toolCalls.length > 0 &&\n        (!m.content || m.content.trim() === \"\")\n      ) {\n        content = null;\n      }\n\n      const out: OpenAIMessage = {\n        role: m.role as OpenAIMessage[\"role\"],\n        content,\n      };\n\n      if (m.toolCallId) {\n        out.tool_call_id = m.toolCallId;\n      }\n\n      if (m.toolCalls && m.toolCalls.length > 0) {\n        out.tool_calls = m.toolCalls.map((tc) => ({\n          id: tc.id,\n          type: \"function\" as const,\n          function: {\n            name: wireToolName(tc.name),\n            arguments:\n              typeof tc.args === \"string\"\n                ? tc.args\n                : JSON.stringify(tc.args ?? {}),\n          },\n          ...(tc.thoughtSignature\n            ? { thoughtSignature: tc.thoughtSignature }\n            : {}),\n        }));\n      }\n\n      return out;\n    });\n  }\n\n  private buildTools(\n    definitions: AgentToolDefinition[],\n    nameToWire: Map<string, string>,\n  ): OpenAITool[] {\n    return definitions.map((def) => ({\n      type: \"function\" as const,\n      function: {\n        name: nameToWire.get(def.name) ?? def.name,\n        description: def.description,\n        parameters: def.parameters,\n      },\n    }));\n  }\n}\n\n// ---------------------------------------------------------------------------\n// Text-based tool call parsing (fallback)\n// ---------------------------------------------------------------------------\n\n/**\n * Parses text-based tool calls from model output.\n *\n * Handles two formats:\n * 1. Llama native: `[{\"name\": \"tool_name\", \"parameters\": {\"arg\": \"val\"}}]`\n * 2. Python-style: `[tool_name(arg1='val1', arg2='val2')]`\n */\nexport function parseTextToolCalls(\n  text: string,\n): Array<{ name: string; args: unknown }> {\n  const trimmed = text.trim();\n\n  const jsonResult = tryParseLlamaJsonToolCalls(trimmed);\n  if (jsonResult.length > 0) return jsonResult;\n\n  const pyResult = tryParsePythonStyleToolCalls(trimmed);\n  if (pyResult.length > 0) return pyResult;\n\n  return [];\n}\n\nfunction isLlamaToolJsonItem(value: unknown): value is Record<\n  string,\n  unknown\n> & {\n  name: string;\n} {\n  if (!isRecord(value)) return false;\n  return typeof value.name === \"string\";\n}\n\nfunction tryParseLlamaJsonToolCalls(\n  text: string,\n): Array<{ name: string; args: unknown }> {\n  const slice = extractLlamaToolJsonSlice(text);\n  if (!slice) return [];\n\n  try {\n    const parsed: unknown = JSON.parse(slice);\n    if (!Array.isArray(parsed)) return [];\n\n    return parsed.filter(isLlamaToolJsonItem).map((item) => ({\n      name: item.name,\n      args: item.parameters ?? item.arguments ?? item.args ?? {},\n    }));\n  } catch {\n    return [];\n  }\n}\n\nfunction tryParsePythonStyleToolCalls(\n  text: string,\n): Array<{ name: string; args: unknown }> {\n  if (text.length > PYTHON_STYLE_TOOL_PARSE_MAX_INPUT) {\n    return [];\n  }\n\n  const pattern = /\\[?([a-zA-Z_][\\w.]*)\\(([^)]*)\\)\\]?/g;\n  const results: Array<{ name: string; args: unknown }> = [];\n\n  for (const match of text.matchAll(pattern)) {\n    const name = match[1];\n    const argsStr = match[2];\n\n    const args: Record<string, unknown> = {};\n    const argPattern = /(\\w+)\\s*=\\s*(?:'([^']*)'|\"([^\"]*)\"|(\\S+))/g;\n    for (const argMatch of argsStr.matchAll(argPattern)) {\n      const key = argMatch[1];\n      const value = argMatch[2] ?? argMatch[3] ?? argMatch[4];\n      args[key] = value;\n    }\n\n    results.push({ name, args });\n  }\n\n  return results;\n}\n"],"mappings":";;;;AAUA,MAAM,6BAA6B,OAAO;;AAG1C,MAAM,gCAAgC,IAAI,OAAO;;AAGjD,MAAM,kCAAkC,IAAI,OAAO;;AAGnD,MAAM,oCAAoC,KAAK;;AAG/C,MAAM,+BAA+B;AAErC,SAAS,SAAS,OAAkD;AAClE,QAAO,OAAO,UAAU,YAAY,UAAU;;AAGhD,SAAS,0BAA0B,MAAkC;CACnE,MAAM,QAAQ,KAAK,QAAQ,KAAK;AAChC,KAAI,QAAQ,EAAG,QAAO;CACtB,MAAM,aAAa,KAAK,YAAY,KAAK;AACzC,KAAI,aAAa,MAAO,QAAO;AAC/B,QAAO,KAAK,MAAM,OAAO,aAAa,EAAE;;;AAI1C,SAAS,mBAAmB,QAA0B;AACpD,KAAI,CAAC,SAAS,OAAO,CAAE,QAAO;CAC9B,MAAM,UAAU,OAAO;AACvB,KAAI,CAAC,MAAM,QAAQ,QAAQ,IAAI,QAAQ,SAAS,EAAG,QAAO;CAC1D,MAAM,QAAQ,QAAQ;AACtB,KAAI,CAAC,SAAS,MAAM,CAAE,QAAO;AAC7B,QAAO,MAAM;;AAGf,SAAS,yBAAyB,OAAwC;AACxE,KAAI,CAAC,SAAS,MAAM,CAAE,QAAO;AAC7B,QAAO,OAAO,MAAM,UAAU;;AAGhC,SAAS,0BACP,OACA,eACA,OACA,KACM;AACN,KAAI,gBAAgB,MAAM,SAAS,IACjC,OAAM,IAAI,MACR,sBAAsB,MAAM,6BAA6B,IAAI,qBAC9D;;AAoDL,SAAS,oBACP,GAC+B;AAC/B,QAAO,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwHzB,IAAa,oBAAb,MAAa,kBAA0C;CACrD,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,YAAY,SAAmC;AAC7C,OAAK,WAAW,QAAQ,YAAY;AACpC,OAAK,YAAY,QAAQ,aAAa;AACtC,OAAK,kBACH,QAAQ,mBAAmB;AAC7B,OAAK,qBACH,QAAQ,sBAAsB;AAChC,OAAK,wBACH,QAAQ,yBAAyB;AAEnC,MAAI,oBAAoB,QAAQ,CAC9B,MAAK,aAAa,QAAQ;OACrB;GACL,MAAM,EAAE,aAAa,iBAAiB;AACtC,QAAK,aAAa,OAAO,MAAM,WAAW;IACxC,MAAM,cACJ,UAAU,YAAY,QAAQ,6BAA6B;IAC7D,MAAM,cAAc,MAAM,cAAc;IACxC,MAAM,WAAW,MAAM,MAAM,aAAa;KACxC,QAAQ;KACR,SAAS;MACP,gBAAgB;MAChB,GAAG;MACJ;KACD,MAAM,KAAK,UAAU,KAAK;KAC1B,QAAQ;KACT,CAAC;AACF,QAAI,CAAC,SAAS,IAAI;KAChB,MAAM,YAAY,MAAM,SAAS,MAAM,CAAC,YAAY,gBAAgB;AACpE,WAAM,IAAI,MACR,yBAAyB,SAAS,OAAO,KAAK,YAC/C;;AAEH,QAAI,CAAC,SAAS,KAAM,OAAM,IAAI,MAAM,mBAAmB;AACvD,WAAO,SAAS;;;;;;;;;;;;CAatB,aAAa,oBACX,SAC4B;EAC5B,MAAM,EACJ,iBACA,cACA,UACA,WACA,iBACA,oBACA,0BACE;AACJ,SAAO,IAAI,kBAAkB;GAC3B,aAAa,MAAM,WAIjBA,OACE,iBACA,cACA,MACA,OACD;GACH;GACA;GACA;GACA;GACA;GACD,CAAC;;;;;;;;;;;;;;;;;;;;;;CAuBJ,aAAa,iBACX,cACA,SAC4B;EAC5B,MAAM,mBACJ,gBAAgB,QAAQ,IAAI;AAE9B,MAAI,CAAC,iBACH,OAAM,IAAI,MACR,oJAED;EAGH,IAAI,kBACF,SAAS;AACX,MAAI,CAAC,gBAEH,mBAAkB,KADN,OAAM,OAAO,kCACC,gBACxB,EAAE,CACH;AAGH,SAAO,kBAAkB,oBAAoB;GAC3C;GACA,cAAc;GACd,UAAU,SAAS;GACnB,WAAW,SAAS;GACpB,iBAAiB,SAAS;GAC1B,oBAAoB,SAAS;GAC7B,uBAAuB,SAAS;GACjC,CAAC;;CAGJ,OAAO,IACL,OACA,SAC2C;EAI3C,MAAM,6BAAa,IAAI,KAAqB;EAC5C,MAAM,6BAAa,IAAI,KAAqB;AAC5C,OAAK,MAAM,QAAQ,MAAM,OAAO;GAC9B,MAAM,OAAO,KAAK,KAAK,QAAQ,OAAO,KAAK;AAC3C,OAAI,WAAW,IAAI,KAAK,IAAI,WAAW,IAAI,KAAK,KAAK,KAAK,KACxD,OAAM,IAAI,MACR,yBAAyB,KAAK,KAAK,SAAS,WAAW,IAAI,KAAK,CAAC,2BAA2B,KAAK,GAClG;AAEH,cAAW,IAAI,KAAK,MAAM,KAAK;AAC/B,cAAW,IAAI,MAAM,KAAK,KAAK;;EAGjC,MAAM,QAAQ,KAAK,WAAW,MAAM,OAAO,WAAW;EACtD,MAAM,WAAW,KAAK,cAAc,MAAM,UAAU,WAAW;AAE/D,QAAM;GAAE,MAAM;GAAU,QAAQ;GAAW;AAE3C,OAAK,IAAI,OAAO,GAAG,OAAO,KAAK,UAAU,QAAQ;AAC/C,OAAI,QAAQ,QAAQ,QAAS;GAE7B,MAAM,EAAE,MAAM,cAAc,OAAO,KAAK,iBACtC,UACA,OACA,QACD;AAED,OAAI,UAAU,WAAW,GAAG;IAC1B,MAAM,SAAS,mBAAmB,KAAK;AACvC,QAAI,OAAO,SAAS,GAAG;AACrB,YAAO,KAAK,iBAAiB,QAAQ,UAAU,SAAS,WAAW;AACnE;;AAEF;;AAGF,YAAS,KAAK;IACZ,MAAM;IACN,SAAS,QAAQ;IACjB,YAAY;IACb,CAAC;AAEF,QAAK,MAAM,MAAM,WAAW;IAC1B,MAAM,WAAW,GAAG,SAAS;IAC7B,MAAM,eAAe,WAAW,IAAI,SAAS,IAAI;AACjD,WAAO,KAAK,kBAAkB,IAAI,cAAc,UAAU,QAAQ;;;;;CAMxE,OAAe,kBACb,IACA,cACA,UACA,SAC2C;EAC3C,IAAI;AACJ,MAAI;AACF,UAAO,KAAK,MAAM,GAAG,SAAS,UAAU;UAClC;AACN,UAAO,EAAE;;AAGX,QAAM;GAAE,MAAM;GAAa,QAAQ,GAAG;GAAI,MAAM;GAAc;GAAM;AAEpE,MAAI;GACF,MAAM,SAAS,MAAM,QAAQ,YAAY,cAAc,KAAK;GAC5D,MAAM,YACJ,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,OAAO;AAE9D,SAAM;IAAE,MAAM;IAAe,QAAQ,GAAG;IAAI;IAAQ;AAEpD,YAAS,KAAK;IACZ,MAAM;IACN,SAAS;IACT,cAAc,GAAG;IAClB,CAAC;WACK,OAAO;GACd,MAAM,SACJ,iBAAiB,QAAQ,MAAM,UAAU;AAE3C,SAAM;IACJ,MAAM;IACN,QAAQ,GAAG;IACX,QAAQ;IACR,OAAO;IACR;AAED,YAAS,KAAK;IACZ,MAAM;IACN,SAAS,KAAK,UAAU,EAAE,OAAO,QAAQ,CAAC;IAC1C,cAAc,GAAG;IAClB,CAAC;;;CAIN,OAAe,iBACb,UACA,OACA,SAKA;EACA,MAAM,OAAgC;GACpC;GACA,QAAQ;GACR,YAAY,KAAK;GAClB;AAED,MAAI,MAAM,SAAS,EACjB,MAAK,QAAQ;EAGf,IAAI;AACJ,MAAI;AACF,kBAAe,MAAM,KAAK,WAAW,MAAM,QAAQ,OAAO;WACnD,KAAK;AAEZ,SAAM;IAAE,MAAM;IAAU,QAAQ;IAAS,OAD7B,eAAe,QAAQ,IAAI,UAAU;IACI;AACrD,SAAM;;EAGR,MAAM,SAAS,aAAa,WAAW;EAEvC,MAAM,UAAU,IAAI,aAAa;EACjC,IAAI,SAAS;EACb,IAAI,WAAW;EACf,MAAM,sCAAsB,IAAI,KAQ7B;AAEH,MAAI;AACF,UAAO,MAAM;AACX,QAAI,QAAQ,QAAQ,QAAS;IAE7B,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AAEV,cAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;IACjD,MAAM,QAAQ,OAAO,MAAM,KAAK;AAChC,aAAS,MAAM,KAAK,IAAI;AAExB,QAAI,OAAO,SAAS,KAAK,gBACvB,OAAM,IAAI,MACR,gEAAgE,KAAK,gBAAgB,qBACtF;AAGH,SAAK,MAAM,QAAQ,OAAO;AACxB,SAAI,KAAK,SAAS,KAAK,gBACrB,OAAM,IAAI,MACR,yDAAyD,KAAK,gBAAgB,qBAC/E;KAGH,MAAM,UAAU,KAAK,MAAM;AAC3B,SAAI,CAAC,QAAQ,WAAW,SAAS,CAAE;KACnC,MAAM,OAAO,QAAQ,MAAM,EAAE;AAC7B,SAAI,SAAS,SAAU;KAEvB,IAAI;AACJ,SAAI;AACF,eAAS,KAAK,MAAM,KAAK;cAClB,UAAU;AACjB,cAAQ,MACN,oDACA,EAAE,MAAM,GAAG,KAAK,MAAM,GAAG,IAAI,GAAG,KAAK,SAAS,MAAM,MAAM,MAAM,EAChE,SACD;AACD;;KAGF,MAAM,eAAe,mBAAmB,OAAO;AAC/C,SAAI,CAAC,SAAS,aAAa,CAAE;AAE7B,SAAI,OAAO,aAAa,YAAY,UAAU;MAC5C,MAAM,UAAU,aAAa;AAC7B,gCACE,2BACA,SAAS,QACT,SACA,KAAK,mBACN;AACD,kBAAY;AACZ,YAAM;OAAE,MAAM;OAA0B;OAAS;;KAGnD,MAAM,eAAe,aAAa;AAClC,SAAI,CAAC,MAAM,QAAQ,aAAa,CAAE;AAElC,UAAK,MAAM,MAAM,cAAc;AAC7B,UAAI,CAAC,yBAAyB,GAAG,CAAE;MACnC,MAAM,MAAM,GAAG;MACf,MAAM,WAAW,oBAAoB,IAAI,GAAG,MAAM;AAClD,UAAI,UAAU;AACZ,WAAI,GAAG,UAAU,WAAW;AAC1B,kCACE,uBACA,SAAS,UAAU,QACnB,GAAG,SAAS,WACZ,KAAK,sBACN;AACD,iBAAS,aAAa,GAAG,SAAS;;AAEpC,WAAI,OAAO,CAAC,SAAS,iBACnB,UAAS,mBAAmB;aAEzB;OACL,MAAM,UAAU,GAAG,UAAU,aAAa;AAC1C,WAAI,QAAQ,SAAS,KAAK,sBACxB,OAAM,IAAI,MACR,mEAAmE,KAAK,sBAAsB,qBAC/F;AAEH,2BAAoB,IAAI,GAAG,OAAO;QAChC,IAAI,GAAG,MAAM,QAAQ,GAAG;QACxB,MAAM,GAAG,UAAU,QAAQ;QAC3B,WAAW;QACX,GAAI,MAAM,EAAE,kBAAkB,KAAK,GAAG,EAAE;QACzC,CAAC;;;;;YAKF;AACR,OAAI;AACF,UAAM,OAAO,QAAQ;YACd,WAAW;AAClB,YAAQ,MACN,8DACA,UACD;;AAEH,OAAI;AACF,WAAO,aAAa;YACb,WAAW;AAClB,YAAQ,MACN,mEACA,UACD;;;EAIL,MAAM,YAA8B,MAAM,KACxC,oBAAoB,QAAQ,CAC7B,CAAC,KAAK,QAAQ;GACb,IAAI,GAAG;GACP,MAAM;GACN,UAAU;IAAE,MAAM,GAAG;IAAM,WAAW,GAAG,aAAa;IAAM;GAC5D,GAAI,GAAG,mBAAmB,EAAE,kBAAkB,GAAG,kBAAkB,GAAG,EAAE;GACzE,EAAE;AAEH,SAAO;GAAE,MAAM;GAAU;GAAW;;CAGtC,OAAe,iBACb,OACA,UACA,SACA,YAC2C;EAC3C,MAAM,gBAAgB,SACpB,WAAW,IAAI,KAAK,IAAI,KAAK,QAAQ,OAAO,KAAK;EAEnD,MAAM,eAAiC,MAAM,KAAK,GAAG,OAAO;GAC1D,IAAI,aAAa;GACjB,MAAM;GACN,UAAU;IACR,MAAM,aAAa,EAAE,KAAK;IAC1B,WAAW,KAAK,UAAU,EAAE,KAAK;IAClC;GACF,EAAE;AAEH,WAAS,KAAK;GACZ,MAAM;GACN,SAAS;GACT,YAAY;GACb,CAAC;AAEF,OAAK,IAAI,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK;GAC5C,MAAM,KAAK,aAAa;GACxB,MAAM,eAAe,MAAM,IAAI,QAAQ,GAAG,SAAS;AACnD,UAAO,KAAK,kBAAkB,IAAI,cAAc,UAAU,QAAQ;;;;;;;;CAStE,AAAQ,cACN,UACA,YACiB;EACjB,MAAM,gBAAgB,SACpB,WAAW,IAAI,KAAK,IAAI,KAAK,QAAQ,OAAO,KAAK;AAEnD,SAAO,SAAS,KAAK,MAAM;GACzB,IAAI,UAAyB,EAAE;AAC/B,OACE,EAAE,SAAS,eACX,EAAE,aACF,EAAE,UAAU,SAAS,MACpB,CAAC,EAAE,WAAW,EAAE,QAAQ,MAAM,KAAK,IAEpC,WAAU;GAGZ,MAAM,MAAqB;IACzB,MAAM,EAAE;IACR;IACD;AAED,OAAI,EAAE,WACJ,KAAI,eAAe,EAAE;AAGvB,OAAI,EAAE,aAAa,EAAE,UAAU,SAAS,EACtC,KAAI,aAAa,EAAE,UAAU,KAAK,QAAQ;IACxC,IAAI,GAAG;IACP,MAAM;IACN,UAAU;KACR,MAAM,aAAa,GAAG,KAAK;KAC3B,WACE,OAAO,GAAG,SAAS,WACf,GAAG,OACH,KAAK,UAAU,GAAG,QAAQ,EAAE,CAAC;KACpC;IACD,GAAI,GAAG,mBACH,EAAE,kBAAkB,GAAG,kBAAkB,GACzC,EAAE;IACP,EAAE;AAGL,UAAO;IACP;;CAGJ,AAAQ,WACN,aACA,YACc;AACd,SAAO,YAAY,KAAK,SAAS;GAC/B,MAAM;GACN,UAAU;IACR,MAAM,WAAW,IAAI,IAAI,KAAK,IAAI,IAAI;IACtC,aAAa,IAAI;IACjB,YAAY,IAAI;IACjB;GACF,EAAE;;;;;;;;;;AAeP,SAAgB,mBACd,MACwC;CACxC,MAAM,UAAU,KAAK,MAAM;CAE3B,MAAM,aAAa,2BAA2B,QAAQ;AACtD,KAAI,WAAW,SAAS,EAAG,QAAO;CAElC,MAAM,WAAW,6BAA6B,QAAQ;AACtD,KAAI,SAAS,SAAS,EAAG,QAAO;AAEhC,QAAO,EAAE;;AAGX,SAAS,oBAAoB,OAK3B;AACA,KAAI,CAAC,SAAS,MAAM,CAAE,QAAO;AAC7B,QAAO,OAAO,MAAM,SAAS;;AAG/B,SAAS,2BACP,MACwC;CACxC,MAAM,QAAQ,0BAA0B,KAAK;AAC7C,KAAI,CAAC,MAAO,QAAO,EAAE;AAErB,KAAI;EACF,MAAM,SAAkB,KAAK,MAAM,MAAM;AACzC,MAAI,CAAC,MAAM,QAAQ,OAAO,CAAE,QAAO,EAAE;AAErC,SAAO,OAAO,OAAO,oBAAoB,CAAC,KAAK,UAAU;GACvD,MAAM,KAAK;GACX,MAAM,KAAK,cAAc,KAAK,aAAa,KAAK,QAAQ,EAAE;GAC3D,EAAE;SACG;AACN,SAAO,EAAE;;;AAIb,SAAS,6BACP,MACwC;AACxC,KAAI,KAAK,SAAS,kCAChB,QAAO,EAAE;CAGX,MAAM,UAAU;CAChB,MAAM,UAAkD,EAAE;AAE1D,MAAK,MAAM,SAAS,KAAK,SAAS,QAAQ,EAAE;EAC1C,MAAM,OAAO,MAAM;EACnB,MAAM,UAAU,MAAM;EAEtB,MAAM,OAAgC,EAAE;AAExC,OAAK,MAAM,YAAY,QAAQ,SADZ,6CACgC,EAAE;GACnD,MAAM,MAAM,SAAS;AAErB,QAAK,OADS,SAAS,MAAM,SAAS,MAAM,SAAS;;AAIvD,UAAQ,KAAK;GAAE;GAAM;GAAM,CAAC;;AAG9B,QAAO"}
+{"version":3,"file":"databricks.js","names":["servingStream"],"sources":["../../src/agents/databricks.ts"],"sourcesContent":["import type {\n  AgentAdapter,\n  AgentEvent,\n  AgentInput,\n  AgentRunContext,\n  AgentToolDefinition,\n} from \"shared\";\nimport { stream as servingStream } from \"../connectors/serving/client\";\nimport { APPKIT_USER_AGENT, getClientOptions } from \"../context/client-options\";\n\n/** Default cap for a single incomplete SSE line tail (DoS guard). */\nconst DEFAULT_MAX_SSE_LINE_CHARS = 1024 * 1024;\n\n/** Default cap for accumulated assistant text from `delta.content`. */\nconst DEFAULT_MAX_STREAM_TEXT_CHARS = 4 * 1024 * 1024;\n\n/** Default cap for accumulated JSON arguments per streamed tool call index. */\nconst DEFAULT_MAX_TOOL_ARGUMENT_CHARS = 2 * 1024 * 1024;\n\n/** Cap text length before running Python-style tool-call regex (ReDoS guard). */\nconst PYTHON_STYLE_TOOL_PARSE_MAX_INPUT = 64 * 1024;\n\n/** Fallback HTTP timeout when the raw fetch adapter path receives no AbortSignal from the runner. */\nconst RAW_FETCH_DEFAULT_TIMEOUT_MS = 120_000;\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return typeof value === \"object\" && value !== null;\n}\n\nfunction extractLlamaToolJsonSlice(text: string): string | undefined {\n  const start = text.indexOf(\"[{\");\n  if (start < 0) return undefined;\n  const endBracket = text.lastIndexOf(\"}]\");\n  if (endBracket < start) return undefined;\n  return text.slice(start, endBracket + 2);\n}\n\n/** OpenAI SSE payload: `{ choices: [{ delta }] }`. */\nfunction openAiChoicesDelta(parsed: unknown): unknown {\n  if (!isRecord(parsed)) return undefined;\n  const choices = parsed.choices;\n  if (!Array.isArray(choices) || choices.length < 1) return undefined;\n  const first = choices[0];\n  if (!isRecord(first)) return undefined;\n  return first.delta;\n}\n\nfunction isStreamingDeltaToolCall(value: unknown): value is DeltaToolCall {\n  if (!isRecord(value)) return false;\n  return typeof value.index === \"number\";\n}\n\nfunction throwIfExceedsStreamLimit(\n  label: string,\n  currentLength: number,\n  chunk: string,\n  max: number,\n): void {\n  if (currentLength + chunk.length > max) {\n    throw new Error(\n      `DatabricksAdapter: ${label} exceeds configured limit (${max} UTF-16 code units)`,\n    );\n  }\n}\n\n/**\n * Transport shim: given an OpenAI-compatible request body, returns the raw\n * SSE byte stream from the serving endpoint. Injected at construction time so\n * callers can swap in the workspace SDK (factory paths), a bare `fetch`\n * (the raw constructor), or a test fake.\n */\ntype StreamBody = (\n  body: Record<string, unknown>,\n  signal?: AbortSignal,\n) => Promise<ReadableStream<Uint8Array>>;\n\n/**\n * Escape-hatch options: provide an `endpointUrl` + `authenticate()` and the\n * adapter uses a bare `fetch()` to call it. Useful for tests and for pointing\n * the adapter at non-workspace endpoints (reverse proxies, mocks).\n */\ninterface RawFetchAdapterOptions {\n  endpointUrl: string;\n  authenticate: () => Promise<Record<string, string>>;\n  maxSteps?: number;\n  maxTokens?: number;\n  /** Max length of one SSE line (including an incomplete tail in the buffer). */\n  maxSseLineChars?: number;\n  /** Max total length of assistant `delta.content` across the stream. */\n  maxStreamTextChars?: number;\n  /** Max length of streamed `function.arguments` per tool call index. */\n  maxToolArgumentsChars?: number;\n}\n\n/**\n * Preferred options: caller provides the transport function directly.\n * The `fromServingEndpoint` / `fromModelServing` factories use this to route\n * through `connectors/serving/stream`, which centralises URL encoding, auth\n * via the SDK's `apiClient.request`, and any future retries/telemetry.\n */\ninterface StreamBodyAdapterOptions {\n  streamBody: StreamBody;\n  maxSteps?: number;\n  maxTokens?: number;\n  maxSseLineChars?: number;\n  maxStreamTextChars?: number;\n  maxToolArgumentsChars?: number;\n}\n\ntype DatabricksAdapterOptions =\n  | RawFetchAdapterOptions\n  | StreamBodyAdapterOptions;\n\nfunction isStreamBodyOptions(\n  o: DatabricksAdapterOptions,\n): o is StreamBodyAdapterOptions {\n  return \"streamBody\" in o;\n}\n\n/**\n * Duck-typed subset of the Databricks SDK `WorkspaceClient`. Callers of\n * `fromServingEndpoint` and `fromModelServing` pass a real `WorkspaceClient`,\n * but we only need the `apiClient.request` surface — so we declare the minimal\n * interface rather than importing the SDK type directly. This keeps the adapter\n * free of a hard compile-time dependency on `@databricks/sdk-experimental`.\n */\ninterface WorkspaceClientLike {\n  apiClient: {\n    request(options: Record<string, unknown>): Promise<unknown>;\n  };\n}\n\ninterface ServingEndpointOptions {\n  workspaceClient: WorkspaceClientLike;\n  endpointName: string;\n  maxSteps?: number;\n  maxTokens?: number;\n  maxSseLineChars?: number;\n  maxStreamTextChars?: number;\n  maxToolArgumentsChars?: number;\n}\n\ninterface ModelServingOptions {\n  maxSteps?: number;\n  maxTokens?: number;\n  workspaceClient?: WorkspaceClientLike;\n  maxSseLineChars?: number;\n  maxStreamTextChars?: number;\n  maxToolArgumentsChars?: number;\n}\n\ninterface OpenAIMessage {\n  role: \"system\" | \"user\" | \"assistant\" | \"tool\";\n  content: string | null;\n  tool_calls?: OpenAIToolCall[];\n  tool_call_id?: string;\n}\n\ninterface OpenAIToolCall {\n  id: string;\n  type: \"function\";\n  function: { name: string; arguments: string };\n  /**\n   * Opaque Vertex/Gemini \"thought signature\" blob the request must echo\n   * back verbatim on the next turn. Vertex's OpenAI-compat proxy emits\n   * this as `thoughtSignature` (camelCase) at the top level of the\n   * tool_call delta (verified against `gemini-3.1-flash-lite-preview`),\n   * and accepts the same spelling back on outbound. Non-Gemini endpoints\n   * (Claude on Databricks, external OpenAI-compat models, Llama, etc.)\n   * leave this undefined and the serializer omits the key.\n   * See https://docs.cloud.google.com/vertex-ai/generative-ai/docs/thought-signatures\n   */\n  thoughtSignature?: string;\n}\n\ninterface OpenAITool {\n  type: \"function\";\n  function: {\n    name: string;\n    description: string;\n    parameters: unknown;\n  };\n}\n\ninterface DeltaToolCall {\n  index: number;\n  id?: string;\n  type?: string;\n  function?: { name?: string; arguments?: string };\n  /** See {@link OpenAIToolCall.thoughtSignature}. */\n  thoughtSignature?: string;\n}\n\n/**\n * Adapter that talks directly to Databricks Model Serving `/invocations` endpoint.\n *\n * No dependency on the Vercel AI SDK or LangChain. Uses raw `fetch()` to POST\n * OpenAI-compatible payloads and parses the SSE stream itself. Calls\n * `authenticate()` per-request so tokens are always fresh.\n *\n * Handles both structured `tool_calls` responses and text-based tool call\n * fallback parsing for models that output tool calls as text.\n *\n * @example Using the factory (recommended)\n * ```ts\n * import { createApp, createAgent, agents } from \"@databricks/appkit\";\n * import { DatabricksAdapter } from \"@databricks/appkit/beta\";\n * import { WorkspaceClient } from \"@databricks/sdk-experimental\";\n *\n * const adapter = DatabricksAdapter.fromServingEndpoint({\n *   workspaceClient: new WorkspaceClient({}),\n *   endpointName: \"my-endpoint\",\n * });\n *\n * await createApp({\n *   plugins: [\n *     agents({\n *       agents: {\n *         assistant: createAgent({\n *           instructions: \"You are a helpful assistant.\",\n *           model: adapter,\n *         }),\n *       },\n *     }),\n *   ],\n * });\n * ```\n *\n * @example Using the raw constructor\n * ```ts\n * const adapter = new DatabricksAdapter({\n *   endpointUrl: \"https://host/serving-endpoints/my-endpoint/invocations\",\n *   authenticate: async () => ({ Authorization: `Bearer ${token}` }),\n * });\n * ```\n */\nexport class DatabricksAdapter implements AgentAdapter {\n  private streamBody: StreamBody;\n  private maxSteps: number;\n  private maxTokens: number;\n  private maxSseLineChars: number;\n  private maxStreamTextChars: number;\n  private maxToolArgumentsChars: number;\n\n  constructor(options: DatabricksAdapterOptions) {\n    this.maxSteps = options.maxSteps ?? 10;\n    this.maxTokens = options.maxTokens ?? 4096;\n    this.maxSseLineChars =\n      options.maxSseLineChars ?? DEFAULT_MAX_SSE_LINE_CHARS;\n    this.maxStreamTextChars =\n      options.maxStreamTextChars ?? DEFAULT_MAX_STREAM_TEXT_CHARS;\n    this.maxToolArgumentsChars =\n      options.maxToolArgumentsChars ?? DEFAULT_MAX_TOOL_ARGUMENT_CHARS;\n\n    if (isStreamBodyOptions(options)) {\n      this.streamBody = options.streamBody;\n    } else {\n      const { endpointUrl, authenticate } = options;\n      this.streamBody = async (body, signal) => {\n        const fetchSignal =\n          signal ?? AbortSignal.timeout(RAW_FETCH_DEFAULT_TIMEOUT_MS);\n        const authHeaders = await authenticate();\n        const response = await fetch(endpointUrl, {\n          method: \"POST\",\n          headers: {\n            \"User-Agent\": APPKIT_USER_AGENT,\n            \"Content-Type\": \"application/json\",\n            ...authHeaders,\n          },\n          body: JSON.stringify(body),\n          signal: fetchSignal,\n        });\n        if (!response.ok) {\n          const errorText = await response.text().catch(() => \"Unknown error\");\n          throw new Error(\n            `Databricks API error (${response.status}): ${errorText}`,\n          );\n        }\n        if (!response.body) throw new Error(\"No response body\");\n        return response.body;\n      };\n    }\n  }\n\n  /**\n   * Creates a DatabricksAdapter for a Databricks Model Serving endpoint.\n   *\n   * Routes through the shared `connectors/serving/stream` helper, which\n   * delegates to the SDK's `apiClient.request({ raw: true })`. That gives the\n   * adapter centralised URL encoding + authentication with the rest of the\n   * serving surface — no bespoke `fetch()` + `authenticate()` plumbing.\n   */\n  static async fromServingEndpoint(\n    options: ServingEndpointOptions,\n  ): Promise<DatabricksAdapter> {\n    const {\n      workspaceClient,\n      endpointName,\n      maxSteps,\n      maxTokens,\n      maxSseLineChars,\n      maxStreamTextChars,\n      maxToolArgumentsChars,\n    } = options;\n    return new DatabricksAdapter({\n      streamBody: (body, signal) =>\n        // Cast through the structural shape: the connector types\n        // `workspaceClient` as the SDK's concrete `WorkspaceClient`, but we\n        // only need `apiClient.request`.\n        servingStream(\n          workspaceClient as unknown as Parameters<typeof servingStream>[0],\n          endpointName,\n          body,\n          signal,\n        ),\n      maxSteps,\n      maxTokens,\n      maxSseLineChars,\n      maxStreamTextChars,\n      maxToolArgumentsChars,\n    });\n  }\n\n  /**\n   * Creates a DatabricksAdapter from a Model Serving endpoint name.\n   * Auto-creates a WorkspaceClient internally. Reads the endpoint name\n   * from the argument or the `DATABRICKS_SERVING_ENDPOINT_NAME` env var.\n   *\n   * @example\n   * ```ts\n   * // Reads endpoint from DATABRICKS_SERVING_ENDPOINT_NAME env var\n   * const adapter = await DatabricksAdapter.fromModelServing();\n   *\n   * // Explicit endpoint\n   * const adapter = await DatabricksAdapter.fromModelServing(\"my-endpoint\");\n   *\n   * // With options\n   * const adapter = await DatabricksAdapter.fromModelServing(\"my-endpoint\", {\n   *   maxSteps: 5,\n   *   maxTokens: 2048,\n   * });\n   * ```\n   */\n  static async fromModelServing(\n    endpointName?: string,\n    options?: ModelServingOptions,\n  ): Promise<DatabricksAdapter> {\n    const resolvedEndpoint =\n      endpointName ?? process.env.DATABRICKS_SERVING_ENDPOINT_NAME;\n\n    if (!resolvedEndpoint) {\n      throw new Error(\n        \"No endpoint name provided and DATABRICKS_SERVING_ENDPOINT_NAME env var is not set. \" +\n          \"Pass an endpoint name or set DATABRICKS_SERVING_ENDPOINT_NAME.\",\n      );\n    }\n\n    let workspaceClient: WorkspaceClientLike | undefined =\n      options?.workspaceClient;\n    if (!workspaceClient) {\n      const sdk = await import(\"@databricks/sdk-experimental\");\n      workspaceClient = new sdk.WorkspaceClient(\n        {},\n        getClientOptions(),\n      ) as unknown as WorkspaceClientLike;\n    }\n\n    return DatabricksAdapter.fromServingEndpoint({\n      workspaceClient,\n      endpointName: resolvedEndpoint,\n      maxSteps: options?.maxSteps,\n      maxTokens: options?.maxTokens,\n      maxSseLineChars: options?.maxSseLineChars,\n      maxStreamTextChars: options?.maxStreamTextChars,\n      maxToolArgumentsChars: options?.maxToolArgumentsChars,\n    });\n  }\n\n  async *run(\n    input: AgentInput,\n    context: AgentRunContext,\n  ): AsyncGenerator<AgentEvent, void, unknown> {\n    // Databricks API requires tool names to match [a-zA-Z0-9_-].\n    // Our tool names use dots (e.g. \"analytics.query\"), so we swap dots\n    // for double-underscores in the wire format and map back on receipt.\n    const nameToWire = new Map<string, string>();\n    const wireToName = new Map<string, string>();\n    for (const tool of input.tools) {\n      const wire = tool.name.replace(/\\./g, \"__\");\n      if (wireToName.has(wire) && wireToName.get(wire) !== tool.name) {\n        throw new Error(\n          `Tool name collision: '${tool.name}' and '${wireToName.get(wire)}' both map to wire name '${wire}'`,\n        );\n      }\n      nameToWire.set(tool.name, wire);\n      wireToName.set(wire, tool.name);\n    }\n\n    const tools = this.buildTools(input.tools, nameToWire);\n    const messages = this.buildMessages(input.messages, nameToWire);\n\n    yield { type: \"status\", status: \"running\" };\n\n    for (let step = 0; step < this.maxSteps; step++) {\n      if (context.signal?.aborted) break;\n\n      const { text, toolCalls } = yield* this.streamCompletion(\n        messages,\n        tools,\n        context,\n      );\n\n      if (toolCalls.length === 0) {\n        const parsed = parseTextToolCalls(text);\n        if (parsed.length > 0) {\n          yield* this.executeToolCalls(parsed, messages, context, nameToWire);\n          continue;\n        }\n        break;\n      }\n\n      messages.push({\n        role: \"assistant\",\n        content: text || null,\n        tool_calls: toolCalls,\n      });\n\n      for (const tc of toolCalls) {\n        const wireName = tc.function.name;\n        const originalName = wireToName.get(wireName) ?? wireName;\n        yield* this.executeSingleTool(tc, originalName, messages, context);\n      }\n    }\n  }\n\n  /** Parse wire arguments, emit tool_call / tool_result, append tool messages. */\n  private async *executeSingleTool(\n    tc: OpenAIToolCall,\n    originalName: string,\n    messages: OpenAIMessage[],\n    context: AgentRunContext,\n  ): AsyncGenerator<AgentEvent, void, unknown> {\n    let args: unknown;\n    try {\n      args = JSON.parse(tc.function.arguments);\n    } catch {\n      args = {};\n    }\n\n    yield { type: \"tool_call\", callId: tc.id, name: originalName, args };\n\n    try {\n      const result = await context.executeTool(originalName, args);\n      const resultStr =\n        typeof result === \"string\" ? result : JSON.stringify(result);\n\n      yield { type: \"tool_result\", callId: tc.id, result };\n\n      messages.push({\n        role: \"tool\",\n        content: resultStr,\n        tool_call_id: tc.id,\n      });\n    } catch (error) {\n      const errMsg =\n        error instanceof Error ? error.message : \"Tool execution failed\";\n\n      yield {\n        type: \"tool_result\",\n        callId: tc.id,\n        result: null,\n        error: errMsg,\n      };\n\n      messages.push({\n        role: \"tool\",\n        content: JSON.stringify({ error: errMsg }),\n        tool_call_id: tc.id,\n      });\n    }\n  }\n\n  private async *streamCompletion(\n    messages: OpenAIMessage[],\n    tools: OpenAITool[],\n    context: AgentRunContext,\n  ): AsyncGenerator<\n    AgentEvent,\n    { text: string; toolCalls: OpenAIToolCall[] },\n    unknown\n  > {\n    const body: Record<string, unknown> = {\n      messages,\n      stream: true,\n      max_tokens: this.maxTokens,\n    };\n\n    if (tools.length > 0) {\n      body.tools = tools;\n    }\n\n    let responseBody: ReadableStream<Uint8Array>;\n    try {\n      responseBody = await this.streamBody(body, context.signal);\n    } catch (err) {\n      const msg = err instanceof Error ? err.message : \"Stream request failed\";\n      yield { type: \"status\", status: \"error\", error: msg };\n      throw err;\n    }\n\n    const reader = responseBody.getReader();\n\n    const decoder = new TextDecoder();\n    let buffer = \"\";\n    let fullText = \"\";\n    const toolCallAccumulator = new Map<\n      number,\n      {\n        id: string;\n        name: string;\n        arguments: string;\n        thoughtSignature?: string;\n      }\n    >();\n\n    try {\n      while (true) {\n        if (context.signal?.aborted) break;\n\n        const { done, value } = await reader.read();\n        if (done) break;\n\n        buffer += decoder.decode(value, { stream: true });\n        const lines = buffer.split(\"\\n\");\n        buffer = lines.pop() ?? \"\";\n\n        if (buffer.length > this.maxSseLineChars) {\n          throw new Error(\n            `DatabricksAdapter: SSE line buffer exceeds configured limit (${this.maxSseLineChars} UTF-16 code units)`,\n          );\n        }\n\n        for (const line of lines) {\n          if (line.length > this.maxSseLineChars) {\n            throw new Error(\n              `DatabricksAdapter: SSE line exceeds configured limit (${this.maxSseLineChars} UTF-16 code units)`,\n            );\n          }\n\n          const trimmed = line.trim();\n          if (!trimmed.startsWith(\"data: \")) continue;\n          const data = trimmed.slice(6);\n          if (data === \"[DONE]\") continue;\n\n          let parsed: unknown;\n          try {\n            parsed = JSON.parse(data);\n          } catch (parseErr) {\n            console.debug(\n              \"[DatabricksAdapter] malformed SSE data line JSON\",\n              { line: `${data.slice(0, 256)}${data.length > 256 ? \"…\" : \"\"}` },\n              parseErr,\n            );\n            continue;\n          }\n\n          const deltaUnknown = openAiChoicesDelta(parsed);\n          if (!isRecord(deltaUnknown)) continue;\n\n          if (typeof deltaUnknown.content === \"string\") {\n            const content = deltaUnknown.content;\n            throwIfExceedsStreamLimit(\n              \"streamed assistant text\",\n              fullText.length,\n              content,\n              this.maxStreamTextChars,\n            );\n            fullText += content;\n            yield { type: \"message_delta\" as const, content };\n          }\n\n          const toolCallsRaw = deltaUnknown.tool_calls;\n          if (!Array.isArray(toolCallsRaw)) continue;\n\n          for (const tc of toolCallsRaw) {\n            if (!isStreamingDeltaToolCall(tc)) continue;\n            const sig = tc.thoughtSignature;\n            const existing = toolCallAccumulator.get(tc.index);\n            if (existing) {\n              if (tc.function?.arguments) {\n                throwIfExceedsStreamLimit(\n                  \"tool call arguments\",\n                  existing.arguments.length,\n                  tc.function.arguments,\n                  this.maxToolArgumentsChars,\n                );\n                existing.arguments += tc.function.arguments;\n              }\n              if (sig && !existing.thoughtSignature) {\n                existing.thoughtSignature = sig;\n              }\n            } else {\n              const initial = tc.function?.arguments ?? \"\";\n              if (initial.length > this.maxToolArgumentsChars) {\n                throw new Error(\n                  `DatabricksAdapter: tool call arguments exceed configured limit (${this.maxToolArgumentsChars} UTF-16 code units)`,\n                );\n              }\n              toolCallAccumulator.set(tc.index, {\n                id: tc.id ?? `call_${tc.index}`,\n                name: tc.function?.name ?? \"\",\n                arguments: initial,\n                ...(sig ? { thoughtSignature: sig } : {}),\n              });\n            }\n          }\n        }\n      }\n    } finally {\n      try {\n        await reader.cancel();\n      } catch (cancelErr) {\n        console.debug(\n          \"[DatabricksAdapter] reader.cancel() failed during teardown\",\n          cancelErr,\n        );\n      }\n      try {\n        reader.releaseLock();\n      } catch (unlockErr) {\n        console.debug(\n          \"[DatabricksAdapter] reader.releaseLock() failed during teardown\",\n          unlockErr,\n        );\n      }\n    }\n\n    const toolCalls: OpenAIToolCall[] = Array.from(\n      toolCallAccumulator.values(),\n    ).map((tc) => ({\n      id: tc.id,\n      type: \"function\" as const,\n      function: { name: tc.name, arguments: tc.arguments || \"{}\" },\n      ...(tc.thoughtSignature ? { thoughtSignature: tc.thoughtSignature } : {}),\n    }));\n\n    return { text: fullText, toolCalls };\n  }\n\n  private async *executeToolCalls(\n    calls: Array<{ name: string; args: unknown }>,\n    messages: OpenAIMessage[],\n    context: AgentRunContext,\n    nameToWire: Map<string, string>,\n  ): AsyncGenerator<AgentEvent, void, unknown> {\n    const wireToolName = (name: string) =>\n      nameToWire.get(name) ?? name.replace(/\\./g, \"__\");\n\n    const toolCallObjs: OpenAIToolCall[] = calls.map((c, i) => ({\n      id: `text_call_${i}`,\n      type: \"function\" as const,\n      function: {\n        name: wireToolName(c.name),\n        arguments: JSON.stringify(c.args),\n      },\n    }));\n\n    messages.push({\n      role: \"assistant\",\n      content: null,\n      tool_calls: toolCallObjs,\n    });\n\n    for (let i = 0; i < toolCallObjs.length; i++) {\n      const tc = toolCallObjs[i];\n      const originalName = calls[i]?.name ?? tc.function.name;\n      yield* this.executeSingleTool(tc, originalName, messages, context);\n    }\n  }\n\n  /**\n   * Maps AppKit {@link AgentInput} messages into OpenAI-compatible wire messages.\n   * Preserves multi-turn tool state (`toolCalls` → `tool_calls`, `toolCallId` →\n   * `tool_call_id`) so resumed threads and hydrated history reach the model.\n   */\n  private buildMessages(\n    messages: AgentInput[\"messages\"],\n    nameToWire: Map<string, string>,\n  ): OpenAIMessage[] {\n    const wireToolName = (name: string) =>\n      nameToWire.get(name) ?? name.replace(/\\./g, \"__\");\n\n    return messages.map((m) => {\n      let content: string | null = m.content;\n      if (\n        m.role === \"assistant\" &&\n        m.toolCalls &&\n        m.toolCalls.length > 0 &&\n        (!m.content || m.content.trim() === \"\")\n      ) {\n        content = null;\n      }\n\n      const out: OpenAIMessage = {\n        role: m.role as OpenAIMessage[\"role\"],\n        content,\n      };\n\n      if (m.toolCallId) {\n        out.tool_call_id = m.toolCallId;\n      }\n\n      if (m.toolCalls && m.toolCalls.length > 0) {\n        out.tool_calls = m.toolCalls.map((tc) => ({\n          id: tc.id,\n          type: \"function\" as const,\n          function: {\n            name: wireToolName(tc.name),\n            arguments:\n              typeof tc.args === \"string\"\n                ? tc.args\n                : JSON.stringify(tc.args ?? {}),\n          },\n          ...(tc.thoughtSignature\n            ? { thoughtSignature: tc.thoughtSignature }\n            : {}),\n        }));\n      }\n\n      return out;\n    });\n  }\n\n  private buildTools(\n    definitions: AgentToolDefinition[],\n    nameToWire: Map<string, string>,\n  ): OpenAITool[] {\n    return definitions.map((def) => ({\n      type: \"function\" as const,\n      function: {\n        name: nameToWire.get(def.name) ?? def.name,\n        description: def.description,\n        parameters: def.parameters,\n      },\n    }));\n  }\n}\n\n// ---------------------------------------------------------------------------\n// Text-based tool call parsing (fallback)\n// ---------------------------------------------------------------------------\n\n/**\n * Parses text-based tool calls from model output.\n *\n * Handles two formats:\n * 1. Llama native: `[{\"name\": \"tool_name\", \"parameters\": {\"arg\": \"val\"}}]`\n * 2. Python-style: `[tool_name(arg1='val1', arg2='val2')]`\n */\nexport function parseTextToolCalls(\n  text: string,\n): Array<{ name: string; args: unknown }> {\n  const trimmed = text.trim();\n\n  const jsonResult = tryParseLlamaJsonToolCalls(trimmed);\n  if (jsonResult.length > 0) return jsonResult;\n\n  const pyResult = tryParsePythonStyleToolCalls(trimmed);\n  if (pyResult.length > 0) return pyResult;\n\n  return [];\n}\n\nfunction isLlamaToolJsonItem(value: unknown): value is Record<\n  string,\n  unknown\n> & {\n  name: string;\n} {\n  if (!isRecord(value)) return false;\n  return typeof value.name === \"string\";\n}\n\nfunction tryParseLlamaJsonToolCalls(\n  text: string,\n): Array<{ name: string; args: unknown }> {\n  const slice = extractLlamaToolJsonSlice(text);\n  if (!slice) return [];\n\n  try {\n    const parsed: unknown = JSON.parse(slice);\n    if (!Array.isArray(parsed)) return [];\n\n    return parsed.filter(isLlamaToolJsonItem).map((item) => ({\n      name: item.name,\n      args: item.parameters ?? item.arguments ?? item.args ?? {},\n    }));\n  } catch {\n    return [];\n  }\n}\n\nfunction tryParsePythonStyleToolCalls(\n  text: string,\n): Array<{ name: string; args: unknown }> {\n  if (text.length > PYTHON_STYLE_TOOL_PARSE_MAX_INPUT) {\n    return [];\n  }\n\n  const pattern = /\\[?([a-zA-Z_][\\w.]*)\\(([^)]*)\\)\\]?/g;\n  const results: Array<{ name: string; args: unknown }> = [];\n\n  for (const match of text.matchAll(pattern)) {\n    const name = match[1];\n    const argsStr = match[2];\n\n    const args: Record<string, unknown> = {};\n    const argPattern = /(\\w+)\\s*=\\s*(?:'([^']*)'|\"([^\"]*)\"|(\\S+))/g;\n    for (const argMatch of argsStr.matchAll(argPattern)) {\n      const key = argMatch[1];\n      const value = argMatch[2] ?? argMatch[3] ?? argMatch[4];\n      args[key] = value;\n    }\n\n    results.push({ name, args });\n  }\n\n  return results;\n}\n"],"mappings":";;;;;AAWA,MAAM,6BAA6B,OAAO;;AAG1C,MAAM,gCAAgC,IAAI,OAAO;;AAGjD,MAAM,kCAAkC,IAAI,OAAO;;AAGnD,MAAM,oCAAoC,KAAK;;AAG/C,MAAM,+BAA+B;AAErC,SAAS,SAAS,OAAkD;AAClE,QAAO,OAAO,UAAU,YAAY,UAAU;;AAGhD,SAAS,0BAA0B,MAAkC;CACnE,MAAM,QAAQ,KAAK,QAAQ,KAAK;AAChC,KAAI,QAAQ,EAAG,QAAO;CACtB,MAAM,aAAa,KAAK,YAAY,KAAK;AACzC,KAAI,aAAa,MAAO,QAAO;AAC/B,QAAO,KAAK,MAAM,OAAO,aAAa,EAAE;;;AAI1C,SAAS,mBAAmB,QAA0B;AACpD,KAAI,CAAC,SAAS,OAAO,CAAE,QAAO;CAC9B,MAAM,UAAU,OAAO;AACvB,KAAI,CAAC,MAAM,QAAQ,QAAQ,IAAI,QAAQ,SAAS,EAAG,QAAO;CAC1D,MAAM,QAAQ,QAAQ;AACtB,KAAI,CAAC,SAAS,MAAM,CAAE,QAAO;AAC7B,QAAO,MAAM;;AAGf,SAAS,yBAAyB,OAAwC;AACxE,KAAI,CAAC,SAAS,MAAM,CAAE,QAAO;AAC7B,QAAO,OAAO,MAAM,UAAU;;AAGhC,SAAS,0BACP,OACA,eACA,OACA,KACM;AACN,KAAI,gBAAgB,MAAM,SAAS,IACjC,OAAM,IAAI,MACR,sBAAsB,MAAM,6BAA6B,IAAI,qBAC9D;;AAoDL,SAAS,oBACP,GAC+B;AAC/B,QAAO,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwHzB,IAAa,oBAAb,MAAa,kBAA0C;CACrD,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,YAAY,SAAmC;AAC7C,OAAK,WAAW,QAAQ,YAAY;AACpC,OAAK,YAAY,QAAQ,aAAa;AACtC,OAAK,kBACH,QAAQ,mBAAmB;AAC7B,OAAK,qBACH,QAAQ,sBAAsB;AAChC,OAAK,wBACH,QAAQ,yBAAyB;AAEnC,MAAI,oBAAoB,QAAQ,CAC9B,MAAK,aAAa,QAAQ;OACrB;GACL,MAAM,EAAE,aAAa,iBAAiB;AACtC,QAAK,aAAa,OAAO,MAAM,WAAW;IACxC,MAAM,cACJ,UAAU,YAAY,QAAQ,6BAA6B;IAC7D,MAAM,cAAc,MAAM,cAAc;IACxC,MAAM,WAAW,MAAM,MAAM,aAAa;KACxC,QAAQ;KACR,SAAS;MACP,cAAc;MACd,gBAAgB;MAChB,GAAG;MACJ;KACD,MAAM,KAAK,UAAU,KAAK;KAC1B,QAAQ;KACT,CAAC;AACF,QAAI,CAAC,SAAS,IAAI;KAChB,MAAM,YAAY,MAAM,SAAS,MAAM,CAAC,YAAY,gBAAgB;AACpE,WAAM,IAAI,MACR,yBAAyB,SAAS,OAAO,KAAK,YAC/C;;AAEH,QAAI,CAAC,SAAS,KAAM,OAAM,IAAI,MAAM,mBAAmB;AACvD,WAAO,SAAS;;;;;;;;;;;;CAatB,aAAa,oBACX,SAC4B;EAC5B,MAAM,EACJ,iBACA,cACA,UACA,WACA,iBACA,oBACA,0BACE;AACJ,SAAO,IAAI,kBAAkB;GAC3B,aAAa,MAAM,WAIjBA,OACE,iBACA,cACA,MACA,OACD;GACH;GACA;GACA;GACA;GACA;GACD,CAAC;;;;;;;;;;;;;;;;;;;;;;CAuBJ,aAAa,iBACX,cACA,SAC4B;EAC5B,MAAM,mBACJ,gBAAgB,QAAQ,IAAI;AAE9B,MAAI,CAAC,iBACH,OAAM,IAAI,MACR,oJAED;EAGH,IAAI,kBACF,SAAS;AACX,MAAI,CAAC,gBAEH,mBAAkB,KADN,OAAM,OAAO,kCACC,gBACxB,EAAE,EACF,kBAAkB,CACnB;AAGH,SAAO,kBAAkB,oBAAoB;GAC3C;GACA,cAAc;GACd,UAAU,SAAS;GACnB,WAAW,SAAS;GACpB,iBAAiB,SAAS;GAC1B,oBAAoB,SAAS;GAC7B,uBAAuB,SAAS;GACjC,CAAC;;CAGJ,OAAO,IACL,OACA,SAC2C;EAI3C,MAAM,6BAAa,IAAI,KAAqB;EAC5C,MAAM,6BAAa,IAAI,KAAqB;AAC5C,OAAK,MAAM,QAAQ,MAAM,OAAO;GAC9B,MAAM,OAAO,KAAK,KAAK,QAAQ,OAAO,KAAK;AAC3C,OAAI,WAAW,IAAI,KAAK,IAAI,WAAW,IAAI,KAAK,KAAK,KAAK,KACxD,OAAM,IAAI,MACR,yBAAyB,KAAK,KAAK,SAAS,WAAW,IAAI,KAAK,CAAC,2BAA2B,KAAK,GAClG;AAEH,cAAW,IAAI,KAAK,MAAM,KAAK;AAC/B,cAAW,IAAI,MAAM,KAAK,KAAK;;EAGjC,MAAM,QAAQ,KAAK,WAAW,MAAM,OAAO,WAAW;EACtD,MAAM,WAAW,KAAK,cAAc,MAAM,UAAU,WAAW;AAE/D,QAAM;GAAE,MAAM;GAAU,QAAQ;GAAW;AAE3C,OAAK,IAAI,OAAO,GAAG,OAAO,KAAK,UAAU,QAAQ;AAC/C,OAAI,QAAQ,QAAQ,QAAS;GAE7B,MAAM,EAAE,MAAM,cAAc,OAAO,KAAK,iBACtC,UACA,OACA,QACD;AAED,OAAI,UAAU,WAAW,GAAG;IAC1B,MAAM,SAAS,mBAAmB,KAAK;AACvC,QAAI,OAAO,SAAS,GAAG;AACrB,YAAO,KAAK,iBAAiB,QAAQ,UAAU,SAAS,WAAW;AACnE;;AAEF;;AAGF,YAAS,KAAK;IACZ,MAAM;IACN,SAAS,QAAQ;IACjB,YAAY;IACb,CAAC;AAEF,QAAK,MAAM,MAAM,WAAW;IAC1B,MAAM,WAAW,GAAG,SAAS;IAC7B,MAAM,eAAe,WAAW,IAAI,SAAS,IAAI;AACjD,WAAO,KAAK,kBAAkB,IAAI,cAAc,UAAU,QAAQ;;;;;CAMxE,OAAe,kBACb,IACA,cACA,UACA,SAC2C;EAC3C,IAAI;AACJ,MAAI;AACF,UAAO,KAAK,MAAM,GAAG,SAAS,UAAU;UAClC;AACN,UAAO,EAAE;;AAGX,QAAM;GAAE,MAAM;GAAa,QAAQ,GAAG;GAAI,MAAM;GAAc;GAAM;AAEpE,MAAI;GACF,MAAM,SAAS,MAAM,QAAQ,YAAY,cAAc,KAAK;GAC5D,MAAM,YACJ,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,OAAO;AAE9D,SAAM;IAAE,MAAM;IAAe,QAAQ,GAAG;IAAI;IAAQ;AAEpD,YAAS,KAAK;IACZ,MAAM;IACN,SAAS;IACT,cAAc,GAAG;IAClB,CAAC;WACK,OAAO;GACd,MAAM,SACJ,iBAAiB,QAAQ,MAAM,UAAU;AAE3C,SAAM;IACJ,MAAM;IACN,QAAQ,GAAG;IACX,QAAQ;IACR,OAAO;IACR;AAED,YAAS,KAAK;IACZ,MAAM;IACN,SAAS,KAAK,UAAU,EAAE,OAAO,QAAQ,CAAC;IAC1C,cAAc,GAAG;IAClB,CAAC;;;CAIN,OAAe,iBACb,UACA,OACA,SAKA;EACA,MAAM,OAAgC;GACpC;GACA,QAAQ;GACR,YAAY,KAAK;GAClB;AAED,MAAI,MAAM,SAAS,EACjB,MAAK,QAAQ;EAGf,IAAI;AACJ,MAAI;AACF,kBAAe,MAAM,KAAK,WAAW,MAAM,QAAQ,OAAO;WACnD,KAAK;AAEZ,SAAM;IAAE,MAAM;IAAU,QAAQ;IAAS,OAD7B,eAAe,QAAQ,IAAI,UAAU;IACI;AACrD,SAAM;;EAGR,MAAM,SAAS,aAAa,WAAW;EAEvC,MAAM,UAAU,IAAI,aAAa;EACjC,IAAI,SAAS;EACb,IAAI,WAAW;EACf,MAAM,sCAAsB,IAAI,KAQ7B;AAEH,MAAI;AACF,UAAO,MAAM;AACX,QAAI,QAAQ,QAAQ,QAAS;IAE7B,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AAEV,cAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;IACjD,MAAM,QAAQ,OAAO,MAAM,KAAK;AAChC,aAAS,MAAM,KAAK,IAAI;AAExB,QAAI,OAAO,SAAS,KAAK,gBACvB,OAAM,IAAI,MACR,gEAAgE,KAAK,gBAAgB,qBACtF;AAGH,SAAK,MAAM,QAAQ,OAAO;AACxB,SAAI,KAAK,SAAS,KAAK,gBACrB,OAAM,IAAI,MACR,yDAAyD,KAAK,gBAAgB,qBAC/E;KAGH,MAAM,UAAU,KAAK,MAAM;AAC3B,SAAI,CAAC,QAAQ,WAAW,SAAS,CAAE;KACnC,MAAM,OAAO,QAAQ,MAAM,EAAE;AAC7B,SAAI,SAAS,SAAU;KAEvB,IAAI;AACJ,SAAI;AACF,eAAS,KAAK,MAAM,KAAK;cAClB,UAAU;AACjB,cAAQ,MACN,oDACA,EAAE,MAAM,GAAG,KAAK,MAAM,GAAG,IAAI,GAAG,KAAK,SAAS,MAAM,MAAM,MAAM,EAChE,SACD;AACD;;KAGF,MAAM,eAAe,mBAAmB,OAAO;AAC/C,SAAI,CAAC,SAAS,aAAa,CAAE;AAE7B,SAAI,OAAO,aAAa,YAAY,UAAU;MAC5C,MAAM,UAAU,aAAa;AAC7B,gCACE,2BACA,SAAS,QACT,SACA,KAAK,mBACN;AACD,kBAAY;AACZ,YAAM;OAAE,MAAM;OAA0B;OAAS;;KAGnD,MAAM,eAAe,aAAa;AAClC,SAAI,CAAC,MAAM,QAAQ,aAAa,CAAE;AAElC,UAAK,MAAM,MAAM,cAAc;AAC7B,UAAI,CAAC,yBAAyB,GAAG,CAAE;MACnC,MAAM,MAAM,GAAG;MACf,MAAM,WAAW,oBAAoB,IAAI,GAAG,MAAM;AAClD,UAAI,UAAU;AACZ,WAAI,GAAG,UAAU,WAAW;AAC1B,kCACE,uBACA,SAAS,UAAU,QACnB,GAAG,SAAS,WACZ,KAAK,sBACN;AACD,iBAAS,aAAa,GAAG,SAAS;;AAEpC,WAAI,OAAO,CAAC,SAAS,iBACnB,UAAS,mBAAmB;aAEzB;OACL,MAAM,UAAU,GAAG,UAAU,aAAa;AAC1C,WAAI,QAAQ,SAAS,KAAK,sBACxB,OAAM,IAAI,MACR,mEAAmE,KAAK,sBAAsB,qBAC/F;AAEH,2BAAoB,IAAI,GAAG,OAAO;QAChC,IAAI,GAAG,MAAM,QAAQ,GAAG;QACxB,MAAM,GAAG,UAAU,QAAQ;QAC3B,WAAW;QACX,GAAI,MAAM,EAAE,kBAAkB,KAAK,GAAG,EAAE;QACzC,CAAC;;;;;YAKF;AACR,OAAI;AACF,UAAM,OAAO,QAAQ;YACd,WAAW;AAClB,YAAQ,MACN,8DACA,UACD;;AAEH,OAAI;AACF,WAAO,aAAa;YACb,WAAW;AAClB,YAAQ,MACN,mEACA,UACD;;;EAIL,MAAM,YAA8B,MAAM,KACxC,oBAAoB,QAAQ,CAC7B,CAAC,KAAK,QAAQ;GACb,IAAI,GAAG;GACP,MAAM;GACN,UAAU;IAAE,MAAM,GAAG;IAAM,WAAW,GAAG,aAAa;IAAM;GAC5D,GAAI,GAAG,mBAAmB,EAAE,kBAAkB,GAAG,kBAAkB,GAAG,EAAE;GACzE,EAAE;AAEH,SAAO;GAAE,MAAM;GAAU;GAAW;;CAGtC,OAAe,iBACb,OACA,UACA,SACA,YAC2C;EAC3C,MAAM,gBAAgB,SACpB,WAAW,IAAI,KAAK,IAAI,KAAK,QAAQ,OAAO,KAAK;EAEnD,MAAM,eAAiC,MAAM,KAAK,GAAG,OAAO;GAC1D,IAAI,aAAa;GACjB,MAAM;GACN,UAAU;IACR,MAAM,aAAa,EAAE,KAAK;IAC1B,WAAW,KAAK,UAAU,EAAE,KAAK;IAClC;GACF,EAAE;AAEH,WAAS,KAAK;GACZ,MAAM;GACN,SAAS;GACT,YAAY;GACb,CAAC;AAEF,OAAK,IAAI,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK;GAC5C,MAAM,KAAK,aAAa;GACxB,MAAM,eAAe,MAAM,IAAI,QAAQ,GAAG,SAAS;AACnD,UAAO,KAAK,kBAAkB,IAAI,cAAc,UAAU,QAAQ;;;;;;;;CAStE,AAAQ,cACN,UACA,YACiB;EACjB,MAAM,gBAAgB,SACpB,WAAW,IAAI,KAAK,IAAI,KAAK,QAAQ,OAAO,KAAK;AAEnD,SAAO,SAAS,KAAK,MAAM;GACzB,IAAI,UAAyB,EAAE;AAC/B,OACE,EAAE,SAAS,eACX,EAAE,aACF,EAAE,UAAU,SAAS,MACpB,CAAC,EAAE,WAAW,EAAE,QAAQ,MAAM,KAAK,IAEpC,WAAU;GAGZ,MAAM,MAAqB;IACzB,MAAM,EAAE;IACR;IACD;AAED,OAAI,EAAE,WACJ,KAAI,eAAe,EAAE;AAGvB,OAAI,EAAE,aAAa,EAAE,UAAU,SAAS,EACtC,KAAI,aAAa,EAAE,UAAU,KAAK,QAAQ;IACxC,IAAI,GAAG;IACP,MAAM;IACN,UAAU;KACR,MAAM,aAAa,GAAG,KAAK;KAC3B,WACE,OAAO,GAAG,SAAS,WACf,GAAG,OACH,KAAK,UAAU,GAAG,QAAQ,EAAE,CAAC;KACpC;IACD,GAAI,GAAG,mBACH,EAAE,kBAAkB,GAAG,kBAAkB,GACzC,EAAE;IACP,EAAE;AAGL,UAAO;IACP;;CAGJ,AAAQ,WACN,aACA,YACc;AACd,SAAO,YAAY,KAAK,SAAS;GAC/B,MAAM;GACN,UAAU;IACR,MAAM,WAAW,IAAI,IAAI,KAAK,IAAI,IAAI;IACtC,aAAa,IAAI;IACjB,YAAY,IAAI;IACjB;GACF,EAAE;;;;;;;;;;AAeP,SAAgB,mBACd,MACwC;CACxC,MAAM,UAAU,KAAK,MAAM;CAE3B,MAAM,aAAa,2BAA2B,QAAQ;AACtD,KAAI,WAAW,SAAS,EAAG,QAAO;CAElC,MAAM,WAAW,6BAA6B,QAAQ;AACtD,KAAI,SAAS,SAAS,EAAG,QAAO;AAEhC,QAAO,EAAE;;AAGX,SAAS,oBAAoB,OAK3B;AACA,KAAI,CAAC,SAAS,MAAM,CAAE,QAAO;AAC7B,QAAO,OAAO,MAAM,SAAS;;AAG/B,SAAS,2BACP,MACwC;CACxC,MAAM,QAAQ,0BAA0B,KAAK;AAC7C,KAAI,CAAC,MAAO,QAAO,EAAE;AAErB,KAAI;EACF,MAAM,SAAkB,KAAK,MAAM,MAAM;AACzC,MAAI,CAAC,MAAM,QAAQ,OAAO,CAAE,QAAO,EAAE;AAErC,SAAO,OAAO,OAAO,oBAAoB,CAAC,KAAK,UAAU;GACvD,MAAM,KAAK;GACX,MAAM,KAAK,cAAc,KAAK,aAAa,KAAK,QAAQ,EAAE;GAC3D,EAAE;SACG;AACN,SAAO,EAAE;;;AAIb,SAAS,6BACP,MACwC;AACxC,KAAI,KAAK,SAAS,kCAChB,QAAO,EAAE;CAGX,MAAM,UAAU;CAChB,MAAM,UAAkD,EAAE;AAE1D,MAAK,MAAM,SAAS,KAAK,SAAS,QAAQ,EAAE;EAC1C,MAAM,OAAO,MAAM;EACnB,MAAM,UAAU,MAAM;EAEtB,MAAM,OAAgC,EAAE;AAExC,OAAK,MAAM,YAAY,QAAQ,SADZ,6CACgC,EAAE;GACnD,MAAM,MAAM,SAAS;AAErB,QAAK,OADS,SAAS,MAAM,SAAS,MAAM,SAAS;;AAIvD,UAAQ,KAAK;GAAE;GAAM;GAAM,CAAC;;AAG9B,QAAO"}

package/dist/appkit/package.js

@@ -1,6 +1,6 @@
 //#region package.json
 var name = "@databricks/appkit";
-var version = "0.42.0";
+var version = "0.43.0";
 
 //#endregion
 export { name, version };

package/dist/cache/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","names":[],"sources":["../../src/cache/index.ts"],"mappings":";;;;;;;AAmDA;;;;;;;;;;;cAAa,YAAA;EAAA,wBACa,uBAAA;EAAA,wBACA,qBAAA;EAAA,iBACP,IAAA;EAAA,eACF,QAAA;EAAA,eACA,WAAA;EAAA,QAEP,OAAA;EAAA,QACA,MAAA;EAAA,QACA,gBAAA;EAAA,QACA,iBAAA;EAAA,QACA,kBAAA;EAAA,QAEA,SAAA;EAAA,QACA,gBAAA;EAAA,QAKD,WAAA,CAAA;EAlBiB;;;;;;EAAA,OA+CjB,eAAA,CAAA,GAAmB,YAAA;EAvClB;;;;;;;EAAA,OAyDK,WAAA,CACX,UAAA,GAAa,OAAA,CAAQ,WAAA,IACpB,OAAA,CAAQ,YAAA;EAFE;;;;;;;;;;;;EAAA,eA+BQ,MAAA;EAyEyB;;;;;;;;;;;;;;;;;;;;EAFxC,YAAA,GAAA,CACJ,GAAA,gCACA,EAAA,GAAK,YAAA,GAAe,WAAA,KAAgB,OAAA,CAAQ,CAAA,GAC5C,OAAA,UACA,OAAA;IAAY,GAAA;IAAc,YAAA,GAAe,WAAA;EAAA,IACxC,OAAA,CAAQ,CAAA;EAkRR;;;;;;;EAAA,QA7HK,iBAAA;EA6JgB;;;;;EAhGlB,GAAA,GAAA,CAAO,GAAA,WAAc,OAAA,CAAQ,CAAA;EA4H7B;;;;;;;EAAA,QA3GQ,QAAA;;UAYN,YAAA;;;;;;;;EA+BF,GAAA,GAAA,CACJ,GAAA,UACA,KAAA,EAAO,CAAA,EACP,OAAA;IAAY,GAAA;EAAA,IACX,OAAA;;;;;;EAaG,MAAA,CAAO,GAAA,WAAc,OAAA;;EAMrB,KAAA,CAAA,GAAS,OAAA;;;;;;EAaT,GAAA,CAAI,GAAA,WAAc,OAAA;;;;;;;EAaxB,WAAA,CAAY,KAAA,gCAAqC,OAAA;;EAO3C,KAAA,CAAA,GAAS,OAAA;;;;;EAQT,gBAAA,CAAA,GAAoB,OAAA;AAAA"}
+{"version":3,"file":"index.d.ts","names":[],"sources":["../../src/cache/index.ts"],"mappings":";;;;;;;AAoDA;;;;;;;;;;;cAAa,YAAA;EAAA,wBACa,uBAAA;EAAA,wBACA,qBAAA;EAAA,iBACP,IAAA;EAAA,eACF,QAAA;EAAA,eACA,WAAA;EAAA,QAEP,OAAA;EAAA,QACA,MAAA;EAAA,QACA,gBAAA;EAAA,QACA,iBAAA;EAAA,QACA,kBAAA;EAAA,QAEA,SAAA;EAAA,QACA,gBAAA;EAAA,QAKD,WAAA,CAAA;EAlBiB;;;;;;EAAA,OA+CjB,eAAA,CAAA,GAAmB,YAAA;EAvClB;;;;;;;EAAA,OAyDK,WAAA,CACX,UAAA,GAAa,OAAA,CAAQ,WAAA,IACpB,OAAA,CAAQ,YAAA;EAFE;;;;;;;;;;;;EAAA,eA+BQ,MAAA;EAyEyB;;;;;;;;;;;;;;;;;;;;EAFxC,YAAA,GAAA,CACJ,GAAA,gCACA,EAAA,GAAK,YAAA,GAAe,WAAA,KAAgB,OAAA,CAAQ,CAAA,GAC5C,OAAA,UACA,OAAA;IAAY,GAAA;IAAc,YAAA,GAAe,WAAA;EAAA,IACxC,OAAA,CAAQ,CAAA;EAkRR;;;;;;;EAAA,QA7HK,iBAAA;EA6JgB;;;;;EAhGlB,GAAA,GAAA,CAAO,GAAA,WAAc,OAAA,CAAQ,CAAA;EA4H7B;;;;;;;EAAA,QA3GQ,QAAA;;UAYN,YAAA;;;;;;;;EA+BF,GAAA,GAAA,CACJ,GAAA,UACA,KAAA,EAAO,CAAA,EACP,OAAA;IAAY,GAAA;EAAA,IACX,OAAA;;;;;;EAaG,MAAA,CAAO,GAAA,WAAc,OAAA;;EAMrB,KAAA,CAAA,GAAS,OAAA;;;;;;EAaT,GAAA,CAAI,GAAA,WAAc,OAAA;;;;;;;EAaxB,WAAA,CAAY,KAAA,gCAAqC,OAAA;;EAO3C,KAAA,CAAA,GAAS,OAAA;;;;;EAQT,gBAAA,CAAA,GAAoB,OAAA;AAAA"}

package/dist/cache/index.js

@@ -3,6 +3,7 @@ import { AppKitError } from "../errors/base.js";
 import { ExecutionError } from "../errors/execution.js";
 import { InitializationError } from "../errors/initialization.js";
 import "../errors/index.js";
+import { getClientOptions } from "../context/client-options.js";
 import { createLakebasePool } from "../connectors/lakebase/index.js";
 import { TelemetryManager } from "../telemetry/telemetry-manager.js";
 import { SpanStatusCode } from "../telemetry/index.js";
@@ -116,7 +117,7 @@ var CacheManager = class CacheManager {
 			return new CacheManager(new InMemoryStorage(config), config);
 		}
 		try {
-			const pool = createLakebasePool({ workspaceClient: new WorkspaceClient({}) });
+			const pool = createLakebasePool({ workspaceClient: new WorkspaceClient({}, getClientOptions()) });
 			const persistentStorage = new PersistentStorage(config, pool);
 			if (await persistentStorage.healthCheck()) {
 				await persistentStorage.initialize();

package/dist/cache/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../../src/cache/index.ts"],"sourcesContent":["import { createHash } from \"node:crypto\";\nimport { ApiError, WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport type { CacheConfig, CacheEntry, CacheStorage } from \"shared\";\nimport { createLakebasePool } from \"../connectors/lakebase\";\nimport { AppKitError, ExecutionError, InitializationError } from \"../errors\";\nimport { createLogger } from \"../logging/logger\";\nimport type { Counter, TelemetryProvider } from \"../telemetry\";\nimport { SpanStatusCode, TelemetryManager } from \"../telemetry\";\nimport { deepMerge } from \"../utils\";\nimport { cacheDefaults } from \"./defaults\";\nimport { InMemoryStorage, PersistentStorage } from \"./storage\";\n\nconst logger = createLogger(\"cache\");\n\n/**\n * Reference-counted in-flight cache execution entry.\n *\n * `sharedController` decouples the cached `fn()` from any single caller's\n * abort signal. Callers join an in-flight entry by incrementing `refCount`;\n * when a caller aborts, refCount is decremented. The shared controller is\n * aborted only when refCount drops to 0 — i.e. all callers have abandoned\n * the request. This prevents one caller's cancellation (e.g. React\n * StrictMode unmount) from poisoning the in-flight result for other still-\n * connected awaiters.\n */\ninterface InFlightEntry<T> {\n  promise: Promise<T>;\n  refCount: number;\n  sharedController: AbortController;\n  abortTimer?: ReturnType<typeof setTimeout>;\n}\n\nfunction createAbortError(signal: AbortSignal): unknown {\n  if (signal.reason !== undefined) return signal.reason;\n  return new DOMException(\"The operation was aborted.\", \"AbortError\");\n}\n\n/**\n * Cache manager class to handle cache operations.\n * Can be used with in-memory storage or persistent storage (Lakebase).\n *\n * The cache is automatically initialized by AppKit. Use `getInstanceSync()` to access\n * the singleton instance after initialization.\n *\n * @internal\n * @example\n * ```typescript\n * const cache = CacheManager.getInstanceSync();\n * const result = await cache.getOrExecute([\"users\", userId], () => fetchUser(userId), userKey);\n * ```\n */\nexport class CacheManager {\n  private static readonly MIN_CLEANUP_INTERVAL_MS = 60_000;\n  private static readonly ABORT_GRACE_PERIOD_MS = 100;\n  private readonly name: string = \"cache-manager\";\n  private static instance: CacheManager | null = null;\n  private static initPromise: Promise<CacheManager> | null = null;\n\n  private storage: CacheStorage;\n  private config: CacheConfig;\n  private inFlightRequests: Map<string, InFlightEntry<unknown>>;\n  private cleanupInProgress: boolean;\n  private lastCleanupAttempt: number;\n\n  private telemetry: TelemetryProvider;\n  private telemetryMetrics: {\n    cacheHitCount: Counter;\n    cacheMissCount: Counter;\n  };\n\n  private constructor(storage: CacheStorage, config: CacheConfig) {\n    this.storage = storage;\n    this.config = config;\n    this.inFlightRequests = new Map();\n    this.cleanupInProgress = false;\n    this.lastCleanupAttempt = 0;\n\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      cacheHitCount: this.telemetry.getMeter().createCounter(\"cache.hit\", {\n        description: \"Total number of cache hits\",\n        unit: \"1\",\n      }),\n      cacheMissCount: this.telemetry.getMeter().createCounter(\"cache.miss\", {\n        description: \"Total number of cache misses\",\n        unit: \"1\",\n      }),\n    };\n  }\n\n  /**\n   * Get the singleton instance of the cache manager (sync version).\n   *\n   * Throws if not initialized - ensure AppKit.create() has completed first.\n   * @returns CacheManager instance\n   */\n  static getInstanceSync(): CacheManager {\n    if (!CacheManager.instance) {\n      throw InitializationError.notInitialized(\n        \"CacheManager\",\n        \"Ensure AppKit.create() has completed before accessing the cache\",\n      );\n    }\n\n    return CacheManager.instance;\n  }\n\n  /**\n   * Initialize and get the singleton instance of the cache manager.\n   * Called internally by AppKit - prefer `getInstanceSync()` for plugin access.\n   * @param userConfig - User configuration for the cache manager\n   * @returns CacheManager instance\n   * @internal\n   */\n  static async getInstance(\n    userConfig?: Partial<CacheConfig>,\n  ): Promise<CacheManager> {\n    if (CacheManager.instance) {\n      return CacheManager.instance;\n    }\n\n    if (!CacheManager.initPromise) {\n      CacheManager.initPromise = CacheManager.create(userConfig).then(\n        (instance) => {\n          CacheManager.instance = instance;\n          return instance;\n        },\n      );\n    }\n\n    return CacheManager.initPromise;\n  }\n\n  /**\n   * Create a new cache manager instance\n   *\n   * Storage selection logic:\n   * 1. If `storage` provided and healthy → use provided storage\n   * 2. If `storage` provided but unhealthy → fallback to InMemory (or disable if strictPersistence)\n   * 3. If no `storage` provided and Lakebase available → use Lakebase\n   * 4. If no `storage` provided and Lakebase unavailable → fallback to InMemory (or disable if strictPersistence)\n   *\n   * @param userConfig - User configuration for the cache manager\n   * @returns CacheManager instance\n   */\n  private static async create(\n    userConfig?: Partial<CacheConfig>,\n  ): Promise<CacheManager> {\n    const config = deepMerge(cacheDefaults, userConfig);\n\n    if (config.storage) {\n      const isHealthy = await config.storage.healthCheck();\n      if (isHealthy) {\n        return new CacheManager(config.storage, config);\n      }\n\n      if (config.strictPersistence) {\n        const disabledConfig = { ...config, enabled: false };\n        return new CacheManager(\n          new InMemoryStorage(disabledConfig),\n          disabledConfig,\n        );\n      }\n\n      return new CacheManager(new InMemoryStorage(config), config);\n    }\n\n    // try to use lakebase storage\n    try {\n      const workspaceClient = new WorkspaceClient({});\n      const pool = createLakebasePool({ workspaceClient });\n      const persistentStorage = new PersistentStorage(config, pool);\n\n      const isHealthy = await persistentStorage.healthCheck();\n      if (isHealthy) {\n        await persistentStorage.initialize();\n        return new CacheManager(persistentStorage, config);\n      }\n\n      // Health check failed, close the pool and fallback\n      await pool.end();\n    } catch {\n      // lakebase unavailable, continue with in-memory storage\n    }\n\n    if (config.strictPersistence) {\n      const disabledConfig = { ...config, enabled: false };\n      return new CacheManager(\n        new InMemoryStorage(disabledConfig),\n        disabledConfig,\n      );\n    }\n\n    return new CacheManager(new InMemoryStorage(config), config);\n  }\n\n  /**\n   * Get or execute a function and cache the result.\n   *\n   * Multiple concurrent callers with the same `cacheKey` are deduplicated\n   * onto a single in-flight execution. Each caller may pass its own\n   * `callerSignal`; the underlying `fn()` is run with a shared, internally\n   * managed `AbortSignal` that aborts only when *all* callers have\n   * abandoned the request (reference counted). This decouples a single\n   * caller's cancellation (e.g. React StrictMode unmount) from the shared\n   * result, so other still-connected callers receive the cached value\n   * normally.\n   *\n   * @param key - Cache key parts\n   * @param fn - Function to execute. Receives the cache-owned shared signal;\n   *   pass it through to the underlying I/O so the work is cancelled when\n   *   no caller is left waiting.\n   * @param userKey - User key for cache namespacing\n   * @param options - Options for the cache\n   * @returns Promise of the result\n   */\n  async getOrExecute<T>(\n    key: (string | number | object)[],\n    fn: (sharedSignal?: AbortSignal) => Promise<T>,\n    userKey: string,\n    options?: { ttl?: number; callerSignal?: AbortSignal },\n  ): Promise<T> {\n    if (!this.config.enabled) return fn(options?.callerSignal);\n\n    const callerSignal = options?.callerSignal;\n    if (callerSignal?.aborted) {\n      throw createAbortError(callerSignal);\n    }\n\n    const cacheKey = this.generateKey(key, userKey);\n\n    return this.telemetry.startActiveSpan(\n      \"cache.getOrExecute\",\n      {\n        attributes: {\n          \"cache.key\": cacheKey,\n          \"cache.enabled\": this.config.enabled,\n          \"cache.persistent\": this.storage.isPersistent(),\n        },\n      },\n      async (span) => {\n        try {\n          const cached = await this.getValid<T>(cacheKey);\n          if (cached !== null) {\n            span.setAttribute(\"cache.hit\", true);\n            span.setStatus({ code: SpanStatusCode.OK });\n            this.telemetryMetrics.cacheHitCount.add(1, {\n              \"cache.key\": cacheKey,\n            });\n\n            logger.event()?.setExecution({\n              cache_hit: true,\n              cache_key: cacheKey,\n            });\n\n            return cached.value;\n          }\n\n          // check if the value is being processed by another request — join\n          // the existing in-flight entry under reference counting so this\n          // caller's abort doesn't poison the shared result.\n          const existing = this.inFlightRequests.get(cacheKey) as\n            | InFlightEntry<T>\n            | undefined;\n          if (existing && !existing.sharedController.signal.aborted) {\n            existing.refCount++;\n            // Cancel any pending abort timer — a new caller has joined\n            if (existing.abortTimer) {\n              clearTimeout(existing.abortTimer);\n              existing.abortTimer = undefined;\n            }\n            span.setAttribute(\"cache.hit\", true);\n            span.setAttribute(\"cache.deduplication\", true);\n            span.addEvent(\"cache.deduplication_used\", {\n              \"cache.key\": cacheKey,\n            });\n            this.telemetryMetrics.cacheHitCount.add(1, {\n              \"cache.key\": cacheKey,\n              \"cache.deduplication\": \"true\",\n            });\n\n            logger.event()?.setExecution({\n              cache_hit: true,\n              cache_key: cacheKey,\n              cache_deduplication: true,\n            });\n\n            return await this._waitWithRefCount(existing, callerSignal);\n          }\n\n          // cache miss - execute function under a shared abort controller\n          span.setAttribute(\"cache.hit\", false);\n          span.addEvent(\"cache.miss\", { \"cache.key\": cacheKey });\n          this.telemetryMetrics.cacheMissCount.add(1, {\n            \"cache.key\": cacheKey,\n          });\n\n          logger.event()?.setExecution({\n            cache_hit: false,\n            cache_key: cacheKey,\n          });\n\n          const sharedController = new AbortController();\n          const entry: InFlightEntry<T> = {\n            promise: undefined as unknown as Promise<T>,\n            refCount: 1,\n            sharedController,\n          };\n\n          entry.promise = fn(sharedController.signal)\n            .then(async (result) => {\n              await this.set(cacheKey, result, options);\n              span.addEvent(\"cache.value_stored\", {\n                \"cache.key\": cacheKey,\n                \"cache.ttl\": options?.ttl ?? this.config.ttl ?? 3600,\n              });\n              return result;\n            })\n            .catch((error) => {\n              span.recordException(error);\n              span.setStatus({ code: SpanStatusCode.ERROR });\n              // If the shared controller aborted, all callers have already\n              // abandoned the request (or are about to via their own signals)\n              // — propagate the original error without wrapping. No live\n              // awaiter will observe this rejection.\n              if (sharedController.signal.aborted) {\n                throw error;\n              }\n              if (error instanceof AppKitError || error instanceof ApiError) {\n                throw error;\n              }\n              throw ExecutionError.statementFailed(\n                error instanceof Error ? error.message : String(error),\n              );\n            })\n            .finally(() => {\n              if (this.inFlightRequests.get(cacheKey) === entry) {\n                this.inFlightRequests.delete(cacheKey);\n              }\n            });\n\n          // Suppress unhandled rejection warnings when every caller bailed\n          // before fn() resolved (their own promises rejected via\n          // waitWithRefCount; the underlying entry.promise has no awaiter).\n          entry.promise.catch(() => {});\n\n          this.inFlightRequests.set(cacheKey, entry as InFlightEntry<unknown>);\n\n          const result = await this._waitWithRefCount(entry, callerSignal);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n          throw error;\n        } finally {\n          span.end();\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  /**\n   * Wait on an in-flight entry, racing the underlying promise against the\n   * caller's abort signal. When the caller aborts, the entry's refCount is\n   * decremented; if it hits zero the shared controller is aborted so the\n   * underlying `fn()` can stop. Other callers continue to await the same\n   * entry and receive the result when it arrives.\n   */\n  private _waitWithRefCount<T>(\n    entry: InFlightEntry<T>,\n    callerSignal?: AbortSignal,\n  ): Promise<T> {\n    if (!callerSignal) return entry.promise;\n\n    return new Promise<T>((resolve, reject) => {\n      let settled = false;\n\n      const release = () => {\n        if (entry.refCount > 0) entry.refCount--;\n        if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n          // Grace period: delay abort so a StrictMode remount can join\n          // the in-flight entry before the shared execution is cancelled.\n          entry.abortTimer = setTimeout(() => {\n            if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n              entry.sharedController.abort(\n                callerSignal.reason ?? \"all cache callers aborted\",\n              );\n            }\n          }, CacheManager.ABORT_GRACE_PERIOD_MS);\n        }\n      };\n\n      const onAbort = () => {\n        if (settled) return;\n        settled = true;\n        callerSignal.removeEventListener(\"abort\", onAbort);\n        release();\n        reject(createAbortError(callerSignal));\n      };\n\n      if (callerSignal.aborted) {\n        onAbort();\n        return;\n      }\n\n      callerSignal.addEventListener(\"abort\", onAbort, { once: true });\n\n      entry.promise.then(\n        (value) => {\n          if (settled) return;\n          settled = true;\n          callerSignal.removeEventListener(\"abort\", onAbort);\n          resolve(value);\n        },\n        (error) => {\n          if (settled) return;\n          settled = true;\n          callerSignal.removeEventListener(\"abort\", onAbort);\n          reject(error);\n        },\n      );\n    });\n  }\n\n  /**\n   * Get a cached value\n   * @param key - Cache key\n   * @returns Promise of the value or null if not found or expired\n   */\n  async get<T>(key: string): Promise<T | null> {\n    if (!this.config.enabled) return null;\n\n    // probabilistic cleanup trigger\n    this.maybeCleanup();\n\n    const entry = await this.getValid<T>(key);\n    return entry?.value ?? null;\n  }\n\n  /**\n   * Get a cached entry only if it has not expired.\n   * Returns null on miss or expired (and deletes the expired entry).\n   *\n   * Storage implementations return entries unconditionally — expiry handling\n   * lives at the CacheManager layer.\n   */\n  private async getValid<T>(key: string): Promise<CacheEntry<T> | null> {\n    const entry = await this.storage.get<T>(key);\n    if (!entry) return null;\n\n    if (Date.now() > entry.expiry) {\n      await this.storage.delete(key);\n      return null;\n    }\n    return entry;\n  }\n\n  /** Probabilistically trigger cleanup of expired entries (fire-and-forget) */\n  private maybeCleanup(): void {\n    if (this.cleanupInProgress) return;\n    if (!this.storage.isPersistent()) return;\n    const now = Date.now();\n    if (now - this.lastCleanupAttempt < CacheManager.MIN_CLEANUP_INTERVAL_MS)\n      return;\n\n    const probability = this.config.cleanupProbability ?? 0.01;\n\n    if (Math.random() > probability) return;\n\n    this.lastCleanupAttempt = now;\n\n    this.cleanupInProgress = true;\n    (this.storage as PersistentStorage)\n      .cleanupExpired()\n      .catch((error) => {\n        logger.debug(\"Error cleaning up expired entries: %O\", error);\n      })\n      .finally(() => {\n        this.cleanupInProgress = false;\n      });\n  }\n\n  /**\n   * Set a value in the cache\n   * @param key - Cache key\n   * @param value - Value to set\n   * @param options - Options for the cache\n   * @returns Promise of the result\n   */\n  async set<T>(\n    key: string,\n    value: T,\n    options?: { ttl?: number },\n  ): Promise<void> {\n    if (!this.config.enabled) return;\n\n    const ttl = options?.ttl ?? this.config.ttl ?? 3600;\n    const expiryTime = Date.now() + ttl * 1000;\n    await this.storage.set(key, { value, expiry: expiryTime });\n  }\n\n  /**\n   * Delete a value from the cache\n   * @param key - Cache key\n   * @returns Promise of the result\n   */\n  async delete(key: string): Promise<void> {\n    if (!this.config.enabled) return;\n    await this.storage.delete(key);\n  }\n\n  /** Clear the cache */\n  async clear(): Promise<void> {\n    await this.storage.clear();\n    for (const entry of this.inFlightRequests.values()) {\n      if (entry.abortTimer) clearTimeout(entry.abortTimer);\n    }\n    this.inFlightRequests.clear();\n  }\n\n  /**\n   * Check if a value exists in the cache\n   * @param key - Cache key\n   * @returns Promise of true if the value exists, false otherwise\n   */\n  async has(key: string): Promise<boolean> {\n    if (!this.config.enabled) return false;\n\n    const entry = await this.getValid(key);\n    return entry !== null;\n  }\n\n  /**\n   * Generate a cache key\n   * @param parts - Parts of the key\n   * @param userKey - User key\n   * @returns Cache key\n   */\n  generateKey(parts: (string | number | object)[], userKey: string): string {\n    const allParts = [userKey, ...parts];\n    const serialized = JSON.stringify(allParts);\n    return createHash(\"sha256\").update(serialized).digest(\"hex\");\n  }\n\n  /** Close the cache */\n  async close(): Promise<void> {\n    await this.storage.close();\n  }\n\n  /**\n   * Check if the storage is healthy\n   * @returns Promise of true if the storage is healthy, false otherwise\n   */\n  async isStorageHealthy(): Promise<boolean> {\n    return this.storage.healthCheck();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAYA,MAAM,SAAS,aAAa,QAAQ;AAoBpC,SAAS,iBAAiB,QAA8B;AACtD,KAAI,OAAO,WAAW,OAAW,QAAO,OAAO;AAC/C,QAAO,IAAI,aAAa,8BAA8B,aAAa;;;;;;;;;;;;;;;;AAiBrE,IAAa,eAAb,MAAa,aAAa;CACxB,OAAwB,0BAA0B;CAClD,OAAwB,wBAAwB;CAChD,AAAiB,OAAe;CAChC,OAAe,WAAgC;CAC/C,OAAe,cAA4C;CAE3D,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,AAAQ;CACR,AAAQ;CAKR,AAAQ,YAAY,SAAuB,QAAqB;AAC9D,OAAK,UAAU;AACf,OAAK,SAAS;AACd,OAAK,mCAAmB,IAAI,KAAK;AACjC,OAAK,oBAAoB;AACzB,OAAK,qBAAqB;AAE1B,OAAK,YAAY,iBAAiB,YAChC,KAAK,MACL,KAAK,OAAO,UACb;AACD,OAAK,mBAAmB;GACtB,eAAe,KAAK,UAAU,UAAU,CAAC,cAAc,aAAa;IAClE,aAAa;IACb,MAAM;IACP,CAAC;GACF,gBAAgB,KAAK,UAAU,UAAU,CAAC,cAAc,cAAc;IACpE,aAAa;IACb,MAAM;IACP,CAAC;GACH;;;;;;;;CASH,OAAO,kBAAgC;AACrC,MAAI,CAAC,aAAa,SAChB,OAAM,oBAAoB,eACxB,gBACA,kEACD;AAGH,SAAO,aAAa;;;;;;;;;CAUtB,aAAa,YACX,YACuB;AACvB,MAAI,aAAa,SACf,QAAO,aAAa;AAGtB,MAAI,CAAC,aAAa,YAChB,cAAa,cAAc,aAAa,OAAO,WAAW,CAAC,MACxD,aAAa;AACZ,gBAAa,WAAW;AACxB,UAAO;IAEV;AAGH,SAAO,aAAa;;;;;;;;;;;;;;CAetB,aAAqB,OACnB,YACuB;EACvB,MAAM,SAAS,UAAU,eAAe,WAAW;AAEnD,MAAI,OAAO,SAAS;AAElB,OADkB,MAAM,OAAO,QAAQ,aAAa,CAElD,QAAO,IAAI,aAAa,OAAO,SAAS,OAAO;AAGjD,OAAI,OAAO,mBAAmB;IAC5B,MAAM,iBAAiB;KAAE,GAAG;KAAQ,SAAS;KAAO;AACpD,WAAO,IAAI,aACT,IAAI,gBAAgB,eAAe,EACnC,eACD;;AAGH,UAAO,IAAI,aAAa,IAAI,gBAAgB,OAAO,EAAE,OAAO;;AAI9D,MAAI;GAEF,MAAM,OAAO,mBAAmB,EAAE,iBADV,IAAI,gBAAgB,EAAE,CAAC,EACI,CAAC;GACpD,MAAM,oBAAoB,IAAI,kBAAkB,QAAQ,KAAK;AAG7D,OADkB,MAAM,kBAAkB,aAAa,EACxC;AACb,UAAM,kBAAkB,YAAY;AACpC,WAAO,IAAI,aAAa,mBAAmB,OAAO;;AAIpD,SAAM,KAAK,KAAK;UACV;AAIR,MAAI,OAAO,mBAAmB;GAC5B,MAAM,iBAAiB;IAAE,GAAG;IAAQ,SAAS;IAAO;AACpD,UAAO,IAAI,aACT,IAAI,gBAAgB,eAAe,EACnC,eACD;;AAGH,SAAO,IAAI,aAAa,IAAI,gBAAgB,OAAO,EAAE,OAAO;;;;;;;;;;;;;;;;;;;;;;CAuB9D,MAAM,aACJ,KACA,IACA,SACA,SACY;AACZ,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO,GAAG,SAAS,aAAa;EAE1D,MAAM,eAAe,SAAS;AAC9B,MAAI,cAAc,QAChB,OAAM,iBAAiB,aAAa;EAGtC,MAAM,WAAW,KAAK,YAAY,KAAK,QAAQ;AAE/C,SAAO,KAAK,UAAU,gBACpB,sBACA,EACE,YAAY;GACV,aAAa;GACb,iBAAiB,KAAK,OAAO;GAC7B,oBAAoB,KAAK,QAAQ,cAAc;GAChD,EACF,EACD,OAAO,SAAS;AACd,OAAI;IACF,MAAM,SAAS,MAAM,KAAK,SAAY,SAAS;AAC/C,QAAI,WAAW,MAAM;AACnB,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,UAAK,iBAAiB,cAAc,IAAI,GAAG,EACzC,aAAa,UACd,CAAC;AAEF,YAAO,OAAO,EAAE,aAAa;MAC3B,WAAW;MACX,WAAW;MACZ,CAAC;AAEF,YAAO,OAAO;;IAMhB,MAAM,WAAW,KAAK,iBAAiB,IAAI,SAAS;AAGpD,QAAI,YAAY,CAAC,SAAS,iBAAiB,OAAO,SAAS;AACzD,cAAS;AAET,SAAI,SAAS,YAAY;AACvB,mBAAa,SAAS,WAAW;AACjC,eAAS,aAAa;;AAExB,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,aAAa,uBAAuB,KAAK;AAC9C,UAAK,SAAS,4BAA4B,EACxC,aAAa,UACd,CAAC;AACF,UAAK,iBAAiB,cAAc,IAAI,GAAG;MACzC,aAAa;MACb,uBAAuB;MACxB,CAAC;AAEF,YAAO,OAAO,EAAE,aAAa;MAC3B,WAAW;MACX,WAAW;MACX,qBAAqB;MACtB,CAAC;AAEF,YAAO,MAAM,KAAK,kBAAkB,UAAU,aAAa;;AAI7D,SAAK,aAAa,aAAa,MAAM;AACrC,SAAK,SAAS,cAAc,EAAE,aAAa,UAAU,CAAC;AACtD,SAAK,iBAAiB,eAAe,IAAI,GAAG,EAC1C,aAAa,UACd,CAAC;AAEF,WAAO,OAAO,EAAE,aAAa;KAC3B,WAAW;KACX,WAAW;KACZ,CAAC;IAEF,MAAM,mBAAmB,IAAI,iBAAiB;IAC9C,MAAM,QAA0B;KAC9B,SAAS;KACT,UAAU;KACV;KACD;AAED,UAAM,UAAU,GAAG,iBAAiB,OAAO,CACxC,KAAK,OAAO,WAAW;AACtB,WAAM,KAAK,IAAI,UAAU,QAAQ,QAAQ;AACzC,UAAK,SAAS,sBAAsB;MAClC,aAAa;MACb,aAAa,SAAS,OAAO,KAAK,OAAO,OAAO;MACjD,CAAC;AACF,YAAO;MACP,CACD,OAAO,UAAU;AAChB,UAAK,gBAAgB,MAAM;AAC3B,UAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAK9C,SAAI,iBAAiB,OAAO,QAC1B,OAAM;AAER,SAAI,iBAAiB,eAAe,iBAAiB,SACnD,OAAM;AAER,WAAM,eAAe,gBACnB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;MACD,CACD,cAAc;AACb,SAAI,KAAK,iBAAiB,IAAI,SAAS,KAAK,MAC1C,MAAK,iBAAiB,OAAO,SAAS;MAExC;AAKJ,UAAM,QAAQ,YAAY,GAAG;AAE7B,SAAK,iBAAiB,IAAI,UAAU,MAAgC;IAEpE,MAAM,SAAS,MAAM,KAAK,kBAAkB,OAAO,aAAa;AAChE,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,UAAM;aACE;AACR,SAAK,KAAK;;KAGd;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;;;;;;;;CAUH,AAAQ,kBACN,OACA,cACY;AACZ,MAAI,CAAC,aAAc,QAAO,MAAM;AAEhC,SAAO,IAAI,SAAY,SAAS,WAAW;GACzC,IAAI,UAAU;GAEd,MAAM,gBAAgB;AACpB,QAAI,MAAM,WAAW,EAAG,OAAM;AAC9B,QAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QAGxD,OAAM,aAAa,iBAAiB;AAClC,SAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QACxD,OAAM,iBAAiB,MACrB,aAAa,UAAU,4BACxB;OAEF,aAAa,sBAAsB;;GAI1C,MAAM,gBAAgB;AACpB,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,aAAS;AACT,WAAO,iBAAiB,aAAa,CAAC;;AAGxC,OAAI,aAAa,SAAS;AACxB,aAAS;AACT;;AAGF,gBAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;AAE/D,SAAM,QAAQ,MACX,UAAU;AACT,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,YAAQ,MAAM;OAEf,UAAU;AACT,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,WAAO,MAAM;KAEhB;IACD;;;;;;;CAQJ,MAAM,IAAO,KAAgC;AAC3C,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO;AAGjC,OAAK,cAAc;AAGnB,UADc,MAAM,KAAK,SAAY,IAAI,GAC3B,SAAS;;;;;;;;;CAUzB,MAAc,SAAY,KAA4C;EACpE,MAAM,QAAQ,MAAM,KAAK,QAAQ,IAAO,IAAI;AAC5C,MAAI,CAAC,MAAO,QAAO;AAEnB,MAAI,KAAK,KAAK,GAAG,MAAM,QAAQ;AAC7B,SAAM,KAAK,QAAQ,OAAO,IAAI;AAC9B,UAAO;;AAET,SAAO;;;CAIT,AAAQ,eAAqB;AAC3B,MAAI,KAAK,kBAAmB;AAC5B,MAAI,CAAC,KAAK,QAAQ,cAAc,CAAE;EAClC,MAAM,MAAM,KAAK,KAAK;AACtB,MAAI,MAAM,KAAK,qBAAqB,aAAa,wBAC/C;EAEF,MAAM,cAAc,KAAK,OAAO,sBAAsB;AAEtD,MAAI,KAAK,QAAQ,GAAG,YAAa;AAEjC,OAAK,qBAAqB;AAE1B,OAAK,oBAAoB;AACzB,EAAC,KAAK,QACH,gBAAgB,CAChB,OAAO,UAAU;AAChB,UAAO,MAAM,yCAAyC,MAAM;IAC5D,CACD,cAAc;AACb,QAAK,oBAAoB;IACzB;;;;;;;;;CAUN,MAAM,IACJ,KACA,OACA,SACe;AACf,MAAI,CAAC,KAAK,OAAO,QAAS;EAE1B,MAAM,MAAM,SAAS,OAAO,KAAK,OAAO,OAAO;EAC/C,MAAM,aAAa,KAAK,KAAK,GAAG,MAAM;AACtC,QAAM,KAAK,QAAQ,IAAI,KAAK;GAAE;GAAO,QAAQ;GAAY,CAAC;;;;;;;CAQ5D,MAAM,OAAO,KAA4B;AACvC,MAAI,CAAC,KAAK,OAAO,QAAS;AAC1B,QAAM,KAAK,QAAQ,OAAO,IAAI;;;CAIhC,MAAM,QAAuB;AAC3B,QAAM,KAAK,QAAQ,OAAO;AAC1B,OAAK,MAAM,SAAS,KAAK,iBAAiB,QAAQ,CAChD,KAAI,MAAM,WAAY,cAAa,MAAM,WAAW;AAEtD,OAAK,iBAAiB,OAAO;;;;;;;CAQ/B,MAAM,IAAI,KAA+B;AACvC,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO;AAGjC,SADc,MAAM,KAAK,SAAS,IAAI,KACrB;;;;;;;;CASnB,YAAY,OAAqC,SAAyB;EACxE,MAAM,WAAW,CAAC,SAAS,GAAG,MAAM;EACpC,MAAM,aAAa,KAAK,UAAU,SAAS;AAC3C,SAAO,WAAW,SAAS,CAAC,OAAO,WAAW,CAAC,OAAO,MAAM;;;CAI9D,MAAM,QAAuB;AAC3B,QAAM,KAAK,QAAQ,OAAO;;;;;;CAO5B,MAAM,mBAAqC;AACzC,SAAO,KAAK,QAAQ,aAAa"}
+{"version":3,"file":"index.js","names":[],"sources":["../../src/cache/index.ts"],"sourcesContent":["import { createHash } from \"node:crypto\";\nimport { ApiError, WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport type { CacheConfig, CacheEntry, CacheStorage } from \"shared\";\nimport { createLakebasePool } from \"../connectors/lakebase\";\nimport { getClientOptions } from \"../context/client-options\";\nimport { AppKitError, ExecutionError, InitializationError } from \"../errors\";\nimport { createLogger } from \"../logging/logger\";\nimport type { Counter, TelemetryProvider } from \"../telemetry\";\nimport { SpanStatusCode, TelemetryManager } from \"../telemetry\";\nimport { deepMerge } from \"../utils\";\nimport { cacheDefaults } from \"./defaults\";\nimport { InMemoryStorage, PersistentStorage } from \"./storage\";\n\nconst logger = createLogger(\"cache\");\n\n/**\n * Reference-counted in-flight cache execution entry.\n *\n * `sharedController` decouples the cached `fn()` from any single caller's\n * abort signal. Callers join an in-flight entry by incrementing `refCount`;\n * when a caller aborts, refCount is decremented. The shared controller is\n * aborted only when refCount drops to 0 — i.e. all callers have abandoned\n * the request. This prevents one caller's cancellation (e.g. React\n * StrictMode unmount) from poisoning the in-flight result for other still-\n * connected awaiters.\n */\ninterface InFlightEntry<T> {\n  promise: Promise<T>;\n  refCount: number;\n  sharedController: AbortController;\n  abortTimer?: ReturnType<typeof setTimeout>;\n}\n\nfunction createAbortError(signal: AbortSignal): unknown {\n  if (signal.reason !== undefined) return signal.reason;\n  return new DOMException(\"The operation was aborted.\", \"AbortError\");\n}\n\n/**\n * Cache manager class to handle cache operations.\n * Can be used with in-memory storage or persistent storage (Lakebase).\n *\n * The cache is automatically initialized by AppKit. Use `getInstanceSync()` to access\n * the singleton instance after initialization.\n *\n * @internal\n * @example\n * ```typescript\n * const cache = CacheManager.getInstanceSync();\n * const result = await cache.getOrExecute([\"users\", userId], () => fetchUser(userId), userKey);\n * ```\n */\nexport class CacheManager {\n  private static readonly MIN_CLEANUP_INTERVAL_MS = 60_000;\n  private static readonly ABORT_GRACE_PERIOD_MS = 100;\n  private readonly name: string = \"cache-manager\";\n  private static instance: CacheManager | null = null;\n  private static initPromise: Promise<CacheManager> | null = null;\n\n  private storage: CacheStorage;\n  private config: CacheConfig;\n  private inFlightRequests: Map<string, InFlightEntry<unknown>>;\n  private cleanupInProgress: boolean;\n  private lastCleanupAttempt: number;\n\n  private telemetry: TelemetryProvider;\n  private telemetryMetrics: {\n    cacheHitCount: Counter;\n    cacheMissCount: Counter;\n  };\n\n  private constructor(storage: CacheStorage, config: CacheConfig) {\n    this.storage = storage;\n    this.config = config;\n    this.inFlightRequests = new Map();\n    this.cleanupInProgress = false;\n    this.lastCleanupAttempt = 0;\n\n    this.telemetry = TelemetryManager.getProvider(\n      this.name,\n      this.config.telemetry,\n    );\n    this.telemetryMetrics = {\n      cacheHitCount: this.telemetry.getMeter().createCounter(\"cache.hit\", {\n        description: \"Total number of cache hits\",\n        unit: \"1\",\n      }),\n      cacheMissCount: this.telemetry.getMeter().createCounter(\"cache.miss\", {\n        description: \"Total number of cache misses\",\n        unit: \"1\",\n      }),\n    };\n  }\n\n  /**\n   * Get the singleton instance of the cache manager (sync version).\n   *\n   * Throws if not initialized - ensure AppKit.create() has completed first.\n   * @returns CacheManager instance\n   */\n  static getInstanceSync(): CacheManager {\n    if (!CacheManager.instance) {\n      throw InitializationError.notInitialized(\n        \"CacheManager\",\n        \"Ensure AppKit.create() has completed before accessing the cache\",\n      );\n    }\n\n    return CacheManager.instance;\n  }\n\n  /**\n   * Initialize and get the singleton instance of the cache manager.\n   * Called internally by AppKit - prefer `getInstanceSync()` for plugin access.\n   * @param userConfig - User configuration for the cache manager\n   * @returns CacheManager instance\n   * @internal\n   */\n  static async getInstance(\n    userConfig?: Partial<CacheConfig>,\n  ): Promise<CacheManager> {\n    if (CacheManager.instance) {\n      return CacheManager.instance;\n    }\n\n    if (!CacheManager.initPromise) {\n      CacheManager.initPromise = CacheManager.create(userConfig).then(\n        (instance) => {\n          CacheManager.instance = instance;\n          return instance;\n        },\n      );\n    }\n\n    return CacheManager.initPromise;\n  }\n\n  /**\n   * Create a new cache manager instance\n   *\n   * Storage selection logic:\n   * 1. If `storage` provided and healthy → use provided storage\n   * 2. If `storage` provided but unhealthy → fallback to InMemory (or disable if strictPersistence)\n   * 3. If no `storage` provided and Lakebase available → use Lakebase\n   * 4. If no `storage` provided and Lakebase unavailable → fallback to InMemory (or disable if strictPersistence)\n   *\n   * @param userConfig - User configuration for the cache manager\n   * @returns CacheManager instance\n   */\n  private static async create(\n    userConfig?: Partial<CacheConfig>,\n  ): Promise<CacheManager> {\n    const config = deepMerge(cacheDefaults, userConfig);\n\n    if (config.storage) {\n      const isHealthy = await config.storage.healthCheck();\n      if (isHealthy) {\n        return new CacheManager(config.storage, config);\n      }\n\n      if (config.strictPersistence) {\n        const disabledConfig = { ...config, enabled: false };\n        return new CacheManager(\n          new InMemoryStorage(disabledConfig),\n          disabledConfig,\n        );\n      }\n\n      return new CacheManager(new InMemoryStorage(config), config);\n    }\n\n    // try to use lakebase storage\n    try {\n      const workspaceClient = new WorkspaceClient({}, getClientOptions());\n      const pool = createLakebasePool({ workspaceClient });\n      const persistentStorage = new PersistentStorage(config, pool);\n\n      const isHealthy = await persistentStorage.healthCheck();\n      if (isHealthy) {\n        await persistentStorage.initialize();\n        return new CacheManager(persistentStorage, config);\n      }\n\n      // Health check failed, close the pool and fallback\n      await pool.end();\n    } catch {\n      // lakebase unavailable, continue with in-memory storage\n    }\n\n    if (config.strictPersistence) {\n      const disabledConfig = { ...config, enabled: false };\n      return new CacheManager(\n        new InMemoryStorage(disabledConfig),\n        disabledConfig,\n      );\n    }\n\n    return new CacheManager(new InMemoryStorage(config), config);\n  }\n\n  /**\n   * Get or execute a function and cache the result.\n   *\n   * Multiple concurrent callers with the same `cacheKey` are deduplicated\n   * onto a single in-flight execution. Each caller may pass its own\n   * `callerSignal`; the underlying `fn()` is run with a shared, internally\n   * managed `AbortSignal` that aborts only when *all* callers have\n   * abandoned the request (reference counted). This decouples a single\n   * caller's cancellation (e.g. React StrictMode unmount) from the shared\n   * result, so other still-connected callers receive the cached value\n   * normally.\n   *\n   * @param key - Cache key parts\n   * @param fn - Function to execute. Receives the cache-owned shared signal;\n   *   pass it through to the underlying I/O so the work is cancelled when\n   *   no caller is left waiting.\n   * @param userKey - User key for cache namespacing\n   * @param options - Options for the cache\n   * @returns Promise of the result\n   */\n  async getOrExecute<T>(\n    key: (string | number | object)[],\n    fn: (sharedSignal?: AbortSignal) => Promise<T>,\n    userKey: string,\n    options?: { ttl?: number; callerSignal?: AbortSignal },\n  ): Promise<T> {\n    if (!this.config.enabled) return fn(options?.callerSignal);\n\n    const callerSignal = options?.callerSignal;\n    if (callerSignal?.aborted) {\n      throw createAbortError(callerSignal);\n    }\n\n    const cacheKey = this.generateKey(key, userKey);\n\n    return this.telemetry.startActiveSpan(\n      \"cache.getOrExecute\",\n      {\n        attributes: {\n          \"cache.key\": cacheKey,\n          \"cache.enabled\": this.config.enabled,\n          \"cache.persistent\": this.storage.isPersistent(),\n        },\n      },\n      async (span) => {\n        try {\n          const cached = await this.getValid<T>(cacheKey);\n          if (cached !== null) {\n            span.setAttribute(\"cache.hit\", true);\n            span.setStatus({ code: SpanStatusCode.OK });\n            this.telemetryMetrics.cacheHitCount.add(1, {\n              \"cache.key\": cacheKey,\n            });\n\n            logger.event()?.setExecution({\n              cache_hit: true,\n              cache_key: cacheKey,\n            });\n\n            return cached.value;\n          }\n\n          // check if the value is being processed by another request — join\n          // the existing in-flight entry under reference counting so this\n          // caller's abort doesn't poison the shared result.\n          const existing = this.inFlightRequests.get(cacheKey) as\n            | InFlightEntry<T>\n            | undefined;\n          if (existing && !existing.sharedController.signal.aborted) {\n            existing.refCount++;\n            // Cancel any pending abort timer — a new caller has joined\n            if (existing.abortTimer) {\n              clearTimeout(existing.abortTimer);\n              existing.abortTimer = undefined;\n            }\n            span.setAttribute(\"cache.hit\", true);\n            span.setAttribute(\"cache.deduplication\", true);\n            span.addEvent(\"cache.deduplication_used\", {\n              \"cache.key\": cacheKey,\n            });\n            this.telemetryMetrics.cacheHitCount.add(1, {\n              \"cache.key\": cacheKey,\n              \"cache.deduplication\": \"true\",\n            });\n\n            logger.event()?.setExecution({\n              cache_hit: true,\n              cache_key: cacheKey,\n              cache_deduplication: true,\n            });\n\n            return await this._waitWithRefCount(existing, callerSignal);\n          }\n\n          // cache miss - execute function under a shared abort controller\n          span.setAttribute(\"cache.hit\", false);\n          span.addEvent(\"cache.miss\", { \"cache.key\": cacheKey });\n          this.telemetryMetrics.cacheMissCount.add(1, {\n            \"cache.key\": cacheKey,\n          });\n\n          logger.event()?.setExecution({\n            cache_hit: false,\n            cache_key: cacheKey,\n          });\n\n          const sharedController = new AbortController();\n          const entry: InFlightEntry<T> = {\n            promise: undefined as unknown as Promise<T>,\n            refCount: 1,\n            sharedController,\n          };\n\n          entry.promise = fn(sharedController.signal)\n            .then(async (result) => {\n              await this.set(cacheKey, result, options);\n              span.addEvent(\"cache.value_stored\", {\n                \"cache.key\": cacheKey,\n                \"cache.ttl\": options?.ttl ?? this.config.ttl ?? 3600,\n              });\n              return result;\n            })\n            .catch((error) => {\n              span.recordException(error);\n              span.setStatus({ code: SpanStatusCode.ERROR });\n              // If the shared controller aborted, all callers have already\n              // abandoned the request (or are about to via their own signals)\n              // — propagate the original error without wrapping. No live\n              // awaiter will observe this rejection.\n              if (sharedController.signal.aborted) {\n                throw error;\n              }\n              if (error instanceof AppKitError || error instanceof ApiError) {\n                throw error;\n              }\n              throw ExecutionError.statementFailed(\n                error instanceof Error ? error.message : String(error),\n              );\n            })\n            .finally(() => {\n              if (this.inFlightRequests.get(cacheKey) === entry) {\n                this.inFlightRequests.delete(cacheKey);\n              }\n            });\n\n          // Suppress unhandled rejection warnings when every caller bailed\n          // before fn() resolved (their own promises rejected via\n          // waitWithRefCount; the underlying entry.promise has no awaiter).\n          entry.promise.catch(() => {});\n\n          this.inFlightRequests.set(cacheKey, entry as InFlightEntry<unknown>);\n\n          const result = await this._waitWithRefCount(entry, callerSignal);\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          span.recordException(error as Error);\n          span.setStatus({ code: SpanStatusCode.ERROR });\n          throw error;\n        } finally {\n          span.end();\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  /**\n   * Wait on an in-flight entry, racing the underlying promise against the\n   * caller's abort signal. When the caller aborts, the entry's refCount is\n   * decremented; if it hits zero the shared controller is aborted so the\n   * underlying `fn()` can stop. Other callers continue to await the same\n   * entry and receive the result when it arrives.\n   */\n  private _waitWithRefCount<T>(\n    entry: InFlightEntry<T>,\n    callerSignal?: AbortSignal,\n  ): Promise<T> {\n    if (!callerSignal) return entry.promise;\n\n    return new Promise<T>((resolve, reject) => {\n      let settled = false;\n\n      const release = () => {\n        if (entry.refCount > 0) entry.refCount--;\n        if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n          // Grace period: delay abort so a StrictMode remount can join\n          // the in-flight entry before the shared execution is cancelled.\n          entry.abortTimer = setTimeout(() => {\n            if (entry.refCount <= 0 && !entry.sharedController.signal.aborted) {\n              entry.sharedController.abort(\n                callerSignal.reason ?? \"all cache callers aborted\",\n              );\n            }\n          }, CacheManager.ABORT_GRACE_PERIOD_MS);\n        }\n      };\n\n      const onAbort = () => {\n        if (settled) return;\n        settled = true;\n        callerSignal.removeEventListener(\"abort\", onAbort);\n        release();\n        reject(createAbortError(callerSignal));\n      };\n\n      if (callerSignal.aborted) {\n        onAbort();\n        return;\n      }\n\n      callerSignal.addEventListener(\"abort\", onAbort, { once: true });\n\n      entry.promise.then(\n        (value) => {\n          if (settled) return;\n          settled = true;\n          callerSignal.removeEventListener(\"abort\", onAbort);\n          resolve(value);\n        },\n        (error) => {\n          if (settled) return;\n          settled = true;\n          callerSignal.removeEventListener(\"abort\", onAbort);\n          reject(error);\n        },\n      );\n    });\n  }\n\n  /**\n   * Get a cached value\n   * @param key - Cache key\n   * @returns Promise of the value or null if not found or expired\n   */\n  async get<T>(key: string): Promise<T | null> {\n    if (!this.config.enabled) return null;\n\n    // probabilistic cleanup trigger\n    this.maybeCleanup();\n\n    const entry = await this.getValid<T>(key);\n    return entry?.value ?? null;\n  }\n\n  /**\n   * Get a cached entry only if it has not expired.\n   * Returns null on miss or expired (and deletes the expired entry).\n   *\n   * Storage implementations return entries unconditionally — expiry handling\n   * lives at the CacheManager layer.\n   */\n  private async getValid<T>(key: string): Promise<CacheEntry<T> | null> {\n    const entry = await this.storage.get<T>(key);\n    if (!entry) return null;\n\n    if (Date.now() > entry.expiry) {\n      await this.storage.delete(key);\n      return null;\n    }\n    return entry;\n  }\n\n  /** Probabilistically trigger cleanup of expired entries (fire-and-forget) */\n  private maybeCleanup(): void {\n    if (this.cleanupInProgress) return;\n    if (!this.storage.isPersistent()) return;\n    const now = Date.now();\n    if (now - this.lastCleanupAttempt < CacheManager.MIN_CLEANUP_INTERVAL_MS)\n      return;\n\n    const probability = this.config.cleanupProbability ?? 0.01;\n\n    if (Math.random() > probability) return;\n\n    this.lastCleanupAttempt = now;\n\n    this.cleanupInProgress = true;\n    (this.storage as PersistentStorage)\n      .cleanupExpired()\n      .catch((error) => {\n        logger.debug(\"Error cleaning up expired entries: %O\", error);\n      })\n      .finally(() => {\n        this.cleanupInProgress = false;\n      });\n  }\n\n  /**\n   * Set a value in the cache\n   * @param key - Cache key\n   * @param value - Value to set\n   * @param options - Options for the cache\n   * @returns Promise of the result\n   */\n  async set<T>(\n    key: string,\n    value: T,\n    options?: { ttl?: number },\n  ): Promise<void> {\n    if (!this.config.enabled) return;\n\n    const ttl = options?.ttl ?? this.config.ttl ?? 3600;\n    const expiryTime = Date.now() + ttl * 1000;\n    await this.storage.set(key, { value, expiry: expiryTime });\n  }\n\n  /**\n   * Delete a value from the cache\n   * @param key - Cache key\n   * @returns Promise of the result\n   */\n  async delete(key: string): Promise<void> {\n    if (!this.config.enabled) return;\n    await this.storage.delete(key);\n  }\n\n  /** Clear the cache */\n  async clear(): Promise<void> {\n    await this.storage.clear();\n    for (const entry of this.inFlightRequests.values()) {\n      if (entry.abortTimer) clearTimeout(entry.abortTimer);\n    }\n    this.inFlightRequests.clear();\n  }\n\n  /**\n   * Check if a value exists in the cache\n   * @param key - Cache key\n   * @returns Promise of true if the value exists, false otherwise\n   */\n  async has(key: string): Promise<boolean> {\n    if (!this.config.enabled) return false;\n\n    const entry = await this.getValid(key);\n    return entry !== null;\n  }\n\n  /**\n   * Generate a cache key\n   * @param parts - Parts of the key\n   * @param userKey - User key\n   * @returns Cache key\n   */\n  generateKey(parts: (string | number | object)[], userKey: string): string {\n    const allParts = [userKey, ...parts];\n    const serialized = JSON.stringify(allParts);\n    return createHash(\"sha256\").update(serialized).digest(\"hex\");\n  }\n\n  /** Close the cache */\n  async close(): Promise<void> {\n    await this.storage.close();\n  }\n\n  /**\n   * Check if the storage is healthy\n   * @returns Promise of true if the storage is healthy, false otherwise\n   */\n  async isStorageHealthy(): Promise<boolean> {\n    return this.storage.healthCheck();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAaA,MAAM,SAAS,aAAa,QAAQ;AAoBpC,SAAS,iBAAiB,QAA8B;AACtD,KAAI,OAAO,WAAW,OAAW,QAAO,OAAO;AAC/C,QAAO,IAAI,aAAa,8BAA8B,aAAa;;;;;;;;;;;;;;;;AAiBrE,IAAa,eAAb,MAAa,aAAa;CACxB,OAAwB,0BAA0B;CAClD,OAAwB,wBAAwB;CAChD,AAAiB,OAAe;CAChC,OAAe,WAAgC;CAC/C,OAAe,cAA4C;CAE3D,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,AAAQ;CACR,AAAQ;CAKR,AAAQ,YAAY,SAAuB,QAAqB;AAC9D,OAAK,UAAU;AACf,OAAK,SAAS;AACd,OAAK,mCAAmB,IAAI,KAAK;AACjC,OAAK,oBAAoB;AACzB,OAAK,qBAAqB;AAE1B,OAAK,YAAY,iBAAiB,YAChC,KAAK,MACL,KAAK,OAAO,UACb;AACD,OAAK,mBAAmB;GACtB,eAAe,KAAK,UAAU,UAAU,CAAC,cAAc,aAAa;IAClE,aAAa;IACb,MAAM;IACP,CAAC;GACF,gBAAgB,KAAK,UAAU,UAAU,CAAC,cAAc,cAAc;IACpE,aAAa;IACb,MAAM;IACP,CAAC;GACH;;;;;;;;CASH,OAAO,kBAAgC;AACrC,MAAI,CAAC,aAAa,SAChB,OAAM,oBAAoB,eACxB,gBACA,kEACD;AAGH,SAAO,aAAa;;;;;;;;;CAUtB,aAAa,YACX,YACuB;AACvB,MAAI,aAAa,SACf,QAAO,aAAa;AAGtB,MAAI,CAAC,aAAa,YAChB,cAAa,cAAc,aAAa,OAAO,WAAW,CAAC,MACxD,aAAa;AACZ,gBAAa,WAAW;AACxB,UAAO;IAEV;AAGH,SAAO,aAAa;;;;;;;;;;;;;;CAetB,aAAqB,OACnB,YACuB;EACvB,MAAM,SAAS,UAAU,eAAe,WAAW;AAEnD,MAAI,OAAO,SAAS;AAElB,OADkB,MAAM,OAAO,QAAQ,aAAa,CAElD,QAAO,IAAI,aAAa,OAAO,SAAS,OAAO;AAGjD,OAAI,OAAO,mBAAmB;IAC5B,MAAM,iBAAiB;KAAE,GAAG;KAAQ,SAAS;KAAO;AACpD,WAAO,IAAI,aACT,IAAI,gBAAgB,eAAe,EACnC,eACD;;AAGH,UAAO,IAAI,aAAa,IAAI,gBAAgB,OAAO,EAAE,OAAO;;AAI9D,MAAI;GAEF,MAAM,OAAO,mBAAmB,EAAE,iBADV,IAAI,gBAAgB,EAAE,EAAE,kBAAkB,CAAC,EAChB,CAAC;GACpD,MAAM,oBAAoB,IAAI,kBAAkB,QAAQ,KAAK;AAG7D,OADkB,MAAM,kBAAkB,aAAa,EACxC;AACb,UAAM,kBAAkB,YAAY;AACpC,WAAO,IAAI,aAAa,mBAAmB,OAAO;;AAIpD,SAAM,KAAK,KAAK;UACV;AAIR,MAAI,OAAO,mBAAmB;GAC5B,MAAM,iBAAiB;IAAE,GAAG;IAAQ,SAAS;IAAO;AACpD,UAAO,IAAI,aACT,IAAI,gBAAgB,eAAe,EACnC,eACD;;AAGH,SAAO,IAAI,aAAa,IAAI,gBAAgB,OAAO,EAAE,OAAO;;;;;;;;;;;;;;;;;;;;;;CAuB9D,MAAM,aACJ,KACA,IACA,SACA,SACY;AACZ,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO,GAAG,SAAS,aAAa;EAE1D,MAAM,eAAe,SAAS;AAC9B,MAAI,cAAc,QAChB,OAAM,iBAAiB,aAAa;EAGtC,MAAM,WAAW,KAAK,YAAY,KAAK,QAAQ;AAE/C,SAAO,KAAK,UAAU,gBACpB,sBACA,EACE,YAAY;GACV,aAAa;GACb,iBAAiB,KAAK,OAAO;GAC7B,oBAAoB,KAAK,QAAQ,cAAc;GAChD,EACF,EACD,OAAO,SAAS;AACd,OAAI;IACF,MAAM,SAAS,MAAM,KAAK,SAAY,SAAS;AAC/C,QAAI,WAAW,MAAM;AACnB,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,UAAK,iBAAiB,cAAc,IAAI,GAAG,EACzC,aAAa,UACd,CAAC;AAEF,YAAO,OAAO,EAAE,aAAa;MAC3B,WAAW;MACX,WAAW;MACZ,CAAC;AAEF,YAAO,OAAO;;IAMhB,MAAM,WAAW,KAAK,iBAAiB,IAAI,SAAS;AAGpD,QAAI,YAAY,CAAC,SAAS,iBAAiB,OAAO,SAAS;AACzD,cAAS;AAET,SAAI,SAAS,YAAY;AACvB,mBAAa,SAAS,WAAW;AACjC,eAAS,aAAa;;AAExB,UAAK,aAAa,aAAa,KAAK;AACpC,UAAK,aAAa,uBAAuB,KAAK;AAC9C,UAAK,SAAS,4BAA4B,EACxC,aAAa,UACd,CAAC;AACF,UAAK,iBAAiB,cAAc,IAAI,GAAG;MACzC,aAAa;MACb,uBAAuB;MACxB,CAAC;AAEF,YAAO,OAAO,EAAE,aAAa;MAC3B,WAAW;MACX,WAAW;MACX,qBAAqB;MACtB,CAAC;AAEF,YAAO,MAAM,KAAK,kBAAkB,UAAU,aAAa;;AAI7D,SAAK,aAAa,aAAa,MAAM;AACrC,SAAK,SAAS,cAAc,EAAE,aAAa,UAAU,CAAC;AACtD,SAAK,iBAAiB,eAAe,IAAI,GAAG,EAC1C,aAAa,UACd,CAAC;AAEF,WAAO,OAAO,EAAE,aAAa;KAC3B,WAAW;KACX,WAAW;KACZ,CAAC;IAEF,MAAM,mBAAmB,IAAI,iBAAiB;IAC9C,MAAM,QAA0B;KAC9B,SAAS;KACT,UAAU;KACV;KACD;AAED,UAAM,UAAU,GAAG,iBAAiB,OAAO,CACxC,KAAK,OAAO,WAAW;AACtB,WAAM,KAAK,IAAI,UAAU,QAAQ,QAAQ;AACzC,UAAK,SAAS,sBAAsB;MAClC,aAAa;MACb,aAAa,SAAS,OAAO,KAAK,OAAO,OAAO;MACjD,CAAC;AACF,YAAO;MACP,CACD,OAAO,UAAU;AAChB,UAAK,gBAAgB,MAAM;AAC3B,UAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAK9C,SAAI,iBAAiB,OAAO,QAC1B,OAAM;AAER,SAAI,iBAAiB,eAAe,iBAAiB,SACnD,OAAM;AAER,WAAM,eAAe,gBACnB,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;MACD,CACD,cAAc;AACb,SAAI,KAAK,iBAAiB,IAAI,SAAS,KAAK,MAC1C,MAAK,iBAAiB,OAAO,SAAS;MAExC;AAKJ,UAAM,QAAQ,YAAY,GAAG;AAE7B,SAAK,iBAAiB,IAAI,UAAU,MAAgC;IAEpE,MAAM,SAAS,MAAM,KAAK,kBAAkB,OAAO,aAAa;AAChE,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU,EAAE,MAAM,eAAe,OAAO,CAAC;AAC9C,UAAM;aACE;AACR,SAAK,KAAK;;KAGd;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;;;;;;;;CAUH,AAAQ,kBACN,OACA,cACY;AACZ,MAAI,CAAC,aAAc,QAAO,MAAM;AAEhC,SAAO,IAAI,SAAY,SAAS,WAAW;GACzC,IAAI,UAAU;GAEd,MAAM,gBAAgB;AACpB,QAAI,MAAM,WAAW,EAAG,OAAM;AAC9B,QAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QAGxD,OAAM,aAAa,iBAAiB;AAClC,SAAI,MAAM,YAAY,KAAK,CAAC,MAAM,iBAAiB,OAAO,QACxD,OAAM,iBAAiB,MACrB,aAAa,UAAU,4BACxB;OAEF,aAAa,sBAAsB;;GAI1C,MAAM,gBAAgB;AACpB,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,aAAS;AACT,WAAO,iBAAiB,aAAa,CAAC;;AAGxC,OAAI,aAAa,SAAS;AACxB,aAAS;AACT;;AAGF,gBAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM,CAAC;AAE/D,SAAM,QAAQ,MACX,UAAU;AACT,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,YAAQ,MAAM;OAEf,UAAU;AACT,QAAI,QAAS;AACb,cAAU;AACV,iBAAa,oBAAoB,SAAS,QAAQ;AAClD,WAAO,MAAM;KAEhB;IACD;;;;;;;CAQJ,MAAM,IAAO,KAAgC;AAC3C,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO;AAGjC,OAAK,cAAc;AAGnB,UADc,MAAM,KAAK,SAAY,IAAI,GAC3B,SAAS;;;;;;;;;CAUzB,MAAc,SAAY,KAA4C;EACpE,MAAM,QAAQ,MAAM,KAAK,QAAQ,IAAO,IAAI;AAC5C,MAAI,CAAC,MAAO,QAAO;AAEnB,MAAI,KAAK,KAAK,GAAG,MAAM,QAAQ;AAC7B,SAAM,KAAK,QAAQ,OAAO,IAAI;AAC9B,UAAO;;AAET,SAAO;;;CAIT,AAAQ,eAAqB;AAC3B,MAAI,KAAK,kBAAmB;AAC5B,MAAI,CAAC,KAAK,QAAQ,cAAc,CAAE;EAClC,MAAM,MAAM,KAAK,KAAK;AACtB,MAAI,MAAM,KAAK,qBAAqB,aAAa,wBAC/C;EAEF,MAAM,cAAc,KAAK,OAAO,sBAAsB;AAEtD,MAAI,KAAK,QAAQ,GAAG,YAAa;AAEjC,OAAK,qBAAqB;AAE1B,OAAK,oBAAoB;AACzB,EAAC,KAAK,QACH,gBAAgB,CAChB,OAAO,UAAU;AAChB,UAAO,MAAM,yCAAyC,MAAM;IAC5D,CACD,cAAc;AACb,QAAK,oBAAoB;IACzB;;;;;;;;;CAUN,MAAM,IACJ,KACA,OACA,SACe;AACf,MAAI,CAAC,KAAK,OAAO,QAAS;EAE1B,MAAM,MAAM,SAAS,OAAO,KAAK,OAAO,OAAO;EAC/C,MAAM,aAAa,KAAK,KAAK,GAAG,MAAM;AACtC,QAAM,KAAK,QAAQ,IAAI,KAAK;GAAE;GAAO,QAAQ;GAAY,CAAC;;;;;;;CAQ5D,MAAM,OAAO,KAA4B;AACvC,MAAI,CAAC,KAAK,OAAO,QAAS;AAC1B,QAAM,KAAK,QAAQ,OAAO,IAAI;;;CAIhC,MAAM,QAAuB;AAC3B,QAAM,KAAK,QAAQ,OAAO;AAC1B,OAAK,MAAM,SAAS,KAAK,iBAAiB,QAAQ,CAChD,KAAI,MAAM,WAAY,cAAa,MAAM,WAAW;AAEtD,OAAK,iBAAiB,OAAO;;;;;;;CAQ/B,MAAM,IAAI,KAA+B;AACvC,MAAI,CAAC,KAAK,OAAO,QAAS,QAAO;AAGjC,SADc,MAAM,KAAK,SAAS,IAAI,KACrB;;;;;;;;CASnB,YAAY,OAAqC,SAAyB;EACxE,MAAM,WAAW,CAAC,SAAS,GAAG,MAAM;EACpC,MAAM,aAAa,KAAK,UAAU,SAAS;AAC3C,SAAO,WAAW,SAAS,CAAC,OAAO,WAAW,CAAC,OAAO,MAAM;;;CAI9D,MAAM,QAAuB;AAC3B,QAAM,KAAK,QAAQ,OAAO;;;;;;CAO5B,MAAM,mBAAqC;AACzC,SAAO,KAAK,QAAQ,aAAa"}

package/dist/connectors/files/client.js

@@ -174,7 +174,10 @@ var FilesConnector = class {
 			const host = hostValue.startsWith("http") ? hostValue : `https://${hostValue}`;
 			const url = new URL(`/api/2.0/fs/files${resolvedPath}`, host);
 			url.searchParams.set("overwrite", String(overwrite));
-			const headers = new Headers({ "Content-Type": "application/octet-stream" });
+			const headers = new Headers({
+				"Content-Type": "application/octet-stream",
+				"User-Agent": client.apiClient.userAgent()
+			});
 			const fetchOptions = {
 				method: "PUT",
 				headers,

package/dist/connectors/files/client.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/files/client.ts"],"sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport { ApiError, type WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport type { TelemetryOptions } from \"shared\";\nimport { createLogger } from \"../../logging/logger\";\nimport type {\n  DirectoryEntry,\n  DownloadResponse,\n  FileMetadata,\n  FilePreview,\n} from \"../../plugins/files/types\";\nimport type { TelemetryProvider } from \"../../telemetry\";\nimport {\n  type Counter,\n  type Histogram,\n  type Span,\n  SpanKind,\n  SpanStatusCode,\n  TelemetryManager,\n} from \"../../telemetry\";\nimport {\n  contentTypeFromPath,\n  FILES_MAX_READ_SIZE,\n  isTextContentType,\n} from \"./defaults\";\n\nconst logger = createLogger(\"connectors:files\");\n\n/**\n * Ambient span-attribute propagation for `FilesConnector.traced()`.\n *\n * Callers (e.g. the plugin's `_withAuthModeAttributes` wrapper) set extra\n * span attributes here via `runWithFilesSpanAttributes(attrs, fn)`. The\n * connector's `traced()` decorator reads them and merges them into the\n * span it creates around the SDK call. This lets the plugin tag spans with\n * `files.auth_mode` without opening a duplicate `files.<op>` span.\n *\n * AsyncLocalStorage is used so concurrent requests don't see each other's\n * attributes. Outside an active scope, `getStore()` returns `undefined` and\n * the connector falls back to the static attribute set.\n */\nconst filesSpanAttributesStorage = new AsyncLocalStorage<\n  Record<string, string>\n>();\n\n/**\n * Run `fn` with the supplied attributes attached to whatever span the\n * `FilesConnector` opens for its SDK call. Used to propagate request-scoped\n * attributes (e.g. `files.auth_mode`) onto the connector's span without\n * opening a parent span — avoids the 2x span allocation that\n * `startActiveSpan` parented otherwise.\n */\nexport function runWithFilesSpanAttributes<T>(\n  attributes: Record<string, string>,\n  fn: () => Promise<T>,\n): Promise<T> {\n  return filesSpanAttributesStorage.run(attributes, fn);\n}\n\ninterface FilesConnectorConfig {\n  defaultVolume?: string;\n  timeout?: number;\n  telemetry?: TelemetryOptions;\n  customContentTypes?: Record<string, string>;\n}\n\nexport class FilesConnector {\n  private readonly name = \"files\";\n  private defaultVolume: string | undefined;\n  private readonly customContentTypes: Record<string, string> | undefined;\n\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    operationCount: Counter;\n    operationDuration: Histogram;\n  };\n\n  constructor(config: FilesConnectorConfig) {\n    this.defaultVolume = config.defaultVolume;\n    this.customContentTypes = config.customContentTypes;\n\n    this.telemetry = TelemetryManager.getProvider(this.name, config.telemetry);\n    this.telemetryMetrics = {\n      operationCount: this.telemetry\n        .getMeter()\n        .createCounter(\"files.operation.count\", {\n          description: \"Total number of file operations\",\n          unit: \"1\",\n        }),\n      operationDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"files.operation.duration\", {\n          description: \"Duration of file operations\",\n          unit: \"ms\",\n        }),\n    };\n  }\n\n  resolvePath(filePath: string): string {\n    if (filePath.length > 4096) {\n      throw new Error(\n        `Path exceeds maximum length of 4096 characters (got ${filePath.length}).`,\n      );\n    }\n    if (filePath.includes(\"\\0\")) {\n      throw new Error(\"Path must not contain null bytes.\");\n    }\n\n    const segments = filePath.split(\"/\");\n    if (segments.some((s) => s === \"..\")) {\n      throw new Error('Path traversal (\"../\") is not allowed.');\n    }\n    if (filePath.startsWith(\"/\")) {\n      if (!filePath.startsWith(\"/Volumes/\")) {\n        throw new Error(\n          'Absolute paths must start with \"/Volumes/\". ' +\n            \"Unity Catalog volume paths follow the format: /Volumes/<catalog>/<schema>/<volume>/\",\n        );\n      }\n      return filePath;\n    }\n    if (!this.defaultVolume) {\n      throw new Error(\n        \"Cannot resolve relative path: no default volume set. Use an absolute path or set a default volume.\",\n      );\n    }\n    return `${this.defaultVolume}/${filePath}`;\n  }\n\n  private async traced<T>(\n    operation: string,\n    attributes: Record<string, string>,\n    fn: (span: Span) => Promise<T>,\n  ): Promise<T> {\n    const startTime = Date.now();\n    let success = false;\n\n    // Pull any ambient attributes set by `runWithFilesSpanAttributes` (e.g.\n    // `files.auth_mode` from the plugin layer). Static `attributes` win on\n    // collision so callers can override per-call.\n    const ambient = filesSpanAttributesStorage.getStore();\n\n    return this.telemetry.startActiveSpan(\n      `files.${operation}`,\n      {\n        kind: SpanKind.CLIENT,\n        attributes: {\n          \"files.operation\": operation,\n          ...(ambient ?? {}),\n          ...attributes,\n        },\n      },\n      async (span: Span) => {\n        try {\n          const result = await fn(span);\n          success = true;\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          span.recordException(error as Error);\n          span.setStatus({\n            code: SpanStatusCode.ERROR,\n            message: error instanceof Error ? error.message : String(error),\n          });\n          throw error;\n        } finally {\n          span.end();\n          const duration = Date.now() - startTime;\n          const metricAttrs = {\n            \"files.operation\": operation,\n            success: String(success),\n          };\n          this.telemetryMetrics.operationCount.add(1, metricAttrs);\n          this.telemetryMetrics.operationDuration.record(duration, metricAttrs);\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  async list(\n    client: WorkspaceClient,\n    directoryPath?: string,\n  ): Promise<DirectoryEntry[]> {\n    const resolvedPath = directoryPath\n      ? this.resolvePath(directoryPath)\n      : this.defaultVolume;\n    if (!resolvedPath) {\n      throw new Error(\"No directory path provided and no default volume set.\");\n    }\n\n    return this.traced(\"list\", { \"files.path\": resolvedPath }, async () => {\n      const entries: DirectoryEntry[] = [];\n      for await (const entry of client.files.listDirectoryContents({\n        directory_path: resolvedPath,\n      })) {\n        entries.push(entry);\n      }\n      return entries;\n    });\n  }\n\n  async read(\n    client: WorkspaceClient,\n    filePath: string,\n    options?: { maxSize?: number },\n  ): Promise<string> {\n    const resolvedPath = this.resolvePath(filePath);\n    const maxSize = options?.maxSize ?? FILES_MAX_READ_SIZE;\n    return this.traced(\"read\", { \"files.path\": resolvedPath }, async () => {\n      const response = await this.download(client, filePath);\n      if (!response.contents) {\n        return \"\";\n      }\n      const reader = response.contents.getReader();\n      const decoder = new TextDecoder();\n      let result = \"\";\n      let bytesRead = 0;\n      while (true) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        bytesRead += value.byteLength;\n        if (bytesRead > maxSize) {\n          await reader.cancel();\n          throw new Error(\n            `File exceeds maximum read size (${maxSize} bytes). Use download() for large files.`,\n          );\n        }\n        result += decoder.decode(value, { stream: true });\n      }\n      result += decoder.decode();\n      return result;\n    });\n  }\n\n  async download(\n    client: WorkspaceClient,\n    filePath: string,\n  ): Promise<DownloadResponse> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"download\", { \"files.path\": resolvedPath }, async () => {\n      return client.files.download({\n        file_path: resolvedPath,\n      });\n    });\n  }\n\n  async exists(client: WorkspaceClient, filePath: string): Promise<boolean> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"exists\", { \"files.path\": resolvedPath }, async () => {\n      try {\n        await this.metadata(client, filePath);\n        return true;\n      } catch (error) {\n        if (error instanceof ApiError && error.statusCode === 404) {\n          return false;\n        }\n        throw error;\n      }\n    });\n  }\n\n  async metadata(\n    client: WorkspaceClient,\n    filePath: string,\n  ): Promise<FileMetadata> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"metadata\", { \"files.path\": resolvedPath }, async () => {\n      const response = await client.files.getMetadata({\n        file_path: resolvedPath,\n      });\n      return {\n        contentLength: response[\"content-length\"],\n        contentType: contentTypeFromPath(\n          filePath,\n          response[\"content-type\"],\n          this.customContentTypes,\n        ),\n        lastModified: response[\"last-modified\"],\n      };\n    });\n  }\n\n  async upload(\n    client: WorkspaceClient,\n    filePath: string,\n    contents: ReadableStream | Buffer | string,\n    options?: { overwrite?: boolean },\n  ): Promise<void> {\n    const resolvedPath = this.resolvePath(filePath);\n\n    return this.traced(\"upload\", { \"files.path\": resolvedPath }, async () => {\n      const body = contents;\n      const overwrite = options?.overwrite ?? true;\n\n      // Workaround: The SDK's files.upload() has two bugs:\n      // 1. It ignores the `contents` field (sets body to undefined)\n      // 2. apiClient.request() checks `instanceof` against its own ReadableStream\n      //    subclass, so standard ReadableStream instances get JSON.stringified to \"{}\"\n      // Bypass both by calling the REST API directly with SDK-provided auth.\n      const hostValue = client.config.host;\n      if (!hostValue) {\n        throw new Error(\n          \"Databricks host is not configured. Set DATABRICKS_HOST or configure client.config.host.\",\n        );\n      }\n      const host = hostValue.startsWith(\"http\")\n        ? hostValue\n        : `https://${hostValue}`;\n      const url = new URL(`/api/2.0/fs/files${resolvedPath}`, host);\n      url.searchParams.set(\"overwrite\", String(overwrite));\n\n      const headers = new Headers({\n        \"Content-Type\": \"application/octet-stream\",\n      });\n      const fetchOptions: RequestInit = { method: \"PUT\", headers, body };\n\n      if (body instanceof ReadableStream) {\n        fetchOptions.duplex = \"half\";\n      } else if (body instanceof Buffer) {\n        headers.set(\"Content-Length\", String(body.length));\n      } else if (typeof body === \"string\") {\n        headers.set(\"Content-Length\", String(Buffer.byteLength(body)));\n      }\n\n      await client.config.authenticate(headers);\n\n      const res = await fetch(url.toString(), fetchOptions);\n\n      if (!res.ok) {\n        const text = await res.text();\n        logger.error(`Upload failed (${res.status}): ${text}`);\n        const safeMessage = text.length > 200 ? `${text.slice(0, 200)}…` : text;\n        throw new ApiError(\n          `Upload failed: ${safeMessage}`,\n          \"UPLOAD_FAILED\",\n          res.status,\n          undefined,\n          [],\n        );\n      }\n    });\n  }\n\n  async createDirectory(\n    client: WorkspaceClient,\n    directoryPath: string,\n  ): Promise<void> {\n    const resolvedPath = this.resolvePath(directoryPath);\n    return this.traced(\n      \"createDirectory\",\n      { \"files.path\": resolvedPath },\n      async () => {\n        await client.files.createDirectory({\n          directory_path: resolvedPath,\n        });\n      },\n    );\n  }\n\n  async delete(client: WorkspaceClient, filePath: string): Promise<void> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"delete\", { \"files.path\": resolvedPath }, async () => {\n      await client.files.delete({\n        file_path: resolvedPath,\n      });\n    });\n  }\n\n  async preview(\n    client: WorkspaceClient,\n    filePath: string,\n    options?: { maxChars?: number },\n  ): Promise<FilePreview> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"preview\", { \"files.path\": resolvedPath }, async () => {\n      const meta = await this.metadata(client, filePath);\n      const isText = isTextContentType(meta.contentType);\n      const isImage = meta.contentType?.startsWith(\"image/\") || false;\n\n      if (!isText) {\n        return { ...meta, textPreview: null, isText: false, isImage };\n      }\n\n      const response = await client.files.download({\n        file_path: resolvedPath,\n      });\n      if (!response.contents) {\n        return { ...meta, textPreview: \"\", isText: true, isImage: false };\n      }\n\n      const reader = response.contents.getReader();\n      const decoder = new TextDecoder();\n      let preview = \"\";\n      const maxChars = options?.maxChars ?? 1024;\n\n      while (preview.length < maxChars) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        preview += decoder.decode(value, { stream: true });\n      }\n      preview += decoder.decode();\n      await reader.cancel();\n\n      if (preview.length > maxChars) {\n        preview = preview.slice(0, maxChars);\n      }\n\n      return { ...meta, textPreview: preview, isText: true, isImage: false };\n    });\n  }\n}\n"],"mappings":";;;;;;;;AAyBA,MAAM,SAAS,aAAa,mBAAmB;;;;;;;;;;;;;;AAe/C,MAAM,6BAA6B,IAAI,mBAEpC;;;;;;;;AASH,SAAgB,2BACd,YACA,IACY;AACZ,QAAO,2BAA2B,IAAI,YAAY,GAAG;;AAUvD,IAAa,iBAAb,MAA4B;CAC1B,AAAiB,OAAO;CACxB,AAAQ;CACR,AAAiB;CAEjB,AAAiB;CACjB,AAAiB;CAKjB,YAAY,QAA8B;AACxC,OAAK,gBAAgB,OAAO;AAC5B,OAAK,qBAAqB,OAAO;AAEjC,OAAK,YAAY,iBAAiB,YAAY,KAAK,MAAM,OAAO,UAAU;AAC1E,OAAK,mBAAmB;GACtB,gBAAgB,KAAK,UAClB,UAAU,CACV,cAAc,yBAAyB;IACtC,aAAa;IACb,MAAM;IACP,CAAC;GACJ,mBAAmB,KAAK,UACrB,UAAU,CACV,gBAAgB,4BAA4B;IAC3C,aAAa;IACb,MAAM;IACP,CAAC;GACL;;CAGH,YAAY,UAA0B;AACpC,MAAI,SAAS,SAAS,KACpB,OAAM,IAAI,MACR,uDAAuD,SAAS,OAAO,IACxE;AAEH,MAAI,SAAS,SAAS,KAAK,CACzB,OAAM,IAAI,MAAM,oCAAoC;AAItD,MADiB,SAAS,MAAM,IAAI,CACvB,MAAM,MAAM,MAAM,KAAK,CAClC,OAAM,IAAI,MAAM,2CAAyC;AAE3D,MAAI,SAAS,WAAW,IAAI,EAAE;AAC5B,OAAI,CAAC,SAAS,WAAW,YAAY,CACnC,OAAM,IAAI,MACR,oIAED;AAEH,UAAO;;AAET,MAAI,CAAC,KAAK,cACR,OAAM,IAAI,MACR,qGACD;AAEH,SAAO,GAAG,KAAK,cAAc,GAAG;;CAGlC,MAAc,OACZ,WACA,YACA,IACY;EACZ,MAAM,YAAY,KAAK,KAAK;EAC5B,IAAI,UAAU;EAKd,MAAM,UAAU,2BAA2B,UAAU;AAErD,SAAO,KAAK,UAAU,gBACpB,SAAS,aACT;GACE,MAAM,SAAS;GACf,YAAY;IACV,mBAAmB;IACnB,GAAI,WAAW,EAAE;IACjB,GAAG;IACJ;GACF,EACD,OAAO,SAAe;AACpB,OAAI;IACF,MAAM,SAAS,MAAM,GAAG,KAAK;AAC7B,cAAU;AACV,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU;KACb,MAAM,eAAe;KACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;KAChE,CAAC;AACF,UAAM;aACE;AACR,SAAK,KAAK;IACV,MAAM,WAAW,KAAK,KAAK,GAAG;IAC9B,MAAM,cAAc;KAClB,mBAAmB;KACnB,SAAS,OAAO,QAAQ;KACzB;AACD,SAAK,iBAAiB,eAAe,IAAI,GAAG,YAAY;AACxD,SAAK,iBAAiB,kBAAkB,OAAO,UAAU,YAAY;;KAGzE;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;CAGH,MAAM,KACJ,QACA,eAC2B;EAC3B,MAAM,eAAe,gBACjB,KAAK,YAAY,cAAc,GAC/B,KAAK;AACT,MAAI,CAAC,aACH,OAAM,IAAI,MAAM,wDAAwD;AAG1E,SAAO,KAAK,OAAO,QAAQ,EAAE,cAAc,cAAc,EAAE,YAAY;GACrE,MAAM,UAA4B,EAAE;AACpC,cAAW,MAAM,SAAS,OAAO,MAAM,sBAAsB,EAC3D,gBAAgB,cACjB,CAAC,CACA,SAAQ,KAAK,MAAM;AAErB,UAAO;IACP;;CAGJ,MAAM,KACJ,QACA,UACA,SACiB;EACjB,MAAM,eAAe,KAAK,YAAY,SAAS;EAC/C,MAAM,UAAU,SAAS,WAAW;AACpC,SAAO,KAAK,OAAO,QAAQ,EAAE,cAAc,cAAc,EAAE,YAAY;GACrE,MAAM,WAAW,MAAM,KAAK,SAAS,QAAQ,SAAS;AACtD,OAAI,CAAC,SAAS,SACZ,QAAO;GAET,MAAM,SAAS,SAAS,SAAS,WAAW;GAC5C,MAAM,UAAU,IAAI,aAAa;GACjC,IAAI,SAAS;GACb,IAAI,YAAY;AAChB,UAAO,MAAM;IACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AACV,iBAAa,MAAM;AACnB,QAAI,YAAY,SAAS;AACvB,WAAM,OAAO,QAAQ;AACrB,WAAM,IAAI,MACR,mCAAmC,QAAQ,0CAC5C;;AAEH,cAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;;AAEnD,aAAU,QAAQ,QAAQ;AAC1B,UAAO;IACP;;CAGJ,MAAM,SACJ,QACA,UAC2B;EAC3B,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,YAAY,EAAE,cAAc,cAAc,EAAE,YAAY;AACzE,UAAO,OAAO,MAAM,SAAS,EAC3B,WAAW,cACZ,CAAC;IACF;;CAGJ,MAAM,OAAO,QAAyB,UAAoC;EACxE,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;AACvE,OAAI;AACF,UAAM,KAAK,SAAS,QAAQ,SAAS;AACrC,WAAO;YACA,OAAO;AACd,QAAI,iBAAiB,YAAY,MAAM,eAAe,IACpD,QAAO;AAET,UAAM;;IAER;;CAGJ,MAAM,SACJ,QACA,UACuB;EACvB,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,YAAY,EAAE,cAAc,cAAc,EAAE,YAAY;GACzE,MAAM,WAAW,MAAM,OAAO,MAAM,YAAY,EAC9C,WAAW,cACZ,CAAC;AACF,UAAO;IACL,eAAe,SAAS;IACxB,aAAa,oBACX,UACA,SAAS,iBACT,KAAK,mBACN;IACD,cAAc,SAAS;IACxB;IACD;;CAGJ,MAAM,OACJ,QACA,UACA,UACA,SACe;EACf,MAAM,eAAe,KAAK,YAAY,SAAS;AAE/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;GACvE,MAAM,OAAO;GACb,MAAM,YAAY,SAAS,aAAa;GAOxC,MAAM,YAAY,OAAO,OAAO;AAChC,OAAI,CAAC,UACH,OAAM,IAAI,MACR,0FACD;GAEH,MAAM,OAAO,UAAU,WAAW,OAAO,GACrC,YACA,WAAW;GACf,MAAM,MAAM,IAAI,IAAI,oBAAoB,gBAAgB,KAAK;AAC7D,OAAI,aAAa,IAAI,aAAa,OAAO,UAAU,CAAC;GAEpD,MAAM,UAAU,IAAI,QAAQ,EAC1B,gBAAgB,4BACjB,CAAC;GACF,MAAM,eAA4B;IAAE,QAAQ;IAAO;IAAS;IAAM;AAElE,OAAI,gBAAgB,eAClB,cAAa,SAAS;YACb,gBAAgB,OACzB,SAAQ,IAAI,kBAAkB,OAAO,KAAK,OAAO,CAAC;YACzC,OAAO,SAAS,SACzB,SAAQ,IAAI,kBAAkB,OAAO,OAAO,WAAW,KAAK,CAAC,CAAC;AAGhE,SAAM,OAAO,OAAO,aAAa,QAAQ;GAEzC,MAAM,MAAM,MAAM,MAAM,IAAI,UAAU,EAAE,aAAa;AAErD,OAAI,CAAC,IAAI,IAAI;IACX,MAAM,OAAO,MAAM,IAAI,MAAM;AAC7B,WAAO,MAAM,kBAAkB,IAAI,OAAO,KAAK,OAAO;AAEtD,UAAM,IAAI,SACR,kBAFkB,KAAK,SAAS,MAAM,GAAG,KAAK,MAAM,GAAG,IAAI,CAAC,KAAK,QAGjE,iBACA,IAAI,QACJ,QACA,EAAE,CACH;;IAEH;;CAGJ,MAAM,gBACJ,QACA,eACe;EACf,MAAM,eAAe,KAAK,YAAY,cAAc;AACpD,SAAO,KAAK,OACV,mBACA,EAAE,cAAc,cAAc,EAC9B,YAAY;AACV,SAAM,OAAO,MAAM,gBAAgB,EACjC,gBAAgB,cACjB,CAAC;IAEL;;CAGH,MAAM,OAAO,QAAyB,UAAiC;EACrE,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;AACvE,SAAM,OAAO,MAAM,OAAO,EACxB,WAAW,cACZ,CAAC;IACF;;CAGJ,MAAM,QACJ,QACA,UACA,SACsB;EACtB,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,WAAW,EAAE,cAAc,cAAc,EAAE,YAAY;GACxE,MAAM,OAAO,MAAM,KAAK,SAAS,QAAQ,SAAS;GAClD,MAAM,SAAS,kBAAkB,KAAK,YAAY;GAClD,MAAM,UAAU,KAAK,aAAa,WAAW,SAAS,IAAI;AAE1D,OAAI,CAAC,OACH,QAAO;IAAE,GAAG;IAAM,aAAa;IAAM,QAAQ;IAAO;IAAS;GAG/D,MAAM,WAAW,MAAM,OAAO,MAAM,SAAS,EAC3C,WAAW,cACZ,CAAC;AACF,OAAI,CAAC,SAAS,SACZ,QAAO;IAAE,GAAG;IAAM,aAAa;IAAI,QAAQ;IAAM,SAAS;IAAO;GAGnE,MAAM,SAAS,SAAS,SAAS,WAAW;GAC5C,MAAM,UAAU,IAAI,aAAa;GACjC,IAAI,UAAU;GACd,MAAM,WAAW,SAAS,YAAY;AAEtC,UAAO,QAAQ,SAAS,UAAU;IAChC,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AACV,eAAW,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;;AAEpD,cAAW,QAAQ,QAAQ;AAC3B,SAAM,OAAO,QAAQ;AAErB,OAAI,QAAQ,SAAS,SACnB,WAAU,QAAQ,MAAM,GAAG,SAAS;AAGtC,UAAO;IAAE,GAAG;IAAM,aAAa;IAAS,QAAQ;IAAM,SAAS;IAAO;IACtE"}
+{"version":3,"file":"client.js","names":[],"sources":["../../../src/connectors/files/client.ts"],"sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport { ApiError, type WorkspaceClient } from \"@databricks/sdk-experimental\";\nimport type { TelemetryOptions } from \"shared\";\nimport { createLogger } from \"../../logging/logger\";\nimport type {\n  DirectoryEntry,\n  DownloadResponse,\n  FileMetadata,\n  FilePreview,\n} from \"../../plugins/files/types\";\nimport type { TelemetryProvider } from \"../../telemetry\";\nimport {\n  type Counter,\n  type Histogram,\n  type Span,\n  SpanKind,\n  SpanStatusCode,\n  TelemetryManager,\n} from \"../../telemetry\";\nimport {\n  contentTypeFromPath,\n  FILES_MAX_READ_SIZE,\n  isTextContentType,\n} from \"./defaults\";\n\nconst logger = createLogger(\"connectors:files\");\n\n/**\n * Ambient span-attribute propagation for `FilesConnector.traced()`.\n *\n * Callers (e.g. the plugin's `_withAuthModeAttributes` wrapper) set extra\n * span attributes here via `runWithFilesSpanAttributes(attrs, fn)`. The\n * connector's `traced()` decorator reads them and merges them into the\n * span it creates around the SDK call. This lets the plugin tag spans with\n * `files.auth_mode` without opening a duplicate `files.<op>` span.\n *\n * AsyncLocalStorage is used so concurrent requests don't see each other's\n * attributes. Outside an active scope, `getStore()` returns `undefined` and\n * the connector falls back to the static attribute set.\n */\nconst filesSpanAttributesStorage = new AsyncLocalStorage<\n  Record<string, string>\n>();\n\n/**\n * Run `fn` with the supplied attributes attached to whatever span the\n * `FilesConnector` opens for its SDK call. Used to propagate request-scoped\n * attributes (e.g. `files.auth_mode`) onto the connector's span without\n * opening a parent span — avoids the 2x span allocation that\n * `startActiveSpan` parented otherwise.\n */\nexport function runWithFilesSpanAttributes<T>(\n  attributes: Record<string, string>,\n  fn: () => Promise<T>,\n): Promise<T> {\n  return filesSpanAttributesStorage.run(attributes, fn);\n}\n\ninterface FilesConnectorConfig {\n  defaultVolume?: string;\n  timeout?: number;\n  telemetry?: TelemetryOptions;\n  customContentTypes?: Record<string, string>;\n}\n\nexport class FilesConnector {\n  private readonly name = \"files\";\n  private defaultVolume: string | undefined;\n  private readonly customContentTypes: Record<string, string> | undefined;\n\n  private readonly telemetry: TelemetryProvider;\n  private readonly telemetryMetrics: {\n    operationCount: Counter;\n    operationDuration: Histogram;\n  };\n\n  constructor(config: FilesConnectorConfig) {\n    this.defaultVolume = config.defaultVolume;\n    this.customContentTypes = config.customContentTypes;\n\n    this.telemetry = TelemetryManager.getProvider(this.name, config.telemetry);\n    this.telemetryMetrics = {\n      operationCount: this.telemetry\n        .getMeter()\n        .createCounter(\"files.operation.count\", {\n          description: \"Total number of file operations\",\n          unit: \"1\",\n        }),\n      operationDuration: this.telemetry\n        .getMeter()\n        .createHistogram(\"files.operation.duration\", {\n          description: \"Duration of file operations\",\n          unit: \"ms\",\n        }),\n    };\n  }\n\n  resolvePath(filePath: string): string {\n    if (filePath.length > 4096) {\n      throw new Error(\n        `Path exceeds maximum length of 4096 characters (got ${filePath.length}).`,\n      );\n    }\n    if (filePath.includes(\"\\0\")) {\n      throw new Error(\"Path must not contain null bytes.\");\n    }\n\n    const segments = filePath.split(\"/\");\n    if (segments.some((s) => s === \"..\")) {\n      throw new Error('Path traversal (\"../\") is not allowed.');\n    }\n    if (filePath.startsWith(\"/\")) {\n      if (!filePath.startsWith(\"/Volumes/\")) {\n        throw new Error(\n          'Absolute paths must start with \"/Volumes/\". ' +\n            \"Unity Catalog volume paths follow the format: /Volumes/<catalog>/<schema>/<volume>/\",\n        );\n      }\n      return filePath;\n    }\n    if (!this.defaultVolume) {\n      throw new Error(\n        \"Cannot resolve relative path: no default volume set. Use an absolute path or set a default volume.\",\n      );\n    }\n    return `${this.defaultVolume}/${filePath}`;\n  }\n\n  private async traced<T>(\n    operation: string,\n    attributes: Record<string, string>,\n    fn: (span: Span) => Promise<T>,\n  ): Promise<T> {\n    const startTime = Date.now();\n    let success = false;\n\n    // Pull any ambient attributes set by `runWithFilesSpanAttributes` (e.g.\n    // `files.auth_mode` from the plugin layer). Static `attributes` win on\n    // collision so callers can override per-call.\n    const ambient = filesSpanAttributesStorage.getStore();\n\n    return this.telemetry.startActiveSpan(\n      `files.${operation}`,\n      {\n        kind: SpanKind.CLIENT,\n        attributes: {\n          \"files.operation\": operation,\n          ...(ambient ?? {}),\n          ...attributes,\n        },\n      },\n      async (span: Span) => {\n        try {\n          const result = await fn(span);\n          success = true;\n          span.setStatus({ code: SpanStatusCode.OK });\n          return result;\n        } catch (error) {\n          span.recordException(error as Error);\n          span.setStatus({\n            code: SpanStatusCode.ERROR,\n            message: error instanceof Error ? error.message : String(error),\n          });\n          throw error;\n        } finally {\n          span.end();\n          const duration = Date.now() - startTime;\n          const metricAttrs = {\n            \"files.operation\": operation,\n            success: String(success),\n          };\n          this.telemetryMetrics.operationCount.add(1, metricAttrs);\n          this.telemetryMetrics.operationDuration.record(duration, metricAttrs);\n        }\n      },\n      { name: this.name, includePrefix: true },\n    );\n  }\n\n  async list(\n    client: WorkspaceClient,\n    directoryPath?: string,\n  ): Promise<DirectoryEntry[]> {\n    const resolvedPath = directoryPath\n      ? this.resolvePath(directoryPath)\n      : this.defaultVolume;\n    if (!resolvedPath) {\n      throw new Error(\"No directory path provided and no default volume set.\");\n    }\n\n    return this.traced(\"list\", { \"files.path\": resolvedPath }, async () => {\n      const entries: DirectoryEntry[] = [];\n      for await (const entry of client.files.listDirectoryContents({\n        directory_path: resolvedPath,\n      })) {\n        entries.push(entry);\n      }\n      return entries;\n    });\n  }\n\n  async read(\n    client: WorkspaceClient,\n    filePath: string,\n    options?: { maxSize?: number },\n  ): Promise<string> {\n    const resolvedPath = this.resolvePath(filePath);\n    const maxSize = options?.maxSize ?? FILES_MAX_READ_SIZE;\n    return this.traced(\"read\", { \"files.path\": resolvedPath }, async () => {\n      const response = await this.download(client, filePath);\n      if (!response.contents) {\n        return \"\";\n      }\n      const reader = response.contents.getReader();\n      const decoder = new TextDecoder();\n      let result = \"\";\n      let bytesRead = 0;\n      while (true) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        bytesRead += value.byteLength;\n        if (bytesRead > maxSize) {\n          await reader.cancel();\n          throw new Error(\n            `File exceeds maximum read size (${maxSize} bytes). Use download() for large files.`,\n          );\n        }\n        result += decoder.decode(value, { stream: true });\n      }\n      result += decoder.decode();\n      return result;\n    });\n  }\n\n  async download(\n    client: WorkspaceClient,\n    filePath: string,\n  ): Promise<DownloadResponse> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"download\", { \"files.path\": resolvedPath }, async () => {\n      return client.files.download({\n        file_path: resolvedPath,\n      });\n    });\n  }\n\n  async exists(client: WorkspaceClient, filePath: string): Promise<boolean> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"exists\", { \"files.path\": resolvedPath }, async () => {\n      try {\n        await this.metadata(client, filePath);\n        return true;\n      } catch (error) {\n        if (error instanceof ApiError && error.statusCode === 404) {\n          return false;\n        }\n        throw error;\n      }\n    });\n  }\n\n  async metadata(\n    client: WorkspaceClient,\n    filePath: string,\n  ): Promise<FileMetadata> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"metadata\", { \"files.path\": resolvedPath }, async () => {\n      const response = await client.files.getMetadata({\n        file_path: resolvedPath,\n      });\n      return {\n        contentLength: response[\"content-length\"],\n        contentType: contentTypeFromPath(\n          filePath,\n          response[\"content-type\"],\n          this.customContentTypes,\n        ),\n        lastModified: response[\"last-modified\"],\n      };\n    });\n  }\n\n  async upload(\n    client: WorkspaceClient,\n    filePath: string,\n    contents: ReadableStream | Buffer | string,\n    options?: { overwrite?: boolean },\n  ): Promise<void> {\n    const resolvedPath = this.resolvePath(filePath);\n\n    return this.traced(\"upload\", { \"files.path\": resolvedPath }, async () => {\n      const body = contents;\n      const overwrite = options?.overwrite ?? true;\n\n      // Workaround: The SDK's files.upload() has two bugs:\n      // 1. It ignores the `contents` field (sets body to undefined)\n      // 2. apiClient.request() checks `instanceof` against its own ReadableStream\n      //    subclass, so standard ReadableStream instances get JSON.stringified to \"{}\"\n      // Bypass both by calling the REST API directly with SDK-provided auth.\n      const hostValue = client.config.host;\n      if (!hostValue) {\n        throw new Error(\n          \"Databricks host is not configured. Set DATABRICKS_HOST or configure client.config.host.\",\n        );\n      }\n      const host = hostValue.startsWith(\"http\")\n        ? hostValue\n        : `https://${hostValue}`;\n      const url = new URL(`/api/2.0/fs/files${resolvedPath}`, host);\n      url.searchParams.set(\"overwrite\", String(overwrite));\n\n      const headers = new Headers({\n        \"Content-Type\": \"application/octet-stream\",\n        \"User-Agent\": client.apiClient.userAgent(),\n      });\n      const fetchOptions: RequestInit = { method: \"PUT\", headers, body };\n\n      if (body instanceof ReadableStream) {\n        fetchOptions.duplex = \"half\";\n      } else if (body instanceof Buffer) {\n        headers.set(\"Content-Length\", String(body.length));\n      } else if (typeof body === \"string\") {\n        headers.set(\"Content-Length\", String(Buffer.byteLength(body)));\n      }\n\n      await client.config.authenticate(headers);\n\n      const res = await fetch(url.toString(), fetchOptions);\n\n      if (!res.ok) {\n        const text = await res.text();\n        logger.error(`Upload failed (${res.status}): ${text}`);\n        const safeMessage = text.length > 200 ? `${text.slice(0, 200)}…` : text;\n        throw new ApiError(\n          `Upload failed: ${safeMessage}`,\n          \"UPLOAD_FAILED\",\n          res.status,\n          undefined,\n          [],\n        );\n      }\n    });\n  }\n\n  async createDirectory(\n    client: WorkspaceClient,\n    directoryPath: string,\n  ): Promise<void> {\n    const resolvedPath = this.resolvePath(directoryPath);\n    return this.traced(\n      \"createDirectory\",\n      { \"files.path\": resolvedPath },\n      async () => {\n        await client.files.createDirectory({\n          directory_path: resolvedPath,\n        });\n      },\n    );\n  }\n\n  async delete(client: WorkspaceClient, filePath: string): Promise<void> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"delete\", { \"files.path\": resolvedPath }, async () => {\n      await client.files.delete({\n        file_path: resolvedPath,\n      });\n    });\n  }\n\n  async preview(\n    client: WorkspaceClient,\n    filePath: string,\n    options?: { maxChars?: number },\n  ): Promise<FilePreview> {\n    const resolvedPath = this.resolvePath(filePath);\n    return this.traced(\"preview\", { \"files.path\": resolvedPath }, async () => {\n      const meta = await this.metadata(client, filePath);\n      const isText = isTextContentType(meta.contentType);\n      const isImage = meta.contentType?.startsWith(\"image/\") || false;\n\n      if (!isText) {\n        return { ...meta, textPreview: null, isText: false, isImage };\n      }\n\n      const response = await client.files.download({\n        file_path: resolvedPath,\n      });\n      if (!response.contents) {\n        return { ...meta, textPreview: \"\", isText: true, isImage: false };\n      }\n\n      const reader = response.contents.getReader();\n      const decoder = new TextDecoder();\n      let preview = \"\";\n      const maxChars = options?.maxChars ?? 1024;\n\n      while (preview.length < maxChars) {\n        const { done, value } = await reader.read();\n        if (done) break;\n        preview += decoder.decode(value, { stream: true });\n      }\n      preview += decoder.decode();\n      await reader.cancel();\n\n      if (preview.length > maxChars) {\n        preview = preview.slice(0, maxChars);\n      }\n\n      return { ...meta, textPreview: preview, isText: true, isImage: false };\n    });\n  }\n}\n"],"mappings":";;;;;;;;AAyBA,MAAM,SAAS,aAAa,mBAAmB;;;;;;;;;;;;;;AAe/C,MAAM,6BAA6B,IAAI,mBAEpC;;;;;;;;AASH,SAAgB,2BACd,YACA,IACY;AACZ,QAAO,2BAA2B,IAAI,YAAY,GAAG;;AAUvD,IAAa,iBAAb,MAA4B;CAC1B,AAAiB,OAAO;CACxB,AAAQ;CACR,AAAiB;CAEjB,AAAiB;CACjB,AAAiB;CAKjB,YAAY,QAA8B;AACxC,OAAK,gBAAgB,OAAO;AAC5B,OAAK,qBAAqB,OAAO;AAEjC,OAAK,YAAY,iBAAiB,YAAY,KAAK,MAAM,OAAO,UAAU;AAC1E,OAAK,mBAAmB;GACtB,gBAAgB,KAAK,UAClB,UAAU,CACV,cAAc,yBAAyB;IACtC,aAAa;IACb,MAAM;IACP,CAAC;GACJ,mBAAmB,KAAK,UACrB,UAAU,CACV,gBAAgB,4BAA4B;IAC3C,aAAa;IACb,MAAM;IACP,CAAC;GACL;;CAGH,YAAY,UAA0B;AACpC,MAAI,SAAS,SAAS,KACpB,OAAM,IAAI,MACR,uDAAuD,SAAS,OAAO,IACxE;AAEH,MAAI,SAAS,SAAS,KAAK,CACzB,OAAM,IAAI,MAAM,oCAAoC;AAItD,MADiB,SAAS,MAAM,IAAI,CACvB,MAAM,MAAM,MAAM,KAAK,CAClC,OAAM,IAAI,MAAM,2CAAyC;AAE3D,MAAI,SAAS,WAAW,IAAI,EAAE;AAC5B,OAAI,CAAC,SAAS,WAAW,YAAY,CACnC,OAAM,IAAI,MACR,oIAED;AAEH,UAAO;;AAET,MAAI,CAAC,KAAK,cACR,OAAM,IAAI,MACR,qGACD;AAEH,SAAO,GAAG,KAAK,cAAc,GAAG;;CAGlC,MAAc,OACZ,WACA,YACA,IACY;EACZ,MAAM,YAAY,KAAK,KAAK;EAC5B,IAAI,UAAU;EAKd,MAAM,UAAU,2BAA2B,UAAU;AAErD,SAAO,KAAK,UAAU,gBACpB,SAAS,aACT;GACE,MAAM,SAAS;GACf,YAAY;IACV,mBAAmB;IACnB,GAAI,WAAW,EAAE;IACjB,GAAG;IACJ;GACF,EACD,OAAO,SAAe;AACpB,OAAI;IACF,MAAM,SAAS,MAAM,GAAG,KAAK;AAC7B,cAAU;AACV,SAAK,UAAU,EAAE,MAAM,eAAe,IAAI,CAAC;AAC3C,WAAO;YACA,OAAO;AACd,SAAK,gBAAgB,MAAe;AACpC,SAAK,UAAU;KACb,MAAM,eAAe;KACrB,SAAS,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;KAChE,CAAC;AACF,UAAM;aACE;AACR,SAAK,KAAK;IACV,MAAM,WAAW,KAAK,KAAK,GAAG;IAC9B,MAAM,cAAc;KAClB,mBAAmB;KACnB,SAAS,OAAO,QAAQ;KACzB;AACD,SAAK,iBAAiB,eAAe,IAAI,GAAG,YAAY;AACxD,SAAK,iBAAiB,kBAAkB,OAAO,UAAU,YAAY;;KAGzE;GAAE,MAAM,KAAK;GAAM,eAAe;GAAM,CACzC;;CAGH,MAAM,KACJ,QACA,eAC2B;EAC3B,MAAM,eAAe,gBACjB,KAAK,YAAY,cAAc,GAC/B,KAAK;AACT,MAAI,CAAC,aACH,OAAM,IAAI,MAAM,wDAAwD;AAG1E,SAAO,KAAK,OAAO,QAAQ,EAAE,cAAc,cAAc,EAAE,YAAY;GACrE,MAAM,UAA4B,EAAE;AACpC,cAAW,MAAM,SAAS,OAAO,MAAM,sBAAsB,EAC3D,gBAAgB,cACjB,CAAC,CACA,SAAQ,KAAK,MAAM;AAErB,UAAO;IACP;;CAGJ,MAAM,KACJ,QACA,UACA,SACiB;EACjB,MAAM,eAAe,KAAK,YAAY,SAAS;EAC/C,MAAM,UAAU,SAAS,WAAW;AACpC,SAAO,KAAK,OAAO,QAAQ,EAAE,cAAc,cAAc,EAAE,YAAY;GACrE,MAAM,WAAW,MAAM,KAAK,SAAS,QAAQ,SAAS;AACtD,OAAI,CAAC,SAAS,SACZ,QAAO;GAET,MAAM,SAAS,SAAS,SAAS,WAAW;GAC5C,MAAM,UAAU,IAAI,aAAa;GACjC,IAAI,SAAS;GACb,IAAI,YAAY;AAChB,UAAO,MAAM;IACX,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AACV,iBAAa,MAAM;AACnB,QAAI,YAAY,SAAS;AACvB,WAAM,OAAO,QAAQ;AACrB,WAAM,IAAI,MACR,mCAAmC,QAAQ,0CAC5C;;AAEH,cAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;;AAEnD,aAAU,QAAQ,QAAQ;AAC1B,UAAO;IACP;;CAGJ,MAAM,SACJ,QACA,UAC2B;EAC3B,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,YAAY,EAAE,cAAc,cAAc,EAAE,YAAY;AACzE,UAAO,OAAO,MAAM,SAAS,EAC3B,WAAW,cACZ,CAAC;IACF;;CAGJ,MAAM,OAAO,QAAyB,UAAoC;EACxE,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;AACvE,OAAI;AACF,UAAM,KAAK,SAAS,QAAQ,SAAS;AACrC,WAAO;YACA,OAAO;AACd,QAAI,iBAAiB,YAAY,MAAM,eAAe,IACpD,QAAO;AAET,UAAM;;IAER;;CAGJ,MAAM,SACJ,QACA,UACuB;EACvB,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,YAAY,EAAE,cAAc,cAAc,EAAE,YAAY;GACzE,MAAM,WAAW,MAAM,OAAO,MAAM,YAAY,EAC9C,WAAW,cACZ,CAAC;AACF,UAAO;IACL,eAAe,SAAS;IACxB,aAAa,oBACX,UACA,SAAS,iBACT,KAAK,mBACN;IACD,cAAc,SAAS;IACxB;IACD;;CAGJ,MAAM,OACJ,QACA,UACA,UACA,SACe;EACf,MAAM,eAAe,KAAK,YAAY,SAAS;AAE/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;GACvE,MAAM,OAAO;GACb,MAAM,YAAY,SAAS,aAAa;GAOxC,MAAM,YAAY,OAAO,OAAO;AAChC,OAAI,CAAC,UACH,OAAM,IAAI,MACR,0FACD;GAEH,MAAM,OAAO,UAAU,WAAW,OAAO,GACrC,YACA,WAAW;GACf,MAAM,MAAM,IAAI,IAAI,oBAAoB,gBAAgB,KAAK;AAC7D,OAAI,aAAa,IAAI,aAAa,OAAO,UAAU,CAAC;GAEpD,MAAM,UAAU,IAAI,QAAQ;IAC1B,gBAAgB;IAChB,cAAc,OAAO,UAAU,WAAW;IAC3C,CAAC;GACF,MAAM,eAA4B;IAAE,QAAQ;IAAO;IAAS;IAAM;AAElE,OAAI,gBAAgB,eAClB,cAAa,SAAS;YACb,gBAAgB,OACzB,SAAQ,IAAI,kBAAkB,OAAO,KAAK,OAAO,CAAC;YACzC,OAAO,SAAS,SACzB,SAAQ,IAAI,kBAAkB,OAAO,OAAO,WAAW,KAAK,CAAC,CAAC;AAGhE,SAAM,OAAO,OAAO,aAAa,QAAQ;GAEzC,MAAM,MAAM,MAAM,MAAM,IAAI,UAAU,EAAE,aAAa;AAErD,OAAI,CAAC,IAAI,IAAI;IACX,MAAM,OAAO,MAAM,IAAI,MAAM;AAC7B,WAAO,MAAM,kBAAkB,IAAI,OAAO,KAAK,OAAO;AAEtD,UAAM,IAAI,SACR,kBAFkB,KAAK,SAAS,MAAM,GAAG,KAAK,MAAM,GAAG,IAAI,CAAC,KAAK,QAGjE,iBACA,IAAI,QACJ,QACA,EAAE,CACH;;IAEH;;CAGJ,MAAM,gBACJ,QACA,eACe;EACf,MAAM,eAAe,KAAK,YAAY,cAAc;AACpD,SAAO,KAAK,OACV,mBACA,EAAE,cAAc,cAAc,EAC9B,YAAY;AACV,SAAM,OAAO,MAAM,gBAAgB,EACjC,gBAAgB,cACjB,CAAC;IAEL;;CAGH,MAAM,OAAO,QAAyB,UAAiC;EACrE,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,UAAU,EAAE,cAAc,cAAc,EAAE,YAAY;AACvE,SAAM,OAAO,MAAM,OAAO,EACxB,WAAW,cACZ,CAAC;IACF;;CAGJ,MAAM,QACJ,QACA,UACA,SACsB;EACtB,MAAM,eAAe,KAAK,YAAY,SAAS;AAC/C,SAAO,KAAK,OAAO,WAAW,EAAE,cAAc,cAAc,EAAE,YAAY;GACxE,MAAM,OAAO,MAAM,KAAK,SAAS,QAAQ,SAAS;GAClD,MAAM,SAAS,kBAAkB,KAAK,YAAY;GAClD,MAAM,UAAU,KAAK,aAAa,WAAW,SAAS,IAAI;AAE1D,OAAI,CAAC,OACH,QAAO;IAAE,GAAG;IAAM,aAAa;IAAM,QAAQ;IAAO;IAAS;GAG/D,MAAM,WAAW,MAAM,OAAO,MAAM,SAAS,EAC3C,WAAW,cACZ,CAAC;AACF,OAAI,CAAC,SAAS,SACZ,QAAO;IAAE,GAAG;IAAM,aAAa;IAAI,QAAQ;IAAM,SAAS;IAAO;GAGnE,MAAM,SAAS,SAAS,SAAS,WAAW;GAC5C,MAAM,UAAU,IAAI,aAAa;GACjC,IAAI,UAAU;GACd,MAAM,WAAW,SAAS,YAAY;AAEtC,UAAO,QAAQ,SAAS,UAAU;IAChC,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,MAAM;AAC3C,QAAI,KAAM;AACV,eAAW,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM,CAAC;;AAEpD,cAAW,QAAQ,QAAQ;AAC3B,SAAM,OAAO,QAAQ;AAErB,OAAI,QAAQ,SAAS,SACnB,WAAU,QAAQ,MAAM,GAAG,SAAS;AAGtC,UAAO;IAAE,GAAG;IAAM,aAAa;IAAS,QAAQ;IAAM,SAAS;IAAO;IACtE"}

package/dist/connectors/mcp/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","names":[],"sources":["../../../src/connectors/mcp/client.ts"],"mappings":";;;;;;;;;;;UAsJiB,mBAAA;EACf,SAAA;EACA,MAAA,EAAQ,KAAA;IAAQ,IAAA;IAAc,KAAA,EAAO,KAAA;EAAA;AAAA;;;;;;;;;;;;;;;;;cA+B1B,eAAA;EAAA,QAOD,aAAA;EAAA,QACA,YAAA;EAAA,QACA,MAAA;EAAA,QACA,OAAA;EAAA,QATF,WAAA;EAAA,QACA,UAAA;EAAA,QACA,SAAA;EAAA,QACA,MAAA;cAGE,aAAA,UACA,YAAA,QAAoB,OAAA,CAAQ,MAAA,mBAC5B,MAAA,EAAQ,aAAA,EACR,OAAA;IAAW,SAAA,GAAY,SAAA;IAAW,SAAA,UAAmB,KAAA;EAAA;EAqH/D;;;;;;;;;;;;;EArGM,UAAA,CACJ,SAAA,EAAW,iBAAA,KACV,OAAA,CAAQ,mBAAA;EAAA,QAoBH,UAAA;EAUF,OAAA,CAAQ,QAAA,EAAU,iBAAA,GAAoB,OAAA;EAqE5C,qBAAA,CAAA,GAAyB,mBAAA;EA2OO;;;;;;EAvNhC,uBAAA,CAAwB,UAAA;EAIlB,QAAA,CACJ,aAAA,UACA,IAAA,WACA,WAAA,GAAc,MAAA,kBACd,YAAA,GAAe,WAAA,GACd,OAAA;EAoDG,KAAA,CAAA,GAAS,OAAA;EAAA,QAMD,OAAA;EAAA,QA0FA,gBAAA;;;;;;UA0DA,kBAAA;AAAA"}
+{"version":3,"file":"client.d.ts","names":[],"sources":["../../../src/connectors/mcp/client.ts"],"mappings":";;;;;;;;;;;UAuJiB,mBAAA;EACf,SAAA;EACA,MAAA,EAAQ,KAAA;IAAQ,IAAA;IAAc,KAAA,EAAO,KAAA;EAAA;AAAA;;;;;;;;;;;;;;;;;cA+B1B,eAAA;EAAA,QAOD,aAAA;EAAA,QACA,YAAA;EAAA,QACA,MAAA;EAAA,QACA,OAAA;EAAA,QATF,WAAA;EAAA,QACA,UAAA;EAAA,QACA,SAAA;EAAA,QACA,MAAA;cAGE,aAAA,UACA,YAAA,QAAoB,OAAA,CAAQ,MAAA,mBAC5B,MAAA,EAAQ,aAAA,EACR,OAAA;IAAW,SAAA,GAAY,SAAA;IAAW,SAAA,UAAmB,KAAA;EAAA;EAqH/D;;;;;;;;;;;;;EArGM,UAAA,CACJ,SAAA,EAAW,iBAAA,KACV,OAAA,CAAQ,mBAAA;EAAA,QAoBH,UAAA;EAUF,OAAA,CAAQ,QAAA,EAAU,iBAAA,GAAoB,OAAA;EAqE5C,qBAAA,CAAA,GAAyB,mBAAA;EA6OO;;;;;;EAzNhC,uBAAA,CAAwB,UAAA;EAIlB,QAAA,CACJ,aAAA,UACA,IAAA,WACA,WAAA,GAAc,MAAA,kBACd,YAAA,GAAe,WAAA,GACd,OAAA;EAoDG,KAAA,CAAA,GAAS,OAAA;EAAA,QAMD,OAAA;EAAA,QA2FA,gBAAA;;;;;;UA2DA,kBAAA;AAAA"}

package/dist/connectors/mcp/client.js

@@ -1,4 +1,5 @@
 import { createLogger } from "../../logging/logger.js";
+import { APPKIT_USER_AGENT } from "../../context/client-options.js";
 import { assertResolvedHostSafe, checkMcpUrl } from "./host-policy.js";
 
 //#region src/connectors/mcp/client.ts
@@ -221,6 +222,7 @@ var AppKitMcpClient = class {
 		};
 		const authHeaders = await this.resolveAuthHeaders(options);
 		const headers = {
+			"User-Agent": APPKIT_USER_AGENT,
 			"Content-Type": "application/json",
 			Accept: "application/json, text/event-stream",
 			...authHeaders
@@ -258,6 +260,7 @@ var AppKitMcpClient = class {
 		if (this.closed) return;
 		const authHeaders = await this.resolveAuthHeaders(options);
 		const headers = {
+			"User-Agent": APPKIT_USER_AGENT,
 			"Content-Type": "application/json",
 			Accept: "application/json, text/event-stream",
 			...authHeaders