scream-code 0.8.4 → 0.8.6

package/dist/main.mjs

@@ -6,7 +6,7 @@ const __dirname = __cjsShimDirname(__filename);
 import "./suppress-sqlite-warning-C2VB0doZ.mjs";
 //#region src/main.ts
 try {
-	(await import("./app-DYDC5U75.mjs")).main();
+	(await import("./app-a4CtjPlv.mjs")).main();
 } catch (error) {
 	process.stderr.write(`${error instanceof Error ? error.stack ?? error.message : String(error)}\n`);
 	process.exit(1);

package/dist/{src-CptHxtUY.mjs → src-BHYcdFfg.mjs}

@@ -1392,6 +1392,19 @@ Only include entities that explicitly appear in the query. If no clear entities,
 }
 //#endregion
 //#region ../../packages/knowledge/src/ingest.ts
+/**
+* Embed texts, returning null-array if embeddings are skipped or on failure.
+* Caller should set skipEmbedding=true when the model didn't load in time,
+* so the entire ingestion is consistent (all or nothing for embeddings).
+* Search falls back to FTS5 keyword match for chunks without embeddings.
+*/
+async function embedOrNulls(engine, texts, skipEmbedding) {
+	if (engine === void 0 || texts.length === 0 || skipEmbedding) return texts.map(() => null);
+	if (!engine.available) return texts.map(() => null);
+	const vectors = await engine.embedBatch(texts);
+	if (vectors === null || vectors.length !== texts.length) return texts.map(() => null);
+	return vectors;
+}
 /** LLM concurrency for extraction — kept low to avoid rate-limit bursts. */
 const LLM_CONCURRENCY = 3;
 const SUPPORTED_EXTENSIONS = new Set([
@@ -1454,12 +1467,25 @@ async function mapWithConcurrency(items, limit, fn, onProgress) {
 */
 async function ingestFile(store, llm, filePath, onProgress) {
 	const engine = store.getEmbeddingEngine();
-	if (engine === void 0 || !engine.available) {
+	let skipEmbed = engine === void 0;
+	if (engine !== void 0) {
 		onProgress?.({
-			stage: "error",
-			message: "向量模型未就绪"
+			stage: "embedding-check",
+			message: "检查向量模型..."
 		});
-		throw new Error("向量模型未就绪，请检查网络后重试");
+		try {
+			skipEmbed = !await Promise.race([engine.ensureReady(), new Promise((resolve) => setTimeout(() => resolve(false), 15e3))]);
+			if (skipEmbed) onProgress?.({
+				stage: "embedding-check",
+				message: "向量模型未就绪，本次摄入跳过向量嵌入"
+			});
+		} catch {
+			skipEmbed = true;
+			onProgress?.({
+				stage: "embedding-check",
+				message: "向量模型未就绪，本次摄入跳过向量嵌入"
+			});
+		}
 	}
 	const existing = await store.findSourceByFilePath(filePath);
 	if (existing !== void 0) {
@@ -1501,14 +1527,7 @@ async function ingestFile(store, llm, filePath, onProgress) {
 			totalChunks: sections.length,
 			message: `嵌入 chunks: 0/${sections.length}`
 		});
-		const chunkEmbeddings = await engine.embedBatch(sections.map((s) => s.heading !== null ? `${s.heading}\n${s.content}` : s.content));
-		if (chunkEmbeddings === null) {
-			onProgress?.({
-				stage: "error",
-				message: "chunk 向量嵌入失败"
-			});
-			throw new Error("chunk 向量嵌入失败");
-		}
+		const chunkEmbeddings = await embedOrNulls(engine, sections.map((s) => s.heading !== null ? `${s.heading}\n${s.content}` : s.content), skipEmbed);
 		const chunks = [];
 		for (let i = 0; i < sections.length; i++) {
 			const section = sections[i];
@@ -1552,14 +1571,7 @@ async function ingestFile(store, llm, filePath, onProgress) {
 		});
 		const titleTexts = extractedEvents.map((e) => e.title);
 		const contentTexts = extractedEvents.map((e) => `${e.title}\n\n${e.content}`);
-		const [titleEmbeddings, contentEmbeddings] = await Promise.all([engine.embedBatch(titleTexts), engine.embedBatch(contentTexts)]);
-		if (titleEmbeddings === null || contentEmbeddings === null) {
-			onProgress?.({
-				stage: "error",
-				message: "event 向量嵌入失败"
-			});
-			throw new Error("event 向量嵌入失败");
-		}
+		const [titleEmbeddings, contentEmbeddings] = await Promise.all([embedOrNulls(engine, titleTexts, skipEmbed), embedOrNulls(engine, contentTexts, skipEmbed)]);
 		const events = [];
 		for (let i = 0; i < extractedEvents.length; i++) {
 			const extracted = extractedEvents[i];
@@ -1595,18 +1607,11 @@ async function ingestFile(store, llm, filePath, onProgress) {
 			if (!entityMap.has(key)) entityMap.set(key, entity);
 		}
 		const uniqueEntities = Array.from(entityMap.values());
-		const entityEmbeddings = uniqueEntities.length > 0 ? await engine.embedBatch(uniqueEntities.map((e) => e.name)) : [];
-		if (uniqueEntities.length > 0 && entityEmbeddings === null) {
-			onProgress?.({
-				stage: "error",
-				message: "entity 向量嵌入失败"
-			});
-			throw new Error("entity 向量嵌入失败");
-		}
+		const entityEmbeddings = uniqueEntities.length > 0 ? await embedOrNulls(engine, uniqueEntities.map((e) => e.name), skipEmbed) : [];
 		const entityIdByEntityKey = /* @__PURE__ */ new Map();
 		for (let i = 0; i < uniqueEntities.length; i++) {
 			const entity = uniqueEntities[i];
-			const embedding = entityEmbeddings?.[i] ?? null;
+			const embedding = entityEmbeddings[i] ?? null;
 			const stored = await store.upsertEntity({
 				sourceId: source.id,
 				type: entity.type,
@@ -1632,14 +1637,7 @@ async function ingestFile(store, llm, filePath, onProgress) {
 			const event = events[eventIndex];
 			return entity.description.length > 0 ? entity.description : `${event.title} ${entity.name}`;
 		});
-		const relationEmbeddings = relationPairs.length > 0 ? await engine.embedBatch(relationTexts) : [];
-		if (relationPairs.length > 0 && relationEmbeddings === null) {
-			onProgress?.({
-				stage: "error",
-				message: "relation 向量嵌入失败"
-			});
-			throw new Error("relation 向量嵌入失败");
-		}
+		const relationEmbeddings = relationPairs.length > 0 ? await embedOrNulls(engine, relationTexts, skipEmbed) : [];
 		for (let i = 0; i < relationPairs.length; i++) {
 			const pair = relationPairs[i];
 			const event = events[pair.eventIndex];

package/dist/{src-DhT3eU2C.mjs → src-CC8SSs-H.mjs}

@@ -3,5 +3,5 @@ import { fileURLToPath as __cjsShimFileURLToPath } from 'node:url';
 import { dirname as __cjsShimDirname } from 'node:path';
 const __filename = __cjsShimFileURLToPath(import.meta.url);
 const __dirname = __cjsShimDirname(__filename);
-import { n as multiSearchWithTrace } from "./src-CptHxtUY.mjs";
+import { n as multiSearchWithTrace } from "./src-BHYcdFfg.mjs";
 export { multiSearchWithTrace };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "scream-code",
-  "version": "0.8.4",
+  "version": "0.8.6",
   "description": "A terminal-native AI agent for builders",
   "license": "MIT",
   "author": "ScreamCli",