extrait 0.1.0 → 0.1.2

package/README.md

@@ -2,7 +2,13 @@
 
 Structured JSON extraction from LLMs with validation, repair, and streaming.
 
-Features:
+<p align="left">
+  <a href="https://www.npmjs.com/package/extrait">
+    <img src="https://img.shields.io/npm/v/extrait?logo=npm&style=plastic" alt="npm package">
+  </a>
+</p>
+
+**Features:**
 - Multi-candidate JSON extraction from LLM responses
 - Automatic repair with jsonrepair
 - Zod schema validation and coercion
@@ -13,7 +19,11 @@ Features:
 ## Installation
 
 ```bash
-bun install
+bun add extrait
+# or
+npm install extrait
+# or
+deno add npm:extrait
 ```
 
 ## Quick Start
@@ -205,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/format.d.ts

@@ -2,7 +2,7 @@ import type { z } from "zod";
 export interface WithFormatOptions {
     schemaInstruction?: string;
 }
-export declare const DEFAULT_SCHEMA_INSTRUCTION = "Strictly follow this schema:";
+export declare const DEFAULT_SCHEMA_INSTRUCTION = "Strictly follow this JSON schema:";
 export declare function withFormat(schema: z.ZodTypeAny, options?: WithFormatOptions): string;
 export declare function formatPrompt(schema: z.ZodTypeAny, task: string, options?: WithFormatOptions): string;
 export declare function resolveSchemaInstruction(instruction: string | undefined): string;

package/dist/index.cjs

@@ -976,7 +976,7 @@ function sanitizeDescription(value) {
 }
 
 // src/format.ts
-var DEFAULT_SCHEMA_INSTRUCTION = "Strictly follow this schema:";
+var DEFAULT_SCHEMA_INSTRUCTION = "Strictly follow this JSON schema:";
 function withFormat(schema, options = {}) {
   const schemaType = formatZodSchemaLikeTypeScript(schema);
   const instruction = resolveSchemaInstruction(options.schemaInstruction);
@@ -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

@@ -897,7 +897,7 @@ function sanitizeDescription(value) {
 }
 
 // src/format.ts
-var DEFAULT_SCHEMA_INSTRUCTION = "Strictly follow this schema:";
+var DEFAULT_SCHEMA_INSTRUCTION = "Strictly follow this JSON schema:";
 function withFormat(schema, options = {}) {
   const schemaType = formatZodSchemaLikeTypeScript(schema);
   const instruction = resolveSchemaInstruction(options.schemaInstruction);
@@ -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,7 +1,14 @@
 {
   "name": "extrait",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/tterrasson/extrait.git"
+  },
+  "bugs": {
+    "url": "https://github.com/tterrasson/extrait/issues"
+  },
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -27,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"
   }
 }