memeloop 0.2.4 → 0.2.5

package/dist/loop-api.cjs

@@ -1986,7 +1986,57 @@ var init_orchestration = __esm({
   }
 });
 
+// src/loopAPI/scriptRuntime.ts
+function messageStep(message) {
+  return { type: "message", data: message };
+}
+function createScriptStepEmitter(emittedSteps, onEmit) {
+  return (step) => {
+    emittedSteps.push(step);
+    onEmit?.(step);
+  };
+}
+function isAsyncIterable(value) {
+  return Boolean(value && typeof value === "object" && Symbol.asyncIterator in value);
+}
+async function* drainEmittedSteps(steps) {
+  while (steps.length > 0) {
+    const step = steps.shift();
+    if (step) yield step;
+  }
+}
+async function* yieldScriptResult(result, emittedSteps) {
+  yield* drainEmittedSteps(emittedSteps);
+  if (isAsyncIterable(result)) {
+    yield* result;
+    yield* drainEmittedSteps(emittedSteps);
+  } else if (typeof result === "string") {
+    yield messageStep(result);
+  } else if (Array.isArray(result)) {
+    yield* result;
+  } else if (result) {
+    yield result;
+  }
+}
+var init_scriptRuntime2 = __esm({
+  "src/loopAPI/scriptRuntime.ts"() {
+    "use strict";
+  }
+});
+
 // src/loopAPI/scriptLoader.ts
+function createBuiltinAgentLoopScriptLoader(loaderOptions) {
+  let builtinDigestsRegistered;
+  return async (scriptReference, policy = {}) => {
+    builtinDigestsRegistered ??= registerBuiltinScriptSources(loaderOptions.sources);
+    await builtinDigestsRegistered;
+    return loadAgentLoopScript(scriptReference, {
+      ...policy,
+      getBuiltinScriptSource: loaderOptions.getBuiltinScriptSource,
+      scriptType: loaderOptions.scriptType
+    });
+  };
+}
 function isLoadedScriptFunction(value) {
   return typeof value === "function";
 }
@@ -2168,34 +2218,24 @@ var init_scriptLoader = __esm({
 });
 
 // src/loopAPI/agent-agent-loop/scriptLoader.ts
-function ensureBuiltinDigestsRegistered() {
-  builtinDigestsRegistered ??= registerBuiltinScriptSources(builtinAgentAgentLoopSources);
-  return builtinDigestsRegistered;
-}
 async function loadAgentAgentLoopScript(scriptReference, options = {}) {
-  await ensureBuiltinDigestsRegistered();
-  return loadAgentLoopScript(scriptReference, {
-    ...options,
-    getBuiltinScriptSource: getBuiltinAgentAgentLoopSource,
-    scriptType: "AgentAgentLoop script"
-  });
+  return loadScript(scriptReference, options);
 }
-var builtinDigestsRegistered;
+var loadScript;
 var init_scriptLoader2 = __esm({
   "src/loopAPI/agent-agent-loop/scriptLoader.ts"() {
     "use strict";
     init_builtinLoopSources();
     init_scriptLoader();
+    loadScript = createBuiltinAgentLoopScriptLoader({
+      sources: builtinAgentAgentLoopSources,
+      getBuiltinScriptSource: getBuiltinAgentAgentLoopSource,
+      scriptType: "AgentAgentLoop script"
+    });
   }
 });
 
 // src/loopAPI/agent-agent-loop/loop.ts
-function isAsyncIterable(value) {
-  return Boolean(value && typeof value === "object" && Symbol.asyncIterator in value);
-}
-function messageStep(message) {
-  return { type: "message", data: message };
-}
 function stepText(step) {
   if (step.type !== "message") return void 0;
   if (typeof step.data === "string") return step.data;
@@ -2249,10 +2289,7 @@ function createScriptArguments(input, context, emittedSteps) {
     if (key === "agents") return agentEntries;
     return readProfileAgentEntries(context.profile, key) ?? [];
   };
-  const emit = (step) => {
-    emittedSteps.push(step);
-    context.runtime?.emit?.(step);
-  };
+  const emit = createScriptStepEmitter(emittedSteps, context.runtime?.emit);
   const log = (event, data) => {
     context.runtime?.log?.(event, data);
   };
@@ -2367,27 +2404,11 @@ function createScriptArguments(input, context, emittedSteps) {
     loadCheckpoint: context.runtime?.loadCheckpoint ?? (async () => void 0)
   };
 }
-async function* drainEmittedSteps(steps) {
-  while (steps.length > 0) {
-    const step = steps.shift();
-    if (step) yield step;
-  }
-}
 async function* runScript(script, input, context) {
   const emittedSteps = [];
   const scriptArguments = createScriptArguments(input, context, emittedSteps);
   const result = await script(scriptArguments);
-  yield* drainEmittedSteps(emittedSteps);
-  if (isAsyncIterable(result)) {
-    yield* result;
-    yield* drainEmittedSteps(emittedSteps);
-  } else if (typeof result === "string") {
-    yield messageStep(result);
-  } else if (Array.isArray(result)) {
-    yield* result;
-  } else if (result) {
-    yield result;
-  }
+  yield* yieldScriptResult(result, emittedSteps);
 }
 function asAgentAgentContext(context) {
   return context;
@@ -2450,6 +2471,7 @@ var init_loop = __esm({
   "src/loopAPI/agent-agent-loop/loop.ts"() {
     "use strict";
     init_orchestration();
+    init_scriptRuntime2();
     init_scriptLoader2();
     LOOP_ID = "agent-agent-loop";
     LOOP_NAME = "AgentAgent Loop";
@@ -4344,6 +4366,9 @@ function resolveMaxIterations(context) {
   const configured = context.agentToolLoop?.maxIterations;
   return configured != null && configured > 0 ? configured : DEFAULT_MAX_ITERATIONS;
 }
+function resolveLocalNodeId(context, input) {
+  return context.localNodeId?.trim() || input?.userMessage?.originNodeId?.trim() || "local";
+}
 function pluginToolCallSignature(calls) {
   return calls.map((call) => `${call.toolId}:${JSON.stringify(call.parameters)}`).join("|");
 }
@@ -4365,7 +4390,7 @@ async function blockRepeatedPluginToolCalls(context, input, state, calls, hookCo
   const toolMessage = createChatMessage({
     messageId: `${input.conversationId}:t:doom-loop:${state.iteration}:${Date.now().toString(36)}`,
     conversationId: input.conversationId,
-    originNodeId: "local",
+    originNodeId: resolveLocalNodeId(context, input),
     lamportClock,
     role: "tool",
     parts: [{
@@ -4410,11 +4435,12 @@ async function startAgentToolLoopTurn(context, input, state) {
     input.conversationId
   );
   const hostUserMessage = input.userMessage;
+  const localNodeId = resolveLocalNodeId(context, input);
   const userMessage = context.normalizeMessage?.({
     ...hostUserMessage,
     messageId: hostUserMessage?.messageId ?? `${input.conversationId}:${now.toString(36)}`,
     conversationId: input.conversationId,
-    originNodeId: hostUserMessage?.originNodeId ?? "local",
+    originNodeId: hostUserMessage?.originNodeId ?? localNodeId,
     timestamp: hostUserMessage?.timestamp ?? now,
     lamportClock: hostUserMessage?.lamportClock ?? lamportClock,
     role: "user",
@@ -4423,7 +4449,7 @@ async function startAgentToolLoopTurn(context, input, state) {
     ...hostUserMessage,
     messageId: hostUserMessage?.messageId ?? `${input.conversationId}:${now.toString(36)}`,
     conversationId: input.conversationId,
-    originNodeId: hostUserMessage?.originNodeId ?? "local",
+    originNodeId: hostUserMessage?.originNodeId ?? localNodeId,
     timestamp: hostUserMessage?.timestamp ?? now,
     lamportClock: hostUserMessage?.lamportClock ?? lamportClock,
     role: "user",
@@ -4534,10 +4560,11 @@ async function* runAgentToolLoopIteration(context, input, state) {
     context.storage,
     input.conversationId
   );
+  const localNodeId = resolveLocalNodeId(context, input);
   const buildAssistantMessage = (content) => context.normalizeMessage?.({
     messageId: assistantMessageId,
     conversationId: input.conversationId,
-    originNodeId: "local",
+    originNodeId: localNodeId,
     timestamp: Date.now(),
     lamportClock: assistantLamportClock,
     role: "assistant",
@@ -4545,7 +4572,7 @@ async function* runAgentToolLoopIteration(context, input, state) {
   }) ?? {
     messageId: assistantMessageId,
     conversationId: input.conversationId,
-    originNodeId: "local",
+    originNodeId: localNodeId,
     timestamp: Date.now(),
     lamportClock: assistantLamportClock,
     role: "assistant",
@@ -4764,40 +4791,24 @@ var init_directRunner = __esm({
 });
 
 // src/loopAPI/agent-tool-loop/scriptLoader.ts
-function ensureBuiltinDigestsRegistered2() {
-  builtinDigestsRegistered2 ??= registerBuiltinScriptSources(builtinAgentToolLoopSources);
-  return builtinDigestsRegistered2;
-}
 async function loadAgentToolLoopScript(scriptReference, options = {}) {
-  await ensureBuiltinDigestsRegistered2();
-  return loadAgentLoopScript(scriptReference, {
-    ...options,
-    getBuiltinScriptSource: getBuiltinAgentToolLoopSource,
-    scriptType: "AgentToolLoop script"
-  });
+  return loadScript2(scriptReference, options);
 }
-var builtinDigestsRegistered2;
+var loadScript2;
 var init_scriptLoader3 = __esm({
   "src/loopAPI/agent-tool-loop/scriptLoader.ts"() {
     "use strict";
     init_builtinLoopSources2();
     init_scriptLoader();
+    loadScript2 = createBuiltinAgentLoopScriptLoader({
+      sources: builtinAgentToolLoopSources,
+      getBuiltinScriptSource: getBuiltinAgentToolLoopSource,
+      scriptType: "AgentToolLoop script"
+    });
   }
 });
 
 // src/loopAPI/agent-tool-loop/scriptRunner.ts
-function isAsyncIterable3(value) {
-  return Boolean(value && typeof value === "object" && Symbol.asyncIterator in value);
-}
-function messageStep2(message) {
-  return { type: "message", data: message };
-}
-async function* drainEmittedSteps2(steps) {
-  while (steps.length > 0) {
-    const step = steps.shift();
-    if (step) yield step;
-  }
-}
 function asAgentToolLoopContext(rawContext) {
   return rawContext;
 }
@@ -4809,10 +4820,7 @@ async function resolveAgentToolLoopScript(context) {
   return loadAgentToolLoopScript(scriptReference, context.scriptPolicy);
 }
 function createAgentToolLoopScriptContext(input, context, emittedSteps) {
-  const emit = (step) => {
-    emittedSteps.push(step);
-    context.runtime?.emit?.(step);
-  };
+  const emit = createScriptStepEmitter(emittedSteps, context.runtime?.emit);
   return {
     input,
     context,
@@ -4823,7 +4831,7 @@ function createAgentToolLoopScriptContext(input, context, emittedSteps) {
     stopTurn: (state, reason) => stopAgentToolLoopTurn(context, input, state, reason),
     emit,
     finish: (message) => {
-      emit(typeof message === "string" ? messageStep2(message) : message);
+      emit(typeof message === "string" ? messageStep(message) : message);
     },
     isCancelled: () => context.runtime?.signal?.cancelled === true || context.agentToolLoop?.isCancelled?.(input.conversationId) === true,
     log: (event, data) => {
@@ -4835,21 +4843,12 @@ function createAgentToolLoopScriptContext(input, context, emittedSteps) {
 async function* runAgentToolLoopScript(script, input, context) {
   const emittedSteps = [];
   const result = await script(createAgentToolLoopScriptContext(input, context, emittedSteps));
-  yield* drainEmittedSteps2(emittedSteps);
-  if (isAsyncIterable3(result)) {
-    yield* result;
-    yield* drainEmittedSteps2(emittedSteps);
-  } else if (typeof result === "string") {
-    yield messageStep2(result);
-  } else if (Array.isArray(result)) {
-    yield* result;
-  } else if (result) {
-    yield result;
-  }
+  yield* yieldScriptResult(result, emittedSteps);
 }
 var init_scriptRunner = __esm({
   "src/loopAPI/agent-tool-loop/scriptRunner.ts"() {
     "use strict";
+    init_scriptRuntime2();
     init_scriptLoader3();
     init_turnPrimitives();
   }