tinywebui 0.1.4 → 0.1.5

package/src/store/scope.js

@@ -1,28 +1,28 @@
 /* Helpers every area of the store shares: whose rows a query sees, and safe FTS5 queries. */
-
 /**
  * Every read of user-owned rows names whose rows it wants. There is no
  * default: `undefined` throws, so a caller that forgot to pass the
  * authenticated user fails loudly instead of silently seeing everyone's data.
  *   string     that user's rows
  *   null       rows with no owner (legacy / pre-auth rows)
- *   ALL_USERS  no filter -- authMode 'none', admin views, internal plumbing
+ *   ALL_USERS  no filter -- authMode 'local', admin views, internal plumbing
  */
 export const ALL_USERS = Symbol('all-users');
-
 export function scope(userId, col = 'user_id') {
-  if (userId === ALL_USERS) return { sql: '1=1', params: [] };
-  if (userId === null) return { sql: `${col} IS NULL`, params: [] };
-  if (typeof userId === 'string' && userId) return { sql: `${col} = ?`, params: [userId] };
-  throw new TypeError('store: a user scope is required (pass a user id, null, or ALL_USERS)');
+    if (userId === ALL_USERS)
+        return { sql: '1=1', params: [] };
+    if (userId === null)
+        return { sql: `${col} IS NULL`, params: [] };
+    if (typeof userId === 'string' && userId)
+        return { sql: `${col} = ?`, params: [userId] };
+    throw new TypeError('store: a user scope is required (pass a user id, null, or ALL_USERS)');
 }
-
 /** The owner to write onto a new row: ALL_USERS (no auth) owns as null. */
 export function ownerOf(userId) {
-  if (userId === ALL_USERS || userId === null || userId === undefined) return null;
-  return String(userId);
+    if (userId === ALL_USERS || userId === null || userId === undefined)
+        return null;
+    return String(userId);
 }
