@toolforge-js/sdk 0.8.6 → 0.8.8

package/dist/{tool-BHVNhWPh.js → agent-DzYymaXh.js}

@@ -1,8 +1,9 @@
 import { createRequire } from "node:module";
 import * as z$1 from "zod";
-import { nanoid } from "nanoid";
 import { P, match } from "ts-pattern";
 import { setTimeout } from "node:timers/promises";
+import { nanoid } from "nanoid";
+import { AsyncLocalStorage } from "node:async_hooks";
 
 //#region rolldown:runtime
 var __create = Object.create;
@@ -30,22 +31,6 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 }) : target, mod));
 var __require = /* @__PURE__ */ createRequire(import.meta.url);
 
-//#endregion
-//#region src/lib/utils.ts
-function convertToWords(input) {
-	if (input.includes("-") || input.includes("_")) return input.split(/[-_]+/).filter((word) => word.length > 0).map((word) => word.toLowerCase());
-	return input.split(/(?=[A-Z])/).filter((word) => word.length > 0).map((word) => word.toLowerCase());
-}
-function invariant(cond, message) {
-	if (!cond) throw new Error(message);
-}
-function getErrorMessage(error, defaultMessage = "Something went wrong. Please try again") {
-	let errorMessage = defaultMessage;
-	if (error instanceof z$1.ZodError) errorMessage = error.issues.length ? error.issues.map((e) => e.message).join(", ") : error.message;
-	else if (error instanceof Error) errorMessage = error.message;
-	return errorMessage;
-}
-
 //#endregion
 //#region ../../node_modules/.bun/dayjs@1.11.18/node_modules/dayjs/dayjs.min.js
 var require_dayjs_min = /* @__PURE__ */ __commonJS({ "../../node_modules/.bun/dayjs@1.11.18/node_modules/dayjs/dayjs.min.js": ((exports, module) => {
@@ -2094,6 +2079,27 @@ function runWithRetries(fn, retries = 3, config = {}) {
 	return execute();
 }
 
+//#endregion
+//#region src/lib/utils.ts
+function convertToWords(input) {
+	if (input.includes("-") || input.includes("_")) return input.split(/[-_]+/).filter((word) => word.length > 0).map((word) => word.toLowerCase());
+	return input.split(/(?=[A-Z])/).filter((word) => word.length > 0).map((word) => word.toLowerCase());
+}
+function invariant(cond, message) {
+	if (!cond) throw new Error(message);
+}
+function getErrorMessage(error, defaultMessage = "Something went wrong. Please try again") {
+	let errorMessage = defaultMessage;
+	if (error instanceof z$1.ZodError) errorMessage = error.issues.length ? error.issues.map((e) => e.message).join(", ") : error.message;
+	else if (error instanceof Error) errorMessage = error.message;
+	return errorMessage;
+}
+
+//#endregion
+//#region src/internal/store.ts
+const runnerIdStore = new AsyncLocalStorage();
+const sessionIdStore = new AsyncLocalStorage();
+
 //#endregion
 //#region src/internal/loader.ts
 var Loader = class {
@@ -2105,7 +2111,9 @@ var Loader = class {
 	#webSocketClient;
 	#signal;
 	constructor(options) {
-		this.#sessionId = options.sessionId;
+		const sessionId = sessionIdStore.getStore();
+		if (!sessionId) throw new Error("sessionId not found");
+		this.#sessionId = sessionId;
 		this.#executionContext = options.executionContext;
 		this.#title = options.title;
 		this.#description = options.description;
@@ -2159,7 +2167,9 @@ var ProgressLoader = class {
 		this.#title = options.title;
 		this.#description = options.description;
 		this.#itemsInQueue = options.itemsInQueue;
-		this.#sessionId = options.sessionId;
+		const sessionId = sessionIdStore.getStore();
+		if (!sessionId) throw new Error("sessionId not found");
+		this.#sessionId = sessionId;
 		this.#executionContext = options.executionContext;
 		this.#webSocketClient = options.webSocketClient;
 		this.#signal = options.signal;
@@ -2247,10 +2257,15 @@ var IO = class {
 	*/
 	#pendingIOs = /* @__PURE__ */ new Map();
 	constructor(options, logger) {
-		this.#sessionId = options.sessionId;
-		this.#runnerId = options.runnerId;
 		this.#executionContext = options.executionContext;
+		const runnerId$1 = runnerIdStore.getStore();
+		if (!runnerId$1) throw new Error("runnerId not found");
+		this.#runnerId = runnerId$1;
+		const sessionId = sessionIdStore.getStore();
+		if (!sessionId) throw new Error("sessionId not found");
+		this.#sessionId = sessionId;
 		this.#logger = logger.child({
+			runnerId: this.#runnerId,
 			sessionId: this.#sessionId,
 			component: "IO"
 		});
@@ -2580,7 +2595,6 @@ var IO = class {
 		return new Loader({
 			title,
 			description,
-			sessionId: this.#sessionId,
 			webSocketClient: this.#webSocketClient,
 			signal: this.#signal,
 			executionContext: this.#executionContext
@@ -2596,7 +2610,6 @@ var IO = class {
 			description,
 			itemsInQueue,
 			webSocketClient: this.#webSocketClient,
-			sessionId: this.#sessionId,
 			signal: this.#signal,
 			executionContext: this.#executionContext
 		});
@@ -2677,6 +2690,88 @@ var IO = class {
 	}
 };
 
+//#endregion
+//#region src/internal/tool.ts
+const TOOL_HANDLER_TAG_NAME = Symbol("TOOL_HANDLER");
+const TOOL_TAG_NAME = Symbol("TOOL");
+/**
+* Tool class is responsible for running the tool's logic.
+* It manages the tool lifecycle, communication with the worker, and handles sending messages back to the ForgeRunner.
+* It uses the IO component to facilitate input/output operations for the tool.
+*/
+var Tool = class {
+	#runnerId;
+	#sessionId;
+	#metadata;
+	#abortController = new AbortController();
+	#handler;
+	#io;
+	#block;
+	#logger;
+	constructor(options, logger) {
+		this.#handler = options.handler;
+		const runnerId$1 = runnerIdStore.getStore();
+		if (!runnerId$1) throw new Error("runnerId not found");
+		this.#runnerId = runnerId$1;
+		const sessionId = sessionIdStore.getStore();
+		if (!sessionId) throw new Error("sessionId not found");
+		this.#sessionId = sessionId;
+		this.#metadata = {
+			toolId: options.toolId,
+			toolName: options.toolName,
+			toolDescription: options.toolDescription
+		};
+		this.#logger = logger.child({
+			runnerId: this.#runnerId,
+			sessionId: this.#sessionId,
+			component: "Tool"
+		});
+		this.#io = new IO({
+			webSocketClient: options.webSocketClient,
+			signal: this.#abortController.signal,
+			executionContext: { type: "TOOL" }
+		}, this.#logger);
+		this.#block = new Block({ signal: this.#abortController.signal });
+		this.#logger.debug("initialized Tool instance");
+	}
+	async run() {
+		return new Promise(async (resolve, reject) => {
+			if (this.#abortController.signal.aborted) {
+				this.#logger.info("tool execution aborted before start");
+				return Promise.reject(/* @__PURE__ */ new Error("tool execution aborted"));
+			}
+			this.#abortController.signal.addEventListener("abort", () => {
+				this.#logger.info("tool execution aborted");
+				reject(/* @__PURE__ */ new Error("Tool execution aborted"));
+			});
+			try {
+				this.#logger.info("running tool handler...");
+				const output = await this.#handler({
+					io: this.#io,
+					signal: this.#abortController.signal,
+					block: this.#block,
+					metadata: {
+						...this.#metadata,
+						sessionId: this.#sessionId,
+						runnerId: this.#runnerId
+					}
+				});
+				this.#logger.info("tool handler completed successfully");
+				this.#io[STOP_SYMBOL]();
+				resolve(output);
+			} catch (error) {
+				this.#logger.error(error, "error running tool handler");
+				return reject(error);
+			}
+		});
+	}
+	stop() {
+		this.#logger.info("stopping Tool...");
+		this.#abortController.abort();
+		this.#logger.info("Tool stopped");
+	}
+};
+
 //#endregion
 //#region src/internal/workflow-builder.ts
 var WorkflowBuilder = class {
@@ -2827,7 +2922,9 @@ var Agent = class {
 	#logger;
 	#webSocketClient;
 	#abortController = new AbortController();
-	#metadata;
+	#runnerId;
+	#sessionId;
+	#agentId;
 	#stepDefinitions;
 	#bootstrap;
 	#workflowBuilder;
@@ -2838,13 +2935,20 @@ var Agent = class {
 	#contextSchema;
 	#context;
 	constructor(definition, options, logger) {
-		this.#metadata = options.metadata;
+		const runnerId$1 = runnerIdStore.getStore();
+		if (!runnerId$1) throw new Error("runnerId not found");
+		this.#runnerId = runnerId$1;
+		const sessionId = sessionIdStore.getStore();
+		if (!sessionId) throw new Error("sessionId not found");
+		this.#sessionId = sessionId;
+		this.#agentId = options.agentId;
 		this.#stepDefinitions = definition.steps;
 		this.#bootstrap = definition.bootstrap;
 		this.#contextSchema = definition.contextSchema;
 		this.#webSocketClient = options.webSocketClient;
 		this.#logger = logger.child({
-			sessionId: this.#metadata.sessionId,
+			runnerId: this.#runnerId,
+			sessionId: this.#sessionId,
 			component: "AGENT"
 		});
 		this.#workflowBuilder = new WorkflowBuilder(this.#logger);
@@ -2885,7 +2989,7 @@ var Agent = class {
 			timestamp: Date.now(),
 			type: "AGENT_START",
 			data: {
-				sessionId: this.#metadata.sessionId,
+				sessionId: this.#sessionId,
 				agentWorkflow: this.#agentWorkflow,
 				executionContext: { type: "AGENT" }
 			}
@@ -2942,8 +3046,6 @@ var Agent = class {
 			const io = new IO({
 				webSocketClient: this.#webSocketClient,
 				executionContext,
-				runnerId: this.#metadata.runnerId,
-				sessionId: this.#metadata.sessionId,
 				signal: this.#abortController.signal
 			}, this.#logger);
 			const block = new Block({ signal: this.#abortController.signal });
@@ -2953,7 +3055,7 @@ var Agent = class {
 					timestamp: Date.now(),
 					type: "STEP_START",
 					data: {
-						sessionId: this.#metadata.sessionId,
+						sessionId: this.#sessionId,
 						executionContext
 					}
 				});
@@ -2964,9 +3066,9 @@ var Agent = class {
 					context: this.#context,
 					updateContext: this.#updateContext.bind(this),
 					metadata: {
-						agentId: this.#metadata.agentId,
-						runnerId: this.#metadata.runnerId,
-						sessionId: this.#metadata.sessionId,
+						agentId: this.#agentId,
+						runnerId: this.#runnerId,
+						sessionId: this.#sessionId,
 						stepName: stepDefinition.name,
 						stepDescription: stepDefinition.description
 					},
@@ -2982,7 +3084,7 @@ var Agent = class {
 					type: "STEP_COMPLETE",
 					data: {
 						output,
-						sessionId: this.#metadata.sessionId,
+						sessionId: this.#sessionId,
 						executionContext
 					}
 				});
@@ -2998,7 +3100,7 @@ var Agent = class {
 							stack: error instanceof Error ? error.stack : void 0,
 							code: error instanceof Error && "code" in error ? String(error.code) : void 0
 						},
-						sessionId: this.#metadata.sessionId
+						sessionId: this.#sessionId
 					}
 				});
 				this.#logger.error({ error }, "error executing step");
@@ -3018,8 +3120,6 @@ var Agent = class {
 			const io = new IO({
 				webSocketClient: this.#webSocketClient,
 				executionContext,
-				runnerId: this.#metadata.runnerId,
-				sessionId: this.#metadata.sessionId,
 				signal: this.#abortController.signal
 			}, this.#logger);
 			const block = new Block({ signal: this.#abortController.signal });
@@ -3029,7 +3129,7 @@ var Agent = class {
 					timestamp: Date.now(),
 					type: "STEP_START",
 					data: {
-						sessionId: this.#metadata.sessionId,
+						sessionId: this.#sessionId,
 						executionContext
 					}
 				});
@@ -3044,7 +3144,7 @@ var Agent = class {
 					type: "STEP_COMPLETE",
 					data: {
 						output: "Bootstrap completed successfully",
-						sessionId: this.#metadata.sessionId,
+						sessionId: this.#sessionId,
 						executionContext
 					}
 				});
@@ -3061,7 +3161,7 @@ var Agent = class {
 							stack: error instanceof Error ? error.stack : void 0,
 							code: error instanceof Error && "code" in error ? String(error.code) : void 0
 						},
-						sessionId: this.#metadata.sessionId
+						sessionId: this.#sessionId
 					}
 				});
 				throw error;
@@ -3081,71 +3181,4 @@ var Agent = class {
 };
 
 //#endregion
-//#region src/internal/tool.ts
-const TOOL_HANDLER_TAG_NAME = Symbol("TOOL_HANDLER");
-const TOOL_TAG_NAME = Symbol("TOOL");
-/**
-* Tool class is responsible for running the tool's logic.
-* It manages the tool lifecycle, communication with the worker, and handles sending messages back to the ForgeRunner.
-* It uses the IO component to facilitate input/output operations for the tool.
-*/
-var Tool = class {
-	#metadata;
-	#abortController = new AbortController();
-	#handler;
-	#io;
-	#block;
-	#logger;
-	constructor(options, logger) {
-		this.#metadata = options.metadata;
-		this.#handler = options.handler;
-		this.#logger = logger.child({
-			sessionId: this.#metadata.sessionId,
-			component: "TOOL"
-		});
-		this.#io = new IO({
-			sessionId: this.#metadata.sessionId,
-			runnerId: this.#metadata.runnerId,
-			webSocketClient: options.webSocketClient,
-			signal: this.#abortController.signal,
-			executionContext: { type: "TOOL" }
-		}, this.#logger);
-		this.#block = new Block({ signal: this.#abortController.signal });
-		this.#logger.debug("initialized Tool instance");
-	}
-	async run() {
-		return new Promise(async (resolve, reject) => {
-			if (this.#abortController.signal.aborted) {
-				this.#logger.info("tool execution aborted before start");
-				return Promise.reject(/* @__PURE__ */ new Error("tool execution aborted"));
-			}
-			this.#abortController.signal.addEventListener("abort", () => {
-				this.#logger.info("tool execution aborted");
-				reject(/* @__PURE__ */ new Error("Tool execution aborted"));
-			});
-			try {
-				this.#logger.info("running tool handler...");
-				const output = await this.#handler({
-					io: this.#io,
-					signal: this.#abortController.signal,
-					block: this.#block,
-					metadata: this.#metadata
-				});
-				this.#logger.info("tool handler completed successfully");
-				this.#io[STOP_SYMBOL]();
-				resolve(output);
-			} catch (error) {
-				this.#logger.error(error, "error running tool handler");
-				return reject(error);
-			}
-		});
-	}
-	stop() {
-		this.#logger.info("stopping Tool...");
-		this.#abortController.abort();
-		this.#logger.info("Tool stopped");
-	}
-};
-
-//#endregion
-export { __toESM as C, __require as S, stopToolMessageSchema as _, AGENT_TAG_NAME as a, invariant as b, runWithRetries as c, heartbeatAckMessageSchema as d, initCommunicationMessageSchema as f, stopAgentMessageSchema as g, startToolMessageSchema as h, AGENT_STEP_TAG_NAME as i, ackMessageSchema as l, startAgentMessageSchema as m, TOOL_TAG_NAME as n, Agent as o, runnerId as p, Tool as r, exponentialBackoff as s, TOOL_HANDLER_TAG_NAME as t, baseMessageSchema as u, convertToWords as v, __commonJS as x, getErrorMessage as y };
+export { __require as C, __commonJS as S, initCommunicationMessageSchema as _, TOOL_TAG_NAME as a, stopAgentMessageSchema as b, sessionIdStore as c, invariant as d, exponentialBackoff as f, heartbeatAckMessageSchema as g, baseMessageSchema as h, TOOL_HANDLER_TAG_NAME as i, convertToWords as l, ackMessageSchema as m, AGENT_TAG_NAME as n, Tool as o, runWithRetries as p, Agent as r, runnerIdStore as s, AGENT_STEP_TAG_NAME as t, getErrorMessage as u, startAgentMessageSchema as v, __toESM as w, stopToolMessageSchema as x, startToolMessageSchema as y };