@anvia/core 0.2.0 → 0.2.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (56) hide show
  1. package/dist/agent/index.d.ts +8 -7
  2. package/dist/agent/index.js +2 -2
  3. package/dist/{agent-CkVbMA1A.d.ts → agent-ufaeGoaz.d.ts} +5 -3
  4. package/dist/audio-generation/index.d.ts +1 -1
  5. package/dist/{chunk-MQVFDXPC.js → chunk-4F7RVGFR.js} +17 -1
  6. package/dist/chunk-4F7RVGFR.js.map +1 -0
  7. package/dist/{chunk-CP47FBJV.js → chunk-6GJDBBDC.js} +1 -1
  8. package/dist/chunk-6GJDBBDC.js.map +1 -0
  9. package/dist/{chunk-HGJDC56F.js → chunk-DO4XW2ZB.js} +54 -7
  10. package/dist/chunk-DO4XW2ZB.js.map +1 -0
  11. package/dist/{chunk-RLA7SST2.js → chunk-NYUOBGVI.js} +14 -1
  12. package/dist/chunk-NYUOBGVI.js.map +1 -0
  13. package/dist/{chunk-BT3I6O56.js → chunk-OD6TY44B.js} +1 -1
  14. package/dist/chunk-OD6TY44B.js.map +1 -0
  15. package/dist/{chunk-JSCITGBH.js → chunk-QAPEP4ON.js} +2 -2
  16. package/dist/{chunk-4ILKTES4.js → chunk-QSIIO4YN.js} +3 -3
  17. package/dist/{chunk-X2LR54ZO.js → chunk-V5EETE3U.js} +6 -2
  18. package/dist/chunk-V5EETE3U.js.map +1 -0
  19. package/dist/completion/index.d.ts +2 -2
  20. package/dist/completion/index.js +1 -1
  21. package/dist/evals/index.d.ts +4 -4
  22. package/dist/evals/index.js +6 -6
  23. package/dist/extractor/index.d.ts +4 -4
  24. package/dist/extractor/index.js +5 -5
  25. package/dist/image-generation/index.d.ts +1 -1
  26. package/dist/index.d.ts +8 -7
  27. package/dist/index.js +8 -8
  28. package/dist/loaders/index.d.ts +1 -1
  29. package/dist/mcp/index.d.ts +7 -5
  30. package/dist/mcp/index.js +1 -1
  31. package/dist/memory/index.d.ts +1 -1
  32. package/dist/{middleware-2iCGCsB6.d.ts → middleware-CKsY_zZE.d.ts} +2 -2
  33. package/dist/observability/index.d.ts +10 -2
  34. package/dist/observability/index.js +1 -1
  35. package/dist/pipeline/index.d.ts +4 -4
  36. package/dist/skills/index.d.ts +4 -4
  37. package/dist/skills/index.js +2 -2
  38. package/dist/tool/index.d.ts +5 -5
  39. package/dist/tool/index.js +2 -2
  40. package/dist/{tool-DiWtAf_Q.d.ts → tool-FEHplpd9.d.ts} +1 -1
  41. package/dist/transcription/index.d.ts +1 -1
  42. package/dist/{types-BrxLd7ay.d.ts → types-BM8-Y8Hy.d.ts} +3 -0
  43. package/dist/{types-21n32ltl.d.ts → types-CW3K1MJm.d.ts} +1 -1
  44. package/dist/{types-ChJoZ0OH.d.ts → types-DpFoVF_g.d.ts} +9 -3
  45. package/dist/vector-store/index.d.ts +20 -3
  46. package/dist/vector-store/index.js +1 -1
  47. package/package.json +7 -7
  48. package/LICENSE +0 -21
  49. package/dist/chunk-BT3I6O56.js.map +0 -1
  50. package/dist/chunk-CP47FBJV.js.map +0 -1
  51. package/dist/chunk-HGJDC56F.js.map +0 -1
  52. package/dist/chunk-MQVFDXPC.js.map +0 -1
  53. package/dist/chunk-RLA7SST2.js.map +0 -1
  54. package/dist/chunk-X2LR54ZO.js.map +0 -1
  55. /package/dist/{chunk-JSCITGBH.js.map → chunk-QAPEP4ON.js.map} +0 -0
  56. /package/dist/{chunk-4ILKTES4.js.map → chunk-QSIIO4YN.js.map} +0 -0
@@ -87,6 +87,7 @@ async function connectMcp(connection) {
87
87
 
88
88
  // src/mcp/connections.ts
89
89
  import { Client } from "@modelcontextprotocol/sdk/client/index.js";
90
+ import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
90
91
  import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
91
92
  import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
92
93
  var mcp = {
@@ -114,6 +115,18 @@ var mcp = {
114
115
  return client;
115
116
  }
116
117
  };
118
+ },
119
+ sse(options) {
120
+ return {
121
+ name: options.name,
122
+ async connect() {
123
+ const client = createSdkClient();
124
+ await client.connect(
125
+ asSdkTransport(new SSEClientTransport(new URL(options.url), options.transport))
126
+ );
127
+ return client;
128
+ }
129
+ };
117
130
  }
118
131
  };
