@anvia/core 1.1.2 → 1.2.1

package/dist/chunk-F3E2RLC7.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/internal/agent-runtime/memory.ts"],"sourcesContent":["import type { Agent } from \"../../agent/agent\";\nimport type { AgentMemory } from \"../../agent/types\";\nimport { type Message as MessageType, Usage } from \"../../completion/index\";\nimport {\n  createMemoryCompactionSummary,\n  cumulativeCompactedMessageCount,\n} from \"../../memory/compaction\";\nimport { MemoryCompactionConflictError, MemoryCompactionError } from \"../../memory/errors\";\nimport type {\n  MemoryCompactionInfo,\n  MemoryCompactionResult,\n  MemorySavePolicy,\n  MemoryScope,\n  MemoryTokenCounter,\n} from \"../../memory/types\";\nimport { throwIfAborted } from \"../abort\";\n\nexport type MemoryPreparation = {\n  history: MessageType[];\n  usage: ReturnType<typeof Usage.empty>;\n  compaction?: MemoryCompactionInfo | undefined;\n  originalTokenCount?: number | undefined;\n};\n\ntype MemoryAgent = Pick<Agent, \"memory\">;\n\nexport class AgentRunMemory {\n  constructor(\n    private readonly agent: MemoryAgent,\n    private readonly memoryScope: MemoryScope | undefined,\n    private readonly initialHistory: MessageType[],\n  ) {}\n\n  memoryPolicy(): MemorySavePolicy | undefined {\n    return this.memory()?.savePolicy;\n  }\n\n  pendingTurnMessages(newMessages: MessageType[]): MessageType[] {\n    return this.memoryPolicy() === \"turn\" ? [...newMessages] : [];\n  }\n\n  async prepareHistory(\n    runId: string,\n    incomingMessages: readonly MessageType[],\n    abortSignal?: AbortSignal | undefined,\n  ): Promise<MemoryPreparation> {\n    const memory = this.memory();\n    if (memory === undefined || this.memoryScope === undefined) {\n      return {\n        history: this.initialHistory,\n        usage: Usage.empty(),\n      };\n    }\n\n    const preparation = await this.prepareStoredHistory(\n      memory,\n      runId,\n      incomingMessages,\n      abortSignal,\n    );\n    const memoryHistory = preparation.history;\n    const chatHistory = [...memoryHistory, ...this.initialHistory];\n    return {\n      ...preparation,\n      history: chatHistory,\n    };\n  }\n\n  async compact(\n    runId: string,\n    abortSignal?: AbortSignal | undefined,\n  ): Promise<MemoryCompactionResult> {\n    const memory = this.memory();\n    const scope = this.memoryScope;\n    if (memory === undefined || scope === undefined) {\n      throw new TypeError(\"Manual memory compaction requires an Agent with configured memory.\");\n    }\n    if (memory.compaction === undefined || memory.store.compaction === undefined) {\n      throw new TypeError(\"Manual memory compaction requires a configured compaction policy.\");\n    }\n    const preparation = await this.prepareStoredHistory(memory, runId, [], abortSignal, true);\n    if (preparation.compaction !== undefined) {\n      return { type: \"compacted\", ...preparation.compaction };\n    }\n    return {\n      type: \"skipped\",\n      reason: \"nothing_to_compact\",\n      originalMessageCount: preparation.history.length,\n      originalTokenCount:\n        preparation.originalTokenCount ??\n        (await countTokens(memory.compaction.tokenCounter, preparation.history)),\n    };\n  }\n\n  async commitAcceptedInput(runId: string, messages: MessageType[]): Promise<void> {\n    const memory = this.memory();\n    if (\n      memory === undefined ||\n      this.memoryScope === undefined ||\n      memory.savePolicy !== \"message\" ||\n      messages.length === 0\n    ) {\n      return;\n    }\n\n    await memory.store.append({\n      scope: this.memoryScope,\n      runId,\n      turn: 1,\n      messages,\n    });\n  }\n\n  async commitMessages(\n    runId: string,\n    turn: number,\n    messages: MessageType[],\n    pendingTurnMessages: MessageType[],\n  ): Promise<void> {\n    const memory = this.memory();\n    if (memory === undefined || this.memoryScope === undefined || messages.length === 0) {\n      return;\n    }\n    if (memory.savePolicy === \"message\") {\n      await memory.store.append({\n        scope: this.memoryScope,\n        runId,\n        turn,\n        messages,\n      });\n    } else if (memory.savePolicy === \"turn\") {\n      pendingTurnMessages.push(...messages);\n    }\n  }\n\n  async commitCompletedTurn(\n    runId: string,\n    turn: number,\n    pendingTurnMessages: MessageType[],\n  ): Promise<void> {\n    const memory = this.memory();\n    if (\n      memory === undefined ||\n      this.memoryScope === undefined ||\n      memory.savePolicy !== \"turn\" ||\n      pendingTurnMessages.length === 0\n    ) {\n      return;\n    }\n    await memory.store.append({\n      scope: this.memoryScope,\n      runId,\n      turn,\n      messages: [...pendingTurnMessages],\n    });\n    pendingTurnMessages.length = 0;\n  }\n\n  async commitCompletedRun(\n    runId: string,\n    turn: number,\n    newMessages: MessageType[],\n    pendingTurnMessages: MessageType[],\n  ): Promise<void> {\n    await this.commitCompletedTurn(runId, turn, pendingTurnMessages);\n    const memory = this.memory();\n    if (memory === undefined || this.memoryScope === undefined || memory.savePolicy !== \"run\") {\n      return;\n    }\n    await memory.store.append({\n      scope: this.memoryScope,\n      runId,\n      turn,\n      messages: [...newMessages],\n    });\n  }\n\n  async recordError(runId: string, error: unknown, newMessages: MessageType[]): Promise<void> {\n    const memory = this.memory();\n    if (memory === undefined || this.memoryScope === undefined) {\n      return;\n    }\n    await memory.store.recordError?.({\n      scope: this.memoryScope,\n      runId,\n      error,\n      messages: [...newMessages],\n    });\n  }\n\n  private memory(): AgentMemory | undefined {\n    return this.memoryScope === undefined ? undefined : this.agent.memory;\n  }\n\n  private async prepareStoredHistory(\n    memory: AgentMemory,\n    runId: string,\n    incomingMessages: readonly MessageType[],\n    abortSignal?: AbortSignal | undefined,\n    force = false,\n  ): Promise<MemoryPreparation> {\n    const scope = this.memoryScope;\n    if (scope === undefined) {\n      return { history: [], usage: Usage.empty() };\n    }\n    const options = memory.compaction;\n    const capability = memory.store.compaction;\n    if (options === undefined || capability === undefined) {\n      return {\n        history: await memory.store.load({ scope }),\n        usage: Usage.empty(),\n      };\n    }\n\n    let usage = Usage.empty();\n    const maxAttempts = options.conflictRetries === false ? 1 : options.conflictRetries.maxAttempts;\n    for (let attempt = 1; attempt <= maxAttempts; attempt += 1) {\n      throwIfAborted(abortSignal);\n      const snapshot = await capability.snapshot({ scope });\n      throwIfAborted(abortSignal);\n      const selection = await selectCompactionPrefix(\n        snapshot.messages,\n        incomingMessages,\n        options.trigger.afterTokens,\n        options.retention.recentTokens,\n        options.tokenCounter,\n        force,\n      );\n      throwIfAborted(abortSignal);\n      const compactedMessageCount = selection.compactedMessageCount;\n      if (compactedMessageCount === 0) {\n        return {\n          history: snapshot.messages,\n          usage,\n          originalTokenCount: selection.originalTokenCount,\n        };\n      }\n\n      const prefix = snapshot.messages.slice(0, compactedMessageCount);\n      let result: Awaited<ReturnType<typeof options.compactor>>;\n      try {\n        result = await options.compactor({\n          scope,\n          messages: prefix,\n          abortSignal,\n        });\n      } catch (error) {\n        throw remappedCompactorError(error, usage);\n      }\n      if (result.usage !== undefined) {\n        usage = Usage.add(usage, result.usage);\n      }\n      if (typeof result?.summary !== \"string\") {\n        throw new MemoryCompactionError(\"Memory compactor must return a summary string.\", {\n          usage,\n        });\n      }\n      const summaryText = result.summary.trim();\n      if (summaryText.length === 0) {\n        throw new MemoryCompactionError(\"Memory compactor returned an empty summary.\", {\n          usage,\n        });\n      }\n      const summary = createMemoryCompactionSummary(\n        summaryText,\n        cumulativeCompactedMessageCount(prefix),\n      );\n      const retained = snapshot.messages.slice(compactedMessageCount);\n      let resultTokenCount: number;\n      try {\n        resultTokenCount = await countTokens(options.tokenCounter, [summary, ...retained]);\n      } catch (error) {\n        throw remappedCompactorError(error, usage);\n      }\n      throwIfAborted(abortSignal);\n      let replacement: Awaited<ReturnType<typeof capability.replacePrefix>>;\n      try {\n        replacement = await capability.replacePrefix({\n          scope,\n          revision: snapshot.revision,\n          messageCount: compactedMessageCount,\n          replacement: summary,\n          runId: `memory-compaction:${runId}:${attempt}`,\n        });\n      } catch (error) {\n        throw new MemoryCompactionError(\"Memory compaction prefix replacement failed.\", {\n          cause: error,\n          usage,\n        });\n      }\n      if (replacement.status === \"committed\") {\n        return {\n          history: [summary, ...retained],\n          usage,\n          compaction: {\n            originalMessageCount: snapshot.messages.length,\n            compactedMessageCount,\n            retainedMessageCount: retained.length,\n            originalTokenCount: selection.originalTokenCount,\n            compactedTokenCount: selection.compactedTokenCount,\n            retainedTokenCount: selection.retainedTokenCount,\n            resultTokenCount,\n            attempts: attempt,\n            usage,\n          },\n        };\n      }\n      throwIfAborted(abortSignal);\n    }\n\n    throw new MemoryCompactionConflictError(maxAttempts, usage);\n  }\n}\n\ntype CompactionSelection = {\n  compactedMessageCount: number;\n  originalTokenCount: number;\n  compactedTokenCount: number;\n  retainedTokenCount: number;\n};\n\nasync function selectCompactionPrefix(\n  messages: readonly MessageType[],\n  incomingMessages: readonly MessageType[],\n  afterTokens: number,\n  recentTokens: number,\n  tokenCounter: MemoryTokenCounter,\n  force: boolean,\n): Promise<CompactionSelection> {\n  const originalTokenCount = await countTokens(tokenCounter, messages);\n  const incomingTokenCount =\n    incomingMessages.length === 0 ? 0 : await countTokens(tokenCounter, incomingMessages);\n  const none = {\n    compactedMessageCount: 0,\n    originalTokenCount,\n    compactedTokenCount: 0,\n    retainedTokenCount: originalTokenCount,\n  };\n  if (!force && originalTokenCount + incomingTokenCount <= afterTokens) {\n    return none;\n  }\n  const userMessageIndexes = messages.flatMap((message, index) =>\n    message.role === \"user\" ? [index] : [],\n  );\n  if (userMessageIndexes.length <= 1) {\n    return none;\n  }\n\n  // Retain complete user-led turns. The newest turn is always kept even when it alone exceeds the\n  // retention budget. Find the earliest newer turn whose tail fits with logarithmic counter calls.\n  const tailTokenCounts = new Map<number, number>();\n  const countTail = async (messageIndex: number): Promise<number> => {\n    const cached = tailTokenCounts.get(messageIndex);\n    if (cached !== undefined) return cached;\n    const count = await countTokens(tokenCounter, messages.slice(messageIndex));\n    tailTokenCounts.set(messageIndex, count);\n    return count;\n  };\n  let lower = 0;\n  let upper = userMessageIndexes.length - 1;\n  let retainedBoundary = upper;\n  while (lower <= upper) {\n    const middle = Math.floor((lower + upper) / 2);\n    const candidate = userMessageIndexes[middle] ?? 0;\n    if ((await countTail(candidate)) <= recentTokens) {\n      retainedBoundary = middle;\n      upper = middle - 1;\n    } else {\n      lower = middle + 1;\n    }\n  }\n  const compactedMessageCount = userMessageIndexes[retainedBoundary] ?? 0;\n  if (compactedMessageCount === 0) {\n    return none;\n  }\n  const compactedTokenCount = await countTokens(\n    tokenCounter,\n    messages.slice(0, compactedMessageCount),\n  );\n  const retainedTokenCount =\n    tailTokenCounts.get(compactedMessageCount) ??\n    (await countTokens(tokenCounter, messages.slice(compactedMessageCount)));\n  return {\n    compactedMessageCount,\n    originalTokenCount,\n    compactedTokenCount,\n    retainedTokenCount,\n  };\n}\n\nasync function countTokens(\n  tokenCounter: MemoryTokenCounter,\n  messages: readonly MessageType[],\n): Promise<number> {\n  let count: number;\n  try {\n    count = await tokenCounter(messages);\n  } catch (error) {\n    throw new MemoryCompactionError(\"Memory token counter failed.\", { cause: error });\n  }\n  if (!Number.isSafeInteger(count) || count < 0) {\n    throw new MemoryCompactionError(\"Memory token counter must return a nonnegative safe integer.\");\n  }\n  return count;\n}\n\nfunction remappedCompactorError(error: unknown, usage: ReturnType<typeof Usage.empty>): Error {\n  if (error instanceof MemoryCompactionError) {\n    const mergedUsage =\n      error.usage === undefined\n        ? usage\n        : isEmptyUsage(usage)\n          ? error.usage\n          : Usage.add(usage, error.usage);\n    if (mergedUsage === error.usage || (error.usage === undefined && isEmptyUsage(usage))) {\n      return error;\n    }\n    return new MemoryCompactionError(error.message, {\n      cause: error,\n      usage: mergedUsage,\n    });\n  }\n  return new MemoryCompactionError(\"Memory compactor failed.\", {\n    cause: error,\n    usage,\n  });\n}\n\nfunction isEmptyUsage(usage: ReturnType<typeof Usage.empty>): boolean {\n  return (\n    usage.inputTokens === 0 &&\n    usage.outputTokens === 0 &&\n    usage.totalTokens === 0 &&\n    usage.cachedInputTokens === 0 &&\n    usage.cacheCreationInputTokens === 0\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;;;AA0BO,IAAM,iBAAN,MAAqB;AAAA,EAC1B,YACmB,OACA,aACA,gBACjB;AAHiB;AACA;AACA;AAAA,EAChB;AAAA,EAHgB;AAAA,EACA;AAAA,EACA;AAAA,EAGnB,eAA6C;AAC3C,WAAO,KAAK,OAAO,GAAG;AAAA,EACxB;AAAA,EAEA,oBAAoB,aAA2C;AAC7D,WAAO,KAAK,aAAa,MAAM,SAAS,CAAC,GAAG,WAAW,IAAI,CAAC;AAAA,EAC9D;AAAA,EAEA,MAAM,eACJ,OACA,kBACA,aAC4B;AAC5B,UAAM,SAAS,KAAK,OAAO;AAC3B,QAAI,WAAW,UAAa,KAAK,gBAAgB,QAAW;AAC1D,aAAO;AAAA,QACL,SAAS,KAAK;AAAA,QACd,OAAO,MAAM,MAAM;AAAA,MACrB;AAAA,IACF;AAEA,UAAM,cAAc,MAAM,KAAK;AAAA,MAC7B;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AACA,UAAM,gBAAgB,YAAY;AAClC,UAAM,cAAc,CAAC,GAAG,eAAe,GAAG,KAAK,cAAc;AAC7D,WAAO;AAAA,MACL,GAAG;AAAA,MACH,SAAS;AAAA,IACX;AAAA,EACF;AAAA,EAEA,MAAM,QACJ,OACA,aACiC;AACjC,UAAM,SAAS,KAAK,OAAO;AAC3B,UAAM,QAAQ,KAAK;AACnB,QAAI,WAAW,UAAa,UAAU,QAAW;AAC/C,YAAM,IAAI,UAAU,oEAAoE;AAAA,IAC1F;AACA,QAAI,OAAO,eAAe,UAAa,OAAO,MAAM,eAAe,QAAW;AAC5E,YAAM,IAAI,UAAU,mEAAmE;AAAA,IACzF;AACA,UAAM,cAAc,MAAM,KAAK,qBAAqB,QAAQ,OAAO,CAAC,GAAG,aAAa,IAAI;AACxF,QAAI,YAAY,eAAe,QAAW;AACxC,aAAO,EAAE,MAAM,aAAa,GAAG,YAAY,WAAW;AAAA,IACxD;AACA,WAAO;AAAA,MACL,MAAM;AAAA,MACN,QAAQ;AAAA,MACR,sBAAsB,YAAY,QAAQ;AAAA,MAC1C,oBACE,YAAY,sBACX,MAAM,YAAY,OAAO,WAAW,cAAc,YAAY,OAAO;AAAA,IAC1E;AAAA,EACF;AAAA,EAEA,MAAM,oBAAoB,OAAe,UAAwC;AAC/E,UAAM,SAAS,KAAK,OAAO;AAC3B,QACE,WAAW,UACX,KAAK,gBAAgB,UACrB,OAAO,eAAe,aACtB,SAAS,WAAW,GACpB;AACA;AAAA,IACF;AAEA,UAAM,OAAO,MAAM,OAAO;AAAA,MACxB,OAAO,KAAK;AAAA,MACZ;AAAA,MACA,MAAM;AAAA,MACN;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,MAAM,eACJ,OACA,MACA,UACA,qBACe;AACf,UAAM,SAAS,KAAK,OAAO;AAC3B,QAAI,WAAW,UAAa,KAAK,gBAAgB,UAAa,SAAS,WAAW,GAAG;AACnF;AAAA,IACF;AACA,QAAI,OAAO,eAAe,WAAW;AACnC,YAAM,OAAO,MAAM,OAAO;AAAA,QACxB,OAAO,KAAK;AAAA,QACZ;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH,WAAW,OAAO,eAAe,QAAQ;AACvC,0BAAoB,KAAK,GAAG,QAAQ;AAAA,IACtC;AAAA,EACF;AAAA,EAEA,MAAM,oBACJ,OACA,MACA,qBACe;AACf,UAAM,SAAS,KAAK,OAAO;AAC3B,QACE,WAAW,UACX,KAAK,gBAAgB,UACrB,OAAO,eAAe,UACtB,oBAAoB,WAAW,GAC/B;AACA;AAAA,IACF;AACA,UAAM,OAAO,MAAM,OAAO;AAAA,MACxB,OAAO,KAAK;AAAA,MACZ;AAAA,MACA;AAAA,MACA,UAAU,CAAC,GAAG,mBAAmB;AAAA,IACnC,CAAC;AACD,wBAAoB,SAAS;AAAA,EAC/B;AAAA,EAEA,MAAM,mBACJ,OACA,MACA,aACA,qBACe;AACf,UAAM,KAAK,oBAAoB,OAAO,MAAM,mBAAmB;AAC/D,UAAM,SAAS,KAAK,OAAO;AAC3B,QAAI,WAAW,UAAa,KAAK,gBAAgB,UAAa,OAAO,eAAe,OAAO;AACzF;AAAA,IACF;AACA,UAAM,OAAO,MAAM,OAAO;AAAA,MACxB,OAAO,KAAK;AAAA,MACZ;AAAA,MACA;AAAA,MACA,UAAU,CAAC,GAAG,WAAW;AAAA,IAC3B,CAAC;AAAA,EACH;AAAA,EAEA,MAAM,YAAY,OAAe,OAAgB,aAA2C;AAC1F,UAAM,SAAS,KAAK,OAAO;AAC3B,QAAI,WAAW,UAAa,KAAK,gBAAgB,QAAW;AAC1D;AAAA,IACF;AACA,UAAM,OAAO,MAAM,cAAc;AAAA,MAC/B,OAAO,KAAK;AAAA,MACZ;AAAA,MACA;AAAA,MACA,UAAU,CAAC,GAAG,WAAW;AAAA,IAC3B,CAAC;AAAA,EACH;AAAA,EAEQ,SAAkC;AACxC,WAAO,KAAK,gBAAgB,SAAY,SAAY,KAAK,MAAM;AAAA,EACjE;AAAA,EAEA,MAAc,qBACZ,QACA,OACA,kBACA,aACA,QAAQ,OACoB;AAC5B,UAAM,QAAQ,KAAK;AACnB,QAAI,UAAU,QAAW;AACvB,aAAO,EAAE,SAAS,CAAC,GAAG,OAAO,MAAM,MAAM,EAAE;AAAA,IAC7C;AACA,UAAM,UAAU,OAAO;AACvB,UAAM,aAAa,OAAO,MAAM;AAChC,QAAI,YAAY,UAAa,eAAe,QAAW;AACrD,aAAO;AAAA,QACL,SAAS,MAAM,OAAO,MAAM,KAAK,EAAE,MAAM,CAAC;AAAA,QAC1C,OAAO,MAAM,MAAM;AAAA,MACrB;AAAA,IACF;AAEA,QAAI,QAAQ,MAAM,MAAM;AACxB,UAAM,cAAc,QAAQ,oBAAoB,QAAQ,IAAI,QAAQ,gBAAgB;AACpF,aAAS,UAAU,GAAG,WAAW,aAAa,WAAW,GAAG;AAC1D,qBAAe,WAAW;AAC1B,YAAM,WAAW,MAAM,WAAW,SAAS,EAAE,MAAM,CAAC;AACpD,qBAAe,WAAW;AAC1B,YAAM,YAAY,MAAM;AAAA,QACtB,SAAS;AAAA,QACT;AAAA,QACA,QAAQ,QAAQ;AAAA,QAChB,QAAQ,UAAU;AAAA,QAClB,QAAQ;AAAA,QACR;AAAA,MACF;AACA,qBAAe,WAAW;AAC1B,YAAM,wBAAwB,UAAU;AACxC,UAAI,0BAA0B,GAAG;AAC/B,eAAO;AAAA,UACL,SAAS,SAAS;AAAA,UAClB;AAAA,UACA,oBAAoB,UAAU;AAAA,QAChC;AAAA,MACF;AAEA,YAAM,SAAS,SAAS,SAAS,MAAM,GAAG,qBAAqB;AAC/D,UAAI;AACJ,UAAI;AACF,iBAAS,MAAM,QAAQ,UAAU;AAAA,UAC/B;AAAA,UACA,UAAU;AAAA,UACV;AAAA,QACF,CAAC;AAAA,MACH,SAAS,OAAO;AACd,cAAM,uBAAuB,OAAO,KAAK;AAAA,MAC3C;AACA,UAAI,OAAO,UAAU,QAAW;AAC9B,gBAAQ,MAAM,IAAI,OAAO,OAAO,KAAK;AAAA,MACvC;AACA,UAAI,OAAO,QAAQ,YAAY,UAAU;AACvC,cAAM,IAAI,sBAAsB,kDAAkD;AAAA,UAChF;AAAA,QACF,CAAC;AAAA,MACH;AACA,YAAM,cAAc,OAAO,QAAQ,KAAK;AACxC,UAAI,YAAY,WAAW,GAAG;AAC5B,cAAM,IAAI,sBAAsB,+CAA+C;AAAA,UAC7E;AAAA,QACF,CAAC;AAAA,MACH;AACA,YAAM,UAAU;AAAA,QACd;AAAA,QACA,gCAAgC,MAAM;AAAA,MACxC;AACA,YAAM,WAAW,SAAS,SAAS,MAAM,qBAAqB;AAC9D,UAAI;AACJ,UAAI;AACF,2BAAmB,MAAM,YAAY,QAAQ,cAAc,CAAC,SAAS,GAAG,QAAQ,CAAC;AAAA,MACnF,SAAS,OAAO;AACd,cAAM,uBAAuB,OAAO,KAAK;AAAA,MAC3C;AACA,qBAAe,WAAW;AAC1B,UAAI;AACJ,UAAI;AACF,sBAAc,MAAM,WAAW,cAAc;AAAA,UAC3C;AAAA,UACA,UAAU,SAAS;AAAA,UACnB,cAAc;AAAA,UACd,aAAa;AAAA,UACb,OAAO,qBAAqB,KAAK,IAAI,OAAO;AAAA,QAC9C,CAAC;AAAA,MACH,SAAS,OAAO;AACd,cAAM,IAAI,sBAAsB,gDAAgD;AAAA,UAC9E,OAAO;AAAA,UACP;AAAA,QACF,CAAC;AAAA,MACH;AACA,UAAI,YAAY,WAAW,aAAa;AACtC,eAAO;AAAA,UACL,SAAS,CAAC,SAAS,GAAG,QAAQ;AAAA,UAC9B;AAAA,UACA,YAAY;AAAA,YACV,sBAAsB,SAAS,SAAS;AAAA,YACxC;AAAA,YACA,sBAAsB,SAAS;AAAA,YAC/B,oBAAoB,UAAU;AAAA,YAC9B,qBAAqB,UAAU;AAAA,YAC/B,oBAAoB,UAAU;AAAA,YAC9B;AAAA,YACA,UAAU;AAAA,YACV;AAAA,UACF;AAAA,QACF;AAAA,MACF;AACA,qBAAe,WAAW;AAAA,IAC5B;AAEA,UAAM,IAAI,8BAA8B,aAAa,KAAK;AAAA,EAC5D;AACF;AASA,eAAe,uBACb,UACA,kBACA,aACA,cACA,cACA,OAC8B;AAC9B,QAAM,qBAAqB,MAAM,YAAY,cAAc,QAAQ;AACnE,QAAM,qBACJ,iBAAiB,WAAW,IAAI,IAAI,MAAM,YAAY,cAAc,gBAAgB;AACtF,QAAM,OAAO;AAAA,IACX,uBAAuB;AAAA,IACvB;AAAA,IACA,qBAAqB;AAAA,IACrB,oBAAoB;AAAA,EACtB;AACA,MAAI,CAAC,SAAS,qBAAqB,sBAAsB,aAAa;AACpE,WAAO;AAAA,EACT;AACA,QAAM,qBAAqB,SAAS;AAAA,IAAQ,CAAC,SAAS,UACpD,QAAQ,SAAS,SAAS,CAAC,KAAK,IAAI,CAAC;AAAA,EACvC;AACA,MAAI,mBAAmB,UAAU,GAAG;AAClC,WAAO;AAAA,EACT;AAIA,QAAM,kBAAkB,oBAAI,IAAoB;AAChD,QAAM,YAAY,OAAO,iBAA0C;AACjE,UAAM,SAAS,gBAAgB,IAAI,YAAY;AAC/C,QAAI,WAAW,OAAW,QAAO;AACjC,UAAM,QAAQ,MAAM,YAAY,cAAc,SAAS,MAAM,YAAY,CAAC;AAC1E,oBAAgB,IAAI,cAAc,KAAK;AACvC,WAAO;AAAA,EACT;AACA,MAAI,QAAQ;AACZ,MAAI,QAAQ,mBAAmB,SAAS;AACxC,MAAI,mBAAmB;AACvB,SAAO,SAAS,OAAO;AACrB,UAAM,SAAS,KAAK,OAAO,QAAQ,SAAS,CAAC;AAC7C,UAAM,YAAY,mBAAmB,MAAM,KAAK;AAChD,QAAK,MAAM,UAAU,SAAS,KAAM,cAAc;AAChD,yBAAmB;AACnB,cAAQ,SAAS;AAAA,IACnB,OAAO;AACL,cAAQ,SAAS;AAAA,IACnB;AAAA,EACF;AACA,QAAM,wBAAwB,mBAAmB,gBAAgB,KAAK;AACtE,MAAI,0BAA0B,GAAG;AAC/B,WAAO;AAAA,EACT;AACA,QAAM,sBAAsB,MAAM;AAAA,IAChC;AAAA,IACA,SAAS,MAAM,GAAG,qBAAqB;AAAA,EACzC;AACA,QAAM,qBACJ,gBAAgB,IAAI,qBAAqB,KACxC,MAAM,YAAY,cAAc,SAAS,MAAM,qBAAqB,CAAC;AACxE,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEA,eAAe,YACb,cACA,UACiB;AACjB,MAAI;AACJ,MAAI;AACF,YAAQ,MAAM,aAAa,QAAQ;AAAA,EACrC,SAAS,OAAO;AACd,UAAM,IAAI,sBAAsB,gCAAgC,EAAE,OAAO,MAAM,CAAC;AAAA,EAClF;AACA,MAAI,CAAC,OAAO,cAAc,KAAK,KAAK,QAAQ,GAAG;AAC7C,UAAM,IAAI,sBAAsB,8DAA8D;AAAA,EAChG;AACA,SAAO;AACT;AAEA,SAAS,uBAAuB,OAAgB,OAA8C;AAC5F,MAAI,iBAAiB,uBAAuB;AAC1C,UAAM,cACJ,MAAM,UAAU,SACZ,QACA,aAAa,KAAK,IAChB,MAAM,QACN,MAAM,IAAI,OAAO,MAAM,KAAK;AACpC,QAAI,gBAAgB,MAAM,SAAU,MAAM,UAAU,UAAa,aAAa,KAAK,GAAI;AACrF,aAAO;AAAA,IACT;AACA,WAAO,IAAI,sBAAsB,MAAM,SAAS;AAAA,MAC9C,OAAO;AAAA,MACP,OAAO;AAAA,IACT,CAAC;AAAA,EACH;AACA,SAAO,IAAI,sBAAsB,4BAA4B;AAAA,IAC3D,OAAO;AAAA,IACP;AAAA,EACF,CAAC;AACH;AAEA,SAAS,aAAa,OAAgD;AACpE,SACE,MAAM,gBAAgB,KACtB,MAAM,iBAAiB,KACvB,MAAM,gBAAgB,KACtB,MAAM,sBAAsB,KAC5B,MAAM,6BAA6B;AAEvC;","names":[]}

