extrait 0.1.1 → 0.1.2

package/README.md

@@ -215,6 +215,10 @@ const result = await llm.structured(
       onToolExecution: (execution) => {
         console.log(execution.name, execution.durationMs);
       },
+      // Optional: transform tool output before it is sent back to the LLM
+      transformToolOutput: (output, execution) => {
+        return { ...output, source: execution.name };
+      },
     },
   }
 );

package/dist/index.cjs

@@ -1301,20 +1301,25 @@ async function executeMCPToolCalls(calls, toolset, context) {
         name: tool.remoteName,
         arguments: args
       });
-      metadata.output = output;
-      const execution = {
+      const executionContext = {
         callId,
-        type: metadata.type,
+        type: call.type ?? "function",
         name: toolName,
         clientId: tool.clientId,
         remoteName: tool.remoteName,
         arguments: parsedArguments,
-        output,
         round: context.round,
         provider: context.provider,
         model: context.model,
         handledLocally: true,
         startedAt,
+        error: undefined
+      };
+      const transformedOutput = context.request.transformToolOutput ? await context.request.transformToolOutput(output, executionContext) : output;
+      metadata.output = transformedOutput;
+      const execution = {
+        ...executionContext,
+        output: transformedOutput,
         durationMs: Date.now() - startedAtMs
       };
       emitToolExecution(context.request, execution);

package/dist/index.d.ts

@@ -12,4 +12,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, ExtractJsonCandidatesOptions, ExtractionCandidate, ExtractionHeuristicsOptions, ExtractionParseHint, HTTPHeaders, LLMAdapter, LLMRequest, LLMResponse, LLMStreamCallbacks, LLMStreamChunk, LLMToolCall, LLMToolDebugOptions, LLMToolExecution, 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, ThinkDiagnostics, ThinkBlock, StructuredTraceEvent, } from "./types";
+export type { CandidateDiagnostics, ExtractJsonCandidatesOptions, ExtractionCandidate, ExtractionHeuristicsOptions, ExtractionParseHint, HTTPHeaders, LLMAdapter, LLMRequest, LLMResponse, LLMStreamCallbacks, LLMStreamChunk, LLMToolCall, LLMToolDebugOptions, LLMToolExecution, LLMToolOutputTransformer, 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, ThinkDiagnostics, ThinkBlock, StructuredTraceEvent, } from "./types";

package/dist/index.js

@@ -1222,20 +1222,25 @@ async function executeMCPToolCalls(calls, toolset, context) {
         name: tool.remoteName,
         arguments: args
       });
-      metadata.output = output;
-      const execution = {
+      const executionContext = {
         callId,
-        type: metadata.type,
+        type: call.type ?? "function",
         name: toolName,
         clientId: tool.clientId,
         remoteName: tool.remoteName,
         arguments: parsedArguments,
-        output,
         round: context.round,
         provider: context.provider,
         model: context.model,
         handledLocally: true,
         startedAt,
+        error: undefined
+      };
+      const transformedOutput = context.request.transformToolOutput ? await context.request.transformToolOutput(output, executionContext) : output;
+      metadata.output = transformedOutput;
+      const execution = {
+        ...executionContext,
+        output: transformedOutput,
         durationMs: Date.now() - startedAtMs
       };
       emitToolExecution(context.request, execution);

package/dist/types.d.ts

@@ -129,6 +129,7 @@ export interface LLMRequest {
     parallelToolCalls?: boolean;
     maxToolRounds?: number;
     onToolExecution?: (execution: LLMToolExecution) => void;
+    transformToolOutput?: LLMToolOutputTransformer;
     toolDebug?: boolean | LLMToolDebugOptions;
     body?: Record<string, unknown>;
 }
@@ -189,6 +190,7 @@ export interface LLMToolExecution {
     startedAt: string;
     durationMs?: number;
 }
+export type LLMToolOutputTransformer = (output: unknown, execution: Omit<LLMToolExecution, "output" | "durationMs">) => unknown | Promise<unknown>;
 export interface LLMToolDebugOptions {
     enabled?: boolean;
     logger?: (line: string) => void;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "extrait",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "license": "MIT",
   "repository": {
     "type": "git",
@@ -34,16 +34,16 @@
     "build:types": "bunx tsc -p tsconfig.build.json",
     "lint": "bunx tsc -p tsconfig.lint.json",
     "prepublishOnly": "bun run lint && bun run build && bun run build:types",
-    "test": "bun test",
+    "test": "bun test tests/ --reporter=dots --only-failures",
     "typecheck": "bunx tsc --noEmit"
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.26.0",
-    "jsonrepair": "^3.13.1",
-    "zod": "^3.24.2"
+    "jsonrepair": "^3.13.2",
+    "zod": "^3.25.76"
   },
   "devDependencies": {
     "@types/bun": "latest",
-    "typescript": "^5"
+    "typescript": "^5.9.3"
   }
 }