@economic/agents 0.0.1-alpha.17 → 0.0.1-alpha.18

package/dist/index.d.mts

@@ -113,28 +113,17 @@ declare abstract class AIChatAgent<Env extends Cloudflare.Env = Cloudflare.Env>
    */
   protected log(message: string, payload?: Record<string, unknown>): Promise<void>;
   /**
-   * Records this conversation in the `conversations` D1 table and resets
-   * the idle summarization timer. Called automatically from `persistMessages`
-   * after every turn.
-   *
-   * After each upsert, any pending `generateSummary` schedule is cancelled
-   * and a new one is set for 30 minutes from now. If the user sends another
-   * message before the timer fires, the schedule is cancelled and reset again
-   * (debounce). When the conversation goes idle, `generateSummary` fires and
-   * writes the LLM-generated title and summary to D1.
+   * Records this conversation in the `conversations` D1 table and triggers
+   * LLM-based title/summary generation when appropriate. Called automatically
+   * from `persistMessages` after every turn.
+   *
+   * On the first turn (no existing row), awaits `generateTitleAndSummary` and
+   * inserts the row with title and summary already populated. On subsequent
+   * turns, upserts the timestamp and fire-and-forgets a summary refresh every
+   * `SUMMARY_CONTEXT_MESSAGES` messages (when the context window fully turns
+   * over). Neither path blocks the response to the client.
    */
   private recordConversation;
-  /**
-   * Generates a title and summary for the conversation after 30 minutes of
-   * inactivity. Invoked automatically by the Cloudflare Agents SDK scheduler
-   * — do not call this directly.
-   *
-   * Delegates to `generateConversationSummary` in `features/conversations`,
-   * which fetches the previous summary, slices to the last
-   * `SUMMARY_CONTEXT_MESSAGES` messages, calls `fastModel` with a structured
-   * output schema, and writes the result back to D1.
-   */
-  generateSummary(): Promise<void>;
   /**
    * Builds the parameter object for a `streamText` or `generateText` call,
    * pre-filling `messages`, `activeSkills`, and `fastModel` from this agent instance.

package/dist/index.mjs

@@ -3724,21 +3724,19 @@ function superRefine(fn) {
 * Records a conversation row in the `conversations` D1 table.
 *
 * Called by `AIChatAgent` after every turn. On first call for a given
-* `durableObjectId` the row is inserted with `created_at` set to now.
+* `durableObjectId` the row is inserted with `created_at` set to now,
+* and with the provided `title` and `summary` if supplied.
 * On subsequent calls only `user_id` and `updated_at` are refreshed —
-* `created_at`, `title`, and `summary` are never overwritten.
-*
-* After upserting, `AIChatAgent` cancels any pending `generateSummary`
-* schedule and resets it to fire 30 minutes from now, debouncing the
-* idle timer on every turn.
+* `created_at`, `title`, and `summary` are never overwritten, preserving
+* any user edits.
 */