package/dist/chunk-FAPIZJS2.js

@@ -0,0 +1,54 @@
+// src/hooks/control.ts
+function createHook(hook) {
+  return hook;
+}
+function cancelRun(reason) {
+  return { type: "terminate", reason };
+}
+function skipTool(reason) {
+  return { type: "skip", reason };
+}
+function requestToolApproval(options = {}) {
+  const action = {
+    type: "approval_request"
+  };
+  if (options.reason !== void 0) {
+    action.reason = options.reason;
+  }
+  if (options.rejectMessage !== void 0) {
+    action.rejectMessage = options.rejectMessage;
+  }
+  return action;
+}
+var runControl = {
+  continue() {
+    return { type: "continue" };
+  },
+  cancel(reason) {
+    return cancelRun(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,
+  cancelRun,
+  skipTool,
+  requestToolApproval,
+  runControl,
+  toolCallControl
+};
+//# sourceMappingURL=chunk-FAPIZJS2.js.map

package/dist/chunk-FAPIZJS2.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/hooks/control.ts"],"sourcesContent":["import type {\n  AgentHook,\n  HookAction,\n  RunControl,\n  ToolApprovalRequestOptions,\n  ToolCallControl,\n  ToolCallHookAction,\n} from \"./types\";\n\nexport function createHook<RawResponse = unknown>(\n  hook: AgentHook<RawResponse>,\n): AgentHook<RawResponse> {\n  return hook;\n}\n\nexport function cancelRun(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  const action: ToolCallHookAction = {\n    type: \"approval_request\" as const,\n  };\n  if (options.reason !== undefined) {\n    action.reason = options.reason;\n  }\n  if (options.rejectMessage !== undefined) {\n    action.rejectMessage = options.rejectMessage;\n  }\n  return action;\n}\n\nexport const runControl: RunControl = {\n  continue() {\n    return { type: \"continue\" };\n  },\n  cancel(reason: string) {\n    return cancelRun(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":";AASO,SAAS,WACd,MACwB;AACxB,SAAO;AACT;AAEO,SAAS,UAAU,QAA4B;AACpD,SAAO,EAAE,MAAM,aAAa,OAAO;AACrC;AAEO,SAAS,SAAS,QAAoC;AAC3D,SAAO,EAAE,MAAM,QAAQ,OAAO;AAChC;AAEO,SAAS,oBAAoB,UAAsC,CAAC,GAAuB;AAChG,QAAM,SAA6B;AAAA,IACjC,MAAM;AAAA,EACR;AACA,MAAI,QAAQ,WAAW,QAAW;AAChC,WAAO,SAAS,QAAQ;AAAA,EAC1B;AACA,MAAI,QAAQ,kBAAkB,QAAW;AACvC,WAAO,gBAAgB,QAAQ;AAAA,EACjC;AACA,SAAO;AACT;AAEO,IAAM,aAAyB;AAAA,EACpC,WAAW;AACT,WAAO,EAAE,MAAM,WAAW;AAAA,EAC5B;AAAA,EACA,OAAO,QAAgB;AACrB,WAAO,UAAU,MAAM;AAAA,EACzB;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-2QZJTG2E.js → chunk-G336L2IE.js}

@@ -1,13 +1,16 @@
 import {
-  InMemoryVectorStore,
-  assertFiniteMinScore,
-  assertPositiveSearchLimit,
-  createTool,
+  InMemoryVectorStore
+} from "./chunk-72LBHKYP.js";
+import {
   retrieveDocuments
-} from "./chunk-XC3LVC4K.js";
+} from "./chunk-HAAGSIZH.js";
 import {
   embedDocuments
-} from "./chunk-K5L7R7XM.js";
+} from "./chunk-HHMLE3TZ.js";
+import {
+  assertFiniteMinScore,
+  assertPositiveSearchLimit
+} from "./chunk-5V533R7F.js";
 
 // src/tool/dynamic-tools.ts
 async function embedTools(options) {
@@ -100,72 +103,9 @@ function defaultToolEmbeddingText(definition) {
   return [definition.name, definition.description, JSON.stringify(definition.parameters)];
 }
 
-// src/tool/middleware.ts
-function createMiddleware(middleware) {
-  return middleware;
-}
-
-// src/tool/question-tool.ts
-import { z } from "zod";
-var questionToolMarker = /* @__PURE__ */ Symbol("anvia.question-tool");
-function createQuestionTool(options) {
-  const tool = createTool({
-    name: options.name,
-    description: options.description,
-    inputSchema: z.object({
-      questions: z.array(
-        z.object({
-          id: z.string().trim().min(1),
-          text: z.string().trim().min(1),
-          choices: z.array(
-            z.object({
-              label: z.string().trim().min(1),
-              value: z.string().trim().min(1)
-            }).strict()
-          ).min(1).optional(),
-          allowCustom: z.boolean().optional()
-        }).strict()
-      ).min(1)
-    }).strict(),
-    execute() {
-      throw new Error("Question tools can only be resolved through an Agent interaction.");
-    }
-  });
-  Object.defineProperty(tool, questionToolMarker, {
-    configurable: false,
-    enumerable: false,
-    value: true,
-    writable: false
-  });
-  return tool;
-}
-function isQuestionTool(tool) {
-  return typeof tool === "object" && tool !== null && questionToolMarker in tool;
-}
-
-// src/tool/think-tool.ts
-import { z as z2 } from "zod";
-var defaultThinkToolDescription = "Use this tool to record a thought while reasoning through a complex task. It does not retrieve information, store memory, or change external state.";
-var thinkToolInput = z2.object({
-  thought: z2.string().describe("A thought to record while reasoning through a task.")
-});
-function createThinkTool(options = {}) {
-  return createTool({
-    name: options.name ?? "think",
-    description: options.description ?? defaultThinkToolDescription,
-    inputSchema: thinkToolInput,
-    outputSchema: z2.string(),
-    execute: (args) => args.thought
-  });
-}
-
 export {
-  createQuestionTool,
-  isQuestionTool,
   embedTools,
   createToolIndex,
-  isToolIndex,
-  createMiddleware,
-  createThinkTool
+  isToolIndex
 };
-//# sourceMappingURL=chunk-2QZJTG2E.js.map
+//# sourceMappingURL=chunk-G336L2IE.js.map

package/dist/chunk-G336L2IE.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/tool/dynamic-tools.ts"],"sourcesContent":["import type { ToolDefinition } from \"../completion\";\nimport type { EmbeddedDocument, EmbeddingModel, VectorMetadata } from \"../embeddings\";\nimport { embedDocuments } from \"../embeddings\";\nimport { assertFiniteMinScore, assertPositiveSearchLimit } from \"../internal/vector-search-options\";\nimport type { RetrySetting } from \"../retry\";\nimport {\n  InMemoryVectorStore,\n  retrieveDocuments,\n  type VectorFilter,\n  type VectorInspectPage,\n  type VectorInspectRequest,\n  type VectorSearchResult,\n} from \"../vector-store\";\nimport type { AnyTool } from \"./tool\";\n\nexport type ToolSearchDocument<Metadata extends VectorMetadata = VectorMetadata> = {\n  toolName: string;\n  definition: ToolDefinition;\n  text: string;\n  metadata?: Metadata | undefined;\n};\n\nexport type EmbedToolsOptions<Metadata extends VectorMetadata = VectorMetadata> = {\n  model: EmbeddingModel;\n  tools: readonly AnyTool[];\n  content?: ((tool: AnyTool, definition: ToolDefinition) => string | string[]) | undefined;\n  metadata?: ((tool: AnyTool, definition: ToolDefinition) => Metadata | undefined) | undefined;\n  concurrency?: number | undefined;\n  retries?: RetrySetting | undefined;\n  abortSignal?: AbortSignal | undefined;\n};\n\nexport type CreateToolIndexOptions<Metadata extends VectorMetadata = VectorMetadata> =\n  EmbedToolsOptions<Metadata> & {\n    topK: number;\n    minScore?: number | undefined;\n    filter?: VectorFilter | undefined;\n  };\n\nexport interface ToolIndex<Metadata extends VectorMetadata = VectorMetadata> {\n  readonly kind: \"tool-index\";\n  readonly tools: readonly AnyTool[];\n  readonly topK: number;\n  readonly minScore?: number | undefined;\n  readonly filter?: VectorFilter | undefined;\n  search(options: {\n    query: string;\n    abortSignal?: AbortSignal | undefined;\n  }): Promise<Array<VectorSearchResult<ToolSearchDocument<Metadata>, Metadata>>>;\n  inspect?(\n    request: VectorInspectRequest,\n  ): Promise<VectorInspectPage<ToolSearchDocument<Metadata>, Metadata>>;\n}\n\nexport async function embedTools<Metadata extends VectorMetadata = VectorMetadata>(\n  options: EmbedToolsOptions<Metadata>,\n): Promise<{\n  documents: Array<EmbeddedDocument<ToolSearchDocument<Metadata>, Metadata>>;\n}> {\n  const toolList = dedupeTools(options.tools);\n  const definitions = await Promise.all(\n    toolList.map(async (tool) => ({ tool, definition: await tool.definition(\"\") })),\n  );\n  const documents = definitions.map(({ tool, definition }) => {\n    const content = options.content?.(tool, definition) ?? defaultToolEmbeddingText(definition);\n    const texts = Array.isArray(content) ? content : [content];\n    const metadata = options.metadata?.(tool, definition);\n    let document: ToolSearchDocument<Metadata> = {\n      toolName: tool.name,\n      definition,\n      text: texts.join(\"\\n\"),\n    };\n    if (metadata !== undefined) document = { ...document, metadata };\n    return { tool, document, texts, metadata };\n  });\n\n  const embedded = await embedDocuments({\n    model: options.model,\n    documents,\n    id: (item) => item.tool.name,\n    content: (item) => item.texts,\n    metadata: (item) => item.metadata,\n    concurrency: options.concurrency,\n    retries: options.retries,\n    abortSignal: options.abortSignal,\n  });\n  return {\n    documents: embedded.documents.map((item) => ({\n      ...item,\n      document: item.document.document,\n    })),\n  };\n}\n\nexport async function createToolIndex<Metadata extends VectorMetadata = VectorMetadata>(\n  options: CreateToolIndexOptions<Metadata>,\n): Promise<ToolIndex<Metadata>> {\n  assertPositiveSearchLimit(options.topK);\n  assertFiniteMinScore(options.minScore);\n  const tools = dedupeTools(options.tools);\n  const { documents } = await embedTools({ ...options, tools });\n  const store = InMemoryVectorStore.fromDocuments({ documents });\n  return new ToolSearchIndex(store, options.model, tools, options);\n}\n\nexport function isToolIndex(value: unknown): value is ToolIndex {\n  if (typeof value !== \"object\" || value === null) return false;\n  const candidate = value as {\n    kind?: unknown;\n    tools?: unknown;\n    topK?: unknown;\n    search?: unknown;\n  };\n  return (\n    candidate.kind === \"tool-index\" &&\n    Array.isArray(candidate.tools) &&\n    typeof candidate.topK === \"number\" &&\n    typeof candidate.search === \"function\"\n  );\n}\n\nclass ToolSearchIndex<Metadata extends VectorMetadata> implements ToolIndex<Metadata> {\n  readonly kind = \"tool-index\" as const;\n  readonly tools: readonly AnyTool[];\n  readonly topK: number;\n  readonly minScore: number | undefined;\n  readonly filter: VectorFilter | undefined;\n\n  constructor(\n    private readonly store: InMemoryVectorStore<ToolSearchDocument<Metadata>, Metadata>,\n    private readonly model: EmbeddingModel,\n    tools: readonly AnyTool[],\n    private readonly options: CreateToolIndexOptions<Metadata>,\n  ) {\n    this.tools = Object.freeze([...tools]);\n    this.topK = options.topK;\n    this.minScore = options.minScore;\n    this.filter = options.filter;\n  }\n\n  search(options: {\n    query: string;\n    abortSignal?: AbortSignal | undefined;\n  }): Promise<Array<VectorSearchResult<ToolSearchDocument<Metadata>, Metadata>>> {\n    return retrieveDocuments({\n      store: this.store,\n      model: this.model,\n      query: options.query,\n      topK: this.topK,\n      minScore: this.minScore,\n      filter: this.filter,\n      retries: this.options.retries,\n      abortSignal: options.abortSignal,\n    });\n  }\n\n  inspect(\n    request: VectorInspectRequest,\n  ): Promise<VectorInspectPage<ToolSearchDocument<Metadata>, Metadata>> {\n    return this.store.inspect(request);\n  }\n}\n\nfunction dedupeTools(tools: readonly AnyTool[]): AnyTool[] {\n  const byName = new Map<string, AnyTool>();\n  for (const tool of tools) byName.set(tool.name, tool);\n  return [...byName.values()];\n}\n\nfunction defaultToolEmbeddingText(definition: ToolDefinition): string[] {\n  return [definition.name, definition.description, JSON.stringify(definition.parameters)];\n}\n"],"mappings":";;;;;;;;;;;;;;;AAsDA,eAAsB,WACpB,SAGC;AACD,QAAM,WAAW,YAAY,QAAQ,KAAK;AAC1C,QAAM,cAAc,MAAM,QAAQ;AAAA,IAChC,SAAS,IAAI,OAAO,UAAU,EAAE,MAAM,YAAY,MAAM,KAAK,WAAW,EAAE,EAAE,EAAE;AAAA,EAChF;AACA,QAAM,YAAY,YAAY,IAAI,CAAC,EAAE,MAAM,WAAW,MAAM;AAC1D,UAAM,UAAU,QAAQ,UAAU,MAAM,UAAU,KAAK,yBAAyB,UAAU;AAC1F,UAAM,QAAQ,MAAM,QAAQ,OAAO,IAAI,UAAU,CAAC,OAAO;AACzD,UAAM,WAAW,QAAQ,WAAW,MAAM,UAAU;AACpD,QAAI,WAAyC;AAAA,MAC3C,UAAU,KAAK;AAAA,MACf;AAAA,MACA,MAAM,MAAM,KAAK,IAAI;AAAA,IACvB;AACA,QAAI,aAAa,OAAW,YAAW,EAAE,GAAG,UAAU,SAAS;AAC/D,WAAO,EAAE,MAAM,UAAU,OAAO,SAAS;AAAA,EAC3C,CAAC;AAED,QAAM,WAAW,MAAM,eAAe;AAAA,IACpC,OAAO,QAAQ;AAAA,IACf;AAAA,IACA,IAAI,CAAC,SAAS,KAAK,KAAK;AAAA,IACxB,SAAS,CAAC,SAAS,KAAK;AAAA,IACxB,UAAU,CAAC,SAAS,KAAK;AAAA,IACzB,aAAa,QAAQ;AAAA,IACrB,SAAS,QAAQ;AAAA,IACjB,aAAa,QAAQ;AAAA,EACvB,CAAC;AACD,SAAO;AAAA,IACL,WAAW,SAAS,UAAU,IAAI,CAAC,UAAU;AAAA,MAC3C,GAAG;AAAA,MACH,UAAU,KAAK,SAAS;AAAA,IAC1B,EAAE;AAAA,EACJ;AACF;AAEA,eAAsB,gBACpB,SAC8B;AAC9B,4BAA0B,QAAQ,IAAI;AACtC,uBAAqB,QAAQ,QAAQ;AACrC,QAAM,QAAQ,YAAY,QAAQ,KAAK;AACvC,QAAM,EAAE,UAAU,IAAI,MAAM,WAAW,EAAE,GAAG,SAAS,MAAM,CAAC;AAC5D,QAAM,QAAQ,oBAAoB,cAAc,EAAE,UAAU,CAAC;AAC7D,SAAO,IAAI,gBAAgB,OAAO,QAAQ,OAAO,OAAO,OAAO;AACjE;AAEO,SAAS,YAAY,OAAoC;AAC9D,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAM,YAAY;AAMlB,SACE,UAAU,SAAS,gBACnB,MAAM,QAAQ,UAAU,KAAK,KAC7B,OAAO,UAAU,SAAS,YAC1B,OAAO,UAAU,WAAW;AAEhC;AAEA,IAAM,kBAAN,MAAsF;AAAA,EAOpF,YACmB,OACA,OACjB,OACiB,SACjB;AAJiB;AACA;AAEA;AAEjB,SAAK,QAAQ,OAAO,OAAO,CAAC,GAAG,KAAK,CAAC;AACrC,SAAK,OAAO,QAAQ;AACpB,SAAK,WAAW,QAAQ;AACxB,SAAK,SAAS,QAAQ;AAAA,EACxB;AAAA,EATmB;AAAA,EACA;AAAA,EAEA;AAAA,EAVV,OAAO;AAAA,EACP;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAcT,OAAO,SAGwE;AAC7E,WAAO,kBAAkB;AAAA,MACvB,OAAO,KAAK;AAAA,MACZ,OAAO,KAAK;AAAA,MACZ,OAAO,QAAQ;AAAA,MACf,MAAM,KAAK;AAAA,MACX,UAAU,KAAK;AAAA,MACf,QAAQ,KAAK;AAAA,MACb,SAAS,KAAK,QAAQ;AAAA,MACtB,aAAa,QAAQ;AAAA,IACvB,CAAC;AAAA,EACH;AAAA,EAEA,QACE,SACoE;AACpE,WAAO,KAAK,MAAM,QAAQ,OAAO;AAAA,EACnC;AACF;AAEA,SAAS,YAAY,OAAsC;AACzD,QAAM,SAAS,oBAAI,IAAqB;AACxC,aAAW,QAAQ,MAAO,QAAO,IAAI,KAAK,MAAM,IAAI;AACpD,SAAO,CAAC,GAAG,OAAO,OAAO,CAAC;AAC5B;AAEA,SAAS,yBAAyB,YAAsC;AACtE,SAAO,CAAC,WAAW,MAAM,WAAW,aAAa,KAAK,UAAU,WAAW,UAAU,CAAC;AACxF;","names":[]}

