@xandout/libra-harness 0.1.8

package/dist/ai-sdk-model.js

@@ -0,0 +1,260 @@
+import { messageContentToText, } from './types.js';
+/**
+ * Adapter that wraps a Vercel AI SDK `LanguageModelV4` as a libra `Model`.
+ *
+ * This lets you use any current AI SDK provider (`@ai-sdk/openai`,
+ * `@ai-sdk/google`, `@ai-sdk/anthropic`, etc.) with libra's hook system
+ * and tool-call loop. Provider-specific wire formats are handled by the
+ * AI SDK while libra retains ownership of tool execution and continuation.
+ *
+ * When `ModelRequest.onDelta` is set, the adapter uses `doStream` and
+ * emits text, reasoning, and tool-input deltas via the callback. The
+ * final assembled `ModelResponse` is still returned. When `onDelta` is
+ * not set, it uses `doGenerate` (no streaming overhead).
+ */
+export class AISdkModel {
+    model;
+    constructor(model) {
+        this.model = model;
+    }
+    async generate(request) {
+        if (request.onDelta)
+            return this.generateStream(request);
+        return this.generateBatch(request);
+    }
+    async generateBatch(request) {
+        const result = await this.model.doGenerate(this.toCallOptions(request));
+        return this.fromAISdkResult(result);
+    }
+    async generateStream(request) {
+        const onDelta = request.onDelta;
+        const { stream } = await this.model.doStream(this.toCallOptions(request));
+        const content = [];
+        const textById = new Map();
+        const textOrder = [];
+        const toolCalls = [];
+        let finishReason = 'stop';
+        let usage;
+        const toolInputBuffers = new Map();
+        const reader = stream.getReader();
+        try {
+            while (true) {
+                const { done, value: part } = await reader.read();
+                if (done)
+                    break;
+                switch (part.type) {
+                    case 'text-start':
+                        if (!textById.has(part.id)) {
+                            textById.set(part.id, '');
+                            textOrder.push(part.id);
+                        }
+                        break;
+                    case 'text-delta':
+                        if (!textById.has(part.id))
+                            textOrder.push(part.id);
+                        textById.set(part.id, (textById.get(part.id) ?? '') + part.delta);
+                        onDelta({ type: 'text', content: part.delta });
+                        break;
+                    case 'reasoning-delta':
+                        onDelta({ type: 'reasoning', content: part.delta });
+                        break;
+                    case 'tool-input-start':
+                        toolInputBuffers.set(part.id, {
+                            name: part.toolName,
+                            input: '',
+                            providerExecuted: part.providerExecuted === true,
+                        });
+                        break;
+                    case 'tool-input-delta': {
+                        const buffer = toolInputBuffers.get(part.id);
+                        if (buffer) {
+                            buffer.input += part.delta;
+                            if (!buffer.providerExecuted) {
+                                onDelta({ type: 'tool-input', content: part.delta, toolCallId: part.id, toolName: buffer.name });
+                            }
+                        }
+                        break;
+                    }
+                    case 'tool-input-end': {
+                        const buffer = toolInputBuffers.get(part.id);
+                        if (buffer && !buffer.providerExecuted) {
+                            toolCalls.push({ id: part.id, name: buffer.name, arguments: buffer.input || '{}' });
+                        }
+                        toolInputBuffers.delete(part.id);
+                        break;
+                    }
+                    case 'tool-call':
+                        if (!part.providerExecuted && !toolCalls.some((toolCall) => toolCall.id === part.toolCallId)) {
+                            toolCalls.push({ id: part.toolCallId, name: part.toolName, arguments: typeof part.input === 'string' ? part.input : JSON.stringify(part.input ?? {}) });
+                        }
+                        break;
+                    case 'file':
+                        content.push(fromAISdkFile(part));
+                        break;
+                    case 'finish':
+                        finishReason = mapFinishReason(part.finishReason.unified);
+                        usage = extractUsage(part.usage);
+                        break;
+                    case 'error':
+                        throw part.error;
+                    default:
+                        break;
+                }
+            }
+        }
+        finally {
+            reader.releaseLock();
+        }
+        const text = textOrder.map((id) => textById.get(id) ?? '').join('');
+        if (text)
+            content.unshift({ type: 'text', text });
+        return {
+            message: {
+                role: 'assistant',
+                content: content.length > 1 || content.some((part) => part.type === 'file') ? content : text,
+                ...(toolCalls.length > 0 && { toolCalls }),
+            },
+            finishReason,
+            ...(usage && { usage }),
+        };
+    }
+    toCallOptions(request) {
+        const prompt = this.toAISdkPrompt(request.messages);
+        if (request.systemPrompt)
+            prompt.unshift({ role: 'system', content: request.systemPrompt });
+        return {
+            prompt,
+            tools: request.tools?.map(toAISdkTool),
+            ...(request.temperature !== undefined && { temperature: request.temperature }),
+            ...(request.maxTokens !== undefined && { maxOutputTokens: request.maxTokens }),
+            ...(request.signal && { abortSignal: request.signal }),
+        };
+    }
+    toAISdkPrompt(messages) {
+        return messages.map((message) => {
+            if (message.role === 'system') {
+                return { role: 'system', content: messageContentToText(message.content) };
+            }
+            if (message.role === 'user') {
+                return { role: 'user', content: toAISdkContent(message.content) };
+            }
+            if (message.role === 'assistant') {
+                const content = [
+                    ...toAISdkContent(message.content),
+                ];
+                for (const toolCall of message.toolCalls ?? []) {
+                    content.push({
+                        type: 'tool-call',
+                        toolCallId: toolCall.id,
+                        toolName: toolCall.name,
+                        input: toolCall.arguments ? JSON.parse(toolCall.arguments) : {},
+                    });
+                }
+                return { role: 'assistant', content };
+            }
+            return {
+                role: 'tool',
+                content: [{
+                        type: 'tool-result',
+                        toolCallId: message.toolCallId ?? '',
+                        toolName: message.name ?? '',
+                        output: { type: 'text', value: messageContentToText(message.content) },
+                    }],
+            };
+        });
+    }
+    fromAISdkResult(result) {
+        const messageContent = [];
+        const toolCalls = [];
+        for (const part of result.content) {
+            if (part.type === 'text') {
+                messageContent.push({ type: 'text', text: part.text });
+            }
+            else if (part.type === 'file') {
+                messageContent.push(fromAISdkFile(part));
+            }
+            else if (part.type === 'tool-call' && !part.providerExecuted) {
+                toolCalls.push({
+                    id: part.toolCallId,
+                    name: part.toolName,
+                    arguments: typeof part.input === 'string' ? part.input : JSON.stringify(part.input ?? {}),
+                });
+            }
+        }
+        const text = messageContent
+            .filter((part) => part.type === 'text')
+            .map((part) => part.text)
+            .join('');
+        return {
+            message: {
+                role: 'assistant',
+                content: messageContent.some((part) => part.type === 'file') ? messageContent : text,
+                ...(toolCalls.length > 0 && { toolCalls }),
+            },
+            finishReason: mapFinishReason(result.finishReason.unified),
+            usage: extractUsage(result.usage),
+        };
+    }
+}
+function toAISdkContent(content) {
+    if (typeof content === 'string')
+        return content ? [{ type: 'text', text: content }] : [];
+    return content.map((part) => {
+        if (part.type === 'text')
+            return part;
+        return {
+            type: 'file',
+            ...(part.filename && { filename: part.filename }),
+            mediaType: part.mediaType,
+            data: toAISdkFileData(part.data),
+        };
+    });
+}
+function toAISdkFileData(data) {
+    if (data.type === 'url')
+        return { type: 'url', url: new URL(data.url) };
+    return data;
+}
+function fromAISdkFile(file) {
+    return {
+        type: 'file',
+        mediaType: file.mediaType,
+        data: file.data.type === 'url'
+            ? { type: 'url', url: file.data.url.toString() }
+            : file.data,
+    };
+}
+function extractUsage(usage) {
+    const result = {
+        promptTokens: usage.inputTokens.total ?? 0,
+        completionTokens: usage.outputTokens.total ?? 0,
+    };
+    if (usage.inputTokens.cacheRead)
+        result.cachedPromptTokens = usage.inputTokens.cacheRead;
+    if (usage.inputTokens.cacheWrite)
+        result.cacheWriteTokens = usage.inputTokens.cacheWrite;
+    if (usage.outputTokens.reasoning)
+        result.reasoningTokens = usage.outputTokens.reasoning;
+    return result;
+}
+function toAISdkTool(definition) {
+    return {
+        type: 'function',
+        name: definition.function.name,
+        ...(definition.function.description && { description: definition.function.description }),
+        inputSchema: definition.function.parameters,
+    };
+}
+function mapFinishReason(reason) {
+    switch (reason) {
+        case 'tool-calls':
+            return 'tool_calls';
+        case 'length':
+            return 'length';
+        case 'content-filter':
+            return 'content_filter';
+        default:
+            return 'stop';
+    }
+}
+//# sourceMappingURL=ai-sdk-model.js.map

