@anvia/core 0.11.1 → 0.11.3

package/dist/{chunk-W4DZU6IW.js → chunk-JS5MGVNM.js}

@@ -2,7 +2,51 @@ import {
   mapWithConcurrency
 } from "./chunk-OIMLU4SF.js";
 
-// src/embeddings/index.ts
+// src/embeddings/distance.ts
+function dotProduct(left, right) {
+  assertSameDimensions(left, right);
+  return left.reduce((sum, value, index) => sum + value * right[index], 0);
+}
+function cosineSimilarity(left, right) {
+  assertSameDimensions(left, right);
+  const leftMagnitude = magnitude(left);
+  const rightMagnitude = magnitude(right);
+  if (leftMagnitude === 0 || rightMagnitude === 0) {
+    return 0;
+  }
+  return dotProduct(left, right) / (leftMagnitude * rightMagnitude);
+}
+function angularDistance(left, right) {
+  const similarity = Math.max(-1, Math.min(1, cosineSimilarity(left, right)));
+  return Math.acos(similarity) / Math.PI;
+}
+function euclideanDistance(left, right) {
+  assertSameDimensions(left, right);
+  return Math.sqrt(
+    left.reduce((sum, value, index) => sum + (value - right[index]) ** 2, 0)
+  );
+}
+function manhattanDistance(left, right) {
+  assertSameDimensions(left, right);
+  return left.reduce((sum, value, index) => sum + Math.abs(value - right[index]), 0);
+}
+function chebyshevDistance(left, right) {
+  assertSameDimensions(left, right);
+  return left.reduce(
+    (max, value, index) => Math.max(max, Math.abs(value - right[index])),
+    0
+  );
+}
+function magnitude(vector) {
+  return Math.sqrt(vector.reduce((sum, value) => sum + value ** 2, 0));
+}
+function assertSameDimensions(left, right) {
+  if (left.length !== right.length) {
+    throw new Error(`Vector dimension mismatch: ${left.length} !== ${right.length}`);
+  }
+}
+
+// src/embeddings/embed.ts
 async function embedText(model, text) {
   const embeddings = await embedTexts(model, [text]);
   const embedding = embeddings[0];
@@ -72,48 +116,6 @@ async function embedDocuments(model, documents, options) {
     embeddings: byDocument.get(index) ?? []
   }));
 }
-function dotProduct(left, right) {
-  assertSameDimensions(left, right);
-  return left.reduce((sum, value, index) => sum + value * right[index], 0);
-}
-function cosineSimilarity(left, right) {
-  assertSameDimensions(left, right);
-  const leftMagnitude = magnitude(left);
-  const rightMagnitude = magnitude(right);
-  if (leftMagnitude === 0 || rightMagnitude === 0) {
-    return 0;
-  }
-  return dotProduct(left, right) / (leftMagnitude * rightMagnitude);
-}
-function angularDistance(left, right) {
-  const similarity = Math.max(-1, Math.min(1, cosineSimilarity(left, right)));
-  return Math.acos(similarity) / Math.PI;
-}
-function euclideanDistance(left, right) {
-  assertSameDimensions(left, right);
-  return Math.sqrt(
-    left.reduce((sum, value, index) => sum + (value - right[index]) ** 2, 0)
-  );
-}
-function manhattanDistance(left, right) {
-  assertSameDimensions(left, right);
-  return left.reduce((sum, value, index) => sum + Math.abs(value - right[index]), 0);
-}
-function chebyshevDistance(left, right) {
-  assertSameDimensions(left, right);
-  return left.reduce(
-    (max, value, index) => Math.max(max, Math.abs(value - right[index])),
-    0
-  );
-}
-function magnitude(vector) {
-  return Math.sqrt(vector.reduce((sum, value) => sum + value ** 2, 0));
-}
-function assertSameDimensions(left, right) {
-  if (left.length !== right.length) {
-    throw new Error(`Vector dimension mismatch: ${left.length} !== ${right.length}`);
-  }
-}
 function chunk(items, size) {
   const chunks = [];
   for (let index = 0; index < items.length; index += size) {
@@ -123,14 +125,14 @@ function chunk(items, size) {
 }
 
 export {
-  embedText,
-  embedTexts,
-  embedDocuments,
   dotProduct,
   cosineSimilarity,
   angularDistance,
   euclideanDistance,
   manhattanDistance,
-  chebyshevDistance
+  chebyshevDistance,
+  embedText,
+  embedTexts,
+  embedDocuments
 };
-//# sourceMappingURL=chunk-W4DZU6IW.js.map
+//# sourceMappingURL=chunk-JS5MGVNM.js.map

package/dist/chunk-JS5MGVNM.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/embeddings/distance.ts","../src/embeddings/embed.ts"],"sourcesContent":["export function dotProduct(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  return left.reduce((sum, value, index) => sum + value * (right[index] as number), 0);\n}\n\nexport function cosineSimilarity(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  const leftMagnitude = magnitude(left);\n  const rightMagnitude = magnitude(right);\n  if (leftMagnitude === 0 || rightMagnitude === 0) {\n    return 0;\n  }\n  return dotProduct(left, right) / (leftMagnitude * rightMagnitude);\n}\n\nexport function angularDistance(left: number[], right: number[]): number {\n  const similarity = Math.max(-1, Math.min(1, cosineSimilarity(left, right)));\n  return Math.acos(similarity) / Math.PI;\n}\n\nexport function euclideanDistance(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  return Math.sqrt(\n    left.reduce((sum, value, index) => sum + (value - (right[index] as number)) ** 2, 0),\n  );\n}\n\nexport function manhattanDistance(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  return left.reduce((sum, value, index) => sum + Math.abs(value - (right[index] as number)), 0);\n}\n\nexport function chebyshevDistance(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  return left.reduce(\n    (max, value, index) => Math.max(max, Math.abs(value - (right[index] as number))),\n    0,\n  );\n}\n\nfunction magnitude(vector: number[]): number {\n  return Math.sqrt(vector.reduce((sum, value) => sum + value ** 2, 0));\n}\n\nfunction assertSameDimensions(left: number[], right: number[]): void {\n  if (left.length !== right.length) {\n    throw new Error(`Vector dimension mismatch: ${left.length} !== ${right.length}`);\n  }\n}\n","import { mapWithConcurrency } from \"../internal/concurrency\";\nimport type {\n  EmbedDocumentsOptions,\n  EmbeddedDocument,\n  Embedding,\n  EmbeddingModel,\n  VectorMetadata,\n} from \"./types\";\n\nexport async function embedText(model: EmbeddingModel, text: string): Promise<Embedding> {\n  const embeddings = await embedTexts(model, [text]);\n  const embedding = embeddings[0];\n  if (embedding === undefined) {\n    throw new Error(\"Embedding model returned no embeddings\");\n  }\n  return embedding;\n}\n\nexport async function embedTexts(model: EmbeddingModel, texts: string[]): Promise<Embedding[]> {\n  if (texts.length === 0) {\n    return [];\n  }\n\n  const maxBatchSize = Math.max(1, Math.trunc(model.maxBatchSize ?? texts.length));\n  const batches: string[][] = [];\n  for (let index = 0; index < texts.length; index += maxBatchSize) {\n    batches.push(texts.slice(index, index + maxBatchSize));\n  }\n\n  const results = await mapWithConcurrency(batches, 1, (batch) => model.embedTexts(batch));\n  const embeddings = results.flat();\n  if (embeddings.length !== texts.length) {\n    throw new Error(\n      `Embedding model returned ${embeddings.length} embeddings for ${texts.length} texts`,\n    );\n  }\n  return embeddings;\n}\n\nexport async function embedDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n  model: EmbeddingModel,\n  documents: T[],\n  options: EmbedDocumentsOptions<T, Metadata>,\n): Promise<Array<EmbeddedDocument<T, Metadata>>> {\n  const prepared = documents.map((document, index) => {\n    const content = options.content(document, index);\n    const texts = Array.isArray(content) ? content : [content];\n    return {\n      id: options.id?.(document, index) ?? `doc${index}`,\n      document,\n      metadata: options.metadata?.(document, index),\n      texts,\n    };\n  });\n\n  const flatTexts = prepared.flatMap((item, documentIndex) =>\n    item.texts.map((text) => ({ documentIndex, text })),\n  );\n  const embeddings = await mapWithConcurrency(\n    chunk(flatTexts, Math.max(1, Math.trunc(model.maxBatchSize ?? (flatTexts.length || 1)))),\n    Math.max(1, Math.trunc(options.concurrency ?? 1)),\n    async (batch) => {\n      const batchEmbeddings = await model.embedTexts(batch.map((item) => item.text));\n      if (batchEmbeddings.length !== batch.length) {\n        throw new Error(\n          `Embedding model returned ${batchEmbeddings.length} embeddings for ${batch.length} texts`,\n        );\n      }\n      return batch.map((item, index) => ({\n        documentIndex: item.documentIndex,\n        embedding: batchEmbeddings[index] as Embedding,\n      }));\n    },\n  );\n\n  const byDocument = new Map<number, Embedding[]>();\n  for (const item of embeddings.flat()) {\n    const list = byDocument.get(item.documentIndex) ?? [];\n    list.push(item.embedding);\n    byDocument.set(item.documentIndex, list);\n  }\n\n  return prepared.map((item, index) => ({\n    id: item.id,\n    document: item.document,\n    ...(item.metadata !== undefined && { metadata: item.metadata }),\n    embeddings: byDocument.get(index) ?? [],\n  }));\n}\n\nfunction chunk<T>(items: T[], size: number): T[][] {\n  const chunks: T[][] = [];\n  for (let index = 0; index < items.length; index += size) {\n    chunks.push(items.slice(index, index + size));\n  }\n  return chunks;\n}\n"],"mappings":";;;;;AAAO,SAAS,WAAW,MAAgB,OAAyB;AAClE,uBAAqB,MAAM,KAAK;AAChC,SAAO,KAAK,OAAO,CAAC,KAAK,OAAO,UAAU,MAAM,QAAS,MAAM,KAAK,GAAc,CAAC;AACrF;AAEO,SAAS,iBAAiB,MAAgB,OAAyB;AACxE,uBAAqB,MAAM,KAAK;AAChC,QAAM,gBAAgB,UAAU,IAAI;AACpC,QAAM,iBAAiB,UAAU,KAAK;AACtC,MAAI,kBAAkB,KAAK,mBAAmB,GAAG;AAC/C,WAAO;AAAA,EACT;AACA,SAAO,WAAW,MAAM,KAAK,KAAK,gBAAgB;AACpD;AAEO,SAAS,gBAAgB,MAAgB,OAAyB;AACvE,QAAM,aAAa,KAAK,IAAI,IAAI,KAAK,IAAI,GAAG,iBAAiB,MAAM,KAAK,CAAC,CAAC;AAC1E,SAAO,KAAK,KAAK,UAAU,IAAI,KAAK;AACtC;AAEO,SAAS,kBAAkB,MAAgB,OAAyB;AACzE,uBAAqB,MAAM,KAAK;AAChC,SAAO,KAAK;AAAA,IACV,KAAK,OAAO,CAAC,KAAK,OAAO,UAAU,OAAO,QAAS,MAAM,KAAK,MAAiB,GAAG,CAAC;AAAA,EACrF;AACF;AAEO,SAAS,kBAAkB,MAAgB,OAAyB;AACzE,uBAAqB,MAAM,KAAK;AAChC,SAAO,KAAK,OAAO,CAAC,KAAK,OAAO,UAAU,MAAM,KAAK,IAAI,QAAS,MAAM,KAAK,CAAY,GAAG,CAAC;AAC/F;AAEO,SAAS,kBAAkB,MAAgB,OAAyB;AACzE,uBAAqB,MAAM,KAAK;AAChC,SAAO,KAAK;AAAA,IACV,CAAC,KAAK,OAAO,UAAU,KAAK,IAAI,KAAK,KAAK,IAAI,QAAS,MAAM,KAAK,CAAY,CAAC;AAAA,IAC/E;AAAA,EACF;AACF;AAEA,SAAS,UAAU,QAA0B;AAC3C,SAAO,KAAK,KAAK,OAAO,OAAO,CAAC,KAAK,UAAU,MAAM,SAAS,GAAG,CAAC,CAAC;AACrE;AAEA,SAAS,qBAAqB,MAAgB,OAAuB;AACnE,MAAI,KAAK,WAAW,MAAM,QAAQ;AAChC,UAAM,IAAI,MAAM,8BAA8B,KAAK,MAAM,QAAQ,MAAM,MAAM,EAAE;AAAA,EACjF;AACF;;;ACvCA,eAAsB,UAAU,OAAuB,MAAkC;AACvF,QAAM,aAAa,MAAM,WAAW,OAAO,CAAC,IAAI,CAAC;AACjD,QAAM,YAAY,WAAW,CAAC;AAC9B,MAAI,cAAc,QAAW;AAC3B,UAAM,IAAI,MAAM,wCAAwC;AAAA,EAC1D;AACA,SAAO;AACT;AAEA,eAAsB,WAAW,OAAuB,OAAuC;AAC7F,MAAI,MAAM,WAAW,GAAG;AACtB,WAAO,CAAC;AAAA,EACV;AAEA,QAAM,eAAe,KAAK,IAAI,GAAG,KAAK,MAAM,MAAM,gBAAgB,MAAM,MAAM,CAAC;AAC/E,QAAM,UAAsB,CAAC;AAC7B,WAAS,QAAQ,GAAG,QAAQ,MAAM,QAAQ,SAAS,cAAc;AAC/D,YAAQ,KAAK,MAAM,MAAM,OAAO,QAAQ,YAAY,CAAC;AAAA,EACvD;AAEA,QAAM,UAAU,MAAM,mBAAmB,SAAS,GAAG,CAAC,UAAU,MAAM,WAAW,KAAK,CAAC;AACvF,QAAM,aAAa,QAAQ,KAAK;AAChC,MAAI,WAAW,WAAW,MAAM,QAAQ;AACtC,UAAM,IAAI;AAAA,MACR,4BAA4B,WAAW,MAAM,mBAAmB,MAAM,MAAM;AAAA,IAC9E;AAAA,EACF;AACA,SAAO;AACT;AAEA,eAAsB,eACpB,OACA,WACA,SAC+C;AAC/C,QAAM,WAAW,UAAU,IAAI,CAAC,UAAU,UAAU;AAClD,UAAM,UAAU,QAAQ,QAAQ,UAAU,KAAK;AAC/C,UAAM,QAAQ,MAAM,QAAQ,OAAO,IAAI,UAAU,CAAC,OAAO;AACzD,WAAO;AAAA,MACL,IAAI,QAAQ,KAAK,UAAU,KAAK,KAAK,MAAM,KAAK;AAAA,MAChD;AAAA,MACA,UAAU,QAAQ,WAAW,UAAU,KAAK;AAAA,MAC5C;AAAA,IACF;AAAA,EACF,CAAC;AAED,QAAM,YAAY,SAAS;AAAA,IAAQ,CAAC,MAAM,kBACxC,KAAK,MAAM,IAAI,CAAC,UAAU,EAAE,eAAe,KAAK,EAAE;AAAA,EACpD;AACA,QAAM,aAAa,MAAM;AAAA,IACvB,MAAM,WAAW,KAAK,IAAI,GAAG,KAAK,MAAM,MAAM,iBAAiB,UAAU,UAAU,EAAE,CAAC,CAAC;AAAA,IACvF,KAAK,IAAI,GAAG,KAAK,MAAM,QAAQ,eAAe,CAAC,CAAC;AAAA,IAChD,OAAO,UAAU;AACf,YAAM,kBAAkB,MAAM,MAAM,WAAW,MAAM,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC;AAC7E,UAAI,gBAAgB,WAAW,MAAM,QAAQ;AAC3C,cAAM,IAAI;AAAA,UACR,4BAA4B,gBAAgB,MAAM,mBAAmB,MAAM,MAAM;AAAA,QACnF;AAAA,MACF;AACA,aAAO,MAAM,IAAI,CAAC,MAAM,WAAW;AAAA,QACjC,eAAe,KAAK;AAAA,QACpB,WAAW,gBAAgB,KAAK;AAAA,MAClC,EAAE;AAAA,IACJ;AAAA,EACF;AAEA,QAAM,aAAa,oBAAI,IAAyB;AAChD,aAAW,QAAQ,WAAW,KAAK,GAAG;AACpC,UAAM,OAAO,WAAW,IAAI,KAAK,aAAa,KAAK,CAAC;AACpD,SAAK,KAAK,KAAK,SAAS;AACxB,eAAW,IAAI,KAAK,eAAe,IAAI;AAAA,EACzC;AAEA,SAAO,SAAS,IAAI,CAAC,MAAM,WAAW;AAAA,IACpC,IAAI,KAAK;AAAA,IACT,UAAU,KAAK;AAAA,IACf,GAAI,KAAK,aAAa,UAAa,EAAE,UAAU,KAAK,SAAS;AAAA,IAC7D,YAAY,WAAW,IAAI,KAAK,KAAK,CAAC;AAAA,EACxC,EAAE;AACJ;AAEA,SAAS,MAAS,OAAY,MAAqB;AACjD,QAAM,SAAgB,CAAC;AACvB,WAAS,QAAQ,GAAG,QAAQ,MAAM,QAAQ,SAAS,MAAM;AACvD,WAAO,KAAK,MAAM,MAAM,OAAO,QAAQ,IAAI,CAAC;AAAA,EAC9C;AACA,SAAO;AACT;","names":[]}