package/dist/chunk-GDJNOZ47.js

@@ -0,0 +1 @@
+//# sourceMappingURL=chunk-GDJNOZ47.js.map

package/dist/chunk-GDJNOZ47.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/chunk-GDY2Z2SZ.js

@@ -0,0 +1,83 @@
+import {
+  getAgentToolState
+} from "./chunk-A7BXVZXZ.js";
+import {
+  retrieveDocuments
+} from "./chunk-HAAGSIZH.js";
+import {
+  isVectorContext
+} from "./chunk-ZHC5CD6J.js";
+
+// src/internal/agent-runtime/retrieval.ts
+async function fetchContextDocuments(agent, ragText, abortSignal) {
+  const documents = [];
+  for (const input of agent.context) {
+    if (!isVectorContext(input)) {
+      documents.push(input);
+      continue;
+    }
+    if (ragText === void 0 || ragText.length === 0) continue;
+    const request = {
+      query: ragText,
+      topK: input.topK,
+      minScore: input.minScore,
+      filter: input.filter,
+      retries: input.retries,
+      abortSignal
+    };
+    const results = "models" in input && input.models !== void 0 ? await retrieveDocuments({
+      ...request,
+      store: input.store,
+      models: input.models,
+      fusion: input.fusion
+    }) : await retrieveDocuments({ ...request, store: input.store, model: input.model });
+    for (const result of results) {
+      const formatted = input.format?.(result);
+      if (formatted !== void 0) {
+        documents.push(formatted);
+        continue;
+      }
+      const metadata = formatMetadata(result.metadata);
+      const document = {
+        id: result.id,
+        text: typeof result.document === "string" ? result.document : JSON.stringify(result.document, null, 2)
+      };
+      if (metadata !== void 0) document.additionalProps = metadata;
+      documents.push(document);
+    }
+  }
+  return documents;
+}
+async function fetchToolDefinitions(agent, ragText, abortSignal) {
+  const state = getAgentToolState(agent);
+  const staticDefinitions = await Promise.all(
+    state.staticTools.map((tool) => tool.definition(ragText ?? ""))
+  );
+  if (ragText === void 0 || ragText.length === 0 || state.toolIndexes.length === 0) {
+    return staticDefinitions;
+  }
+  const definitions = [...staticDefinitions];
+  const names = new Set(staticDefinitions.map((definition) => definition.name));
+  for (const index of state.toolIndexes) {
+    const results = await index.search({ query: ragText, abortSignal });
+    for (const result of results) {
+      const toolName = result.document.toolName;
+      if (names.has(toolName)) continue;
+      const tool = index.tools.find((candidate) => candidate.name === toolName);
+      if (tool === void 0) continue;
+      names.add(toolName);
+      definitions.push(await tool.definition(ragText));
+    }
+  }
+  return definitions;
+}
+function formatMetadata(metadata) {
+  if (metadata === void 0) return void 0;
+  return Object.fromEntries(Object.entries(metadata).map(([key, value]) => [key, String(value)]));
+}
+
+export {
+  fetchContextDocuments,
+  fetchToolDefinitions
+};
+//# sourceMappingURL=chunk-GDY2Z2SZ.js.map