package/dist/ai-sdk-model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-sdk-model.js","sourceRoot":"","sources":["../src/ai-sdk-model.ts"],"names":[],"mappings":"AAaA,OAAO,EACL,oBAAoB,GAOrB,MAAM,YAAY,CAAC;AAEpB;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,UAAU;IACQ,KAAK;IAAlC,YAA6B,KAAsB;qBAAtB,KAAK;IAAoB,CAAC;IAEvD,KAAK,CAAC,QAAQ,CAAC,OAAqB;QAClC,IAAI,OAAO,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACzD,OAAO,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,OAAqB;QAC/C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;QACxE,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,OAAqB;QAChD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAQ,CAAC;QACjC,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1E,MAAM,OAAO,GAA4D,EAAE,CAAC;QAC5E,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC3C,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,IAAI,YAAY,GAAiB,MAAM,CAAC;QACxC,IAAI,KAAsD,CAAC;QAE3D,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAsE,CAAC;QAEvG,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;QAClC,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAClD,IAAI,IAAI;oBAAE,MAAM;gBAEhB,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;oBAClB,KAAK,YAAY;wBACf,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;4BAC3B,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;4BAC1B,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBAC1B,CAAC;wBACD,MAAM;oBACR,KAAK,YAAY;wBACf,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;4BAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBACpD,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;wBAClE,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;wBAC/C,MAAM;oBACR,KAAK,iBAAiB;wBACpB,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;wBACpD,MAAM;oBACR,KAAK,kBAAkB;wBACrB,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;4BAC5B,IAAI,EAAE,IAAI,CAAC,QAAQ;4BACnB,KAAK,EAAE,EAAE;4BACT,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,KAAK,IAAI;yBACjD,CAAC,CAAC;wBACH,MAAM;oBACR,KAAK,kBAAkB,EAAE,CAAC;wBACxB,MAAM,MAAM,GAAG,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBAC7C,IAAI,MAAM,EAAE,CAAC;4BACX,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;4BAC3B,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;gCAC7B,OAAO,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;4BACnG,CAAC;wBACH,CAAC;wBACD,MAAM;oBACR,CAAC;oBACD,KAAK,gBAAgB,EAAE,CAAC;wBACtB,MAAM,MAAM,GAAG,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBAC7C,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;4BACvC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,SAAS,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI,EAAE,CAAC,CAAC;wBACtF,CAAC;wBACD,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBACjC,MAAM;oBACR,CAAC;oBACD,KAAK,WAAW;wBACd,IAAI,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;4BAC7F,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;wBAC1J,CAAC;wBACD,MAAM;oBACR,KAAK,MAAM;wBACT,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;wBAClC,MAAM;oBACR,KAAK,QAAQ;wBACX,YAAY,GAAG,eAAe,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;wBAC1D,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBACjC,MAAM;oBACR,KAAK,OAAO;wBACV,MAAM,IAAI,CAAC,KAAK,CAAC;oBACnB;wBACE,MAAM;gBACV,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;QAED,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpE,IAAI,IAAI;YAAE,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAElD,OAAO;YACL,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;gBAC5F,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;aAC3C;YACD,YAAY;YACZ,GAAG,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,CAAC;SACxB,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,OAAqB;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,OAAO,CAAC,YAAY;YAAE,MAAM,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;QAC5F,OAAO;YACL,MAAM;YACN,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,WAAW,CAAC;YACtC,GAAG,CAAC,OAAO,CAAC,WAAW,KAAK,SAAS,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,WAAW,EAAE,CAAC;YAC9E,GAAG,CAAC,OAAO,CAAC,SAAS,KAAK,SAAS,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;YAC9E,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;SACvD,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,QAAmB;QACvC,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;YAC9B,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC9B,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,oBAAoB,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5E,CAAC;YAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACpE,CAAC;YAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACjC,MAAM,OAAO,GAA2F;oBACtG,GAAG,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC;iBACnC,CAAC;gBACF,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,SAAS,IAAI,EAAE,EAAE,CAAC;oBAC/C,OAAO,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,WAAW;wBACjB,UAAU,EAAE,QAAQ,CAAC,EAAE;wBACvB,QAAQ,EAAE,QAAQ,CAAC,IAAI;wBACvB,KAAK,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE;qBAChE,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;YACxC,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC;wBACR,IAAI,EAAE,aAAa;wBACnB,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,EAAE;wBACpC,QAAQ,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;wBAC5B,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,oBAAoB,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;qBACvE,CAAC;aACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,MAAqC;QAC3D,MAAM,cAAc,GAA4D,EAAE,CAAC;QACnF,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACzB,cAAc,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YACzD,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAChC,cAAc,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3C,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBAC/D,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,IAAI,CAAC,UAAU;oBACnB,IAAI,EAAE,IAAI,CAAC,QAAQ;oBACnB,SAAS,EAAE,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;iBAC1F,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,MAAM,IAAI,GAAG,cAAc;aACxB,MAAM,CAAC,CAAC,IAAI,EAA0C,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;aAC9E,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACxB,IAAI,CAAC,EAAE,CAAC,CAAC;QAEZ,OAAO;YACL,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI;gBACpF,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;aAC3C;YACD,YAAY,EAAE,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC;YAC1D,KAAK,EAAE,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC;SAClC,CAAC;IACJ,CAAC;CACF;AAED,SAAS,cAAc,CAAC,OAAuB;IAC7C,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACzF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAC1B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,IAAI,CAAC;QACtC,OAAO;YACL,IAAI,EAAE,MAAM;YACZ,GAAG,CAAC,IAAI,CAAC,QAAQ,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjD,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,IAAI,EAAE,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC;SACjC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,IAAqB;IAC5C,IAAI,IAAI,CAAC,IAAI,KAAK,KAAK;QAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;IACxE,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,aAAa,CAAC,IAAyB;IAC9C,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,KAAK;YAC5B,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE;YAChD,CAAC,CAAC,IAAI,CAAC,IAAI;KACd,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAA2B;IAC/C,MAAM,MAAM,GAAwC;QAClD,YAAY,EAAE,KAAK,CAAC,WAAW,CAAC,KAAK,IAAI,CAAC;QAC1C,gBAAgB,EAAE,KAAK,CAAC,YAAY,CAAC,KAAK,IAAI,CAAC;KAChD,CAAC;IACF,IAAI,KAAK,CAAC,WAAW,CAAC,SAAS;QAAE,MAAM,CAAC,kBAAkB,GAAG,KAAK,CAAC,WAAW,CAAC,SAAS,CAAC;IACzF,IAAI,KAAK,CAAC,WAAW,CAAC,UAAU;QAAE,MAAM,CAAC,gBAAgB,GAAG,KAAK,CAAC,WAAW,CAAC,UAAU,CAAC;IACzF,IAAI,KAAK,CAAC,YAAY,CAAC,SAAS;QAAE,MAAM,CAAC,eAAe,GAAG,KAAK,CAAC,YAAY,CAAC,SAAS,CAAC;IACxF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,WAAW,CAAC,UAA0B;IAC7C,OAAO;QACL,IAAI,EAAE,UAAU;QAChB,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,IAAI;QAC9B,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,UAAU,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;QACxF,WAAW,EAAE,UAAU,CAAC,QAAQ,CAAC,UAAwD;KAC1F,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,MAAgE;IACvF,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,YAAY;YACf,OAAO,YAAY,CAAC;QACtB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,gBAAgB;YACnB,OAAO,gBAAgB,CAAC;QAC1B;YACE,OAAO,MAAM,CAAC;IAClB,CAAC;AACH,CAAC"}

