fix: adjust env's params

langgenius · Nov 27, 2024 · 6f4e2d5 · 6f4e2d5
1 parent 1846793
commit 6f4e2d5
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/api/.env.example b/api/.env.example
@@ -412,4 +412,4 @@ RESET_PASSWORD_TOKEN_EXPIRY_MINUTES=5
 
 CREATE_TIDB_SERVICE_JOB_ENABLED=false
 
-RETRIEVAL_TOP_N=
+RETRIEVAL_TOP_N=1
diff --git a/api/configs/feature/__init__.py b/api/configs/feature/__init__.py
@@ -621,7 +621,7 @@ class DataSetConfig(BaseSettings):
         default=30,
     )
 
-    RETRIEVAL_TOP_N: Optional[PositiveInt] = Field(description="number of retrieval top_n", default=None)
+    RETRIEVAL_TOP_N: int = Field(description="number of retrieval top_n", default=0)
 
 
 class WorkspaceConfig(BaseSettings):

diff --git a/docker/docker-compose.yaml b/docker/docker-compose.yaml
@@ -287,7 +287,7 @@ x-shared-env: &shared-api-worker-env
   OCEANBASE_CLUSTER_NAME: ${OCEANBASE_CLUSTER_NAME:-difyai}
   OCEANBASE_MEMORY_LIMIT: ${OCEANBASE_MEMORY_LIMIT:-6G}
   CREATE_TIDB_SERVICE_JOB_ENABLED: ${CREATE_TIDB_SERVICE_JOB_ENABLED:-false}
-  RETRIEVAL_TOP_N: ${RETRIEVAL_TOP_N:-}
+  RETRIEVAL_TOP_N: ${RETRIEVAL_TOP_N:-0}
 
 services:
   # API service
Original file line number	Diff line number	Diff line change
Expand Up		@@ -412,4 +412,4 @@ RESET_PASSWORD_TOKEN_EXPIRY_MINUTES=5

		CREATE_TIDB_SERVICE_JOB_ENABLED=false

		RETRIEVAL_TOP_N=
		RETRIEVAL_TOP_N=1