package/dist/chunk-GDY2Z2SZ.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/internal/agent-runtime/retrieval.ts"],"sourcesContent":["import type { Agent } from \"../../agent/agent\";\nimport { getAgentToolState } from \"../../agent/tool-state\";\nimport { isVectorContext } from \"../../agent/vector-context\";\nimport type { CompletionModel, Document, ToolDefinition } from \"../../completion/index\";\nimport { retrieveDocuments } from \"../../vector-store\";\n\nexport async function fetchContextDocuments<Output, M extends CompletionModel, ContextDocument>(\n  agent: Agent<Output, M, ContextDocument>,\n  ragText: string | undefined,\n  abortSignal?: AbortSignal | undefined,\n): Promise<Document[]> {\n  const documents: Document[] = [];\n  for (const input of agent.context) {\n    if (!isVectorContext(input)) {\n      documents.push(input);\n      continue;\n    }\n    if (ragText === undefined || ragText.length === 0) continue;\n    const request = {\n      query: ragText,\n      topK: input.topK,\n      minScore: input.minScore,\n      filter: input.filter,\n      retries: input.retries,\n      abortSignal,\n    };\n    const results =\n      \"models\" in input && input.models !== undefined\n        ? await retrieveDocuments({\n            ...request,\n            store: input.store,\n            models: input.models,\n            fusion: input.fusion,\n          })\n        : await retrieveDocuments({ ...request, store: input.store, model: input.model });\n    for (const result of results) {\n      const formatted = input.format?.(result);\n      if (formatted !== undefined) {\n        documents.push(formatted);\n        continue;\n      }\n      const metadata = formatMetadata(result.metadata);\n      const document: Document = {\n        id: result.id,\n        text:\n          typeof result.document === \"string\"\n            ? result.document\n            : JSON.stringify(result.document, null, 2),\n      };\n      if (metadata !== undefined) document.additionalProps = metadata;\n      documents.push(document);\n    }\n  }\n  return documents;\n}\n\nexport async function fetchToolDefinitions<Output, M extends CompletionModel, ContextDocument>(\n  agent: Agent<Output, M, ContextDocument>,\n  ragText: string | undefined,\n  abortSignal?: AbortSignal | undefined,\n): Promise<ToolDefinition[]> {\n  const state = getAgentToolState(agent);\n  const staticDefinitions = await Promise.all(\n    state.staticTools.map((tool) => tool.definition(ragText ?? \"\")),\n  );\n  if (ragText === undefined || ragText.length === 0 || state.toolIndexes.length === 0) {\n    return staticDefinitions;\n  }\n\n  const definitions = [...staticDefinitions];\n  const names = new Set(staticDefinitions.map((definition) => definition.name));\n  for (const index of state.toolIndexes) {\n    const results = await index.search({ query: ragText, abortSignal });\n    for (const result of results) {\n      const toolName = result.document.toolName;\n      if (names.has(toolName)) continue;\n      const tool = index.tools.find((candidate) => candidate.name === toolName);\n      if (tool === undefined) continue;\n      names.add(toolName);\n      definitions.push(await tool.definition(ragText));\n    }\n  }\n  return definitions;\n}\n\nfunction formatMetadata(\n  metadata: Record<string, unknown> | undefined,\n): Record<string, string> | undefined {\n  if (metadata === undefined) return undefined;\n  return Object.fromEntries(Object.entries(metadata).map(([key, value]) => [key, String(value)]));\n}\n"],"mappings":";;;;;;;;;;;AAMA,eAAsB,sBACpB,OACA,SACA,aACqB;AACrB,QAAM,YAAwB,CAAC;AAC/B,aAAW,SAAS,MAAM,SAAS;AACjC,QAAI,CAAC,gBAAgB,KAAK,GAAG;AAC3B,gBAAU,KAAK,KAAK;AACpB;AAAA,IACF;AACA,QAAI,YAAY,UAAa,QAAQ,WAAW,EAAG;AACnD,UAAM,UAAU;AAAA,MACd,OAAO;AAAA,MACP,MAAM,MAAM;AAAA,MACZ,UAAU,MAAM;AAAA,MAChB,QAAQ,MAAM;AAAA,MACd,SAAS,MAAM;AAAA,MACf;AAAA,IACF;AACA,UAAM,UACJ,YAAY,SAAS,MAAM,WAAW,SAClC,MAAM,kBAAkB;AAAA,MACtB,GAAG;AAAA,MACH,OAAO,MAAM;AAAA,MACb,QAAQ,MAAM;AAAA,MACd,QAAQ,MAAM;AAAA,IAChB,CAAC,IACD,MAAM,kBAAkB,EAAE,GAAG,SAAS,OAAO,MAAM,OAAO,OAAO,MAAM,MAAM,CAAC;AACpF,eAAW,UAAU,SAAS;AAC5B,YAAM,YAAY,MAAM,SAAS,MAAM;AACvC,UAAI,cAAc,QAAW;AAC3B,kBAAU,KAAK,SAAS;AACxB;AAAA,MACF;AACA,YAAM,WAAW,eAAe,OAAO,QAAQ;AAC/C,YAAM,WAAqB;AAAA,QACzB,IAAI,OAAO;AAAA,QACX,MACE,OAAO,OAAO,aAAa,WACvB,OAAO,WACP,KAAK,UAAU,OAAO,UAAU,MAAM,CAAC;AAAA,MAC/C;AACA,UAAI,aAAa,OAAW,UAAS,kBAAkB;AACvD,gBAAU,KAAK,QAAQ;AAAA,IACzB;AAAA,EACF;AACA,SAAO;AACT;AAEA,eAAsB,qBACpB,OACA,SACA,aAC2B;AAC3B,QAAM,QAAQ,kBAAkB,KAAK;AACrC,QAAM,oBAAoB,MAAM,QAAQ;AAAA,IACtC,MAAM,YAAY,IAAI,CAAC,SAAS,KAAK,WAAW,WAAW,EAAE,CAAC;AAAA,EAChE;AACA,MAAI,YAAY,UAAa,QAAQ,WAAW,KAAK,MAAM,YAAY,WAAW,GAAG;AACnF,WAAO;AAAA,EACT;AAEA,QAAM,cAAc,CAAC,GAAG,iBAAiB;AACzC,QAAM,QAAQ,IAAI,IAAI,kBAAkB,IAAI,CAAC,eAAe,WAAW,IAAI,CAAC;AAC5E,aAAW,SAAS,MAAM,aAAa;AACrC,UAAM,UAAU,MAAM,MAAM,OAAO,EAAE,OAAO,SAAS,YAAY,CAAC;AAClE,eAAW,UAAU,SAAS;AAC5B,YAAM,WAAW,OAAO,SAAS;AACjC,UAAI,MAAM,IAAI,QAAQ,EAAG;AACzB,YAAM,OAAO,MAAM,MAAM,KAAK,CAAC,cAAc,UAAU,SAAS,QAAQ;AACxE,UAAI,SAAS,OAAW;AACxB,YAAM,IAAI,QAAQ;AAClB,kBAAY,KAAK,MAAM,KAAK,WAAW,OAAO,CAAC;AAAA,IACjD;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,eACP,UACoC;AACpC,MAAI,aAAa,OAAW,QAAO;AACnC,SAAO,OAAO,YAAY,OAAO,QAAQ,QAAQ,EAAE,IAAI,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,KAAK,OAAO,KAAK,CAAC,CAAC,CAAC;AAChG;","names":[]}

package/dist/chunk-GLMYZ6OA.js

@@ -0,0 +1,20 @@
+import {
+  toProviderJsonSchema
+} from "./chunk-SQAAVYJG.js";
+
+// src/agent/output-schema.ts
+var providerOutputSchemas = /* @__PURE__ */ new WeakMap();
+function registerAgentProviderOutputSchema(agent, schema) {
+  if (schema !== void 0) {
+    providerOutputSchemas.set(agent, toProviderJsonSchema(schema));
+  }
+}
+function getAgentProviderOutputSchema(agent) {
+  return providerOutputSchemas.get(agent);
+}
+
+export {
+  registerAgentProviderOutputSchema,
+  getAgentProviderOutputSchema
+};
+//# sourceMappingURL=chunk-GLMYZ6OA.js.map

package/dist/chunk-GLMYZ6OA.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/agent/output-schema.ts"],"sourcesContent":["import type { JsonObject } from \"../completion\";\nimport { toProviderJsonSchema, type ZodSchema } from \"../schema/zod-schema\";\n\nconst providerOutputSchemas = new WeakMap<object, JsonObject>();\n\nexport function registerAgentProviderOutputSchema(\n  agent: object,\n  schema: ZodSchema | undefined,\n): void {\n  if (schema !== undefined) {\n    providerOutputSchemas.set(agent, toProviderJsonSchema(schema));\n  }\n}\n\nexport function getAgentProviderOutputSchema(agent: object): JsonObject | undefined {\n  return providerOutputSchemas.get(agent);\n}\n"],"mappings":";;;;;AAGA,IAAM,wBAAwB,oBAAI,QAA4B;AAEvD,SAAS,kCACd,OACA,QACM;AACN,MAAI,WAAW,QAAW;AACxB,0BAAsB,IAAI,OAAO,qBAAqB,MAAM,CAAC;AAAA,EAC/D;AACF;AAEO,SAAS,6BAA6B,OAAuC;AAClF,SAAO,sBAAsB,IAAI,KAAK;AACxC;","names":[]}