119
132
  function createSdkClient() {
@@ -130,4 +143,4 @@ export {
130
143
  connectMcp,
131
144
  mcp
132
145
  };
133
- //# sourceMappingURL=chunk-RLA7SST2.js.map
146
+ //# sourceMappingURL=chunk-NYUOBGVI.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/mcp/result.ts","../src/mcp/tool.ts","../src/mcp/connect.ts","../src/mcp/connections.ts"],"sourcesContent":["import type { JsonValue } from \"../completion/index\";\nimport type { McpToolCallContent, McpToolCallResult } from \"./types\";\n\nexport function createCallToolParams(\n name: string,\n args: unknown,\n): { name: string; arguments?: Record<string, unknown> } {\n if (args === null || args === undefined) {\n return { name };\n }\n\n if (!isPlainRecord(args)) {\n throw new Error(\"MCP tool arguments must be a JSON object\");\n }\n\n return { name, arguments: args };\n}\n\nexport function mapMcpToolResult(result: McpToolCallResult): string {\n if (\"toolResult\" in result) {\n return serializeMcpValue(result.toolResult);\n }\n\n if (result.isError === true) {\n throw new Error(mcpErrorMessage(result.content));\n }\n\n return result.content.map(mapMcpContent).join(\"\");\n}\n\nfunction mcpErrorMessage(content: McpToolCallContent[]): string {\n const text = content\n .map((item) => (item.type === \"text\" ? item.text : undefined))\n .filter((item): item is string => item !== undefined)\n .join(\"\\n\");\n\n return text === \"\" ? \"MCP tool returned an error\" : text;\n}\n\nfunction mapMcpContent(content: McpToolCallContent): string {\n if (content.type === \"text\") {\n return content.text;\n }\n\n if (content.type === \"image\") {\n return `data:${content.mimeType};base64,${content.data}`;\n }\n\n if (content.type === \"resource\") {\n const mimeType =\n content.resource.mimeType === undefined ? \"\" : `data:${content.resource.mimeType};`;\n if (\"text\" in content.resource) {\n return `${mimeType}${content.resource.uri}:${content.resource.text}`;\n }\n\n return `${mimeType}${content.resource.uri}:${content.resource.blob}`;\n }\n\n throw new Error(`Unsupported MCP tool result content: ${serializeMcpValue(content)}`);\n}\n\nfunction serializeMcpValue(value: unknown): string {\n if (typeof value === \"string\") {\n return value;\n }\n\n const serialized = JSON.stringify(value);\n return serialized === undefined ? String(value) : serialized;\n}\n\nfunction isPlainRecord(value: unknown): value is Record<string, JsonValue> {\n return typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n","import type { ToolDefinition } from \"../completion/index\";\nimport type { Tool } from \"../tool/index\";\nimport { createCallToolParams, mapMcpToolResult } from \"./result\";\nimport type { McpClient, McpToolDefinition } from \"./types\";\n\nconst MCP_TOOL_METADATA_KEY = Symbol.for(\"anvia.mcp.tool.metadata\");\n\nexport function createMcpTool(\n definition: McpToolDefinition,\n client: McpClient,\n serverName?: string,\n): Tool {\n const tool: Tool = {\n name: definition.name,\n definition(): ToolDefinition {\n return {\n name: definition.name,\n description: definition.description ?? \"\",\n parameters: definition.inputSchema,\n };\n },\n async call(args): Promise<string> {\n const result = await client.callTool(createCallToolParams(definition.name, args));\n return mapMcpToolResult(result);\n },\n };\n if (serverName !== undefined) {\n Object.defineProperty(tool, MCP_TOOL_METADATA_KEY, {\n value: { serverName },\n enumerable: false,\n });\n }\n return tool;\n}\n","import { createMcpTool } from \"./tool\";\nimport type { McpConnection, McpServer } from \"./types\";\n\nexport async function connectMcp(connection: McpConnection): Promise<McpServer> {\n const client = await connection.connect();\n const { tools } = await client.listTools();\n\n return {\n name: connection.name,\n tools: tools.map((tool) => createMcpTool(tool, client, connection.name)),\n close: () => client.close(),\n };\n}\n","import { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport type {\n McpClient,\n McpConnection,\n McpHttpOptions,\n McpSseOptions,\n McpStdioOptions,\n} from \"./types\";\n\nexport const mcp = {\n stdio(options: McpStdioOptions): McpConnection {\n return {\n name: options.name,\n async connect(): Promise<McpClient> {\n const { name: _name, ...server } = options;\n const client = createSdkClient();\n await client.connect(asSdkTransport(new StdioClientTransport(server)));\n return client as McpClient;\n },\n };\n },\n\n http(options: McpHttpOptions): McpConnection {\n return {\n name: options.name,\n async connect(): Promise<McpClient> {\n const client = createSdkClient();\n await client.connect(\n asSdkTransport(\n new StreamableHTTPClientTransport(new URL(options.url), options.transport),\n ),\n );\n return client as McpClient;\n },\n };\n },\n\n sse(options: McpSseOptions): McpConnection {\n return {\n name: options.name,\n async connect(): Promise<McpClient> {\n const client = createSdkClient();\n await client.connect(\n asSdkTransport(new SSEClientTransport(new URL(options.url), options.transport)),\n );\n return client as McpClient;\n },\n };\n },\n};\n\nfunction createSdkClient(): Client {\n return new Client({\n name: \"@anvia/core\",\n version: \"0.1.0\",\n });\n}\n\nfunction asSdkTransport(transport: unknown): Parameters<Client[\"connect\"]>[0] {\n return transport as Parameters<Client[\"connect\"]>[0];\n}\n"],"mappings":";AAGO,SAAS,qBACd,MACA,MACuD;AACvD,MAAI,SAAS,QAAQ,SAAS,QAAW;AACvC,WAAO,EAAE,KAAK;AAAA,EAChB;AAEA,MAAI,CAAC,cAAc,IAAI,GAAG;AACxB,UAAM,IAAI,MAAM,0CAA0C;AAAA,EAC5D;AAEA,SAAO,EAAE,MAAM,WAAW,KAAK;AACjC;AAEO,SAAS,iBAAiB,QAAmC;AAClE,MAAI,gBAAgB,QAAQ;AAC1B,WAAO,kBAAkB,OAAO,UAAU;AAAA,EAC5C;AAEA,MAAI,OAAO,YAAY,MAAM;AAC3B,UAAM,IAAI,MAAM,gBAAgB,OAAO,OAAO,CAAC;AAAA,EACjD;AAEA,SAAO,OAAO,QAAQ,IAAI,aAAa,EAAE,KAAK,EAAE;AAClD;AAEA,SAAS,gBAAgB,SAAuC;AAC9D,QAAM,OAAO,QACV,IAAI,CAAC,SAAU,KAAK,SAAS,SAAS,KAAK,OAAO,MAAU,EAC5D,OAAO,CAAC,SAAyB,SAAS,MAAS,EACnD,KAAK,IAAI;AAEZ,SAAO,SAAS,KAAK,+BAA+B;AACtD;AAEA,SAAS,cAAc,SAAqC;AAC1D,MAAI,QAAQ,SAAS,QAAQ;AAC3B,WAAO,QAAQ;AAAA,EACjB;AAEA,MAAI,QAAQ,SAAS,SAAS;AAC5B,WAAO,QAAQ,QAAQ,QAAQ,WAAW,QAAQ,IAAI;AAAA,EACxD;AAEA,MAAI,QAAQ,SAAS,YAAY;AAC/B,UAAM,WACJ,QAAQ,SAAS,aAAa,SAAY,KAAK,QAAQ,QAAQ,SAAS,QAAQ;AAClF,QAAI,UAAU,QAAQ,UAAU;AAC9B,aAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,IACpE;AAEA,WAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,EACpE;AAEA,QAAM,IAAI,MAAM,wCAAwC,kBAAkB,OAAO,CAAC,EAAE;AACtF;AAEA,SAAS,kBAAkB,OAAwB;AACjD,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO;AAAA,EACT;AAEA,QAAM,aAAa,KAAK,UAAU,KAAK;AACvC,SAAO,eAAe,SAAY,OAAO,KAAK,IAAI;AACpD;AAEA,SAAS,cAAc,OAAoD;AACzE,SAAO,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAC5E;;;ACnEA,IAAM,wBAAwB,uBAAO,IAAI,yBAAyB;AAE3D,SAAS,cACd,YACA,QACA,YACM;AACN,QAAM,OAAa;AAAA,IACjB,MAAM,WAAW;AAAA,IACjB,aAA6B;AAC3B,aAAO;AAAA,QACL,MAAM,WAAW;AAAA,QACjB,aAAa,WAAW,eAAe;AAAA,QACvC,YAAY,WAAW;AAAA,MACzB;AAAA,IACF;AAAA,IACA,MAAM,KAAK,MAAuB;AAChC,YAAM,SAAS,MAAM,OAAO,SAAS,qBAAqB,WAAW,MAAM,IAAI,CAAC;AAChF,aAAO,iBAAiB,MAAM;AAAA,IAChC;AAAA,EACF;AACA,MAAI,eAAe,QAAW;AAC5B,WAAO,eAAe,MAAM,uBAAuB;AAAA,MACjD,OAAO,EAAE,WAAW;AAAA,MACpB,YAAY;AAAA,IACd,CAAC;AAAA,EACH;AACA,SAAO;AACT;;;AC9BA,eAAsB,WAAW,YAA+C;AAC9E,QAAM,SAAS,MAAM,WAAW,QAAQ;AACxC,QAAM,EAAE,MAAM,IAAI,MAAM,OAAO,UAAU;AAEzC,SAAO;AAAA,IACL,MAAM,WAAW;AAAA,IACjB,OAAO,MAAM,IAAI,CAAC,SAAS,cAAc,MAAM,QAAQ,WAAW,IAAI,CAAC;AAAA,IACvE,OAAO,MAAM,OAAO,MAAM;AAAA,EAC5B;AACF;;;ACZA,SAAS,cAAc;AACvB,SAAS,0BAA0B;AACnC,SAAS,4BAA4B;AACrC,SAAS,qCAAqC;AASvC,IAAM,MAAM;AAAA,EACjB,MAAM,SAAyC;AAC7C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,EAAE,MAAM,OAAO,GAAG,OAAO,IAAI;AACnC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO,QAAQ,eAAe,IAAI,qBAAqB,MAAM,CAAC,CAAC;AACrE,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,KAAK,SAAwC;AAC3C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX;AAAA,YACE,IAAI,8BAA8B,IAAI,IAAI,QAAQ,GAAG,GAAG,QAAQ,SAAS;AAAA,UAC3E;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,IAAI,SAAuC;AACzC,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX,eAAe,IAAI,mBAAmB,IAAI,IAAI,QAAQ,GAAG,GAAG,QAAQ,SAAS,CAAC;AAAA,QAChF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,kBAA0B;AACjC,SAAO,IAAI,OAAO;AAAA,IAChB,MAAM;AAAA,IACN,SAAS;AAAA,EACX,CAAC;AACH;AAEA,SAAS,eAAe,WAAsD;AAC5E,SAAO;AACT;","names":[]}
@@ -6,4 +6,4 @@ function createObserver(observer) {
6
6
  export {
7
7
  createObserver
8
8
  };
9
- //# sourceMappingURL=chunk-BT3I6O56.js.map
9
+ //# sourceMappingURL=chunk-OD6TY44B.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/observability/types.ts"],"sourcesContent":["import type {\n CompletionModelCapabilities,\n CompletionRequest,\n CompletionResponse,\n JsonObject,\n Message,\n ToolCall,\n ToolDefinition,\n Usage,\n} from \"../completion\";\nimport type { ToolCallStreamEvent } from \"../tool\";\n\nexport type AgentTraceInfo = {\n traceId?: string | undefined;\n observationId?: string | undefined;\n};\n\nexport type AgentTraceOptions = {\n name?: string | undefined;\n userId?: string | undefined;\n sessionId?: string | undefined;\n metadata?: Record<string, unknown> | undefined;\n tags?: string[] | undefined;\n version?: string | undefined;\n traceId?: string | undefined;\n failOnObserverError?: boolean | undefined;\n};\n\nexport type AgentRunStartArgs = {\n agentName?: string | undefined;\n agentDescription?: string | undefined;\n instructions?: string | undefined;\n trace?: AgentTraceOptions | undefined;\n prompt: Message;\n history: Message[];\n maxTurns: number;\n};\n\nexport type AgentRunEndArgs = {\n output: string;\n usage: Usage;\n messages: Message[];\n};\n\nexport type AgentRunErrorArgs = {\n error: unknown;\n usage: Usage;\n messages: Message[];\n};\n\nexport type AgentGenerationStartArgs = {\n turn: number;\n request: CompletionRequest;\n providerRequest?: JsonObject | undefined;\n modelInfo?: {\n provider: string;\n defaultModel: string;\n capabilities?: CompletionModelCapabilities | undefined;\n };\n};\n\nexport type AgentGenerationEndArgs<RawResponse = unknown> = {\n turn: number;\n response: CompletionResponse<RawResponse>;\n firstDeltaMs?: number | undefined;\n};\n\nexport type AgentGenerationErrorArgs = {\n turn: number;\n error: unknown;\n};\n\nexport type AgentToolStartArgs = {\n turn: number;\n toolCall: ToolCall;\n toolName: string;\n args: string;\n internalCallId: string;\n toolCallId?: string | undefined;\n toolDefinition?: ToolDefinition | undefined;\n toolMetadata?: JsonObject | undefined;\n};\n\nexport type AgentToolEndArgs = AgentToolStartArgs & {\n result: string;\n skipped: boolean;\n};\n\nexport type AgentToolErrorArgs = AgentToolStartArgs & {\n error: unknown;\n};\n\nexport type AgentToolStreamEventArgs = AgentToolStartArgs & {\n event: ToolCallStreamEvent;\n};\n\nexport interface AgentGenerationObserver {\n end(args: AgentGenerationEndArgs): void | Promise<void>;\n error?(args: AgentGenerationErrorArgs): void | Promise<void>;\n}\n\nexport interface AgentToolObserver {\n streamEvent?(args: AgentToolStreamEventArgs): void | Promise<void>;\n end(args: AgentToolEndArgs): void | Promise<void>;\n error?(args: AgentToolErrorArgs): void | Promise<void>;\n}\n\nexport interface AgentRunObserver {\n readonly trace?: AgentTraceInfo | undefined;\n startGeneration?(\n args: AgentGenerationStartArgs,\n ): AgentGenerationObserver | undefined | Promise<AgentGenerationObserver | undefined>;\n startTool?(\n args: AgentToolStartArgs,\n ): AgentToolObserver | undefined | Promise<AgentToolObserver | undefined>;\n end(args: AgentRunEndArgs): void | Promise<void>;\n error?(args: AgentRunErrorArgs): void | Promise<void>;\n}\n\nexport interface AgentObserver {\n startRun(\n args: AgentRunStartArgs,\n ): AgentRunObserver | undefined | Promise<AgentRunObserver | undefined>;\n flush?(): void | Promise<void>;\n shutdown?(): void | Promise<void>;\n}\n\nexport type AgentObserverRegistration = {\n observer: AgentObserver;\n failOnObserverError?: boolean | undefined;\n};\n\nexport type ObserveOptions = {\n failOnObserverError?: boolean | undefined;\n};\n\nexport function createObserver(observer: AgentObserver): AgentObserver {\n return observer;\n}\n"],"mappings":";AAwIO,SAAS,eAAe,UAAwC;AACrE,SAAO;AACT;","names":[]}
@@ -1,6 +1,6 @@
1
1
  import {
2
2
  ExtractorBuilder
3
- } from "./chunk-4ILKTES4.js";
3
+ } from "./chunk-QSIIO4YN.js";
4
4
  import {
5
5
  cosineSimilarity,
6
6
  embedText
@@ -301,4 +301,4 @@ export {
301
301
  llmScore,
302
302
  agentEvalTarget
303
303
  };
304
- //# sourceMappingURL=chunk-JSCITGBH.js.map
304
+ //# sourceMappingURL=chunk-QAPEP4ON.js.map
@@ -1,6 +1,6 @@
1
1
  import {
2
2
  AgentBuilder
3
- } from "./chunk-HGJDC56F.js";
3
+ } from "./chunk-DO4XW2ZB.js";
4
4
  import {
5
5
  createTool
6
6
  } from "./chunk-445TT5Q5.js";
@@ -9,7 +9,7 @@ import {
9
9
  CompletionRequestBuilder,
10
10
  Message,
11
11
  Usage
12
- } from "./chunk-CP47FBJV.js";
12
+ } from "./chunk-6GJDBBDC.js";
13
13
 
