@economic/agents 2.14.0 → 2.14.1

package/dist/index.d.mts

@@ -226,6 +226,16 @@ interface TurnSnapshot {
   frameworkToolNames: string[];
   messages: ModelMessage[];
 }
+type CurrentTurnState<RequestContext extends Record<string, unknown>, UserContext extends Record<string, unknown>> = {
+  context?: ToolContext<RequestContext, UserContext>;
+  inputTools: ToolSet;
+  preparedTools: ToolSet;
+  unavailableSkillIds: Set<string>;
+  skillsSystemBlock: string;
+  visibleSkillNames: string[];
+  authorizedSkillNames: Set<string>;
+  baseSystem: string;
+};
 type Authentication<UserContext> = {
   success: true;
   userContext?: UserContext;
@@ -249,8 +259,9 @@ declare abstract class Agent<RequestContext extends Record<string, unknown> = Re
   onStart(): Promise<void>;
   configureSession(session: Session): Session;
   onConnect(connection: Connection, ctx: ConnectionContext): Promise<void>;
-  private _turn;
-  private getToolContextForCurrentTurn;
+  protected _turn: CurrentTurnState<RequestContext, UserContext>;
+  protected createToolContextForCurrentTurn(requestContext?: RequestContext): ToolContext<RequestContext, UserContext>;
+  protected getToolContextForCurrentTurn(): ToolContext<RequestContext, UserContext> | undefined;
   /**
    * Merges the client request `body` into `experimental_context` for tools
    * returned from {@link getTools} only (Think-internal tools are unchanged).

package/dist/index.mjs

@@ -1158,13 +1158,17 @@ var Agent = class extends Think {
 		authorizedSkillNames: /* @__PURE__ */ new Set(),
 		baseSystem: ""
 	};
-	getToolContextForCurrentTurn(requestContext = {}) {
+	createToolContextForCurrentTurn(requestContext = {}) {
 		const connectionState = getCurrentAgent().connection?.state;
 		return {
+			...this._turn.context,
 			...requestContext,
 			_userContext: connectionState?.userContext ?? this.userContext
 		};
 	}
+	getToolContextForCurrentTurn() {
+		return this._turn.context;
+	}
 	/**
 	* Merges the client request `body` into `experimental_context` for tools
 	* returned from {@link getTools} only (Think-internal tools are unchanged).
@@ -1175,8 +1179,11 @@ var Agent = class extends Think {
 	async beforeTurn(ctx) {
 		this.analytics.promptReceived(ctx.messages, ctx.continuation);
 		this.registerInstanceActivity();
-		const body = await this._applyTokenExchange(ctx.body);
-		this._turn.context = this.getToolContextForCurrentTurn(body);
+		const body = await this._applyTokenExchange({
+			...this._turn.context,
+			...ctx.body
+		});
+		this._turn.context = this.createToolContextForCurrentTurn(body);
 		this._turn.inputTools = ctx.tools;
 		const activeSkillName = this._getLastActivatedSkillName(ctx.messages);
 		const { tools, activeTools } = await this._prepareTools(ctx.tools, activeSkillName, true);
@@ -2851,38 +2858,48 @@ var ChatAgent = class extends Agent {
 	* returned immediately and each event reaches the client as it is produced.
 	*/
 	async streamMessage(request) {
-		let message;
+		let payload;
 		try {
-			({message} = await request.json());
+			payload = await request.json();
 		} catch {
-			return Response.json({ error: "Invalid request body" }, { status: 400 });
+			return Response.json({ error: "Request body must be valid JSON" }, { status: 400 });
 		}
-		if (typeof message === "string") message = {
-			id: crypto.randomUUID(),
-			role: "user",
-			parts: [{
-				type: "text",
-				text: message
-			}]
-		};
-		else if (!message || message.role !== "user" || !Array.isArray(message.parts)) return Response.json({ error: "message must be a user message" }, { status: 400 });
+		if (!payload || typeof payload !== "object" || Array.isArray(payload)) return Response.json({ error: "Request body must be an object" }, { status: 400 });
+		let { message, context } = payload;
+		if (context !== void 0 && (context === null || typeof context !== "object" || Array.isArray(context))) return Response.json({ error: "Context must be an object" }, { status: 400 });
+		if (typeof message === "string") {
+			if (!message.trim()) return Response.json({ error: "Message must not be empty" }, { status: 400 });
+			message = {
+				id: crypto.randomUUID(),
+				role: "user",
+				parts: [{
+					type: "text",
+					text: message
+				}]
+			};
+		} else if (!message || typeof message !== "object" || Array.isArray(message) || typeof message.id !== "string" || !message.id || message.role !== "user" || !Array.isArray(message.parts)) return Response.json({ error: "Message must be a non-empty string or a user UIMessage" }, { status: 400 });
+		const input = context ? () => {
+			this._turn.context = this.createToolContextForCurrentTurn(context);
+			return [message];
+		} : message;
 		const encoder = new TextEncoder();
 		const readable = new ReadableStream({ start: async (controller) => {
 			const send = (event, data) => controller.enqueue(encoder.encode(`event: ${event}\ndata: ${data}\n\n`));
+			const sendError = (error) => send("error", JSON.stringify({ error: error instanceof Error ? error.message : String(error) }));
 			try {
 				await this.runTurn({
 					mode: "stream",
-					input: message,
+					input,
 					callback: {
 						onStart: (event) => send("start", JSON.stringify(event)),
 						onEvent: (json) => send("chunk", json),
 						onDone: () => send("done", "{}"),
-						onError: (error) => send("error", JSON.stringify({ error })),
+						onError: sendError,
 						onInterrupted: () => send("interrupted", "{}")
 					}
 				});
 			} catch (error) {
-				send("error", JSON.stringify({ error: error instanceof Error ? error.message : String(error) }));
+				sendError(error);
 			} finally {
 				controller.close();
 			}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@economic/agents",
-  "version": "2.14.0",
+  "version": "2.14.1",
   "description": "A starter for creating a TypeScript package.",
   "license": "MIT",
   "files": [