@anvia/core 0.26.0 → 1.0.0-rc.2

package/dist/chunk-D37WMSP6.js

@@ -0,0 +1,131 @@
+import {
+  InMemoryVectorStore,
+  assertFiniteMinScore,
+  assertPositiveSearchLimit,
+  createTool,
+  retrieveDocuments
+} from "./chunk-77AIYARB.js";
+import {
+  embedDocuments
+} from "./chunk-FMHLROPQ.js";
+
+// src/tool/dynamic-tools.ts
+async function embedTools(options) {
+  const toolList = dedupeTools(options.tools);
+  const definitions = await Promise.all(
+    toolList.map(async (tool) => ({ tool, definition: await tool.definition("") }))
+  );
+  const documents = definitions.map(({ tool, definition }) => {
+    const content = options.content?.(tool, definition) ?? defaultToolEmbeddingText(definition);
+    const texts = Array.isArray(content) ? content : [content];
+    const metadata = options.metadata?.(tool, definition);
+    const document = {
+      toolName: tool.name,
+      definition,
+      text: texts.join("\n"),
+      ...metadata === void 0 ? {} : { metadata }
+    };
+    return { tool, document, texts, metadata };
+  });
+  const embedded = await embedDocuments({
+    model: options.model,
+    documents,
+    id: (item) => item.tool.name,
+    content: (item) => item.texts,
+    metadata: (item) => item.metadata,
+    concurrency: options.concurrency,
+    retries: options.retries,
+    abortSignal: options.abortSignal
+  });
+  return {
+    documents: embedded.documents.map((item) => ({
+      ...item,
+      document: item.document.document
+    }))
+  };
+}
+async function createToolIndex(options) {
+  assertPositiveSearchLimit(options.topK);
+  assertFiniteMinScore(options.minScore);
+  const tools = dedupeTools(options.tools);
+  const { documents } = await embedTools({ ...options, tools });
+  const store = InMemoryVectorStore.fromDocuments({ documents });
+  return new ToolSearchIndex(store, options.model, tools, options);
+}
+function isToolIndex(value) {
+  if (typeof value !== "object" || value === null) return false;
+  const candidate = value;
+  return candidate.kind === "tool-index" && Array.isArray(candidate.tools) && typeof candidate.topK === "number" && typeof candidate.search === "function";
+}
+var ToolSearchIndex = class {
+  constructor(store, model, tools, options) {
+    this.store = store;
+    this.model = model;
+    this.options = options;
+    this.tools = Object.freeze([...tools]);
+    this.topK = options.topK;
+    this.minScore = options.minScore;
+    this.filter = options.filter;
+  }
+  store;
+  model;
+  options;
+  kind = "tool-index";
+  tools;
+  topK;
+  minScore;
+  filter;
+  search(options) {
+    return retrieveDocuments({
+      store: this.store,
+      model: this.model,
+      query: options.query,
+      topK: this.topK,
+      minScore: this.minScore,
+      filter: this.filter,
+      retries: this.options.retries,
+      abortSignal: options.abortSignal
+    });
+  }
+  inspect(request) {
+    return this.store.inspect(request);
+  }
+};
+function dedupeTools(tools) {
+  const byName = /* @__PURE__ */ new Map();
+  for (const tool of tools) byName.set(tool.name, tool);
+  return [...byName.values()];
+}
+function defaultToolEmbeddingText(definition) {
+  return [definition.name, definition.description, JSON.stringify(definition.parameters)];
+}
+
+// src/tool/middleware.ts
+function createMiddleware(middleware) {
+  return middleware;
+}
+
+// src/tool/think-tool.ts
+import { z } 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 = z.object({
+  thought: z.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: z.string(),
+    execute: (args) => args.thought
+  });
+}
+
+export {
+  embedTools,
+  createToolIndex,
+  isToolIndex,
+  createMiddleware,
+  createThinkTool
+};
+//# sourceMappingURL=chunk-D37WMSP6.js.map