package/dist/{chunk-XXT2UCAR.js → chunk-M6RPFHG7.js}

@@ -1,4 +1,4 @@
-// src/memory/index.ts
+// src/memory/options.ts
 function resolveMemoryOptions(options = {}) {
   return {
     savePolicy: options.savePolicy ?? "message"
@@ -8,4 +8,4 @@ function resolveMemoryOptions(options = {}) {
 export {
   resolveMemoryOptions
 };
-//# sourceMappingURL=chunk-XXT2UCAR.js.map
+//# sourceMappingURL=chunk-M6RPFHG7.js.map

package/dist/chunk-M6RPFHG7.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/memory/options.ts"],"sourcesContent":["import type { MemoryOptions, ResolvedMemoryOptions } from \"./types\";\n\nexport function resolveMemoryOptions(options: MemoryOptions = {}): ResolvedMemoryOptions {\n  return {\n    savePolicy: options.savePolicy ?? \"message\",\n  };\n}\n"],"mappings":";AAEO,SAAS,qBAAqB,UAAyB,CAAC,GAA0B;AACvF,SAAO;AAAA,IACL,YAAY,QAAQ,cAAc;AAAA,EACpC;AACF;","names":[]}

package/dist/chunk-MLMXQ2OV.js

@@ -0,0 +1,205 @@
+import {
+  PromptRequest
+} from "./chunk-7CQAERL3.js";
+import {
+  isSkillTool
+} from "./chunk-YK4WAAS4.js";
+import {
+  ToolSet
+} from "./chunk-FV5B4SGK.js";
+import {
+  createTool
+} from "./chunk-5EDHXYH5.js";
+import {
+  isStreamingCompletionModel
+} from "./chunk-SRBW3LD7.js";
+import {
+  compact
+} from "./chunk-MMHG7WAM.js";
+
+// src/agent/agent.ts
+import { z } from "zod";
+
+// src/agent/ids.ts
+function normalizeAgentId(id) {
+  if (typeof id !== "string") {
+    throw new TypeError("Agent id must be a string.");
+  }
+  const normalized = id.trim();
+  if (normalized.length === 0) {
+    throw new TypeError("Agent id must be a non-empty string.");
+  }
+  return normalized;
+}
+
+// src/agent/agent.ts
+var DEFAULT_MAX_TURNS = 20;
+var Agent = class {
+  id;
+  name;
+  description;
+  model;
+  instructions;
+  staticContext;
+  temperature;
+  maxTokens;
+  additionalParams;
+  toolSet;
+  toolChoice;
+  defaultMaxTurns;
+  hook;
+  outputSchema;
+  observers;
+  approvals;
+  dynamicContexts;
+  dynamicTools;
+  middlewares;
+  /**
+   * @deprecated Use `middlewares` instead.
+   */
+  toolMiddlewares;
+  memory;
+  eventStore;
+  constructor(options) {
+    this.id = normalizeAgentId(options.id);
+    this.name = options.name;
+    this.description = options.description;
+    this.model = options.model;
+    this.instructions = options.instructions;
+    this.staticContext = options.staticContext ?? [];
+    this.temperature = options.temperature;
+    this.maxTokens = options.maxTokens;
+    this.additionalParams = options.additionalParams;
+    this.toolSet = options.toolSet ?? new ToolSet();
+    this.toolChoice = options.toolChoice;
+    this.defaultMaxTurns = options.defaultMaxTurns ?? DEFAULT_MAX_TURNS;
+    this.hook = options.hook;
+    this.outputSchema = options.outputSchema;
+    this.observers = options.observers ?? [];
+    this.approvals = options.approvals;
+    this.dynamicContexts = options.dynamicContexts ?? [];
+    this.dynamicTools = options.dynamicTools ?? [];
+    this.middlewares = options.middlewares ?? options.toolMiddlewares ?? [];
+    this.toolMiddlewares = this.middlewares;
+    this.memory = options.memory;
+    this.eventStore = options.eventStore;
+  }
+  prompt(prompt) {
+    return PromptRequest.fromAgent(this, prompt);
+  }
+  session(sessionId, options = {}) {
+    if (this.memory === void 0) {
+      throw new Error(`Agent "${this.id}" has no memory store configured.`);
+    }
+    const normalized = sessionId.trim();
+    if (normalized.length === 0) {
+      throw new TypeError("Session id must be a non-empty string.");
+    }
+    return new AgentSession(this, {
+      sessionId: normalized,
+      ...options.userId !== void 0 && { userId: options.userId },
+      ...options.metadata !== void 0 && { metadata: options.metadata }
+    });
+  }
+  asTool(options) {
+    const description = options.description ?? this.description ?? `Prompt the ${options.name} agent.`;
+    return createTool({
+      name: options.name,
+      description,
+      input: z.object({
+        prompt: z.string().describe("The prompt to send to the agent.")
+      }),
+      output: z.string(),
+      execute: async ({ prompt }, context) => {
+        const request = this.prompt(prompt);
+        const childRequest = options.maxTurns === void 0 ? request : request.maxTurns(options.maxTurns);
+        if (options.stream === true && context.emitStreamEvent !== void 0 && this.model.capabilities.streaming && isStreamingCompletionModel(this.model)) {
+          let output = "";
+          for await (const event of childRequest.stream()) {
+            await context.emitStreamEvent(
+              compact({
+                agentId: this.id,
+                agentName: this.name,
+                event
+              })
+            );
+            if (event.type === "final") {
+              output = event.output;
+            }
+          }
+          return output;
+        }
+        const response = await childRequest.send();
+        return response.output;
+      }
+    });
+  }
+  getTool(toolName) {
+    const staticTool = this.toolSet.get(toolName);
+    if (staticTool !== void 0) {
+      return staticTool;
+    }
+    for (const registration of this.dynamicTools) {
+      const dynamicTool = dynamicToolSetFromIndex(registration.index)?.get(toolName);
+      if (dynamicTool !== void 0) {
+        return dynamicTool;
+      }
+    }
+    return void 0;
+  }
+  async callTool(toolName, args, context) {
+    if (this.toolSet.contains(toolName)) {
+      return this.toolSet.call(toolName, args, context);
+    }
+    for (const registration of this.dynamicTools) {
+      const toolSet = dynamicToolSetFromIndex(registration.index);
+      if (toolSet?.contains(toolName)) {
+        return toolSet.call(toolName, args, context);
+      }
+    }
+    return this.toolSet.call(toolName, args, context);
+  }
+  shouldApplyToolMiddleware(toolName) {
+    return !isSkillTool(this.getTool(toolName));
+  }
+};
+var AgentSession = class {
+  constructor(agent, context) {
+    this.agent = agent;
+    this.context = context;
+  }
+  agent;
+  context;
+  prompt(prompt) {
+    if (Array.isArray(prompt)) {
+      throw new TypeError("AgentSession.prompt does not accept Message[] transcripts.");
+    }
+    return PromptRequest.fromAgent(this.agent, prompt, { memoryContext: this.context });
+  }
+  async messages() {
+    const memory = this.agent.memory;
+    if (memory === void 0) {
+      throw new Error(`Agent "${this.agent.id}" has no memory store configured.`);
+    }
+    return memory.store.load(this.context);
+  }
+  async clear() {
+    const memory = this.agent.memory;
+    if (memory === void 0) {
+      throw new Error(`Agent "${this.agent.id}" has no memory store configured.`);
+    }
+    await memory.store.clear(this.context);
+  }
+};
+function dynamicToolSetFromIndex(index) {
+  const maybeIndex = index;
+  return maybeIndex.toolSet instanceof ToolSet ? maybeIndex.toolSet : void 0;
+}
+
+export {
+  normalizeAgentId,
+  DEFAULT_MAX_TURNS,
+  Agent,
+  AgentSession
+};
+//# sourceMappingURL=chunk-MLMXQ2OV.js.map

package/dist/chunk-MLMXQ2OV.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/agent/agent.ts","../src/agent/ids.ts"],"sourcesContent":["import { z } from \"zod\";\nimport { isStreamingCompletionModel } from \"../completion/create-completion\";\nimport type {\n  CompletionModel,\n  Document,\n  JsonObject,\n  JsonValue,\n  Message as MessageType,\n  ToolChoice,\n} from \"../completion/index\";\nimport type { PromptHook } from \"../hooks\";\nimport { compact } from \"../internal/compact\";\nimport type { MemoryRegistration, SessionOptions } from \"../memory/types\";\nimport type { AgentObserverRegistration } from \"../observability\";\nimport { PromptRequest } from \"../request\";\nimport { createTool } from \"../tool/create-tool\";\nimport type { ToolSearchDocument } from \"../tool/dynamic-tools\";\nimport type { AgentMiddleware } from \"../tool/middleware\";\nimport { isSkillTool } from \"../tool/skill-tool-marker\";\nimport type {\n  AnyTool,\n  NormalizedToolOutput,\n  Tool,\n  ToolApprovalsOptions,\n  ToolCallContext,\n} from \"../tool/tool\";\nimport { ToolSet } from \"../tool/tool-set\";\nimport type { VectorSearchIndex } from \"../vector-store\";\nimport { normalizeAgentId } from \"./ids\";\nimport type {\n  AgentEventStoreRegistration,\n  AgentOptions,\n  AgentToolOptions,\n  DynamicContextRegistration,\n  DynamicToolRegistration,\n} from \"./types\";\n\nexport const DEFAULT_MAX_TURNS = 20;\n\nexport class Agent<M extends CompletionModel = CompletionModel> {\n  readonly id: string;\n  readonly name: string | undefined;\n  readonly description: string | undefined;\n  readonly model: M;\n  readonly instructions: string | undefined;\n  readonly staticContext: Document[];\n  readonly temperature: number | undefined;\n  readonly maxTokens: number | undefined;\n  readonly additionalParams: JsonValue | undefined;\n  readonly toolSet: ToolSet;\n  readonly toolChoice: ToolChoice | undefined;\n  readonly defaultMaxTurns: number | undefined;\n  readonly hook: PromptHook | undefined;\n  readonly outputSchema: JsonObject | undefined;\n  readonly observers: AgentObserverRegistration[];\n  readonly approvals: ToolApprovalsOptions | undefined;\n  readonly dynamicContexts: DynamicContextRegistration[];\n  readonly dynamicTools: DynamicToolRegistration[];\n  readonly middlewares: AgentMiddleware[];\n  /**\n   * @deprecated Use `middlewares` instead.\n   */\n  readonly toolMiddlewares: AgentMiddleware[];\n  readonly memory: MemoryRegistration | undefined;\n  readonly eventStore: AgentEventStoreRegistration | undefined;\n\n  constructor(options: AgentOptions<M>) {\n    this.id = normalizeAgentId(options.id);\n    this.name = options.name;\n    this.description = options.description;\n    this.model = options.model;\n    this.instructions = options.instructions;\n    this.staticContext = options.staticContext ?? [];\n    this.temperature = options.temperature;\n    this.maxTokens = options.maxTokens;\n    this.additionalParams = options.additionalParams;\n    this.toolSet = options.toolSet ?? new ToolSet();\n    this.toolChoice = options.toolChoice;\n    this.defaultMaxTurns = options.defaultMaxTurns ?? DEFAULT_MAX_TURNS;\n    this.hook = options.hook;\n    this.outputSchema = options.outputSchema;\n    this.observers = options.observers ?? [];\n    this.approvals = options.approvals;\n    this.dynamicContexts = options.dynamicContexts ?? [];\n    this.dynamicTools = options.dynamicTools ?? [];\n    this.middlewares = options.middlewares ?? options.toolMiddlewares ?? [];\n    this.toolMiddlewares = this.middlewares;\n    this.memory = options.memory;\n    this.eventStore = options.eventStore;\n  }\n\n  prompt(prompt: string | MessageType | MessageType[]): PromptRequest<M> {\n    return PromptRequest.fromAgent(this, prompt);\n  }\n\n  session(sessionId: string, options: SessionOptions = {}): AgentSession<M> {\n    if (this.memory === undefined) {\n      throw new Error(`Agent \"${this.id}\" has no memory store configured.`);\n    }\n    const normalized = sessionId.trim();\n    if (normalized.length === 0) {\n      throw new TypeError(\"Session id must be a non-empty string.\");\n    }\n    return new AgentSession(this, {\n      sessionId: normalized,\n      ...(options.userId !== undefined && { userId: options.userId }),\n      ...(options.metadata !== undefined && { metadata: options.metadata }),\n    });\n  }\n\n  asTool(options: AgentToolOptions): Tool<{ prompt: string }, string> {\n    const description =\n      options.description ?? this.description ?? `Prompt the ${options.name} agent.`;\n\n    return createTool({\n      name: options.name,\n      description,\n      input: z.object({\n        prompt: z.string().describe(\"The prompt to send to the agent.\"),\n      }),\n      output: z.string(),\n      execute: async ({ prompt }, context: ToolCallContext) => {\n        const request = this.prompt(prompt);\n        const childRequest =\n          options.maxTurns === undefined ? request : request.maxTurns(options.maxTurns);\n        if (\n          options.stream === true &&\n          context.emitStreamEvent !== undefined &&\n          this.model.capabilities.streaming &&\n          isStreamingCompletionModel(this.model)\n        ) {\n          let output = \"\";\n          for await (const event of childRequest.stream()) {\n            await context.emitStreamEvent(\n              compact({\n                agentId: this.id,\n                agentName: this.name,\n                event,\n              }),\n            );\n            if (event.type === \"final\") {\n              output = event.output;\n            }\n          }\n          return output;\n        }\n        const response = await childRequest.send();\n        return response.output;\n      },\n    });\n  }\n\n  getTool(toolName: string): AnyTool | undefined {\n    const staticTool = this.toolSet.get(toolName);\n    if (staticTool !== undefined) {\n      return staticTool;\n    }\n\n    for (const registration of this.dynamicTools) {\n      const dynamicTool = dynamicToolSetFromIndex(registration.index)?.get(toolName);\n      if (dynamicTool !== undefined) {\n        return dynamicTool;\n      }\n    }\n\n    return undefined;\n  }\n\n  async callTool(\n    toolName: string,\n    args: string,\n    context?: ToolCallContext,\n  ): Promise<NormalizedToolOutput> {\n    if (this.toolSet.contains(toolName)) {\n      return this.toolSet.call(toolName, args, context);\n    }\n\n    for (const registration of this.dynamicTools) {\n      const toolSet = dynamicToolSetFromIndex(registration.index);\n      if (toolSet?.contains(toolName)) {\n        return toolSet.call(toolName, args, context);\n      }\n    }\n\n    return this.toolSet.call(toolName, args, context);\n  }\n\n  shouldApplyToolMiddleware(toolName: string): boolean {\n    return !isSkillTool(this.getTool(toolName));\n  }\n}\n\nexport class AgentSession<M extends CompletionModel = CompletionModel> {\n  constructor(\n    private readonly agent: Agent<M>,\n    private readonly context: {\n      sessionId: string;\n      userId?: string | undefined;\n      metadata?: JsonObject | undefined;\n    },\n  ) {}\n\n  prompt(prompt: string | MessageType): PromptRequest<M> {\n    if (Array.isArray(prompt)) {\n      throw new TypeError(\"AgentSession.prompt does not accept Message[] transcripts.\");\n    }\n    return PromptRequest.fromAgent(this.agent, prompt, { memoryContext: this.context });\n  }\n\n  async messages(): Promise<MessageType[]> {\n    const memory = this.agent.memory;\n    if (memory === undefined) {\n      throw new Error(`Agent \"${this.agent.id}\" has no memory store configured.`);\n    }\n    return memory.store.load(this.context);\n  }\n\n  async clear(): Promise<void> {\n    const memory = this.agent.memory;\n    if (memory === undefined) {\n      throw new Error(`Agent \"${this.agent.id}\" has no memory store configured.`);\n    }\n    await memory.store.clear(this.context);\n  }\n}\n\nfunction dynamicToolSetFromIndex(\n  index: VectorSearchIndex<ToolSearchDocument>,\n): ToolSet | undefined {\n  const maybeIndex = index as { toolSet?: unknown };\n  return maybeIndex.toolSet instanceof ToolSet ? maybeIndex.toolSet : undefined;\n}\n","export function normalizeAgentId(id: string): string {\n  if (typeof id !== \"string\") {\n    throw new TypeError(\"Agent id must be a string.\");\n  }\n\n  const normalized = id.trim();\n  if (normalized.length === 0) {\n    throw new TypeError(\"Agent id must be a non-empty string.\");\n  }\n\n  return normalized;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA,SAAS,SAAS;;;ACAX,SAAS,iBAAiB,IAAoB;AACnD,MAAI,OAAO,OAAO,UAAU;AAC1B,UAAM,IAAI,UAAU,4BAA4B;AAAA,EAClD;AAEA,QAAM,aAAa,GAAG,KAAK;AAC3B,MAAI,WAAW,WAAW,GAAG;AAC3B,UAAM,IAAI,UAAU,sCAAsC;AAAA,EAC5D;AAEA,SAAO;AACT;;;AD0BO,IAAM,oBAAoB;AAE1B,IAAM,QAAN,MAAyD;AAAA,EACrD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA;AAAA;AAAA,EAIA;AAAA,EACA;AAAA,EACA;AAAA,EAET,YAAY,SAA0B;AACpC,SAAK,KAAK,iBAAiB,QAAQ,EAAE;AACrC,SAAK,OAAO,QAAQ;AACpB,SAAK,cAAc,QAAQ;AAC3B,SAAK,QAAQ,QAAQ;AACrB,SAAK,eAAe,QAAQ;AAC5B,SAAK,gBAAgB,QAAQ,iBAAiB,CAAC;AAC/C,SAAK,cAAc,QAAQ;AAC3B,SAAK,YAAY,QAAQ;AACzB,SAAK,mBAAmB,QAAQ;AAChC,SAAK,UAAU,QAAQ,WAAW,IAAI,QAAQ;AAC9C,SAAK,aAAa,QAAQ;AAC1B,SAAK,kBAAkB,QAAQ,mBAAmB;AAClD,SAAK,OAAO,QAAQ;AACpB,SAAK,eAAe,QAAQ;AAC5B,SAAK,YAAY,QAAQ,aAAa,CAAC;AACvC,SAAK,YAAY,QAAQ;AACzB,SAAK,kBAAkB,QAAQ,mBAAmB,CAAC;AACnD,SAAK,eAAe,QAAQ,gBAAgB,CAAC;AAC7C,SAAK,cAAc,QAAQ,eAAe,QAAQ,mBAAmB,CAAC;AACtE,SAAK,kBAAkB,KAAK;AAC5B,SAAK,SAAS,QAAQ;AACtB,SAAK,aAAa,QAAQ;AAAA,EAC5B;AAAA,EAEA,OAAO,QAAgE;AACrE,WAAO,cAAc,UAAU,MAAM,MAAM;AAAA,EAC7C;AAAA,EAEA,QAAQ,WAAmB,UAA0B,CAAC,GAAoB;AACxE,QAAI,KAAK,WAAW,QAAW;AAC7B,YAAM,IAAI,MAAM,UAAU,KAAK,EAAE,mCAAmC;AAAA,IACtE;AACA,UAAM,aAAa,UAAU,KAAK;AAClC,QAAI,WAAW,WAAW,GAAG;AAC3B,YAAM,IAAI,UAAU,wCAAwC;AAAA,IAC9D;AACA,WAAO,IAAI,aAAa,MAAM;AAAA,MAC5B,WAAW;AAAA,MACX,GAAI,QAAQ,WAAW,UAAa,EAAE,QAAQ,QAAQ,OAAO;AAAA,MAC7D,GAAI,QAAQ,aAAa,UAAa,EAAE,UAAU,QAAQ,SAAS;AAAA,IACrE,CAAC;AAAA,EACH;AAAA,EAEA,OAAO,SAA6D;AAClE,UAAM,cACJ,QAAQ,eAAe,KAAK,eAAe,cAAc,QAAQ,IAAI;AAEvE,WAAO,WAAW;AAAA,MAChB,MAAM,QAAQ;AAAA,MACd;AAAA,MACA,OAAO,EAAE,OAAO;AAAA,QACd,QAAQ,EAAE,OAAO,EAAE,SAAS,kCAAkC;AAAA,MAChE,CAAC;AAAA,MACD,QAAQ,EAAE,OAAO;AAAA,MACjB,SAAS,OAAO,EAAE,OAAO,GAAG,YAA6B;AACvD,cAAM,UAAU,KAAK,OAAO,MAAM;AAClC,cAAM,eACJ,QAAQ,aAAa,SAAY,UAAU,QAAQ,SAAS,QAAQ,QAAQ;AAC9E,YACE,QAAQ,WAAW,QACnB,QAAQ,oBAAoB,UAC5B,KAAK,MAAM,aAAa,aACxB,2BAA2B,KAAK,KAAK,GACrC;AACA,cAAI,SAAS;AACb,2BAAiB,SAAS,aAAa,OAAO,GAAG;AAC/C,kBAAM,QAAQ;AAAA,cACZ,QAAQ;AAAA,gBACN,SAAS,KAAK;AAAA,gBACd,WAAW,KAAK;AAAA,gBAChB;AAAA,cACF,CAAC;AAAA,YACH;AACA,gBAAI,MAAM,SAAS,SAAS;AAC1B,uBAAS,MAAM;AAAA,YACjB;AAAA,UACF;AACA,iBAAO;AAAA,QACT;AACA,cAAM,WAAW,MAAM,aAAa,KAAK;AACzC,eAAO,SAAS;AAAA,MAClB;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,QAAQ,UAAuC;AAC7C,UAAM,aAAa,KAAK,QAAQ,IAAI,QAAQ;AAC5C,QAAI,eAAe,QAAW;AAC5B,aAAO;AAAA,IACT;AAEA,eAAW,gBAAgB,KAAK,cAAc;AAC5C,YAAM,cAAc,wBAAwB,aAAa,KAAK,GAAG,IAAI,QAAQ;AAC7E,UAAI,gBAAgB,QAAW;AAC7B,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,SACJ,UACA,MACA,SAC+B;AAC/B,QAAI,KAAK,QAAQ,SAAS,QAAQ,GAAG;AACnC,aAAO,KAAK,QAAQ,KAAK,UAAU,MAAM,OAAO;AAAA,IAClD;AAEA,eAAW,gBAAgB,KAAK,cAAc;AAC5C,YAAM,UAAU,wBAAwB,aAAa,KAAK;AAC1D,UAAI,SAAS,SAAS,QAAQ,GAAG;AAC/B,eAAO,QAAQ,KAAK,UAAU,MAAM,OAAO;AAAA,MAC7C;AAAA,IACF;AAEA,WAAO,KAAK,QAAQ,KAAK,UAAU,MAAM,OAAO;AAAA,EAClD;AAAA,EAEA,0BAA0B,UAA2B;AACnD,WAAO,CAAC,YAAY,KAAK,QAAQ,QAAQ,CAAC;AAAA,EAC5C;AACF;AAEO,IAAM,eAAN,MAAgE;AAAA,EACrE,YACmB,OACA,SAKjB;AANiB;AACA;AAAA,EAKhB;AAAA,EANgB;AAAA,EACA;AAAA,EAOnB,OAAO,QAAgD;AACrD,QAAI,MAAM,QAAQ,MAAM,GAAG;AACzB,YAAM,IAAI,UAAU,4DAA4D;AAAA,IAClF;AACA,WAAO,cAAc,UAAU,KAAK,OAAO,QAAQ,EAAE,eAAe,KAAK,QAAQ,CAAC;AAAA,EACpF;AAAA,EAEA,MAAM,WAAmC;AACvC,UAAM,SAAS,KAAK,MAAM;AAC1B,QAAI,WAAW,QAAW;AACxB,YAAM,IAAI,MAAM,UAAU,KAAK,MAAM,EAAE,mCAAmC;AAAA,IAC5E;AACA,WAAO,OAAO,MAAM,KAAK,KAAK,OAAO;AAAA,EACvC;AAAA,EAEA,MAAM,QAAuB;AAC3B,UAAM,SAAS,KAAK,MAAM;AAC1B,QAAI,WAAW,QAAW;AACxB,YAAM,IAAI,MAAM,UAAU,KAAK,MAAM,EAAE,mCAAmC;AAAA,IAC5E;AACA,UAAM,OAAO,MAAM,MAAM,KAAK,OAAO;AAAA,EACvC;AACF;AAEA,SAAS,wBACP,OACqB;AACrB,QAAM,aAAa;AACnB,SAAO,WAAW,mBAAmB,UAAU,WAAW,UAAU;AACtE;","names":[]}

