@economic/agents 1.6.3 → 1.6.5

package/dist/index.d.mts

@@ -57,9 +57,9 @@ declare function buildLLMParams(config: BuildLLMParamsConfig): LLMParams;
  * type MyContext = AgentToolContext<MyBody>;
  * ```
  */
-type AgentToolContext<TBody = Record<string, unknown>, TSession = Record<string, unknown> | undefined> = TBody & {
-  logEvent: (message: string, payload?: Record<string, unknown>) => void | Promise<void>;
-  session?: TSession;
+type AgentToolContext<TBody = Record<string, unknown>, TJwtIdentity = Record<string, unknown> | undefined> = TBody & {
+  _logEvent: (message: string, payload?: Record<string, unknown>) => void | Promise<void>;
+  _jwtIdentity?: TJwtIdentity;
 };
 interface AgentEnv {
   AGENT_DB: D1Database;
@@ -95,12 +95,7 @@ interface JwtAuthConfig<TClaims extends Record<string, unknown> = Record<string,
  * For chat agents with message history, compaction, and conversation recording,
  * extend {@link ChatAgent} instead.
  */
-declare abstract class Agent<Env extends Cloudflare.Env = Cloudflare.Env> extends Agent$1<Env & AgentEnv> {
-  /**
-   * Verified session claims from JWT authentication.
-   * Set in `onConnect` when `getJwtAuthConfig` is implemented.
-   */
-  private session?;
+declare abstract class Agent<Env extends Cloudflare.Env = Cloudflare.Env, TJwtIdentity extends Record<string, unknown> = Record<string, unknown>> extends Agent$1<Env & AgentEnv> {
   /**
    * Override to enable JWT authentication on WebSocket connections.
    * Return the auth config based on the incoming request, or undefined to skip auth.
@@ -109,6 +104,17 @@ declare abstract class Agent<Env extends Cloudflare.Env = Cloudflare.Env> extend
    * @returns JWT auth config or undefined to skip authentication
    */
   protected getJwtAuthConfig?(request: Request): JwtAuthConfig<Record<string, unknown>> | undefined;
+  /**
+   * The identity associated with the session.
+   * Define getIdentity to return the identity from the request.
+   */
+  protected jwtIdentity: TJwtIdentity;
+  /**
+   * Returns the identity following verification of the JWT token - getJwtAuthConfig is required.
+   * @returns The identity from the request.
+   * @param token - The token from the request.
+   */
+  protected getJwtIdentity?(token: string): Promise<TJwtIdentity>;
   /**
    * Returns the user ID from the durable object name.
    */
@@ -152,7 +158,7 @@ interface MessageRating {
  * Skill loading, compaction, and LLM calls use `buildLLMParams` from
  * `@economic/agents` inside `onChatMessage`.
  */
-declare abstract class ChatAgent<Env extends Cloudflare.Env = Cloudflare.Env, TSessionIdentity extends Record<string, unknown> = Record<string, unknown>> extends AIChatAgent<Env & ChatAgentEnv> {
+declare abstract class ChatAgent<Env extends Cloudflare.Env = Cloudflare.Env, TJwtIdentity extends Record<string, unknown> = Record<string, unknown>> extends AIChatAgent<Env & ChatAgentEnv> {
   /**
    * The binding of the Durable Object instance for this agent.
    */
@@ -188,11 +194,6 @@ declare abstract class ChatAgent<Env extends Cloudflare.Env = Cloudflare.Env, TS
    * Default is 15.
    */
   protected maxMessagesBeforeCompaction?: number | undefined;
-  /**
-   * Verified session claims from JWT authentication.
-   * Set in `onConnect` when `getJwtAuthConfig` is implemented.
-   */
-  private session?;
   /**
    * Override to enable JWT authentication on WebSocket connections.
    * Return the auth config based on the incoming request, or undefined to skip auth.
@@ -205,13 +206,13 @@ declare abstract class ChatAgent<Env extends Cloudflare.Env = Cloudflare.Env, TS
    * The identity associated with the session.
    * Define getIdentity to return the identity from the request.
    */
-  protected identity: TSessionIdentity;
+  protected jwtIdentity: TJwtIdentity;
   /**
-   * Returns the identity from the request.
+   * Returns the identity following verification of the JWT token - getJwtAuthConfig is required.
    * @returns The identity from the request.
    * @param token - The token from the request.
    */
-  protected getIdentity(_token: string): Promise<TSessionIdentity>;
+  protected getJwtIdentity?(token: string): Promise<TJwtIdentity>;
   /**
    * Returns the user ID from the durable object name.
    */

package/dist/index.mjs

@@ -474,10 +474,10 @@ async function verifyJwt(request, config) {
 */
 var Agent = class extends Agent$1 {
 	/**
-	* Verified session claims from JWT authentication.
-	* Set in `onConnect` when `getJwtAuthConfig` is implemented.
+	* The identity associated with the session.
+	* Define getIdentity to return the identity from the request.
 	*/
-	session;
+	jwtIdentity = {};
 	/**
 	* Returns the user ID from the durable object name.
 	*/
@@ -510,7 +510,8 @@ var Agent = class extends Agent$1 {
 					connection.close(result.status === 401 ? 4001 : 4003, result.message);
 					return;
 				}
-				this.session = result.claims;
+				const token = extractTokenFromConnectRequest(ctx.request);
+				if (token) this.jwtIdentity = await this.getJwtIdentity?.(token) ?? {};
 			}
 		}
 		return super.onConnect(connection, ctx);
@@ -541,8 +542,8 @@ var Agent = class extends Agent$1 {
 		const experimental_context = {
 			...config.experimental_context,
 			...config.options?.body,
-			session: this.session,
-			logEvent: this.logEvent.bind(this)
+			_jwtIdentity: this.jwtIdentity,
+			_logEvent: this.logEvent.bind(this)
 		};
 		const onFinish = async (event) => {
 			this.logEvent("Turn completed", buildTurnLogPayload(event));
@@ -603,7 +604,7 @@ function formatMessagesForSummary(messages) {
 		}
 		if (parts.length > 0) lines.push(`${roleLabel}: ${parts.join(" ")}`);
 	}
-	return lines.join("\n\n");
+	return lines.join("\n");
 }
 /**
 * Calls the model to produce a concise summary of old + recent message windows.
@@ -661,14 +662,37 @@ async function compactIfNeeded(messages, model, tailSize) {
 }
 //#endregion
 //#region src/server/agent-chat/features/conversations/conversations.ts
+/**
+* Renders only the user- and assistant-visible text parts of a message
+* array for conversation title/summary generation.
+*
+* Tool calls, tool results, system messages, and reasoning parts are
+* intentionally excluded so the model receives a clean transcript.
+*/
+function formatMessagesForConversationSummary(messages) {
+	const lines = [];
+	for (const message of messages) {
+		if (message.role !== "user" && message.role !== "assistant") continue;
+		let text = "";
+		if (typeof message.content === "string") text = message.content.trim();
+		else text = message.content.filter((part) => part.type === "text").map((part) => part.text.trim()).filter(Boolean).join(" ");
+		if (!text) continue;
+		const roleLabel = message.role === "user" ? "User" : "Assistant";
+		lines.push(`${roleLabel}: ${text}`);
+	}
+	return lines.join("\n");
+}
+/**
+* Number of recent messages passed to `generateSummary` for rolling
+* summarization. Keeping this bounded prevents the prompt growing
+* unboundedly regardless of conversation length.
+*/
+const SUMMARY_CONTEXT_MESSAGES = 15;
 async function recordConversationSummary(db, durableObjectName, messages, model) {
 	if (!await getConversationSummary(db, durableObjectName)) {
 		const { title, summary } = await generateTitleAndSummary(messages, model);
 		await upsertConversationSummary(db, durableObjectName, title, summary);
-	} else {
-		await upsertConversationSummary(db, durableObjectName);
-		if (messages.length % SUMMARY_CONTEXT_MESSAGES === 0) await generateConversationSummary(db, durableObjectName, messages, model);
-	}
+	} else if (messages.length <= 7 || messages.length % SUMMARY_CONTEXT_MESSAGES === 0) await generateConversationSummary(db, durableObjectName, messages, model);
 }
 /**
 * Records a conversation row in the `conversations` D1 table.
@@ -688,12 +712,6 @@ async function upsertConversationSummary(db, durableObjectName, title, summary)
          updated_at = excluded.updated_at`).bind(durableObjectName, title ?? null, summary ?? null, now, now).run();
 }
 /**
-* Number of recent messages passed to `generateSummary` for rolling
-* summarization. Keeping this bounded prevents the prompt growing
-* unboundedly regardless of conversation length.
-*/
-const SUMMARY_CONTEXT_MESSAGES = 30;
-/**
 * Returns the current `title` and `summary` for a conversation row,
 * or `null` if the row does not exist yet.
 */
