UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-24 21:13:59 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

feat: tool cache dir

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-18 18:07:51 -07:00
parent eef125d085
commit 6f9a31eba5
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/apps/webui/routers/tools.py
View File

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