package/dist/{chunk-M4T4ID34.js → chunk-NI637AGP.js}

@@ -1,13 +1,13 @@
 import {
   Agent,
   normalizeAgentId
-} from "./chunk-IFQVPV32.js";
-import {
-  resolveMemoryOptions
-} from "./chunk-XXT2UCAR.js";
+} from "./chunk-MLMXQ2OV.js";
 import {
   ToolSet
-} from "./chunk-BVGDCFQW.js";
+} from "./chunk-FV5B4SGK.js";
+import {
+  resolveMemoryOptions
+} from "./chunk-M6RPFHG7.js";
 import {
   toProviderJsonSchema
 } from "./chunk-WQKHFADH.js";
@@ -203,4 +203,4 @@ var AgentBuilder = class {
 export {
   AgentBuilder
 };
-//# sourceMappingURL=chunk-M4T4ID34.js.map
+//# sourceMappingURL=chunk-NI637AGP.js.map

package/dist/chunk-NI637AGP.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/agent/builder.ts"],"sourcesContent":["import type { CompletionModel, Document, JsonObject, JsonValue, ToolChoice } from \"../completion\";\nimport type { PromptHook } from \"../hooks\";\nimport type { McpServer } from \"../mcp\";\nimport { resolveMemoryOptions } from \"../memory/options\";\nimport type { MemoryOptions, MemoryRegistration, MemoryStore } from \"../memory/types\";\nimport type { AgentObserver, AgentObserverRegistration, ObserveOptions } from \"../observability\";\nimport { toProviderJsonSchema, type ZodSchema } from \"../schema/zod-schema\";\nimport type { SkillSet } from \"../skills\";\nimport type { ToolSearchDocument } from \"../tool/dynamic-tools\";\nimport type { AgentMiddleware, ToolMiddleware } from \"../tool/middleware\";\nimport type { AnyTool, ToolApprovalsOptions } from \"../tool/tool\";\nimport { ToolSet } from \"../tool/tool-set\";\nimport type { VectorSearchIndex } from \"../vector-store\";\nimport { Agent } from \"./agent\";\nimport { normalizeAgentId } from \"./ids\";\nimport type {\n  AgentEventStore,\n  AgentEventStoreOptions,\n  AgentEventStoreRegistration,\n  DynamicContextOptions,\n  DynamicContextRegistration,\n  DynamicToolOptions,\n  DynamicToolRegistration,\n} from \"./types\";\n\nexport class AgentBuilder<M extends CompletionModel = CompletionModel> {\n  private readonly agentId: string;\n  private agentName: string | undefined;\n  private agentDescription: string | undefined;\n  private instructionBlocks: string[] = [];\n  private contextDocs: Document[] = [];\n  private temp: number | undefined;\n  private maxTokenCount: number | undefined;\n  private params: JsonValue | undefined;\n  private choice: ToolChoice | undefined;\n  private turns: number | undefined;\n  private requestHook: PromptHook | undefined;\n  private schema: JsonObject | undefined;\n  private approvalOptions: ToolApprovalsOptions | undefined;\n  private skillInstructionBlocks: string[] = [];\n  private observerRegistrations: AgentObserverRegistration[] = [];\n  private dynamicContextRegistrations: DynamicContextRegistration[] = [];\n  private dynamicToolRegistrations: DynamicToolRegistration[] = [];\n  private middlewareRegistrations: AgentMiddleware[] = [];\n  private memoryRegistration: MemoryRegistration | undefined;\n  private eventStoreRegistration: AgentEventStoreRegistration | undefined;\n  private activeToolSet = new ToolSet();\n\n  constructor(\n    agentId: string,\n    private readonly completionModel: M,\n  ) {\n    this.agentId = normalizeAgentId(agentId);\n  }\n\n  name(name: string): this {\n    this.agentName = name;\n    return this;\n  }\n\n  description(description: string): this {\n    this.agentDescription = description;\n    return this;\n  }\n\n  instructions(instructions: string): this {\n    if (instructions.length > 0) {\n      this.instructionBlocks.push(instructions);\n    }\n    return this;\n  }\n\n  context(text: string, id = `static_doc_${this.contextDocs.length}`): this {\n    this.contextDocs.push({ id, text });\n    return this;\n  }\n\n  dynamicContext<T>(index: VectorSearchIndex<T>, options: DynamicContextOptions<T>): this {\n    this.dynamicContextRegistrations.push({ index, options } as DynamicContextRegistration);\n    return this;\n  }\n\n  dynamicTools(index: VectorSearchIndex<ToolSearchDocument>, options: DynamicToolOptions): this {\n    this.dynamicToolRegistrations.push({ index, options });\n    return this;\n  }\n\n  tool(tool: AnyTool): this {\n    this.activeToolSet.addTool(tool);\n    return this;\n  }\n\n  tools(tools: AnyTool[]): this {\n    this.activeToolSet.addTools(tools);\n    return this;\n  }\n\n  mcp(servers: McpServer[]): this {\n    for (const server of servers) {\n      this.activeToolSet.addTools(server.tools);\n    }\n    return this;\n  }\n\n  skills(skillSet: SkillSet): this {\n    if (skillSet.instructions.length > 0) {\n      this.skillInstructionBlocks.push(skillSet.instructions);\n    }\n    this.activeToolSet.addTools(skillSet.tools);\n    return this;\n  }\n\n  useToolSet(toolSet: ToolSet): this {\n    toolSet.addTools(this.activeToolSet);\n    this.activeToolSet = toolSet;\n    return this;\n  }\n\n  temperature(temperature: number): this {\n    this.temp = temperature;\n    return this;\n  }\n\n  maxTokens(maxTokens: number): this {\n    this.maxTokenCount = maxTokens;\n    return this;\n  }\n\n  additionalParams(params: JsonValue): this {\n    this.params = params;\n    return this;\n  }\n\n  toolChoice(toolChoice: ToolChoice): this {\n    this.choice = toolChoice;\n    return this;\n  }\n\n  defaultMaxTurns(defaultMaxTurns: number): this {\n    this.turns = defaultMaxTurns;\n    return this;\n  }\n\n  hook(hook: PromptHook): this {\n    this.requestHook = hook;\n    return this;\n  }\n\n  middleware(middleware: AgentMiddleware): this {\n    this.middlewareRegistrations.push(middleware);\n    return this;\n  }\n\n  middlewares(middlewares: AgentMiddleware[]): this {\n    this.middlewareRegistrations.push(...middlewares);\n    return this;\n  }\n\n  /**\n   * @deprecated Use `middleware` instead.\n   */\n  toolMiddleware(middleware: ToolMiddleware): this {\n    return this.middleware(middleware);\n  }\n\n  /**\n   * @deprecated Use `middlewares` instead.\n   */\n  toolMiddlewares(middlewares: ToolMiddleware[]): this {\n    return this.middlewares(middlewares);\n  }\n\n  observe(observer: AgentObserver, options: ObserveOptions = {}): this {\n    this.observerRegistrations.push({\n      observer,\n      failOnObserverError: options.failOnObserverError,\n    });\n    return this;\n  }\n\n  approvals(options: ToolApprovalsOptions): this {\n    this.approvalOptions = options;\n    return this;\n  }\n\n  memory(store: MemoryStore, options: MemoryOptions = {}): this {\n    this.memoryRegistration = {\n      store,\n      options: resolveMemoryOptions(options),\n    };\n    return this;\n  }\n\n  eventStore(store: AgentEventStore, options: AgentEventStoreOptions = {}): this {\n    this.eventStoreRegistration = {\n      store,\n      options: {\n        include: options.include ?? \"all\",\n      },\n    };\n    return this;\n  }\n\n  outputSchema(schema: ZodSchema): this {\n    this.schema = toProviderJsonSchema(schema);\n    return this;\n  }\n\n  build(): Agent<M> {\n    return new Agent({\n      id: this.agentId,\n      name: this.agentName,\n      description: this.agentDescription,\n      model: this.completionModel,\n      instructions: this.buildInstructions(),\n      staticContext: this.contextDocs,\n      temperature: this.temp,\n      maxTokens: this.maxTokenCount,\n      additionalParams: this.params,\n      toolSet: this.activeToolSet,\n      toolChoice: this.choice,\n      defaultMaxTurns: this.turns,\n      hook: this.requestHook,\n      outputSchema: this.schema,\n      observers: this.observerRegistrations,\n      approvals: this.approvalOptions,\n      dynamicContexts: this.dynamicContextRegistrations,\n      dynamicTools: this.dynamicToolRegistrations,\n      middlewares: this.middlewareRegistrations,\n      memory: this.memoryRegistration,\n      eventStore: this.eventStoreRegistration,\n    });\n  }\n\n  private buildInstructions(): string | undefined {\n    const parts = [...this.instructionBlocks, ...this.skillInstructionBlocks].filter(\n      (part): part is string => part !== undefined && part.length > 0,\n    );\n    return parts.length === 0 ? undefined : parts.join(\"\\n\\n\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;AAyBO,IAAM,eAAN,MAAgE;AAAA,EAuBrE,YACE,SACiB,iBACjB;AADiB;AAEjB,SAAK,UAAU,iBAAiB,OAAO;AAAA,EACzC;AAAA,EAHmB;AAAA,EAxBF;AAAA,EACT;AAAA,EACA;AAAA,EACA,oBAA8B,CAAC;AAAA,EAC/B,cAA0B,CAAC;AAAA,EAC3B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,yBAAmC,CAAC;AAAA,EACpC,wBAAqD,CAAC;AAAA,EACtD,8BAA4D,CAAC;AAAA,EAC7D,2BAAsD,CAAC;AAAA,EACvD,0BAA6C,CAAC;AAAA,EAC9C;AAAA,EACA;AAAA,EACA,gBAAgB,IAAI,QAAQ;AAAA,EASpC,KAAK,MAAoB;AACvB,SAAK,YAAY;AACjB,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAA2B;AACrC,SAAK,mBAAmB;AACxB,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,cAA4B;AACvC,QAAI,aAAa,SAAS,GAAG;AAC3B,WAAK,kBAAkB,KAAK,YAAY;AAAA,IAC1C;AACA,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ,MAAc,KAAK,cAAc,KAAK,YAAY,MAAM,IAAU;AACxE,SAAK,YAAY,KAAK,EAAE,IAAI,KAAK,CAAC;AAClC,WAAO;AAAA,EACT;AAAA,EAEA,eAAkB,OAA6B,SAAyC;AACtF,SAAK,4BAA4B,KAAK,EAAE,OAAO,QAAQ,CAA+B;AACtF,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,OAA8C,SAAmC;AAC5F,SAAK,yBAAyB,KAAK,EAAE,OAAO,QAAQ,CAAC;AACrD,WAAO;AAAA,EACT;AAAA,EAEA,KAAK,MAAqB;AACxB,SAAK,cAAc,QAAQ,IAAI;AAC/B,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAAwB;AAC5B,SAAK,cAAc,SAAS,KAAK;AACjC,WAAO;AAAA,EACT;AAAA,EAEA,IAAI,SAA4B;AAC9B,eAAW,UAAU,SAAS;AAC5B,WAAK,cAAc,SAAS,OAAO,KAAK;AAAA,IAC1C;AACA,WAAO;AAAA,EACT;AAAA,EAEA,OAAO,UAA0B;AAC/B,QAAI,SAAS,aAAa,SAAS,GAAG;AACpC,WAAK,uBAAuB,KAAK,SAAS,YAAY;AAAA,IACxD;AACA,SAAK,cAAc,SAAS,SAAS,KAAK;AAC1C,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,SAAwB;AACjC,YAAQ,SAAS,KAAK,aAAa;AACnC,SAAK,gBAAgB;AACrB,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAA2B;AACrC,SAAK,OAAO;AACZ,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,WAAyB;AACjC,SAAK,gBAAgB;AACrB,WAAO;AAAA,EACT;AAAA,EAEA,iBAAiB,QAAyB;AACxC,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,YAA8B;AACvC,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,gBAAgB,iBAA+B;AAC7C,SAAK,QAAQ;AACb,WAAO;AAAA,EACT;AAAA,EAEA,KAAK,MAAwB;AAC3B,SAAK,cAAc;AACnB,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,YAAmC;AAC5C,SAAK,wBAAwB,KAAK,UAAU;AAC5C,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAAsC;AAChD,SAAK,wBAAwB,KAAK,GAAG,WAAW;AAChD,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,eAAe,YAAkC;AAC/C,WAAO,KAAK,WAAW,UAAU;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA,EAKA,gBAAgB,aAAqC;AACnD,WAAO,KAAK,YAAY,WAAW;AAAA,EACrC;AAAA,EAEA,QAAQ,UAAyB,UAA0B,CAAC,GAAS;AACnE,SAAK,sBAAsB,KAAK;AAAA,MAC9B;AAAA,MACA,qBAAqB,QAAQ;AAAA,IAC/B,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,SAAqC;AAC7C,SAAK,kBAAkB;AACvB,WAAO;AAAA,EACT;AAAA,EAEA,OAAO,OAAoB,UAAyB,CAAC,GAAS;AAC5D,SAAK,qBAAqB;AAAA,MACxB;AAAA,MACA,SAAS,qBAAqB,OAAO;AAAA,IACvC;AACA,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,OAAwB,UAAkC,CAAC,GAAS;AAC7E,SAAK,yBAAyB;AAAA,MAC5B;AAAA,MACA,SAAS;AAAA,QACP,SAAS,QAAQ,WAAW;AAAA,MAC9B;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,QAAyB;AACpC,SAAK,SAAS,qBAAqB,MAAM;AACzC,WAAO;AAAA,EACT;AAAA,EAEA,QAAkB;AAChB,WAAO,IAAI,MAAM;AAAA,MACf,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,OAAO,KAAK;AAAA,MACZ,cAAc,KAAK,kBAAkB;AAAA,MACrC,eAAe,KAAK;AAAA,MACpB,aAAa,KAAK;AAAA,MAClB,WAAW,KAAK;AAAA,MAChB,kBAAkB,KAAK;AAAA,MACvB,SAAS,KAAK;AAAA,MACd,YAAY,KAAK;AAAA,MACjB,iBAAiB,KAAK;AAAA,MACtB,MAAM,KAAK;AAAA,MACX,cAAc,KAAK;AAAA,MACnB,WAAW,KAAK;AAAA,MAChB,WAAW,KAAK;AAAA,MAChB,iBAAiB,KAAK;AAAA,MACtB,cAAc,KAAK;AAAA,MACnB,aAAa,KAAK;AAAA,MAClB,QAAQ,KAAK;AAAA,MACb,YAAY,KAAK;AAAA,IACnB,CAAC;AAAA,EACH;AAAA,EAEQ,oBAAwC;AAC9C,UAAM,QAAQ,CAAC,GAAG,KAAK,mBAAmB,GAAG,KAAK,sBAAsB,EAAE;AAAA,MACxE,CAAC,SAAyB,SAAS,UAAa,KAAK,SAAS;AAAA,IAChE;AACA,WAAO,MAAM,WAAW,IAAI,SAAY,MAAM,KAAK,MAAM;AAAA,EAC3D;AACF;","names":[]}