package/dist/context.d.ts

@@ -0,0 +1,75 @@
+import type { Message, MessageContent, ToolCall } from './types.js';
+import type { Tool } from './tool.js';
+/**
+ * The request that initiates an agent turn.
+ */
+export interface AgentRequest {
+    /** The user's text or multimodal message. */
+    message: MessageContent;
+    /** Override the agent's system prompt for this turn. */
+    systemPrompt?: string;
+    /** Additional tools for this turn (merged with agent-level tools). */
+    tools?: Tool[];
+    /** Extra metadata available to hooks/extensions throughout the turn. */
+    metadata?: Record<string, unknown>;
+    /** Abort the turn by aborting this signal. */
+    signal?: AbortSignal;
+    /** Max LLM iterations before the turn is stopped. Default: 25. */
+    maxIterations?: number;
+}
+/** Why a turn ended. */
+export type TurnFinishReason = 'stop' | 'tool_calls' | 'halted' | 'max_iterations' | 'error';
+/**
+ * The result of an agent turn.
+ */
+export interface AgentResponse {
+    role: 'assistant';
+    /** Final text response from the model (may be empty if halted). */
+    message: string;
+    /** Tool calls made during the turn (for observation/logging). */
+    toolCalls?: ToolCall[];
+    /**
+     * When `finishReason` is `'tool_calls'`, these are the external tool calls
+     * the caller must execute and send back as `tool` messages in a follow-up
+     * request. Internal tool calls (already executed) are in `toolCalls`.
+     */
+    pendingToolCalls?: ToolCall[];
+    /** Why the turn ended. */
+    finishReason: TurnFinishReason;
+    /** Number of LLM iterations executed. */
+    iterations: number;
+    /** Shared metadata bag. */
+    metadata: Record<string, unknown>;
+}
+/**
+ * Mutable execution state for a single turn.
+ *
+ * Extensions receive this through hooks and may mutate it to influence
+ * the turn. This represents **execution state**, not the whole application.
+ * Do not use it as a service locator.
+ */
+export interface TurnContext {
+    request: AgentRequest;
+    /** The conversation messages for this turn (mutable). */
+    messages: Message[];
+    /** Tools available this turn (mutable). */
+    tools: Tool[];
+    /**
+     * The resolved system prompt for this turn (per-turn override ?? agent config).
+     * Available from beforeTurn onward — extensions can read it to audit
+     * what instructions the model will operate under without waiting for
+     * beforeLLM. May be undefined if no system prompt is configured.
+     */
+    systemPrompt?: string;
+    /** The final response once the turn completes. */
+    response?: AgentResponse;
+    /** Shared metadata bag for extensions. */
+    metadata: Record<string, unknown>;
+    /** AbortSignal for this turn. */
+    signal: AbortSignal;
+    /** Inject a steering message into this turn before the next LLM iteration. */
+    steer(message: string): void;
+    /** Halt this turn. Resolves with finishReason 'halted'. */
+    halt(reason?: string): void;
+}
+//# sourceMappingURL=context.d.ts.map

