fix: The proxy configuration is not correctly passed.

CherryHQ · Feb 28, 2025 · 5bacf04 · 5bacf04
1 parent 1d4916c
commit 5bacf04
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/main/services/KnowledgeService.ts b/src/main/services/KnowledgeService.ts
@@ -13,6 +13,7 @@ import { getAllFiles } from '@main/utils/file'
 import type { LoaderReturn } from '@shared/config/types'
 import { FileType, KnowledgeBaseParams, KnowledgeItem } from '@types'
 import { app } from 'electron'
+import { ProxyAgent, setGlobalDispatcher } from 'undici'
 import { v4 as uuidv4 } from 'uuid'
 
 import { proxyManager } from './ProxyManager'
@@ -85,6 +86,7 @@ class KnowledgeService {
     _: Electron.IpcMainInvokeEvent,
     { base, item, forceReload = false }: { base: KnowledgeBaseParams; item: KnowledgeItem; forceReload: boolean }
   ): Promise<LoaderReturn> => {
+    setGlobalDispatcher(new ProxyAgent(proxyManager.getProxyUrl() || ''))
     const ragApplication = await this.getRagApplication(base)
 
     const sendDirectoryProcessingPercent = (totalFiles: number, processedFiles: number) => {

diff --git a/src/main/services/ProxyManager.ts b/src/main/services/ProxyManager.ts
@@ -66,6 +66,7 @@ export class ProxyManager {
   private async setCustomProxy(): Promise<void> {
     try {
       if (this.config.url) {
+        this.proxyUrl = this.config.url
         this.proxyAgent = new HttpsProxyAgent(this.config.url)
         this.setEnvironment(this.config.url)
         await this.setSessionsProxy({ proxyRules: this.config.url })