extrait 0.3.1 → 0.5.1

package/dist/prompt.d.ts

@@ -4,6 +4,8 @@ export interface PromptMessageBuilder extends StructuredPromptResolver {
     system(strings: TemplateStringsArray, ...values: unknown[]): PromptMessageBuilder;
     user(input: string): PromptMessageBuilder;
     user(strings: TemplateStringsArray, ...values: unknown[]): PromptMessageBuilder;
+    assistant(input: string): PromptMessageBuilder;
+    assistant(strings: TemplateStringsArray, ...values: unknown[]): PromptMessageBuilder;
     build(): StructuredPromptPayload;
 }
 export declare function prompt(strings: TemplateStringsArray, ...values: unknown[]): string;

package/dist/types.d.ts

@@ -119,9 +119,14 @@ export interface MCPToolClient {
     callTool(params: MCPCallToolParams): Promise<unknown>;
     close?(): Promise<void>;
 }
+export interface LLMMessage {
+    role: "system" | "user" | "assistant" | "tool";
+    content: unknown;
+}
 export interface LLMRequest {
-    prompt: string;
+    prompt?: string;
     systemPrompt?: string;
+    messages?: LLMMessage[];
     temperature?: number;
     maxTokens?: number;
     mcpClients?: MCPToolClient[];
@@ -130,9 +135,11 @@ export interface LLMRequest {
     maxToolRounds?: number;
     onToolExecution?: (execution: LLMToolExecution) => void;
     transformToolOutput?: LLMToolOutputTransformer;
+    transformToolArguments?: LLMToolArgumentsTransformer;
     unknownToolError?: (toolName: string) => string;
     toolDebug?: boolean | LLMToolDebugOptions;
     body?: Record<string, unknown>;
+    signal?: AbortSignal;
 }
 export interface LLMUsage {
     inputTokens?: number;
@@ -192,6 +199,11 @@ export interface LLMToolExecution {
     durationMs?: number;
 }
 export type LLMToolOutputTransformer = (output: unknown, execution: Omit<LLMToolExecution, "output" | "durationMs">) => unknown | Promise<unknown>;
+export type LLMToolArgumentsTransformer = (args: Record<string, unknown>, context: {
+    name: string;
+    remoteName: string;
+    clientId: string;
+}) => Record<string, unknown> | Promise<Record<string, unknown>>;
 export interface LLMToolDebugOptions {
     enabled?: boolean;
     logger?: (line: string) => void;
@@ -217,8 +229,9 @@ export interface StructuredPromptContext {
     mode: StructuredMode;
 }
 export interface StructuredPromptPayload {
-    prompt: string;
+    prompt?: string;
     systemPrompt?: string;
+    messages?: LLMMessage[];
 }
 export interface StructuredPromptResolver {
     resolvePrompt(context: StructuredPromptContext): StructuredPromptPayload;
@@ -262,7 +275,7 @@ export interface StructuredCallOptions<TSchema extends z.ZodTypeAny> {
     debug?: boolean | StructuredDebugOptions;
     observe?: (event: StructuredTraceEvent) => void;
     systemPrompt?: string;
-    request?: Omit<LLMRequest, "prompt" | "systemPrompt">;
+    request?: Omit<LLMRequest, "prompt" | "systemPrompt" | "messages">;
     schemaInstruction?: string;
 }
 export interface StructuredOptions<TSchema extends z.ZodTypeAny> extends StructuredCallOptions<TSchema> {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "extrait",
-  "version": "0.3.1",
+  "version": "0.5.1",
   "license": "MIT",
   "repository": {
     "type": "git",
@@ -36,15 +36,21 @@
     "prepublishOnly": "bun run lint && bun run build && bun run build:types",
     "test": "bun test tests/ --reporter=dots --only-failures",
     "typecheck": "bunx tsc --noEmit",
-    "pack": "bun run build && npm pack"
+    "pack": "bun run build:types && bun run build && npm pack"
   },
   "dependencies": {
-    "@modelcontextprotocol/sdk": "^1.26.0",
+    "@modelcontextprotocol/sdk": "^1.27.1",
     "jsonrepair": "^3.13.2",
-    "zod": "^3.25.76"
+    "zod": "^4.3.6"
   },
   "devDependencies": {
-    "@types/bun": "latest",
+    "@types/bun": "^1.3.10",
     "typescript": "^5.9.3"
+  },
+  "overrides": {
+    "zod": "^4.3.6"
+  },
+  "resolutions": {
+    "zod": "^4.3.6"
   }
 }