14
14
  // src/extractor/extractor.ts
15
15
  var SUBMIT_TOOL_NAME = "submit";
@@ -139,4 +139,4 @@ export {
139
139
  Extractor,
140
140
  ExtractorBuilder
141
141
  };
142
- //# sourceMappingURL=chunk-4ILKTES4.js.map
142
+ //# sourceMappingURL=chunk-QSIIO4YN.js.map
@@ -1,6 +1,6 @@
1
1
  import {
2
2
  InMemoryVectorStore
3
- } from "./chunk-MQVFDXPC.js";
3
+ } from "./chunk-4F7RVGFR.js";
4
4
  import {
5
5
  ToolSet
6
6
  } from "./chunk-3H7FVGHU.js";
@@ -54,9 +54,13 @@ var DynamicToolSearchIndex = class {
54
54
  constructor(index, toolSet) {
55
55
  this.index = index;
56
56
  this.toolSet = toolSet;
57
+ if (index.inspect !== void 0) {
58
+ this.inspect = (request) => index.inspect?.(request);
59
+ }
57
60
  }
58
61
  index;
59
62
  toolSet;
63
+ inspect;
60
64
  search(request) {
61
65
  return this.index.search(request);
62
66
  }
@@ -99,4 +103,4 @@ export {
99
103
  createToolMiddleware,
100
104
  createThinkTool
101
105
  };
102
- //# sourceMappingURL=chunk-X2LR54ZO.js.map
106
+ //# sourceMappingURL=chunk-V5EETE3U.js.map
@@ -0,0 +1 @@
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 type {\n VectorInspectPage,\n VectorInspectRequest,\n VectorSearchResult,\n VectorSearchToolOptions,\n} from \"../vector-store\";\nimport {\n InMemoryVectorStore,\n type VectorSearchIndex,\n type VectorSearchRequest,\n} from \"../vector-store\";\nimport type { AnyTool, Tool } from \"./tool\";\nimport { ToolSet } from \"./tool-set\";\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 content?: ((tool: AnyTool, definition: ToolDefinition) => string | string[]) | undefined;\n metadata?: ((tool: AnyTool, definition: ToolDefinition) => Metadata | undefined) | undefined;\n concurrency?: number | undefined;\n};\n\nexport interface DynamicToolIndex<Metadata extends VectorMetadata = VectorMetadata>\n extends VectorSearchIndex<ToolSearchDocument<Metadata>, Metadata> {\n readonly toolSet: ToolSet;\n}\n\nexport async function embedTools<Metadata extends VectorMetadata = VectorMetadata>(\n model: EmbeddingModel,\n tools: AnyTool[] | ToolSet,\n options: EmbedToolsOptions<Metadata> = {},\n): Promise<Array<EmbeddedDocument<ToolSearchDocument<Metadata>, Metadata>>> {\n const toolList = Array.isArray(tools) ? tools : tools.values();\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 return embedDocuments(model, documents, {\n id: (item) => item.tool.name,\n content: (item) => item.texts,\n metadata: (item) => item.metadata,\n concurrency: options.concurrency,\n }).then((embedded) =>\n embedded.map((item) => ({\n ...item,\n document: item.document.document,\n })),\n );\n}\n\nexport async function createToolIndex<Metadata extends VectorMetadata = VectorMetadata>(\n model: EmbeddingModel,\n tools: AnyTool[] | ToolSet,\n options: EmbedToolsOptions<Metadata> = {},\n): Promise<DynamicToolIndex<Metadata>> {\n const toolSet = Array.isArray(tools) ? ToolSet.fromTools(tools) : tools;\n const embedded = await embedTools(model, toolSet, options);\n const index = InMemoryVectorStore.fromDocuments(embedded).index(model);\n return new DynamicToolSearchIndex(index, toolSet);\n}\n\nexport function isDynamicToolIndex(value: unknown): value is DynamicToolIndex {\n return (\n typeof value === \"object\" &&\n value !== null &&\n \"toolSet\" in value &&\n (value as { toolSet?: unknown }).toolSet instanceof ToolSet\n );\n}\n\nclass DynamicToolSearchIndex<Metadata extends VectorMetadata>\n implements DynamicToolIndex<Metadata>\n{\n readonly inspect?: (\n request: VectorInspectRequest,\n ) => Promise<VectorInspectPage<ToolSearchDocument<Metadata>, Metadata>>;\n\n constructor(\n private readonly index: VectorSearchIndex<ToolSearchDocument<Metadata>, Metadata>,\n readonly toolSet: ToolSet,\n ) {\n if (index.inspect !== undefined) {\n this.inspect = (request) =>\n index.inspect?.(request) as Promise<\n VectorInspectPage<ToolSearchDocument<Metadata>, Metadata>\n >;\n }\n }\n\n search(\n request: VectorSearchRequest,\n ): Promise<Array<VectorSearchResult<ToolSearchDocument<Metadata>, Metadata>>> {\n return this.index.search(request);\n }\n\n searchIds(request: VectorSearchRequest): Promise<Array<{ score: number; id: string }>> {\n return this.index.searchIds(request);\n }\n\n asTool(options: VectorSearchToolOptions): Tool<{ query: string; topK?: number }, unknown> {\n return this.index.asTool(options);\n }\n}\n\nfunction defaultToolEmbeddingText(definition: ToolDefinition): string[] {\n return [definition.name, definition.description, JSON.stringify(definition.parameters)];\n}\n","export type ToolResultMiddlewareArgs = {\n toolName: string;\n args: string;\n result: string;\n originalResult: string;\n turn: number;\n toolCallId?: string | undefined;\n internalCallId: string;\n};\n\nexport interface ToolMiddleware {\n onResult?(args: ToolResultMiddlewareArgs): string | undefined | Promise<string | undefined>;\n}\n\nexport function createToolMiddleware(middleware: ToolMiddleware): ToolMiddleware {\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 input: thinkToolInput,\n output: z.string(),\n execute: (args) => args.thought,\n });\n}\n"],"mappings":";;;;;;;;;;;;;;AAmCA,eAAsB,WACpB,OACA,OACA,UAAuC,CAAC,GACkC;AAC1E,QAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,MAAM,OAAO;AAC7D,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,SAAO,eAAe,OAAO,WAAW;AAAA,IACtC,IAAI,CAAC,SAAS,KAAK,KAAK;AAAA,IACxB,SAAS,CAAC,SAAS,KAAK;AAAA,IACxB,UAAU,CAAC,SAAS,KAAK;AAAA,IACzB,aAAa,QAAQ;AAAA,EACvB,CAAC,EAAE;AAAA,IAAK,CAAC,aACP,SAAS,IAAI,CAAC,UAAU;AAAA,MACtB,GAAG;AAAA,MACH,UAAU,KAAK,SAAS;AAAA,IAC1B,EAAE;AAAA,EACJ;AACF;AAEA,eAAsB,gBACpB,OACA,OACA,UAAuC,CAAC,GACH;AACrC,QAAM,UAAU,MAAM,QAAQ,KAAK,IAAI,QAAQ,UAAU,KAAK,IAAI;AAClE,QAAM,WAAW,MAAM,WAAW,OAAO,SAAS,OAAO;AACzD,QAAM,QAAQ,oBAAoB,cAAc,QAAQ,EAAE,MAAM,KAAK;AACrE,SAAO,IAAI,uBAAuB,OAAO,OAAO;AAClD;AAEO,SAAS,mBAAmB,OAA2C;AAC5E,SACE,OAAO,UAAU,YACjB,UAAU,QACV,aAAa,SACZ,MAAgC,mBAAmB;AAExD;AAEA,IAAM,yBAAN,MAEA;AAAA,EAKE,YACmB,OACR,SACT;AAFiB;AACR;AAET,QAAI,MAAM,YAAY,QAAW;AAC/B,WAAK,UAAU,CAAC,YACd,MAAM,UAAU,OAAO;AAAA,IAG3B;AAAA,EACF;AAAA,EATmB;AAAA,EACR;AAAA,EANF;AAAA,EAgBT,OACE,SAC4E;AAC5E,WAAO,KAAK,MAAM,OAAO,OAAO;AAAA,EAClC;AAAA,EAEA,UAAU,SAA6E;AACrF,WAAO,KAAK,MAAM,UAAU,OAAO;AAAA,EACrC;AAAA,EAEA,OAAO,SAAmF;AACxF,WAAO,KAAK,MAAM,OAAO,OAAO;AAAA,EAClC;AACF;AAEA,SAAS,yBAAyB,YAAsC;AACtE,SAAO,CAAC,WAAW,MAAM,WAAW,aAAa,KAAK,UAAU,WAAW,UAAU,CAAC;AACxF;;;AChHO,SAAS,qBAAqB,YAA4C;AAC/E,SAAO;AACT;;;AChBA,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,OAAO;AAAA,IACP,QAAQ,EAAE,OAAO;AAAA,IACjB,SAAS,CAAC,SAAS,KAAK;AAAA,EAC1B,CAAC;AACH;","names":[]}
@@ -1,5 +1,5 @@
1
- import { D as Document, M as Message, b as CompletionModel, q as ToolDefinition, o as ToolChoice, j as JsonValue, J as JsonObject, d as CompletionRequest, e as CompletionResponse } from '../types-BrxLd7ay.js';
2
- export { A as AssistantContent, a as AssistantMessage, C as CompletionCapabilityError, c as CompletionModelCapabilities, f as CompletionStreamEvent, g as DocumentContent, I as ImageContent, h as ImageDetail, i as JsonPrimitive, R as Reasoning, k as ReasoningContent, l as ReasoningContentType, S as StreamingCompletionModel, m as SystemMessage, T as Text, n as ToolCall, p as ToolContent, r as ToolFunction, s as ToolMessage, t as ToolResult, u as ToolResultContent, U as Usage, v as UserContent, w as UserMessage, x as assertCompletionRequestSupported, y as reasoningDisplayText, z as textFromAssistantContent } from '../types-BrxLd7ay.js';
1
+ import { D as Document, M as Message, b as CompletionModel, q as ToolDefinition, o as ToolChoice, j as JsonValue, J as JsonObject, d as CompletionRequest, e as CompletionResponse } from '../types-BM8-Y8Hy.js';
2
+ export { A as AssistantContent, a as AssistantMessage, C as CompletionCapabilityError, c as CompletionModelCapabilities, f as CompletionStreamEvent, g as DocumentContent, I as ImageContent, h as ImageDetail, i as JsonPrimitive, R as Reasoning, k as ReasoningContent, l as ReasoningContentType, S as StreamingCompletionModel, m as SystemMessage, T as Text, n as ToolCall, p as ToolContent, r as ToolFunction, s as ToolMessage, t as ToolResult, u as ToolResultContent, U as Usage, v as UserContent, w as UserMessage, x as assertCompletionRequestSupported, y as reasoningDisplayText, z as textFromAssistantContent } from '../types-BM8-Y8Hy.js';
3
3
 
