@langgraph-js/pure-graph 3.2.4 → 3.3.0

package/dist/{stream-8w3eP9hU.js → stream-Bleed1lp.js}

@@ -486,7 +486,8 @@ class KyselyThreadsManager {
           metadata: this.adapter.dbToJson(existing.metadata),
           status: existing.status,
           values: existing.values ? this.adapter.dbToJson(existing.values) : null,
-          interrupts: this.adapter.dbToJson(existing.interrupts)
+          interrupts: this.adapter.dbToJson(existing.interrupts),
+          title: existing.title
         };
       }
     }
@@ -507,7 +508,8 @@ class KyselyThreadsManager {
       metadata,
       status: "idle",
       values: null,
-      interrupts
+      interrupts,
+      title: null
     };
   }
   async search(query) {
@@ -516,7 +518,7 @@ class KyselyThreadsManager {
     if (query?.select) {
       selectedFields = new Set(query.select);
     } else if (query?.withoutDetails) {
-      selectedFields = /* @__PURE__ */ new Set(["thread_id", "created_at", "updated_at", "metadata", "status"]);
+      selectedFields = /* @__PURE__ */ new Set(["thread_id", "created_at", "updated_at", "metadata", "status", "title"]);
     } else {
       selectedFields = /* @__PURE__ */ new Set([
         "thread_id",
@@ -525,7 +527,8 @@ class KyselyThreadsManager {
         "metadata",
         "status",
         "values",
-        "interrupts"
+        "interrupts",
+        "title"
       ]);
     }
     const selections = [];
@@ -536,6 +539,7 @@ class KyselyThreadsManager {
     if (selectedFields.has("status")) selections.push("status");
     if (selectedFields.has("values")) selections.push("values");
     if (selectedFields.has("interrupts")) selections.push("interrupts");
+    if (selectedFields.has("title")) selections.push("title");
     if (selections.length > 0) {
       queryBuilder = queryBuilder.select(selections);
     } else {
@@ -579,6 +583,7 @@ class KyselyThreadsManager {
       if (selectedFields.has("values"))
         result.values = row.values ? this.adapter.dbToJson(row.values) : null;
       if (selectedFields.has("interrupts")) result.interrupts = this.adapter.dbToJson(row.interrupts);
+      if (selectedFields.has("title")) result.title = row.title;
       return result;
     });
   }
@@ -595,7 +600,8 @@ class KyselyThreadsManager {
       metadata: this.adapter.dbToJson(row.metadata),
       status: row.status,
       values: row.values ? this.adapter.dbToJson(row.values) : null,
-      interrupts: this.adapter.dbToJson(row.interrupts)
+      interrupts: this.adapter.dbToJson(row.interrupts),
+      title: row.title
     };
   }
   async set(threadId, thread) {
@@ -618,6 +624,9 @@ class KyselyThreadsManager {
     if (thread.interrupts !== void 0) {
       updates.interrupts = this.adapter.jsonToDb(thread.interrupts);
     }
+    if (thread.title !== void 0) {
+      updates.title = thread.title;
+    }
     await this.db.updateTable("threads").set(updates).where("thread_id", "=", threadId).execute();
   }
   async delete(threadId) {
@@ -747,6 +756,9 @@ class KyselyThreadsManager {
     if (updates.interrupts !== void 0) {
       patchUpdates.interrupts = this.adapter.jsonToDb(updates.interrupts);
     }
+    if (updates.title !== void 0) {
+      patchUpdates.title = updates.title;
+    }
     await this.db.updateTable("threads").set(patchUpdates).where("thread_id", "=", threadId).execute();
     return await this.get(threadId);
   }
@@ -828,7 +840,8 @@ class KyselyThreadsManager {
       metadata: this.adapter.jsonToDb(originalThread.metadata),
       status: originalThread.status,
       values: originalThread.values ? this.adapter.jsonToDb(originalThread.values) : null,
-      interrupts: this.adapter.jsonToDb(originalThread.interrupts)
+      interrupts: this.adapter.jsonToDb(originalThread.interrupts),
+      title: originalThread.title
     }).execute();
     const checkpoints = await this.db.selectFrom("checkpoints").selectAll().where("thread_id", "=", threadId).orderBy("created_at", "asc").execute();
     for (const cp of checkpoints) {
@@ -861,6 +874,13 @@ class KyselyThreadsManager {
       metadata: this.adapter.jsonToDb(metadata || {})
     }).execute();
   }
+  async setTitleIfNull(threadId, title) {
+    const result = await this.db.updateTable("threads").set({
+      title,
+      updated_at: this.adapter.dateToDb(/* @__PURE__ */ new Date())
+    }).where("thread_id", "=", threadId).where("title", "is", null).executeTakeFirst();
+    return result.numUpdatedRows > 0n;
+  }
 }
 
 function _parseShallowKey(key) {
@@ -1194,7 +1214,6 @@ class MemoryStreamQueue extends BaseStreamQueue {
     if (this.isDestroyed) return;
     const data = this.compressMessages ? await this.encodeData(item) : item;
     process.env.LG_TEMP_MESSAGE !== "true" && this.data.push(data);
-    console.log(process.env.LG_TEMP_MESSAGE !== "true");
     this.emit("dataChange", data);
   }
   onDataChange(listener) {
@@ -1367,7 +1386,8 @@ class MemoryThreadsManager {
       metadata: payload?.metadata || {},
       status: "idle",
       values: null,
-      interrupts: {}
+      interrupts: {},
+      title: null
     };
     this.checkpoints.set(threadId, []);
     this.threads.push(thread);
@@ -1445,7 +1465,8 @@ class MemoryThreadsManager {
           "metadata",
           "status",
           "values",
-          "interrupts"
+          "interrupts",
+          "title"
         ]);
       }
       if (includeFields.has("thread_id")) result.thread_id = i.thread_id;
@@ -1455,6 +1476,7 @@ class MemoryThreadsManager {
       if (includeFields.has("status")) result.status = i.status;
       if (includeFields.has("values")) result.values = i.values;
       if (includeFields.has("interrupts")) result.interrupts = i.interrupts;
+      if (includeFields.has("title")) result.title = i.title;
       return result;
     });
   }
