@combycode/llm-sdk 1.1.0 → 1.3.0

package/dist/index.browser.js

@@ -2524,7 +2524,7 @@ var QueueState = class {
       });
       this.processed++;
       if (response.ok) {
-        const body = await parseResponseBody(response, entry.request.responseType ?? "json");
+        const body = entry.request.responseType === "stream" ? response.body : await parseResponseBody(response, entry.request.responseType ?? "json");
         this.semaphore.release();
         entry.resolve({ status: response.status, headers: resHeaders, body });
         return;
@@ -22427,6 +22427,160 @@ async function* wrapParallel(raw, interval, moderate2) {
   }
 }
 
+// src/llm/files/retrieve.ts
+var DEFAULT_BASE = {
+  anthropic: "https://api.anthropic.com",
+  openai: "https://api.openai.com",
+  xai: "https://api.x.ai",
+  google: "https://generativelanguage.googleapis.com",
+  openrouter: "https://openrouter.ai/api"
+};
+var OCTET_STREAM = "application/octet-stream";
+function contentRequest(ctx, file) {
+  const base = ctx.baseURL ?? DEFAULT_BASE[ctx.provider];
+  const id = file.id;
+  if (ctx.provider === "anthropic") {
+    return {
+      url: `${base}/v1/files/${id}/content?beta=true`,
+      headers: {
+        "x-api-key": ctx.apiKey,
+        "anthropic-version": "2023-06-01",
+        "anthropic-beta": "files-api-2025-04-14",
+        accept: "application/binary"
+      }
+    };
+  }
+  if (ctx.provider === "openai" || ctx.provider === "xai" || ctx.provider === "openrouter") {
+    const containerId = file.ref?.containerId;
+    const path = containerId ? `/v1/containers/${containerId}/files/${id}/content` : `/v1/files/${id}/content`;
+    return { url: `${base}${path}`, headers: { authorization: `Bearer ${ctx.apiKey}` } };
+  }
+  if (ctx.provider === "google") {
+    return {
+      url: `${base}/v1beta/files/${id}:download?alt=media`,
+      headers: { "x-goog-api-key": ctx.apiKey }
+    };
+  }
+  throw new Error(`retrieveFile: no file-content endpoint for provider "${ctx.provider}"`);
+}
+function providerAuth(ctx, url) {
+  const base = ctx.baseURL ?? DEFAULT_BASE[ctx.provider];
+  if (!url.startsWith(base)) return {};
+  if (ctx.provider === "anthropic") {
+    return { "x-api-key": ctx.apiKey, "anthropic-version": "2023-06-01" };
+  }
+  if (ctx.provider === "google") return { "x-goog-api-key": ctx.apiKey };
+  return { authorization: `Bearer ${ctx.apiKey}` };
+}
+async function fetchFileResponse(ctx, file, responseType) {
+  if (file.url) {
+    return ctx.fetch({
+      url: file.url,
+      method: "GET",
+      headers: providerAuth(ctx, file.url),
+      body: void 0,
+      provider: ctx.provider,
+      model: "files",
+      responseType
+    });
+  }
+  if (file.id) {
+    const { url, headers } = contentRequest(ctx, file);
+    return ctx.fetch({
+      url,
+      method: "GET",
+      headers,
+      body: void 0,
+      provider: ctx.provider,
+      model: "files",
+      responseType
+    });
+  }
+  throw new Error("retrieveFile: FileOutput has neither `data`, `url`, nor `id`");
+}
+function bytesToBlob(bytes, type) {
+  const ab = new ArrayBuffer(bytes.byteLength);
+  new Uint8Array(ab).set(bytes);
+  return new Blob([ab], { type });
+}
+function singleChunkStream(bytes) {
+  return new ReadableStream({
+    start(controller) {
+      controller.enqueue(bytes);
+      controller.close();
+    }
+  });
+}
+var EXT_MIME = {
+  png: "image/png",
+  jpg: "image/jpeg",
+  jpeg: "image/jpeg",
+  gif: "image/gif",
+  webp: "image/webp",
+  svg: "image/svg+xml",
+  bmp: "image/bmp",
+  csv: "text/csv",
+  tsv: "text/tab-separated-values",
+  txt: "text/plain",
+  json: "application/json",
+  xml: "application/xml",
+  html: "text/html",
+  md: "text/markdown",
+  pdf: "application/pdf",
+  zip: "application/zip",
+  parquet: "application/vnd.apache.parquet",
+  xlsx: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+};
+function mimeFromName(name) {
+  const ext = name?.includes(".") ? name.split(".").pop()?.toLowerCase() : void 0;
+  return ext ? EXT_MIME[ext] : void 0;
+}
+function header(headers, name) {
+  const lower = name.toLowerCase();
+  for (const k in headers) if (k.toLowerCase() === lower) return headers[k];
+  return void 0;
+}
+function filenameFromDisposition(cd) {
+  if (!cd) return void 0;
+  const ext = cd.match(/filename\*=[^']*''([^;]+)/i);
+  if (ext) {
+    try {
+      return decodeURIComponent(ext[1].trim());
+    } catch {
+    }
+  }
+  const plain = cd.match(/filename="?([^";]+)"?/i);
+  return plain ? plain[1].trim() : void 0;
+}
+async function retrieveFile(file, ctx) {
+  if (file.data) {
+    const blob2 = bytesToBlob(base64ToBytes(file.data), file.mimeType ?? OCTET_STREAM);
+    return { blob: blob2, name: file.name, mimeType: blob2.type, size: blob2.size };
+  }
+  const res = await fetchFileResponse(ctx, file, "arraybuffer");
+  const headers = res.headers ?? {};
+  const name = filenameFromDisposition(header(headers, "content-disposition")) ?? file.name;
+  const mimeType = header(headers, "content-type")?.split(";")[0].trim() || file.mimeType || mimeFromName(name) || OCTET_STREAM;
+  const blob = new Blob([res.body], { type: mimeType });
+  return { blob, name, mimeType, size: blob.size };
+}
+async function streamFile(file, ctx) {
+  if (file.data) {
+    const bytes = base64ToBytes(file.data);
+    return { stream: singleChunkStream(bytes), name: file.name, mimeType: file.mimeType, size: bytes.byteLength };
+  }
+  const res = await fetchFileResponse(ctx, file, "stream");
+  const headers = res.headers ?? {};
+  const len = header(headers, "content-length");
+  const name = filenameFromDisposition(header(headers, "content-disposition")) ?? file.name;
+  return {
+    stream: res.body,
+    name,
+    mimeType: header(headers, "content-type")?.split(";")[0].trim() || file.mimeType || mimeFromName(name),
+    size: len ? Number(len) : void 0
+  };
+}
+
 // src/util/duration.ts
 var UNIT_MS = {
   s: 1e3,
@@ -22669,6 +22823,28 @@ var LLMClient = class {
     else next.output = result;
     return next;
   }
+  // ─── File retrieval (hosted-tool output files) ─────────────────────────
+  retrieveContext() {
+    return {
+      provider: this.provider,
+      apiKey: this.apiKey,
+      fetch: this.fetchFn,
+      baseURL: this.adapter.baseURL()
+    };
+  }
+  /** Fetch a hosted-tool output file (e.g. a code-execution file from
+   *  `response.files`): its bytes as a `Blob` plus `name` / `mimeType` / `size`.
+   *  Resolves inline `data`, a `url`, or a provider file `id` — all through this
+   *  client's provider + auth + engine. */
+  retrieveFile(file) {
+    return retrieveFile(file, this.retrieveContext());
+  }
+  /** Stream a hosted-tool output file: a `ReadableStream<Uint8Array>` plus
+   *  best-effort `name` / `mimeType` / `size` (from the response headers). For large
+   *  files piped straight to a file / GridFS / HTTP response without buffering. */
+  streamFile(file) {
+    return streamFile(file, this.retrieveContext());
+  }
   /** Submit a request. Returns the parsed CompletionResponse. */
   async complete(input, options = {}) {
     const rawMessages = normalizeInput(input);
@@ -22880,17 +23056,18 @@ var LLMClient = class {
     let usage = emptyUsage();
     let finishReason = "stop";
     let moderationReport;
+    const files = [];
     const fetchStream = this.fetchStreamFn;
-    const adapter = this.adapter;
     const queueName = this.queueName;
     const priority = this.priority;
+    const parseStream = this.adapter.createStreamParser();
     async function* rawEvents() {
       for await (const sseEvent of fetchStream(httpReq, {
         queueName,
         priority,
         ctx
       })) {
-        for (const ev of adapter.parseStreamEvent(sseEvent)) yield ev;
+        for (const ev of parseStream(sseEvent)) yield ev;
       }
     }
     const mod = options.moderation;
@@ -22928,6 +23105,9 @@ var LLMClient = class {
         case "done":
           finishReason = event.finishReason;
           break;
+        case "file":
+          files.push(event.file);
+          break;
         case "moderation":
           moderationReport = this.mergeModeration(
             moderationReport,
@@ -22950,6 +23130,7 @@ var LLMClient = class {
       toolCalls: [],
       thinking: thinking || null,
       media: [],
+      ...files.length ? { files } : {},
       ...moderationReport ? { moderation: moderationReport } : {},
       latencyMs: performance.now() - start,
       raw: null
@@ -23210,6 +23391,22 @@ function anthropicRequestTier(t) {
 function anthropicBilledTier(raw) {
   return typeof raw === "string" && raw ? { serviceTier: raw, pricingTier: raw } : {};
 }
+function filesFromCodeExecBlock(block) {
+  if (block.type !== "bash_code_execution_tool_result" && block.type !== "code_execution_tool_result") {
+    return [];
+  }
+  const result = block.content;
+  if (!result || result.type !== "bash_code_execution_result" && result.type !== "code_execution_result" || !Array.isArray(result.content)) {
+    return [];
+  }
+  const files = [];
+  for (const out of result.content) {
+    if ((out.type === "bash_code_execution_output" || out.type === "code_execution_output") && typeof out.file_id === "string") {
+      files.push({ id: out.file_id, source: "code_execution" });
+    }
+  }
+  return files;
+}
 var AnthropicAdapter = class {
   name = "anthropic";
   apiKey;
@@ -23305,7 +23502,10 @@ var AnthropicAdapter = class {
     });
     const headers = {};
     if (hasFileRef) headers["anthropic-beta"] = "files-api-2025-04-14";
-    return { body, headers };
+    const usesCodeExec = req.tools?.some(
+      (t) => !isFunctionTool(t) && t.type === "code_interpreter"
+    );
+    return { body, headers, ...usesCodeExec ? { path: "/v1/messages?beta=true" } : {} };
   }
   enableStreaming(providerReq, _req) {
     providerReq.body.stream = true;
@@ -23387,15 +23587,8 @@ var AnthropicAdapter = class {
         };
         content.push(tc);
         toolCalls.push(tc);
-      } else if (block.type === "code_execution_tool_result") {
-        const result = block.content;
-        if (result?.type === "code_execution_result" && Array.isArray(result.content)) {
-          for (const out of result.content) {
-            if (out.type === "code_execution_output" && typeof out.file_id === "string") {
-              files.push({ id: out.file_id, source: "code_execution" });
-            }
-          }
-        }
+      } else {
+        files.push(...filesFromCodeExecBlock(block));
       }
     }
     const finishReason = extractFinishReason(toolCalls.length > 0, r.stop_reason, {
@@ -23434,6 +23627,8 @@ var AnthropicAdapter = class {
       if (block.type === "tool_use") {
         return [{ type: "tool_call_start", id: block.id, name: block.name }];
       }
+      const files = filesFromCodeExecBlock(block);
+      if (files.length) return files.map((file) => ({ type: "file", file }));
     }
     if (type === "content_block_stop") {
     }
@@ -23457,6 +23652,11 @@ var AnthropicAdapter = class {
     }
     return [];
   }
+  /** Stateless — every event is self-contained (code-execution result blocks
+   *  arrive complete in a single content_block_start). */
+  createStreamParser() {
+    return (event) => this.parseStreamEvent(event);
+  }
   parseUsage(u) {
     if (!u) return emptyUsage();
     const inputTokens = u.input_tokens ?? 0;
@@ -23899,7 +24099,9 @@ var GoogleAdapter = class {
     const content = [];
     const toolCalls = [];
     const media = [];
+    const files = [];
     let thinking = null;
+    const hasCodeExec = parts.some((p) => p.executableCode || p.codeExecutionResult);
     for (const part of parts) {
       if (part.text !== void 0 && !part.thought) {
         content.push({ type: "text", text: part.text });
@@ -23910,7 +24112,9 @@ var GoogleAdapter = class {
       if (part.inlineData) {
         const inline = part.inlineData;
         const mime = inline.mimeType;
-        if (mime.startsWith("image/")) {
+        if (hasCodeExec) {
+          files.push({ data: inline.data, mimeType: mime, source: "code_execution" });
+        } else if (mime.startsWith("image/")) {
           const p = {
             type: "image_output",
             mediaId: "",
@@ -23971,11 +24175,24 @@ var GoogleAdapter = class {
       toolCalls,
       thinking,
       media,
+      ...files.length ? { files } : {},
       latencyMs,
       raw
     };
   }
   parseStreamEvent(event) {
+    return this.streamEvents(event, { codeExec: false });
+  }
+  /** Stateful — Google splits the code-execution marker (`executableCode` /
+   *  `codeExecutionResult`) and the produced file (`inlineData`) across parts and
+   *  often across SSE events. The closure remembers "code execution began in this
+   *  stream" so a later `inlineData` blob is routed to `files` (a code-exec
+   *  artifact) rather than `media` (conversational output). */
+  createStreamParser() {
+    const state = { codeExec: false };
+    return (event) => this.streamEvents(event, state);
+  }
+  streamEvents(event, state) {
     const data = JSON.parse(event.data);
     const candidates = data.candidates ?? [];
     const candidate = candidates[0];
@@ -23989,6 +24206,9 @@ var GoogleAdapter = class {
     const rawContent = candidate.content ?? {};
     const parts = rawContent.parts ?? [];
     const events = [];
+    for (const part of parts) {
+      if (part.executableCode || part.codeExecutionResult) state.codeExec = true;
+    }
     for (const part of parts) {
       if (part.text !== void 0 && !part.thought)
         events.push({ type: "text", text: part.text });
@@ -23996,10 +24216,17 @@ var GoogleAdapter = class {
       if (part.inlineData) {
         const inline = part.inlineData;
         const mime = inline.mimeType;
-        const mediaType = mime.startsWith("image/") ? "image" : mime.startsWith("audio/") ? "audio" : "video";
-        events.push({ type: "media_start", mediaType, mimeType: mime });
-        events.push({ type: "media_chunk", data: inline.data });
-        events.push({ type: "media_end" });
+        if (state.codeExec) {
+          events.push({
+            type: "file",
+            file: { data: inline.data, mimeType: mime, source: "code_execution" }
+          });
+        } else {
+          const mediaType = mime.startsWith("image/") ? "image" : mime.startsWith("audio/") ? "audio" : "video";
+          events.push({ type: "media_start", mediaType, mimeType: mime });
+          events.push({ type: "media_chunk", data: inline.data });
+          events.push({ type: "media_end" });
+        }
       }
       if (part.functionCall) {
         const fc = part.functionCall;
@@ -24294,6 +24521,10 @@ var GoogleInteractionsAdapter = class {
     }
     return events;
   }
+  /** Stateless — the Interactions adapter surfaces no code-execution file outputs. */
+  createStreamParser() {
+    return (event) => this.parseStreamEvent(event);
+  }
   parseUsage(u) {
     if (!u) return emptyUsage();
     const input = u.total_input_tokens ?? u.prompt_tokens ?? 0;
@@ -25259,6 +25490,10 @@ var OpenAIAdapter = class {
     }
     return events;
   }
+  /** Stateless — Chat Completions has no hosted code-execution file outputs. */
+  createStreamParser() {
+    return (event) => this.parseStreamEvent(event);
+  }
   parseUsage(u) {
     if (!u) return emptyUsage();
     const input = u.prompt_tokens ?? u.input_tokens ?? 0;
@@ -25774,6 +26009,33 @@ function filenameForMime(mimeType) {
   if (mimeType.startsWith("image/")) return `file.${mimeType.slice("image/".length)}`;
   return "file.bin";
 }
+function filesFromResponsesOutputItem(item) {
+  const files = [];
+  const type = item.type;
+  if (type === "message") {
+    for (const c of item.content ?? []) {
+      if (c.type !== "output_text") continue;
+      for (const a of c.annotations ?? []) {
+        if (a.type === "container_file_citation" && typeof a.file_id === "string") {
+          files.push({
+            id: a.file_id,
+            ...typeof a.filename === "string" ? { name: a.filename } : {},
+            ...typeof a.container_id === "string" ? { ref: { containerId: a.container_id } } : {},
+            source: "code_execution"
+          });
+        }
+      }
+    }
+  }
+  if (type === "code_interpreter_call") {
+    for (const out of item.outputs ?? []) {
+      if (out.type === "image" && typeof out.url === "string") {
+        files.push({ url: out.url, source: "code_execution" });
+      }
+    }
+  }
+  return files;
+}
 var OpenAIResponsesAdapter = class {
   name = "openai";
   apiKey;
@@ -25949,10 +26211,12 @@ var OpenAIResponsesAdapter = class {
     const content = [];
     const toolCalls = [];
     const media = [];
+    const files = [];
     let thinking = null;
     let text = "";
     for (const item of output) {
       const type = item.type;
+      files.push(...filesFromResponsesOutputItem(item));
       if (type === "message") {
         const itemContent = item.content ?? [];
         for (const c of itemContent) {
@@ -26012,6 +26276,7 @@ var OpenAIResponsesAdapter = class {
       toolCalls,
       thinking,
       media,
+      ...files.length ? { files } : {},
       ...moderation ? { moderation } : {},
       latencyMs,
       raw
@@ -26053,6 +26318,7 @@ var OpenAIResponsesAdapter = class {
     }
     if (type === "response.output_item.done") {
       const item = data.item;
+      for (const file of filesFromResponsesOutputItem(item)) events.push({ type: "file", file });
       if (item?.type === "function_call") {
         events.push({ type: "tool_call_end", id: item.call_id ?? "" });
       }
@@ -26083,6 +26349,11 @@ var OpenAIResponsesAdapter = class {
     }
     return events;
   }
+  /** Stateless — each output item finalizes with all its file annotations in a
+   *  single response.output_item.done event. */
+  createStreamParser() {
+    return (event) => this.parseStreamEvent(event);
+  }
   parseUsage(u) {
     if (!u) return emptyUsage();
     const input = u.input_tokens ?? 0;
@@ -27738,6 +28009,7 @@ var AgentLoop = class _AgentLoop {
   _toolTimeout;
   _maxSteps;
   _guardrails;
+  _toolInputGuardrails;
   _policy;
   _approve;
   _checkpoint;
@@ -27765,6 +28037,7 @@ var AgentLoop = class _AgentLoop {
     this._toolTimeout = config.toolTimeout ?? DEFAULT_TOOL_TIMEOUT_MS;
     this._maxSteps = config.maxSteps !== void 0 && config.maxSteps > 0 ? config.maxSteps : DEFAULT_MAX_STEPS;
     this._guardrails = config.guardrails ?? [];
+    this._toolInputGuardrails = config.toolInputGuardrails ?? [];
     this._policy = config.policy ?? null;
     this._approve = config.approve ?? null;
     this._checkpoint = config.checkpoint ?? null;
@@ -27802,6 +28075,19 @@ var AgentLoop = class _AgentLoop {
   get model() {
     return this.client.model;
   }
+  // ─── File retrieval (hosted-tool output files) ──────────────────────────
+  /** Fetch a hosted-tool output file from the run's `response.files` — its bytes
+   *  as a `Blob` plus `name` / `mimeType` / `size`. Delegates to the underlying
+   *  client, so it uses this agent's provider + key (same as `retrieveFile` on a
+   *  `complete()` result). */
+  retrieveFile(file) {
+    return this.client.retrieveFile(file);
+  }
+  /** Stream a hosted-tool output file (large files) — a `ReadableStream` plus
+   *  best-effort `name` / `mimeType` / `size`. Delegates to the underlying client. */
+  streamFile(file) {
+    return this.client.streamFile(file);
+  }
   get system() {
     return this._system;
   }
@@ -27983,6 +28269,10 @@ var AgentLoop = class _AgentLoop {
       toolCalls: lastResponse?.toolCalls ?? [],
       thinking: lastResponse?.thinking ?? null,
       media: lastResponse?.media ?? [],
+      // Propagate hosted-tool file outputs + inline-moderation result from the final
+      // LLM response (e.g. code-execution files produced during the run).
+      ...lastResponse?.files ? { files: lastResponse.files } : {},
+      ...lastResponse?.moderation ? { moderation: lastResponse.moderation } : {},
       latencyMs: performance.now() - startPerf,
       raw: lastResponse?.raw ?? null
     };
@@ -28177,6 +28467,8 @@ var AgentLoop = class _AgentLoop {
       toolCalls: lastResponse?.toolCalls ?? [],
       thinking: lastResponse?.thinking ?? null,
       media: [],
+      ...lastResponse?.files ? { files: lastResponse.files } : {},
+      ...lastResponse?.moderation ? { moderation: lastResponse.moderation } : {},
       latencyMs: performance.now() - startPerf,
       raw: null
     };
@@ -28263,6 +28555,18 @@ var AgentLoop = class _AgentLoop {
       runTrace
     );
     if (!lookup.found) return lookup.errorResult;
+    for (const g of this._toolInputGuardrails) {
+      const decision = await g.check({
+        toolName: tc.name,
+        arguments: tc.arguments,
+        callId: tc.id,
+        step,
+        trace: { sessionId: runTrace.sessionId, requestId: runTrace.requestId, callId: tc.id }
+      });
+      if (!decision.pass) {
+        return this.buildDeniedResult(tc, decision.reason, reports);
+      }
+    }
     if (this._policy !== null) {
       const target = {
         kind: "tool",
@@ -28279,7 +28583,7 @@ var AgentLoop = class _AgentLoop {
     try {
       const baseCtx = { step, callId: tc.id, metrics, trace: { sessionId: runTrace.sessionId, requestId: runTrace.requestId, callId: tc.id } };
       const result = await executeWithTimeout(lookup.tool, tc, baseCtx, this._toolTimeout);
-      return await this.buildSuccessResult(tc, result, runId, step, metrics, reports, toolStart, runTrace);
+      return await this.buildSuccessResult(tc, result, runId, step, metrics, reports, toolStart, runTrace, lookup.tool, baseCtx);
     } catch (e) {
       return handleToolError(e, tc, this.hooks, runId, this.id, step, metrics, reports, toolStart, runTrace);
     }
@@ -28370,7 +28674,7 @@ var AgentLoop = class _AgentLoop {
       try {
         const baseCtx = { step, callId: tc.id, metrics, trace: { sessionId: runTrace.sessionId, requestId: runTrace.requestId, callId: tc.id } };
         const result = await executeWithTimeout(tool, tc, baseCtx, this._toolTimeout);
-        return await this.buildSuccessResult(tc, result, runId, step, metrics, reports, toolStart, runTrace);
+        return await this.buildSuccessResult(tc, result, runId, step, metrics, reports, toolStart, runTrace, tool, baseCtx);
       } catch (e) {
         return handleToolError(e, tc, this.hooks, runId, this.id, step, metrics, reports, toolStart, runTrace);
       }
@@ -28382,9 +28686,16 @@ var AgentLoop = class _AgentLoop {
     return this.buildDeniedResult(tc, denyMsg, reports);
   }
   /** Emit onToolCallComplete, push report, and return success content part. */
-  async buildSuccessResult(tc, result, runId, step, metrics, reports, toolStart, runTrace) {
+  async buildSuccessResult(tc, result, runId, step, metrics, reports, toolStart, runTrace, tool, context) {
     const latencyMs = performance.now() - toolStart;
     const resultStr = typeof result === "string" ? result : JSON.stringify(result);
+    let customData;
+    if (tool?.customDataExtractor && context) {
+      try {
+        customData = await tool.customDataExtractor(result, tc.arguments, context);
+      } catch {
+      }
+    }
     await this.hooks.emit("onToolCallComplete", {
       runId,
       agentId: this.id,
@@ -28406,7 +28717,8 @@ var AgentLoop = class _AgentLoop {
       latencyMs,
       skipped: false,
       error: null,
-      metrics: Object.fromEntries(metrics)
+      metrics: Object.fromEntries(metrics),
+      ...customData !== void 0 ? { customData } : {}
     });
     return { type: "tool_result", id: tc.id, content: resultStr };
   }
@@ -34410,7 +34722,13 @@ async function complete(opts) {
         serviceTier
       });
     }
-    const result = { text: res.text, response: res };
+    const result = {
+      text: res.text,
+      response: res,
+      // Bound to this call's client (same provider/model/key/engine).
+      retrieveFile: (file) => llm.retrieveFile(file),
+      streamFile: (file) => llm.streamFile(file)
+    };
     if (opts.structured?.schema) {
       result.parsed = parseStructured(res.text);
     }

package/dist/index.d.ts

@@ -49,6 +49,7 @@ export { isFunctionTool, isBuiltinTool } from './llm/types/tools';
 export type { FunctionTool, BuiltinTool, McpToolParams, Tool, ToolChoice, JsonSchema } from './llm/types/tools';
 export { emptyUsage } from './llm/types/response';
 export type { CompletionResponse, FileOutput, FinishReason, Usage } from './llm/types/response';
+export type { FileStream, RetrievedFile } from './llm/files/retrieve';
 export type { NormalizedRequest, ReasoningContext, ThinkingConfig, CacheConfig } from './llm/types/request';
 export type { MediaStreamType, StreamEvent } from './llm/types/stream';
 export type { ExecuteOptions } from './llm/types/options';
@@ -98,7 +99,7 @@ export type { ConversationHistoryConfig, HistoryEntry, HistorySnapshot } from '.
 export { AgentLoop } from './agent/loop';
 export type { AgentLoopConfig } from './agent/loop-config';
 export type { AgentLoopSnapshot, AgentRunReport, AgentStreamEvent, AgentTool, ContentClass, LearnInput, StepReport, TokenCountContext, TokenCounter, ToolCallReport, ToolExecutionContext } from './agent/types';
-export type { Guardrail, GuardrailDecision, GuardrailPass, GuardrailTrip, GuardrailCheckContext, InputGuardrailContext, OutputGuardrailContext, GuardrailTriggeredContext } from './agent/guardrail-types';
+export type { Guardrail, GuardrailDecision, GuardrailPass, GuardrailTrip, GuardrailCheckContext, InputGuardrailContext, OutputGuardrailContext, GuardrailTriggeredContext, ToolInputGuardrail, ToolInputGuardrailContext, ToolInputGuardrailDecision } from './agent/guardrail-types';
 export { BearerKeyAuth } from './server/auth';
 export type { AuthPlugin, AuthVerifyResult } from './server/auth';
 export { dispatch } from './server/dispatch';