package/dist/chunk-NPZDYOE6.js

@@ -0,0 +1,53 @@
+import {
+  compact
+} from "./chunk-MMHG7WAM.js";
+
+// src/hooks/control.ts
+function createHook(hook) {
+  return hook;
+}
+function cancelPrompt(reason) {
+  return { type: "terminate", reason };
+}
+function skipTool(reason) {
+  return { type: "skip", reason };
+}
+function requestToolApproval(options = {}) {
+  return compact({
+    type: "approval_request",
+    reason: options.reason,
+    rejectMessage: options.rejectMessage
+  });
+}
+var runControl = {
+  continue() {
+    return { type: "continue" };
+  },
+  cancel(reason) {
+    return cancelPrompt(reason);
+  }
+};
+var toolCallControl = {
+  run() {
+    return { type: "continue" };
+  },
+  skip(reason) {
+    return skipTool(reason);
+  },
+  cancel(reason) {
+    return { type: "terminate", reason };
+  },
+  requestApproval(options) {
+    return requestToolApproval(options);
+  }
+};
+
+export {
+  createHook,
+  cancelPrompt,
+  skipTool,
+  requestToolApproval,
+  runControl,
+  toolCallControl
+};
+//# sourceMappingURL=chunk-NPZDYOE6.js.map