@@ -732,7 +750,7 @@ async function updateConversationSummary(db, durableObjectName, title, summary)
 */
 async function generateTitleAndSummary(messages, model, existingSummary) {
 	const recentMessages = await convertToModelMessages(messages.slice(-SUMMARY_CONTEXT_MESSAGES));
-	const previousContext = existingSummary ? `Previous summary: ${existingSummary}\n\nMost recent messages:` : "Conversation:";
+	const previousContext = existingSummary ? `Previous summary:\n${existingSummary}\n\nMost recent messages:` : "Conversation:";
 	const { output } = await generateText({
 		model,
 		output: Output.object({ schema: jsonSchema({
@@ -744,12 +762,12 @@ async function generateTitleAndSummary(messages, model, existingSummary) {
 				},
 				summary: {
 					type: "string",
-					description: "2-3 sentence summary. If the conversation direction has changed from the previous summary, reflect the new direction."
+					description: "1-2 short sentence summary of the conversation. If the conversation direction has changed from the previous summary, reflect the new direction."
 				}
 			},
 			required: ["title", "summary"]
 		}) }),
-		prompt: `${previousContext}\n\n${formatMessagesForSummary(recentMessages)}`
+		prompt: `${previousContext}\n\n${formatMessagesForConversationSummary(recentMessages)}`
 	});
 	return output;
 }
