@clawchatsai/connector 0.0.36 → 0.0.37

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@clawchatsai/connector",
-  "version": "0.0.36",
+  "version": "0.0.37",
   "type": "module",
   "description": "ClawChats OpenClaw plugin — P2P tunnel + local API bridge",
   "main": "dist/index.js",

package/server.js

@@ -1064,6 +1064,26 @@ async function handleCreateMessage(req, res, params) {
 
     // Bump thread updated_at
     db.prepare('UPDATE threads SET updated_at = ? WHERE id = ?').run(Date.now(), params.id);
+
+    // Heuristic title on first user message (mimics Claude/ChatGPT — title appears immediately on send)
+    if (body.role === 'user' && body.content) {
+      const threadInfo = db.prepare('SELECT title FROM threads WHERE id = ?').get(params.id);
+      if (threadInfo?.title === 'New chat') {
+        const heuristic = body.content.replace(/\n.*/s, '').slice(0, 40).trim()
+          + (body.content.length > 40 ? '...' : '');
+        if (heuristic) {
+          db.prepare('UPDATE threads SET title = ? WHERE id = ?').run(heuristic, params.id);
+          const activeWs = getWorkspaces().active;
+          gatewayClient.broadcastToBrowsers(JSON.stringify({
+            type: 'clawchats',
+            event: 'thread-title-updated',
+            threadId: params.id,
+            workspace: activeWs,
+            title: heuristic
+          }));
+        }
+      }
+    }
   }
 
   const message = db.prepare('SELECT * FROM messages WHERE id = ?').get(body.id);
@@ -2297,6 +2317,15 @@ async function handleRequest(req, res) {
     if ((p = matchRoute(method, urlPath, 'POST /api/threads/:id/context-fill'))) {
       return handleContextFill(req, res, p);
     }
+    if ((p = matchRoute(method, urlPath, 'POST /api/threads/:id/generate-title'))) {
+      const db = getActiveDb();
+      const thread = db.prepare('SELECT * FROM threads WHERE id = ?').get(p.id);
+      if (!thread) return sendError(res, 404, 'Thread not found');
+      // Regenerate: sets heuristic immediately (safe fallback), then fires AI upgrade
+      const activeWs = getWorkspaces().active;
+      gatewayClient.generateThreadTitle(db, p.id, activeWs);
+      return send(res, 200, { ok: true });
+    }
     if ((p = matchRoute(method, urlPath, 'POST /api/threads/:id/upload'))) {
       return await handleUpload(req, res, p);
     }
@@ -2487,6 +2516,22 @@ class GatewayClient {
       this.streamState.delete(sessionKey);
     }
 
+    // Intercept title generation responses (final, error, or aborted)
+    if (sessionKey && sessionKey.includes('__clawchats_title_')) {
+      if (state === 'final') {
+        const content = extractContent(message);
+        if (content && this.handleTitleResponse(sessionKey, content)) return;
+      } else if (state === 'error' || state === 'aborted') {
+        // Clean up pending entry by substring match — heuristic title stays
+        if (this._pendingTitleGens) {
+          for (const key of this._pendingTitleGens.keys()) {
+            if (sessionKey === key || sessionKey.includes(key)) { this._pendingTitleGens.delete(key); break; }
+          }
+        }
+        return;
+      }
+    }
+
     // Save assistant messages on final
     if (state === 'final') {
       this.saveAssistantMessage(sessionKey, message, seq);
@@ -2598,6 +2643,16 @@ class GatewayClient {
       }));
 
       console.log(`Saved assistant message to ${parsed.workspace}/${parsed.threadId} (${pendingMsg ? 'merged into pending' : 'seq: ' + seq})`);
+
+      // Auto-generate AI title upgrade after first assistant response
+      // Heuristic was already set on user message save; this fires the AI upgrade
+      const currentTitle = db.prepare('SELECT title FROM threads WHERE id = ?').get(parsed.threadId)?.title;
+      if (currentTitle && currentTitle !== 'New chat') {
+        const msgCount = db.prepare('SELECT COUNT(*) as c FROM messages WHERE thread_id = ?').get(parsed.threadId).c;
+        if (msgCount >= 2) {
+          this.generateThreadTitle(db, parsed.threadId, parsed.workspace, true);
+        }
+      }
     } catch (e) {
       console.error(`Failed to save assistant message:`, e.message);
     }
@@ -2630,6 +2685,125 @@ class GatewayClient {
     }
   }
 
