@anvia/core 0.4.1 → 0.5.0

package/dist/extractor/index.js

@@ -2,14 +2,14 @@ import {
   ExtractionError,
   Extractor,
   ExtractorBuilder
-} from "../chunk-4PHDOEKY.js";
-import "../chunk-BD66WU2B.js";
-import "../chunk-TVIADATY.js";
+} from "../chunk-Z7HY4WU7.js";
+import "../chunk-365W6FJD.js";
+import "../chunk-ORLEOXD7.js";
 import "../chunk-XUUY2L2D.js";
 import "../chunk-XXT2UCAR.js";
 import "../chunk-YK4WAAS4.js";
-import "../chunk-CY7SB5FG.js";
-import "../chunk-OIA4CN5V.js";
+import "../chunk-S7EOE6EC.js";
+import "../chunk-P425B6GR.js";
 import "../chunk-6GJDBBDC.js";
 import "../chunk-EFGX3EX5.js";
 import "../chunk-OIMLU4SF.js";

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 export { AgentBuilder, MaxTurnsError, PromptCancelledError } from './agent/index.js';
-export { A as AgentChildStreamEvent, a as AgentStreamEvent, P as PromptResponse, c as cancelPrompt, b as createHook, r as requestToolApproval, d as runControl, s as skipTool, t as toolCallControl } from './agent-B-ls5y_g.js';
+export { A as AgentChildStreamEvent, a as AgentStreamEvent, P as PromptResponse, c as cancelPrompt, b as createHook, r as requestToolApproval, d as runControl, s as skipTool, t as toolCallControl } from './agent-CWGuTdMJ.js';
 export { A as AssistantContent, a as AssistantMessage, C as CompletionModel, b as CompletionRequest, c as CompletionResponse, D as Document, I as ImageContent, J as JsonObject, d as JsonPrimitive, e as JsonValue, M as Message, S as SystemMessage, T as Text, f as ToolCall, g as ToolDefinition, h as ToolMessage, i as ToolResult, j as ToolResultContent, U as Usage, k as UserContent, l as UserMessage } from './types-C54aNoCd.js';
 export { MemoryStore } from './memory/index.js';
 export { Z as ZodSchema } from './zod-schema-C7F4clpm.js';
@@ -7,12 +7,12 @@ export { loadSkills, skill } from './skills/index.js';
 export { S as SkillValidationError } from './types-T9rlOIUc.js';
 export { C as CreateToolOptions, c as createThinkTool, a as createTool } from './think-tool-ByArsrxe.js';
 export { A as AnyTool, T as Tool, a as ToolApprovalContext, b as ToolApprovalPolicy, c as ToolCallContext, d as ToolCallStreamEvent } from './tool-C3ciF-VG.js';
+export { A as AgentMiddleware, C as CompletionRequestMiddlewareArgs, a as CompletionRequestMiddlewareResult, b as CompletionResponseMiddlewareArgs, c as CompletionResponseMiddlewareResult, T as ToolInputMiddlewareArgs, d as ToolInputMiddlewareResult, e as ToolMiddleware, f as ToolOutputMiddlewareArgs, g as ToolOutputMiddlewareResult, h as ToolResultMiddlewareArgs, i as createMiddleware, j as createToolMiddleware } from './middleware-qfm1EFKK.js';
 import './types-UhcgW8Fo.js';
 import '@modelcontextprotocol/sdk/client/sse.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import './observability/index.js';
-import './middleware-CGiEIaBx.js';
-import './types-IB2e9u5M.js';
 import './vector-store/index.js';
+import './types-IB2e9u5M.js';
 import 'zod';

package/dist/index.js

