@usestratus/sdk 0.12.0

package/dist/core/codemode/index.js

@@ -0,0 +1,104 @@
+/**
+ * Code mode: let LLMs write code that orchestrates tools instead of calling them one at a time.
+ *
+ * Inspired by Cloudflare's Code Mode and CodeAct — LLMs are better at writing code
+ * than making individual tool calls because they've seen millions of lines of real-world
+ * TypeScript but only contrived tool-calling examples.
+ */
+import { z } from "zod";
+import { isFunctionTool } from "../hosted-tool";
+import { tool } from "../tool";
+import { generateTypes, normalizeCode, sanitizeToolName } from "./types";
+export { FunctionExecutor, WorkerExecutor } from "./executor";
+export { generateTypes, normalizeCode, sanitizeToolName } from "./types";
+// ── Default description ────────────────────────────────────────────
+const DEFAULT_DESCRIPTION = `Execute code to achieve a goal.
+
+Available:
+{{types}}
+
+Write an async arrow function in JavaScript that returns the result.
+Do NOT use TypeScript syntax — no type annotations, interfaces, or generics.
+Do NOT define named functions then call them — just write the arrow function body directly.
+
+Example: async () => { const r = await codemode.searchWeb({ query: "test" }); return r; }`;
+const codeSchema = z.object({
+    code: z.string().describe("JavaScript async arrow function to execute"),
+});
+/**
+ * Create a code mode tool that allows LLMs to write and execute code
+ * with access to your tools in a sandboxed environment.
+ *
+ * Returns a Stratus `FunctionTool` that can be added to any agent's tools array.
+ *
+ * @example
+ * ```ts
+ * import { Agent, tool } from "stratus";
+ * import { createCodeModeTool, FunctionExecutor } from "stratus/core/codemode";
+ *
+ * const weatherTool = tool({
+ *   name: "get_weather",
+ *   description: "Get weather for a location",
+ *   parameters: z.object({ location: z.string() }),
+ *   execute: async (ctx, { location }) => `72°F, sunny in ${location}`,
+ * });
+ *
+ * const executor = new FunctionExecutor({ timeout: 10_000 });
+ * const codemode = createCodeModeTool({
+ *   tools: [weatherTool],
+ *   executor,
+ * });
+ *
+ * const agent = new Agent({
+ *   name: "assistant",
+ *   instructions: "You are a helpful assistant.",
+ *   tools: [codemode],
+ * });
+ * ```
+ */
+export function createCodeModeTool(options) {
+    // Filter to only function tools (hosted tools can't be called locally)
+    const functionTools = options.tools.filter(isFunctionTool);
+    const types = generateTypes(functionTools);
+    const executor = options.executor;
+    return tool({
+        name: "execute_code",
+        description: (options.description ?? DEFAULT_DESCRIPTION).replace("{{types}}", types),
+        parameters: codeSchema,
+        async execute(context, { code }) {
+            // Build the function map for the sandbox
+            const fns = {};
+            for (const t of functionTools) {
+                const safeName = sanitizeToolName(t.name);
+                fns[safeName] = async (args) => {
+                    const validated = t.parameters.parse(args);
+                    const result = await t.execute(context, validated);
+                    // Try to parse as JSON, fall back to raw string
+                    try {
+                        return JSON.parse(result);
+                    }
+                    catch {
+                        return result;
+                    }
+                };
+            }
+            const normalizedCode = normalizeCode(code);
+            const executeResult = await executor.execute(normalizedCode, fns);
+            if (executeResult.error) {
+                const logCtx = executeResult.logs?.length
+                    ? `\n\nConsole output:\n${executeResult.logs.join("\n")}`
+                    : "";
+                throw new Error(`Code execution failed: ${executeResult.error}${logCtx}`);
+            }
+            const output = {
+                code,
+                result: executeResult.result,
+            };
+            if (executeResult.logs?.length) {
+                output.logs = executeResult.logs;
+            }
+            return JSON.stringify(output);
+        },
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/core/codemode/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/codemode/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAI/B,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAGzE,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE9D,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAEzE,sEAAsE;AAEtE,MAAM,mBAAmB,GAAG;;;;;;;;;0FAS8D,CAAC;AAmB3F,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;CACvE,CAAC,CAAC;AAEH;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,UAAU,kBAAkB,CACjC,OAA4B;IAE5B,uEAAuE;IACvE,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;IAC3D,MAAM,KAAK,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;IAC3C,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IAElC,OAAO,IAAI,CAAC;QACX,IAAI,EAAE,cAAc;QACpB,WAAW,EAAE,CAAC,OAAO,CAAC,WAAW,IAAI,mBAAmB,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC;QACrF,UAAU,EAAE,UAAU;QACtB,KAAK,CAAC,OAAO,CAAC,OAAiB,EAAE,EAAE,IAAI,EAAoB;YAC1D,yCAAyC;YACzC,MAAM,GAAG,GAA6D,EAAE,CAAC;YACzE,KAAK,MAAM,CAAC,IAAI,aAAa,EAAE,CAAC;gBAC/B,MAAM,QAAQ,GAAG,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBAC1C,GAAG,CAAC,QAAQ,CAAC,GAAG,KAAK,EAAE,IAAa,EAAE,EAAE;oBACvC,MAAM,SAAS,GAAG,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAC3C,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;oBACnD,gDAAgD;oBAChD,IAAI,CAAC;wBACJ,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;oBAC3B,CAAC;oBAAC,MAAM,CAAC;wBACR,OAAO,MAAM,CAAC;oBACf,CAAC;gBACF,CAAC,CAAC;YACH,CAAC;YAED,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;YAC3C,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;YAElE,IAAI,aAAa,CAAC,KAAK,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,EAAE,MAAM;oBACxC,CAAC,CAAC,wBAAwB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;oBACzD,CAAC,CAAC,EAAE,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,0BAA0B,aAAa,CAAC,KAAK,GAAG,MAAM,EAAE,CAAC,CAAC;YAC3E,CAAC;YAED,MAAM,MAAM,GAAmB;gBAC9B,IAAI;gBACJ,MAAM,EAAE,aAAa,CAAC,MAAM;aAC5B,CAAC;YACF,IAAI,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;gBAChC,MAAM,CAAC,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC;YAClC,CAAC;YACD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC;KACD,CAAC,CAAC;AACJ,CAAC"}

package/dist/core/codemode/types.d.ts

@@ -0,0 +1,24 @@
+/**
+ * Code mode: type generation and code normalization utilities.
+ * Adapted from @cloudflare/codemode for the Stratus SDK.
+ */
+import type { FunctionTool } from "../tool";
+/**
+ * Sanitize a tool name into a valid JavaScript identifier.
+ * Replaces hyphens, dots, and spaces with `_`, strips other invalid chars,
+ * prefixes digit-leading names with `_`, and appends `_` to JS reserved words.
+ */
+export declare function sanitizeToolName(name: string): string;
+/**
+ * Generate TypeScript type definitions from an array of FunctionTools.
+ * These types are included in the LLM's system prompt so it can write
+ * correct code against the `codemode.*` API.
+ */
+export declare function generateTypes(tools: FunctionTool[]): string;
+/**
+ * Normalize LLM-generated code into an async arrow function.
+ * Handles common LLM patterns: bare statements, function declarations,
+ * code fences, and expression statements.
+ */
+export declare function normalizeCode(code: string): string;
+//# sourceMappingURL=types.d.ts.map

package/dist/core/codemode/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/core/codemode/types.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AA2E5C;;;;GAIG;AACH,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAQrD;AAoQD;;;;GAIG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,YAAY,EAAE,GAAG,MAAM,CA6D3D;AAYD;;;;GAIG;AACH,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAYlD"}

package/dist/core/codemode/types.js

@@ -0,0 +1,405 @@
+/**
+ * Code mode: type generation and code normalization utilities.
+ * Adapted from @cloudflare/codemode for the Stratus SDK.
+ */
+import { zodToJsonSchema } from "../utils/zod";
+// ── JS reserved words ──────────────────────────────────────────────
+const JS_RESERVED = new Set([
+    "abstract",
+    "arguments",
+    "await",
+    "boolean",
+    "break",
+    "byte",
+    "case",
+    "catch",
+    "char",
+    "class",
+    "const",
+    "continue",
+    "debugger",
+    "default",
+    "delete",
+    "do",
+    "double",
+    "else",
+    "enum",
+    "eval",
+    "export",
+    "extends",
+    "false",
+    "final",
+    "finally",
+    "float",
+    "for",
+    "function",
+    "goto",
+    "if",
+    "implements",
+    "import",
+    "in",
+    "instanceof",
+    "int",
+    "interface",
+    "let",
+    "long",
+    "native",
+    "new",
+    "null",
+    "package",
+    "private",
+    "protected",
+    "public",
+    "return",
+    "short",
+    "static",
+    "super",
+    "switch",
+    "synchronized",
+    "this",
+    "throw",
+    "throws",
+    "transient",
+    "true",
+    "try",
+    "typeof",
+    "undefined",
+    "var",
+    "void",
+    "volatile",
+    "while",
+    "with",
+    "yield",
+]);
+// ── Name sanitization ──────────────────────────────────────────────
+/**
+ * Sanitize a tool name into a valid JavaScript identifier.
+ * Replaces hyphens, dots, and spaces with `_`, strips other invalid chars,
+ * prefixes digit-leading names with `_`, and appends `_` to JS reserved words.
+ */
+export function sanitizeToolName(name) {
+    if (!name)
+        return "_";
+    let sanitized = name.replace(/[-.\s]/g, "_");
+    sanitized = sanitized.replace(/[^a-zA-Z0-9_$]/g, "");
+    if (!sanitized)
+        return "_";
+    if (/^[0-9]/.test(sanitized))
+        sanitized = `_${sanitized}`;
+    if (JS_RESERVED.has(sanitized))
+        sanitized = `${sanitized}_`;
+    return sanitized;
+}
+function toCamelCase(str) {
+    return str
+        .replace(/_([a-z])/g, (_, letter) => letter.toUpperCase())
+        .replace(/^[a-z]/, (letter) => letter.toUpperCase());
+}
+// ── String escaping ────────────────────────────────────────────────
+function escapeControlChar(ch) {
+    const code = ch.charCodeAt(0);
+    if (code <= 31 || code === 127)
+        return `\\u${code.toString(16).padStart(4, "0")}`;
+    return ch;
+}
+function escapeStringLiteral(s) {
+    let out = "";
+    for (const ch of s) {
+        if (ch === "\\")
+            out += "\\\\";
+        else if (ch === '"')
+            out += '\\"';
+        else if (ch === "\n")
+            out += "\\n";
+        else if (ch === "\r")
+            out += "\\r";
+        else if (ch === "\t")
+            out += "\\t";
+        else if (ch === "\u2028")
+            out += "\\u2028";
+        else if (ch === "\u2029")
+            out += "\\u2029";
+        else
+            out += escapeControlChar(ch);
+    }
+    return out;
+}
+function escapeJsDoc(text) {
+    return text.replace(/\*\//g, "*\\/");
+}
+function needsQuotes(name) {
+    return !/^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(name);
+}
+function quoteProp(name) {
+    if (needsQuotes(name)) {
+        let escaped = "";
+        for (const ch of name) {
+            if (ch === "\\")
+                escaped += "\\\\";
+            else if (ch === '"')
+                escaped += '\\"';
+            else if (ch === "\n")
+                escaped += "\\n";
+            else if (ch === "\r")
+                escaped += "\\r";
+            else if (ch === "\t")
+                escaped += "\\t";
+            else if (ch === "\u2028")
+                escaped += "\\u2028";
+            else if (ch === "\u2029")
+                escaped += "\\u2029";
+            else
+                escaped += escapeControlChar(ch);
+        }
+        return `"${escaped}"`;
+    }
+    return name;
+}
+function resolveRef(ref, root) {
+    if (ref === "#")
+        return root;
+    if (!ref.startsWith("#/"))
+        return null;
+    const segments = ref
+        .slice(2)
+        .split("/")
+        .map((s) => s.replace(/~1/g, "/").replace(/~0/g, "~"));
+    let current = root;
+    for (const seg of segments) {
+        if (current === null || typeof current !== "object")
+            return null;
+        current = current[seg];
+        if (current === undefined)
+            return null;
+    }
+    if (typeof current === "boolean")
+        return current;
+    if (current === null || typeof current !== "object")
+        return null;
+    return current;
+}
+function applyNullable(result, schema) {
+    if (result !== "unknown" && result !== "never" && schema?.nullable === true)
+        return `${result} | null`;
+    return result;
+}
+function jsonSchemaToTypeString(schema, indent, ctx) {
+    if (typeof schema === "boolean")
+        return schema ? "unknown" : "never";
+    if (ctx.depth >= ctx.maxDepth)
+        return "unknown";
+    if (ctx.seen.has(schema))
+        return "unknown";
+    ctx.seen.add(schema);
+    const nextCtx = { ...ctx, depth: ctx.depth + 1 };
+    try {
+        if (schema.$ref) {
+            const resolved = resolveRef(schema.$ref, ctx.root);
+            if (!resolved)
+                return "unknown";
+            return applyNullable(jsonSchemaToTypeString(resolved, indent, nextCtx), schema);
+        }
+        if (schema.anyOf)
+            return applyNullable(schema.anyOf.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(" | "), schema);
+        if (schema.oneOf)
+            return applyNullable(schema.oneOf.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(" | "), schema);
+        if (schema.allOf)
+            return applyNullable(schema.allOf.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(" & "), schema);
+        if (schema.enum) {
+            if (schema.enum.length === 0)
+                return "never";
+            return applyNullable(schema.enum
+                .map((v) => {
+                if (v === null)
+                    return "null";
+                if (typeof v === "string")
+                    return `"${escapeStringLiteral(v)}"`;
+                if (typeof v === "object")
+                    return JSON.stringify(v) ?? "unknown";
+                return String(v);
+            })
+                .join(" | "), schema);
+        }
+        if (schema.const !== undefined) {
+            return applyNullable(schema.const === null
+                ? "null"
+                : typeof schema.const === "string"
+                    ? `"${escapeStringLiteral(schema.const)}"`
+                    : typeof schema.const === "object"
+                        ? (JSON.stringify(schema.const) ?? "unknown")
+                        : String(schema.const), schema);
+        }
+        const type = schema.type;
+        if (type === "string")
+            return applyNullable("string", schema);
+        if (type === "number" || type === "integer")
+            return applyNullable("number", schema);
+        if (type === "boolean")
+            return applyNullable("boolean", schema);
+        if (type === "null")
+            return "null";
+        if (type === "array") {
+            const prefixItems = schema.prefixItems;
+            if (Array.isArray(prefixItems))
+                return applyNullable(`[${prefixItems.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(", ")}]`, schema);
+            if (Array.isArray(schema.items))
+                return applyNullable(`[${schema.items.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(", ")}]`, schema);
+            if (schema.items)
+                return applyNullable(`${jsonSchemaToTypeString(schema.items, indent, nextCtx)}[]`, schema);
+            return applyNullable("unknown[]", schema);
+        }
+        if (type === "object" || schema.properties) {
+            const props = schema.properties || {};
+            const required = new Set(schema.required || []);
+            const lines = [];
+            for (const [propName, propSchema] of Object.entries(props)) {
+                if (typeof propSchema === "boolean") {
+                    const boolType = propSchema ? "unknown" : "never";
+                    const optionalMark = required.has(propName) ? "" : "?";
+                    lines.push(`${indent}    ${quoteProp(propName)}${optionalMark}: ${boolType};`);
+                    continue;
+                }
+                const isRequired = required.has(propName);
+                const propType = jsonSchemaToTypeString(propSchema, `${indent}    `, nextCtx);
+                const desc = propSchema.description;
+                const format = propSchema.format;
+                if (desc || format) {
+                    const descText = desc
+                        ? escapeJsDoc(desc.replace(/\r?\n/g, " "))
+                        : undefined;
+                    const formatTag = format ? `@format ${escapeJsDoc(format)}` : undefined;
+                    if (descText && formatTag) {
+                        lines.push(`${indent}    /**`);
+                        lines.push(`${indent}     * ${descText}`);
+                        lines.push(`${indent}     * ${formatTag}`);
+                        lines.push(`${indent}     */`);
+                    }
+                    else {
+                        lines.push(`${indent}    /** ${descText ?? formatTag} */`);
+                    }
+                }
+                const quotedName = quoteProp(propName);
+                const optionalMark = isRequired ? "" : "?";
+                lines.push(`${indent}    ${quotedName}${optionalMark}: ${propType};`);
+            }
+            if (schema.additionalProperties) {
+                const valueType = schema.additionalProperties === true
+                    ? "unknown"
+                    : jsonSchemaToTypeString(schema.additionalProperties, `${indent}    `, nextCtx);
+                lines.push(`${indent}    [key: string]: ${valueType};`);
+            }
+            if (lines.length === 0) {
+                if (schema.additionalProperties === false)
+                    return applyNullable("{}", schema);
+                return applyNullable("Record<string, unknown>", schema);
+            }
+            return applyNullable(`{\n${lines.join("\n")}\n${indent}}`, schema);
+        }
+        if (Array.isArray(type)) {
+            return applyNullable(type
+                .map((t) => {
+                if (t === "string")
+                    return "string";
+                if (t === "number" || t === "integer")
+                    return "number";
+                if (t === "boolean")
+                    return "boolean";
+                if (t === "null")
+                    return "null";
+                if (t === "array")
+                    return "unknown[]";
+                if (t === "object")
+                    return "Record<string, unknown>";
+                return "unknown";
+            })
+                .join(" | "), schema);
+        }
+        return "unknown";
+    }
+    finally {
+        ctx.seen.delete(schema);
+    }
+}
+// ── Public: generateTypes ──────────────────────────────────────────
+/**
+ * Generate TypeScript type definitions from an array of FunctionTools.
+ * These types are included in the LLM's system prompt so it can write
+ * correct code against the `codemode.*` API.
+ */
+export function generateTypes(tools) {
+    let availableTools = "";
+    let availableTypes = "";
+    for (const t of tools) {
+        const safeName = sanitizeToolName(t.name);
+        const camelName = toCamelCase(safeName);
+        try {
+            const jsonSchema = zodToJsonSchema(t.parameters);
+            const inputType = `type ${camelName}Input = ${jsonSchemaToTypeString(jsonSchema, "", {
+                root: jsonSchema,
+                depth: 0,
+                seen: new Set(),
+                maxDepth: 20,
+            })}`;
+            const outputType = `type ${camelName}Output = unknown`;
+            availableTypes += `\n${inputType.trim()}`;
+            availableTypes += `\n${outputType.trim()}`;
+            const jsdocLines = [];
+            if (t.description?.trim()) {
+                jsdocLines.push(escapeJsDoc(t.description.trim().replace(/\r?\n/g, " ")));
+            }
+            else {
+                jsdocLines.push(escapeJsDoc(t.name));
+            }
+            // Extract param descriptions from JSON Schema properties
+            if (jsonSchema.properties) {
+                for (const [fieldName, propSchema] of Object.entries(jsonSchema.properties)) {
+                    if (typeof propSchema === "object" &&
+                        propSchema !== null &&
+                        "description" in propSchema &&
+                        propSchema.description) {
+                        jsdocLines.push(escapeJsDoc(`@param input.${fieldName} - ${String(propSchema.description).replace(/\r?\n/g, " ")}`));
+                    }
+                }
+            }
+            const jsdocBody = jsdocLines.map((l) => `\t * ${l}`).join("\n");
+            availableTools += `\n\t/**\n${jsdocBody}\n\t */`;
+            availableTools += `\n\t${safeName}: (input: ${camelName}Input) => Promise<${camelName}Output>;`;
+            availableTools += "\n";
+        }
+        catch {
+            availableTypes += `\ntype ${camelName}Input = unknown`;
+            availableTypes += `\ntype ${camelName}Output = unknown`;
+            availableTools += `\n\t/**\n\t * ${escapeJsDoc(t.name)}\n\t */`;
+            availableTools += `\n\t${safeName}: (input: ${camelName}Input) => Promise<${camelName}Output>;`;
+            availableTools += "\n";
+        }
+    }
+    availableTools = `\ndeclare const codemode: {${availableTools}}`;
+    return `\n${availableTypes}\n${availableTools}\n`.trim();
+}
+// ── Public: normalizeCode ──────────────────────────────────────────
+/**
+ * Strip markdown code fences that LLMs commonly wrap code in.
+ */
+function stripCodeFences(code) {
+    const match = code.match(/^```(?:js|javascript|typescript|ts|tsx|jsx)?\s*\n([\s\S]*?)```\s*$/);
+    return match ? match[1] : code;
+}
+/**
+ * Normalize LLM-generated code into an async arrow function.
+ * Handles common LLM patterns: bare statements, function declarations,
+ * code fences, and expression statements.
+ */
+export function normalizeCode(code) {
+    const trimmed = stripCodeFences(code.trim());
+    if (!trimmed.trim())
+        return "async () => {}";
+    const source = trimmed.trim();
+    // Check if it's already an async arrow function
+    if (/^async\s*\(/.test(source) && source.includes("=>")) {
+        return source;
+    }
+    // Wrap bare code in an async arrow function
+    return `async () => {\n${source}\n}`;
+}
+//# sourceMappingURL=types.js.map

package/dist/core/codemode/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/core/codemode/types.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAE/C,sEAAsE;AAEtE,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC;IAC3B,UAAU;IACV,WAAW;IACX,OAAO;IACP,SAAS;IACT,OAAO;IACP,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;IACN,OAAO;IACP,OAAO;IACP,UAAU;IACV,UAAU;IACV,SAAS;IACT,QAAQ;IACR,IAAI;IACJ,QAAQ;IACR,MAAM;IACN,MAAM;IACN,MAAM;IACN,QAAQ;IACR,SAAS;IACT,OAAO;IACP,OAAO;IACP,SAAS;IACT,OAAO;IACP,KAAK;IACL,UAAU;IACV,MAAM;IACN,IAAI;IACJ,YAAY;IACZ,QAAQ;IACR,IAAI;IACJ,YAAY;IACZ,KAAK;IACL,WAAW;IACX,KAAK;IACL,MAAM;IACN,QAAQ;IACR,KAAK;IACL,MAAM;IACN,SAAS;IACT,SAAS;IACT,WAAW;IACX,QAAQ;IACR,QAAQ;IACR,OAAO;IACP,QAAQ;IACR,OAAO;IACP,QAAQ;IACR,cAAc;IACd,MAAM;IACN,OAAO;IACP,QAAQ;IACR,WAAW;IACX,MAAM;IACN,KAAK;IACL,QAAQ;IACR,WAAW;IACX,KAAK;IACL,MAAM;IACN,UAAU;IACV,OAAO;IACP,MAAM;IACN,OAAO;CACP,CAAC,CAAC;AAEH,sEAAsE;AAEtE;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAAC,IAAY;IAC5C,IAAI,CAAC,IAAI;QAAE,OAAO,GAAG,CAAC;IACtB,IAAI,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAC7C,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC,CAAC;IACrD,IAAI,CAAC,SAAS;QAAE,OAAO,GAAG,CAAC;IAC3B,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC;QAAE,SAAS,GAAG,IAAI,SAAS,EAAE,CAAC;IAC1D,IAAI,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC;QAAE,SAAS,GAAG,GAAG,SAAS,GAAG,CAAC;IAC5D,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,SAAS,WAAW,CAAC,GAAW;IAC/B,OAAO,GAAG;SACR,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,MAAc,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;SACjE,OAAO,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;AACvD,CAAC;AAED,sEAAsE;AAEtE,SAAS,iBAAiB,CAAC,EAAU;IACpC,MAAM,IAAI,GAAG,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IAC9B,IAAI,IAAI,IAAI,EAAE,IAAI,IAAI,KAAK,GAAG;QAAE,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;IAClF,OAAO,EAAE,CAAC;AACX,CAAC;AAED,SAAS,mBAAmB,CAAC,CAAS;IACrC,IAAI,GAAG,GAAG,EAAE,CAAC;IACb,KAAK,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC;QACpB,IAAI,EAAE,KAAK,IAAI;YAAE,GAAG,IAAI,MAAM,CAAC;aAC1B,IAAI,EAAE,KAAK,GAAG;YAAE,GAAG,IAAI,KAAK,CAAC;aAC7B,IAAI,EAAE,KAAK,IAAI;YAAE,GAAG,IAAI,KAAK,CAAC;aAC9B,IAAI,EAAE,KAAK,IAAI;YAAE,GAAG,IAAI,KAAK,CAAC;aAC9B,IAAI,EAAE,KAAK,IAAI;YAAE,GAAG,IAAI,KAAK,CAAC;aAC9B,IAAI,EAAE,KAAK,QAAQ;YAAE,GAAG,IAAI,SAAS,CAAC;aACtC,IAAI,EAAE,KAAK,QAAQ;YAAE,GAAG,IAAI,SAAS,CAAC;;YACtC,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;IACnC,CAAC;IACD,OAAO,GAAG,CAAC;AACZ,CAAC;AAED,SAAS,WAAW,CAAC,IAAY;IAChC,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;AACtC,CAAC;AAED,SAAS,WAAW,CAAC,IAAY;IAChC,OAAO,CAAC,4BAA4B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACjD,CAAC;AAED,SAAS,SAAS,CAAC,IAAY;IAC9B,IAAI,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;QACvB,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,KAAK,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC;YACvB,IAAI,EAAE,KAAK,IAAI;gBAAE,OAAO,IAAI,MAAM,CAAC;iBAC9B,IAAI,EAAE,KAAK,GAAG;gBAAE,OAAO,IAAI,KAAK,CAAC;iBACjC,IAAI,EAAE,KAAK,IAAI;gBAAE,OAAO,IAAI,KAAK,CAAC;iBAClC,IAAI,EAAE,KAAK,IAAI;gBAAE,OAAO,IAAI,KAAK,CAAC;iBAClC,IAAI,EAAE,KAAK,IAAI;gBAAE,OAAO,IAAI,KAAK,CAAC;iBAClC,IAAI,EAAE,KAAK,QAAQ;gBAAE,OAAO,IAAI,SAAS,CAAC;iBAC1C,IAAI,EAAE,KAAK,QAAQ;gBAAE,OAAO,IAAI,SAAS,CAAC;;gBAC1C,OAAO,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,IAAI,OAAO,GAAG,CAAC;IACvB,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AA8BD,SAAS,UAAU,CAAC,GAAW,EAAE,IAAoB;IACpD,IAAI,GAAG,KAAK,GAAG;QAAE,OAAO,IAAI,CAAC;IAC7B,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACvC,MAAM,QAAQ,GAAG,GAAG;SAClB,KAAK,CAAC,CAAC,CAAC;SACR,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;IACxD,IAAI,OAAO,GAAY,IAAI,CAAC;IAC5B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,OAAO,KAAK,IAAI,IAAI,OAAO,OAAO,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC;QACjE,OAAO,GAAI,OAAmC,CAAC,GAAG,CAAC,CAAC;QACpD,IAAI,OAAO,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC;IACxC,CAAC;IACD,IAAI,OAAO,OAAO,KAAK,SAAS;QAAE,OAAO,OAAO,CAAC;IACjD,IAAI,OAAO,KAAK,IAAI,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IACjE,OAAO,OAAyB,CAAC;AAClC,CAAC;AAED,SAAS,aAAa,CAAC,MAAc,EAAE,MAAuB;IAC7D,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,OAAO,IAAI,MAAM,EAAE,QAAQ,KAAK,IAAI;QAC1E,OAAO,GAAG,MAAM,SAAS,CAAC;IAC3B,OAAO,MAAM,CAAC;AACf,CAAC;AAED,SAAS,sBAAsB,CAC9B,MAAgC,EAChC,MAAc,EACd,GAAe;IAEf,IAAI,OAAO,MAAM,KAAK,SAAS;QAAE,OAAO,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;IACrE,IAAI,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,QAAQ;QAAE,OAAO,SAAS,CAAC;IAChD,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC;QAAE,OAAO,SAAS,CAAC;IAC3C,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACrB,MAAM,OAAO,GAAe,EAAE,GAAG,GAAG,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC;IAC7D,IAAI,CAAC;QACJ,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,QAAQ,GAAG,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;YACnD,IAAI,CAAC,QAAQ;gBAAE,OAAO,SAAS,CAAC;YAChC,OAAO,aAAa,CAAC,sBAAsB,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,CAAC;QACjF,CAAC;QACD,IAAI,MAAM,CAAC,KAAK;YACf,OAAO,aAAa,CACnB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAC/E,MAAM,CACN,CAAC;QACH,IAAI,MAAM,CAAC,KAAK;YACf,OAAO,aAAa,CACnB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAC/E,MAAM,CACN,CAAC;QACH,IAAI,MAAM,CAAC,KAAK;YACf,OAAO,aAAa,CACnB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAC/E,MAAM,CACN,CAAC;QACH,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,OAAO,CAAC;YAC7C,OAAO,aAAa,CACnB,MAAM,CAAC,IAAI;iBACT,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACV,IAAI,CAAC,KAAK,IAAI;oBAAE,OAAO,MAAM,CAAC;gBAC9B,IAAI,OAAO,CAAC,KAAK,QAAQ;oBAAE,OAAO,IAAI,mBAAmB,CAAC,CAAC,CAAC,GAAG,CAAC;gBAChE,IAAI,OAAO,CAAC,KAAK,QAAQ;oBAAE,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;gBACjE,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC,CAAC;iBACD,IAAI,CAAC,KAAK,CAAC,EACb,MAAM,CACN,CAAC;QACH,CAAC;QACD,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAChC,OAAO,aAAa,CACnB,MAAM,CAAC,KAAK,KAAK,IAAI;gBACpB,CAAC,CAAC,MAAM;gBACR,CAAC,CAAC,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;oBACjC,CAAC,CAAC,IAAI,mBAAmB,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG;oBAC1C,CAAC,CAAC,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;wBACjC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,SAAS,CAAC;wBAC7C,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EACzB,MAAM,CACN,CAAC;QACH,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACzB,IAAI,IAAI,KAAK,QAAQ;YAAE,OAAO,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC9D,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,SAAS;YAAE,OAAO,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACpF,IAAI,IAAI,KAAK,SAAS;YAAE,OAAO,aAAa,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAChE,IAAI,IAAI,KAAK,MAAM;YAAE,OAAO,MAAM,CAAC;QACnC,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACtB,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;YACvC,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC;gBAC7B,OAAO,aAAa,CACnB,IAAI,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EACpF,MAAM,CACN,CAAC;YACH,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;gBAC9B,OAAO,aAAa,CACnB,IAAI,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,sBAAsB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EACrF,MAAM,CACN,CAAC;YACH,IAAI,MAAM,CAAC,KAAK;gBACf,OAAO,aAAa,CACnB,GAAG,sBAAsB,CAAC,MAAM,CAAC,KAAuB,EAAE,MAAM,EAAE,OAAO,CAAC,IAAI,EAC9E,MAAM,CACN,CAAC;YACH,OAAO,aAAa,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QAC3C,CAAC;QACD,IAAI,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YAC5C,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;YACtC,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;YAChD,MAAM,KAAK,GAAa,EAAE,CAAC;YAC3B,KAAK,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5D,IAAI,OAAO,UAAU,KAAK,SAAS,EAAE,CAAC;oBACrC,MAAM,QAAQ,GAAG,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;oBAClD,MAAM,YAAY,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;oBACvD,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,OAAO,SAAS,CAAC,QAAQ,CAAC,GAAG,YAAY,KAAK,QAAQ,GAAG,CAAC,CAAC;oBAC/E,SAAS;gBACV,CAAC;gBACD,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAC1C,MAAM,QAAQ,GAAG,sBAAsB,CAAC,UAAU,EAAE,GAAG,MAAM,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC9E,MAAM,IAAI,GAAG,UAAU,CAAC,WAAW,CAAC;gBACpC,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;gBACjC,IAAI,IAAI,IAAI,MAAM,EAAE,CAAC;oBACpB,MAAM,QAAQ,GAAG,IAAI;wBACpB,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;wBAC1C,CAAC,CAAC,SAAS,CAAC;oBACb,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,WAAW,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;oBACxE,IAAI,QAAQ,IAAI,SAAS,EAAE,CAAC;wBAC3B,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,SAAS,CAAC,CAAC;wBAC/B,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,UAAU,QAAQ,EAAE,CAAC,CAAC;wBAC1C,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,UAAU,SAAS,EAAE,CAAC,CAAC;wBAC3C,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,SAAS,CAAC,CAAC;oBAChC,CAAC;yBAAM,CAAC;wBACP,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,WAAW,QAAQ,IAAI,SAAS,KAAK,CAAC,CAAC;oBAC5D,CAAC;gBACF,CAAC;gBACD,MAAM,UAAU,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;gBACvC,MAAM,YAAY,GAAG,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;gBAC3C,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,OAAO,UAAU,GAAG,YAAY,KAAK,QAAQ,GAAG,CAAC,CAAC;YACvE,CAAC;YACD,IAAI,MAAM,CAAC,oBAAoB,EAAE,CAAC;gBACjC,MAAM,SAAS,GACd,MAAM,CAAC,oBAAoB,KAAK,IAAI;oBACnC,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,sBAAsB,CAAC,MAAM,CAAC,oBAAoB,EAAE,GAAG,MAAM,MAAM,EAAE,OAAO,CAAC,CAAC;gBAClF,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,sBAAsB,SAAS,GAAG,CAAC,CAAC;YACzD,CAAC;YACD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxB,IAAI,MAAM,CAAC,oBAAoB,KAAK,KAAK;oBAAE,OAAO,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBAC9E,OAAO,aAAa,CAAC,yBAAyB,EAAE,MAAM,CAAC,CAAC;YACzD,CAAC;YACD,OAAO,aAAa,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,MAAM,GAAG,EAAE,MAAM,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,OAAO,aAAa,CACnB,IAAI;iBACF,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACV,IAAI,CAAC,KAAK,QAAQ;oBAAE,OAAO,QAAQ,CAAC;gBACpC,IAAI,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,SAAS;oBAAE,OAAO,QAAQ,CAAC;gBACvD,IAAI,CAAC,KAAK,SAAS;oBAAE,OAAO,SAAS,CAAC;gBACtC,IAAI,CAAC,KAAK,MAAM;oBAAE,OAAO,MAAM,CAAC;gBAChC,IAAI,CAAC,KAAK,OAAO;oBAAE,OAAO,WAAW,CAAC;gBACtC,IAAI,CAAC,KAAK,QAAQ;oBAAE,OAAO,yBAAyB,CAAC;gBACrD,OAAO,SAAS,CAAC;YAClB,CAAC,CAAC;iBACD,IAAI,CAAC,KAAK,CAAC,EACb,MAAM,CACN,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;YAAS,CAAC;QACV,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACzB,CAAC;AACF,CAAC;AAED,sEAAsE;AAEtE;;;;GAIG;AACH,MAAM,UAAU,aAAa,CAAC,KAAqB;IAClD,IAAI,cAAc,GAAG,EAAE,CAAC;IACxB,IAAI,cAAc,GAAG,EAAE,CAAC;IAExB,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,SAAS,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;QAExC,IAAI,CAAC;YACJ,MAAM,UAAU,GAAG,eAAe,CAAC,CAAC,CAAC,UAAU,CAAmB,CAAC;YACnE,MAAM,SAAS,GAAG,QAAQ,SAAS,WAAW,sBAAsB,CAAC,UAAU,EAAE,EAAE,EAAE;gBACpF,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,CAAC;gBACR,IAAI,EAAE,IAAI,GAAG,EAAE;gBACf,QAAQ,EAAE,EAAE;aACZ,CAAC,EAAE,CAAC;YACL,MAAM,UAAU,GAAG,QAAQ,SAAS,kBAAkB,CAAC;YAEvD,cAAc,IAAI,KAAK,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC;YAC1C,cAAc,IAAI,KAAK,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC;YAE3C,MAAM,UAAU,GAAa,EAAE,CAAC;YAChC,IAAI,CAAC,CAAC,WAAW,EAAE,IAAI,EAAE,EAAE,CAAC;gBAC3B,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;YAC3E,CAAC;iBAAM,CAAC;gBACP,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YACtC,CAAC;YAED,yDAAyD;YACzD,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;gBAC3B,KAAK,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;oBAC7E,IACC,OAAO,UAAU,KAAK,QAAQ;wBAC9B,UAAU,KAAK,IAAI;wBACnB,aAAa,IAAI,UAAU;wBAC3B,UAAU,CAAC,WAAW,EACrB,CAAC;wBACF,UAAU,CAAC,IAAI,CACd,WAAW,CACV,gBAAgB,SAAS,MAAM,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,CACtF,CACD,CAAC;oBACH,CAAC;gBACF,CAAC;YACF,CAAC;YAED,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChE,cAAc,IAAI,YAAY,SAAS,SAAS,CAAC;YACjD,cAAc,IAAI,OAAO,QAAQ,aAAa,SAAS,qBAAqB,SAAS,UAAU,CAAC;YAChG,cAAc,IAAI,IAAI,CAAC;QACxB,CAAC;QAAC,MAAM,CAAC;YACR,cAAc,IAAI,UAAU,SAAS,iBAAiB,CAAC;YACvD,cAAc,IAAI,UAAU,SAAS,kBAAkB,CAAC;YACxD,cAAc,IAAI,iBAAiB,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;YAChE,cAAc,IAAI,OAAO,QAAQ,aAAa,SAAS,qBAAqB,SAAS,UAAU,CAAC;YAChG,cAAc,IAAI,IAAI,CAAC;QACxB,CAAC;IACF,CAAC;IAED,cAAc,GAAG,8BAA8B,cAAc,GAAG,CAAC;IACjE,OAAO,KAAK,cAAc,KAAK,cAAc,IAAI,CAAC,IAAI,EAAE,CAAC;AAC1D,CAAC;AAED,sEAAsE;AAEtE;;GAEG;AACH,SAAS,eAAe,CAAC,IAAY;IACpC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,oEAAoE,CAAC,CAAC;IAC/F,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACjC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,aAAa,CAAC,IAAY;IACzC,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IAC7C,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE;QAAE,OAAO,gBAAgB,CAAC;IAC7C,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IAE9B,gDAAgD;IAChD,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACzD,OAAO,MAAM,CAAC;IACf,CAAC;IAED,4CAA4C;IAC5C,OAAO,kBAAkB,MAAM,KAAK,CAAC;AACtC,CAAC"}

package/dist/core/context.d.ts

@@ -0,0 +1,10 @@
+import type { UsageInfo } from "./model";
+export declare class RunContext<TContext> {
+    readonly context: TContext;
+    readonly usage: UsageInfo;
+    numTurns: number;
+    totalCostUsd: number;
+    constructor(context: TContext);
+    addUsage(usage: UsageInfo | undefined): void;
+}
+//# sourceMappingURL=context.d.ts.map

package/dist/core/context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../src/core/context.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEzC,qBAAa,UAAU,CAAC,QAAQ;IAC/B,QAAQ,CAAC,OAAO,EAAE,QAAQ,CAAC;IAC3B,QAAQ,CAAC,KAAK,EAAE,SAAS,CAAC;IAC1B,QAAQ,SAAK;IACb,YAAY,SAAK;gBAEL,OAAO,EAAE,QAAQ;IAS7B,QAAQ,CAAC,KAAK,EAAE,SAAS,GAAG,SAAS,GAAG,IAAI;CAgB5C"}

package/dist/core/context.js

@@ -0,0 +1,32 @@
+export class RunContext {
+    context;
+    usage;
+    numTurns = 0;
+    totalCostUsd = 0;
+    constructor(context) {
+        this.context = context;
+        this.usage = {
+            promptTokens: 0,
+            completionTokens: 0,
+            totalTokens: 0,
+        };
+    }
+    addUsage(usage) {
+        if (!usage)
+            return;
+        const u = this.usage;
+        u.promptTokens += usage.promptTokens;
+        u.completionTokens += usage.completionTokens;
+        u.totalTokens += usage.totalTokens;
+        if (usage.cacheReadTokens !== undefined) {
+            u.cacheReadTokens = (u.cacheReadTokens ?? 0) + usage.cacheReadTokens;
+        }
+        if (usage.cacheCreationTokens !== undefined) {
+            u.cacheCreationTokens = (u.cacheCreationTokens ?? 0) + usage.cacheCreationTokens;
+        }
+        if (usage.reasoningTokens !== undefined) {
+            u.reasoningTokens = (u.reasoningTokens ?? 0) + usage.reasoningTokens;
+        }
+    }
+}
+//# sourceMappingURL=context.js.map

package/dist/core/context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../../src/core/context.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,UAAU;IACb,OAAO,CAAW;IAClB,KAAK,CAAY;IAC1B,QAAQ,GAAG,CAAC,CAAC;IACb,YAAY,GAAG,CAAC,CAAC;IAEjB,YAAY,OAAiB;QAC5B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,KAAK,GAAG;YACZ,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,CAAC;SACd,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,KAA4B;QACpC,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;QACrB,CAAC,CAAC,YAAY,IAAI,KAAK,CAAC,YAAY,CAAC;QACrC,CAAC,CAAC,gBAAgB,IAAI,KAAK,CAAC,gBAAgB,CAAC;QAC7C,CAAC,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC;QACnC,IAAI,KAAK,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YACzC,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC,eAAe,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,eAAe,CAAC;QACtE,CAAC;QACD,IAAI,KAAK,CAAC,mBAAmB,KAAK,SAAS,EAAE,CAAC;YAC7C,CAAC,CAAC,mBAAmB,GAAG,CAAC,CAAC,CAAC,mBAAmB,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,mBAAmB,CAAC;QAClF,CAAC;QACD,IAAI,KAAK,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YACzC,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC,CAAC,eAAe,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,eAAe,CAAC;QACtE,CAAC;IACF,CAAC;CACD"}

package/dist/core/cost.d.ts

@@ -0,0 +1,9 @@
+import type { UsageInfo } from "./model";
+export interface PricingConfig {
+    inputTokenCostPer1k: number;
+    outputTokenCostPer1k: number;
+    cachedInputTokenCostPer1k?: number;
+}
+export type CostEstimator = (usage: UsageInfo) => number;
+export declare function createCostEstimator(pricing: PricingConfig): CostEstimator;
+//# sourceMappingURL=cost.d.ts.map

package/dist/core/cost.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost.d.ts","sourceRoot":"","sources":["../../src/core/cost.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEzC,MAAM,WAAW,aAAa;IAC7B,mBAAmB,EAAE,MAAM,CAAC;IAC5B,oBAAoB,EAAE,MAAM,CAAC;IAC7B,yBAAyB,CAAC,EAAE,MAAM,CAAC;CACnC;AAED,MAAM,MAAM,aAAa,GAAG,CAAC,KAAK,EAAE,SAAS,KAAK,MAAM,CAAC;AAEzD,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,aAAa,GAAG,aAAa,CAezE"}

package/dist/core/cost.js

@@ -0,0 +1,14 @@
+export function createCostEstimator(pricing) {
+    return (usage) => {
+        let inputTokens = usage.promptTokens;
+        let cachedCost = 0;
+        if (pricing.cachedInputTokenCostPer1k !== undefined && usage.cacheReadTokens) {
+            cachedCost = (usage.cacheReadTokens / 1000) * pricing.cachedInputTokenCostPer1k;
+            inputTokens -= usage.cacheReadTokens;
+        }
+        const inputCost = (inputTokens / 1000) * pricing.inputTokenCostPer1k;
+        const outputCost = (usage.completionTokens / 1000) * pricing.outputTokenCostPer1k;
+        return inputCost + outputCost + cachedCost;
+    };
+}
+//# sourceMappingURL=cost.js.map

package/dist/core/cost.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost.js","sourceRoot":"","sources":["../../src/core/cost.ts"],"names":[],"mappings":"AAUA,MAAM,UAAU,mBAAmB,CAAC,OAAsB;IACzD,OAAO,CAAC,KAAgB,EAAU,EAAE;QACnC,IAAI,WAAW,GAAG,KAAK,CAAC,YAAY,CAAC;QACrC,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,IAAI,OAAO,CAAC,yBAAyB,KAAK,SAAS,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;YAC9E,UAAU,GAAG,CAAC,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,yBAAyB,CAAC;YAChF,WAAW,IAAI,KAAK,CAAC,eAAe,CAAC;QACtC,CAAC;QAED,MAAM,SAAS,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,mBAAmB,CAAC;QACrE,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,oBAAoB,CAAC;QAElF,OAAO,SAAS,GAAG,UAAU,GAAG,UAAU,CAAC;IAC5C,CAAC,CAAC;AACH,CAAC"}