Skip to content

Commit 5411e93

Browse files
committed
refactor: auto create mcp images folder if not exist
1 parent 4d3f60b commit 5411e93

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

backend/main.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
import os
2+
13
import sqlbot_xpack
24
from alembic.config import Config
35
from fastapi import FastAPI
@@ -56,7 +58,9 @@ def custom_generate_unique_id(route: APIRoute) -> str:
5658

5759
mcp_app = FastAPI()
5860
# mcp server, images path
59-
mcp_app.mount("/images", StaticFiles(directory=settings.MCP_IMAGE_PATH), name="images")
61+
images_path = settings.MCP_IMAGE_PATH
62+
os.makedirs(images_path, exist_ok=True)
63+
mcp_app.mount("/images", StaticFiles(directory=images_path), name="images")
6064

6165
mcp = FastApiMCP(
6266
app,

0 commit comments

Comments
 (0)