4
4
  declare function normalizeDocuments(documents: Document[]): Message | undefined;
5
5
  declare function formatDocument(document: Document): string;
@@ -11,7 +11,7 @@ import {
11
11
  normalizeDocuments,
12
12
  reasoningDisplayText,
13
13
  textFromAssistantContent
14
- } from "../chunk-CP47FBJV.js";
14
+ } from "../chunk-6GJDBBDC.js";
15
15
  export {
16
16
  AssistantContent,
17
17
  CompletionCapabilityError,
@@ -1,10 +1,10 @@
1
- import { j as JsonValue, M as Message, b as CompletionModel } from '../types-BrxLd7ay.js';
2
- import { s as PromptResponse, A as Agent } from '../agent-CkVbMA1A.js';
3
- import { Z as ZodSchema } from '../middleware-2iCGCsB6.js';
1
+ import { j as JsonValue, M as Message, b as CompletionModel } from '../types-BM8-Y8Hy.js';
2
+ import { s as PromptResponse, A as Agent } from '../agent-ufaeGoaz.js';
3
+ import { Z as ZodSchema } from '../middleware-CKsY_zZE.js';
4
4
  import { EmbeddingModel } from '../embeddings/index.js';
5
5
  import '../memory/index.js';
6
6
  import '../observability/index.js';
7
- import '../tool-DiWtAf_Q.js';
7
+ import '../tool-FEHplpd9.js';
8
8
  import '../vector-store/index.js';
9
9
  import 'zod';
10
10
 
@@ -7,17 +7,17 @@ import {
7
7
  llmScore,
8
8
  runEvalSuite,
9
9
  semanticSimilarity
10
- } from "../chunk-JSCITGBH.js";
11
- import "../chunk-4ILKTES4.js";
12
- import "../chunk-X2LR54ZO.js";
13
- import "../chunk-MQVFDXPC.js";
14
- import "../chunk-HGJDC56F.js";
10
+ } from "../chunk-QAPEP4ON.js";
11
+ import "../chunk-QSIIO4YN.js";
12
+ import "../chunk-V5EETE3U.js";
13
+ import "../chunk-4F7RVGFR.js";
14
+ import "../chunk-DO4XW2ZB.js";
15
15
  import "../chunk-XXT2UCAR.js";
