@agimon-ai/po 0.0.0-alpha.21 → 0.0.0-alpha.23

package/dist/src/builders/server/runtime.mjs

@@ -33,6 +33,7 @@ import { loadServerPackages, selectedServerProviderOverrides } from "@agimon-ai/
 import crypto from "node:crypto";
 import { createOpenSessionRegistry, createRequestReceipts, createWorkspaceRegistry, identifyWorkspace, listSavedSessionRecords, listSavedSessions, readSqliteTranscript, readWorkspaceMarker, writeWorkspaceMarker } from "@agimon-ai/po-core/history";
 import { createNodePlatform } from "@agimon-ai/po-core/nodePlatform";
+import { readSessionFileDirectories, removeSessionFiles } from "@agimon-ai/po-core/platform";
 import { readPoBackgroundWorkService } from "@agimon-ai/po-core/backgroundWork";
 import { createHeadlessHub } from "@agimon-ai/po-core/headlessHub";
 import { serveHeadlessServer } from "@agimon-ai/po-core/headlessServer";
@@ -77,6 +78,7 @@ async function runServerRuntime(options, runtime) {
 	const serverDirectory = path.join(piAgentDirectory(baseEnvironment), "server");
 	const platform = createNodePlatform({
 		homeDirectory,
+		temporaryDirectory: path.resolve(baseEnvironment.TMPDIR || baseEnvironment.TMP || baseEnvironment.TEMP || os.tmpdir()),
 		sessionsRoot: path.join(serverDirectory, "sessions")
 	});
 	/** Called on a watched change; assigned once every scope can be checked. */
@@ -206,6 +208,17 @@ async function runServerRuntime(options, runtime) {
 	* it exists to survive.
 	*/
 	let shuttingDown = false;
+	const removeSessionOutputs = async (sessionId) => {
+		const directories = await readSessionFileDirectories(platform.sessions, sessionId);
+		await removeSessionFiles({
+			platform,
+			sessionId,
+			environment: baseEnvironment,
+			artifactDirectory: path.join(directories?.agentDirectory ?? piAgentDirectory(baseEnvironment, homeDirectory), "po-artifacts", sessionId),
+			...directories === void 0 ? {} : { directories }
+		});
+		openSessions.remove(sessionId);
+	};
 	const closeManagedSession = async (sessionId, options) => {
 		if (!shuttingDown && options?.keepDormant !== true) openSessions.remove(sessionId);
 		const artifacts = sessionArtifacts.get(sessionId);
@@ -1298,7 +1311,8 @@ async function runServerRuntime(options, runtime) {
 				return target.summary.id;
 			},
 			dormantSessions: () => openSessions.list(),
-			removeDormantSession: (record) => openSessions.remove(record.sessionId),
+			removeDormantSession: (record) => removeSessionOutputs(record.sessionId),
+			removeSessionFiles: removeSessionOutputs,
 			setDormantGroups: (record, groups) => {
 				if (!openSessions.add({
 					...record,