package/dist/context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../src/context.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEtC;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,6CAA6C;IAC7C,OAAO,EAAE,cAAc,CAAC;IACxB,wDAAwD;IACxD,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,sEAAsE;IACtE,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,wEAAwE;IACxE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,8CAA8C;IAC9C,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,kEAAkE;IAClE,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,wBAAwB;AACxB,MAAM,MAAM,gBAAgB,GACxB,MAAM,GACN,YAAY,GACZ,QAAQ,GACR,gBAAgB,GAChB,OAAO,CAAC;AAEZ;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,mEAAmE;IACnE,OAAO,EAAE,MAAM,CAAC;IAChB,iEAAiE;IACjE,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;IACvB;;;;OAIG;IACH,gBAAgB,CAAC,EAAE,QAAQ,EAAE,CAAC;IAC9B,0BAA0B;IAC1B,YAAY,EAAE,gBAAgB,CAAC;IAC/B,yCAAyC;IACzC,UAAU,EAAE,MAAM,CAAC;IACnB,2BAA2B;IAC3B,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED;;;;;;GAMG;AACH,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,YAAY,CAAC;IACtB,yDAAyD;IACzD,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,2CAA2C;IAC3C,KAAK,EAAE,IAAI,EAAE,CAAC;IACd;;;;;OAKG;IACH,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,kDAAkD;IAClD,QAAQ,CAAC,EAAE,aAAa,CAAC;IACzB,0CAA0C;IAC1C,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,iCAAiC;IACjC,MAAM,EAAE,WAAW,CAAC;IACpB,8EAA8E;IAC9E,KAAK,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,2DAA2D;IAC3D,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7B"}

