extrait 0.5.5 → 0.5.6

package/README.md

@@ -485,6 +485,14 @@ const result = await llm.structured(
       },
       // Optional: transform tool arguments before the tool is called
       transformToolArguments: (args, call) => args,
+      // Optional: transform the full MCP call payload, including _meta
+      transformToolCallParams: (params, call) => ({
+        ...params,
+        _meta: {
+          source: "extrait-docs",
+          clientId: call.clientId,
+        },
+      }),
       // Optional: custom error message when an unknown tool is called
       unknownToolError: (toolName) => `Tool "${toolName}" is not available.`,
     },
@@ -494,6 +502,18 @@ const result = await llm.structured(
 await mcpClient.close?.();
 ```
 
+`transformToolArguments()` only receives the tool input object. `transformToolCallParams()` runs after it and receives the full `MCPCallToolParams` payload that will be sent to the MCP client:
+
+```typescript
+type MCPCallToolParams = {
+  name: string;
+  arguments?: Record<string, unknown>;
+  _meta?: Record<string, unknown>;
+};
+```
+
+Use `transformToolCallParams()` when you need to attach MCP-specific metadata, override the final remote tool name, or otherwise change the full request passed to `client.callTool()`. This hook is exported as `LLMToolCallParamsTransformer`.
+
 ### Timeouts
 
 Use `timeout` to set per-request and per-tool-call time limits without managing `AbortSignal` manually.

package/dist/index.cjs

@@ -1317,6 +1317,17 @@ async function executeMCPToolCalls(calls, toolset, context) {
       remoteName: tool.remoteName,
       clientId: tool.clientId
     }) : rawArgs;
+    const toolParams = context.request.transformToolCallParams ? await context.request.transformToolCallParams({
+      name: tool.remoteName,
+      arguments: args
+    }, {
+      name: toolName,
+      remoteName: tool.remoteName,
+      clientId: tool.clientId
+    }) : {
+      name: tool.remoteName,
+      arguments: args
+    };
     const metadata = {
       id: callId,
       type: call.type ?? "function",
@@ -1326,10 +1337,7 @@ async function executeMCPToolCalls(calls, toolset, context) {
     const startedAt = new Date().toISOString();
     const startedAtMs = Date.now();
     try {
-      const output = await tool.client.callTool({
-        name: tool.remoteName,
-        arguments: args
-      });
+      const output = await tool.client.callTool(toolParams);
       const executionContext = {
         callId,
         type: call.type ?? "function",
@@ -4495,6 +4503,10 @@ async function callModel(adapter, options) {
     parallelToolCalls: options.request?.parallelToolCalls,
     maxToolRounds: options.request?.maxToolRounds,
     onToolExecution: options.request?.onToolExecution,
+    transformToolOutput: options.request?.transformToolOutput,
+    transformToolArguments: options.request?.transformToolArguments,
+    transformToolCallParams: options.request?.transformToolCallParams,
+    unknownToolError: options.request?.unknownToolError,
     toolDebug: options.request?.toolDebug,
     body: options.request?.body,
     signal: requestSignal

package/dist/index.d.ts

@@ -14,4 +14,4 @@ export { createOpenAICompatibleAdapter, type OpenAICompatibleAdapterOptions, } f
 export { createAnthropicCompatibleAdapter, DEFAULT_ANTHROPIC_MAX_TOKENS, DEFAULT_ANTHROPIC_VERSION, type AnthropicCompatibleAdapterOptions, } from "./providers/anthropic-compatible";
 export { DEFAULT_MAX_TOOL_ROUNDS } from "./providers/mcp-runtime";
 export { createDefaultProviderRegistry, createModelAdapter, createProviderRegistry, registerBuiltinProviders, type BuiltinProviderKind, type ModelAdapterConfig, type ProviderFactory, type ProviderRegistry, type ProviderTransportConfig, } from "./providers/registry";
-export type { CandidateDiagnostics, EmbeddingRequest, EmbeddingResult, LLMImageContent, LLMMessageContent, LLMTextContent, ExtractJsonCandidatesOptions, ExtractionCandidate, ExtractionHeuristicsOptions, ExtractionParseHint, HTTPHeaders, LLMAdapter, LLMMessage, LLMRequest, LLMResponse, LLMStreamCallbacks, LLMStreamChunk, LLMToolCall, LLMToolCallRef, LLMToolDebugOptions, LLMToolExecution, LLMToolOutputTransformer, LLMToolArgumentsTransformer, LLMToolChoice, MCPCallToolParams, MCPListToolsResult, MCPToolClient, MCPToolDescriptor, MCPToolSchema, LLMUsage, MarkdownCodeBlock, MarkdownCodeOptions, ParseLLMOutputOptions, ParseLLMOutputResult, ParseTraceEvent, PipelineError, StructuredAttempt, StructuredCallOptions, StructuredDebugOptions, StructuredError, StructuredMode, StructuredOptions, StructuredPromptBuilder, StructuredPromptContext, StructuredPromptPayload, StructuredPromptResolver, StructuredPromptValue, StructuredResult, StructuredStreamData, StructuredStreamEvent, StructuredStreamInput, StructuredStreamOptions, StructuredSelfHealInput, StructuredTimeoutOptions, ThinkDiagnostics, ThinkBlock, StructuredTraceEvent, } from "./types";
+export type { CandidateDiagnostics, EmbeddingRequest, EmbeddingResult, LLMImageContent, LLMMessageContent, LLMTextContent, ExtractJsonCandidatesOptions, ExtractionCandidate, ExtractionHeuristicsOptions, ExtractionParseHint, HTTPHeaders, LLMAdapter, LLMMessage, LLMRequest, LLMResponse, LLMStreamCallbacks, LLMStreamChunk, LLMToolCall, LLMToolCallRef, LLMToolDebugOptions, LLMToolExecution, LLMToolOutputTransformer, LLMToolArgumentsTransformer, LLMToolCallParamsTransformer, LLMToolChoice, MCPCallToolParams, MCPListToolsResult, MCPToolClient, MCPToolDescriptor, MCPToolSchema, LLMUsage, MarkdownCodeBlock, MarkdownCodeOptions, ParseLLMOutputOptions, ParseLLMOutputResult, ParseTraceEvent, PipelineError, StructuredAttempt, StructuredCallOptions, StructuredDebugOptions, StructuredError, StructuredMode, StructuredOptions, StructuredPromptBuilder, StructuredPromptContext, StructuredPromptPayload, StructuredPromptResolver, StructuredPromptValue, StructuredResult, StructuredStreamData, StructuredStreamEvent, StructuredStreamInput, StructuredStreamOptions, StructuredSelfHealInput, StructuredTimeoutOptions, ThinkDiagnostics, ThinkBlock, StructuredTraceEvent, } from "./types";

package/dist/index.js

@@ -1228,6 +1228,17 @@ async function executeMCPToolCalls(calls, toolset, context) {
       remoteName: tool.remoteName,
       clientId: tool.clientId
     }) : rawArgs;
+    const toolParams = context.request.transformToolCallParams ? await context.request.transformToolCallParams({
+      name: tool.remoteName,
+      arguments: args
+    }, {
+      name: toolName,
+      remoteName: tool.remoteName,
+      clientId: tool.clientId
+    }) : {
+      name: tool.remoteName,
+      arguments: args
+    };
     const metadata = {
       id: callId,
       type: call.type ?? "function",
@@ -1237,10 +1248,7 @@ async function executeMCPToolCalls(calls, toolset, context) {
     const startedAt = new Date().toISOString();
     const startedAtMs = Date.now();
     try {
-      const output = await tool.client.callTool({
-        name: tool.remoteName,
-        arguments: args
-      });
+      const output = await tool.client.callTool(toolParams);
       const executionContext = {
         callId,
         type: call.type ?? "function",
@@ -4406,6 +4414,10 @@ async function callModel(adapter, options) {
     parallelToolCalls: options.request?.parallelToolCalls,
     maxToolRounds: options.request?.maxToolRounds,
     onToolExecution: options.request?.onToolExecution,
+    transformToolOutput: options.request?.transformToolOutput,
+    transformToolArguments: options.request?.transformToolArguments,
+    transformToolCallParams: options.request?.transformToolCallParams,
+    unknownToolError: options.request?.unknownToolError,
     toolDebug: options.request?.toolDebug,
     body: options.request?.body,
     signal: requestSignal

package/dist/types.d.ts

@@ -110,6 +110,7 @@ export interface MCPListToolsResult {
 export interface MCPCallToolParams {
     name: string;
     arguments?: Record<string, unknown>;
+    _meta?: Record<string, unknown>;
 }
 export interface MCPToolClient {
     id: string;
@@ -156,6 +157,7 @@ export interface LLMRequest {
     onToolExecution?: (execution: LLMToolExecution) => void;
     transformToolOutput?: LLMToolOutputTransformer;
     transformToolArguments?: LLMToolArgumentsTransformer;
+    transformToolCallParams?: LLMToolCallParamsTransformer;
     unknownToolError?: (toolName: string) => string;
     toolDebug?: boolean | LLMToolDebugOptions;
     body?: Record<string, unknown>;
@@ -237,6 +239,11 @@ export type LLMToolArgumentsTransformer = (args: Record<string, unknown>, contex
     remoteName: string;
     clientId: string;
 }) => Record<string, unknown> | Promise<Record<string, unknown>>;
+export type LLMToolCallParamsTransformer = (params: MCPCallToolParams, context: {
+    name: string;
+    remoteName: string;
+    clientId: string;
+}) => MCPCallToolParams | Promise<MCPCallToolParams>;
 export interface LLMToolDebugOptions {
     enabled?: boolean;
     logger?: (line: string) => void;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "extrait",
-  "version": "0.5.5",
+  "version": "0.5.6",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/tterrasson/extrait.git"