@memtensor/memos-local-openclaw-plugin 1.0.0 → 1.0.2-beta.1

package/dist/viewer/server.js

@@ -210,6 +210,12 @@ class ViewerServer {
                 this.serveConfig(res);
             else if (p === "/api/config" && req.method === "PUT")
                 this.handleSaveConfig(req, res);
+            else if (p === "/api/test-model" && req.method === "POST")
+                this.handleTestModel(req, res);
+            else if (p === "/api/fallback-model" && req.method === "GET")
+                this.serveFallbackModel(res);
+            else if (p === "/api/update-check" && req.method === "GET")
+                this.handleUpdateCheck(res);
             else if (p === "/api/auth/logout" && req.method === "POST")
                 this.handleLogout(req, res);
             else if (p === "/api/migrate/scan" && req.method === "GET")
@@ -553,7 +559,8 @@ class ViewerServer {
         try {
             ftsResults = db.prepare("SELECT c.* FROM chunks_fts f JOIN chunks c ON f.rowid = c.rowid WHERE chunks_fts MATCH ? ORDER BY rank LIMIT 100").all(q).filter(passesFilter);
         }
-        catch {
+        catch { /* FTS syntax error, fall through */ }
+        if (ftsResults.length === 0) {
             ftsResults = db.prepare("SELECT * FROM chunks WHERE content LIKE ? OR summary LIKE ? ORDER BY created_at DESC LIMIT 100").all(`%${q}%`, `%${q}%`).filter(passesFilter);
         }
         const SEMANTIC_THRESHOLD = 0.64;
@@ -584,16 +591,12 @@ class ViewerServer {
             }
         }
         for (const r of ftsResults) {
-            if (seenIds.has(r.id))
-                continue;
-            const vscore = scoreMap.get(r.id);
-            if (vscore !== undefined && vscore < SEMANTIC_THRESHOLD)
-                continue;
-            seenIds.add(r.id);
-            merged.push(r);
-        }
-        const fallback = merged.length === 0 && ftsResults.length > 0;
-        const results = fallback ? ftsResults.slice(0, 20) : merged;
+            if (!seenIds.has(r.id)) {
+                seenIds.add(r.id);
+                merged.push(r);
+            }
+        }
+        const results = merged.length > 0 ? merged : ftsResults.slice(0, 20);
         this.store.recordViewerEvent("search");
         this.jsonResponse(res, {
             results,
@@ -601,7 +604,6 @@ class ViewerServer {
             vectorCount: vectorResults.length,
             ftsCount: ftsResults.length,
             total: results.length,
-            fallbackFts: fallback,
         });
     }
     // ─── Skills API ───
@@ -754,9 +756,10 @@ class ViewerServer {
                 this.jsonResponse(res, { ok: true, skillId, visibility });
             }
             catch (err) {
-                this.log.error(`handleSkillVisibility error: skillId=${skillId}, body=${body}, err=${err}`);
-                res.writeHead(400, { "Content-Type": "application/json" });
-                res.end(JSON.stringify({ error: String(err) }));
+                const errMsg = err instanceof Error ? `${err.name}: ${err.message}` : String(err);
+                this.log.error(`handleSkillVisibility error: skillId=${skillId}, body=${body}, err=${errMsg}`);
+                res.writeHead(500, { "Content-Type": "application/json" });
+                res.end(JSON.stringify({ error: errMsg }));
             }
         });
     }
@@ -962,20 +965,27 @@ class ViewerServer {
         this.jsonResponse(res, { ok: true, deleted: count });
     }
     handleDeleteAll(res) {
-        const result = this.store.deleteAll();
-        // Clean up skills-store directory
-        const skillsStoreDir = node_path_1.default.join(this.dataDir, "skills-store");
         try {
-            if (node_fs_1.default.existsSync(skillsStoreDir)) {
-                node_fs_1.default.rmSync(skillsStoreDir, { recursive: true });
-                node_fs_1.default.mkdirSync(skillsStoreDir, { recursive: true });
-                this.log.info("Cleared skills-store directory");
+            const result = this.store.deleteAll();
+            const skillsStoreDir = node_path_1.default.join(this.dataDir, "skills-store");
+            try {
+                if (node_fs_1.default.existsSync(skillsStoreDir)) {
+                    node_fs_1.default.rmSync(skillsStoreDir, { recursive: true });
+                    node_fs_1.default.mkdirSync(skillsStoreDir, { recursive: true });
+                    this.log.info("Cleared skills-store directory");
+                }
             }
+            catch (err) {
+                this.log.warn(`Failed to clear skills-store: ${err}`);
+            }
+            this.jsonResponse(res, { ok: true, deleted: result });
         }
         catch (err) {
-            this.log.warn(`Failed to clear skills-store: ${err}`);
+            const msg = err instanceof Error ? err.message : String(err);
+            this.log.error(`handleDeleteAll error: ${msg}`);
+            res.writeHead(500, { "Content-Type": "application/json" });
+            res.end(JSON.stringify({ ok: false, error: msg }));
         }
-        this.jsonResponse(res, { ok: true, deleted: result });
     }
     // ─── Helpers ───
     // ─── Config API ───
@@ -1058,6 +1068,208 @@ class ViewerServer {
             }
         });
     }
