🐛 fix: fix file relative chunks (lobehub#3676)

* 🌐 chore: update i18n * 🐛 fix: fix relative files
valarpirai · Aug 30, 2024 · afe1906 · afe1906
1 parent fe62aee
commit afe1906
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/locales/en-US/components.json b/locales/en-US/components.json
@@ -59,7 +59,7 @@
       "status": {
         "error": "Chunking failed",
         "errorResult": "Chunking failed, please check and try again. Error detail:",
-        "processing": "Chunking in progress.",
+        "processing": "Chunking",
         "processingTip": "The server is splitting text chunks; closing the page will not affect the chunking progress."
       }
     }

diff --git a/src/database/server/models/chunk.ts b/src/database/server/models/chunk.ts
@@ -171,6 +171,10 @@ export class ChunkModel {
   }) {
     const similarity = sql<number>`1 - (${cosineDistance(embeddings.embeddings, embedding)})`;
 
+    const hasFiles = fileIds && fileIds.length > 0;
+
+    if (!hasFiles) return [];
+
     const result = await serverDB
       .select({
         fileId: files.id,
@@ -186,7 +190,7 @@ export class ChunkModel {
       .leftJoin(embeddings, eq(chunks.id, embeddings.chunkId))
       .leftJoin(fileChunks, eq(chunks.id, fileChunks.chunkId))
       .leftJoin(files, eq(files.id, fileChunks.fileId))
-      .where(and(fileIds && fileIds.length > 0 ? inArray(fileChunks.fileId, fileIds) : undefined))
+      .where(inArray(fileChunks.fileId, fileIds))
       .orderBy((t) => desc(t.similarity))
       .limit(5);