@linnlabs/linnkit 0.9.0 → 0.10.0

package/dist/cli.js

@@ -200,6 +200,7 @@ var AgentSpecBudgetPolicy = z.object({
 });
 var AgentSpecToolHistoryPolicy = z.object({
   strategy: z.enum(["per-pair", "per-run", "none"]).optional(),
+  retentionMode: z.enum(["drop", "compress"]).optional(),
   keepLatestToolPairs: z.number().int().nonnegative().optional(),
   keepLatestRuns: z.number().int().nonnegative().optional(),
   maxInteractionGroups: z.number().int().nonnegative().optional(),
@@ -1081,6 +1082,7 @@ var Logger = class {
   constructor(moduleName) {
     this.moduleName = moduleName;
   }
+  moduleName;
   debug(message, data) {
     this.log(0 /* DEBUG */, "debug", message, data);
   }
@@ -1153,6 +1155,17 @@ var logger = new Logger("GraphExecutor");
 function asLocalRecord(local) {
   return local && typeof local === "object" ? { ...local } : {};
 }
+function readNonEmptyString(value) {
+  return typeof value === "string" && value.trim().length > 0 ? value : void 0;
+}
+function readRuntimeConversationId(state) {
+  const local = state?.local && typeof state.local === "object" ? state.local : void 0;
+  return readNonEmptyString(local?.conversationId);
+}
+function readRuntimeTurnId(state) {
+  const local = state?.local && typeof state.local === "object" ? state.local : void 0;
+  return readNonEmptyString(local?.turnId);
+}
 var GraphExecutor = class {
   constructor(checkpointer, config = {}) {
     this.checkpointer = checkpointer;
@@ -1162,6 +1175,7 @@ var GraphExecutor = class {
     };
     this.telemetryPort = config.telemetryPort ?? noopTelemetry;
   }
+  checkpointer;
   nodes = /* @__PURE__ */ new Map();
   ephemeralLocals = /* @__PURE__ */ new Map();
   config;
@@ -1169,8 +1183,8 @@ var GraphExecutor = class {
   registerNode(node) {
     this.nodes.set(node.id, node);
   }
-  async peekCheckpoint(conversationId) {
-    return await this.checkpointer.load(conversationId);
+  async peekCheckpoint(checkpointKey) {
+    return await this.checkpointer.load(checkpointKey);
   }
   sanitize(state) {
     const local = asLocalRecord(state.local);
@@ -1182,8 +1196,8 @@ var GraphExecutor = class {
       local
     };
   }
-  async prime(conversationId, local, nodeId = "user") {
-    this.ephemeralLocals.set(conversationId, { ...local || {} });
+  async prime(checkpointKey, local, nodeId = "user") {
+    this.ephemeralLocals.set(checkpointKey, { ...local || {} });
     const localSansMemory = { ...local || {} };
     if ("memory" in localSansMemory) delete localSansMemory.memory;
     const state = {
@@ -1191,10 +1205,10 @@ var GraphExecutor = class {
       schemaVersion: ENGINE_STATE_SCHEMA_VERSION,
       local: localSansMemory
     };
-    await this.checkpointer.save(conversationId, state);
+    await this.checkpointer.save(checkpointKey, state);
   }
-  async setNode(conversationId, nodeId, localPatch) {
-    const current = await this.checkpointer.load(conversationId) || {
+  async setNode(checkpointKey, nodeId, localPatch) {
+    const current = await this.checkpointer.load(checkpointKey) || {
       schemaVersion: ENGINE_STATE_SCHEMA_VERSION,
       local: {}
     };
@@ -1205,19 +1219,46 @@ var GraphExecutor = class {
       schemaVersion: current.schemaVersion ?? ENGINE_STATE_SCHEMA_VERSION,
       local: mergedLocal
     };
-    await this.checkpointer.save(conversationId, next);
+    await this.checkpointer.save(checkpointKey, next);
   }
-  async runUntilYield(conversationId) {
+  async runUntilYield(checkpointKey) {
     const runId = `run_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
     let lifecyclePhase = "completed";
+    let initialState = null;
+    try {
+      initialState = await this.loadInitialState(checkpointKey);
+    } catch (err) {
+      lifecyclePhase = "failed";
+      this.telemetryPort.emit({
+        kind: "run_lifecycle",
+        runId,
+        phase: "spawned",
+        scope: {}
+      });
+      this.telemetryPort.emit({
+        kind: "run_lifecycle",
+        runId,
+        phase: lifecyclePhase,
+        scope: {}
+      });
+      throw err;
+    }
+    let lifecycleConversationId = readRuntimeConversationId(initialState);
+    let lifecycleTurnId = readRuntimeTurnId(initialState);
     this.telemetryPort.emit({
       kind: "run_lifecycle",
       runId,
       phase: "spawned",
-      scope: { conversationId: conversationId || void 0 }
+      scope: {
+        conversationId: lifecycleConversationId,
+        turnId: lifecycleTurnId
+      }
     });
     try {
-      return await this.runUntilYieldInternal(conversationId);
+      const result = await this.runUntilYieldInternal(checkpointKey, initialState);
+      lifecycleConversationId = readRuntimeConversationId(result.checkpoint);
+      lifecycleTurnId = readRuntimeTurnId(result.checkpoint);
+      return result;
     } catch (err) {
       lifecyclePhase = err?.name === "AbortError" ? "cancelled" : "failed";
       throw err;
@@ -1226,17 +1267,23 @@ var GraphExecutor = class {
         kind: "run_lifecycle",
         runId,
         phase: lifecyclePhase,
-        scope: { conversationId: conversationId || void 0 }
+        scope: {
+          conversationId: lifecycleConversationId,
+          turnId: lifecycleTurnId
+        }
       });
     }
   }
-  async runUntilYieldInternal(conversationId) {
-    let state = await this.checkpointer.load(conversationId) || {
+  async loadInitialState(checkpointKey) {
+    return await this.checkpointer.load(checkpointKey) || {
       nodeId: "user",
       schemaVersion: ENGINE_STATE_SCHEMA_VERSION,
       local: {}
     };
-    const ephemeral = this.ephemeralLocals.get(conversationId) || {};
+  }
+  async runUntilYieldInternal(checkpointKey, initialState) {
+    let state = initialState;
+    const ephemeral = this.ephemeralLocals.get(checkpointKey) || {};
     state = {
       ...state,
       schemaVersion: state.schemaVersion ?? ENGINE_STATE_SCHEMA_VERSION,
@@ -1265,7 +1312,7 @@ var GraphExecutor = class {
       const signalRaw = state.local?.signal;
       if (isAbortSignal2(signalRaw) && signalRaw.aborted) {
         logger.warn("[GraphExecutor] \u6536\u5230 AbortSignal\uFF0C\u7ACB\u5373\u505C\u6B62\u63A8\u7406\u5FAA\u73AF");
-        this.ephemeralLocals.delete(conversationId);
+        this.ephemeralLocals.delete(checkpointKey);
         throwAbortError();
       }
       const isLastStep = cycleStepCount >= this.config.maxSteps;
@@ -1313,8 +1360,8 @@ var GraphExecutor = class {
           checkpointCount
         });
         const cp2 = this.sanitize(state);
-        await this.checkpointer.save(conversationId, cp2);
-        this.ephemeralLocals.delete(conversationId);
+        await this.checkpointer.save(checkpointKey, cp2);
+        this.ephemeralLocals.delete(checkpointKey);
         return { events: allEvents, checkpoint: cp2, stepCount };
       }
       logger.info("[GraphExecutor] \u8282\u70B9\u5207\u6362", {
@@ -1325,18 +1372,18 @@ var GraphExecutor = class {
       });
       const nodeRunStartedAt = Date.now();
       const nodeIdForTelemetry = state.nodeId;
+      const conversationIdForTelemetry = readRuntimeConversationId(state);
       let result;
       try {
         result = await node.run(state);
       } finally {
-        const turnIdForTelemetry = typeof state.local?.turnId === "string" ? state.local.turnId : void 0;
         this.telemetryPort.emit({
           kind: "graph_node",
           nodeId: nodeIdForTelemetry,
           durationMs: Date.now() - nodeRunStartedAt,
           scope: {
-            conversationId: conversationId || void 0,
-            turnId: turnIdForTelemetry
+            conversationId: conversationIdForTelemetry,
+            turnId: readRuntimeTurnId(state)
           }
         });
       }
@@ -1375,7 +1422,7 @@ var GraphExecutor = class {
         });
         state = { ...state, nodeId: nextNodeId };
         const cp2 = this.sanitize(state);
-        await this.checkpointer.save(conversationId, cp2);
+        await this.checkpointer.save(checkpointKey, cp2);
         continue;
       }
       if (result.kind === "yield") {
@@ -1386,7 +1433,7 @@ var GraphExecutor = class {
           checkpointCount
         });
         const cp2 = this.sanitize(state);
-        await this.checkpointer.save(conversationId, cp2);
+        await this.checkpointer.save(checkpointKey, cp2);
         return { events: allEvents, checkpoint: cp2, stepCount };
       }
       if (result.kind === "pause") {
@@ -1397,7 +1444,7 @@ var GraphExecutor = class {
           checkpointCount
         });
         const cp2 = this.sanitize(state);
-        await this.checkpointer.save(conversationId, cp2);
+        await this.checkpointer.save(checkpointKey, cp2);
         return { events: allEvents, checkpoint: cp2, stepCount };
       }
     }
@@ -1408,8 +1455,8 @@ var GraphExecutor = class {
       checkpointCount
     });
     const cp = this.sanitize(state);
-    await this.checkpointer.save(conversationId, cp);
-    this.ephemeralLocals.delete(conversationId);
+    await this.checkpointer.save(checkpointKey, cp);
+    this.ephemeralLocals.delete(checkpointKey);
     return { events: allEvents, checkpoint: cp, stepCount };
   }
 };
@@ -1572,15 +1619,15 @@ function splitConcatenatedJsonObjects(input) {
 }
 
 // src/runtime-kernel/graph-engine/tick-pipeline/helpers.ts
-function readNonEmptyString(value) {
+function readNonEmptyString2(value) {
   if (typeof value !== "string") return void 0;
   const trimmed = value.trim();
   return trimmed.length > 0 ? trimmed : void 0;
 }
 function resolveConversationIdForRuntimeEvents(toolContext) {
-  const fromCamel = readNonEmptyString(toolContext?.conversationId);
+  const fromCamel = readNonEmptyString2(toolContext?.conversationId);
   if (fromCamel) return fromCamel;
-  const fromSnake = toolContext ? readNonEmptyString(toolContext["conversation_id"]) : void 0;
+  const fromSnake = toolContext ? readNonEmptyString2(toolContext["conversation_id"]) : void 0;
   if (fromSnake) return fromSnake;
   return generateMessageId();
 }
@@ -1937,7 +1984,7 @@ var runModelLockMiddleware = async (ctx, stage, next) => {
   if (stage.id !== "execute_llm") {
     return;
   }
-  const normalized = readNonEmptyString(ctx.cloudQuotaFallbackAppliedModelId);
+  const normalized = readNonEmptyString2(ctx.cloudQuotaFallbackAppliedModelId);
   if (!normalized) {
     return;
   }
@@ -2525,7 +2572,7 @@ function createExecuteLlmStage(dependencies) {
         streamEventHandler,
         ctx.signal,
         (fallbackModelId) => {
-          ctx.cloudQuotaFallbackAppliedModelId = readNonEmptyString(fallbackModelId);
+          ctx.cloudQuotaFallbackAppliedModelId = readNonEmptyString2(fallbackModelId);
         },
         (info) => {
           ctx.modelFallbackAudit = info;
@@ -2569,7 +2616,7 @@ function createPrepareCallStage(dependencies) {
   return {
     id: "prepare_call",
     async run(ctx) {
-      const lockedRunModelId = readNonEmptyString(ctx.executorLocal?.runLockedModelId);
+      const lockedRunModelId = readNonEmptyString2(ctx.executorLocal?.runLockedModelId);
       const requestedModelId = lockedRunModelId ?? ctx.request.model_id;
       ctx.modelId = dependencies.modelResolver.resolveModelId(requestedModelId);
       await emitAuditEnvelope(ctx.audit, {
@@ -2641,7 +2688,7 @@ var GraphAgentExecutor = class {
     this.llmCaller = dependencies.llmCaller;
     this.toolRuntime = dependencies.toolRuntime;
     this.contextBuilder = dependencies.contextBuilder;
-    this.cloudQuotaFallbackModelId = readNonEmptyString(dependencies.cloudQuotaFallbackModelId);
+    this.cloudQuotaFallbackModelId = readNonEmptyString2(dependencies.cloudQuotaFallbackModelId);
     this.modelCatalog = dependencies.modelCatalog ?? createEmptyModelCatalog();
     this.modelResolver = dependencies.modelResolver ?? new ModelResolver({
       modelCatalog: this.modelCatalog
@@ -2689,7 +2736,7 @@ var GraphAgentExecutor = class {
       llmMessages: [],
       mode: input.request.mode === "chat" ? "chat" : "agent",
       conversationId: resolveConversationIdForRuntimeEvents(input.toolContext),
-      turnId: readNonEmptyString(input.toolContext?.turnId) ?? `turn_${Date.now()}`,
+      turnId: readNonEmptyString2(input.toolContext?.turnId) ?? `turn_${Date.now()}`,
       telemetry: this.telemetryPort,
       audit: this.auditPort
     };
@@ -2739,18 +2786,18 @@ function isSummarizationCallbacks(value) {
 function isGraphSseSink(value) {
   return typeof value === "function";
 }
-function readNonEmptyString2(value) {
+function readNonEmptyString3(value) {
   if (typeof value !== "string") return void 0;
   const trimmed = value.trim();
   return trimmed.length > 0 ? trimmed : void 0;
 }
 function readGraphAgentLocal(local) {
   const source = local ?? {};
-  const answerId = readNonEmptyString2(source.answerId);
+  const answerId = readNonEmptyString3(source.answerId);
   const chunkSeq = Number.isInteger(source.chunkSeq) ? Number(source.chunkSeq) : 0;
   return {
-    conversationId: readNonEmptyString2(source.conversationId) ?? "",
-    turnId: readNonEmptyString2(source.turnId),
+    conversationId: readNonEmptyString3(source.conversationId) ?? "",
+    turnId: readNonEmptyString3(source.turnId),
     request: isAgentInvocationRequest(source.request) ? source.request : void 0,
     toolContext: isToolExecutionContext(source.toolContext) ? source.toolContext : void 0,
     history: isRuntimeEventArray(source.history) ? source.history : [],
@@ -3515,7 +3562,7 @@ var LlmNode = class {
 function isRecord8(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
 }
-function readNonEmptyString3(value) {
+function readNonEmptyString4(value) {
   if (typeof value !== "string") {
     return void 0;
   }
@@ -3537,7 +3584,7 @@ function resolveFinalAnswerFromToolResult(toolName, parsedResult) {
     return void 0;
   }
   if (toolName === "write_report") {
-    const report = readNonEmptyString3(data.report);
+    const report = readNonEmptyString4(data.report);
     if (!report) {
       throw new Error("[write_report] \u5DE5\u5177\u8F93\u51FA\u7F3A\u5C11 data.report\uFF08\u5FC5\u987B\u63D0\u4F9B\u5B8C\u6574\u62A5\u544A\u6B63\u6587\uFF09");
     }
@@ -3547,7 +3594,7 @@ function resolveFinalAnswerFromToolResult(toolName, parsedResult) {
     if (data.success !== true) {
       return void 0;
     }
-    const finalAnswer = readNonEmptyString3(data.final_answer);
+    const finalAnswer = readNonEmptyString4(data.final_answer);
     if (!finalAnswer) {
       throw new Error("[research_run_writer] \u5DE5\u5177\u8F93\u51FA\u7F3A\u5C11 data.final_answer\uFF08success=true \u65F6\u5FC5\u987B\u63D0\u4F9B\u6700\u7EC8\u62A5\u544A\u6B63\u6587\uFF09");
     }
@@ -4881,12 +4928,12 @@ function asRecord(value) {
   }
   return value;
 }
-function summarizeCheckpoint(conversationId, state, savedAt) {
+function summarizeCheckpoint(checkpointKey, state, savedAt) {
   const local = asRecord(state.local);
   const executorLocal = asRecord(local?.executorLocal);
   const pendingToolCalls = local?.pendingToolCalls;
   return {
-    conversationId,
+    checkpointKey,
     schemaVersion: state.schemaVersion ?? 1,
     savedAt,
     currentNode: state.nodeId,
@@ -4905,25 +4952,25 @@ var MemoryCheckpointer = class {
       local: state.local && typeof state.local === "object" && !Array.isArray(state.local) ? { ...state.local } : state.local
     };
   }
-  async load(conversationId) {
-    const entry = this.store.get(conversationId);
+  async load(checkpointKey) {
+    const entry = this.store.get(checkpointKey);
     return entry ? this.cloneState(entry.state) : null;
   }
-  async save(conversationId, state) {
-    this.store.set(conversationId, {
+  async save(checkpointKey, state) {
+    this.store.set(checkpointKey, {
       state: this.cloneState(state),
       savedAt: Date.now()
     });
   }
-  async clear(conversationId) {
-    this.store.delete(conversationId);
+  async clear(checkpointKey) {
+    this.store.delete(checkpointKey);
   }
-  async peekMeta(conversationId) {
-    const entry = this.store.get(conversationId);
-    return entry ? summarizeCheckpoint(conversationId, entry.state, entry.savedAt) : null;
+  async peekMeta(checkpointKey) {
+    const entry = this.store.get(checkpointKey);
+    return entry ? summarizeCheckpoint(checkpointKey, entry.state, entry.savedAt) : null;
   }
   async list(filter = {}) {
-    const summaries = Array.from(this.store.entries()).map(([conversationId, entry]) => summarizeCheckpoint(conversationId, entry.state, entry.savedAt)).filter((summary) => filter.savedAfter === void 0 ? true : summary.savedAt > filter.savedAfter).sort((left, right) => right.savedAt - left.savedAt);
+    const summaries = Array.from(this.store.entries()).map(([checkpointKey, entry]) => summarizeCheckpoint(checkpointKey, entry.state, entry.savedAt)).filter((summary) => filter.savedAfter === void 0 ? true : summary.savedAt > filter.savedAfter).sort((left, right) => right.savedAt - left.savedAt);
     if (filter.limit === void 0) {
       return summaries;
     }
@@ -6463,7 +6510,7 @@ function cloneRunRecord(record) {
     ...record,
     iterationBudget: record.iterationBudget ? { ...record.iterationBudget } : void 0,
     errorIfAny: record.errorIfAny ? { ...record.errorIfAny } : void 0,
-    metadata: record.metadata ? { ...record.metadata } : void 0
+    metadata: record.metadata ? structuredClone(record.metadata) : void 0
   };
 }
 function matchesStatus(candidate, filter) {
@@ -6875,7 +6922,7 @@ function isTerminalStatus(status) {
   return status === "completed" || status === "failed" || status === "cancelled";
 }
 function cloneMetadata(metadata) {
-  return metadata ? { ...metadata } : void 0;
+  return metadata ? structuredClone(metadata) : void 0;
 }
 function recordToSnapshot(record) {
   return {
@@ -6962,7 +7009,7 @@ var DefaultRunSupervisor = class {
       startedAt,
       updatedAt: startedAt,
       iterationBudget: spec.iterationBudget ? { ...spec.iterationBudget } : void 0,
-      metadata: spec.metadata ? { ...spec.metadata } : void 0
+      metadata: cloneMetadata(spec.metadata)
     };
     await this.registryStore.save(record);
     const handle = new DefaultRunHandle({
@@ -7136,12 +7183,13 @@ var DefaultRunSupervisor = class {
     }
     try {
       await handle.markRunning({ currentNode: "detached" });
+      const registeredRecord = await this.registryStore.load(handle.runId);
       const executorOutcome = await this.executor.execute({
         runId: handle.runId,
-        parentRunId: spec.parentRunId,
-        conversationId: spec.conversationId,
-        agentSpec: spec.agentSpec,
-        request: spec.request,
+        parentRunId: handle.parentRunId,
+        conversationId: registeredRecord?.conversationId ?? spec.conversationId,
+        agentSpec: await handle.spec(),
+        request: await handle.request(),
         signal: handle.signal,
         eventBus: spec.eventBus,
         eventStore: spec.eventStore,
@@ -7150,7 +7198,7 @@ var DefaultRunSupervisor = class {
         contextFences: spec.contextFences,
         wakeSource: spec.wakeSource,
         ephemeral: spec.ephemeral,
-        metadata: spec.metadata
+        metadata: cloneMetadata(registeredRecord?.metadata ?? spec.metadata)
       });
       const outcome = await this.persistExecutorOutcome(handle, executorOutcome);
       this.notifyTerminalWaiters(outcome);
@@ -7214,10 +7262,14 @@ var DefaultRunSupervisor = class {
       await this.registryStore.save(record);
     }
     const completedAt = executorOutcome?.completedAt ?? this.now();
+    const fallbackMeta = record ? void 0 : await handle.meta();
     const outcome = recordToTerminalOutcome(record ?? {
       runId: handle.runId,
       parentRunId: handle.parentRunId,
-      status: executorOutcome?.status ?? "completed"}, completedAt);
+      conversationId: fallbackMeta?.conversationId ?? "",
+      agentSpecId: fallbackMeta?.agentSpecId,
+      status: executorOutcome?.status ?? "completed",
+      startedAt: fallbackMeta?.startedAt ?? completedAt}, completedAt);
     const nextOutcome = {
       ...outcome,
       metadata: {
@@ -7545,6 +7597,7 @@ async function emitRunEvent(event, sink) {
 async function runAgent(agent, options) {
   const modelId = resolveModelId(agent, options);
   const conversationId = options.conversationId ?? `conv_${Date.now()}`;
+  const checkpointKey = conversationId;
   const runId = options.runId ?? generateRunId();
   const turnId = runId;
   const costCollector = new QuickstartRunCostCollector();
@@ -7599,7 +7652,7 @@ async function runAgent(agent, options) {
   };
   await handle.markRunning({ currentNode: "user" });
   try {
-    await executor.prime(conversationId, {
+    await executor.prime(checkpointKey, {
       conversationId,
       turnId,
       request: {
@@ -7624,7 +7677,7 @@ async function runAgent(agent, options) {
         return void 0;
       }
     });
-    const result = await executor.runUntilYield(conversationId);
+    const result = await executor.runUntilYield(checkpointKey);
     runtimeEvents.push(...result.events);
     for (const event of result.events) {
       if (event.type === "final_answer_chunk") continue;
@@ -7787,7 +7840,7 @@ function createQuickstartTemplateFiles(projectName) {
           start: 'linnkit run hello --input "\u4F60\u597D\uFF0C\u4ECB\u7ECD\u4E00\u4E0B\u4F60\u81EA\u5DF1"'
         },
         dependencies: {
-          "@linnlabs/linnkit": "^0.8.0",
+          "@linnlabs/linnkit": "^0.10.0",
           zod: "^3.22.0"
         },
         devDependencies: {}