package/dist/{chunk-XEVOC433.js → chunk-GPVXCY5I.js}

@@ -179,78 +179,7 @@ function createChunk(text, index, start, end) {
   return { index, text: text.slice(start, end), start, end };
 }
 
-// src/documents/text-document.ts
-function chunkTextDocuments(options) {
-  if (typeof options !== "object" || options === null) {
-    throw new TypeError("chunkTextDocuments options must contain a documents array");
-  }
-  const documents = options.documents;
-  if (!Array.isArray(documents)) {
-    throw new TypeError("chunkTextDocuments options must contain a documents array");
-  }
-  const ids = /* @__PURE__ */ new Set();
-  const chunks = [];
-  for (const document of documents) {
-    validateDocument(document, ids);
-    chunks.push(...chunkDocument(document, options.chunking));
-  }
-  return chunks;
-}
-function validateDocument(document, ids) {
-  if (typeof document !== "object" || document === null) {
-    throw new TypeError("Text documents must be objects");
-  }
-  if (typeof document.id !== "string" || document.id.length === 0) {
-    throw new TypeError("Text document ids must be non-empty strings");
-  }
-  if (ids.has(document.id)) {
-    throw new TypeError(`Duplicate text document id: ${document.id}`);
-  }
-  if (typeof document.text !== "string" || document.text.length === 0) {
-    throw new TypeError(`Text document ${document.id} must contain non-empty text`);
-  }
-  ids.add(document.id);
-}
-function chunkDocument(document, chunking) {
-  if (chunking === void 0 || chunking.strategy === "none") {
-    return [createDocumentChunk(document, 0, document.text)];
-  }
-  const textChunks = chunkDocumentText(document.text, chunking);
-  return textChunks.map((chunk) => createDocumentChunk(document, chunk.index, chunk.text));
-}
-function chunkDocumentText(text, chunking) {
-  const overlap = chunking.overlap ?? 0;
-  if (chunking.strategy === "fixed") {
-    return chunkText({
-      text,
-      strategy: "fixed",
-      maxSize: chunking.maxSize,
-      overlap
-    });
-  }
-  return chunkText({
-    text,
-    strategy: "recursive",
-    maxSize: chunking.maxSize,
-    overlap,
-    separators: chunking.separators
-  });
-}
-function createDocumentChunk(document, index, text) {
-  const chunk = {
-    id: `${document.id}:chunk:${index}`,
-    documentId: document.id,
-    index,
-    text
-  };
-  if (document.metadata === void 0) {
-    return chunk;
-  }
-  return { ...chunk, metadata: document.metadata };
-}
-
 export {
-  chunkText,
-  chunkTextDocuments
+  chunkText
 };
-//# sourceMappingURL=chunk-XEVOC433.js.map
+//# sourceMappingURL=chunk-GPVXCY5I.js.map

package/dist/chunk-GPVXCY5I.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/documents/chunk-text.ts"],"sourcesContent":["export type TextChunk = Readonly<{\n  index: number;\n  text: string;\n  start: number;\n  end: number;\n}>;\n\nexport type ChunkTextOptions =\n  | Readonly<{\n      text: string;\n      strategy: \"fixed\";\n      maxSize: number;\n      overlap?: number;\n    }>\n  | Readonly<{\n      text: string;\n      strategy: \"recursive\";\n      maxSize: number;\n      overlap?: number;\n      separators: readonly string[];\n    }>;\n\ntype TextRange = Readonly<{\n  start: number;\n  end: number;\n}>;\n\nexport function chunkText(options: ChunkTextOptions): readonly TextChunk[] {\n  const validated = validateOptions(options);\n  if (validated.text.length === 0) {\n    return [];\n  }\n\n  if (validated.strategy === \"fixed\") {\n    return chunkFixed(validated.text, validated.maxSize, validated.overlap);\n  }\n\n  const ranges = resolveRecursiveRanges(\n    validated.text,\n    { start: 0, end: validated.text.length },\n    validated.maxSize,\n    validated.separators,\n  );\n  return applyRecursiveOverlap(validated.text, ranges, validated.maxSize, validated.overlap);\n}\n\nfunction validateOptions(options: ChunkTextOptions):\n  | Readonly<{\n      text: string;\n      strategy: \"fixed\";\n      maxSize: number;\n      overlap: number;\n    }>\n  | Readonly<{\n      text: string;\n      strategy: \"recursive\";\n      maxSize: number;\n      overlap: number;\n      separators: readonly string[];\n    }> {\n  if (typeof options !== \"object\" || options === null) {\n    throw new TypeError(\"chunkText options must be an object\");\n  }\n\n  const value = options as unknown as Record<string, unknown>;\n  if (typeof value.text !== \"string\") {\n    throw new TypeError(\"chunkText text must be a string\");\n  }\n  if (!Number.isSafeInteger(value.maxSize) || (value.maxSize as number) <= 0) {\n    throw new RangeError(\"chunkText maxSize must be a positive safe integer\");\n  }\n\n  const maxSize = value.maxSize as number;\n  const overlap = value.overlap === undefined ? 0 : value.overlap;\n  if (!Number.isSafeInteger(overlap) || (overlap as number) < 0 || (overlap as number) >= maxSize) {\n    throw new RangeError(\n      \"chunkText overlap must be a non-negative safe integer smaller than maxSize\",\n    );\n  }\n\n  if (value.strategy === \"fixed\") {\n    if (\"separators\" in value) {\n      throw new TypeError(\"chunkText fixed strategy does not accept separators\");\n    }\n    return {\n      text: value.text,\n      strategy: \"fixed\",\n      maxSize,\n      overlap: overlap as number,\n    };\n  }\n\n  if (value.strategy !== \"recursive\") {\n    throw new TypeError('chunkText strategy must be either \"fixed\" or \"recursive\"');\n  }\n  if (!Array.isArray(value.separators) || value.separators.length === 0) {\n    throw new TypeError(\"chunkText recursive separators must be a non-empty array\");\n  }\n\n  const separators = value.separators as unknown[];\n  const seen = new Set<string>();\n  for (const separator of separators) {\n    if (typeof separator !== \"string\" || separator.length === 0) {\n      throw new TypeError(\"chunkText recursive separators must contain non-empty strings\");\n    }\n    if (seen.has(separator)) {\n      throw new TypeError(\"chunkText recursive separators must not contain duplicates\");\n    }\n    seen.add(separator);\n  }\n\n  return {\n    text: value.text,\n    strategy: \"recursive\",\n    maxSize,\n    overlap: overlap as number,\n    separators: separators as string[],\n  };\n}\n\nfunction chunkFixed(text: string, maxSize: number, overlap: number): readonly TextChunk[] {\n  const chunks: TextChunk[] = [];\n  let start = 0;\n\n  while (start < text.length) {\n    const end = Math.min(start + maxSize, text.length);\n    chunks.push(createChunk(text, chunks.length, start, end));\n    if (end === text.length) {\n      break;\n    }\n    start = end - overlap;\n  }\n\n  return chunks;\n}\n\nfunction resolveRecursiveRanges(\n  text: string,\n  initialRange: TextRange,\n  maxSize: number,\n  separators: readonly string[],\n): readonly TextRange[] {\n  const resolved: TextRange[] = [];\n  const pending: Array<Readonly<{ range: TextRange; separatorIndex: number }>> = [\n    { range: initialRange, separatorIndex: 0 },\n  ];\n\n  while (pending.length > 0) {\n    const current = pending.pop();\n    if (current === undefined) {\n      break;\n    }\n    if (current.range.end - current.range.start <= maxSize) {\n      resolved.push(current.range);\n      continue;\n    }\n\n    let pieces: readonly TextRange[] | undefined;\n    let nextSeparatorIndex = separators.length;\n    for (let index = current.separatorIndex; index < separators.length; index += 1) {\n      const candidate = splitAfterSeparator(text, current.range, separators[index] as string);\n      if (candidate.length > 1) {\n        pieces = candidate;\n        nextSeparatorIndex = index + 1;\n        break;\n      }\n    }\n\n    if (pieces === undefined) {\n      resolved.push(...hardSplit(current.range, maxSize));\n      continue;\n    }\n\n    for (let index = pieces.length - 1; index >= 0; index -= 1) {\n      pending.push({ range: pieces[index] as TextRange, separatorIndex: nextSeparatorIndex });\n    }\n  }\n\n  return resolved;\n}\n\nfunction splitAfterSeparator(text: string, range: TextRange, separator: string): TextRange[] {\n  const ranges: TextRange[] = [];\n  let start = range.start;\n  let searchFrom = range.start;\n\n  while (searchFrom < range.end) {\n    const separatorStart = text.indexOf(separator, searchFrom);\n    if (separatorStart < 0 || separatorStart + separator.length > range.end) {\n      break;\n    }\n    const end = separatorStart + separator.length;\n    if (end > start) {\n      ranges.push({ start, end });\n    }\n    start = end;\n    searchFrom = end;\n  }\n\n  if (start < range.end) {\n    ranges.push({ start, end: range.end });\n  }\n  return ranges;\n}\n\nfunction hardSplit(range: TextRange, maxSize: number): readonly TextRange[] {\n  const ranges: TextRange[] = [];\n  for (let start = range.start; start < range.end; start += maxSize) {\n    ranges.push({ start, end: Math.min(start + maxSize, range.end) });\n  }\n  return ranges;\n}\n\nfunction applyRecursiveOverlap(\n  text: string,\n  ranges: readonly TextRange[],\n  maxSize: number,\n  overlap: number,\n): readonly TextChunk[] {\n  const boundaries = ranges.map((range) => range.end);\n  const chunks: TextChunk[] = [];\n  let start = 0;\n  let previousEnd = 0;\n\n  while (start < text.length) {\n    const limit = Math.min(start + maxSize, text.length);\n    const boundary = findLastBoundaryAtMost(boundaries, limit);\n    const end = boundary !== undefined && boundary > previousEnd ? boundary : limit;\n\n    chunks.push(createChunk(text, chunks.length, start, end));\n    if (end === text.length) {\n      break;\n    }\n    previousEnd = end;\n    start = Math.max(0, end - overlap);\n  }\n\n  return chunks;\n}\n\nfunction findLastBoundaryAtMost(boundaries: readonly number[], limit: number): number | undefined {\n  let low = 0;\n  let high = boundaries.length;\n  while (low < high) {\n    const middle = low + Math.floor((high - low) / 2);\n    if ((boundaries[middle] as number) <= limit) {\n      low = middle + 1;\n    } else {\n      high = middle;\n    }\n  }\n  return boundaries[low - 1];\n}\n\nfunction createChunk(text: string, index: number, start: number, end: number): TextChunk {\n  return { index, text: text.slice(start, end), start, end };\n}\n"],"mappings":";AA2BO,SAAS,UAAU,SAAiD;AACzE,QAAM,YAAY,gBAAgB,OAAO;AACzC,MAAI,UAAU,KAAK,WAAW,GAAG;AAC/B,WAAO,CAAC;AAAA,EACV;AAEA,MAAI,UAAU,aAAa,SAAS;AAClC,WAAO,WAAW,UAAU,MAAM,UAAU,SAAS,UAAU,OAAO;AAAA,EACxE;AAEA,QAAM,SAAS;AAAA,IACb,UAAU;AAAA,IACV,EAAE,OAAO,GAAG,KAAK,UAAU,KAAK,OAAO;AAAA,IACvC,UAAU;AAAA,IACV,UAAU;AAAA,EACZ;AACA,SAAO,sBAAsB,UAAU,MAAM,QAAQ,UAAU,SAAS,UAAU,OAAO;AAC3F;AAEA,SAAS,gBAAgB,SAalB;AACL,MAAI,OAAO,YAAY,YAAY,YAAY,MAAM;AACnD,UAAM,IAAI,UAAU,qCAAqC;AAAA,EAC3D;AAEA,QAAM,QAAQ;AACd,MAAI,OAAO,MAAM,SAAS,UAAU;AAClC,UAAM,IAAI,UAAU,iCAAiC;AAAA,EACvD;AACA,MAAI,CAAC,OAAO,cAAc,MAAM,OAAO,KAAM,MAAM,WAAsB,GAAG;AAC1E,UAAM,IAAI,WAAW,mDAAmD;AAAA,EAC1E;AAEA,QAAM,UAAU,MAAM;AACtB,QAAM,UAAU,MAAM,YAAY,SAAY,IAAI,MAAM;AACxD,MAAI,CAAC,OAAO,cAAc,OAAO,KAAM,UAAqB,KAAM,WAAsB,SAAS;AAC/F,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,MAAI,MAAM,aAAa,SAAS;AAC9B,QAAI,gBAAgB,OAAO;AACzB,YAAM,IAAI,UAAU,qDAAqD;AAAA,IAC3E;AACA,WAAO;AAAA,MACL,MAAM,MAAM;AAAA,MACZ,UAAU;AAAA,MACV;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,MAAI,MAAM,aAAa,aAAa;AAClC,UAAM,IAAI,UAAU,0DAA0D;AAAA,EAChF;AACA,MAAI,CAAC,MAAM,QAAQ,MAAM,UAAU,KAAK,MAAM,WAAW,WAAW,GAAG;AACrE,UAAM,IAAI,UAAU,0DAA0D;AAAA,EAChF;AAEA,QAAM,aAAa,MAAM;AACzB,QAAM,OAAO,oBAAI,IAAY;AAC7B,aAAW,aAAa,YAAY;AAClC,QAAI,OAAO,cAAc,YAAY,UAAU,WAAW,GAAG;AAC3D,YAAM,IAAI,UAAU,+DAA+D;AAAA,IACrF;AACA,QAAI,KAAK,IAAI,SAAS,GAAG;AACvB,YAAM,IAAI,UAAU,4DAA4D;AAAA,IAClF;AACA,SAAK,IAAI,SAAS;AAAA,EACpB;AAEA,SAAO;AAAA,IACL,MAAM,MAAM;AAAA,IACZ,UAAU;AAAA,IACV;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEA,SAAS,WAAW,MAAc,SAAiB,SAAuC;AACxF,QAAM,SAAsB,CAAC;AAC7B,MAAI,QAAQ;AAEZ,SAAO,QAAQ,KAAK,QAAQ;AAC1B,UAAM,MAAM,KAAK,IAAI,QAAQ,SAAS,KAAK,MAAM;AACjD,WAAO,KAAK,YAAY,MAAM,OAAO,QAAQ,OAAO,GAAG,CAAC;AACxD,QAAI,QAAQ,KAAK,QAAQ;AACvB;AAAA,IACF;AACA,YAAQ,MAAM;AAAA,EAChB;AAEA,SAAO;AACT;AAEA,SAAS,uBACP,MACA,cACA,SACA,YACsB;AACtB,QAAM,WAAwB,CAAC;AAC/B,QAAM,UAAyE;AAAA,IAC7E,EAAE,OAAO,cAAc,gBAAgB,EAAE;AAAA,EAC3C;AAEA,SAAO,QAAQ,SAAS,GAAG;AACzB,UAAM,UAAU,QAAQ,IAAI;AAC5B,QAAI,YAAY,QAAW;AACzB;AAAA,IACF;AACA,QAAI,QAAQ,MAAM,MAAM,QAAQ,MAAM,SAAS,SAAS;AACtD,eAAS,KAAK,QAAQ,KAAK;AAC3B;AAAA,IACF;AAEA,QAAI;AACJ,QAAI,qBAAqB,WAAW;AACpC,aAAS,QAAQ,QAAQ,gBAAgB,QAAQ,WAAW,QAAQ,SAAS,GAAG;AAC9E,YAAM,YAAY,oBAAoB,MAAM,QAAQ,OAAO,WAAW,KAAK,CAAW;AACtF,UAAI,UAAU,SAAS,GAAG;AACxB,iBAAS;AACT,6BAAqB,QAAQ;AAC7B;AAAA,MACF;AAAA,IACF;AAEA,QAAI,WAAW,QAAW;AACxB,eAAS,KAAK,GAAG,UAAU,QAAQ,OAAO,OAAO,CAAC;AAClD;AAAA,IACF;AAEA,aAAS,QAAQ,OAAO,SAAS,GAAG,SAAS,GAAG,SAAS,GAAG;AAC1D,cAAQ,KAAK,EAAE,OAAO,OAAO,KAAK,GAAgB,gBAAgB,mBAAmB,CAAC;AAAA,IACxF;AAAA,EACF;AAEA,SAAO;AACT;AAEA,SAAS,oBAAoB,MAAc,OAAkB,WAAgC;AAC3F,QAAM,SAAsB,CAAC;AAC7B,MAAI,QAAQ,MAAM;AAClB,MAAI,aAAa,MAAM;AAEvB,SAAO,aAAa,MAAM,KAAK;AAC7B,UAAM,iBAAiB,KAAK,QAAQ,WAAW,UAAU;AACzD,QAAI,iBAAiB,KAAK,iBAAiB,UAAU,SAAS,MAAM,KAAK;AACvE;AAAA,IACF;AACA,UAAM,MAAM,iBAAiB,UAAU;AACvC,QAAI,MAAM,OAAO;AACf,aAAO,KAAK,EAAE,OAAO,IAAI,CAAC;AAAA,IAC5B;AACA,YAAQ;AACR,iBAAa;AAAA,EACf;AAEA,MAAI,QAAQ,MAAM,KAAK;AACrB,WAAO,KAAK,EAAE,OAAO,KAAK,MAAM,IAAI,CAAC;AAAA,EACvC;AACA,SAAO;AACT;AAEA,SAAS,UAAU,OAAkB,SAAuC;AAC1E,QAAM,SAAsB,CAAC;AAC7B,WAAS,QAAQ,MAAM,OAAO,QAAQ,MAAM,KAAK,SAAS,SAAS;AACjE,WAAO,KAAK,EAAE,OAAO,KAAK,KAAK,IAAI,QAAQ,SAAS,MAAM,GAAG,EAAE,CAAC;AAAA,EAClE;AACA,SAAO;AACT;AAEA,SAAS,sBACP,MACA,QACA,SACA,SACsB;AACtB,QAAM,aAAa,OAAO,IAAI,CAAC,UAAU,MAAM,GAAG;AAClD,QAAM,SAAsB,CAAC;AAC7B,MAAI,QAAQ;AACZ,MAAI,cAAc;AAElB,SAAO,QAAQ,KAAK,QAAQ;AAC1B,UAAM,QAAQ,KAAK,IAAI,QAAQ,SAAS,KAAK,MAAM;AACnD,UAAM,WAAW,uBAAuB,YAAY,KAAK;AACzD,UAAM,MAAM,aAAa,UAAa,WAAW,cAAc,WAAW;AAE1E,WAAO,KAAK,YAAY,MAAM,OAAO,QAAQ,OAAO,GAAG,CAAC;AACxD,QAAI,QAAQ,KAAK,QAAQ;AACvB;AAAA,IACF;AACA,kBAAc;AACd,YAAQ,KAAK,IAAI,GAAG,MAAM,OAAO;AAAA,EACnC;AAEA,SAAO;AACT;AAEA,SAAS,uBAAuB,YAA+B,OAAmC;AAChG,MAAI,MAAM;AACV,MAAI,OAAO,WAAW;AACtB,SAAO,MAAM,MAAM;AACjB,UAAM,SAAS,MAAM,KAAK,OAAO,OAAO,OAAO,CAAC;AAChD,QAAK,WAAW,MAAM,KAAgB,OAAO;AAC3C,YAAM,SAAS;AAAA,IACjB,OAAO;AACL,aAAO;AAAA,IACT;AAAA,EACF;AACA,SAAO,WAAW,MAAM,CAAC;AAC3B;AAEA,SAAS,YAAY,MAAc,OAAe,OAAe,KAAwB;AACvF,SAAO,EAAE,OAAO,MAAM,KAAK,MAAM,OAAO,GAAG,GAAG,OAAO,IAAI;AAC3D;","names":[]}