16
16
  import "../chunk-YK4WAAS4.js";
17
17
  import "../chunk-3H7FVGHU.js";
18
18
  import "../chunk-445TT5Q5.js";
19
19
  import "../chunk-XUUY2L2D.js";
20
- import "../chunk-CP47FBJV.js";
20
+ import "../chunk-6GJDBBDC.js";
21
21
  import "../chunk-A7VDIZQN.js";
22
22
  export {
23
23
  EvalOutcome,
@@ -1,9 +1,9 @@
1
- import { U as Usage, M as Message, b as CompletionModel, j as JsonValue, o as ToolChoice } from '../types-BrxLd7ay.js';
2
- import { A as Agent } from '../agent-CkVbMA1A.js';
3
- import { Z as ZodSchema } from '../middleware-2iCGCsB6.js';
1
+ import { U as Usage, M as Message, b as CompletionModel, j as JsonValue, o as ToolChoice } from '../types-BM8-Y8Hy.js';
2
+ import { A as Agent } from '../agent-ufaeGoaz.js';
3
+ import { Z as ZodSchema } from '../middleware-CKsY_zZE.js';
4
4
  import '../memory/index.js';
5
5
  import '../observability/index.js';
6
- import '../tool-DiWtAf_Q.js';
6
+ import '../tool-FEHplpd9.js';
7
7
  import '../vector-store/index.js';
8
8
  import '../embeddings/index.js';
9
9
  import 'zod';
@@ -2,16 +2,16 @@ import {
2
2
  ExtractionError,
3
3
  Extractor,
4
4
  ExtractorBuilder
5
- } from "../chunk-4ILKTES4.js";
6
- import "../chunk-X2LR54ZO.js";
7
- import "../chunk-MQVFDXPC.js";
8
- import "../chunk-HGJDC56F.js";
5
+ } from "../chunk-QSIIO4YN.js";
6
+ import "../chunk-V5EETE3U.js";
7
+ import "../chunk-4F7RVGFR.js";
8
+ import "../chunk-DO4XW2ZB.js";
9
9
  import "../chunk-XXT2UCAR.js";
10
10
  import "../chunk-YK4WAAS4.js";
11
11
  import "../chunk-3H7FVGHU.js";
12
12
  import "../chunk-445TT5Q5.js";
13
13
  import "../chunk-XUUY2L2D.js";
14
- import "../chunk-CP47FBJV.js";
14
+ import "../chunk-6GJDBBDC.js";
15
15
  import "../chunk-A7VDIZQN.js";