-async function recordConversation(db, durableObjectId, userId) {
+async function recordConversation(db, durableObjectId, userId, title, summary) {
 	const now = (/* @__PURE__ */ new Date()).toISOString();
 	await db.prepare(`INSERT INTO conversations (durable_object_id, user_id, title, summary, created_at, updated_at)
-       VALUES (?, ?, NULL, NULL, ?, ?)
+       VALUES (?, ?, ?, ?, ?, ?)
        ON CONFLICT(durable_object_id) DO UPDATE SET
          user_id    = excluded.user_id,
-         updated_at = excluded.updated_at`).bind(durableObjectId, userId, now, now).run();
+         updated_at = excluded.updated_at`).bind(durableObjectId, userId, title ?? null, summary ?? null, now, now).run();
 }
 /**
 * Returns the current `title` and `summary` for a conversation row,
@@ -3754,19 +3752,19 @@ async function updateConversationSummary(db, durableObjectId, title, summary) {
 	await db.prepare(`UPDATE conversations SET title = ?, summary = ? WHERE durable_object_id = ?`).bind(title, summary, durableObjectId).run();
 }
 /**
-* Generates a title and summary for a conversation using the provided model
-* and writes the result back to D1.
+* Generates a title and summary for a conversation using the provided model.
+* Returns the result without writing to D1.
 *
-* Fetches any existing summary first so the model can detect direction changes.
-* Only the last `SUMMARY_CONTEXT_MESSAGES` messages are passed to keep the
-* prompt bounded regardless of total conversation length.
+* Pass `existingSummary` so the model can detect direction changes when
+* updating an existing summary. Omit it (or pass undefined) for the initial
+* generation.
 *
-* Called by `AIChatAgent.generateSummary()` after the idle timer fires.
+* Only the last `SUMMARY_CONTEXT_MESSAGES` messages are used to keep the
+* prompt bounded regardless of total conversation length.
 */
-async function generateConversationSummary(db, durableObjectId, messages, model) {
-	const existing = await getConversationSummary(db, durableObjectId);
+async function generateTitleAndSummary(messages, model, existingSummary) {
 	const recentMessages = await convertToModelMessages(messages.slice(-30));
-	const previousContext = existing?.summary ? `Previous summary: ${existing.summary}\n\nMost recent messages:` : "Conversation:";
+	const previousContext = existingSummary ? `Previous summary: ${existingSummary}\n\nMost recent messages:` : "Conversation:";
 	const { output } = await generateText({
 		model,
 		output: Output.object({ schema: object({
@@ -3775,7 +3773,22 @@ async function generateConversationSummary(db, durableObjectId, messages, model)
 		}) }),
 		prompt: `${previousContext}\n\n${formatMessagesForSummary(recentMessages)}`
 	});
-	await updateConversationSummary(db, durableObjectId, output.title, output.summary);
+	return output;
+}
+/**
+* Generates a title and summary for a conversation using the provided model
+* and writes the result back to D1.
+*
+* Fetches any existing summary first so the model can detect direction changes.
+* Only the last `SUMMARY_CONTEXT_MESSAGES` messages are passed to keep the
+* prompt bounded regardless of total conversation length.
+*
+* Called by `AIChatAgent` every `SUMMARY_CONTEXT_MESSAGES` messages after
+* the first turn.
+*/
+async function generateConversationSummary(db, durableObjectId, messages, model) {
+	const { title, summary } = await generateTitleAndSummary(messages, model, (await getConversationSummary(db, durableObjectId))?.summary ?? void 0);
+	await updateConversationSummary(db, durableObjectId, title, summary);
 }
 //#endregion
 //#region src/agents/AIChatAgent.ts
@@ -3828,39 +3841,31 @@ var AIChatAgent = class extends AIChatAgent$1 {
 		await insertAuditEvent(context.db, this.ctx.id.toString(), context.userId, message, payload);
 	}
 	/**
-	* Records this conversation in the `conversations` D1 table and resets
-	* the idle summarization timer. Called automatically from `persistMessages`
-	* after every turn.
+	* Records this conversation in the `conversations` D1 table and triggers
+	* LLM-based title/summary generation when appropriate. Called automatically
+	* from `persistMessages` after every turn.
 	*
-	* After each upsert, any pending `generateSummary` schedule is cancelled
-	* and a new one is set for 30 minutes from now. If the user sends another
-	* message before the timer fires, the schedule is cancelled and reset again
-	* (debounce). When the conversation goes idle, `generateSummary` fires and
-	* writes the LLM-generated title and summary to D1.
+	* On the first turn (no existing row), awaits `generateTitleAndSummary` and
+	* inserts the row with title and summary already populated. On subsequent
+	* turns, upserts the timestamp and fire-and-forgets a summary refresh every
+	* `SUMMARY_CONTEXT_MESSAGES` messages (when the context window fully turns
+	* over). Neither path blocks the response to the client.
 	*/
-	async recordConversation() {
+	async recordConversation(messageCount) {
 		const context = this.resolveD1Context();
 		if (!context) return;
-		await recordConversation(context.db, this.ctx.id.toString(), context.userId);
-		const existing = this.getSchedules({ type: "delayed" }).find((s) => s.callback === "generateSummary");
-		if (existing) await this.cancelSchedule(existing.id);
-		await this.schedule(1800, "generateSummary", {});
-	}
-	/**
-	* Generates a title and summary for the conversation after 30 minutes of
-	* inactivity. Invoked automatically by the Cloudflare Agents SDK scheduler
-	* — do not call this directly.
-	*
-	* Delegates to `generateConversationSummary` in `features/conversations`,
-	* which fetches the previous summary, slices to the last
-	* `SUMMARY_CONTEXT_MESSAGES` messages, calls `fastModel` with a structured
-	* output schema, and writes the result back to D1.
-	*/
-	async generateSummary() {
-		const context = this.resolveD1Context();
-		if (!context) return;
-		await generateConversationSummary(context.db, this.ctx.id.toString(), this.messages, this.fastModel);
-		this.log("conversation summary generated");
+		const durableObjectId = this.ctx.id.toString();
+		if (!await getConversationSummary(context.db, durableObjectId)) {
+			const { title, summary } = await generateTitleAndSummary(this.messages, this.fastModel);
+			await recordConversation(context.db, durableObjectId, context.userId, title, summary);
+			this.log("conversation summary generated");
+		} else {
+			await recordConversation(context.db, durableObjectId, context.userId);
+			if (messageCount % 30 === 0) {
+				generateConversationSummary(context.db, durableObjectId, this.messages, this.fastModel);
+				this.log("conversation summary updated");
+			}
+		}
 	}
 	/**
 	* Builds the parameter object for a `streamText` or `generateText` call,
@@ -3913,7 +3918,7 @@ var AIChatAgent = class extends AIChatAgent$1 {
 	* Returns an empty array if no skills have been loaded yet.
 	*/
 	async getLoadedSkills() {
-		return getStoredSkills(this.sql);
+		return getStoredSkills(this.sql.bind(this));
 	}
 	/**
 	* Extracts skill state from activate_skill results, persists to DO SQLite,
@@ -3944,9 +3949,9 @@ var AIChatAgent = class extends AIChatAgent$1 {
 				} catch {}
 			}
 		}
-		if (latestSkillState !== void 0) saveStoredSkills(this.sql, latestSkillState);
+		if (latestSkillState !== void 0) saveStoredSkills(this.sql.bind(this), latestSkillState);
 		this.log("turn completed", buildTurnSummary(messages, latestSkillState ?? []));
-		this.recordConversation();
+		this.recordConversation(messages.length);
 		const filtered = filterEphemeralMessages(messages);
 		return super.persistMessages(filtered, excludeBroadcastIds, options);
 	}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@economic/agents",
-  "version": "0.0.1-alpha.17",
+  "version": "0.0.1-alpha.18",
   "description": "A starter for creating a TypeScript package.",
   "homepage": "https://github.com/author/library#readme",
   "bugs": {