@@ -844,23 +862,10 @@ var ChatAgent = class extends AIChatAgent {
 	*/
 	maxMessagesBeforeCompaction = 15;
 	/**
-	* Verified session claims from JWT authentication.
-	* Set in `onConnect` when `getJwtAuthConfig` is implemented.
-	*/
-	session;
-	/**
 	* The identity associated with the session.
 	* Define getIdentity to return the identity from the request.
 	*/
-	identity = {};
-	/**
-	* Returns the identity from the request.
-	* @returns The identity from the request.
-	* @param token - The token from the request.
-	*/
-	async getIdentity(_token) {
-		return Promise.resolve({});
-	}
+	jwtIdentity = {};
 	/**
 	* Returns the user ID from the durable object name.
 	*/
@@ -893,11 +898,10 @@ var ChatAgent = class extends AIChatAgent {
 					connection.close(result.status === 401 ? 4001 : 4003, result.message);
 					return;
 				}
-				this.session = result.claims;
 			}
+			const token = extractTokenFromConnectRequest(ctx.request);
+			if (token) this.jwtIdentity = await this.getJwtIdentity?.(token) ?? {};
 		}
-		const token = extractTokenFromConnectRequest(ctx.request);
-		if (token) this.identity = await this.getIdentity(token);
 		return super.onConnect(connection, ctx);
 	}
 	/**
@@ -931,8 +935,8 @@ var ChatAgent = class extends AIChatAgent {
 		const experimental_context = {
 			...config.experimental_context,
 			...config.options?.body,
-			session: this.session,
-			logEvent: this.logEvent.bind(this)
+			_jwtIdentity: this.jwtIdentity,
+			_logEvent: this.logEvent.bind(this)
 		};
 		const messages = await convertToModelMessages(this.messages);
 		const fastModel = this.getFastModel();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@economic/agents",
-  "version": "1.6.3",
+  "version": "1.6.5",
   "description": "A starter for creating a TypeScript package.",
   "license": "MIT",
   "bin": {