Browse Source

feat: tool cache dir

Timothy J. Baek 10 tháng trước cách đây
mục cha
commit
6f9a31eba5
1 tập tin đã thay đổi với 5 bổ sung1 xóa
  1. 5 1
      backend/apps/webui/routers/tools.py

+ 5 - 1
backend/apps/webui/routers/tools.py

@@ -15,8 +15,9 @@ from constants import ERROR_MESSAGES
 
 from importlib import util
 import os
+from pathlib import Path
 
-from config import DATA_DIR
+from config import DATA_DIR, CACHE_DIR
 
 
 TOOLS_DIR = f"{DATA_DIR}/tools"
@@ -79,6 +80,9 @@ async def create_new_toolkit(
             specs = get_tools_specs(TOOLS[form_data.id])
             toolkit = Tools.insert_new_tool(user.id, form_data, specs)
 
+            tool_cache_dir = Path(CACHE_DIR) / "tools" / form_data.id
+            tool_cache_dir.mkdir(parents=True, exist_ok=True)
+
             if toolkit:
                 return toolkit
             else: