AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 05:55:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Bowen Liang 2025-02-19 09:13:36 +08:00 committed by GitHub
parent 2af81d1ee3
commit dfdd6dfa20
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/configs/middleware/__init__.py
View File

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

3
api/core/rag/datasource/retrieval_service.py
View File

@ -1,5 +1,6 @@
import concurrent.futures import concurrent.futures
import json import json
from concurrent.futures import ThreadPoolExecutor
from typing import Optional from typing import Optional
from flask import Flask, current_app from flask import Flask, current_app
@ -52,7 +53,7 @@ class RetrievalService:
exceptions: list[str] = [] exceptions: list[str] = []
# Optimize multithreading with thread pools # 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 = [] futures = []
if retrieval_method == "keyword_search": if retrieval_method == "keyword_search":
futures.append( futures.append(