updates

Author: Shaw

cloud/packages/lib/services/containers/agent-warm-pool.ts

@@ -383,13 +383,4 @@ function emptyState(): PoolStateSnapshot {
   };
 }
 
-/**
- * Helper used by the claim flow to persist a row update + delete pool row
- * in one transaction. Lives on the repo; this re-export is purely for
- * call-site clarity.
- */
-export const claimWarmContainer = agentSandboxesRepository.claimWarmContainer.bind(
-  agentSandboxesRepository,
-);
-
 export type { AgentSandbox };

cloud/packages/tests/unit/agent-warm-pool-manager.test.ts

@@ -187,21 +187,27 @@ describe("WarmPoolManager", () => {
     });
     const { WarmPoolManager } = await importManager();
     const creator: CreatorMock = {
-      ...makeCreator(),
+      createCalls: 0,
+      destroyCalls: [],
+      probeReturns: new Map(),
       async createPoolContainer() {
         this.createCalls++;
         if (this.createCalls === 2) throw new Error("hcloud quota exceeded");
         return { id: `c-${this.createCalls}`, nodeId: "node-1" };
       },
+      async destroyPoolContainer(id: string) {
+        this.destroyCalls.push(id);
+      },
+      async healthProbe() {
+        return true;
+      },
     } as CreatorMock;
+    const mgr = new WarmPoolManager(creator, DEFAULT_WARM_POOL_POLICY, () => NOW);
 
-    const result = await mgr.replenishWithCreator(creator);
+    const result = await mgr.replenish(IMAGE);
     expect(result.created.length).toBe(1);
     expect(result.failed.length).toBe(1);
     expect(result.failed[0]?.error).toContain("hcloud quota");
