Browse Source

fix: change the config name and fix typo in description of the number of retrieval executors (#13856)

Bowen Liang 2 tháng trước cách đây
mục cha
commit
dfdd6dfa20

+ 2 - 2
api/configs/middleware/__init__.py

@@ -167,8 +167,8 @@ class DatabaseConfig(BaseSettings):
         default=False,
     )
 
-    RETRIEVAL_SERVICE_WORKER: NonNegativeInt = Field(
-        description="If True, enables the retrieval service worker.",
+    RETRIEVAL_SERVICE_EXECUTORS: NonNegativeInt = Field(
+        description="Number of processes for the retrieval service, default to CPU cores.",
         default=os.cpu_count(),
     )
 

+ 2 - 1
api/core/rag/datasource/retrieval_service.py

@@ -1,5 +1,6 @@
 import concurrent.futures
 import json
+from concurrent.futures import ThreadPoolExecutor
 from typing import Optional
 
 from flask import Flask, current_app
@@ -52,7 +53,7 @@ class RetrievalService:
         exceptions: list[str] = []
 
         # Optimize multithreading with thread pools
-        with concurrent.futures.ThreadPoolExecutor(max_workers=dify_config.RETRIEVAL_SERVICE_WORKER) as executor:  # type: ignore
+        with ThreadPoolExecutor(max_workers=dify_config.RETRIEVAL_SERVICE_EXECUTORS) as executor:  # type: ignore
             futures = []
             if retrieval_method == "keyword_search":
                 futures.append(