From bfa61ae3ee117ab9eef5358c9fcbcbe8ab538e41 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Tue, 4 Feb 2025 12:35:42 +0800
Subject: [PATCH] feat: reduce embedding batch size to 5

---
 src/main/services/KnowledgeService.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/services/KnowledgeService.ts b/src/main/services/KnowledgeService.ts
index 9010f4c2..ea178dec 100644
--- a/src/main/services/KnowledgeService.ts
+++ b/src/main/services/KnowledgeService.ts
@@ -45,14 +45,14 @@ class KnowledgeService {
               azureOpenAIApiDeploymentName: model,
               azureOpenAIApiInstanceName: getInstanceName(baseURL),
               dimensions,
-              batchSize: 10
+              batchSize: 5
             })
           : new OpenAiEmbeddings({
               model,
               apiKey,
               configuration: { baseURL },
               dimensions,
-              batchSize: 10
+              batchSize: 5
             })
       )
       .setVectorDatabase(new LibSqlDb({ path: path.join(this.storageDir, id) }))