+  /**
+   * Generate a title for a thread. Optionally sets a heuristic title immediately,
+   * then fires an async AI title upgrade via the gateway.
+   * @param {boolean} skipHeuristic - If true, skip heuristic step (already set on user message save)
+   */
+  generateThreadTitle(db, threadId, workspace, skipHeuristic = false) {
+    const thread = db.prepare('SELECT title FROM threads WHERE id = ?').get(threadId);
+    if (!thread) return;
+
+    // Skip if AI title gen already in progress for this thread
+    const titleKey = `__clawchats_title_${threadId}`;
+    if (this._pendingTitleGens?.has(titleKey)) return;
+
+    // Get first user message for heuristic title
+    const firstUserMsg = db.prepare(
+      "SELECT content FROM messages WHERE thread_id = ? AND role = 'user' ORDER BY created_at ASC LIMIT 1"
+    ).get(threadId);
+    if (!firstUserMsg?.content) return;
+
+    // Step 1: Heuristic title (immediate) — skip if already set on user message save
+    if (!skipHeuristic) {
+      const heuristic = firstUserMsg.content.replace(/\n.*/s, '').slice(0, 40).trim()
+        + (firstUserMsg.content.length > 40 ? '...' : '');
+      db.prepare('UPDATE threads SET title = ? WHERE id = ?').run(heuristic, threadId);
+
+      this.broadcastToBrowsers(JSON.stringify({
+        type: 'clawchats',
+        event: 'thread-title-updated',
+        threadId,
+        workspace,
+        title: heuristic
+      }));
+    }
+
+    // Step 2: AI title upgrade (async, best-effort)
+    const messages = db.prepare(
+      'SELECT role, content FROM messages WHERE thread_id = ? ORDER BY created_at ASC LIMIT 6'
+    ).all(threadId);
+
+    // Need at least 2 messages (user + assistant) for meaningful AI title
+    if (messages.length < 2) return;
+
+    const conversation = messages.map(m => {
+      const role = m.role === 'user' ? 'User' : 'Assistant';
+      const content = m.content.length > 300 ? m.content.slice(0, 300) + '...' : m.content;
+      return `${role}: ${content}`;
+    }).join('\n\n');
+
+    const prompt = `Based on this conversation, generate a concise 3-5 word title. Return ONLY the title text, no quotes, no explanation:\n\n${conversation}\n\nTitle:`;
+
+    const reqId = `title-${threadId}-${Date.now()}`;
+    if (!this._pendingTitleGens) this._pendingTitleGens = new Map();
+    this._pendingTitleGens.set(titleKey, { threadId, workspace, reqId });
+
+    // Timeout cleanup — prevent unbounded map growth if gateway never responds
+    setTimeout(() => {
+      if (this._pendingTitleGens?.has(titleKey)) {
+        this._pendingTitleGens.delete(titleKey);
+        console.log(`Title gen timeout for ${threadId} — keeping heuristic title`);
+      }
+    }, 30000);
+
+    this.sendToGateway(JSON.stringify({
+      type: 'req',
+      id: reqId,
+      method: 'chat.send',
+      params: {
+        sessionKey: titleKey,
+        message: prompt,
+        deliver: false,
+        idempotencyKey: reqId
+      }
+    }));
+  }
+
+  /**
+   * Handle AI title response from gateway.
+   * Returns true if the event was consumed (was a title gen response).
+   */
+  handleTitleResponse(sessionKey, content) {
+    if (!this._pendingTitleGens) return false;
+    // Gateway may prefix sessionKey (e.g. agent:main:__clawchats_title_xxx)
+    // Find the matching pending entry by substring
+    let matchKey = null, pending = null;
+    for (const [key, val] of this._pendingTitleGens) {
+      if (sessionKey === key || sessionKey.includes(key)) {
+        matchKey = key;
+        pending = val;
+        break;
+      }
+    }
+    if (!pending) return false;
+
+    this._pendingTitleGens.delete(matchKey);
+
+    let title = content.trim()
+      .replace(/^["']|["']$/g, '')
+      .replace(/^Title:\s*/i, '')
+      .replace(/\n.*/s, '')
+      .trim();
+
+    if (title.length > 50) title = title.substring(0, 47) + '...';
+    if (title.length === 0 || title.length >= 100) return true; // bad response, keep heuristic
+
+    const db = getDb(pending.workspace);
+    db.prepare('UPDATE threads SET title = ? WHERE id = ?').run(title, pending.threadId);
+
+    this.broadcastToBrowsers(JSON.stringify({
+      type: 'clawchats',
+      event: 'thread-title-updated',
+      threadId: pending.threadId,
+      workspace: pending.workspace,
+      title
+    }));
+
+    console.log(`AI title generated for ${pending.threadId}: "${title}"`);
+    return true;
+  }
+
   handleAgentEvent(payload) {
     const { runId, stream, data, sessionKey } = payload;
     if (!runId) return;
@@ -3390,6 +3564,26 @@ export function createApp(config = {}) {
     } else {
       db.prepare('INSERT INTO messages (id, thread_id, role, content, status, metadata, seq, timestamp, created_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)').run(body.id, params.id, body.role, body.content, body.status || 'sent', metadata, body.seq || null, body.timestamp, Date.now());
       db.prepare('UPDATE threads SET updated_at = ? WHERE id = ?').run(Date.now(), params.id);
+
+      // Heuristic title on first user message (mimics Claude/ChatGPT — title appears immediately on send)
+      if (body.role === 'user' && body.content) {
+        const threadInfo = db.prepare('SELECT title FROM threads WHERE id = ?').get(params.id);
+        if (threadInfo?.title === 'New chat') {
+          const heuristic = body.content.replace(/\n.*/s, '').slice(0, 40).trim()
+            + (body.content.length > 40 ? '...' : '');
+          if (heuristic) {
+            db.prepare('UPDATE threads SET title = ? WHERE id = ?').run(heuristic, params.id);
+            const activeWs = _getWorkspaces().active;
+            _gatewayClient.broadcastToBrowsers(JSON.stringify({
+              type: 'clawchats',
+              event: 'thread-title-updated',
+              threadId: params.id,
+              workspace: activeWs,
+              title: heuristic
+            }));
+          }
+        }
+      }
     }
     const message = db.prepare('SELECT * FROM messages WHERE id = ?').get(body.id);
     if (message && message.metadata) { try { message.metadata = JSON.parse(message.metadata); } catch { /* ok */ } }
@@ -3703,6 +3897,22 @@ export function createApp(config = {}) {
         return;
       }
       if (state === 'final' || state === 'aborted' || state === 'error') this.streamState.delete(sessionKey);
+
+      // Intercept title generation responses (final, error, or aborted)
+      if (sessionKey && sessionKey.includes('__clawchats_title_')) {
+        if (state === 'final') {
+          const content = extractContent(message);
+          if (content && this.handleTitleResponse(sessionKey, content)) return;
+        } else if (state === 'error' || state === 'aborted') {
+          if (this._pendingTitleGens) {
+            for (const key of this._pendingTitleGens.keys()) {
+              if (sessionKey === key || sessionKey.includes(key)) { this._pendingTitleGens.delete(key); break; }
+            }
+          }
+          return;
+        }
+      }
+
       if (state === 'final') this.saveAssistantMessage(sessionKey, message, seq);
       if (state === 'error') this.saveErrorMarker(sessionKey, message);
     }
@@ -3758,6 +3968,15 @@ export function createApp(config = {}) {
         const workspaceUnreadTotal = db.prepare('SELECT COALESCE(SUM(unread_count), 0) as total FROM threads').get().total;
         this.broadcastToBrowsers(JSON.stringify({ type: 'clawchats', event: 'unread-update', workspace: parsed.workspace, threadId: parsed.threadId, messageId, action: 'new', unreadCount, workspaceUnreadTotal, title: threadInfo?.title || 'Chat', preview, timestamp: now }));
         console.log(`Saved assistant message to ${parsed.workspace}/${parsed.threadId} (${pendingMsg ? 'merged into pending' : 'seq: ' + seq})`);
+
+        // Auto-generate AI title upgrade after first assistant response
+        const currentTitle = db.prepare('SELECT title FROM threads WHERE id = ?').get(parsed.threadId)?.title;
+        if (currentTitle && currentTitle !== 'New chat') {
+          const msgCount = db.prepare('SELECT COUNT(*) as c FROM messages WHERE thread_id = ?').get(parsed.threadId).c;
+          if (msgCount >= 2) {
+            this.generateThreadTitle(db, parsed.threadId, parsed.workspace, true);
+          }
+        }
       } catch (e) { console.error(`Failed to save assistant message:`, e.message); }
     }
 
@@ -3779,6 +3998,50 @@ export function createApp(config = {}) {
       } catch (e) { console.error(`Failed to save error marker:`, e.message); }
     }
 