@@ -1685,6 +1707,21 @@ class MemoryThreadsManager {
     checkpoints.push(checkpoint);
     this.checkpoints.set(threadId, checkpoints);
   }
+  async setTitleIfNull(threadId, title) {
+    const index = this.threads.findIndex((t) => t.thread_id === threadId);
+    if (index === -1) {
+      throw new Error(`Thread with ID ${threadId} not found.`);
+    }
+    if (this.threads[index].title === null) {
+      this.threads[index] = {
+        ...this.threads[index],
+        title,
+        updated_at: (/* @__PURE__ */ new Date()).toISOString()
+      };
+      return true;
+    }
+    return false;
+  }
 }
 
 const createCheckPointer = async () => {
@@ -1736,7 +1773,7 @@ const createMessageQueue = async () => {
   let q;
   if (process.env.REDIS_URL) {
     console.debug("LG | Using redis as stream queue");
-    const { RedisStreamQueue } = await import('./queue-CPJTzcgl.js');
+    const { RedisStreamQueue } = await import('./queue-CS3K-zd_.js');
     q = RedisStreamQueue;
   } else {
     q = MemoryStreamQueue;
@@ -1755,7 +1792,7 @@ const createThreadManager = async (config) => {
     const dbType = getDatabaseType(process.env.DATABASE_URL);
     if (dbType === "remote") {
       console.debug("LG | Using Remote PostgreSQL ThreadsManager");
-      const { RemoteKyselyThreadsManager } = await import('./remote-threads-CrG03ZS7.js');
+      const { RemoteKyselyThreadsManager } = await import('./remote-threads-UpSB8X-e.js');
       const threadsManager = new RemoteKyselyThreadsManager(process.env.DATABASE_URL);
       if (process.env.DATABASE_INIT === "true") {
         await threadsManager.setup();
@@ -1776,7 +1813,7 @@ const createThreadManager = async (config) => {
   }
   if (process.env.SQLITE_DATABASE_URI && config.checkpointer) {
     console.debug("LG | Using SQLite ThreadsManager");
-    const { SQLiteAdapter } = await import('./sqlite-adapter-CJXgit1j.js');
+    const { SQLiteAdapter } = await import('./sqlite-adapter-D5g0LbUq.js');
     const database = config.checkpointer.db;
     const threadsManager = new KyselyThreadsManager(new SQLiteAdapter(database));
     await threadsManager.setup();
@@ -1785,11 +1822,64 @@ const createThreadManager = async (config) => {
   return new MemoryThreadsManager();
 };
 
+function extractTextContent(content) {
+  if (typeof content === "string") {
+    return content;
+  }
+  if (Array.isArray(content)) {
+    const textPart = content.find((part) => part.type === "text");
+    return textPart?.text || "";
+  }
+  if (content?.text) {
+    return content.text;
+  }
+  return "";
+}
+const defaultTitleGenerator = (state, context) => {
+  const messages = state?.messages;
+  if (!messages || !Array.isArray(messages) || messages.length === 0) {
+    return null;
+  }
+  const firstMessage = messages[0];
+  if (!firstMessage) {
+    return null;
+  }
+  const content = extractTextContent(firstMessage.content);
+  if (!content) {
+    return null;
+  }
+  const cleanedContent = content.trim().replace(/\n/g, " ");
+  if (!cleanedContent) {
+    return null;
+  }
+  const maxLength = 15;
+  const title = cleanedContent.slice(0, maxLength);
+  return title.length < cleanedContent.length ? `${title}...` : title;
+};
+
 class LangGraphGlobal {
   static globalMessageQueue = null;
   static globalCheckPointer = null;
   static globalThreadsManager = null;
   static isInitialized = null;
+  /**
+   * 全局标题生成器
+   * 可通过 setTitleGenerator 替换
+   */
+  static _titleGenerator = defaultTitleGenerator;
+  /**
+   * 设置自定义标题生成器
+   * @param generator 标题生成函数，传入 null 可禁用标题生成
+   */
+  static setTitleGenerator(generator) {
+    LangGraphGlobal._titleGenerator = generator;
+  }
+  /**
+   * 获取当前标题生成器
+   */
+  static getTitleGenerator() {
+    return LangGraphGlobal._titleGenerator;
+  }
   static async initGlobal() {
     if (LangGraphGlobal.isInitialized) {
       return LangGraphGlobal.isInitialized;
@@ -2018,5 +2108,5 @@ async function* streamState(threads, run, payload, options) {
   }
 }
 
-export { BaseStreamQueue as B, CancelEventMessage as C, GRAPHS as G, KyselyThreadsManager as K, LangGraphGlobal as L, streamState as a, getGraph as g, registerGraph as r, serialiseAsDict as s };
-//# sourceMappingURL=stream-8w3eP9hU.js.map
+export { BaseStreamQueue as B, CancelEventMessage as C, GRAPHS as G, KyselyThreadsManager as K, LangGraphGlobal as L, streamState as a, defaultTitleGenerator as d, getGraph as g, registerGraph as r, serialiseAsDict as s };
+//# sourceMappingURL=stream-Bleed1lp.js.map