-
 /**
  * Turns free text from a search box into an FTS5 MATCH expression that cannot
  * fail to parse. FTS5's own query grammar has ANDs, ORs, dashes, colons and
@@ -34,10 +34,11 @@ export function ownerOf(userId) {
  * a whole word is finished.
  */
 export function ftsQuery(raw, { any = false } = {}) {
-  const tokens = String(raw ?? '').trim().split(/\s+/).filter(Boolean).slice(0, 12);
-  if (!tokens.length) return '';
-  const esc = (t) => t.replace(/"/g, '""');
-  return tokens
-    .map((t, i) => (i === tokens.length - 1 ? `"${esc(t)}"*` : `"${esc(t)}"`))
-    .join(any ? ' OR ' : ' ');
+    const tokens = String(raw ?? '').trim().split(/\s+/).filter(Boolean).slice(0, 12);
+    if (!tokens.length)
+        return '';
+    const esc = (t) => t.replace(/"/g, '""');
+    return tokens
+        .map((t, i) => (i === tokens.length - 1 ? `"${esc(t)}"*` : `"${esc(t)}"`))
+        .join(any ? ' OR ' : ' ');
 }

package/src/store/search.js

@@ -1,33 +1,31 @@
 import { scope, ftsQuery } from './scope.js';
-
 // Sidebar search: full-text over what was said, one row per chat.
 // One area of the Store; see index.js.
-
 export class SearchStore {
-  constructor(db, deps = {}) {
-    this.db = db;
-    Object.assign(this, deps);
-  }
-
-  /**
-   * Full-text search, one row per matching CHAT, not per matching message --
-   * a search box is for finding a conversation, and a chat where the same
-   * word landed in five messages should not crowd out four other chats that
-   * only said it once. Each chat's own best-ranked message stands for it,
-   * and chats are then ordered against each other by that same rank.
-   *
-   * `bm25()` and `snippet()` are FTS5 auxiliary functions: usable only in a
-   * query that itself matches the virtual table, not through an arbitrary
-   * subquery. Computing both once in `hits` and carrying the values through
-   * `ranked` (rather than recomputing `bm25(messages_fts)` inside the window
-   * function's own ORDER BY) is what keeps them legal here -- SQLite raises
-   * "unable to use function bm25 in the requested context" otherwise.
-   */
-  chats(query, limit, userId) {
-    const s = scope(userId, 'c.user_id');
-    const q = ftsQuery(query);
-    if (!q) return [];
-    return this.db.prepare(`
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    /**
+     * Full-text search, one row per matching CHAT, not per matching message --
+     * a search box is for finding a conversation, and a chat where the same
+     * word landed in five messages should not crowd out four other chats that
+     * only said it once. Each chat's own best-ranked message stands for it,
+     * and chats are then ordered against each other by that same rank.
+     *
+     * `bm25()` and `snippet()` are FTS5 auxiliary functions: usable only in a
+     * query that itself matches the virtual table, not through an arbitrary
+     * subquery. Computing both once in `hits` and carrying the values through
+     * `ranked` (rather than recomputing `bm25(messages_fts)` inside the window
+     * function's own ORDER BY) is what keeps them legal here -- SQLite raises
+     * "unable to use function bm25 in the requested context" otherwise.
+     */
+    chats(query, limit, userId) {
+        const s = scope(userId, 'c.user_id');
+        const q = ftsQuery(query);
+        if (!q)
+            return [];
+        return this.db.prepare(`
       WITH hits AS (
         SELECT
           m.chat_id  AS chatId,
@@ -52,5 +50,5 @@ export class SearchStore {
       ORDER BY rank
       LIMIT ?
     `).all(q, ...s.params, limit);
-  }
+    }
 }

package/src/store/turns.js

@@ -1,78 +1,78 @@
 import { scope, ftsQuery } from './scope.js';
 import { CORPUS } from './embeddings.js';
-
 // Chat turns -- a question and its final answer -- the 'chats' retrieval corpus.
 // One area of the Store; see index.js.
-
 export class TurnStore {
-  constructor(db, deps = {}) {
-    this.db = db;
-    Object.assign(this, deps);
-  }
-
-  /**
-   * A chat's turns: each question with the answer it finally got -- the last
-   * assistant message with text before the next question, the one the
-   * transcript shows as the answer. A turn's id is its user message's id.
-   */
-  forChat(chatId) {
-    const rows = this.db.prepare(`
+    constructor(db, deps) {
+        this.db = db;
+        Object.assign(this, deps);
+    }
+    /**
+     * A chat's turns: each question with the answer it finally got -- the last
+     * assistant message with text before the next question, the one the
+     * transcript shows as the answer. A turn's id is its user message's id.
+     */
+    forChat(chatId) {
+        const rows = this.db.prepare(`
       SELECT id, seq, role, content, created_at FROM messages
       WHERE chat_id = ? AND role IN ('user', 'assistant') ORDER BY seq
     `).all(chatId);
-    const turns = [];
-    for (const r of rows) {
-      if (r.role === 'user') turns.push({ id: r.id, chatId, seq: r.seq, question: r.content || '', answer: '', createdAt: r.created_at });
-      else if (turns.length && r.content?.trim()) turns.at(-1).answer = r.content;
+        const turns = [];
+        for (const r of rows) {
+            if (r.role === 'user')
+                turns.push({ id: r.id, chatId, seq: r.seq, question: r.content || '', answer: '', createdAt: r.created_at });
+            else if (turns.length && r.content?.trim())
+                turns.at(-1).answer = r.content;
+        }
+        return turns;
     }
-    return turns;
-  }
-
-  /** The turns with these ids, each with its chat's title: Map(id -> turn). */
-  byIds(ids) {
-    const chatIds = this.db.prepare(`
+    /** The turns with these ids, each with its chat's title: Map(id -> turn). */
+    byIds(ids) {
+        const chatIds = this.db.prepare(`
       SELECT DISTINCT chat_id FROM messages WHERE role = 'user' AND id IN (SELECT value FROM json_each(?))
     `).all(JSON.stringify(ids)).map((r) => r.chat_id);
-    const wanted = new Set(ids);
-    const out = new Map();
-    for (const chatId of chatIds) {
-      const title = this.chats.byId(chatId)?.title || 'Untitled chat';
-      for (const t of this.forChat(chatId)) if (wanted.has(t.id)) out.set(t.id, { ...t, chatTitle: title });
+        const wanted = new Set(ids);
+        const out = new Map();
+        for (const chatId of chatIds) {
+            const title = this.chats.byId(chatId)?.title || 'Untitled chat';
+            for (const t of this.forChat(chatId))
+                if (wanted.has(t.id))
+                    out.set(t.id, { ...t, chatTitle: title });
+        }
+        return out;
     }
-    return out;
-  }
-
-  /** Every turn id a user may search, leaving one chat out (the one asking). */
-  ids(userId, { excludeChatId = null } = {}) {
-    const s = scope(userId, 'c.user_id');
-    return this.db.prepare(`
+    /** Every turn id a user may search, leaving one chat out (the one asking). */
+    ids(userId, { excludeChatId = null } = {}) {
+        const s = scope(userId, 'c.user_id');
+        return this.db.prepare(`
       SELECT m.id FROM messages m JOIN chats c ON c.id = m.chat_id
       WHERE m.role = 'user' AND m.chat_id IS NOT ? AND ${s.sql}
     `).all(excludeChatId, ...s.params).map((r) => r.id);
-  }
-
-  /** BM25 per turn (its best message), higher is better. `any`: any query word may match. */
-  lexicalScores(userId, query, { excludeChatId = null, any = true } = {}) {
-    const q = ftsQuery(query, { any });
-    if (!q) return new Map();
-    const s = scope(userId, 'c.user_id');
-    const rows = this.db.prepare(`
+    }
+    /** BM25 per turn (its best message), higher is better. `any`: any query word may match. */
+    lexicalScores(userId, query, { excludeChatId = null, any = true } = {}) {
+        const q = ftsQuery(query, { any });
+        if (!q)
+            return new Map();
+        const s = scope(userId, 'c.user_id');
+        const rows = this.db.prepare(`
       SELECT bm25(messages_fts) AS rank,
         (SELECT u.id FROM messages u WHERE u.chat_id = m.chat_id AND u.role = 'user' AND u.seq <= m.seq ORDER BY u.seq DESC LIMIT 1) AS turn
       FROM messages_fts JOIN messages m ON m.id = messages_fts.rowid JOIN chats c ON c.id = m.chat_id
       WHERE messages_fts MATCH ? AND m.chat_id IS NOT ? AND ${s.sql}
     `).all(q, excludeChatId, ...s.params);
-    const out = new Map();
-    for (const r of rows) if (r.turn != null) out.set(r.turn, Math.max(out.get(r.turn) ?? -Infinity, -r.rank));
-    return out;
-  }
-
-  /** Chats with a question not yet embedded under this model (for backfill). */
-  chatsMissingVectors(modelKey) {
-    return this.db.prepare(`
+        const out = new Map();
+        for (const r of rows)
+            if (r.turn != null)
+                out.set(r.turn, Math.max(out.get(r.turn) ?? -Infinity, -r.rank));
+        return out;
+    }
+    /** Chats with a question not yet embedded under this model (for backfill). */
+    chatsMissingVectors(modelKey) {
+        return this.db.prepare(`
       SELECT DISTINCT m.chat_id AS id FROM messages m
       LEFT JOIN embeddings e ON e.corpus = ? AND e.unit_id = m.id AND e.model_key = ? AND e.chunk = 0
       WHERE m.role = 'user' AND e.unit_id IS NULL
     `).all(CORPUS.CHATS, modelKey).map((r) => r.id);
-  }
+    }
 }