16
16
  export {
17
17
  ExtractionError,
@@ -1,4 +1,4 @@
1
- import { j as JsonValue } from '../types-BrxLd7ay.js';
1
+ import { j as JsonValue } from '../types-BM8-Y8Hy.js';
2
2
 
3
3
  type ImageGenerationRequest = {
4
4
  prompt: string;
package/dist/index.d.ts CHANGED
@@ -1,26 +1,27 @@
1
1
  export { MemoryAppendInput, MemoryContext, MemoryErrorInput, MemoryOptions, MemoryRegistration, MemorySavePolicy, MemoryStore, ResolvedMemoryOptions, SessionOptions, resolveMemoryOptions } from './memory/index.js';
2
- export { A as Agent, a as AgentChildStreamEvent, b as AgentEventAppendInput, c as AgentEventRecord, d as AgentEventStore, e as AgentEventStoreInclude, f as AgentEventStoreOptions, g as AgentEventStoreRegistration, h as AgentOptions, i as AgentSession, j as AgentStreamEvent, k as AgentToolOptions, C as CompletionCallHookArgs, l as CompletionResponseHookArgs, D as DEFAULT_MAX_TURNS, m as DynamicContextOptions, n as DynamicContextRegistration, o as DynamicToolOptions, p as DynamicToolRegistration, H as HookAction, q as HookResult, P as PromptHook, r as PromptRequest, s as PromptResponse, R as RunControl, T as ToolApprovalRequestOptions, t as ToolCallControl, u as ToolCallHookAction, v as ToolCallHookArgs, w as ToolCallHookResult, x as ToolHookArgs, y as ToolResultHookArgs, z as cancelPrompt, B as createHook, E as requestToolApproval, F as runControl, G as skipTool, I as toolCallControl } from './agent-CkVbMA1A.js';
2
+ export { A as Agent, a as AgentChildStreamEvent, b as AgentEventAppendInput, c as AgentEventRecord, d as AgentEventStore, e as AgentEventStoreInclude, f as AgentEventStoreOptions, g as AgentEventStoreRegistration, h as AgentOptions, i as AgentSession, j as AgentStreamEvent, k as AgentToolOptions, C as CompletionCallHookArgs, l as CompletionResponseHookArgs, D as DEFAULT_MAX_TURNS, m as DynamicContextOptions, n as DynamicContextRegistration, o as DynamicToolOptions, p as DynamicToolRegistration, H as HookAction, q as HookResult, P as PromptHook, r as PromptRequest, s as PromptResponse, R as RunControl, T as ToolApprovalRequestOptions, t as ToolCallControl, u as ToolCallHookAction, v as ToolCallHookArgs, w as ToolCallHookResult, x as ToolHookArgs, y as ToolResultHookArgs, z as cancelPrompt, B as createHook, E as requestToolApproval, F as runControl, G as skipTool, I as toolCallControl } from './agent-ufaeGoaz.js';
3
3
  export { AgentBuilder, MaxTurnsError, PromptCancelledError } from './agent/index.js';
4
4
  export { AudioGenerationModel, AudioGenerationRequest, AudioGenerationRequestBuilder, AudioGenerationResponse, audioGenerationRequest } from './audio-generation/index.js';
5
5
  export { CompletionRequestBuilder, formatDocument, normalizeDocuments } from './completion/index.js';
6
- export { A as AssistantContent, a as AssistantMessage, C as CompletionCapabilityError, b as CompletionModel, c as CompletionModelCapabilities, d as CompletionRequest, e as CompletionResponse, f as CompletionStreamEvent, D as Document, g as DocumentContent, I as ImageContent, h as ImageDetail, J as JsonObject, i as JsonPrimitive, j as JsonValue, M as Message, R as Reasoning, k as ReasoningContent, l as ReasoningContentType, S as StreamingCompletionModel, m as SystemMessage, T as Text, n as ToolCall, o as ToolChoice, p as ToolContent, q as ToolDefinition, r as ToolFunction, s as ToolMessage, t as ToolResult, u as ToolResultContent, U as Usage, v as UserContent, w as UserMessage, x as assertCompletionRequestSupported, y as reasoningDisplayText, z as textFromAssistantContent } from './types-BrxLd7ay.js';
6
+ export { A as AssistantContent, a as AssistantMessage, C as CompletionCapabilityError, b as CompletionModel, c as CompletionModelCapabilities, d as CompletionRequest, e as CompletionResponse, f as CompletionStreamEvent, D as Document, g as DocumentContent, I as ImageContent, h as ImageDetail, J as JsonObject, i as JsonPrimitive, j as JsonValue, M as Message, R as Reasoning, k as ReasoningContent, l as ReasoningContentType, S as StreamingCompletionModel, m as SystemMessage, T as Text, n as ToolCall, o as ToolChoice, p as ToolContent, q as ToolDefinition, r as ToolFunction, s as ToolMessage, t as ToolResult, u as ToolResultContent, U as Usage, v as UserContent, w as UserMessage, x as assertCompletionRequestSupported, y as reasoningDisplayText, z as textFromAssistantContent } from './types-BM8-Y8Hy.js';
7
7
  export { EmbedDocumentsOptions, EmbeddedDocument, Embedding, EmbeddingModel, VectorMetadata, VectorMetadataValue, angularDistance, chebyshevDistance, cosineSimilarity, dotProduct, embedDocuments, embedText, embedTexts, euclideanDistance, manhattanDistance } from './embeddings/index.js';
8
8
  export { AgentEvalTargetOptions, ContainsOptions, EvalCase, EvalCaseResult, EvalMetadata, EvalMetric, EvalMetricArgs, EvalMetricResult, EvalOutcome, EvalOutcomeStatus, EvalReportArgs, EvalReporter, EvalSuiteResult, EvalTarget, ExactMatchOptions, LlmJudgeOptions, LlmScoreMetricScore, LlmScoreOptions, RunEvalSuiteOptions, SelectorOrValue, SemanticSimilarityOptions, ValueSelector, agentEvalTarget, contains, exactMatch, llmJudge, llmScore, runEvalSuite, semanticSimilarity } from './evals/index.js';
9
9
  export { ExtractionError, ExtractionResponse, Extractor, ExtractorBuilder } from './extractor/index.js';
10
10
  export { GeneratedImage, ImageGenerationModel, ImageGenerationRequest, ImageGenerationRequestBuilder, ImageGenerationResponse, imageGenerationRequest } from './image-generation/index.js';
11
11
  export { connectMcp, mcp } from './mcp/index.js';
12
- export { M as McpClient, a as McpConnection, b as McpHttpOptions, c as McpServer, d as McpStdioOptions, e as McpToolCallContent, f as McpToolCallResult, g as McpToolDefinition } from './types-ChJoZ0OH.js';
12
+ export { M as McpClient, a as McpConnection, b as McpHttpOptions, c as McpServer, d as McpSseOptions, e as McpStdioOptions, f as McpToolCallContent, g as McpToolCallResult, h as McpToolDefinition } from './types-DpFoVF_g.js';
13
13
  export { ListedModel, ModelList, ModelListingClient, ModelListingError } from './model-listing/index.js';
14
14
  export { AgentGenerationEndArgs, AgentGenerationErrorArgs, AgentGenerationObserver, AgentGenerationStartArgs, AgentObserver, AgentObserverRegistration, AgentRunEndArgs, AgentRunErrorArgs, AgentRunObserver, AgentRunStartArgs, AgentToolEndArgs, AgentToolErrorArgs, AgentToolObserver, AgentToolStartArgs, AgentToolStreamEventArgs, AgentTraceInfo, AgentTraceOptions, ObserveOptions, createObserver } from './observability/index.js';
15
15
  export { Pipeline, PipelineBatchOptions, PipelineBuilder, PipelineGraph, PipelineGraphEdge, PipelineGraphNode, PipelineMetadata, PipelineOp, PipelineRunEvent, PipelineRunObserver, PipelineRunOptions, PipelineStageKind, PipelineStageMetadata } from './pipeline/index.js';
16
- export { D as DynamicToolIndex, E as EmbedToolsOptions, T as ToolMiddleware, a as ToolResultMiddlewareArgs, b as ToolSearchDocument, c as ToolSet, Z as ZodSchema, d as createToolIndex, e as createToolMiddleware, f as embedTools, i as isDynamicToolIndex } from './middleware-2iCGCsB6.js';
16
+ export { D as DynamicToolIndex, E as EmbedToolsOptions, T as ToolMiddleware, a as ToolResultMiddlewareArgs, b as ToolSearchDocument, c as ToolSet, Z as ZodSchema, d as createToolIndex, e as createToolMiddleware, f as embedTools, i as isDynamicToolIndex } from './middleware-CKsY_zZE.js';
17
17
  export { loadSkills, skill } from './skills/index.js';
18
- export { S as Skill, a as SkillLoader, b as SkillSet, c as SkillValidationError, d as SkillValidationIssue } from './types-21n32ltl.js';
18
+ export { S as Skill, a as SkillLoader, b as SkillSet, c as SkillValidationError, d as SkillValidationIssue } from './types-CW3K1MJm.js';
19
19
  export { ReadableStreamOptions, toReadableStream } from './streaming/index.js';
20
20
  export { CreateThinkToolOptions, CreateToolOptions, ToolCallError, ToolJsonError, ToolNotFoundError, createThinkTool, createTool } from './tool/index.js';
21
- export { A as AnyTool, T as Tool, a as ToolApprovalContext, b as ToolApprovalPolicy, c as ToolApprovalRunContext, d as ToolCallContext, e as ToolCallStreamEvent, p as parseToolArgs, s as serializeToolOutput } from './tool-DiWtAf_Q.js';
21
+ export { A as AnyTool, T as Tool, a as ToolApprovalContext, b as ToolApprovalPolicy, c as ToolApprovalRunContext, d as ToolCallContext, e as ToolCallStreamEvent, p as parseToolArgs, s as serializeToolOutput } from './tool-FEHplpd9.js';
22
22
  export { TranscriptionModel, TranscriptionRequest, TranscriptionRequestBuilder, TranscriptionResponse, transcriptionRequest } from './transcription/index.js';
23
- export { InMemoryVectorIndex, InMemoryVectorStore, IndexStrategy, VectorFilter, VectorSearchIndex, VectorSearchRequest, VectorSearchResult, VectorSearchToolOptions, createVectorSearchTool, vectorFilter } from './vector-store/index.js';
23
+ export { InMemoryVectorIndex, InMemoryVectorStore, IndexStrategy, VectorFilter, VectorInspectItem, VectorInspectPage, VectorInspectRequest, VectorSearchIndex, VectorSearchRequest, VectorSearchResult, VectorSearchToolOptions, createVectorSearchTool, vectorFilter } from './vector-store/index.js';
24
+ import '@modelcontextprotocol/sdk/client/sse.js';
24
25
  import '@modelcontextprotocol/sdk/client/stdio.js';
25
26
  import '@modelcontextprotocol/sdk/client/streamableHttp.js';
26
27
  import 'zod';
package/dist/index.js CHANGED
@@ -5,13 +5,13 @@ import {
5
5
  import {
6
6
  connectMcp,
7
7
  mcp
8
- } from "./chunk-RLA7SST2.js";
8
+ } from "./chunk-NYUOBGVI.js";
9
9
  import {
10
10
  ModelListingError
11
11
  } from "./chunk-WJT22R3T.js";
12
12
  import {
13
13
  createObserver
14
- } from "./chunk-BT3I6O56.js";
14
+ } from "./chunk-OD6TY44B.js";
15
15
  import {
16
16
  Pipeline,
17
17
  PipelineBuilder
@@ -34,25 +34,25 @@ import {
34
34
  llmScore,
35
35
  runEvalSuite,
36
36
  semanticSimilarity
37
- } from "./chunk-JSCITGBH.js";
37
+ } from "./chunk-QAPEP4ON.js";
38
38
  import {
39
39
  ExtractionError,
40
40
  Extractor,
41
41
  ExtractorBuilder
42
- } from "./chunk-4ILKTES4.js";
42
+ } from "./chunk-QSIIO4YN.js";
43
43
  import {
44
44
  createThinkTool,
45
45
  createToolIndex,
46
46
  createToolMiddleware,
47
47
  embedTools,
48
48
  isDynamicToolIndex
49
- } from "./chunk-X2LR54ZO.js";
49
+ } from "./chunk-V5EETE3U.js";
50
50
  import {
51
51
  InMemoryVectorIndex,
52
52
  InMemoryVectorStore,
53
53
  createVectorSearchTool,
54
54
  vectorFilter
55
- } from "./chunk-MQVFDXPC.js";
55
+ } from "./chunk-4F7RVGFR.js";
56
56
  import {
57
57
  Agent,
58
58
  AgentBuilder,
@@ -67,7 +67,7 @@ import {
67
67
  runControl,
68
68
  skipTool,
69
69
  toolCallControl
70
- } from "./chunk-HGJDC56F.js";
70
+ } from "./chunk-DO4XW2ZB.js";
71
71
  import {
72
72
  resolveMemoryOptions
73
73
  } from "./chunk-XXT2UCAR.js";
