luckytri 1.0.0 → 1.0.2

package/server/knowledge/schema.js

@@ -57,6 +57,16 @@ function mergeLegacyMemories(db) {
   }
 }
 
+export const KNOWLEDGE_TRIGGERS = [
+  `CREATE TRIGGER IF NOT EXISTS core_vector_changed AFTER UPDATE OF embedding ON core_chunks BEGIN
+      DELETE FROM core_vector_buckets WHERE chunk_id=NEW.id;
+      UPDATE core_chunks SET vector_indexed=0 WHERE id=NEW.id;
+    END`,
+  `CREATE TRIGGER IF NOT EXISTS core_vector_deleted AFTER DELETE ON core_chunks BEGIN
+      DELETE FROM core_vector_buckets WHERE chunk_id=OLD.id;
+    END`,
+];
+
 export function migrateKnowledge(db) {
   db.exec(`
     CREATE TABLE IF NOT EXISTS core_collections (
@@ -124,14 +134,8 @@ export function migrateKnowledge(db) {
   db.exec(`CREATE INDEX IF NOT EXISTS core_chunks_collection ON core_chunks(collection_id, document_id);
     CREATE INDEX IF NOT EXISTS core_chunks_unindexed ON core_chunks(vector_indexed) WHERE embedding IS NOT NULL;
     CREATE TABLE IF NOT EXISTS core_vector_buckets(dimension INTEGER NOT NULL,band INTEGER NOT NULL,bucket INTEGER NOT NULL,chunk_id TEXT NOT NULL,PRIMARY KEY(dimension,band,bucket,chunk_id));
-    CREATE INDEX IF NOT EXISTS core_vector_buckets_chunk ON core_vector_buckets(chunk_id);
-    CREATE TRIGGER IF NOT EXISTS core_vector_changed AFTER UPDATE OF embedding ON core_chunks BEGIN
-      DELETE FROM core_vector_buckets WHERE chunk_id=NEW.id;
-      UPDATE core_chunks SET vector_indexed=0 WHERE id=NEW.id;
-    END;
-    CREATE TRIGGER IF NOT EXISTS core_vector_deleted AFTER DELETE ON core_chunks BEGIN
-      DELETE FROM core_vector_buckets WHERE chunk_id=OLD.id;
-    END;`);
+    CREATE INDEX IF NOT EXISTS core_vector_buckets_chunk ON core_vector_buckets(chunk_id);`);
+  db.exec(KNOWLEDGE_TRIGGERS.join(";") + ";");
   if (
     !db
       .prepare("SELECT id FROM core_collections WHERE id='shared-default'")

package/server/local-access.js

@@ -0,0 +1,32 @@
+import { randomBytes } from "node:crypto";
+import { lstatSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+import { runtimePaths } from "./paths.js";
+
+export function isLoopback(address = "") {
+  return (
+    address === "::1" ||
+    address === "127.0.0.1" ||
+    address === "::ffff:127.0.0.1"
+  );
+}
+
+// A machine-local capability for CLI status/stop. It is never a user setting,
+// never returned to the browser, and never grants access to the studio APIs.
+export function localServiceKey() {
+  const directory = runtimePaths().data;
+  const file = join(directory, "local-api.key");
+  mkdirSync(directory, { recursive: true, mode: 0o700 });
+  try {
+    writeFileSync(file, randomBytes(32).toString("base64url"), {
+      flag: "wx",
+      mode: 0o600,
+    });
+  } catch (error) {
+    if (error.code !== "EEXIST") throw error;
+  }
+  if (!lstatSync(file).isFile()) throw Error("本机服务凭据必须是普通文件");
+  const key = readFileSync(file, "utf8");
+  if (!/^[\w-]{43}$/.test(key)) throw Error("本机服务凭据无效，请检查实例目录");
+  return key;
+}