package/dist/context.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=context.js.map

package/dist/context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../src/context.ts"],"names":[],"mappings":""}

package/dist/errors.d.ts

@@ -0,0 +1,33 @@
+/** Base error class for all libra errors. */
+export declare class LibraError extends Error {
+    constructor(message: string);
+}
+/** The model call failed. */
+export declare class ModelError extends LibraError {
+    readonly status?: number | undefined;
+    readonly body?: unknown;
+    constructor(message: string, status?: number | undefined, body?: unknown);
+}
+/** A tool execution failed. */
+export declare class ToolError extends LibraError {
+    readonly toolName: string;
+    constructor(message: string, toolName: string);
+}
+/** A hook threw and the failure policy aborted the turn. */
+export declare class HookError extends LibraError {
+    readonly hookName: string;
+    readonly extensionName: string;
+    readonly cause?: unknown;
+    constructor(message: string, hookName: string, extensionName: string, cause?: unknown);
+}
+/** The turn was halted (via AbortSignal or agent.halt()). */
+export declare class HaltedError extends LibraError {
+    readonly reason: string;
+    constructor(message: string, reason: string);
+}
+/** The turn exceeded the maximum number of LLM iterations. */
+export declare class MaxIterationsError extends LibraError {
+    readonly iterations: number;
+    constructor(message: string, iterations: number);
+}
+//# sourceMappingURL=errors.d.ts.map

