From 97257839dea81c76c83bc3d04e1489753bcf7fb2 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Wed, 23 Apr 2025 18:59:42 +0800
Subject: [PATCH] chore(release): increase Node.js memory limit in release
 workflow

- Added NODE_OPTIONS to set max-old-space-size to 8192 in the release workflow for Mac, Windows, and Linux builds.
---
 .github/workflows/release.yml                 | 3 +++
 src/renderer/src/services/KnowledgeService.ts | 9 ++++++---
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 7b55fbcd7e..b8b1731d05 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -76,6 +76,7 @@ jobs:
         env:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
+          NODE_OPTIONS: --max-old-space-size=8192
 
       - name: Build Mac
         if: matrix.os == 'macos-latest'
@@ -91,6 +92,7 @@ jobs:
           APPLE_TEAM_ID: ${{ vars.APPLE_TEAM_ID }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          NODE_OPTIONS: --max-old-space-size=8192
 
       - name: Build Windows
         if: matrix.os == 'windows-latest'
@@ -100,6 +102,7 @@ jobs:
         env:
           GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
           RENDERER_VITE_AIHUBMIX_SECRET: ${{ vars.RENDERER_VITE_AIHUBMIX_SECRET }}
+          NODE_OPTIONS: --max-old-space-size=8192
 
       - name: Release
         uses: ncipollo/release-action@v1
diff --git a/src/renderer/src/services/KnowledgeService.ts b/src/renderer/src/services/KnowledgeService.ts
index 5a0a1e5377..f594a44d7f 100644
--- a/src/renderer/src/services/KnowledgeService.ts
+++ b/src/renderer/src/services/KnowledgeService.ts
@@ -131,11 +131,10 @@ export const processKnowledgeSearch = async (
 
       const searchResults = Array.from(
         new Map(allSearchResults.flat().map((item) => [item.metadata.uniqueId || item.pageContent, item])).values()
-      )
-        .sort((a, b) => b.score - a.score)
-        .slice(0, documentCount)
+      ).sort((a, b) => b.score - a.score)
 
       console.log(`Knowledge base ${base.name} search results:`, searchResults)
+
       let rerankResults = searchResults
       if (base.rerankModel && searchResults.length > 0) {
         rerankResults = await window.api.knowledgeBase.rerank({
@@ -145,6 +144,10 @@ export const processKnowledgeSearch = async (
         })
       }
 
+      if (rerankResults.length > 0) {
+        rerankResults = rerankResults.slice(0, documentCount)
+      }
+
       const processdResults = await Promise.all(
         rerankResults.map(async (item) => {
           const file = await getFileFromUrl(item.metadata.source)