package/dist/chunk-NPZDYOE6.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/control.ts"],"sourcesContent":["import { compact } from \"../internal/compact\";\nimport type {\n  HookAction,\n  PromptHook,\n  RunControl,\n  ToolApprovalRequestOptions,\n  ToolCallControl,\n  ToolCallHookAction,\n} from \"./types\";\n\nexport function createHook<RawResponse = unknown>(\n  hook: PromptHook<RawResponse>,\n): PromptHook<RawResponse> {\n  return hook;\n}\n\nexport function cancelPrompt(reason: string): HookAction {\n  return { type: \"terminate\", reason };\n}\n\nexport function skipTool(reason: string): ToolCallHookAction {\n  return { type: \"skip\", reason };\n}\n\nexport function requestToolApproval(options: ToolApprovalRequestOptions = {}): ToolCallHookAction {\n  return compact({\n    type: \"approval_request\" as const,\n    reason: options.reason,\n    rejectMessage: options.rejectMessage,\n  }) as ToolCallHookAction;\n}\n\nexport const runControl: RunControl = {\n  continue() {\n    return { type: \"continue\" };\n  },\n  cancel(reason: string) {\n    return cancelPrompt(reason);\n  },\n};\n\nexport const toolCallControl: ToolCallControl = {\n  run() {\n    return { type: \"continue\" };\n  },\n  skip(reason: string) {\n    return skipTool(reason);\n  },\n  cancel(reason: string) {\n    return { type: \"terminate\", reason };\n  },\n  requestApproval(options) {\n    return requestToolApproval(options);\n  },\n};\n"],"mappings":";;;;;AAUO,SAAS,WACd,MACyB;AACzB,SAAO;AACT;AAEO,SAAS,aAAa,QAA4B;AACvD,SAAO,EAAE,MAAM,aAAa,OAAO;AACrC;AAEO,SAAS,SAAS,QAAoC;AAC3D,SAAO,EAAE,MAAM,QAAQ,OAAO;AAChC;AAEO,SAAS,oBAAoB,UAAsC,CAAC,GAAuB;AAChG,SAAO,QAAQ;AAAA,IACb,MAAM;AAAA,IACN,QAAQ,QAAQ;AAAA,IAChB,eAAe,QAAQ;AAAA,EACzB,CAAC;AACH;AAEO,IAAM,aAAyB;AAAA,EACpC,WAAW;AACT,WAAO,EAAE,MAAM,WAAW;AAAA,EAC5B;AAAA,EACA,OAAO,QAAgB;AACrB,WAAO,aAAa,MAAM;AAAA,EAC5B;AACF;AAEO,IAAM,kBAAmC;AAAA,EAC9C,MAAM;AACJ,WAAO,EAAE,MAAM,WAAW;AAAA,EAC5B;AAAA,EACA,KAAK,QAAgB;AACnB,WAAO,SAAS,MAAM;AAAA,EACxB;AAAA,EACA,OAAO,QAAgB;AACrB,WAAO,EAAE,MAAM,aAAa,OAAO;AAAA,EACrC;AAAA,EACA,gBAAgB,SAAS;AACvB,WAAO,oBAAoB,OAAO;AAAA,EACpC;AACF;","names":[]}

package/dist/{chunk-5G3VME2D.js → chunk-SRBW3LD7.js}

@@ -2,7 +2,7 @@ import {
   Message,
   assertCompletionRequestSupported,
   textFromAssistantContent
-} from "./chunk-WJO6NVE2.js";
+} from "./chunk-5LFGEMIW.js";
 import {
   toProviderJsonSchema
 } from "./chunk-WQKHFADH.js";
@@ -335,4 +335,4 @@ export {
   formatDocument,
   CompletionRequestBuilder
 };