package/dist/chunk-D37WMSP6.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/tool/dynamic-tools.ts","../src/tool/middleware.ts","../src/tool/think-tool.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    const document: ToolSearchDocument<Metadata> = {\n      toolName: tool.name,\n      definition,\n      text: texts.join(\"\\n\"),\n      ...(metadata === undefined ? {} : { metadata }),\n    };\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","import type {\n  CompletionRequest,\n  CompletionResponse,\n  JsonValue,\n  ToolResultContentPart,\n} from \"../completion\";\nimport type { MaybePromise } from \"../internal/type-utils\";\n\nexport type CompletionRequestMiddlewareArgs = {\n  turn: number;\n  request: CompletionRequest;\n  originalRequest: CompletionRequest;\n};\n\nexport type CompletionRequestMiddlewareResult =\n  | {\n      request: CompletionRequest;\n    }\n  | undefined;\n\nexport type CompletionResponseMiddlewareArgs<RawResponse = unknown> = {\n  turn: number;\n  request: CompletionRequest;\n  response: CompletionResponse<RawResponse>;\n  originalResponse: CompletionResponse<RawResponse>;\n};\n\nexport type CompletionResponseMiddlewareResult<RawResponse = unknown> =\n  | {\n      response: CompletionResponse<RawResponse>;\n    }\n  | undefined;\n\nexport type ToolInputMiddlewareArgs = {\n  toolName: string;\n  args: string;\n  originalArgs: string;\n  turn: number;\n  toolCallId?: string | undefined;\n  internalCallId: string;\n};\n\nexport type ToolInputMiddlewareResult =\n  | {\n      args: JsonValue | string;\n    }\n  | undefined;\n\nexport type ToolResultMiddlewareArgs = {\n  toolName: string;\n  args: string;\n  result: string;\n  originalResult: string;\n  structuredResult?: readonly ToolResultContentPart[] | undefined;\n  originalStructuredResult?: readonly ToolResultContentPart[] | undefined;\n  turn: number;\n  toolCallId?: string | undefined;\n  internalCallId: string;\n};\n\nexport type ToolOutputMiddlewareArgs = ToolResultMiddlewareArgs;\n\nexport type ToolOutputMiddlewareResult =\n  | string\n  | {\n      result?: string | undefined;\n      structuredResult?: readonly ToolResultContentPart[] | undefined;\n    }\n  | undefined;\n\nexport interface AgentMiddleware<RawResponse = unknown> {\n  onCompletionRequest?(\n    args: CompletionRequestMiddlewareArgs,\n  ): MaybePromise<CompletionRequestMiddlewareResult>;\n  onCompletionResponse?(\n    args: CompletionResponseMiddlewareArgs<RawResponse>,\n  ): MaybePromise<CompletionResponseMiddlewareResult<RawResponse>>;\n  onToolInput?(args: ToolInputMiddlewareArgs): MaybePromise<ToolInputMiddlewareResult>;\n  onToolOutput?(args: ToolOutputMiddlewareArgs): MaybePromise<ToolOutputMiddlewareResult>;\n}\n\nexport function createMiddleware<RawResponse = unknown>(\n  middleware: AgentMiddleware<RawResponse>,\n): AgentMiddleware<RawResponse> {\n  return middleware;\n}\n","import { z } from \"zod\";\nimport { createTool } from \"./create-tool\";\n\nconst defaultThinkToolDescription =\n  \"Use this tool to record a thought while reasoning through a complex task. It does not retrieve information, store memory, or change external state.\";\n\nconst thinkToolInput = z.object({\n  thought: z.string().describe(\"A thought to record while reasoning through a task.\"),\n});\n\nexport type CreateThinkToolOptions = {\n  name?: string;\n  description?: string;\n};\n\nexport function createThinkTool(options: CreateThinkToolOptions = {}) {\n  return createTool({\n    name: options.name ?? \"think\",\n    description: options.description ?? defaultThinkToolDescription,\n    inputSchema: thinkToolInput,\n    outputSchema: z.string(),\n    execute: (args) => args.thought,\n  });\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,UAAM,WAAyC;AAAA,MAC7C,UAAU,KAAK;AAAA,MACf;AAAA,MACA,MAAM,MAAM,KAAK,IAAI;AAAA,MACrB,GAAI,aAAa,SAAY,CAAC,IAAI,EAAE,SAAS;AAAA,IAC/C;AACA,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;;;AC1FO,SAAS,iBACd,YAC8B;AAC9B,SAAO;AACT;;;ACrFA,SAAS,SAAS;AAGlB,IAAM,8BACJ;AAEF,IAAM,iBAAiB,EAAE,OAAO;AAAA,EAC9B,SAAS,EAAE,OAAO,EAAE,SAAS,qDAAqD;AACpF,CAAC;AAOM,SAAS,gBAAgB,UAAkC,CAAC,GAAG;AACpE,SAAO,WAAW;AAAA,IAChB,MAAM,QAAQ,QAAQ;AAAA,IACtB,aAAa,QAAQ,eAAe;AAAA,IACpC,aAAa;AAAA,IACb,cAAc,EAAE,OAAO;AAAA,IACvB,SAAS,CAAC,SAAS,KAAK;AAAA,EAC1B,CAAC;AACH;","names":[]}