improve reasoning toggle tests and refactor model capability checks

src/db/index.js

@@ -1,6 +1,7 @@
 import fs from "fs";
 import Database from "better-sqlite3";
 import {initializeSchema, runMigrations} from "./schema.js";
+import {initializeAttachments} from "./attachments.js";
 
 export function createDatabase(config) {
     if (config.dbPath !== ":memory:") {
@@ -10,5 +11,6 @@ export function createDatabase(config) {
     const db = new Database(config.dbPath);
     initializeSchema(db);
     runMigrations(db, config);
+    initializeAttachments(db);
     return db;
 }

src/db/repositories.js

@@ -1,3 +1,5 @@
+import {attachmentMetadata} from "./attachments.js";
+
 const parseJsonArray = (value) => {
     if (!value) return [];
     try {
@@ -869,6 +871,11 @@ export function createRepositories(db, config) {
                 if (message.sourceId != null) importedMessageIds.set(message.sourceId, Number(result.lastInsertRowid));
                 if (message.sources) statements.updateChatMessageSources.run(JSON.stringify(message.sources), result.lastInsertRowid, chatId);
                 if (message.thinking) statements.updateChatMessageThinking.run(message.thinking, result.lastInsertRowid, chatId);
+                for (const [position, attachment] of (message.attachments || []).slice(0, config.uploadMaxFiles || 5).entries()) {
+                    // JSON backups contain references, not files. Only restore existing, owned images.
+                    db.prepare("INSERT OR IGNORE INTO message_attachments(message_id,attachment_id,position) SELECT ?,id,? FROM attachments WHERE id=? AND username=? AND status='attached'")
+                        .run(result.lastInsertRowid, position, attachment.id, username);
+                }
             }
             statements.updateChatGenerationSettings.run(
                 chat.preferred_model ?? null,
@@ -1271,6 +1278,7 @@ export function createRepositories(db, config) {
                     persona_state: this.getPersonaChatState(chat.id, username),
                     messages: this.listChatMessages(chat.id).map((message) => ({
                         ...message,
+                        attachments: db.prepare("SELECT a.* FROM attachments a JOIN message_attachments m ON a.id=m.attachment_id WHERE m.message_id=? ORDER BY m.position").all(message.id).map(attachmentMetadata),
                         retryVariants: parseJsonArray(message.retry_variants)
                     }))
                 })),