-//# sourceMappingURL=chunk-5G3VME2D.js.map
+//# sourceMappingURL=chunk-SRBW3LD7.js.map

package/dist/chunk-SRBW3LD7.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/completion/create-completion.ts","../src/completion/documents.ts","../src/completion/request.ts"],"sourcesContent":["import { toProviderJsonSchema, type ZodSchema } from \"../schema/zod-schema\";\nimport type {\n  AssistantContent,\n  CompletionModel,\n  CompletionRequest,\n  CompletionResponse,\n  CompletionStreamEvent,\n  Document,\n  JsonObject,\n  JsonValue,\n  Message as MessageType,\n  StreamingCompletionModel,\n  ToolChoice,\n  ToolDefinition,\n  Usage,\n} from \"./types\";\nimport { assertCompletionRequestSupported, Message, textFromAssistantContent } from \"./types\";\n\nexport type CreateCompletionInput = string | MessageType | MessageType[];\n\nexport type CreateCompletionBaseOptions = {\n  input?: CreateCompletionInput | undefined;\n  messages?: MessageType[] | undefined;\n  instructions?: string | undefined;\n  documents?: Document[] | undefined;\n  tools?: ToolDefinition[] | undefined;\n  temperature?: number | undefined;\n  maxTokens?: number | undefined;\n  toolChoice?: ToolChoice | undefined;\n  outputSchema?: JsonObject | undefined;\n  params?: JsonValue | undefined;\n};\n\nexport type CreateCompletionOptions = CreateCompletionBaseOptions;\n\nexport type CreateCompletionStreamOptions = CreateCompletionBaseOptions;\n\nexport type CreateParsedCompletionOptions<T> = Omit<CreateCompletionBaseOptions, \"outputSchema\"> & {\n  schema: ZodSchema<T>;\n};\n\nexport type CreateCompletionResult<RawResponse = unknown> = {\n  text: string;\n  content: AssistantContent[];\n  usage: Usage;\n  response: CompletionResponse<RawResponse>;\n};\n\nexport type CreateParsedCompletionResult<\n  T,\n  RawResponse = unknown,\n> = CreateCompletionResult<RawResponse> & {\n  data: T;\n};\n\ntype RawResponseOf<Model> =\n  Model extends CompletionModel<infer RawResponse, infer _ModelName> ? RawResponse : unknown;\n\nexport function createCompletion<Model extends CompletionModel>(\n  model: Model,\n  options: CreateCompletionOptions,\n): Promise<CreateCompletionResult<RawResponseOf<Model>>> {\n  return sendCompletion(model, options);\n}\n\nexport function createCompletionStream<Model extends StreamingCompletionModel>(\n  model: Model,\n  options: CreateCompletionStreamOptions,\n): AsyncIterable<CompletionStreamEvent<RawResponseOf<Model>>> {\n  const request = toCompletionRequest(options);\n  if (!isStreamingCompletionModel(model) || !model.capabilities.streaming) {\n    throw new Error(\"This completion model does not support streaming\");\n  }\n  assertCompletionRequestSupported(model, request, { streaming: true });\n  return model.streamCompletion(request) as AsyncIterable<\n    CompletionStreamEvent<RawResponseOf<Model>>\n  >;\n}\n\nexport async function createParsedCompletion<T, Model extends CompletionModel>(\n  model: Model,\n  options: CreateParsedCompletionOptions<T>,\n): Promise<CreateParsedCompletionResult<T, RawResponseOf<Model>>> {\n  const { schema, ...completionOptions } = options;\n  const request = toCompletionRequest(\n    {\n      ...completionOptions,\n      outputSchema: toProviderJsonSchema(schema),\n    },\n    \"createParsedCompletion\",\n  );\n  assertCompletionRequestSupported(model, request);\n  const response = (await model.completion(request)) as CompletionResponse<RawResponseOf<Model>>;\n  const text = textFromAssistantContent(response.choice);\n  return {\n    data: parseCompletionData(text, schema),\n    text,\n    content: response.choice,\n    usage: response.usage,\n    response,\n  };\n}\n\nasync function sendCompletion<Model extends CompletionModel>(\n  model: Model,\n  options: CreateCompletionOptions,\n): Promise<CreateCompletionResult<RawResponseOf<Model>>> {\n  const request = toCompletionRequest(options);\n  assertCompletionRequestSupported(model, request);\n  const response = (await model.completion(request)) as CompletionResponse<RawResponseOf<Model>>;\n  return {\n    text: textFromAssistantContent(response.choice),\n    content: response.choice,\n    usage: response.usage,\n    response,\n  };\n}\n\nfunction toCompletionRequest(\n  options: CreateCompletionBaseOptions,\n  helperName = \"createCompletion\",\n): CompletionRequest {\n  const chatHistory = [\n    ...messagesFromMessages(options.messages),\n    ...messagesFromInput(options.input),\n  ];\n\n  if (chatHistory.length === 0) {\n    throw new Error(`${helperName} requires input or messages.`);\n  }\n\n  const request: CompletionRequest = {\n    chatHistory,\n    documents: [...(options.documents ?? [])],\n    tools: [...(options.tools ?? [])],\n  };\n\n  if (options.instructions !== undefined && options.instructions.length > 0) {\n    request.instructions = options.instructions;\n  }\n  if (options.temperature !== undefined) request.temperature = options.temperature;\n  if (options.maxTokens !== undefined) request.maxTokens = options.maxTokens;\n  if (options.toolChoice !== undefined) request.toolChoice = options.toolChoice;\n  if (options.outputSchema !== undefined) request.outputSchema = options.outputSchema;\n  if (options.params !== undefined) request.additionalParams = options.params;\n\n  return request;\n}\n\nfunction messagesFromInput(input: CreateCompletionInput | undefined): MessageType[] {\n  if (input === undefined) {\n    return [];\n  }\n  if (typeof input === \"string\") {\n    return [Message.user(input)];\n  }\n  if (Array.isArray(input)) {\n    return normalizeMessageArray(input, \"input\");\n  }\n  if (!isCoreMessage(input)) {\n    throw new TypeError(\"input must be a string, Message, or Message[].\");\n  }\n  return [input];\n}\n\nfunction messagesFromMessages(messages: MessageType[] | undefined): MessageType[] {\n  return messages === undefined ? [] : normalizeMessageArray(messages, \"messages\");\n}\n\nfunction normalizeMessageArray(\n  messages: MessageType[],\n  fieldName: \"input\" | \"messages\",\n): MessageType[] {\n  if (!messages.every(isCoreMessage)) {\n    throw new TypeError(`${fieldName} must contain only Message values.`);\n  }\n\n  return [...messages];\n}\n\nfunction isCoreMessage(value: unknown): value is MessageType {\n  if (!isRecord(value)) {\n    return false;\n  }\n\n  if (value.role === \"system\") {\n    return typeof value.content === \"string\";\n  }\n\n  if (value.role === \"user\") {\n    return Array.isArray(value.content) && value.content.every(isUserContent);\n  }\n\n  if (value.role === \"assistant\") {\n    return (\n      (value.id === undefined || typeof value.id === \"string\") &&\n      Array.isArray(value.content) &&\n      value.content.every(isAssistantContent)\n    );\n  }\n\n  if (value.role === \"tool\") {\n    return Array.isArray(value.content) && value.content.every(isToolContent);\n  }\n\n  return false;\n}\n\nfunction isUserContent(value: unknown): boolean {\n  if (!isRecord(value)) {\n    return false;\n  }\n\n  if (value.type === \"text\") {\n    return typeof value.text === \"string\";\n  }\n\n  if (value.type === \"image\") {\n    return isImageContent(value);\n  }\n\n  if (value.type === \"document\") {\n    return isDocumentContent(value);\n  }\n\n  return false;\n}\n\nfunction isAssistantContent(value: unknown): boolean {\n  if (!isRecord(value)) {\n    return false;\n  }\n\n  if (value.type === \"text\") {\n    return typeof value.text === \"string\";\n  }\n\n  if (value.type === \"reasoning\") {\n    return (\n      typeof value.text === \"string\" &&\n      (value.id === undefined || typeof value.id === \"string\") &&\n      (value.content === undefined || Array.isArray(value.content))\n    );\n  }\n\n  if (value.type === \"tool_call\") {\n    return (\n      typeof value.id === \"string\" &&\n      (value.callId === undefined || typeof value.callId === \"string\") &&\n      isRecord(value.function) &&\n      typeof value.function.name === \"string\" &&\n      \"arguments\" in value.function\n    );\n  }\n\n  if (value.type === \"image\") {\n    return isImageContent(value);\n  }\n\n  return false;\n}\n\nfunction isToolContent(value: unknown): boolean {\n  return (\n    isRecord(value) &&\n    value.type === \"tool_result\" &&\n    typeof value.id === \"string\" &&\n    (value.callId === undefined || typeof value.callId === \"string\") &&\n    Array.isArray(value.content) &&\n    value.content.every(isToolResultContent)\n  );\n}\n\nfunction isToolResultContent(value: unknown): boolean {\n  if (!isRecord(value)) {\n    return false;\n  }\n\n  if (value.type === \"text\") {\n    return typeof value.text === \"string\";\n  }\n\n  if (value.type === \"image\") {\n    return (\n      typeof value.data === \"string\" &&\n      (value.mediaType === undefined || typeof value.mediaType === \"string\")\n    );\n  }\n\n  return false;\n}\n\nfunction isImageContent(value: Record<string, unknown>): boolean {\n  if (!isRecord(value.source)) {\n    return false;\n  }\n\n  if (value.source.type === \"url\") {\n    return typeof value.source.url === \"string\";\n  }\n\n  if (value.source.type === \"base64\") {\n    return typeof value.source.data === \"string\" && typeof value.source.mediaType === \"string\";\n  }\n\n  return false;\n}\n\nfunction isDocumentContent(value: Record<string, unknown>): boolean {\n  if (!isRecord(value.source)) {\n    return false;\n  }\n\n  if (value.source.type === \"url\") {\n    return typeof value.source.url === \"string\" && typeof value.source.mediaType === \"string\";\n  }\n\n  if (value.source.type === \"base64\") {\n    return typeof value.source.data === \"string\" && typeof value.source.mediaType === \"string\";\n  }\n\n  if (value.source.type === \"text\") {\n    return (\n      typeof value.source.text === \"string\" &&\n      (value.source.mediaType === undefined || typeof value.source.mediaType === \"string\")\n    );\n  }\n\n  return false;\n}\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return typeof value === \"object\" && value !== null;\n}\n\nexport function isStreamingCompletionModel(\n  model: CompletionModel,\n): model is StreamingCompletionModel {\n  return typeof (model as { streamCompletion?: unknown }).streamCompletion === \"function\";\n}\n\nfunction parseCompletionData<T>(text: string, schema: ZodSchema<T>): T {\n  let json: unknown;\n  try {\n    json = JSON.parse(text);\n  } catch (error) {\n    throw new Error(\"createParsedCompletion expected the model response to be valid JSON.\", {\n      cause: error,\n    });\n  }\n\n  return schema.parse(json);\n}\n","import { type Document, Message, type Message as MessageType } from \"./types\";\n\nexport function normalizeDocuments(documents: Document[]): MessageType | undefined {\n  if (documents.length === 0) {\n    return undefined;\n  }\n\n  return Message.user(documents.map(formatDocument).join(\"\\n\"));\n}\n\nexport function formatDocument(document: Document): string {\n  return `<file id: ${document.id}>\\n${formatDocumentBody(document)}\\n</file>\\n`;\n}\n\nfunction formatDocumentBody(document: Document): string {\n  const metadata = formatMetadata(document.additionalProps);\n  return metadata === undefined ? document.text : `${metadata}\\n${document.text}`;\n}\n\nfunction formatMetadata(additionalProps: Record<string, string> | undefined): string | undefined {\n  if (additionalProps === undefined) {\n    return undefined;\n  }\n\n  const entries = Object.entries(additionalProps).sort(([left], [right]) =>\n    left.localeCompare(right),\n  );\n  if (entries.length === 0) {\n    return undefined;\n  }\n\n  const metadata = entries.map(([key, value]) => `${key}: ${JSON.stringify(value)}`).join(\" \");\n  return `<metadata ${metadata} />`;\n}\n","import type {\n  CompletionModel,\n  CompletionRequest,\n  CompletionResponse,\n  Document,\n  JsonObject,\n  JsonValue,\n  Message as MessageType,\n  ToolChoice,\n  ToolDefinition,\n} from \"./types\";\nimport { assertCompletionRequestSupported } from \"./types\";\n\ntype ModelNameOf<M extends CompletionModel> =\n  M extends CompletionModel<unknown, infer ModelName> ? ModelName : string;\n\ntype RawResponseOf<M extends CompletionModel> =\n  M extends CompletionModel<infer RawResponse, infer _ModelName> ? RawResponse : unknown;\n\nexport class CompletionRequestBuilder<M extends CompletionModel = CompletionModel> {\n  private requestModel: ModelNameOf<M> | undefined;\n  private instructionBlocks: string[] = [];\n  private history: MessageType[] = [];\n  private docs: Document[] = [];\n  private toolDefs: ToolDefinition[] = [];\n  private temp: number | undefined;\n  private maxTokenCount: number | undefined;\n  private choice: ToolChoice | undefined;\n  private params: JsonValue | undefined;\n  private schema: JsonObject | undefined;\n\n  constructor(\n    private readonly model: M,\n    private readonly promptMessage: MessageType,\n  ) {}\n\n  modelOverride(model: ModelNameOf<M> | undefined): this {\n    this.requestModel = model;\n    return this;\n  }\n\n  instructions(instructions: string | undefined): this {\n    if (instructions !== undefined && instructions.length > 0) {\n      this.instructionBlocks.push(instructions);\n    }\n    return this;\n  }\n\n  messages(messages: MessageType[]): this {\n    this.history.push(...messages);\n    return this;\n  }\n\n  documents(documents: Document[]): this {\n    this.docs.push(...documents);\n    return this;\n  }\n\n  tools(tools: ToolDefinition[]): this {\n    this.toolDefs.push(...tools);\n    return this;\n  }\n\n  temperature(temperature: number | undefined): this {\n    this.temp = temperature;\n    return this;\n  }\n\n  maxTokens(maxTokens: number | undefined): this {\n    this.maxTokenCount = maxTokens;\n    return this;\n  }\n\n  toolChoice(toolChoice: ToolChoice | undefined): this {\n    this.choice = toolChoice;\n    return this;\n  }\n\n  additionalParams(additionalParams: JsonValue | undefined): this {\n    this.params = additionalParams;\n    return this;\n  }\n\n  outputSchema(outputSchema: JsonObject | undefined): this {\n    this.schema = outputSchema;\n    return this;\n  }\n\n  build(): CompletionRequest<ModelNameOf<M>> {\n    const instructions = this.buildInstructions();\n    const request: CompletionRequest<ModelNameOf<M>> = {\n      chatHistory: [...this.history, this.promptMessage],\n      documents: [...this.docs],\n      tools: [...this.toolDefs],\n    };\n\n    if (this.requestModel !== undefined) request.model = this.requestModel;\n    if (instructions !== undefined) request.instructions = instructions;\n    if (this.temp !== undefined) request.temperature = this.temp;\n    if (this.maxTokenCount !== undefined) request.maxTokens = this.maxTokenCount;\n    if (this.choice !== undefined) request.toolChoice = this.choice;\n    if (this.params !== undefined) request.additionalParams = this.params;\n    if (this.schema !== undefined) request.outputSchema = this.schema;\n\n    return request;\n  }\n\n  async send(): Promise<CompletionResponse<RawResponseOf<M>>> {\n    const request = this.build();\n    assertCompletionRequestSupported(this.model, request);\n    return this.model.completion(request) as Promise<CompletionResponse<RawResponseOf<M>>>;\n  }\n\n  private buildInstructions(): string | undefined {\n    return this.instructionBlocks.length === 0 ? undefined : this.instructionBlocks.join(\"\\n\\n\");\n  }\n}\n"],"mappings":";;;;;;;;;;AA0DO,SAAS,iBACd,OACA,SACuD;AACvD,SAAO,eAAe,OAAO,OAAO;AACtC;AAEO,SAAS,uBACd,OACA,SAC4D;AAC5D,QAAM,UAAU,oBAAoB,OAAO;AAC3C,MAAI,CAAC,2BAA2B,KAAK,KAAK,CAAC,MAAM,aAAa,WAAW;AACvE,UAAM,IAAI,MAAM,kDAAkD;AAAA,EACpE;AACA,mCAAiC,OAAO,SAAS,EAAE,WAAW,KAAK,CAAC;AACpE,SAAO,MAAM,iBAAiB,OAAO;AAGvC;AAEA,eAAsB,uBACpB,OACA,SACgE;AAChE,QAAM,EAAE,QAAQ,GAAG,kBAAkB,IAAI;AACzC,QAAM,UAAU;AAAA,IACd;AAAA,MACE,GAAG;AAAA,MACH,cAAc,qBAAqB,MAAM;AAAA,IAC3C;AAAA,IACA;AAAA,EACF;AACA,mCAAiC,OAAO,OAAO;AAC/C,QAAM,WAAY,MAAM,MAAM,WAAW,OAAO;AAChD,QAAM,OAAO,yBAAyB,SAAS,MAAM;AACrD,SAAO;AAAA,IACL,MAAM,oBAAoB,MAAM,MAAM;AAAA,IACtC;AAAA,IACA,SAAS,SAAS;AAAA,IAClB,OAAO,SAAS;AAAA,IAChB;AAAA,EACF;AACF;AAEA,eAAe,eACb,OACA,SACuD;AACvD,QAAM,UAAU,oBAAoB,OAAO;AAC3C,mCAAiC,OAAO,OAAO;AAC/C,QAAM,WAAY,MAAM,MAAM,WAAW,OAAO;AAChD,SAAO;AAAA,IACL,MAAM,yBAAyB,SAAS,MAAM;AAAA,IAC9C,SAAS,SAAS;AAAA,IAClB,OAAO,SAAS;AAAA,IAChB;AAAA,EACF;AACF;AAEA,SAAS,oBACP,SACA,aAAa,oBACM;AACnB,QAAM,cAAc;AAAA,IAClB,GAAG,qBAAqB,QAAQ,QAAQ;AAAA,IACxC,GAAG,kBAAkB,QAAQ,KAAK;AAAA,EACpC;AAEA,MAAI,YAAY,WAAW,GAAG;AAC5B,UAAM,IAAI,MAAM,GAAG,UAAU,8BAA8B;AAAA,EAC7D;AAEA,QAAM,UAA6B;AAAA,IACjC;AAAA,IACA,WAAW,CAAC,GAAI,QAAQ,aAAa,CAAC,CAAE;AAAA,IACxC,OAAO,CAAC,GAAI,QAAQ,SAAS,CAAC,CAAE;AAAA,EAClC;AAEA,MAAI,QAAQ,iBAAiB,UAAa,QAAQ,aAAa,SAAS,GAAG;AACzE,YAAQ,eAAe,QAAQ;AAAA,EACjC;AACA,MAAI,QAAQ,gBAAgB,OAAW,SAAQ,cAAc,QAAQ;AACrE,MAAI,QAAQ,cAAc,OAAW,SAAQ,YAAY,QAAQ;AACjE,MAAI,QAAQ,eAAe,OAAW,SAAQ,aAAa,QAAQ;AACnE,MAAI,QAAQ,iBAAiB,OAAW,SAAQ,eAAe,QAAQ;AACvE,MAAI,QAAQ,WAAW,OAAW,SAAQ,mBAAmB,QAAQ;AAErE,SAAO;AACT;AAEA,SAAS,kBAAkB,OAAyD;AAClF,MAAI,UAAU,QAAW;AACvB,WAAO,CAAC;AAAA,EACV;AACA,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO,CAAC,QAAQ,KAAK,KAAK,CAAC;AAAA,EAC7B;AACA,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,WAAO,sBAAsB,OAAO,OAAO;AAAA,EAC7C;AACA,MAAI,CAAC,cAAc,KAAK,GAAG;AACzB,UAAM,IAAI,UAAU,gDAAgD;AAAA,EACtE;AACA,SAAO,CAAC,KAAK;AACf;AAEA,SAAS,qBAAqB,UAAoD;AAChF,SAAO,aAAa,SAAY,CAAC,IAAI,sBAAsB,UAAU,UAAU;AACjF;AAEA,SAAS,sBACP,UACA,WACe;AACf,MAAI,CAAC,SAAS,MAAM,aAAa,GAAG;AAClC,UAAM,IAAI,UAAU,GAAG,SAAS,oCAAoC;AAAA,EACtE;AAEA,SAAO,CAAC,GAAG,QAAQ;AACrB;AAEA,SAAS,cAAc,OAAsC;AAC3D,MAAI,CAAC,SAAS,KAAK,GAAG;AACpB,WAAO;AAAA,EACT;AAEA,MAAI,MAAM,SAAS,UAAU;AAC3B,WAAO,OAAO,MAAM,YAAY;AAAA,EAClC;AAEA,MAAI,MAAM,SAAS,QAAQ;AACzB,WAAO,MAAM,QAAQ,MAAM,OAAO,KAAK,MAAM,QAAQ,MAAM,aAAa;AAAA,EAC1E;AAEA,MAAI,MAAM,SAAS,aAAa;AAC9B,YACG,MAAM,OAAO,UAAa,OAAO,MAAM,OAAO,aAC/C,MAAM,QAAQ,MAAM,OAAO,KAC3B,MAAM,QAAQ,MAAM,kBAAkB;AAAA,EAE1C;AAEA,MAAI,MAAM,SAAS,QAAQ;AACzB,WAAO,MAAM,QAAQ,MAAM,OAAO,KAAK,MAAM,QAAQ,MAAM,aAAa;AAAA,EAC1E;AAEA,SAAO;AACT;AAEA,SAAS,cAAc,OAAyB;AAC9C,MAAI,CAAC,SAAS,KAAK,GAAG;AACpB,WAAO;AAAA,EACT;AAEA,MAAI,MAAM,SAAS,QAAQ;AACzB,WAAO,OAAO,MAAM,SAAS;AAAA,EAC/B;AAEA,MAAI,MAAM,SAAS,SAAS;AAC1B,WAAO,eAAe,KAAK;AAAA,EAC7B;AAEA,MAAI,MAAM,SAAS,YAAY;AAC7B,WAAO,kBAAkB,KAAK;AAAA,EAChC;AAEA,SAAO;AACT;AAEA,SAAS,mBAAmB,OAAyB;AACnD,MAAI,CAAC,SAAS,KAAK,GAAG;AACpB,WAAO;AAAA,EACT;AAEA,MAAI,MAAM,SAAS,QAAQ;AACzB,WAAO,OAAO,MAAM,SAAS;AAAA,EAC/B;AAEA,MAAI,MAAM,SAAS,aAAa;AAC9B,WACE,OAAO,MAAM,SAAS,aACrB,MAAM,OAAO,UAAa,OAAO,MAAM,OAAO,cAC9C,MAAM,YAAY,UAAa,MAAM,QAAQ,MAAM,OAAO;AAAA,EAE/D;AAEA,MAAI,MAAM,SAAS,aAAa;AAC9B,WACE,OAAO,MAAM,OAAO,aACnB,MAAM,WAAW,UAAa,OAAO,MAAM,WAAW,aACvD,SAAS,MAAM,QAAQ,KACvB,OAAO,MAAM,SAAS,SAAS,YAC/B,eAAe,MAAM;AAAA,EAEzB;AAEA,MAAI,MAAM,SAAS,SAAS;AAC1B,WAAO,eAAe,KAAK;AAAA,EAC7B;AAEA,SAAO;AACT;AAEA,SAAS,cAAc,OAAyB;AAC9C,SACE,SAAS,KAAK,KACd,MAAM,SAAS,iBACf,OAAO,MAAM,OAAO,aACnB,MAAM,WAAW,UAAa,OAAO,MAAM,WAAW,aACvD,MAAM,QAAQ,MAAM,OAAO,KAC3B,MAAM,QAAQ,MAAM,mBAAmB;AAE3C;AAEA,SAAS,oBAAoB,OAAyB;AACpD,MAAI,CAAC,SAAS,KAAK,GAAG;AACpB,WAAO;AAAA,EACT;AAEA,MAAI,MAAM,SAAS,QAAQ;AACzB,WAAO,OAAO,MAAM,SAAS;AAAA,EAC/B;AAEA,MAAI,MAAM,SAAS,SAAS;AAC1B,WACE,OAAO,MAAM,SAAS,aACrB,MAAM,cAAc,UAAa,OAAO,MAAM,cAAc;AAAA,EAEjE;AAEA,SAAO;AACT;AAEA,SAAS,eAAe,OAAyC;AAC/D,MAAI,CAAC,SAAS,MAAM,MAAM,GAAG;AAC3B,WAAO;AAAA,EACT;AAEA,MAAI,MAAM,OAAO,SAAS,OAAO;AAC/B,WAAO,OAAO,MAAM,OAAO,QAAQ;AAAA,EACrC;AAEA,MAAI,MAAM,OAAO,SAAS,UAAU;AAClC,WAAO,OAAO,MAAM,OAAO,SAAS,YAAY,OAAO,MAAM,OAAO,cAAc;AAAA,EACpF;AAEA,SAAO;AACT;AAEA,SAAS,kBAAkB,OAAyC;AAClE,MAAI,CAAC,SAAS,MAAM,MAAM,GAAG;AAC3B,WAAO;AAAA,EACT;AAEA,MAAI,MAAM,OAAO,SAAS,OAAO;AAC/B,WAAO,OAAO,MAAM,OAAO,QAAQ,YAAY,OAAO,MAAM,OAAO,cAAc;AAAA,EACnF;AAEA,MAAI,MAAM,OAAO,SAAS,UAAU;AAClC,WAAO,OAAO,MAAM,OAAO,SAAS,YAAY,OAAO,MAAM,OAAO,cAAc;AAAA,EACpF;AAEA,MAAI,MAAM,OAAO,SAAS,QAAQ;AAChC,WACE,OAAO,MAAM,OAAO,SAAS,aAC5B,MAAM,OAAO,cAAc,UAAa,OAAO,MAAM,OAAO,cAAc;AAAA,EAE/E;AAEA,SAAO;AACT;AAEA,SAAS,SAAS,OAAkD;AAClE,SAAO,OAAO,UAAU,YAAY,UAAU;AAChD;AAEO,SAAS,2BACd,OACmC;AACnC,SAAO,OAAQ,MAAyC,qBAAqB;AAC/E;AAEA,SAAS,oBAAuB,MAAc,QAAyB;AACrE,MAAI;AACJ,MAAI;AACF,WAAO,KAAK,MAAM,IAAI;AAAA,EACxB,SAAS,OAAO;AACd,UAAM,IAAI,MAAM,wEAAwE;AAAA,MACtF,OAAO;AAAA,IACT,CAAC;AAAA,EACH;AAEA,SAAO,OAAO,MAAM,IAAI;AAC1B;;;AC9VO,SAAS,mBAAmB,WAAgD;AACjF,MAAI,UAAU,WAAW,GAAG;AAC1B,WAAO;AAAA,EACT;AAEA,SAAO,QAAQ,KAAK,UAAU,IAAI,cAAc,EAAE,KAAK,IAAI,CAAC;AAC9D;AAEO,SAAS,eAAe,UAA4B;AACzD,SAAO,aAAa,SAAS,EAAE;AAAA,EAAM,mBAAmB,QAAQ,CAAC;AAAA;AAAA;AACnE;AAEA,SAAS,mBAAmB,UAA4B;AACtD,QAAM,WAAW,eAAe,SAAS,eAAe;AACxD,SAAO,aAAa,SAAY,SAAS,OAAO,GAAG,QAAQ;AAAA,EAAK,SAAS,IAAI;AAC/E;AAEA,SAAS,eAAe,iBAAyE;AAC/F,MAAI,oBAAoB,QAAW;AACjC,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,OAAO,QAAQ,eAAe,EAAE;AAAA,IAAK,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,MAClE,KAAK,cAAc,KAAK;AAAA,EAC1B;AACA,MAAI,QAAQ,WAAW,GAAG;AACxB,WAAO;AAAA,EACT;AAEA,QAAM,WAAW,QAAQ,IAAI,CAAC,CAAC,KAAK,KAAK,MAAM,GAAG,GAAG,KAAK,KAAK,UAAU,KAAK,CAAC,EAAE,EAAE,KAAK,GAAG;AAC3F,SAAO,aAAa,QAAQ;AAC9B;;;ACdO,IAAM,2BAAN,MAA4E;AAAA,EAYjF,YACmB,OACA,eACjB;AAFiB;AACA;AAAA,EAChB;AAAA,EAFgB;AAAA,EACA;AAAA,EAbX;AAAA,EACA,oBAA8B,CAAC;AAAA,EAC/B,UAAyB,CAAC;AAAA,EAC1B,OAAmB,CAAC;AAAA,EACpB,WAA6B,CAAC;AAAA,EAC9B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAOR,cAAc,OAAyC;AACrD,SAAK,eAAe;AACpB,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,cAAwC;AACnD,QAAI,iBAAiB,UAAa,aAAa,SAAS,GAAG;AACzD,WAAK,kBAAkB,KAAK,YAAY;AAAA,IAC1C;AACA,WAAO;AAAA,EACT;AAAA,EAEA,SAAS,UAA+B;AACtC,SAAK,QAAQ,KAAK,GAAG,QAAQ;AAC7B,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,WAA6B;AACrC,SAAK,KAAK,KAAK,GAAG,SAAS;AAC3B,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAA+B;AACnC,SAAK,SAAS,KAAK,GAAG,KAAK;AAC3B,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAAuC;AACjD,SAAK,OAAO;AACZ,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,WAAqC;AAC7C,SAAK,gBAAgB;AACrB,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,YAA0C;AACnD,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,iBAAiB,kBAA+C;AAC9D,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,cAA4C;AACvD,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,QAA2C;AACzC,UAAM,eAAe,KAAK,kBAAkB;AAC5C,UAAM,UAA6C;AAAA,MACjD,aAAa,CAAC,GAAG,KAAK,SAAS,KAAK,aAAa;AAAA,MACjD,WAAW,CAAC,GAAG,KAAK,IAAI;AAAA,MACxB,OAAO,CAAC,GAAG,KAAK,QAAQ;AAAA,IAC1B;AAEA,QAAI,KAAK,iBAAiB,OAAW,SAAQ,QAAQ,KAAK;AAC1D,QAAI,iBAAiB,OAAW,SAAQ,eAAe;AACvD,QAAI,KAAK,SAAS,OAAW,SAAQ,cAAc,KAAK;AACxD,QAAI,KAAK,kBAAkB,OAAW,SAAQ,YAAY,KAAK;AAC/D,QAAI,KAAK,WAAW,OAAW,SAAQ,aAAa,KAAK;AACzD,QAAI,KAAK,WAAW,OAAW,SAAQ,mBAAmB,KAAK;AAC/D,QAAI,KAAK,WAAW,OAAW,SAAQ,eAAe,KAAK;AAE3D,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAAsD;AAC1D,UAAM,UAAU,KAAK,MAAM;AAC3B,qCAAiC,KAAK,OAAO,OAAO;AACpD,WAAO,KAAK,MAAM,WAAW,OAAO;AAAA,EACtC;AAAA,EAEQ,oBAAwC;AAC9C,WAAO,KAAK,kBAAkB,WAAW,IAAI,SAAY,KAAK,kBAAkB,KAAK,MAAM;AAAA,EAC7F;AACF;","names":[]}