package/dist/errors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.d.ts","sourceRoot":"","sources":["../src/errors.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,qBAAa,UAAW,SAAQ,KAAK;IACnC,YAAY,OAAO,EAAE,MAAM,EAG1B;CACF;AAED,6BAA6B;AAC7B,qBAAa,UAAW,SAAQ,UAAU;aAGtB,MAAM,CAAC,EAAE,MAAM;aACf,IAAI,CAAC,EAAE,OAAO;IAHhC,YACE,OAAO,EAAE,MAAM,EACC,MAAM,CAAC,EAAE,MAAM,YAAA,EACf,IAAI,CAAC,EAAE,OAAO,EAI/B;CACF;AAED,+BAA+B;AAC/B,qBAAa,SAAU,SAAQ,UAAU;aAGrB,QAAQ,EAAE,MAAM;IAFlC,YACE,OAAO,EAAE,MAAM,EACC,QAAQ,EAAE,MAAM,EAIjC;CACF;AAED,4DAA4D;AAC5D,qBAAa,SAAU,SAAQ,UAAU;aAGrB,QAAQ,EAAE,MAAM;aAChB,aAAa,EAAE,MAAM;aACZ,KAAK,CAAC,EAAE,OAAO;IAJ1C,YACE,OAAO,EAAE,MAAM,EACC,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,MAAM,EACZ,KAAK,CAAC,EAAE,OAAO,EAIzC;CACF;AAED,6DAA6D;AAC7D,qBAAa,WAAY,SAAQ,UAAU;aAGvB,MAAM,EAAE,MAAM;IAFhC,YACE,OAAO,EAAE,MAAM,EACC,MAAM,EAAE,MAAM,EAI/B;CACF;AAED,8DAA8D;AAC9D,qBAAa,kBAAmB,SAAQ,UAAU;aAG9B,UAAU,EAAE,MAAM;IAFpC,YACE,OAAO,EAAE,MAAM,EACC,UAAU,EAAE,MAAM,EAInC;CACF"}

