@lloyal-labs/lloyal-agents 1.7.0 → 2.1.0

package/dist/shared-root.js

@@ -1,125 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.withSharedRoot = withSharedRoot;
-const effection_1 = require("effection");
-const sdk_1 = require("@lloyal-labs/sdk");
-const context_1 = require("./context");
-const trace_scope_1 = require("./trace-scope");
-/**
- * Scoped shared root branch with guaranteed cleanup
- *
- * Creates (or forks) a root branch, prefills the system prompt, and passes
- * it to the body function. The root is pruned via try/finally when the body
- * returns or throws, regardless of whether children still exist.
- *
- * **Cold path** (no `parent`): creates root at position 0, prefills system
- * prompt. Use for top-level research where no prior context exists.
- *
- * **Warm path** (`parent` provided): forks from parent branch, prefills
- * system prompt as a delta. Sub-agents inherit the parent's full KV state.
- * Use for recursive tools (web_research, research) where sub-agents should
- * attend over the calling agent's accumulated evidence.
- *
- * @param opts - System prompt, tools, sampling parameters, and optional parent branch
- * @param body - Operation that receives the root branch and prefix length.
- *   Typically calls {@link runAgents} or {@link useAgentPool} inside.
- * @returns The body's return value
- *
- * @category Agents
- */
-function* withSharedRoot(opts, body) {
-    const ctx = yield* context_1.Ctx.expect();
-    const tw = yield* context_1.Trace.expect();
-    // Read parent trace ID — connects nested pools to the outer DISPATCH that spawned them
-    let parentTraceId = null;
-    try {
-        const p = yield* context_1.TraceParent.get();
-        if (p != null)
-            parentTraceId = p;
-    }
-    catch {
-        /* no parent — top level */
-    }
-    const scope = (0, trace_scope_1.traceScope)(tw, parentTraceId, "withSharedRoot", {
-        hasTools: !!opts.tools,
-        systemPromptLength: opts.systemPrompt.length,
-        hasParent: !!opts.parent,
-    });
-    const messages = [{ role: "system", content: opts.systemPrompt }];
-    const fmtOpts = {
-        addGenerationPrompt: false,
-        enableThinking: false,
-    };
-    if (opts.tools)
-        fmtOpts.tools = opts.tools;
-    const fmt = ctx.formatChatSync(JSON.stringify(messages), fmtOpts);
-    const sharedTokens = ctx.tokenizeSync(fmt.prompt);
-    tw.write({
-        traceId: tw.nextId(),
-        parentTraceId: scope.traceId,
-        ts: performance.now(),
-        type: "prompt:format",
-        promptText: fmt.prompt,
-        tokenCount: sharedTokens.length,
-        messages: JSON.stringify(messages),
-        tools: opts.tools,
-        grammar: fmt.grammar || undefined,
-        role: "sharedRoot",
-    });
-    // Warm path: fork from parent branch (inherits full KV state)
-    // Cold path: create fresh root at position 0
-    let root;
-    let prefillTokens;
-    if (opts.parent) {
-        root = opts.parent.forkSync();
-        // Warm path: parent already has system prompt + tools in KV.
-        // Only prefill turn separator — the prompt is inherited via fork.
-        // This saves ~760 tokens per recursive fork.
-        const sep = ctx.getTurnSeparator();
-        prefillTokens = sep;
-    }
-    else {
-        root = sdk_1.Branch.create(ctx, 0, opts.params ?? { temperature: 0.5 });
-        prefillTokens = sharedTokens;
-    }
-    tw.write({
-        traceId: tw.nextId(),
-        parentTraceId: scope.traceId,
-        ts: performance.now(),
-        type: "branch:create",
-        branchHandle: root.handle,
-        parentHandle: opts.parent?.handle ?? null,
-        position: opts.parent ? opts.parent.position : 0,
-        role: "sharedRoot",
-    });
-    yield* (0, effection_1.call)(() => root.prefill(prefillTokens));
-    tw.write({
-        traceId: tw.nextId(),
-        parentTraceId: scope.traceId,
-        ts: performance.now(),
-        type: "branch:prefill",
-        branchHandle: root.handle,
-        tokenCount: prefillTokens.length,
-        role: "sharedPrefix",
-    });
-    try {
-        if (opts.enableScratchpad)
-            yield* context_1.ScratchpadParent.set(root);
-        return yield* body(root, prefillTokens.length);
-    }
-    finally {
-        if (!root.disposed) {
-            tw.write({
-                traceId: tw.nextId(),
-                parentTraceId: scope.traceId,
-                ts: performance.now(),
-                type: "branch:prune",
-                branchHandle: root.handle,
-                position: 0,
-            });
-            root.pruneSubtreeSync();
-        }
-        scope.close();
-    }
-}
-//# sourceMappingURL=shared-root.js.map