package/dist/{chunk-246MPVBE.js → chunk-V3WTT7TE.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-YK4WAAS4.js";
 import {
   createTool
-} from "./chunk-T22FYPE2.js";
+} from "./chunk-5EDHXYH5.js";
 import {
   isRecord
 } from "./chunk-MMHG7WAM.js";
@@ -395,4 +395,4 @@ export {
   SkillValidationError,
   skill
 };
-//# sourceMappingURL=chunk-246MPVBE.js.map
+//# sourceMappingURL=chunk-V3WTT7TE.js.map

package/dist/completion/index.d.ts

@@ -1,12 +1,14 @@
-export { C as CreateCompletionBaseOptions, a as CreateCompletionInput, b as CreateCompletionOptions, c as CreateCompletionResult, d as CreateCompletionStreamOptions, e as CreateParsedCompletionOptions, f as CreateParsedCompletionResult, g as createCompletion, h as createCompletionStream, i as createParsedCompletion, j as isStreamingCompletionModel } from '../create-completion-7GNUka0b.js';
-import { D as Document, M as Message, C as CompletionModel, h as ToolDefinition, o as ToolChoice, e as JsonValue, J as JsonObject, b as CompletionRequest, c as CompletionResponse } from '../types-CP3iyh6j.js';
-export { A as AssistantContent, a as AssistantMessage, r as CompletionCapabilityError, t as CompletionModelCapabilities, q as CompletionStreamEvent, u as DocumentContent, I as ImageContent, v as ImageDetail, d as JsonPrimitive, w as Reasoning, x as ReasoningContent, R as ReasoningContentType, p as StreamingCompletionModel, S as SystemMessage, T as Text, f as ToolCall, g as ToolContent, y as ToolFunction, i as ToolMessage, j as ToolResult, k as ToolResultContent, U as Usage, l as UserContent, m as UserMessage, z as assertCompletionRequestSupported, n as isToolResultContentArray, B as reasoningDisplayText, s as serializeToolResultOutput, E as textFromAssistantContent } from '../types-CP3iyh6j.js';
+export { C as CreateCompletionBaseOptions, a as CreateCompletionInput, b as CreateCompletionOptions, c as CreateCompletionResult, d as CreateCompletionStreamOptions, e as CreateParsedCompletionOptions, f as CreateParsedCompletionResult, g as createCompletion, h as createCompletionStream, i as createParsedCompletion, j as isStreamingCompletionModel } from '../create-completion-Y6YsYbQB.js';
+import { D as Document, M as Message, C as CompletionModel, h as ToolDefinition, n as ToolChoice, e as JsonValue, J as JsonObject, b as CompletionRequest, c as CompletionResponse } from '../types-q9G7Bvwl.js';
+export { A as AssistantContent, a as AssistantMessage, r as CompletionCapabilityError, t as CompletionModelCapabilities, p as CompletionStreamEvent, u as DocumentContent, I as ImageContent, v as ImageDetail, d as JsonPrimitive, R as Reasoning, w as ReasoningContent, x as ReasoningContentType, o as StreamingCompletionModel, S as SystemMessage, T as Text, f as ToolCall, g as ToolContent, y as ToolFunction, i as ToolMessage, j as ToolResult, k as ToolResultContent, U as Usage, l as UserContent, m as UserMessage, z as assertCompletionRequestSupported, q as isToolResultContentArray, B as reasoningDisplayText, s as serializeToolResultOutput, E as textFromAssistantContent } from '../types-q9G7Bvwl.js';
 import '../zod-schema-C7F4clpm.js';
 import 'zod';
 
 declare function normalizeDocuments(documents: Document[]): Message | undefined;
 declare function formatDocument(document: Document): string;
 