@@ -99,7 +99,7 @@ import {
99
99
  normalizeDocuments,
100
100
  reasoningDisplayText,
101
101
  textFromAssistantContent
102
- } from "./chunk-CP47FBJV.js";
102
+ } from "./chunk-6GJDBBDC.js";
103
103
  import {
104
104
  angularDistance,
105
105
  chebyshevDistance,
@@ -1,4 +1,4 @@
1
- import { D as Document } from '../types-BrxLd7ay.js';
1
+ import { D as Document } from '../types-BM8-Y8Hy.js';
2
2
 
3
3
  type LoaderResult<T> = {
4
4
  ok: true;
@@ -1,15 +1,17 @@
1
- import { a as McpConnection, c as McpServer, d as McpStdioOptions, b as McpHttpOptions } from '../types-ChJoZ0OH.js';
2
- export { M as McpClient, e as McpToolCallContent, f as McpToolCallResult, g as McpToolDefinition } from '../types-ChJoZ0OH.js';
1
+ import { a as McpConnection, c as McpServer, e as McpStdioOptions, b as McpHttpOptions, d as McpSseOptions } from '../types-DpFoVF_g.js';
2
+ export { M as McpClient, f as McpToolCallContent, g as McpToolCallResult, h as McpToolDefinition } from '../types-DpFoVF_g.js';
3
+ import '@modelcontextprotocol/sdk/client/sse.js';
3
4
  import '@modelcontextprotocol/sdk/client/stdio.js';
4
5
  import '@modelcontextprotocol/sdk/client/streamableHttp.js';
5
- import '../types-BrxLd7ay.js';
6
- import '../tool-DiWtAf_Q.js';
6
+ import '../types-BM8-Y8Hy.js';
7
+ import '../tool-FEHplpd9.js';
7
8
 
8
9
  declare function connectMcp(connection: McpConnection): Promise<McpServer>;
9
10
 
10
11
  declare const mcp: {
11
12
  stdio(options: McpStdioOptions): McpConnection;
12
13
  http(options: McpHttpOptions): McpConnection;
14
+ sse(options: McpSseOptions): McpConnection;
13
15
  };
14
16
 
15
- export { McpConnection, McpHttpOptions, McpServer, McpStdioOptions, connectMcp, mcp };
17
+ export { McpConnection, McpHttpOptions, McpServer, McpSseOptions, McpStdioOptions, connectMcp, mcp };
package/dist/mcp/index.js CHANGED
@@ -1,7 +1,7 @@
1
1
  import {
2
2
  connectMcp,
3
3
  mcp
4
- } from "../chunk-RLA7SST2.js";
4
+ } from "../chunk-NYUOBGVI.js";
5
5
  export {
6
6
  connectMcp,
7
7
  mcp
@@ -1,4 +1,4 @@
1
- import { J as JsonObject, M as Message } from '../types-BrxLd7ay.js';
1
+ import { J as JsonObject, M as Message } from '../types-BM8-Y8Hy.js';
2
2
 
3
3
  type MemorySavePolicy = "message" | "turn" | "run";
4
4
  type MemoryContext = {
@@ -1,8 +1,8 @@
1
1
  import { z } from 'zod';
2
- import { q as ToolDefinition } from './types-BrxLd7ay.js';
2
+ import { q as ToolDefinition } from './types-BM8-Y8Hy.js';
3
3
  import { VectorMetadata, EmbeddingModel, EmbeddedDocument } from './embeddings/index.js';
4
4
  import { VectorSearchIndex } from './vector-store/index.js';
5
- import { A as AnyTool, d as ToolCallContext } from './tool-DiWtAf_Q.js';
5
+ import { A as AnyTool, d as ToolCallContext } from './tool-FEHplpd9.js';
6
6
 
7
7
  type ZodSchema<T = unknown> = z.ZodType<T>;
8
8
 
@@ -1,5 +1,5 @@
1
- import { M as Message, d as CompletionRequest, e as CompletionResponse, n as ToolCall, U as Usage } from '../types-BrxLd7ay.js';
2
- import { e as ToolCallStreamEvent } from '../tool-DiWtAf_Q.js';
1
+ import { M as Message, d as CompletionRequest, J as JsonObject, c as CompletionModelCapabilities, e as CompletionResponse, n as ToolCall, q as ToolDefinition, U as Usage } from '../types-BM8-Y8Hy.js';
2
+ import { e as ToolCallStreamEvent } from '../tool-FEHplpd9.js';
3
3
 
4
4
  type AgentTraceInfo = {
5
5
  traceId?: string | undefined;
@@ -37,6 +37,12 @@ type AgentRunErrorArgs = {
37
37
  type AgentGenerationStartArgs = {
38
38
  turn: number;
39
39
  request: CompletionRequest;
40
+ providerRequest?: JsonObject | undefined;
41
+ modelInfo?: {
42
+ provider: string;
43
+ defaultModel: string;
44
+ capabilities?: CompletionModelCapabilities | undefined;
45
+ };
40
46
  };
41
47
  type AgentGenerationEndArgs<RawResponse = unknown> = {
42
48
  turn: number;
@@ -54,6 +60,8 @@ type AgentToolStartArgs = {
54
60
  args: string;
55
61
  internalCallId: string;
56
62
  toolCallId?: string | undefined;
63
+ toolDefinition?: ToolDefinition | undefined;
64
+ toolMetadata?: JsonObject | undefined;
57
65
  };
58
66
  type AgentToolEndArgs = AgentToolStartArgs & {
59
67
  result: string;
@@ -1,6 +1,6 @@
1
1
  import {
2
2
  createObserver
3
- } from "../chunk-BT3I6O56.js";
3
+ } from "../chunk-OD6TY44B.js";
4
4
  export {
5
5
  createObserver
6
6
  };
@@ -1,10 +1,10 @@
1
- import { J as JsonObject, b as CompletionModel } from '../types-BrxLd7ay.js';
2
- import { A as Agent } from '../agent-CkVbMA1A.js';
1
+ import { J as JsonObject, b as CompletionModel } from '../types-BM8-Y8Hy.js';
2
+ import { A as Agent } from '../agent-ufaeGoaz.js';
3
3
  import { Extractor } from '../extractor/index.js';
4
4
  import '../memory/index.js';
5
5
  import '../observability/index.js';
6
- import '../tool-DiWtAf_Q.js';
7
- import '../middleware-2iCGCsB6.js';
6
+ import '../tool-FEHplpd9.js';
7
+ import '../middleware-CKsY_zZE.js';
8
8
  import 'zod';
9
9
  import '../embeddings/index.js';
10
10
  import '../vector-store/index.js';
@@ -1,7 +1,7 @@
1
- import { a as SkillLoader, b as SkillSet } from '../types-21n32ltl.js';
2
- export { S as Skill, c as SkillValidationError, d as SkillValidationIssue } from '../types-21n32ltl.js';
3
- import '../tool-DiWtAf_Q.js';
4
- import '../types-BrxLd7ay.js';
1
+ import { a as SkillLoader, b as SkillSet } from '../types-CW3K1MJm.js';
2
+ export { S as Skill, c as SkillValidationError, d as SkillValidationIssue } from '../types-CW3K1MJm.js';
3
+ import '../tool-FEHplpd9.js';
4
+ import '../types-BM8-Y8Hy.js';
5
5
 
6
6
  declare function loadSkills(loaders: SkillLoader | SkillLoader[]): Promise<SkillSet>;
7
7
 
@@ -3,8 +3,8 @@ import {
3
3
  loadSkills,
4
4
  skill
5
5
  } from "../chunk-2VCWYCPV.js";
6
- import "../chunk-X2LR54ZO.js";
7
- import "../chunk-MQVFDXPC.js";
6
+ import "../chunk-V5EETE3U.js";
7
+ import "../chunk-4F7RVGFR.js";
8
8
  import "../chunk-YK4WAAS4.js";
9
9
  import "../chunk-3H7FVGHU.js";
10
10
  import "../chunk-445TT5Q5.js";
@@ -1,9 +1,9 @@
1
1
  import { z } from 'zod';
2
- import { Z as ZodSchema } from '../middleware-2iCGCsB6.js';
3
- export { D as DynamicToolIndex, E as EmbedToolsOptions, T as ToolMiddleware, a as ToolResultMiddlewareArgs, b as ToolSearchDocument, c as ToolSet, d as createToolIndex, e as createToolMiddleware, f as embedTools, i as isDynamicToolIndex } from '../middleware-2iCGCsB6.js';
4
- import { b as ToolApprovalPolicy, d as ToolCallContext, T as Tool } from '../tool-DiWtAf_Q.js';
5
- export { A as AnyTool, a as ToolApprovalContext, c as ToolApprovalRunContext, e as ToolCallStreamEvent, p as parseToolArgs, s as serializeToolOutput } from '../tool-DiWtAf_Q.js';
6
- import '../types-BrxLd7ay.js';
2
+ import { Z as ZodSchema } from '../middleware-CKsY_zZE.js';
3
+ export { D as DynamicToolIndex, E as EmbedToolsOptions, T as ToolMiddleware, a as ToolResultMiddlewareArgs, b as ToolSearchDocument, c as ToolSet, d as createToolIndex, e as createToolMiddleware, f as embedTools, i as isDynamicToolIndex } from '../middleware-CKsY_zZE.js';
4
+ import { b as ToolApprovalPolicy, d as ToolCallContext, T as Tool } from '../tool-FEHplpd9.js';
5
+ export { A as AnyTool, a as ToolApprovalContext, c as ToolApprovalRunContext, e as ToolCallStreamEvent, p as parseToolArgs, s as serializeToolOutput } from '../tool-FEHplpd9.js';
6
+ import '../types-BM8-Y8Hy.js';
7
7
  import '../embeddings/index.js';
8
8
  import '../vector-store/index.js';
9
9
 
@@ -4,8 +4,8 @@ import {
4
4
  createToolMiddleware,
5
5
  embedTools,
6
6
  isDynamicToolIndex
7
- } from "../chunk-X2LR54ZO.js";
8
- import "../chunk-MQVFDXPC.js";
7
+ } from "../chunk-V5EETE3U.js";
8
+ import "../chunk-4F7RVGFR.js";
9
9
  import {
10
10
  ToolCallError,
11
11
  ToolJsonError,
@@ -1,4 +1,4 @@
1
- import { J as JsonObject, q as ToolDefinition, j as JsonValue } from './types-BrxLd7ay.js';
1
+ import { J as JsonObject, q as ToolDefinition, j as JsonValue } from './types-BM8-Y8Hy.js';
2
2
 
3
3
  type ToolApprovalRunContext = {
4
4
  agentId: string;
@@ -1,4 +1,4 @@
1
- import { j as JsonValue } from '../types-BrxLd7ay.js';
1
+ import { j as JsonValue } from '../types-BM8-Y8Hy.js';
2
2
 
3
3
  type TranscriptionRequest = {
4
4
  data: Uint8Array;
@@ -205,6 +205,9 @@ interface CompletionModel<RawResponse = unknown> {
205
205
  readonly provider: string;
206
206
  readonly defaultModel: string;
207
207
  readonly capabilities: CompletionModelCapabilities;
208
+ traceRequest?(request: CompletionRequest, options?: {
209
+ stream?: boolean | undefined;
210
+ }): JsonObject | undefined;
208
211
  completion(request: CompletionRequest): Promise<CompletionResponse<RawResponse>>;
209
212
  }
210
213
  type CompletionStreamEvent<RawResponse = unknown> = {
@@ -1,4 +1,4 @@
1
- import { A as AnyTool } from './tool-DiWtAf_Q.js';
1
+ import { A as AnyTool } from './tool-FEHplpd9.js';
2
2
 
3
3
  type Skill = {
4
4
  readonly name: string;
@@ -1,7 +1,8 @@
1
+ import { SSEClientTransportOptions } from '@modelcontextprotocol/sdk/client/sse.js';
1
2
  import { StdioServerParameters } from '@modelcontextprotocol/sdk/client/stdio.js';
2
3
  import { StreamableHTTPClientTransportOptions } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
3
- import { J as JsonObject } from './types-BrxLd7ay.js';
4
- import { A as AnyTool } from './tool-DiWtAf_Q.js';
4
+ import { J as JsonObject } from './types-BM8-Y8Hy.js';
5
+ import { A as AnyTool } from './tool-FEHplpd9.js';
5
6
 
6
7
  type McpToolDefinition = {
7
8
  name: string;
@@ -60,5 +61,10 @@ type McpHttpOptions = {
60
61
  url: string | URL;
61
62
  transport?: StreamableHTTPClientTransportOptions | undefined;
62
63
  };
64
+ type McpSseOptions = {
65
+ name: string;
66
+ url: string | URL;
67
+ transport?: SSEClientTransportOptions | undefined;
68
+ };
63
69
 
64
- export type { McpClient as M, McpConnection as a, McpHttpOptions as b, McpServer as c, McpStdioOptions as d, McpToolCallContent as e, McpToolCallResult as f, McpToolDefinition as g };
70
+ export type { McpClient as M, McpConnection as a, McpHttpOptions as b, McpServer as c, McpSseOptions as d, McpStdioOptions as e, McpToolCallContent as f, McpToolCallResult as g, McpToolDefinition as h };