package/dist/errors.js

@@ -0,0 +1,59 @@
+/** Base error class for all libra errors. */
+export class LibraError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'LibraError';
+    }
+}
+/** The model call failed. */
+export class ModelError extends LibraError {
+    status;
+    body;
+    constructor(message, status, body) {
+        super(message);
+        this.status = status;
+        this.body = body;
+        this.name = 'ModelError';
+    }
+}
+/** A tool execution failed. */
+export class ToolError extends LibraError {
+    toolName;
+    constructor(message, toolName) {
+        super(message);
+        this.toolName = toolName;
+        this.name = 'ToolError';
+    }
+}
+/** A hook threw and the failure policy aborted the turn. */
+export class HookError extends LibraError {
+    hookName;
+    extensionName;
+    cause;
+    constructor(message, hookName, extensionName, cause) {
+        super(message);
+        this.hookName = hookName;
+        this.extensionName = extensionName;
+        this.cause = cause;
+        this.name = 'HookError';
+    }
+}
+/** The turn was halted (via AbortSignal or agent.halt()). */
+export class HaltedError extends LibraError {
+    reason;
+    constructor(message, reason) {
+        super(message);
+        this.reason = reason;
+        this.name = 'HaltedError';
+    }
+}
+/** The turn exceeded the maximum number of LLM iterations. */
+export class MaxIterationsError extends LibraError {
+    iterations;
+    constructor(message, iterations) {
+        super(message);
+        this.iterations = iterations;
+        this.name = 'MaxIterationsError';
+    }
+}
+//# sourceMappingURL=errors.js.map

package/dist/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../src/errors.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,MAAM,OAAO,UAAW,SAAQ,KAAK;IACnC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC;IAC3B,CAAC;CACF;AAED,6BAA6B;AAC7B,MAAM,OAAO,UAAW,SAAQ,UAAU;IAGtB,MAAM;IACN,IAAI;IAHtB,YACE,OAAe,EACC,MAAe,EACf,IAAc;QAE9B,KAAK,CAAC,OAAO,CAAC,CAAC;sBAHC,MAAM;oBACN,IAAI;QAGpB,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC;IAC3B,CAAC;CACF;AAED,+BAA+B;AAC/B,MAAM,OAAO,SAAU,SAAQ,UAAU;IAGrB,QAAQ;IAF1B,YACE,OAAe,EACC,QAAgB;QAEhC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAFC,QAAQ;QAGxB,IAAI,CAAC,IAAI,GAAG,WAAW,CAAC;IAC1B,CAAC;CACF;AAED,4DAA4D;AAC5D,MAAM,OAAO,SAAU,SAAQ,UAAU;IAGrB,QAAQ;IACR,aAAa;IACJ,KAAK;IAJhC,YACE,OAAe,EACC,QAAgB,EAChB,aAAqB,EACZ,KAAe;QAExC,KAAK,CAAC,OAAO,CAAC,CAAC;wBAJC,QAAQ;6BACR,aAAa;qBACJ,KAAK;QAG9B,IAAI,CAAC,IAAI,GAAG,WAAW,CAAC;IAC1B,CAAC;CACF;AAED,6DAA6D;AAC7D,MAAM,OAAO,WAAY,SAAQ,UAAU;IAGvB,MAAM;IAFxB,YACE,OAAe,EACC,MAAc;QAE9B,KAAK,CAAC,OAAO,CAAC,CAAC;sBAFC,MAAM;QAGtB,IAAI,CAAC,IAAI,GAAG,aAAa,CAAC;IAC5B,CAAC;CACF;AAED,8DAA8D;AAC9D,MAAM,OAAO,kBAAmB,SAAQ,UAAU;IAG9B,UAAU;IAF5B,YACE,OAAe,EACC,UAAkB;QAElC,KAAK,CAAC,OAAO,CAAC,CAAC;0BAFC,UAAU;QAG1B,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC;IACnC,CAAC;CACF"}