+type ModelNameOf<M extends CompletionModel> = M extends CompletionModel<unknown, infer ModelName> ? ModelName : string;
+type RawResponseOf<M extends CompletionModel> = M extends CompletionModel<infer RawResponse, infer _ModelName> ? RawResponse : unknown;
 declare class CompletionRequestBuilder<M extends CompletionModel = CompletionModel> {
     private readonly model;
     private readonly promptMessage;
@@ -21,7 +23,7 @@ declare class CompletionRequestBuilder<M extends CompletionModel = CompletionMod
     private params;
     private schema;
     constructor(model: M, promptMessage: Message);
-    modelOverride(model: string | undefined): this;
+    modelOverride(model: ModelNameOf<M> | undefined): this;
     instructions(instructions: string | undefined): this;
     messages(messages: Message[]): this;
     documents(documents: Document[]): this;
@@ -31,8 +33,8 @@ declare class CompletionRequestBuilder<M extends CompletionModel = CompletionMod
     toolChoice(toolChoice: ToolChoice | undefined): this;
     additionalParams(additionalParams: JsonValue | undefined): this;
     outputSchema(outputSchema: JsonObject | undefined): this;
-    build(): CompletionRequest;
-    send(): Promise<CompletionResponse>;
+    build(): CompletionRequest<ModelNameOf<M>>;
+    send(): Promise<CompletionResponse<RawResponseOf<M>>>;
     private buildInstructions;
 }
 

package/dist/completion/index.js

@@ -6,7 +6,7 @@ import {
   formatDocument,
   isStreamingCompletionModel,
   normalizeDocuments
-} from "../chunk-5G3VME2D.js";
+} from "../chunk-SRBW3LD7.js";
 import {
   AssistantContent,
   CompletionCapabilityError,
@@ -19,7 +19,7 @@ import {
   reasoningDisplayText,
   serializeToolResultOutput,
   textFromAssistantContent
-} from "../chunk-WJO6NVE2.js";
+} from "../chunk-5LFGEMIW.js";
 import "../chunk-WQKHFADH.js";
 export {
   AssistantContent,