package/dist/{chunk-KSKST3KP.js → chunk-H3H74RWJ.js}

@@ -83,4 +83,4 @@ function containsError(value, seen) {
 export {
   observerSnapshot
 };
-//# sourceMappingURL=chunk-KSKST3KP.js.map
+//# sourceMappingURL=chunk-H3H74RWJ.js.map

package/dist/chunk-HAAGSIZH.js

@@ -0,0 +1,89 @@
+import {
+  embedSparseQuery,
+  embedText
+} from "./chunk-HHMLE3TZ.js";
+import {
+  assertFiniteMinScore,
+  assertPositiveSearchLimit
+} from "./chunk-5V533R7F.js";
+import {
+  resolveRetryOptions,
+  runWithRetries
+} from "./chunk-POXHJF3H.js";
+import {
+  throwIfAborted
+} from "./chunk-USRKPEQN.js";
+
+// src/vector-store/retrieve.ts
+async function retrieveDocuments(options) {
+  assertPositiveSearchLimit(options.topK);
+  assertFiniteMinScore(options.minScore);
+  throwIfAborted(options.abortSignal);
+  const retries = resolveRetries(options.retries);
+  const search = (operation) => runWithRetries(operation, retries, {
+    streaming: false,
+    abortSignal: options.abortSignal
+  });
+  if ("model" in options && options.model !== void 0) {
+    const { embedding } = await embedText({
+      model: options.model,
+      text: options.query,
+      retries: options.retries,
+      abortSignal: options.abortSignal
+    });
+    return deduplicateResults(
+      await search(
+        () => options.store.search({
+          vector: embedding.vector,
+          topK: options.topK,
+          minScore: options.minScore,
+          filter: options.filter,
+          abortSignal: options.abortSignal
+        })
+      )
+    );
+  }
+  const [{ embedding: dense }, { embedding: sparse }] = await Promise.all([
+    embedText({
+      model: options.models.dense,
+      text: options.query,
+      retries: options.retries,
+      abortSignal: options.abortSignal
+    }),
+    embedSparseQuery({
+      model: options.models.sparse,
+      query: options.query,
+      retries: options.retries,
+      abortSignal: options.abortSignal
+    })
+  ]);
+  return deduplicateResults(
+    await search(
+      () => options.store.searchHybrid({
+        vector: dense.vector,
+        sparseVector: sparse.vector,
+        fusion: options.fusion,
+        topK: options.topK,
+        minScore: options.minScore,
+        filter: options.filter,
+        abortSignal: options.abortSignal
+      })
+    )
+  );
+}
+function deduplicateResults(results) {
+  const bestById = /* @__PURE__ */ new Map();
+  for (const result of results) {
+    const current = bestById.get(result.id);
+    if (current === void 0 || result.score > current.score) bestById.set(result.id, result);
+  }
+  return [...bestById.values()].sort((left, right) => right.score - left.score);
+}
+function resolveRetries(setting) {
+  return setting === void 0 || setting === false ? void 0 : resolveRetryOptions(setting);
+}
+
+export {
+  retrieveDocuments
+};
+//# sourceMappingURL=chunk-HAAGSIZH.js.map

package/dist/chunk-HAAGSIZH.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/vector-store/retrieve.ts"],"sourcesContent":["import { embedSparseQuery, embedText, type VectorMetadata } from \"../embeddings\";\nimport { throwIfAborted } from \"../internal/abort\";\nimport { assertFiniteMinScore, assertPositiveSearchLimit } from \"../internal/vector-search-options\";\nimport { type ResolvedRetryOptions, resolveRetryOptions, runWithRetries } from \"../retry\";\nimport type {\n  RetrieveDocumentsOptions,\n  RetrieveHybridDocumentsOptions,\n  VectorSearchResult,\n} from \"./types\";\n\nexport function retrieveDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n  options: RetrieveDocumentsOptions<T, Metadata>,\n): Promise<Array<VectorSearchResult<T, Metadata>>>;\nexport function retrieveDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n  options: RetrieveHybridDocumentsOptions<T, Metadata>,\n): Promise<Array<VectorSearchResult<T, Metadata>>>;\nexport async function retrieveDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n  options: RetrieveDocumentsOptions<T, Metadata> | RetrieveHybridDocumentsOptions<T, Metadata>,\n): Promise<Array<VectorSearchResult<T, Metadata>>> {\n  assertPositiveSearchLimit(options.topK);\n  assertFiniteMinScore(options.minScore);\n  throwIfAborted(options.abortSignal);\n  const retries = resolveRetries(options.retries);\n  const search = (operation: () => Promise<Array<VectorSearchResult<T, Metadata>>>) =>\n    runWithRetries(operation, retries, {\n      streaming: false,\n      abortSignal: options.abortSignal,\n    });\n\n  if (\"model\" in options && options.model !== undefined) {\n    const { embedding } = await embedText({\n      model: options.model,\n      text: options.query,\n      retries: options.retries,\n      abortSignal: options.abortSignal,\n    });\n    return deduplicateResults(\n      await search(() =>\n        options.store.search({\n          vector: embedding.vector,\n          topK: options.topK,\n          minScore: options.minScore,\n          filter: options.filter,\n          abortSignal: options.abortSignal,\n        }),\n      ),\n    );\n  }\n\n  const [{ embedding: dense }, { embedding: sparse }] = await Promise.all([\n    embedText({\n      model: options.models.dense,\n      text: options.query,\n      retries: options.retries,\n      abortSignal: options.abortSignal,\n    }),\n    embedSparseQuery({\n      model: options.models.sparse,\n      query: options.query,\n      retries: options.retries,\n      abortSignal: options.abortSignal,\n    }),\n  ]);\n  return deduplicateResults(\n    await search(() =>\n      options.store.searchHybrid({\n        vector: dense.vector,\n        sparseVector: sparse.vector,\n        fusion: options.fusion,\n        topK: options.topK,\n        minScore: options.minScore,\n        filter: options.filter,\n        abortSignal: options.abortSignal,\n      }),\n    ),\n  );\n}\n\nfunction deduplicateResults<T, Metadata extends VectorMetadata>(\n  results: Array<VectorSearchResult<T, Metadata>>,\n): Array<VectorSearchResult<T, Metadata>> {\n  const bestById = new Map<string, VectorSearchResult<T, Metadata>>();\n  for (const result of results) {\n    const current = bestById.get(result.id);\n    if (current === undefined || result.score > current.score) bestById.set(result.id, result);\n  }\n  return [...bestById.values()].sort((left, right) => right.score - left.score);\n}\n\nfunction resolveRetries(\n  setting: RetrieveDocumentsOptions<unknown>[\"retries\"],\n): ResolvedRetryOptions | undefined {\n  return setting === undefined || setting === false ? undefined : resolveRetryOptions(setting);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAgBA,eAAsB,kBACpB,SACiD;AACjD,4BAA0B,QAAQ,IAAI;AACtC,uBAAqB,QAAQ,QAAQ;AACrC,iBAAe,QAAQ,WAAW;AAClC,QAAM,UAAU,eAAe,QAAQ,OAAO;AAC9C,QAAM,SAAS,CAAC,cACd,eAAe,WAAW,SAAS;AAAA,IACjC,WAAW;AAAA,IACX,aAAa,QAAQ;AAAA,EACvB,CAAC;AAEH,MAAI,WAAW,WAAW,QAAQ,UAAU,QAAW;AACrD,UAAM,EAAE,UAAU,IAAI,MAAM,UAAU;AAAA,MACpC,OAAO,QAAQ;AAAA,MACf,MAAM,QAAQ;AAAA,MACd,SAAS,QAAQ;AAAA,MACjB,aAAa,QAAQ;AAAA,IACvB,CAAC;AACD,WAAO;AAAA,MACL,MAAM;AAAA,QAAO,MACX,QAAQ,MAAM,OAAO;AAAA,UACnB,QAAQ,UAAU;AAAA,UAClB,MAAM,QAAQ;AAAA,UACd,UAAU,QAAQ;AAAA,UAClB,QAAQ,QAAQ;AAAA,UAChB,aAAa,QAAQ;AAAA,QACvB,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AAEA,QAAM,CAAC,EAAE,WAAW,MAAM,GAAG,EAAE,WAAW,OAAO,CAAC,IAAI,MAAM,QAAQ,IAAI;AAAA,IACtE,UAAU;AAAA,MACR,OAAO,QAAQ,OAAO;AAAA,MACtB,MAAM,QAAQ;AAAA,MACd,SAAS,QAAQ;AAAA,MACjB,aAAa,QAAQ;AAAA,IACvB,CAAC;AAAA,IACD,iBAAiB;AAAA,MACf,OAAO,QAAQ,OAAO;AAAA,MACtB,OAAO,QAAQ;AAAA,MACf,SAAS,QAAQ;AAAA,MACjB,aAAa,QAAQ;AAAA,IACvB,CAAC;AAAA,EACH,CAAC;AACD,SAAO;AAAA,IACL,MAAM;AAAA,MAAO,MACX,QAAQ,MAAM,aAAa;AAAA,QACzB,QAAQ,MAAM;AAAA,QACd,cAAc,OAAO;AAAA,QACrB,QAAQ,QAAQ;AAAA,QAChB,MAAM,QAAQ;AAAA,QACd,UAAU,QAAQ;AAAA,QAClB,QAAQ,QAAQ;AAAA,QAChB,aAAa,QAAQ;AAAA,MACvB,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEA,SAAS,mBACP,SACwC;AACxC,QAAM,WAAW,oBAAI,IAA6C;AAClE,aAAW,UAAU,SAAS;AAC5B,UAAM,UAAU,SAAS,IAAI,OAAO,EAAE;AACtC,QAAI,YAAY,UAAa,OAAO,QAAQ,QAAQ,MAAO,UAAS,IAAI,OAAO,IAAI,MAAM;AAAA,EAC3F;AACA,SAAO,CAAC,GAAG,SAAS,OAAO,CAAC,EAAE,KAAK,CAAC,MAAM,UAAU,MAAM,QAAQ,KAAK,KAAK;AAC9E;AAEA,SAAS,eACP,SACkC;AAClC,SAAO,YAAY,UAAa,YAAY,QAAQ,SAAY,oBAAoB,OAAO;AAC7F;","names":[]}

package/dist/{chunk-K5L7R7XM.js → chunk-HHMLE3TZ.js}

@@ -1,55 +1,13 @@
 import {
   mapWithConcurrency
-} from "./chunk-IZNOP6JG.js";
+} from "./chunk-4FIVU53H.js";
 import {
   resolveRetryOptions,
-  runWithRetries,
+  runWithRetries
+} from "./chunk-POXHJF3H.js";
+import {
   throwIfAborted
-} from "./chunk-3RWESPUG.js";
-
-// 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}`);
-  }
-}
+} from "./chunk-USRKPEQN.js";
 
 // src/embeddings/embed.ts
 async function embedText(options) {
@@ -316,16 +274,10 @@ function validateSparseEmbedding(embedding, index) {
 }
 
 export {
-  dotProduct,
-  cosineSimilarity,
-  angularDistance,
-  euclideanDistance,
-  manhattanDistance,
-  chebyshevDistance,
   embedText,
   embedTexts,
   embedSparseTexts,
   embedSparseQuery,
   embedDocuments
 };
-//# sourceMappingURL=chunk-K5L7R7XM.js.map
+//# sourceMappingURL=chunk-HHMLE3TZ.js.map