+    generateThreadTitle(db, threadId, workspace, skipHeuristic = false) {
+      const thread = db.prepare('SELECT title FROM threads WHERE id = ?').get(threadId);
+      if (!thread) return;
+      const titleKey = `__clawchats_title_${threadId}`;
+      if (this._pendingTitleGens?.has(titleKey)) return;
+      const firstUserMsg = db.prepare("SELECT content FROM messages WHERE thread_id = ? AND role = 'user' ORDER BY created_at ASC LIMIT 1").get(threadId);
+      if (!firstUserMsg?.content) return;
+
+      if (!skipHeuristic) {
+        const heuristic = firstUserMsg.content.replace(/\n.*/s, '').slice(0, 40).trim() + (firstUserMsg.content.length > 40 ? '...' : '');
+        db.prepare('UPDATE threads SET title = ? WHERE id = ?').run(heuristic, threadId);
+        this.broadcastToBrowsers(JSON.stringify({ type: 'clawchats', event: 'thread-title-updated', threadId, workspace, title: heuristic }));
+      }
+
+      const messages = db.prepare('SELECT role, content FROM messages WHERE thread_id = ? ORDER BY created_at ASC LIMIT 6').all(threadId);
+      if (messages.length < 2) return;
+      const conversation = messages.map(m => `${m.role === 'user' ? 'User' : 'Assistant'}: ${m.content.length > 300 ? m.content.slice(0, 300) + '...' : m.content}`).join('\n\n');
+      const prompt = `Based on this conversation, generate a concise 3-5 word title. Return ONLY the title text, no quotes, no explanation:\n\n${conversation}\n\nTitle:`;
+      const reqId = `title-${threadId}-${Date.now()}`;
+      if (!this._pendingTitleGens) this._pendingTitleGens = new Map();
+      this._pendingTitleGens.set(titleKey, { threadId, workspace, reqId });
+      setTimeout(() => { if (this._pendingTitleGens?.has(titleKey)) { this._pendingTitleGens.delete(titleKey); console.log(`Title gen timeout for ${threadId} — keeping heuristic title`); } }, 30000);
+      this.sendToGateway(JSON.stringify({ type: 'req', id: reqId, method: 'chat.send', params: { sessionKey: titleKey, message: prompt, deliver: false, idempotencyKey: reqId } }));
+    }
+
+    handleTitleResponse(sessionKey, content) {
+      if (!this._pendingTitleGens) return false;
+      // Gateway may prefix sessionKey (e.g. agent:main:__clawchats_title_xxx)
+      let matchKey = null, pending = null;
+      for (const [key, val] of this._pendingTitleGens) {
+        if (sessionKey === key || sessionKey.includes(key)) { matchKey = key; pending = val; break; }
+      }
+      if (!pending) return false;
+      this._pendingTitleGens.delete(matchKey);
+      let title = content.trim().replace(/^["']|["']$/g, '').replace(/^Title:\s*/i, '').replace(/\n.*/s, '').trim();
+      if (title.length > 50) title = title.substring(0, 47) + '...';
+      if (title.length === 0 || title.length >= 100) return true;
+      const db = _getDb(pending.workspace);
+      db.prepare('UPDATE threads SET title = ? WHERE id = ?').run(title, pending.threadId);
+      this.broadcastToBrowsers(JSON.stringify({ type: 'clawchats', event: 'thread-title-updated', threadId: pending.threadId, workspace: pending.workspace, title }));
+      console.log(`AI title generated for ${pending.threadId}: "${title}"`);
+      return true;
+    }
+
     handleAgentEvent(payload) {
       const { runId, stream, data, sessionKey } = payload;
       if (!runId) return;
@@ -4122,6 +4385,15 @@ export function createApp(config = {}) {
       if ((p = matchRoute(method, urlPath, 'POST /api/threads/:id/messages'))) return await _handleCreateMessage(req, res, p);
       if ((p = matchRoute(method, urlPath, 'DELETE /api/threads/:id/messages/:messageId'))) return _handleDeleteMessage(req, res, p);
       if ((p = matchRoute(method, urlPath, 'POST /api/threads/:id/context-fill'))) return _handleContextFill(req, res, p);
+      if ((p = matchRoute(method, urlPath, 'POST /api/threads/:id/generate-title'))) {
+        const db = _getActiveDb();
+        const thread = db.prepare('SELECT * FROM threads WHERE id = ?').get(p.id);
+        if (!thread) return sendError(res, 404, 'Thread not found');
+        // Regenerate: sets heuristic immediately (safe fallback), then fires AI upgrade
+        const activeWs = _getWorkspaces().active;
+        _gatewayClient.generateThreadTitle(db, p.id, activeWs);
+        return send(res, 200, { ok: true });
+      }
       if ((p = matchRoute(method, urlPath, 'POST /api/threads/:id/upload'))) return await _handleUpload(req, res, p);
       if ((p = matchRoute(method, urlPath, 'GET /api/threads/:id/intelligence'))) return _handleGetIntelligence(req, res, p);
       if ((p = matchRoute(method, urlPath, 'POST /api/threads/:id/intelligence'))) return await _handleSaveIntelligence(req, res, p);