package/dist/extension.d.ts

@@ -0,0 +1,63 @@
+import type { Agent } from './agent.js';
+/**
+ * An extension adds behavior to an agent without requiring the core
+ * to understand that behavior.
+ *
+ * `install` is called when the extension is added via `agent.use()`.
+ * The extension typically registers hooks on the agent inside `install`.
+ *
+ * Example:
+ * ```typescript
+ * const loggingExtension: Extension = {
+ *   name: 'logging',
+ *   install(agent) {
+ *     agent.hook('beforeLLM', 'logging', async (ctx) => {
+ *       console.log('LLM call with', ctx.turn.messages.length, 'messages')
+ *     })
+ *   },
+ * }
+ * ```
+ */
+export interface Extension {
+    /** Unique name for this extension (used in error reporting and ordering). */
+    name: string;
+    /** Called when the extension is installed via `agent.use()`. */
+    install(agent: Agent): void;
+    /**
+     * Optional priority that controls hook execution order. Higher =
+     * hooks run first within each lifecycle stage. Extensions with the
+     * same priority retain registration order (stable sort). Default: 0.
+     *
+     * This applies to **all** registration paths:
+     * - `agent.use(extension)` — hooks registered inside `install()`
+     *   inherit the extension's priority.
+     * - The extension loader (`loadExtensions` / `installExtensions`) —
+     *   sorts extensions by priority before calling `use()`, and the
+     *   priority is also passed through to each hook.
+     * - Direct `agent.hook()` calls outside `use()` — look up the
+     *   priority of an installed extension with a matching name, or
+     *   default to 0.
+     *
+     * For directory-loaded extensions, the `priority` in
+     * `extension.json` takes precedence over this field.
+     *
+     * Use this to ensure enrichment extensions (e.g. keyword extraction)
+     * run before persistence extensions (e.g. session) without relying on
+     * `use()` call order:
+     *
+     * ```typescript
+     * const keywordLogger = { name: 'keyword-logger', priority: 10, install(a) { ... } };
+     * const session = { name: 'session', priority: -100, install(a) { ... } };
+     * // Hooks run keyword-logger first regardless of use() order.
+     * agent.use(session).use(keywordLogger);
+     * ```
+     */
+    priority?: number;
+    /**
+     * Optional cleanup method called when the extension is unloaded or
+     * the agent shuts down. Use this to close connections, stop
+     * processes, release resources, etc.
+     */
+    close?(): Promise<void>;
+}
+//# sourceMappingURL=extension.d.ts.map

package/dist/extension.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"extension.d.ts","sourceRoot":"","sources":["../src/extension.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAExC;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,WAAW,SAAS;IACxB,6EAA6E;IAC7E,IAAI,EAAE,MAAM,CAAC;IACb,gEAAgE;IAChE,OAAO,CAAC,KAAK,EAAE,KAAK,GAAG,IAAI,CAAC;IAC5B;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4BG;IACH,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;;OAIG;IACH,KAAK,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACzB"}

package/dist/extension.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=extension.js.map

package/dist/extension.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extension.js","sourceRoot":"","sources":["../src/extension.ts"],"names":[],"mappings":""}