-
-    function _ignored<T>(_v: T): void {}
-    _ignored(creator);
   });
 
   test("drainIdle removes only past-idle-window entries when target == floor", async () => {

packages/core/src/features/advanced-capabilities/actions/roles.ts

@@ -24,15 +24,6 @@ interface RoleAssignmentToon {
 	newRole?: string;
 }
 
-/** Shape of the role extraction structured response. */
-interface RoleExtractionResult {
-	assignments?:
-		| {
-				assignment?: RoleAssignmentToon | RoleAssignmentToon[];
-		  }
-		| RoleAssignmentToon[];
-}
-
 function isRecord(value: unknown): value is Record<string, unknown> {
 	return typeof value === "object" && value !== null && !Array.isArray(value);
 }

packages/core/src/services/message.ts

@@ -28,6 +28,9 @@ import {
 } from "../prompts/message-handler";
 import { filterByContextGate } from "../runtime/context-gates";
 import { createContextObject } from "../runtime/context-object";
+import type { ContextRegistry } from "../runtime/context-registry";
+import { checkSenderRole } from "../roles";
+import type { ContextDefinition, RoleGateRole } from "../types/contexts";
 import {
 	type EvaluatorEffects,
 	type EvaluatorOutput,
@@ -51,6 +54,11 @@ import {
 	runPlannerLoop,
 } from "../runtime/planner-loop";
 import { actionHasSubActions, runSubPlanner } from "../runtime/sub-planner";
+import {
+	createJsonFileTrajectoryRecorder,
+	isTrajectoryRecordingEnabled,
+	type TrajectoryRecorder,
+} from "../runtime/trajectory-recorder";
 import { isExplicitSelfModificationRequest } from "../should-respond";
 import {
 	getModelStreamChunkDeliveryDepth,

plugins/plugin-discord/actions/chatWithAttachments.ts

@@ -14,7 +14,6 @@ import {
 	type Memory,
 	MemoryType,
 	ModelType,
-	parseToonKeyValue,
 	type State,
 	trimTokens,
 } from "@elizaos/core";
@@ -23,6 +22,7 @@ import {
 	attachmentSummarizationTemplate as summarizationTemplate,
 } from "../generated/prompts/typescript/prompts.js";
 import { requireActionSpec } from "../generated/specs/spec-helpers";
+import { getActionParameters, parseJsonObjectFromText } from "../utils";
 
 const SPEC_NAME = "DISCORD_CHAT_WITH_ATTACHMENTS";
 
@@ -37,7 +37,16 @@ const getAttachmentIds = async (
 	runtime: IAgentRuntime,
 	_message: Memory,
 	state: State,
+	options?: HandlerOptions,
 ): Promise<{ objective: string; attachmentIds: string[] } | null> => {
+	const parameters = getActionParameters(options);
+	if (parameters.objective && Array.isArray(parameters.attachmentIds)) {
+		return {
+			objective: String(parameters.objective),
+			attachmentIds: parameters.attachmentIds.map(String),
+		};
+	}
+
 	const prompt = composePromptFromState({
 		state,
 		template: attachmentIdsTemplate,
@@ -47,10 +56,7 @@ const getAttachmentIds = async (
 		const response = await runtime.useModel(ModelType.TEXT_SMALL, {
 			prompt,
 		});
-		// Try parsing the TOON response.
-		const parsedResponse = parseToonKeyValue<Record<string, unknown>>(
-			response,
-		) as {
+		const parsedResponse = parseJsonObjectFromText(response) as {
 			objective: string;
 			attachmentIds: string[];
 		} | null;
@@ -176,7 +182,7 @@ export const chatWithAttachments: Action = {
 		runtime: IAgentRuntime,
 		message: Memory,
 		state?: State,
-		_options?: HandlerOptions,
+		options?: HandlerOptions,
 		callback?: HandlerCallback,
 	): Promise<ActionResult | undefined> => {
 		if (!state) {
@@ -196,7 +202,7 @@ export const chatWithAttachments: Action = {
 		};
 
 		// 1. extract attachment IDs from the message
-		const attachmentData = await getAttachmentIds(runtime, message, state);
+		const attachmentData = await getAttachmentIds(runtime, message, state, options);
 		if (!attachmentData) {
 			runtime.logger.warn(
 				{

plugins/plugin-discord/actions/summarizeConversation.ts

@@ -15,12 +15,12 @@ import {
 	type Memory,
 	MemoryType,
 	ModelType,
-	parseToonKeyValue,
 	type State,
 	splitChunks,
 	trimTokens,
 } from "@elizaos/core";
 import { requireActionSpec } from "../generated/specs/spec-helpers";
+import { getActionParameters, parseJsonObjectFromText } from "../utils";
 
 /**
  * Normalizes a numeric timestamp to milliseconds.
@@ -137,7 +137,28 @@ const getDateRange = async (
 	runtime: IAgentRuntime,
 	_message: Memory,
 	state: State,
+	options?: HandlerOptions,
 ): Promise<{ objective: string; start: number; end: number } | null> => {
+	const parameters = getActionParameters(options);
+	if (parameters.objective && parameters.start && parameters.end) {
+		const startRaw = parseTimeToTimestamp(
+			parameters.start as string | number,
+		);
+		const endRaw = parseTimeToTimestamp(parameters.end as string | number);
+		if (Number.isFinite(startRaw) && Number.isFinite(endRaw)) {
+			let start = startRaw <= endRaw ? startRaw : endRaw;
+			const end = startRaw <= endRaw ? endRaw : startRaw;
+			if (start === end) {
+				start = end - 3600 * 1000;
+			}
+			return {
+				objective: String(parameters.objective),
+				start,
+				end,
+			};
+		}
+	}
+
 	const prompt = composePromptFromState({
 		state,
 		template: dateRangeTemplate,
@@ -148,10 +169,7 @@ const getDateRange = async (
 			prompt,
 		});
 
-		// Try parsing the TOON response.
-		const parsedResponse = parseToonKeyValue<Record<string, unknown>>(
-			response,
-		) as {
+		const parsedResponse = parseJsonObjectFromText(response) as {
 			objective: string;
 			start: string | number;
 			end: string | number;
@@ -311,7 +329,7 @@ export const summarize: Action = {
 		runtime: IAgentRuntime,
 		message: Memory,
 		state?: State,
-		_options?: HandlerOptions,
+		options?: HandlerOptions,
 		callback?: HandlerCallback,
 	): Promise<ActionResult | undefined> => {
 		if (!state) {
@@ -333,7 +351,7 @@ export const summarize: Action = {
 		const { roomId } = message;
 
 		// 1. extract date range from the message
-		const dateRange = await getDateRange(runtime, message, state);
+		const dateRange = await getDateRange(runtime, message, state, options);
 		if (!dateRange) {
 			runtime.logger.warn(
 				{