+    handleTestModel(req, res) {
+        this.readBody(req, async (body) => {
+            try {
+                const { type, provider, model, endpoint, apiKey } = JSON.parse(body);
+                if (!provider) {
+                    this.jsonResponse(res, { ok: false, error: "provider is required" });
+                    return;
+                }
+                if (type === "embedding") {
+                    await this.testEmbeddingModel(provider, model, endpoint, apiKey);
+                    this.jsonResponse(res, { ok: true, detail: `${provider}/${model}` });
+                }
+                else {
+                    await this.testChatModel(provider, model, endpoint, apiKey);
+                    this.jsonResponse(res, { ok: true, detail: `${provider}/${model}` });
+                }
+            }
+            catch (e) {
+                const msg = e instanceof Error ? e.message : String(e);
+                this.log.warn(`test-model failed: ${msg}`);
+                this.jsonResponse(res, { ok: false, error: msg });
+            }
+        });
+    }
+    serveFallbackModel(res) {
+        try {
+            const cfgPath = this.getOpenClawConfigPath();
+            if (!node_fs_1.default.existsSync(cfgPath)) {
+                this.jsonResponse(res, { available: false });
+                return;
+            }
+            const raw = JSON.parse(node_fs_1.default.readFileSync(cfgPath, "utf-8"));
+            const agentModel = raw?.agents?.defaults?.model?.primary;
+            if (!agentModel) {
+                this.jsonResponse(res, { available: false });
+                return;
+            }
+            const [providerKey, modelId] = agentModel.includes("/")
+                ? agentModel.split("/", 2)
+                : [undefined, agentModel];
+            const providerCfg = providerKey
+                ? raw?.models?.providers?.[providerKey]
+                : Object.values(raw?.models?.providers ?? {})[0];
+            if (!providerCfg || !providerCfg.baseUrl || !providerCfg.apiKey) {
+                this.jsonResponse(res, { available: false });
+                return;
+            }
+            this.jsonResponse(res, { available: true, model: modelId || agentModel, baseUrl: providerCfg.baseUrl });
+        }
+        catch {
+            this.jsonResponse(res, { available: false });
+        }
+    }
+    findPluginPackageJson() {
+        let dir = __dirname;
+        for (let i = 0; i < 6; i++) {
+            const candidate = node_path_1.default.join(dir, "package.json");
+            if (node_fs_1.default.existsSync(candidate)) {
+                try {
+                    const pkg = JSON.parse(node_fs_1.default.readFileSync(candidate, "utf-8"));
+                    if (pkg.name && pkg.name.includes("memos-local"))
+                        return candidate;
+                }
+                catch { /* skip */ }
+            }
+            dir = node_path_1.default.dirname(dir);
+        }
+        return null;
+    }
+    async handleUpdateCheck(res) {
+        try {
+            const pkgPath = this.findPluginPackageJson();
+            if (!pkgPath) {
+                this.jsonResponse(res, { updateAvailable: false, error: "package.json not found" });
+                return;
+            }
+            const pkg = JSON.parse(node_fs_1.default.readFileSync(pkgPath, "utf-8"));
+            const current = pkg.version;
+            const name = pkg.name;
+            if (!current || !name) {
+                this.jsonResponse(res, { updateAvailable: false, current });
+                return;
+            }
+            const npmResp = await fetch(`https://registry.npmjs.org/${name}/latest`, {
+                signal: AbortSignal.timeout(6_000),
+            });
+            if (!npmResp.ok) {
+                this.jsonResponse(res, { updateAvailable: false, current });
+                return;
+            }
+            const data = await npmResp.json();
+            const latest = data.version ?? current;
+            this.jsonResponse(res, {
+                updateAvailable: latest !== current,
+                current,
+                latest,
+                packageName: name,
+            });
+        }
+        catch (e) {
+            this.log.warn(`handleUpdateCheck error: ${e}`);
+            this.jsonResponse(res, { updateAvailable: false, error: String(e) });
+        }
+    }
+    async testEmbeddingModel(provider, model, endpoint, apiKey) {
+        if (provider === "local") {
+            return;
+        }
+        const baseUrl = (endpoint || "https://api.openai.com/v1").replace(/\/+$/, "");
+        const embUrl = baseUrl.endsWith("/embeddings") ? baseUrl : `${baseUrl}/embeddings`;
+        const headers = {
+            "Content-Type": "application/json",
+            "Authorization": `Bearer ${apiKey}`,
+        };
+        if (provider === "cohere") {
+            headers["Authorization"] = `Bearer ${apiKey}`;
+            const resp = await fetch(baseUrl.replace(/\/v\d+.*/, "/v2/embed"), {
+                method: "POST",
+                headers,
+                body: JSON.stringify({ texts: ["test"], model: model || "embed-english-v3.0", input_type: "search_query", embedding_types: ["float"] }),
+                signal: AbortSignal.timeout(15_000),
+            });
+            if (!resp.ok) {
+                const txt = await resp.text();
+                throw new Error(`Cohere embed ${resp.status}: ${txt}`);
+            }
+            return;
+        }
+        if (provider === "gemini") {
+            const url = `https://generativelanguage.googleapis.com/v1/models/${model || "text-embedding-004"}:embedContent?key=${apiKey}`;
+            const resp = await fetch(url, {
+                method: "POST",
+                headers: { "Content-Type": "application/json" },
+                body: JSON.stringify({ content: { parts: [{ text: "test" }] } }),
+                signal: AbortSignal.timeout(15_000),
+            });
+            if (!resp.ok) {
+                const txt = await resp.text();
+                throw new Error(`Gemini embed ${resp.status}: ${txt}`);
+            }
+            return;
+        }
+        const resp = await fetch(embUrl, {
+            method: "POST",
+            headers,
+            body: JSON.stringify({ input: ["test"], model: model || "text-embedding-3-small" }),
+            signal: AbortSignal.timeout(15_000),
+        });
+        if (!resp.ok) {
+            const txt = await resp.text();
+            throw new Error(`${resp.status}: ${txt}`);
+        }
+    }
+    async testChatModel(provider, model, endpoint, apiKey) {
+        const baseUrl = (endpoint || "https://api.openai.com/v1").replace(/\/+$/, "");
+        if (provider === "anthropic") {
+            const url = endpoint || "https://api.anthropic.com/v1/messages";
+            const resp = await fetch(url, {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                    "x-api-key": apiKey,
+                    "anthropic-version": "2023-06-01",
+                },
+                body: JSON.stringify({ model: model || "claude-3-haiku-20240307", max_tokens: 5, messages: [{ role: "user", content: "hi" }] }),
+                signal: AbortSignal.timeout(15_000),
+            });
+            if (!resp.ok) {
+                const txt = await resp.text();
+                throw new Error(`Anthropic ${resp.status}: ${txt}`);
+            }
+            return;
+        }
+        if (provider === "gemini") {
+            const url = `https://generativelanguage.googleapis.com/v1/models/${model || "gemini-1.5-flash"}:generateContent?key=${apiKey}`;
+            const resp = await fetch(url, {
+                method: "POST",
+                headers: { "Content-Type": "application/json" },
+                body: JSON.stringify({ contents: [{ parts: [{ text: "hi" }] }], generationConfig: { maxOutputTokens: 5 } }),
+                signal: AbortSignal.timeout(15_000),
+            });
+            if (!resp.ok) {
+                const txt = await resp.text();
+                throw new Error(`Gemini ${resp.status}: ${txt}`);
+            }
+            return;
+        }
+        const chatUrl = baseUrl.endsWith("/chat/completions") ? baseUrl : `${baseUrl}/chat/completions`;
+        const resp = await fetch(chatUrl, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                "Authorization": `Bearer ${apiKey}`,
+            },
+            body: JSON.stringify({ model: model || "gpt-4o-mini", max_tokens: 5, messages: [{ role: "user", content: "hi" }] }),
+            signal: AbortSignal.timeout(15_000),
+        });
+        if (!resp.ok) {
+            const txt = await resp.text();
+            throw new Error(`${resp.status}: ${txt}`);
+        }
+    }
     serveLogs(res, url) {
         const limit = Math.min(Number(url.searchParams.get("limit") ?? 20), 200);
         const offset = Math.max(0, Number(url.searchParams.get("offset") ?? 0));