ToolConfigs

langgenius · Jun 15, 2024 · 006dc60 · 006dc60
1 parent cf47865
commit 006dc60
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 15 deletions.
diff --git a/api/config.py b/api/config.py
@@ -46,7 +46,6 @@
     'INVITE_EXPIRY_HOURS': 72,
     'ETL_TYPE': 'dify',
     'KEYWORD_STORE': 'jieba',
-    'TOOL_ICON_CACHE_MAX_AGE': 3600,
     'MILVUS_DATABASE': 'default',
     'KEYWORD_DATA_SOURCE_TYPE': 'database',
     'INDEXING_MAX_SEGMENTATION_TOKENS_LENGTH': 1000,
@@ -261,9 +260,6 @@ def __init__(self):
         # Indexing Configurations.
         self.INDEXING_MAX_SEGMENTATION_TOKENS_LENGTH = get_env('INDEXING_MAX_SEGMENTATION_TOKENS_LENGTH')
 
-        # Tool Configurations.
-        self.TOOL_ICON_CACHE_MAX_AGE = get_env('TOOL_ICON_CACHE_MAX_AGE')
-
         # ------------------------
         # Platform Configurations.
         # ------------------------

diff --git a/api/configs/feature/__init__.py b/api/configs/feature/__init__.py
@@ -259,21 +259,33 @@ class ModerationConfigs(BaseModel):
     )
 
 
+class ToolConfigs(BaseModel):
+    """
+    Tool configs
+    """
+
+    TOOL_ICON_CACHE_MAX_AGE: PositiveInt = Field(
+        description='max age in seconds for tool icon caching',
+        default=3600,
+    )
+
+
 class FeatureConfigs(
-    OAuthConfigs,
-    ModerationConfigs,
+    AppExecutionConfigs,
+    BillingConfigs,
+    CodeExecutionSandboxConfigs,
     EndpointConfigs,
+    FileAccessConfigs,
+    FileUploadConfigs,
     HttpConfigs,
+    InnerAPIConfigs,
     LoggingConfigs,
+    ModelLoadBalanceConfigs,
+    ModerationConfigs,
+    OAuthConfigs,
     SecurityConfigs,
-    InnerAPIConfigs,
-    AppExecutionConfigs,
-    WorkflowConfigs,
-    CodeExecutionSandboxConfigs,
-    FileUploadConfigs,
-    FileAccessConfigs,
+    ToolConfigs,
     UpdateConfigs,
-    ModelLoadBalanceConfigs,
-    BillingConfigs,
+    WorkflowConfigs,
 ):
     pass
diff --git a/api/controllers/console/workspace/tool_providers.py b/api/controllers/console/workspace/tool_providers.py
@@ -104,7 +104,7 @@ class ToolBuiltinProviderIconApi(Resource):
     @setup_required
     def get(self, provider):
         icon_bytes, mimetype = BuiltinToolManageService.get_builtin_tool_provider_icon(provider)
-        icon_cache_max_age = int(current_app.config.get('TOOL_ICON_CACHE_MAX_AGE'))
+        icon_cache_max_age = current_app.config.get('TOOL_ICON_CACHE_MAX_AGE')
         return send_file(io.BytesIO(icon_bytes), mimetype=mimetype, max_age=icon_cache_max_age)
 
 class ToolApiProviderAddApi(Resource):