package/dist/shared-root.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"shared-root.js","sourceRoot":"","sources":["../src/shared-root.ts"],"names":[],"mappings":";;AA8DA,wCAqGC;AAnKD,yCAAiC;AAEjC,0CAA0C;AAE1C,uCAAsE;AACtE,+CAA2C;AAmC3C;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,QAAe,CAAC,CAAC,cAAc,CAC7B,IAAuB,EACvB,IAAgE;IAEhE,MAAM,GAAG,GAAmB,KAAK,CAAC,CAAC,aAAG,CAAC,MAAM,EAAE,CAAC;IAChD,MAAM,EAAE,GAAG,KAAK,CAAC,CAAC,eAAK,CAAC,MAAM,EAAE,CAAC;IAEjC,uFAAuF;IACvF,IAAI,aAAa,GAAkB,IAAI,CAAC;IACxC,IAAI,CAAC;QACH,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,qBAAW,CAAC,GAAG,EAAE,CAAC;QACnC,IAAI,CAAC,IAAI,IAAI;YAAE,aAAa,GAAG,CAAC,CAAC;IACnC,CAAC;IAAC,MAAM,CAAC;QACP,2BAA2B;IAC7B,CAAC;IAED,MAAM,KAAK,GAAG,IAAA,wBAAU,EAAC,EAAE,EAAE,aAAa,EAAE,gBAAgB,EAAE;QAC5D,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK;QACtB,kBAAkB,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM;QAC5C,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM;KACzB,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;IAClE,MAAM,OAAO,GAA4B;QACvC,mBAAmB,EAAE,KAAK;QAC1B,cAAc,EAAE,KAAK;KACtB,CAAC;IACF,IAAI,IAAI,CAAC,KAAK;QAAE,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;IAC3C,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC,CAAC;IAClE,MAAM,YAAY,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAElD,EAAE,CAAC,KAAK,CAAC;QACP,OAAO,EAAE,EAAE,CAAC,MAAM,EAAE;QACpB,aAAa,EAAE,KAAK,CAAC,OAAO;QAC5B,EAAE,EAAE,WAAW,CAAC,GAAG,EAAE;QACrB,IAAI,EAAE,eAAe;QACrB,UAAU,EAAE,GAAG,CAAC,MAAM;QACtB,UAAU,EAAE,YAAY,CAAC,MAAM;QAC/B,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;QAClC,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,SAAS;QACjC,IAAI,EAAE,YAAY;KACnB,CAAC,CAAC;IAEH,8DAA8D;IAC9D,6CAA6C;IAC7C,IAAI,IAAY,CAAC;IACjB,IAAI,aAAuB,CAAC;IAE5B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QAC9B,6DAA6D;QAC7D,kEAAkE;QAClE,6CAA6C;QAC7C,MAAM,GAAG,GAAG,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACnC,aAAa,GAAG,GAAG,CAAC;IACtB,CAAC;SAAM,CAAC;QACN,IAAI,GAAG,YAAM,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QAClE,aAAa,GAAG,YAAY,CAAC;IAC/B,CAAC;IAED,EAAE,CAAC,KAAK,CAAC;QACP,OAAO,EAAE,EAAE,CAAC,MAAM,EAAE;QACpB,aAAa,EAAE,KAAK,CAAC,OAAO;QAC5B,EAAE,EAAE,WAAW,CAAC,GAAG,EAAE;QACrB,IAAI,EAAE,eAAe;QACrB,YAAY,EAAE,IAAI,CAAC,MAAM;QACzB,YAAY,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,IAAI,IAAI;QACzC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAChD,IAAI,EAAE,YAAY;KACnB,CAAC,CAAC;IAEH,KAAK,CAAC,CAAC,IAAA,gBAAI,EAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC;IAE/C,EAAE,CAAC,KAAK,CAAC;QACP,OAAO,EAAE,EAAE,CAAC,MAAM,EAAE;QACpB,aAAa,EAAE,KAAK,CAAC,OAAO;QAC5B,EAAE,EAAE,WAAW,CAAC,GAAG,EAAE;QACrB,IAAI,EAAE,gBAAgB;QACtB,YAAY,EAAE,IAAI,CAAC,MAAM;QACzB,UAAU,EAAE,aAAa,CAAC,MAAM;QAChC,IAAI,EAAE,cAAc;KACrB,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,IAAI,IAAI,CAAC,gBAAgB;YAAE,KAAK,CAAC,CAAC,0BAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC7D,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;IACjD,CAAC;YAAS,CAAC;QACT,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnB,EAAE,CAAC,KAAK,CAAC;gBACP,OAAO,EAAE,EAAE,CAAC,MAAM,EAAE;gBACpB,aAAa,EAAE,KAAK,CAAC,OAAO;gBAC5B,EAAE,EAAE,WAAW,CAAC,GAAG,EAAE;gBACrB,IAAI,EAAE,cAAc;gBACpB,YAAY,EAAE,IAAI,CAAC,MAAM;gBACzB,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC;YACH,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1B,CAAC;QACD,KAAK,CAAC,KAAK,EAAE,CAAC;IAChB,CAAC;AACH,CAAC"}