@@ -2,10 +2,10 @@ import {
   SkillValidationError,
   loadSkills,
   skill
-} from "./chunk-L6JCKFAX.js";
+} from "./chunk-DUP7FMAF.js";
 import {
   AgentBuilder
-} from "./chunk-BD66WU2B.js";
+} from "./chunk-365W6FJD.js";
 import {
   MaxTurnsError,
   PromptCancelledError,
@@ -15,16 +15,18 @@ import {
   runControl,
   skipTool,
   toolCallControl
-} from "./chunk-TVIADATY.js";
+} from "./chunk-ORLEOXD7.js";
 import "./chunk-XUUY2L2D.js";
 import "./chunk-XXT2UCAR.js";
 import "./chunk-YK4WAAS4.js";
 import {
-  createThinkTool
-} from "./chunk-CY7SB5FG.js";
+  createMiddleware,
+  createThinkTool,
+  createToolMiddleware
+} from "./chunk-S7EOE6EC.js";
 import {
   createTool
-} from "./chunk-OIA4CN5V.js";
+} from "./chunk-P425B6GR.js";
 import {
   AssistantContent,
   Message,
@@ -44,8 +46,10 @@ export {
   UserContent,
   cancelPrompt,
   createHook,
+  createMiddleware,
   createThinkTool,
   createTool,
+  createToolMiddleware,
   loadSkills,
   requestToolApproval,
   runControl,

package/dist/internal/agent.d.ts

@@ -1,8 +1,8 @@
-export { e as Agent, j as AgentEventAppendInput, k as AgentEventRecord, h as AgentEventStore, l as AgentEventStoreInclude, i as AgentEventStoreOptions, x as AgentEventStoreRegistration, y as AgentOptions, z as AgentSession, B as AgentToolOptions, E as DEFAULT_MAX_TURNS, D as DynamicContextOptions, F as DynamicContextRegistration, f as DynamicToolOptions, G as DynamicToolRegistration } from '../agent-B-ls5y_g.js';
+export { e as Agent, j as AgentEventAppendInput, k as AgentEventRecord, h as AgentEventStore, l as AgentEventStoreInclude, i as AgentEventStoreOptions, I as AgentEventStoreRegistration, J as AgentOptions, K as AgentSession, L as AgentToolOptions, M as DEFAULT_MAX_TURNS, D as DynamicContextOptions, N as DynamicContextRegistration, f as DynamicToolOptions, O as DynamicToolRegistration } from '../agent-CWGuTdMJ.js';
 import '../types-C54aNoCd.js';
 import '../memory/index.js';
 import '../observability/index.js';
 import '../tool-C3ciF-VG.js';
-import '../middleware-CGiEIaBx.js';
+import '../middleware-qfm1EFKK.js';
 import '../types-IB2e9u5M.js';
 import '../vector-store/index.js';

package/dist/internal/agent.js

@@ -2,11 +2,11 @@ import {
   Agent,
   AgentSession,
   DEFAULT_MAX_TURNS
-} from "../chunk-TVIADATY.js";
+} from "../chunk-ORLEOXD7.js";
 import "../chunk-XUUY2L2D.js";
 import "../chunk-YK4WAAS4.js";
-import "../chunk-CY7SB5FG.js";
-import "../chunk-OIA4CN5V.js";
+import "../chunk-S7EOE6EC.js";
+import "../chunk-P425B6GR.js";
 import "../chunk-6GJDBBDC.js";
 import "../chunk-EFGX3EX5.js";
 import "../chunk-OIMLU4SF.js";

package/dist/loaders/index.d.ts

@@ -36,6 +36,19 @@ type PdfPageWithPath = {
     pageNumber: number;
     text: string;
 };
+
+declare function fileToDocument(file: FileReadWithPath): Document;
+declare function fileLoaderToDocuments(loader: AsyncIterable<FileReadWithPath>): Promise<Document[]>;
+declare function pdfToDocument(pdf: PdfReadWithPath): Document;
+declare function pdfLoaderToDocuments(loader: AsyncIterable<PdfReadWithPath>): Promise<Document[]>;
+declare function pdfPageToDocument(page: PdfPageWithPath): Document;
+declare function pdfPageLoaderToDocuments(loader: AsyncIterable<PdfPageWithPath>): Promise<Document[]>;
+
+type LoaderValue$1<T> = T extends {
+    ok: true;
+    value: infer Value;
+} ? Value : never;
+type UnwrapLoaderResult$1<T> = [LoaderValue$1<T>] extends [never] ? T : LoaderValue$1<T>;
 declare class FileLoader<T = LoaderResult<FileSource>> implements AsyncIterable<T> {
     private readonly source;
     private readonly mode;
@@ -47,9 +60,15 @@ declare class FileLoader<T = LoaderResult<FileSource>> implements AsyncIterable<
     static fromBytesMany(bytes: Array<Uint8Array | ArrayBuffer>): FileLoader<LoaderResult<FileSource>>;
     read(): FileLoader<LoaderResult<string>>;
     readWithPath(): FileLoader<LoaderResult<FileReadWithPath>>;
-    ignoreErrors(): FileLoader<UnwrapLoaderResult<T>>;
+    ignoreErrors(): FileLoader<UnwrapLoaderResult$1<T>>;
     [Symbol.asyncIterator](): AsyncIterator<T>;
 }
+
+type LoaderValue<T> = T extends {
+    ok: true;
+    value: infer Value;
+} ? Value : never;
+type UnwrapLoaderResult<T> = [LoaderValue<T>] extends [never] ? T : LoaderValue<T>;
 declare class PdfFileLoader<T = LoaderResult<PdfSource>> implements AsyncIterable<T> {
     private readonly source;
     private readonly mode;
@@ -67,16 +86,5 @@ declare class PdfFileLoader<T = LoaderResult<PdfSource>> implements AsyncIterabl
     ignoreErrors(): PdfFileLoader<UnwrapLoaderResult<T>>;
     [Symbol.asyncIterator](): AsyncIterator<T>;
 }
-declare function fileToDocument(file: FileReadWithPath): Document;
-declare function fileLoaderToDocuments(loader: AsyncIterable<FileReadWithPath>): Promise<Document[]>;
-declare function pdfToDocument(pdf: PdfReadWithPath): Document;
-declare function pdfLoaderToDocuments(loader: AsyncIterable<PdfReadWithPath>): Promise<Document[]>;
-declare function pdfPageToDocument(page: PdfPageWithPath): Document;
-declare function pdfPageLoaderToDocuments(loader: AsyncIterable<PdfPageWithPath>): Promise<Document[]>;
-type LoaderValue<T> = T extends {
-    ok: true;
-    value: infer Value;
-} ? Value : never;
-type UnwrapLoaderResult<T> = [LoaderValue<T>] extends [never] ? T : LoaderValue<T>;
 
 export { FileLoader, type FileReadWithPath, type FileSource, type LoaderResult, PdfFileLoader, type PdfPage, type PdfPageWithPath, type PdfReadWithPath, type PdfSource, fileLoaderToDocuments, fileToDocument, pdfLoaderToDocuments, pdfPageLoaderToDocuments, pdfPageToDocument, pdfToDocument };

package/dist/loaders/index.js

@@ -1,7 +1,85 @@
-// src/loaders/index.ts
+// src/loaders/documents.ts
+function fileToDocument(file) {
+  return {
+    id: file.path,
+    text: file.text,
+    additionalProps: {
+      source: file.path,
+      mediaType: "text/plain"
+    }
+  };
+}
+async function fileLoaderToDocuments(loader) {
+  const documents = [];
+  for await (const file of loader) {
+    documents.push(fileToDocument(file));
+  }
+  return documents;
+}
+function pdfToDocument(pdf) {
+  return {
+    id: pdf.path,
+    text: pdf.text,
+    additionalProps: {
+      source: pdf.path,
+      mediaType: "application/pdf"
+    }
+  };
+}
+async function pdfLoaderToDocuments(loader) {
+  const documents = [];
+  for await (const pdf of loader) {
+    documents.push(pdfToDocument(pdf));
+  }
+  return documents;
+}
+function pdfPageToDocument(page) {
+  return {
+    id: `${page.path}#page=${page.pageNumber}`,
+    text: page.text,
+    additionalProps: {
+      source: page.path,
+      mediaType: "application/pdf",
+      pageNumber: String(page.pageNumber)
+    }
+  };
+}
+async function pdfPageLoaderToDocuments(loader) {
+  const documents = [];
+  for await (const page of loader) {
+    documents.push(pdfPageToDocument(page));
+  }
+  return documents;
+}
+
+// src/loaders/file.ts
 import { readdir, readFile } from "fs/promises";
-import { isAbsolute, join } from "path";
+import { join } from "path";
+
+// src/loaders/shared.ts
+import { isAbsolute } from "path";
 import { glob } from "tinyglobby";
+function ok(value) {
+  return { ok: true, value };
+}
+function err(error) {
+  return { ok: false, error };
+}
+async function sortedGlob(pattern) {
+  return (await glob(pattern, { absolute: isAbsolute(pattern), onlyFiles: true })).sort(
+    (left, right) => left.localeCompare(right)
+  );
+}
+function toUint8Array(bytes) {
+  if (bytes instanceof Uint8Array) {
+    return new Uint8Array(
+      bytes.buffer.slice(bytes.byteOffset, bytes.byteOffset + bytes.byteLength)
+    );
+  }
+  return new Uint8Array(bytes);
+}
+
+// src/loaders/file.ts
 var FileLoader = class _FileLoader {
   constructor(source, mode, skipErrors) {
     this.source = source;
@@ -68,6 +146,30 @@ var FileLoader = class _FileLoader {
     }
   }
 };
+async function mapFileResult(item, mode) {
+  if (!item.ok) {
+    return item;
+  }
+  if (mode === "source") {
+    return item;
+  }
+  try {
+    const text = await readFileSource(item.value);
+    return mode === "read" ? ok(text) : ok({ path: item.value.path, text });
+  } catch (error) {
+    return err(error);
+  }
+}
+async function readFileSource(source) {
+  if ("bytes" in source) {
+    return new TextDecoder().decode(source.bytes);
+  }
+  return readFile(source.path, "utf8");
+}
+
+// src/loaders/pdf.ts
+import { readdir as readdir2, readFile as readFile2 } from "fs/promises";
+import { join as join2 } from "path";
 var PdfFileLoader = class _PdfFileLoader {
   constructor(source, mode, skipErrors) {
     this.source = source;
@@ -91,8 +193,8 @@ var PdfFileLoader = class _PdfFileLoader {
   static withDir(directory) {
     return new _PdfFileLoader(
       async function* () {
-        const entries = await readdir(directory, { withFileTypes: true });
-        const files = entries.filter((entry) => entry.isFile()).map((entry) => join(directory, entry.name)).sort((left, right) => left.localeCompare(right));
+        const entries = await readdir2(directory, { withFileTypes: true });
+        const files = entries.filter((entry) => entry.isFile()).map((entry) => join2(directory, entry.name)).sort((left, right) => left.localeCompare(right));
         for (const path of files) {
           yield ok({ path });
         }
@@ -140,83 +242,6 @@ var PdfFileLoader = class _PdfFileLoader {
     }
   }
 };
-function fileToDocument(file) {
-  return {
-    id: file.path,
-    text: file.text,
-    additionalProps: {
-      source: file.path,
-      mediaType: "text/plain"
-    }
-  };
-}
-async function fileLoaderToDocuments(loader) {
-  const documents = [];
-  for await (const file of loader) {
-    documents.push(fileToDocument(file));
-  }
-  return documents;
-}
-function pdfToDocument(pdf) {
-  return {
-    id: pdf.path,
-    text: pdf.text,
-    additionalProps: {
-      source: pdf.path,
-      mediaType: "application/pdf"
-    }
-  };
-}
-async function pdfLoaderToDocuments(loader) {
-  const documents = [];
-  for await (const pdf of loader) {
-    documents.push(pdfToDocument(pdf));
-  }
-  return documents;
-}
-function pdfPageToDocument(page) {
-  return {
-    id: `${page.path}#page=${page.pageNumber}`,
-    text: page.text,
-    additionalProps: {
-      source: page.path,
-      mediaType: "application/pdf",
-      pageNumber: String(page.pageNumber)
-    }
-  };
-}
-async function pdfPageLoaderToDocuments(loader) {
-  const documents = [];
-  for await (const page of loader) {
-    documents.push(pdfPageToDocument(page));
-  }
-  return documents;
-}
-function ok(value) {
-  return { ok: true, value };
-}
-function err(error) {
-  return { ok: false, error };
-}
-async function sortedGlob(pattern) {
-  return (await glob(pattern, { absolute: isAbsolute(pattern), onlyFiles: true })).sort(
-    (left, right) => left.localeCompare(right)
-  );
-}
-async function mapFileResult(item, mode) {
-  if (!item.ok) {
-    return item;
-  }
-  if (mode === "source") {
-    return item;
-  }
-  try {
-    const text = await readFileSource(item.value);
-    return mode === "read" ? ok(text) : ok({ path: item.value.path, text });
-  } catch (error) {
-    return err(error);
-  }
-}
 async function mapPdfResult(item, mode) {
   if (!item.ok) {
     return [item];
@@ -240,15 +265,9 @@ async function mapPdfResult(item, mode) {
     return [err(error)];
   }
 }
-async function readFileSource(source) {
-  if ("bytes" in source) {
-    return new TextDecoder().decode(source.bytes);
-  }
-  return readFile(source.path, "utf8");
-}
 async function readPdfPages(source) {
   const pdfjs = await import("pdfjs-dist/legacy/build/pdf.mjs");
-  const bytes = "bytes" in source ? source.bytes : toUint8Array(await readFile(source.path));
+  const bytes = "bytes" in source ? source.bytes : toUint8Array(await readFile2(source.path));
   const loadingTask = pdfjs.getDocument({ data: bytes });
   const document = await loadingTask.promise;
   const pages = [];
@@ -267,14 +286,6 @@ async function readPdfPages(source) {
   }
   return pages;
 }
-function toUint8Array(bytes) {
-  if (bytes instanceof Uint8Array) {
-    return new Uint8Array(
-      bytes.buffer.slice(bytes.byteOffset, bytes.byteOffset + bytes.byteLength)
-    );
-  }
-  return new Uint8Array(bytes);
-}
 export {
   FileLoader,
   PdfFileLoader,

package/dist/loaders/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/loaders/index.ts"],"sourcesContent":["import { readdir, readFile } from \"node:fs/promises\";\nimport { isAbsolute, join } from \"node:path\";\nimport { glob } from \"tinyglobby\";\nimport type { Document } from \"../completion\";\n\nexport type LoaderResult<T> = { ok: true; value: T } | { ok: false; error: unknown };\n\nexport type FileSource = { path: string } | { path: \"<memory>\"; bytes: Uint8Array };\nexport type FileReadWithPath = { path: string; text: string };\n\nexport type PdfSource = { path: string } | { path: \"<memory>\"; bytes: Uint8Array };\nexport type PdfReadWithPath = { path: string; text: string };\nexport type PdfPage = { pageNumber: number; text: string };\nexport type PdfPageWithPath = { path: string; pageNumber: number; text: string };\n\ntype FileMode = \"source\" | \"read\" | \"readWithPath\";\ntype PdfMode = \"source\" | \"read\" | \"readWithPath\" | \"byPage\" | \"byPageWithPath\";\n\nexport class FileLoader<T = LoaderResult<FileSource>> implements AsyncIterable<T> {\n  private constructor(\n    private readonly source: () => AsyncIterable<LoaderResult<FileSource>>,\n    private readonly mode: FileMode,\n    private readonly skipErrors: boolean,\n  ) {}\n\n  static withGlob(pattern: string): FileLoader<LoaderResult<FileSource>> {\n    return new FileLoader(\n      async function* () {\n        for (const path of await sortedGlob(pattern)) {\n          yield ok({ path });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  static withDir(directory: string): FileLoader<LoaderResult<FileSource>> {\n    return new FileLoader(\n      async function* () {\n        const entries = await readdir(directory, { withFileTypes: true });\n        const files = entries\n          .filter((entry) => entry.isFile())\n          .map((entry) => join(directory, entry.name))\n          .sort((left, right) => left.localeCompare(right));\n        for (const path of files) {\n          yield ok({ path });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  static fromBytes(bytes: Uint8Array | ArrayBuffer): FileLoader<LoaderResult<FileSource>> {\n    return FileLoader.fromBytesMany([bytes]);\n  }\n\n  static fromBytesMany(\n    bytes: Array<Uint8Array | ArrayBuffer>,\n  ): FileLoader<LoaderResult<FileSource>> {\n    return new FileLoader(\n      async function* () {\n        for (const value of bytes) {\n          yield ok({ path: \"<memory>\", bytes: toUint8Array(value) });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  read(): FileLoader<LoaderResult<string>> {\n    return new FileLoader(this.source, \"read\", this.skipErrors);\n  }\n\n  readWithPath(): FileLoader<LoaderResult<FileReadWithPath>> {\n    return new FileLoader(this.source, \"readWithPath\", this.skipErrors);\n  }\n\n  ignoreErrors(): FileLoader<UnwrapLoaderResult<T>> {\n    return new FileLoader(this.source, this.mode, true) as FileLoader<UnwrapLoaderResult<T>>;\n  }\n\n  async *[Symbol.asyncIterator](): AsyncIterator<T> {\n    for await (const item of this.source()) {\n      const result = await mapFileResult(item, this.mode);\n      if (!result.ok && this.skipErrors) {\n        continue;\n      }\n      yield (this.skipErrors && result.ok ? result.value : result) as T;\n    }\n  }\n}\n\nexport class PdfFileLoader<T = LoaderResult<PdfSource>> implements AsyncIterable<T> {\n  private constructor(\n    private readonly source: () => AsyncIterable<LoaderResult<PdfSource>>,\n    private readonly mode: PdfMode,\n    private readonly skipErrors: boolean,\n  ) {}\n\n  static withGlob(pattern: string): PdfFileLoader<LoaderResult<PdfSource>> {\n    return new PdfFileLoader(\n      async function* () {\n        for (const path of await sortedGlob(pattern)) {\n          yield ok({ path });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  static withDir(directory: string): PdfFileLoader<LoaderResult<PdfSource>> {\n    return new PdfFileLoader(\n      async function* () {\n        const entries = await readdir(directory, { withFileTypes: true });\n        const files = entries\n          .filter((entry) => entry.isFile())\n          .map((entry) => join(directory, entry.name))\n          .sort((left, right) => left.localeCompare(right));\n        for (const path of files) {\n          yield ok({ path });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  static fromBytes(bytes: Uint8Array | ArrayBuffer): PdfFileLoader<LoaderResult<PdfSource>> {\n    return PdfFileLoader.fromBytesMany([bytes]);\n  }\n\n  static fromBytesMany(\n    bytes: Array<Uint8Array | ArrayBuffer>,\n  ): PdfFileLoader<LoaderResult<PdfSource>> {\n    return new PdfFileLoader(\n      async function* () {\n        for (const value of bytes) {\n          yield ok({ path: \"<memory>\", bytes: toUint8Array(value) });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  read(): PdfFileLoader<LoaderResult<string>> {\n    return new PdfFileLoader(this.source, \"read\", this.skipErrors);\n  }\n\n  readWithPath(): PdfFileLoader<LoaderResult<PdfReadWithPath>> {\n    return new PdfFileLoader(this.source, \"readWithPath\", this.skipErrors);\n  }\n\n  byPage(\n    this: PdfFileLoader<LoaderResult<PdfReadWithPath>>,\n  ): PdfFileLoader<LoaderResult<PdfPageWithPath>>;\n  byPage(this: PdfFileLoader<PdfReadWithPath>): PdfFileLoader<PdfPageWithPath>;\n  byPage(): PdfFileLoader<LoaderResult<PdfPage>>;\n  byPage(): PdfFileLoader<LoaderResult<PdfPage> | LoaderResult<PdfPageWithPath> | PdfPageWithPath> {\n    const mode = this.mode === \"readWithPath\" ? \"byPageWithPath\" : \"byPage\";\n    return new PdfFileLoader(this.source, mode, this.skipErrors);\n  }\n\n  ignoreErrors(): PdfFileLoader<UnwrapLoaderResult<T>> {\n    return new PdfFileLoader(this.source, this.mode, true) as PdfFileLoader<UnwrapLoaderResult<T>>;\n  }\n\n  async *[Symbol.asyncIterator](): AsyncIterator<T> {\n    for await (const item of this.source()) {\n      const results = await mapPdfResult(item, this.mode);\n      for (const result of results) {\n        if (!result.ok && this.skipErrors) {\n          continue;\n        }\n        yield (this.skipErrors && result.ok ? result.value : result) as T;\n      }\n    }\n  }\n}\n\nexport function fileToDocument(file: FileReadWithPath): Document {\n  return {\n    id: file.path,\n    text: file.text,\n    additionalProps: {\n      source: file.path,\n      mediaType: \"text/plain\",\n    },\n  };\n}\n\nexport async function fileLoaderToDocuments(\n  loader: AsyncIterable<FileReadWithPath>,\n): Promise<Document[]> {\n  const documents: Document[] = [];\n  for await (const file of loader) {\n    documents.push(fileToDocument(file));\n  }\n  return documents;\n}\n\nexport function pdfToDocument(pdf: PdfReadWithPath): Document {\n  return {\n    id: pdf.path,\n    text: pdf.text,\n    additionalProps: {\n      source: pdf.path,\n      mediaType: \"application/pdf\",\n    },\n  };\n}\n\nexport async function pdfLoaderToDocuments(\n  loader: AsyncIterable<PdfReadWithPath>,\n): Promise<Document[]> {\n  const documents: Document[] = [];\n  for await (const pdf of loader) {\n    documents.push(pdfToDocument(pdf));\n  }\n  return documents;\n}\n\nexport function pdfPageToDocument(page: PdfPageWithPath): Document {\n  return {\n    id: `${page.path}#page=${page.pageNumber}`,\n    text: page.text,\n    additionalProps: {\n      source: page.path,\n      mediaType: \"application/pdf\",\n      pageNumber: String(page.pageNumber),\n    },\n  };\n}\n\nexport async function pdfPageLoaderToDocuments(\n  loader: AsyncIterable<PdfPageWithPath>,\n): Promise<Document[]> {\n  const documents: Document[] = [];\n  for await (const page of loader) {\n    documents.push(pdfPageToDocument(page));\n  }\n  return documents;\n}\n\ntype LoaderValue<T> = T extends { ok: true; value: infer Value } ? Value : never;\ntype UnwrapLoaderResult<T> = [LoaderValue<T>] extends [never] ? T : LoaderValue<T>;\n\nfunction ok<T>(value: T): LoaderResult<T> {\n  return { ok: true, value };\n}\n\nfunction err(error: unknown): LoaderResult<never> {\n  return { ok: false, error };\n}\n\nasync function sortedGlob(pattern: string): Promise<string[]> {\n  return (await glob(pattern, { absolute: isAbsolute(pattern), onlyFiles: true })).sort(\n    (left, right) => left.localeCompare(right),\n  );\n}\n\nasync function mapFileResult(\n  item: LoaderResult<FileSource>,\n  mode: FileMode,\n): Promise<LoaderResult<FileSource | string | FileReadWithPath>> {\n  if (!item.ok) {\n    return item;\n  }\n  if (mode === \"source\") {\n    return item;\n  }\n  try {\n    const text = await readFileSource(item.value);\n    return mode === \"read\" ? ok(text) : ok({ path: item.value.path, text });\n  } catch (error) {\n    return err(error);\n  }\n}\n\nasync function mapPdfResult(\n  item: LoaderResult<PdfSource>,\n  mode: PdfMode,\n): Promise<Array<LoaderResult<PdfSource | string | PdfReadWithPath | PdfPage | PdfPageWithPath>>> {\n  if (!item.ok) {\n    return [item];\n  }\n  if (mode === \"source\") {\n    return [item];\n  }\n  try {\n    const pages = await readPdfPages(item.value);\n    if (mode === \"read\") {\n      return [ok(pages.map((page) => page.text).join(\"\"))];\n    }\n    if (mode === \"readWithPath\") {\n      return [ok({ path: item.value.path, text: pages.map((page) => page.text).join(\"\") })];\n    }\n    if (mode === \"byPageWithPath\") {\n      return pages.map((page) => ok({ path: item.value.path, ...page }));\n    }\n    return pages.map(ok);\n  } catch (error) {\n    return [err(error)];\n  }\n}\n\nasync function readFileSource(source: FileSource): Promise<string> {\n  if (\"bytes\" in source) {\n    return new TextDecoder().decode(source.bytes);\n  }\n  return readFile(source.path, \"utf8\");\n}\n\nasync function readPdfPages(source: PdfSource): Promise<PdfPage[]> {\n  const pdfjs = await import(\"pdfjs-dist/legacy/build/pdf.mjs\");\n  const bytes = \"bytes\" in source ? source.bytes : toUint8Array(await readFile(source.path));\n  const loadingTask = pdfjs.getDocument({ data: bytes });\n  const document = await loadingTask.promise;\n  const pages: PdfPage[] = [];\n  try {\n    for (let index = 1; index <= document.numPages; index += 1) {\n      const page = await document.getPage(index);\n      const content = await page.getTextContent();\n      const text = content.items\n        .flatMap((item) =>\n          typeof item === \"object\" && item !== null && \"str\" in item ? [String(item.str)] : [],\n        )\n        .join(\"\\n\");\n      pages.push({ pageNumber: index - 1, text: text.length > 0 ? `${text}\\n` : \"\" });\n    }\n  } finally {\n    await loadingTask.destroy();\n  }\n  return pages;\n}\n\nfunction toUint8Array(bytes: Uint8Array | ArrayBuffer): Uint8Array {\n  if (bytes instanceof Uint8Array) {\n    return new Uint8Array(\n      bytes.buffer.slice(bytes.byteOffset, bytes.byteOffset + bytes.byteLength),\n    );\n  }\n  return new Uint8Array(bytes);\n}\n"],"mappings":";AAAA,SAAS,SAAS,gBAAgB;AAClC,SAAS,YAAY,YAAY;AACjC,SAAS,YAAY;AAgBd,IAAM,aAAN,MAAM,YAAqE;AAAA,EACxE,YACW,QACA,MACA,YACjB;AAHiB;AACA;AACA;AAAA,EAChB;AAAA,EAHgB;AAAA,EACA;AAAA,EACA;AAAA,EAGnB,OAAO,SAAS,SAAuD;AACrE,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,mBAAW,QAAQ,MAAM,WAAW,OAAO,GAAG;AAC5C,gBAAM,GAAG,EAAE,KAAK,CAAC;AAAA,QACnB;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,QAAQ,WAAyD;AACtE,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,cAAM,UAAU,MAAM,QAAQ,WAAW,EAAE,eAAe,KAAK,CAAC;AAChE,cAAM,QAAQ,QACX,OAAO,CAAC,UAAU,MAAM,OAAO,CAAC,EAChC,IAAI,CAAC,UAAU,KAAK,WAAW,MAAM,IAAI,CAAC,EAC1C,KAAK,CAAC,MAAM,UAAU,KAAK,cAAc,KAAK,CAAC;AAClD,mBAAW,QAAQ,OAAO;AACxB,gBAAM,GAAG,EAAE,KAAK,CAAC;AAAA,QACnB;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,UAAU,OAAuE;AACtF,WAAO,YAAW,cAAc,CAAC,KAAK,CAAC;AAAA,EACzC;AAAA,EAEA,OAAO,cACL,OACsC;AACtC,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,mBAAW,SAAS,OAAO;AACzB,gBAAM,GAAG,EAAE,MAAM,YAAY,OAAO,aAAa,KAAK,EAAE,CAAC;AAAA,QAC3D;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAyC;AACvC,WAAO,IAAI,YAAW,KAAK,QAAQ,QAAQ,KAAK,UAAU;AAAA,EAC5D;AAAA,EAEA,eAA2D;AACzD,WAAO,IAAI,YAAW,KAAK,QAAQ,gBAAgB,KAAK,UAAU;AAAA,EACpE;AAAA,EAEA,eAAkD;AAChD,WAAO,IAAI,YAAW,KAAK,QAAQ,KAAK,MAAM,IAAI;AAAA,EACpD;AAAA,EAEA,QAAQ,OAAO,aAAa,IAAsB;AAChD,qBAAiB,QAAQ,KAAK,OAAO,GAAG;AACtC,YAAM,SAAS,MAAM,cAAc,MAAM,KAAK,IAAI;AAClD,UAAI,CAAC,OAAO,MAAM,KAAK,YAAY;AACjC;AAAA,MACF;AACA,YAAO,KAAK,cAAc,OAAO,KAAK,OAAO,QAAQ;AAAA,IACvD;AAAA,EACF;AACF;AAEO,IAAM,gBAAN,MAAM,eAAuE;AAAA,EAC1E,YACW,QACA,MACA,YACjB;AAHiB;AACA;AACA;AAAA,EAChB;AAAA,EAHgB;AAAA,EACA;AAAA,EACA;AAAA,EAGnB,OAAO,SAAS,SAAyD;AACvE,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,mBAAW,QAAQ,MAAM,WAAW,OAAO,GAAG;AAC5C,gBAAM,GAAG,EAAE,KAAK,CAAC;AAAA,QACnB;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,QAAQ,WAA2D;AACxE,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,cAAM,UAAU,MAAM,QAAQ,WAAW,EAAE,eAAe,KAAK,CAAC;AAChE,cAAM,QAAQ,QACX,OAAO,CAAC,UAAU,MAAM,OAAO,CAAC,EAChC,IAAI,CAAC,UAAU,KAAK,WAAW,MAAM,IAAI,CAAC,EAC1C,KAAK,CAAC,MAAM,UAAU,KAAK,cAAc,KAAK,CAAC;AAClD,mBAAW,QAAQ,OAAO;AACxB,gBAAM,GAAG,EAAE,KAAK,CAAC;AAAA,QACnB;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,UAAU,OAAyE;AACxF,WAAO,eAAc,cAAc,CAAC,KAAK,CAAC;AAAA,EAC5C;AAAA,EAEA,OAAO,cACL,OACwC;AACxC,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,mBAAW,SAAS,OAAO;AACzB,gBAAM,GAAG,EAAE,MAAM,YAAY,OAAO,aAAa,KAAK,EAAE,CAAC;AAAA,QAC3D;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAA4C;AAC1C,WAAO,IAAI,eAAc,KAAK,QAAQ,QAAQ,KAAK,UAAU;AAAA,EAC/D;AAAA,EAEA,eAA6D;AAC3D,WAAO,IAAI,eAAc,KAAK,QAAQ,gBAAgB,KAAK,UAAU;AAAA,EACvE;AAAA,EAOA,SAAiG;AAC/F,UAAM,OAAO,KAAK,SAAS,iBAAiB,mBAAmB;AAC/D,WAAO,IAAI,eAAc,KAAK,QAAQ,MAAM,KAAK,UAAU;AAAA,EAC7D;AAAA,EAEA,eAAqD;AACnD,WAAO,IAAI,eAAc,KAAK,QAAQ,KAAK,MAAM,IAAI;AAAA,EACvD;AAAA,EAEA,QAAQ,OAAO,aAAa,IAAsB;AAChD,qBAAiB,QAAQ,KAAK,OAAO,GAAG;AACtC,YAAM,UAAU,MAAM,aAAa,MAAM,KAAK,IAAI;AAClD,iBAAW,UAAU,SAAS;AAC5B,YAAI,CAAC,OAAO,MAAM,KAAK,YAAY;AACjC;AAAA,QACF;AACA,cAAO,KAAK,cAAc,OAAO,KAAK,OAAO,QAAQ;AAAA,MACvD;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,eAAe,MAAkC;AAC/D,SAAO;AAAA,IACL,IAAI,KAAK;AAAA,IACT,MAAM,KAAK;AAAA,IACX,iBAAiB;AAAA,MACf,QAAQ,KAAK;AAAA,MACb,WAAW;AAAA,IACb;AAAA,EACF;AACF;AAEA,eAAsB,sBACpB,QACqB;AACrB,QAAM,YAAwB,CAAC;AAC/B,mBAAiB,QAAQ,QAAQ;AAC/B,cAAU,KAAK,eAAe,IAAI,CAAC;AAAA,EACrC;AACA,SAAO;AACT;AAEO,SAAS,cAAc,KAAgC;AAC5D,SAAO;AAAA,IACL,IAAI,IAAI;AAAA,IACR,MAAM,IAAI;AAAA,IACV,iBAAiB;AAAA,MACf,QAAQ,IAAI;AAAA,MACZ,WAAW;AAAA,IACb;AAAA,EACF;AACF;AAEA,eAAsB,qBACpB,QACqB;AACrB,QAAM,YAAwB,CAAC;AAC/B,mBAAiB,OAAO,QAAQ;AAC9B,cAAU,KAAK,cAAc,GAAG,CAAC;AAAA,EACnC;AACA,SAAO;AACT;AAEO,SAAS,kBAAkB,MAAiC;AACjE,SAAO;AAAA,IACL,IAAI,GAAG,KAAK,IAAI,SAAS,KAAK,UAAU;AAAA,IACxC,MAAM,KAAK;AAAA,IACX,iBAAiB;AAAA,MACf,QAAQ,KAAK;AAAA,MACb,WAAW;AAAA,MACX,YAAY,OAAO,KAAK,UAAU;AAAA,IACpC;AAAA,EACF;AACF;AAEA,eAAsB,yBACpB,QACqB;AACrB,QAAM,YAAwB,CAAC;AAC/B,mBAAiB,QAAQ,QAAQ;AAC/B,cAAU,KAAK,kBAAkB,IAAI,CAAC;AAAA,EACxC;AACA,SAAO;AACT;AAKA,SAAS,GAAM,OAA2B;AACxC,SAAO,EAAE,IAAI,MAAM,MAAM;AAC3B;AAEA,SAAS,IAAI,OAAqC;AAChD,SAAO,EAAE,IAAI,OAAO,MAAM;AAC5B;AAEA,eAAe,WAAW,SAAoC;AAC5D,UAAQ,MAAM,KAAK,SAAS,EAAE,UAAU,WAAW,OAAO,GAAG,WAAW,KAAK,CAAC,GAAG;AAAA,IAC/E,CAAC,MAAM,UAAU,KAAK,cAAc,KAAK;AAAA,EAC3C;AACF;AAEA,eAAe,cACb,MACA,MAC+D;AAC/D,MAAI,CAAC,KAAK,IAAI;AACZ,WAAO;AAAA,EACT;AACA,MAAI,SAAS,UAAU;AACrB,WAAO;AAAA,EACT;AACA,MAAI;AACF,UAAM,OAAO,MAAM,eAAe,KAAK,KAAK;AAC5C,WAAO,SAAS,SAAS,GAAG,IAAI,IAAI,GAAG,EAAE,MAAM,KAAK,MAAM,MAAM,KAAK,CAAC;AAAA,EACxE,SAAS,OAAO;AACd,WAAO,IAAI,KAAK;AAAA,EAClB;AACF;AAEA,eAAe,aACb,MACA,MACgG;AAChG,MAAI,CAAC,KAAK,IAAI;AACZ,WAAO,CAAC,IAAI;AAAA,EACd;AACA,MAAI,SAAS,UAAU;AACrB,WAAO,CAAC,IAAI;AAAA,EACd;AACA,MAAI;AACF,UAAM,QAAQ,MAAM,aAAa,KAAK,KAAK;AAC3C,QAAI,SAAS,QAAQ;AACnB,aAAO,CAAC,GAAG,MAAM,IAAI,CAAC,SAAS,KAAK,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;AAAA,IACrD;AACA,QAAI,SAAS,gBAAgB;AAC3B,aAAO,CAAC,GAAG,EAAE,MAAM,KAAK,MAAM,MAAM,MAAM,MAAM,IAAI,CAAC,SAAS,KAAK,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;AAAA,IACtF;AACA,QAAI,SAAS,kBAAkB;AAC7B,aAAO,MAAM,IAAI,CAAC,SAAS,GAAG,EAAE,MAAM,KAAK,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC;AAAA,IACnE;AACA,WAAO,MAAM,IAAI,EAAE;AAAA,EACrB,SAAS,OAAO;AACd,WAAO,CAAC,IAAI,KAAK,CAAC;AAAA,EACpB;AACF;AAEA,eAAe,eAAe,QAAqC;AACjE,MAAI,WAAW,QAAQ;AACrB,WAAO,IAAI,YAAY,EAAE,OAAO,OAAO,KAAK;AAAA,EAC9C;AACA,SAAO,SAAS,OAAO,MAAM,MAAM;AACrC;AAEA,eAAe,aAAa,QAAuC;AACjE,QAAM,QAAQ,MAAM,OAAO,iCAAiC;AAC5D,QAAM,QAAQ,WAAW,SAAS,OAAO,QAAQ,aAAa,MAAM,SAAS,OAAO,IAAI,CAAC;AACzF,QAAM,cAAc,MAAM,YAAY,EAAE,MAAM,MAAM,CAAC;AACrD,QAAM,WAAW,MAAM,YAAY;AACnC,QAAM,QAAmB,CAAC;AAC1B,MAAI;AACF,aAAS,QAAQ,GAAG,SAAS,SAAS,UAAU,SAAS,GAAG;AAC1D,YAAM,OAAO,MAAM,SAAS,QAAQ,KAAK;AACzC,YAAM,UAAU,MAAM,KAAK,eAAe;AAC1C,YAAM,OAAO,QAAQ,MAClB;AAAA,QAAQ,CAAC,SACR,OAAO,SAAS,YAAY,SAAS,QAAQ,SAAS,OAAO,CAAC,OAAO,KAAK,GAAG,CAAC,IAAI,CAAC;AAAA,MACrF,EACC,KAAK,IAAI;AACZ,YAAM,KAAK,EAAE,YAAY,QAAQ,GAAG,MAAM,KAAK,SAAS,IAAI,GAAG,IAAI;AAAA,IAAO,GAAG,CAAC;AAAA,IAChF;AAAA,EACF,UAAE;AACA,UAAM,YAAY,QAAQ;AAAA,EAC5B;AACA,SAAO;AACT;AAEA,SAAS,aAAa,OAA6C;AACjE,MAAI,iBAAiB,YAAY;AAC/B,WAAO,IAAI;AAAA,MACT,MAAM,OAAO,MAAM,MAAM,YAAY,MAAM,aAAa,MAAM,UAAU;AAAA,IAC1E;AAAA,EACF;AACA,SAAO,IAAI,WAAW,KAAK;AAC7B;","names":[]}
+{"version":3,"sources":["../../src/loaders/documents.ts","../../src/loaders/file.ts","../../src/loaders/shared.ts","../../src/loaders/pdf.ts"],"sourcesContent":["import type { Document } from \"../completion\";\nimport type { FileReadWithPath, PdfPageWithPath, PdfReadWithPath } from \"./types\";\n\nexport function fileToDocument(file: FileReadWithPath): Document {\n  return {\n    id: file.path,\n    text: file.text,\n    additionalProps: {\n      source: file.path,\n      mediaType: \"text/plain\",\n    },\n  };\n}\n\nexport async function fileLoaderToDocuments(\n  loader: AsyncIterable<FileReadWithPath>,\n): Promise<Document[]> {\n  const documents: Document[] = [];\n  for await (const file of loader) {\n    documents.push(fileToDocument(file));\n  }\n  return documents;\n}\n\nexport function pdfToDocument(pdf: PdfReadWithPath): Document {\n  return {\n    id: pdf.path,\n    text: pdf.text,\n    additionalProps: {\n      source: pdf.path,\n      mediaType: \"application/pdf\",\n    },\n  };\n}\n\nexport async function pdfLoaderToDocuments(\n  loader: AsyncIterable<PdfReadWithPath>,\n): Promise<Document[]> {\n  const documents: Document[] = [];\n  for await (const pdf of loader) {\n    documents.push(pdfToDocument(pdf));\n  }\n  return documents;\n}\n\nexport function pdfPageToDocument(page: PdfPageWithPath): Document {\n  return {\n    id: `${page.path}#page=${page.pageNumber}`,\n    text: page.text,\n    additionalProps: {\n      source: page.path,\n      mediaType: \"application/pdf\",\n      pageNumber: String(page.pageNumber),\n    },\n  };\n}\n\nexport async function pdfPageLoaderToDocuments(\n  loader: AsyncIterable<PdfPageWithPath>,\n): Promise<Document[]> {\n  const documents: Document[] = [];\n  for await (const page of loader) {\n    documents.push(pdfPageToDocument(page));\n  }\n  return documents;\n}\n","import { readdir, readFile } from \"node:fs/promises\";\nimport { join } from \"node:path\";\nimport { err, ok, sortedGlob, toUint8Array } from \"./shared\";\nimport type { FileReadWithPath, FileSource, LoaderResult } from \"./types\";\n\ntype FileMode = \"source\" | \"read\" | \"readWithPath\";\ntype LoaderValue<T> = T extends { ok: true; value: infer Value } ? Value : never;\ntype UnwrapLoaderResult<T> = [LoaderValue<T>] extends [never] ? T : LoaderValue<T>;\n\nexport class FileLoader<T = LoaderResult<FileSource>> implements AsyncIterable<T> {\n  private constructor(\n    private readonly source: () => AsyncIterable<LoaderResult<FileSource>>,\n    private readonly mode: FileMode,\n    private readonly skipErrors: boolean,\n  ) {}\n\n  static withGlob(pattern: string): FileLoader<LoaderResult<FileSource>> {\n    return new FileLoader(\n      async function* () {\n        for (const path of await sortedGlob(pattern)) {\n          yield ok({ path });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  static withDir(directory: string): FileLoader<LoaderResult<FileSource>> {\n    return new FileLoader(\n      async function* () {\n        const entries = await readdir(directory, { withFileTypes: true });\n        const files = entries\n          .filter((entry) => entry.isFile())\n          .map((entry) => join(directory, entry.name))\n          .sort((left, right) => left.localeCompare(right));\n        for (const path of files) {\n          yield ok({ path });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  static fromBytes(bytes: Uint8Array | ArrayBuffer): FileLoader<LoaderResult<FileSource>> {\n    return FileLoader.fromBytesMany([bytes]);\n  }\n\n  static fromBytesMany(\n    bytes: Array<Uint8Array | ArrayBuffer>,\n  ): FileLoader<LoaderResult<FileSource>> {\n    return new FileLoader(\n      async function* () {\n        for (const value of bytes) {\n          yield ok({ path: \"<memory>\", bytes: toUint8Array(value) });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  read(): FileLoader<LoaderResult<string>> {\n    return new FileLoader(this.source, \"read\", this.skipErrors);\n  }\n\n  readWithPath(): FileLoader<LoaderResult<FileReadWithPath>> {\n    return new FileLoader(this.source, \"readWithPath\", this.skipErrors);\n  }\n\n  ignoreErrors(): FileLoader<UnwrapLoaderResult<T>> {\n    return new FileLoader(this.source, this.mode, true) as FileLoader<UnwrapLoaderResult<T>>;\n  }\n\n  async *[Symbol.asyncIterator](): AsyncIterator<T> {\n    for await (const item of this.source()) {\n      const result = await mapFileResult(item, this.mode);\n      if (!result.ok && this.skipErrors) {\n        continue;\n      }\n      yield (this.skipErrors && result.ok ? result.value : result) as T;\n    }\n  }\n}\n\nasync function mapFileResult(\n  item: LoaderResult<FileSource>,\n  mode: FileMode,\n): Promise<LoaderResult<FileSource | string | FileReadWithPath>> {\n  if (!item.ok) {\n    return item;\n  }\n  if (mode === \"source\") {\n    return item;\n  }\n  try {\n    const text = await readFileSource(item.value);\n    return mode === \"read\" ? ok(text) : ok({ path: item.value.path, text });\n  } catch (error) {\n    return err(error);\n  }\n}\n\nasync function readFileSource(source: FileSource): Promise<string> {\n  if (\"bytes\" in source) {\n    return new TextDecoder().decode(source.bytes);\n  }\n  return readFile(source.path, \"utf8\");\n}\n","import { isAbsolute } from \"node:path\";\nimport { glob } from \"tinyglobby\";\nimport type { LoaderResult } from \"./types\";\n\nexport function ok<T>(value: T): LoaderResult<T> {\n  return { ok: true, value };\n}\n\nexport function err(error: unknown): LoaderResult<never> {\n  return { ok: false, error };\n}\n\nexport async function sortedGlob(pattern: string): Promise<string[]> {\n  return (await glob(pattern, { absolute: isAbsolute(pattern), onlyFiles: true })).sort(\n    (left, right) => left.localeCompare(right),\n  );\n}\n\nexport function toUint8Array(bytes: Uint8Array | ArrayBuffer): Uint8Array {\n  if (bytes instanceof Uint8Array) {\n    return new Uint8Array(\n      bytes.buffer.slice(bytes.byteOffset, bytes.byteOffset + bytes.byteLength),\n    );\n  }\n  return new Uint8Array(bytes);\n}\n","import { readdir, readFile } from \"node:fs/promises\";\nimport { join } from \"node:path\";\nimport { err, ok, sortedGlob, toUint8Array } from \"./shared\";\nimport type { LoaderResult, PdfPage, PdfPageWithPath, PdfReadWithPath, PdfSource } from \"./types\";\n\ntype PdfMode = \"source\" | \"read\" | \"readWithPath\" | \"byPage\" | \"byPageWithPath\";\ntype LoaderValue<T> = T extends { ok: true; value: infer Value } ? Value : never;\ntype UnwrapLoaderResult<T> = [LoaderValue<T>] extends [never] ? T : LoaderValue<T>;\n\nexport class PdfFileLoader<T = LoaderResult<PdfSource>> implements AsyncIterable<T> {\n  private constructor(\n    private readonly source: () => AsyncIterable<LoaderResult<PdfSource>>,\n    private readonly mode: PdfMode,\n    private readonly skipErrors: boolean,\n  ) {}\n\n  static withGlob(pattern: string): PdfFileLoader<LoaderResult<PdfSource>> {\n    return new PdfFileLoader(\n      async function* () {\n        for (const path of await sortedGlob(pattern)) {\n          yield ok({ path });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  static withDir(directory: string): PdfFileLoader<LoaderResult<PdfSource>> {\n    return new PdfFileLoader(\n      async function* () {\n        const entries = await readdir(directory, { withFileTypes: true });\n        const files = entries\n          .filter((entry) => entry.isFile())\n          .map((entry) => join(directory, entry.name))\n          .sort((left, right) => left.localeCompare(right));\n        for (const path of files) {\n          yield ok({ path });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  static fromBytes(bytes: Uint8Array | ArrayBuffer): PdfFileLoader<LoaderResult<PdfSource>> {\n    return PdfFileLoader.fromBytesMany([bytes]);\n  }\n\n  static fromBytesMany(\n    bytes: Array<Uint8Array | ArrayBuffer>,\n  ): PdfFileLoader<LoaderResult<PdfSource>> {\n    return new PdfFileLoader(\n      async function* () {\n        for (const value of bytes) {\n          yield ok({ path: \"<memory>\", bytes: toUint8Array(value) });\n        }\n      },\n      \"source\",\n      false,\n    );\n  }\n\n  read(): PdfFileLoader<LoaderResult<string>> {\n    return new PdfFileLoader(this.source, \"read\", this.skipErrors);\n  }\n\n  readWithPath(): PdfFileLoader<LoaderResult<PdfReadWithPath>> {\n    return new PdfFileLoader(this.source, \"readWithPath\", this.skipErrors);\n  }\n\n  byPage(\n    this: PdfFileLoader<LoaderResult<PdfReadWithPath>>,\n  ): PdfFileLoader<LoaderResult<PdfPageWithPath>>;\n  byPage(this: PdfFileLoader<PdfReadWithPath>): PdfFileLoader<PdfPageWithPath>;\n  byPage(): PdfFileLoader<LoaderResult<PdfPage>>;\n  byPage(): PdfFileLoader<LoaderResult<PdfPage> | LoaderResult<PdfPageWithPath> | PdfPageWithPath> {\n    const mode = this.mode === \"readWithPath\" ? \"byPageWithPath\" : \"byPage\";\n    return new PdfFileLoader(this.source, mode, this.skipErrors);\n  }\n\n  ignoreErrors(): PdfFileLoader<UnwrapLoaderResult<T>> {\n    return new PdfFileLoader(this.source, this.mode, true) as PdfFileLoader<UnwrapLoaderResult<T>>;\n  }\n\n  async *[Symbol.asyncIterator](): AsyncIterator<T> {\n    for await (const item of this.source()) {\n      const results = await mapPdfResult(item, this.mode);\n      for (const result of results) {\n        if (!result.ok && this.skipErrors) {\n          continue;\n        }\n        yield (this.skipErrors && result.ok ? result.value : result) as T;\n      }\n    }\n  }\n}\n\nasync function mapPdfResult(\n  item: LoaderResult<PdfSource>,\n  mode: PdfMode,\n): Promise<Array<LoaderResult<PdfSource | string | PdfReadWithPath | PdfPage | PdfPageWithPath>>> {\n  if (!item.ok) {\n    return [item];\n  }\n  if (mode === \"source\") {\n    return [item];\n  }\n  try {\n    const pages = await readPdfPages(item.value);\n    if (mode === \"read\") {\n      return [ok(pages.map((page) => page.text).join(\"\"))];\n    }\n    if (mode === \"readWithPath\") {\n      return [ok({ path: item.value.path, text: pages.map((page) => page.text).join(\"\") })];\n    }\n    if (mode === \"byPageWithPath\") {\n      return pages.map((page) => ok({ path: item.value.path, ...page }));\n    }\n    return pages.map(ok);\n  } catch (error) {\n    return [err(error)];\n  }\n}\n\nasync function readPdfPages(source: PdfSource): Promise<PdfPage[]> {\n  const pdfjs = await import(\"pdfjs-dist/legacy/build/pdf.mjs\");\n  const bytes = \"bytes\" in source ? source.bytes : toUint8Array(await readFile(source.path));\n  const loadingTask = pdfjs.getDocument({ data: bytes });\n  const document = await loadingTask.promise;\n  const pages: PdfPage[] = [];\n  try {\n    for (let index = 1; index <= document.numPages; index += 1) {\n      const page = await document.getPage(index);\n      const content = await page.getTextContent();\n      const text = content.items\n        .flatMap((item) =>\n          typeof item === \"object\" && item !== null && \"str\" in item ? [String(item.str)] : [],\n        )\n        .join(\"\\n\");\n      pages.push({ pageNumber: index - 1, text: text.length > 0 ? `${text}\\n` : \"\" });\n    }\n  } finally {\n    await loadingTask.destroy();\n  }\n  return pages;\n}\n"],"mappings":";AAGO,SAAS,eAAe,MAAkC;AAC/D,SAAO;AAAA,IACL,IAAI,KAAK;AAAA,IACT,MAAM,KAAK;AAAA,IACX,iBAAiB;AAAA,MACf,QAAQ,KAAK;AAAA,MACb,WAAW;AAAA,IACb;AAAA,EACF;AACF;AAEA,eAAsB,sBACpB,QACqB;AACrB,QAAM,YAAwB,CAAC;AAC/B,mBAAiB,QAAQ,QAAQ;AAC/B,cAAU,KAAK,eAAe,IAAI,CAAC;AAAA,EACrC;AACA,SAAO;AACT;AAEO,SAAS,cAAc,KAAgC;AAC5D,SAAO;AAAA,IACL,IAAI,IAAI;AAAA,IACR,MAAM,IAAI;AAAA,IACV,iBAAiB;AAAA,MACf,QAAQ,IAAI;AAAA,MACZ,WAAW;AAAA,IACb;AAAA,EACF;AACF;AAEA,eAAsB,qBACpB,QACqB;AACrB,QAAM,YAAwB,CAAC;AAC/B,mBAAiB,OAAO,QAAQ;AAC9B,cAAU,KAAK,cAAc,GAAG,CAAC;AAAA,EACnC;AACA,SAAO;AACT;AAEO,SAAS,kBAAkB,MAAiC;AACjE,SAAO;AAAA,IACL,IAAI,GAAG,KAAK,IAAI,SAAS,KAAK,UAAU;AAAA,IACxC,MAAM,KAAK;AAAA,IACX,iBAAiB;AAAA,MACf,QAAQ,KAAK;AAAA,MACb,WAAW;AAAA,MACX,YAAY,OAAO,KAAK,UAAU;AAAA,IACpC;AAAA,EACF;AACF;AAEA,eAAsB,yBACpB,QACqB;AACrB,QAAM,YAAwB,CAAC;AAC/B,mBAAiB,QAAQ,QAAQ;AAC/B,cAAU,KAAK,kBAAkB,IAAI,CAAC;AAAA,EACxC;AACA,SAAO;AACT;;;ACjEA,SAAS,SAAS,gBAAgB;AAClC,SAAS,YAAY;;;ACDrB,SAAS,kBAAkB;AAC3B,SAAS,YAAY;AAGd,SAAS,GAAM,OAA2B;AAC/C,SAAO,EAAE,IAAI,MAAM,MAAM;AAC3B;AAEO,SAAS,IAAI,OAAqC;AACvD,SAAO,EAAE,IAAI,OAAO,MAAM;AAC5B;AAEA,eAAsB,WAAW,SAAoC;AACnE,UAAQ,MAAM,KAAK,SAAS,EAAE,UAAU,WAAW,OAAO,GAAG,WAAW,KAAK,CAAC,GAAG;AAAA,IAC/E,CAAC,MAAM,UAAU,KAAK,cAAc,KAAK;AAAA,EAC3C;AACF;AAEO,SAAS,aAAa,OAA6C;AACxE,MAAI,iBAAiB,YAAY;AAC/B,WAAO,IAAI;AAAA,MACT,MAAM,OAAO,MAAM,MAAM,YAAY,MAAM,aAAa,MAAM,UAAU;AAAA,IAC1E;AAAA,EACF;AACA,SAAO,IAAI,WAAW,KAAK;AAC7B;;;ADhBO,IAAM,aAAN,MAAM,YAAqE;AAAA,EACxE,YACW,QACA,MACA,YACjB;AAHiB;AACA;AACA;AAAA,EAChB;AAAA,EAHgB;AAAA,EACA;AAAA,EACA;AAAA,EAGnB,OAAO,SAAS,SAAuD;AACrE,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,mBAAW,QAAQ,MAAM,WAAW,OAAO,GAAG;AAC5C,gBAAM,GAAG,EAAE,KAAK,CAAC;AAAA,QACnB;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,QAAQ,WAAyD;AACtE,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,cAAM,UAAU,MAAM,QAAQ,WAAW,EAAE,eAAe,KAAK,CAAC;AAChE,cAAM,QAAQ,QACX,OAAO,CAAC,UAAU,MAAM,OAAO,CAAC,EAChC,IAAI,CAAC,UAAU,KAAK,WAAW,MAAM,IAAI,CAAC,EAC1C,KAAK,CAAC,MAAM,UAAU,KAAK,cAAc,KAAK,CAAC;AAClD,mBAAW,QAAQ,OAAO;AACxB,gBAAM,GAAG,EAAE,KAAK,CAAC;AAAA,QACnB;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,UAAU,OAAuE;AACtF,WAAO,YAAW,cAAc,CAAC,KAAK,CAAC;AAAA,EACzC;AAAA,EAEA,OAAO,cACL,OACsC;AACtC,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,mBAAW,SAAS,OAAO;AACzB,gBAAM,GAAG,EAAE,MAAM,YAAY,OAAO,aAAa,KAAK,EAAE,CAAC;AAAA,QAC3D;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAyC;AACvC,WAAO,IAAI,YAAW,KAAK,QAAQ,QAAQ,KAAK,UAAU;AAAA,EAC5D;AAAA,EAEA,eAA2D;AACzD,WAAO,IAAI,YAAW,KAAK,QAAQ,gBAAgB,KAAK,UAAU;AAAA,EACpE;AAAA,EAEA,eAAkD;AAChD,WAAO,IAAI,YAAW,KAAK,QAAQ,KAAK,MAAM,IAAI;AAAA,EACpD;AAAA,EAEA,QAAQ,OAAO,aAAa,IAAsB;AAChD,qBAAiB,QAAQ,KAAK,OAAO,GAAG;AACtC,YAAM,SAAS,MAAM,cAAc,MAAM,KAAK,IAAI;AAClD,UAAI,CAAC,OAAO,MAAM,KAAK,YAAY;AACjC;AAAA,MACF;AACA,YAAO,KAAK,cAAc,OAAO,KAAK,OAAO,QAAQ;AAAA,IACvD;AAAA,EACF;AACF;AAEA,eAAe,cACb,MACA,MAC+D;AAC/D,MAAI,CAAC,KAAK,IAAI;AACZ,WAAO;AAAA,EACT;AACA,MAAI,SAAS,UAAU;AACrB,WAAO;AAAA,EACT;AACA,MAAI;AACF,UAAM,OAAO,MAAM,eAAe,KAAK,KAAK;AAC5C,WAAO,SAAS,SAAS,GAAG,IAAI,IAAI,GAAG,EAAE,MAAM,KAAK,MAAM,MAAM,KAAK,CAAC;AAAA,EACxE,SAAS,OAAO;AACd,WAAO,IAAI,KAAK;AAAA,EAClB;AACF;AAEA,eAAe,eAAe,QAAqC;AACjE,MAAI,WAAW,QAAQ;AACrB,WAAO,IAAI,YAAY,EAAE,OAAO,OAAO,KAAK;AAAA,EAC9C;AACA,SAAO,SAAS,OAAO,MAAM,MAAM;AACrC;;;AE7GA,SAAS,WAAAA,UAAS,YAAAC,iBAAgB;AAClC,SAAS,QAAAC,aAAY;AAQd,IAAM,gBAAN,MAAM,eAAuE;AAAA,EAC1E,YACW,QACA,MACA,YACjB;AAHiB;AACA;AACA;AAAA,EAChB;AAAA,EAHgB;AAAA,EACA;AAAA,EACA;AAAA,EAGnB,OAAO,SAAS,SAAyD;AACvE,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,mBAAW,QAAQ,MAAM,WAAW,OAAO,GAAG;AAC5C,gBAAM,GAAG,EAAE,KAAK,CAAC;AAAA,QACnB;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,QAAQ,WAA2D;AACxE,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,cAAM,UAAU,MAAMC,SAAQ,WAAW,EAAE,eAAe,KAAK,CAAC;AAChE,cAAM,QAAQ,QACX,OAAO,CAAC,UAAU,MAAM,OAAO,CAAC,EAChC,IAAI,CAAC,UAAUC,MAAK,WAAW,MAAM,IAAI,CAAC,EAC1C,KAAK,CAAC,MAAM,UAAU,KAAK,cAAc,KAAK,CAAC;AAClD,mBAAW,QAAQ,OAAO;AACxB,gBAAM,GAAG,EAAE,KAAK,CAAC;AAAA,QACnB;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,UAAU,OAAyE;AACxF,WAAO,eAAc,cAAc,CAAC,KAAK,CAAC;AAAA,EAC5C;AAAA,EAEA,OAAO,cACL,OACwC;AACxC,WAAO,IAAI;AAAA,MACT,mBAAmB;AACjB,mBAAW,SAAS,OAAO;AACzB,gBAAM,GAAG,EAAE,MAAM,YAAY,OAAO,aAAa,KAAK,EAAE,CAAC;AAAA,QAC3D;AAAA,MACF;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAA4C;AAC1C,WAAO,IAAI,eAAc,KAAK,QAAQ,QAAQ,KAAK,UAAU;AAAA,EAC/D;AAAA,EAEA,eAA6D;AAC3D,WAAO,IAAI,eAAc,KAAK,QAAQ,gBAAgB,KAAK,UAAU;AAAA,EACvE;AAAA,EAOA,SAAiG;AAC/F,UAAM,OAAO,KAAK,SAAS,iBAAiB,mBAAmB;AAC/D,WAAO,IAAI,eAAc,KAAK,QAAQ,MAAM,KAAK,UAAU;AAAA,EAC7D;AAAA,EAEA,eAAqD;AACnD,WAAO,IAAI,eAAc,KAAK,QAAQ,KAAK,MAAM,IAAI;AAAA,EACvD;AAAA,EAEA,QAAQ,OAAO,aAAa,IAAsB;AAChD,qBAAiB,QAAQ,KAAK,OAAO,GAAG;AACtC,YAAM,UAAU,MAAM,aAAa,MAAM,KAAK,IAAI;AAClD,iBAAW,UAAU,SAAS;AAC5B,YAAI,CAAC,OAAO,MAAM,KAAK,YAAY;AACjC;AAAA,QACF;AACA,cAAO,KAAK,cAAc,OAAO,KAAK,OAAO,QAAQ;AAAA,MACvD;AAAA,IACF;AAAA,EACF;AACF;AAEA,eAAe,aACb,MACA,MACgG;AAChG,MAAI,CAAC,KAAK,IAAI;AACZ,WAAO,CAAC,IAAI;AAAA,EACd;AACA,MAAI,SAAS,UAAU;AACrB,WAAO,CAAC,IAAI;AAAA,EACd;AACA,MAAI;AACF,UAAM,QAAQ,MAAM,aAAa,KAAK,KAAK;AAC3C,QAAI,SAAS,QAAQ;AACnB,aAAO,CAAC,GAAG,MAAM,IAAI,CAAC,SAAS,KAAK,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;AAAA,IACrD;AACA,QAAI,SAAS,gBAAgB;AAC3B,aAAO,CAAC,GAAG,EAAE,MAAM,KAAK,MAAM,MAAM,MAAM,MAAM,IAAI,CAAC,SAAS,KAAK,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;AAAA,IACtF;AACA,QAAI,SAAS,kBAAkB;AAC7B,aAAO,MAAM,IAAI,CAAC,SAAS,GAAG,EAAE,MAAM,KAAK,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC;AAAA,IACnE;AACA,WAAO,MAAM,IAAI,EAAE;AAAA,EACrB,SAAS,OAAO;AACd,WAAO,CAAC,IAAI,KAAK,CAAC;AAAA,EACpB;AACF;AAEA,eAAe,aAAa,QAAuC;AACjE,QAAM,QAAQ,MAAM,OAAO,iCAAiC;AAC5D,QAAM,QAAQ,WAAW,SAAS,OAAO,QAAQ,aAAa,MAAMC,UAAS,OAAO,IAAI,CAAC;AACzF,QAAM,cAAc,MAAM,YAAY,EAAE,MAAM,MAAM,CAAC;AACrD,QAAM,WAAW,MAAM,YAAY;AACnC,QAAM,QAAmB,CAAC;AAC1B,MAAI;AACF,aAAS,QAAQ,GAAG,SAAS,SAAS,UAAU,SAAS,GAAG;AAC1D,YAAM,OAAO,MAAM,SAAS,QAAQ,KAAK;AACzC,YAAM,UAAU,MAAM,KAAK,eAAe;AAC1C,YAAM,OAAO,QAAQ,MAClB;AAAA,QAAQ,CAAC,SACR,OAAO,SAAS,YAAY,SAAS,QAAQ,SAAS,OAAO,CAAC,OAAO,KAAK,GAAG,CAAC,IAAI,CAAC;AAAA,MACrF,EACC,KAAK,IAAI;AACZ,YAAM,KAAK,EAAE,YAAY,QAAQ,GAAG,MAAM,KAAK,SAAS,IAAI,GAAG,IAAI;AAAA,IAAO,GAAG,CAAC;AAAA,IAChF;AAAA,EACF,UAAE;AACA,UAAM,YAAY,QAAQ;AAAA,EAC5B;AACA,SAAO;AACT;","names":["readdir","readFile","join","readdir","join","readFile"]}

package/dist/mcp/index.js

@@ -77,7 +77,16 @@ function createMcpTool(definition, client, serverName) {
 // src/mcp/connect.ts
 async function connectMcp(connection) {
   const client = await connection.connect();
-  const { tools } = await client.listTools();
+  let tools;
+  try {
+    ({ tools } = await client.listTools());
+  } catch (error) {
+    try {
+      await client.close();
+    } catch {
+    }
+    throw error;
+  }
   return {
     name: connection.name,
     tools: tools.map((tool) => createMcpTool(tool, client, connection.name)),

package/dist/mcp/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/mcp/result.ts","../../src/mcp/tool.ts","../../src/mcp/connect.ts","../../src/mcp/connections.ts"],"sourcesContent":["import type { JsonValue } from \"../completion/index\";\nimport type { McpToolCallContent, McpToolCallResult } from \"./types\";\n\nexport function createCallToolParams(\n  name: string,\n  args: unknown,\n): { name: string; arguments?: Record<string, unknown> } {\n  if (args === null || args === undefined) {\n    return { name };\n  }\n\n  if (!isPlainRecord(args)) {\n    throw new Error(\"MCP tool arguments must be a JSON object\");\n  }\n\n  return { name, arguments: args };\n}\n\nexport function mapMcpToolResult(result: McpToolCallResult): string {\n  if (\"toolResult\" in result) {\n    return serializeMcpValue(result.toolResult);\n  }\n\n  if (result.isError === true) {\n    throw new Error(mcpErrorMessage(result.content));\n  }\n\n  return result.content.map(mapMcpContent).join(\"\");\n}\n\nfunction mcpErrorMessage(content: McpToolCallContent[]): string {\n  const text = content\n    .map((item) => (item.type === \"text\" ? item.text : undefined))\n    .filter((item): item is string => item !== undefined)\n    .join(\"\\n\");\n\n  return text === \"\" ? \"MCP tool returned an error\" : text;\n}\n\nfunction mapMcpContent(content: McpToolCallContent): string {\n  if (content.type === \"text\") {\n    return content.text;\n  }\n\n  if (content.type === \"image\") {\n    return `data:${content.mimeType};base64,${content.data}`;\n  }\n\n  if (content.type === \"resource\") {\n    const mimeType =\n      content.resource.mimeType === undefined ? \"\" : `data:${content.resource.mimeType};`;\n    if (\"text\" in content.resource) {\n      return `${mimeType}${content.resource.uri}:${content.resource.text}`;\n    }\n\n    return `${mimeType}${content.resource.uri}:${content.resource.blob}`;\n  }\n\n  throw new Error(`Unsupported MCP tool result content: ${serializeMcpValue(content)}`);\n}\n\nfunction serializeMcpValue(value: unknown): string {\n  if (typeof value === \"string\") {\n    return value;\n  }\n\n  const serialized = JSON.stringify(value);\n  return serialized === undefined ? String(value) : serialized;\n}\n\nfunction isPlainRecord(value: unknown): value is Record<string, JsonValue> {\n  return typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n","import type { ToolDefinition } from \"../completion/index\";\nimport type { Tool } from \"../tool/index\";\nimport { createCallToolParams, mapMcpToolResult } from \"./result\";\nimport type { McpClient, McpToolDefinition } from \"./types\";\n\nconst MCP_TOOL_METADATA_KEY = Symbol.for(\"anvia.mcp.tool.metadata\");\n\nexport function createMcpTool(\n  definition: McpToolDefinition,\n  client: McpClient,\n  serverName?: string,\n): Tool {\n  const tool: Tool = {\n    name: definition.name,\n    definition(): ToolDefinition {\n      return {\n        name: definition.name,\n        description: definition.description ?? \"\",\n        parameters: definition.inputSchema,\n      };\n    },\n    async call(args): Promise<string> {\n      const result = await client.callTool(createCallToolParams(definition.name, args));\n      return mapMcpToolResult(result);\n    },\n  };\n  if (serverName !== undefined) {\n    Object.defineProperty(tool, MCP_TOOL_METADATA_KEY, {\n      value: { serverName },\n      enumerable: false,\n    });\n  }\n  return tool;\n}\n","import { createMcpTool } from \"./tool\";\nimport type { McpConnection, McpServer } from \"./types\";\n\nexport async function connectMcp(connection: McpConnection): Promise<McpServer> {\n  const client = await connection.connect();\n  const { tools } = await client.listTools();\n\n  return {\n    name: connection.name,\n    tools: tools.map((tool) => createMcpTool(tool, client, connection.name)),\n    close: () => client.close(),\n  };\n}\n","import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport type {\n  McpClient,\n  McpConnection,\n  McpHttpOptions,\n  McpSseOptions,\n  McpStdioOptions,\n} from \"./types\";\n\nexport const mcp = {\n  stdio(options: McpStdioOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        const { name: _name, ...server } = options;\n        const client = createSdkClient();\n        await client.connect(asSdkTransport(new StdioClientTransport(server)));\n        return client as McpClient;\n      },\n    };\n  },\n\n  http(options: McpHttpOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        const client = createSdkClient();\n        await client.connect(\n          asSdkTransport(\n            new StreamableHTTPClientTransport(new URL(options.url), options.transport),\n          ),\n        );\n        return client as McpClient;\n      },\n    };\n  },\n\n  sse(options: McpSseOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        const client = createSdkClient();\n        await client.connect(\n          asSdkTransport(new SSEClientTransport(new URL(options.url), options.transport)),\n        );\n        return client as McpClient;\n      },\n    };\n  },\n};\n\nfunction createSdkClient(): Client {\n  return new Client({\n    name: \"@anvia/core\",\n    version: \"0.1.0\",\n  });\n}\n\nfunction asSdkTransport(transport: unknown): Parameters<Client[\"connect\"]>[0] {\n  return transport as Parameters<Client[\"connect\"]>[0];\n}\n"],"mappings":";AAGO,SAAS,qBACd,MACA,MACuD;AACvD,MAAI,SAAS,QAAQ,SAAS,QAAW;AACvC,WAAO,EAAE,KAAK;AAAA,EAChB;AAEA,MAAI,CAAC,cAAc,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,0CAA0C;AAAA,EAC5D;AAEA,SAAO,EAAE,MAAM,WAAW,KAAK;AACjC;AAEO,SAAS,iBAAiB,QAAmC;AAClE,MAAI,gBAAgB,QAAQ;AAC1B,WAAO,kBAAkB,OAAO,UAAU;AAAA,EAC5C;AAEA,MAAI,OAAO,YAAY,MAAM;AAC3B,UAAM,IAAI,MAAM,gBAAgB,OAAO,OAAO,CAAC;AAAA,EACjD;AAEA,SAAO,OAAO,QAAQ,IAAI,aAAa,EAAE,KAAK,EAAE;AAClD;AAEA,SAAS,gBAAgB,SAAuC;AAC9D,QAAM,OAAO,QACV,IAAI,CAAC,SAAU,KAAK,SAAS,SAAS,KAAK,OAAO,MAAU,EAC5D,OAAO,CAAC,SAAyB,SAAS,MAAS,EACnD,KAAK,IAAI;AAEZ,SAAO,SAAS,KAAK,+BAA+B;AACtD;AAEA,SAAS,cAAc,SAAqC;AAC1D,MAAI,QAAQ,SAAS,QAAQ;AAC3B,WAAO,QAAQ;AAAA,EACjB;AAEA,MAAI,QAAQ,SAAS,SAAS;AAC5B,WAAO,QAAQ,QAAQ,QAAQ,WAAW,QAAQ,IAAI;AAAA,EACxD;AAEA,MAAI,QAAQ,SAAS,YAAY;AAC/B,UAAM,WACJ,QAAQ,SAAS,aAAa,SAAY,KAAK,QAAQ,QAAQ,SAAS,QAAQ;AAClF,QAAI,UAAU,QAAQ,UAAU;AAC9B,aAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,IACpE;AAEA,WAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,EACpE;AAEA,QAAM,IAAI,MAAM,wCAAwC,kBAAkB,OAAO,CAAC,EAAE;AACtF;AAEA,SAAS,kBAAkB,OAAwB;AACjD,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO;AAAA,EACT;AAEA,QAAM,aAAa,KAAK,UAAU,KAAK;AACvC,SAAO,eAAe,SAAY,OAAO,KAAK,IAAI;AACpD;AAEA,SAAS,cAAc,OAAoD;AACzE,SAAO,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAC5E;;;ACnEA,IAAM,wBAAwB,uBAAO,IAAI,yBAAyB;AAE3D,SAAS,cACd,YACA,QACA,YACM;AACN,QAAM,OAAa;AAAA,IACjB,MAAM,WAAW;AAAA,IACjB,aAA6B;AAC3B,aAAO;AAAA,QACL,MAAM,WAAW;AAAA,QACjB,aAAa,WAAW,eAAe;AAAA,QACvC,YAAY,WAAW;AAAA,MACzB;AAAA,IACF;AAAA,IACA,MAAM,KAAK,MAAuB;AAChC,YAAM,SAAS,MAAM,OAAO,SAAS,qBAAqB,WAAW,MAAM,IAAI,CAAC;AAChF,aAAO,iBAAiB,MAAM;AAAA,IAChC;AAAA,EACF;AACA,MAAI,eAAe,QAAW;AAC5B,WAAO,eAAe,MAAM,uBAAuB;AAAA,MACjD,OAAO,EAAE,WAAW;AAAA,MACpB,YAAY;AAAA,IACd,CAAC;AAAA,EACH;AACA,SAAO;AACT;;;AC9BA,eAAsB,WAAW,YAA+C;AAC9E,QAAM,SAAS,MAAM,WAAW,QAAQ;AACxC,QAAM,EAAE,MAAM,IAAI,MAAM,OAAO,UAAU;AAEzC,SAAO;AAAA,IACL,MAAM,WAAW;AAAA,IACjB,OAAO,MAAM,IAAI,CAAC,SAAS,cAAc,MAAM,QAAQ,WAAW,IAAI,CAAC;AAAA,IACvE,OAAO,MAAM,OAAO,MAAM;AAAA,EAC5B;AACF;;;ACZA,SAAS,cAAc;AACvB,SAAS,0BAA0B;AACnC,SAAS,4BAA4B;AACrC,SAAS,qCAAqC;AASvC,IAAM,MAAM;AAAA,EACjB,MAAM,SAAyC;AAC7C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,EAAE,MAAM,OAAO,GAAG,OAAO,IAAI;AACnC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO,QAAQ,eAAe,IAAI,qBAAqB,MAAM,CAAC,CAAC;AACrE,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,KAAK,SAAwC;AAC3C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX;AAAA,YACE,IAAI,8BAA8B,IAAI,IAAI,QAAQ,GAAG,GAAG,QAAQ,SAAS;AAAA,UAC3E;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,IAAI,SAAuC;AACzC,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX,eAAe,IAAI,mBAAmB,IAAI,IAAI,QAAQ,GAAG,GAAG,QAAQ,SAAS,CAAC;AAAA,QAChF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,kBAA0B;AACjC,SAAO,IAAI,OAAO;AAAA,IAChB,MAAM;AAAA,IACN,SAAS;AAAA,EACX,CAAC;AACH;AAEA,SAAS,eAAe,WAAsD;AAC5E,SAAO;AACT;","names":[]}
+{"version":3,"sources":["../../src/mcp/result.ts","../../src/mcp/tool.ts","../../src/mcp/connect.ts","../../src/mcp/connections.ts"],"sourcesContent":["import type { JsonValue } from \"../completion/index\";\nimport type { McpToolCallContent, McpToolCallResult } from \"./types\";\n\nexport function createCallToolParams(\n  name: string,\n  args: unknown,\n): { name: string; arguments?: Record<string, unknown> } {\n  if (args === null || args === undefined) {\n    return { name };\n  }\n\n  if (!isPlainRecord(args)) {\n    throw new Error(\"MCP tool arguments must be a JSON object\");\n  }\n\n  return { name, arguments: args };\n}\n\nexport function mapMcpToolResult(result: McpToolCallResult): string {\n  if (\"toolResult\" in result) {\n    return serializeMcpValue(result.toolResult);\n  }\n\n  if (result.isError === true) {\n    throw new Error(mcpErrorMessage(result.content));\n  }\n\n  return result.content.map(mapMcpContent).join(\"\");\n}\n\nfunction mcpErrorMessage(content: McpToolCallContent[]): string {\n  const text = content\n    .map((item) => (item.type === \"text\" ? item.text : undefined))\n    .filter((item): item is string => item !== undefined)\n    .join(\"\\n\");\n\n  return text === \"\" ? \"MCP tool returned an error\" : text;\n}\n\nfunction mapMcpContent(content: McpToolCallContent): string {\n  if (content.type === \"text\") {\n    return content.text;\n  }\n\n  if (content.type === \"image\") {\n    return `data:${content.mimeType};base64,${content.data}`;\n  }\n\n  if (content.type === \"resource\") {\n    const mimeType =\n      content.resource.mimeType === undefined ? \"\" : `data:${content.resource.mimeType};`;\n    if (\"text\" in content.resource) {\n      return `${mimeType}${content.resource.uri}:${content.resource.text}`;\n    }\n\n    return `${mimeType}${content.resource.uri}:${content.resource.blob}`;\n  }\n\n  throw new Error(`Unsupported MCP tool result content: ${serializeMcpValue(content)}`);\n}\n\nfunction serializeMcpValue(value: unknown): string {\n  if (typeof value === \"string\") {\n    return value;\n  }\n\n  const serialized = JSON.stringify(value);\n  return serialized === undefined ? String(value) : serialized;\n}\n\nfunction isPlainRecord(value: unknown): value is Record<string, JsonValue> {\n  return typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n","import type { ToolDefinition } from \"../completion/index\";\nimport type { Tool } from \"../tool/index\";\nimport { createCallToolParams, mapMcpToolResult } from \"./result\";\nimport type { McpClient, McpToolDefinition } from \"./types\";\n\nconst MCP_TOOL_METADATA_KEY = Symbol.for(\"anvia.mcp.tool.metadata\");\n\nexport function createMcpTool(\n  definition: McpToolDefinition,\n  client: McpClient,\n  serverName?: string,\n): Tool {\n  const tool: Tool = {\n    name: definition.name,\n    definition(): ToolDefinition {\n      return {\n        name: definition.name,\n        description: definition.description ?? \"\",\n        parameters: definition.inputSchema,\n      };\n    },\n    async call(args): Promise<string> {\n      const result = await client.callTool(createCallToolParams(definition.name, args));\n      return mapMcpToolResult(result);\n    },\n  };\n  if (serverName !== undefined) {\n    Object.defineProperty(tool, MCP_TOOL_METADATA_KEY, {\n      value: { serverName },\n      enumerable: false,\n    });\n  }\n  return tool;\n}\n","import { createMcpTool } from \"./tool\";\nimport type { McpConnection, McpServer } from \"./types\";\n\nexport async function connectMcp(connection: McpConnection): Promise<McpServer> {\n  const client = await connection.connect();\n  let tools: Awaited<ReturnType<typeof client.listTools>>[\"tools\"];\n  try {\n    ({ tools } = await client.listTools());\n  } catch (error) {\n    try {\n      await client.close();\n    } catch {\n      // Preserve the initialization failure so callers see the actionable MCP error.\n    }\n    throw error;\n  }\n\n  return {\n    name: connection.name,\n    tools: tools.map((tool) => createMcpTool(tool, client, connection.name)),\n    close: () => client.close(),\n  };\n}\n","import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport type {\n  McpClient,\n  McpConnection,\n  McpHttpOptions,\n  McpSseOptions,\n  McpStdioOptions,\n} from \"./types\";\n\nexport const mcp = {\n  stdio(options: McpStdioOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        const { name: _name, ...server } = options;\n        const client = createSdkClient();\n        await client.connect(asSdkTransport(new StdioClientTransport(server)));\n        return client as McpClient;\n      },\n    };\n  },\n\n  http(options: McpHttpOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        const client = createSdkClient();\n        await client.connect(\n          asSdkTransport(\n            new StreamableHTTPClientTransport(new URL(options.url), options.transport),\n          ),\n        );\n        return client as McpClient;\n      },\n    };\n  },\n\n  sse(options: McpSseOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        const client = createSdkClient();\n        await client.connect(\n          asSdkTransport(new SSEClientTransport(new URL(options.url), options.transport)),\n        );\n        return client as McpClient;\n      },\n    };\n  },\n};\n\nfunction createSdkClient(): Client {\n  return new Client({\n    name: \"@anvia/core\",\n    version: \"0.1.0\",\n  });\n}\n\nfunction asSdkTransport(transport: unknown): Parameters<Client[\"connect\"]>[0] {\n  return transport as Parameters<Client[\"connect\"]>[0];\n}\n"],"mappings":";AAGO,SAAS,qBACd,MACA,MACuD;AACvD,MAAI,SAAS,QAAQ,SAAS,QAAW;AACvC,WAAO,EAAE,KAAK;AAAA,EAChB;AAEA,MAAI,CAAC,cAAc,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,0CAA0C;AAAA,EAC5D;AAEA,SAAO,EAAE,MAAM,WAAW,KAAK;AACjC;AAEO,SAAS,iBAAiB,QAAmC;AAClE,MAAI,gBAAgB,QAAQ;AAC1B,WAAO,kBAAkB,OAAO,UAAU;AAAA,EAC5C;AAEA,MAAI,OAAO,YAAY,MAAM;AAC3B,UAAM,IAAI,MAAM,gBAAgB,OAAO,OAAO,CAAC;AAAA,EACjD;AAEA,SAAO,OAAO,QAAQ,IAAI,aAAa,EAAE,KAAK,EAAE;AAClD;AAEA,SAAS,gBAAgB,SAAuC;AAC9D,QAAM,OAAO,QACV,IAAI,CAAC,SAAU,KAAK,SAAS,SAAS,KAAK,OAAO,MAAU,EAC5D,OAAO,CAAC,SAAyB,SAAS,MAAS,EACnD,KAAK,IAAI;AAEZ,SAAO,SAAS,KAAK,+BAA+B;AACtD;AAEA,SAAS,cAAc,SAAqC;AAC1D,MAAI,QAAQ,SAAS,QAAQ;AAC3B,WAAO,QAAQ;AAAA,EACjB;AAEA,MAAI,QAAQ,SAAS,SAAS;AAC5B,WAAO,QAAQ,QAAQ,QAAQ,WAAW,QAAQ,IAAI;AAAA,EACxD;AAEA,MAAI,QAAQ,SAAS,YAAY;AAC/B,UAAM,WACJ,QAAQ,SAAS,aAAa,SAAY,KAAK,QAAQ,QAAQ,SAAS,QAAQ;AAClF,QAAI,UAAU,QAAQ,UAAU;AAC9B,aAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,IACpE;AAEA,WAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,EACpE;AAEA,QAAM,IAAI,MAAM,wCAAwC,kBAAkB,OAAO,CAAC,EAAE;AACtF;AAEA,SAAS,kBAAkB,OAAwB;AACjD,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO;AAAA,EACT;AAEA,QAAM,aAAa,KAAK,UAAU,KAAK;AACvC,SAAO,eAAe,SAAY,OAAO,KAAK,IAAI;AACpD;AAEA,SAAS,cAAc,OAAoD;AACzE,SAAO,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAC5E;;;ACnEA,IAAM,wBAAwB,uBAAO,IAAI,yBAAyB;AAE3D,SAAS,cACd,YACA,QACA,YACM;AACN,QAAM,OAAa;AAAA,IACjB,MAAM,WAAW;AAAA,IACjB,aAA6B;AAC3B,aAAO;AAAA,QACL,MAAM,WAAW;AAAA,QACjB,aAAa,WAAW,eAAe;AAAA,QACvC,YAAY,WAAW;AAAA,MACzB;AAAA,IACF;AAAA,IACA,MAAM,KAAK,MAAuB;AAChC,YAAM,SAAS,MAAM,OAAO,SAAS,qBAAqB,WAAW,MAAM,IAAI,CAAC;AAChF,aAAO,iBAAiB,MAAM;AAAA,IAChC;AAAA,EACF;AACA,MAAI,eAAe,QAAW;AAC5B,WAAO,eAAe,MAAM,uBAAuB;AAAA,MACjD,OAAO,EAAE,WAAW;AAAA,MACpB,YAAY;AAAA,IACd,CAAC;AAAA,EACH;AACA,SAAO;AACT;;;AC9BA,eAAsB,WAAW,YAA+C;AAC9E,QAAM,SAAS,MAAM,WAAW,QAAQ;AACxC,MAAI;AACJ,MAAI;AACF,KAAC,EAAE,MAAM,IAAI,MAAM,OAAO,UAAU;AAAA,EACtC,SAAS,OAAO;AACd,QAAI;AACF,YAAM,OAAO,MAAM;AAAA,IACrB,QAAQ;AAAA,IAER;AACA,UAAM;AAAA,EACR;AAEA,SAAO;AAAA,IACL,MAAM,WAAW;AAAA,IACjB,OAAO,MAAM,IAAI,CAAC,SAAS,cAAc,MAAM,QAAQ,WAAW,IAAI,CAAC;AAAA,IACvE,OAAO,MAAM,OAAO,MAAM;AAAA,EAC5B;AACF;;;ACtBA,SAAS,cAAc;AACvB,SAAS,0BAA0B;AACnC,SAAS,4BAA4B;AACrC,SAAS,qCAAqC;AASvC,IAAM,MAAM;AAAA,EACjB,MAAM,SAAyC;AAC7C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,EAAE,MAAM,OAAO,GAAG,OAAO,IAAI;AACnC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO,QAAQ,eAAe,IAAI,qBAAqB,MAAM,CAAC,CAAC;AACrE,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,KAAK,SAAwC;AAC3C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX;AAAA,YACE,IAAI,8BAA8B,IAAI,IAAI,QAAQ,GAAG,GAAG,QAAQ,SAAS;AAAA,UAC3E;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,IAAI,SAAuC;AACzC,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX,eAAe,IAAI,mBAAmB,IAAI,IAAI,QAAQ,GAAG,GAAG,QAAQ,SAAS,CAAC;AAAA,QAChF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,kBAA0B;AACjC,SAAO,IAAI,OAAO;AAAA,IAChB,MAAM;AAAA,IACN,SAAS;AAAA,EACX,CAAC;AACH;AAEA,SAAS,eAAe,WAAsD;AAC5E,SAAO;AACT;","names":[]}