"test/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "9f32a06f16774fe2226f6c4b9b0da014fbcb77c1"
Commit d2c5f3d5 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

refac: convert str var to f-string

parent 5b26d2a6
...@@ -2,5 +2,5 @@ from peewee import * ...@@ -2,5 +2,5 @@ from peewee import *
from config import DATA_DIR from config import DATA_DIR
DB = SqliteDatabase(str(DATA_DIR / "ollama.db")) DB = SqliteDatabase(f"{DATA_DIR}/ollama.db")
DB.connect() DB.connect()
...@@ -96,7 +96,7 @@ async def download( ...@@ -96,7 +96,7 @@ async def download(
file_name = parse_huggingface_url(url) file_name = parse_huggingface_url(url)
if file_name: if file_name:
file_path = str(UPLOAD_DIR / file_name) file_path = f"{UPLOAD_DIR}/{file_name}"
return StreamingResponse( return StreamingResponse(
download_file_stream(url, file_path, file_name), download_file_stream(url, file_path, file_name),
...@@ -108,7 +108,7 @@ async def download( ...@@ -108,7 +108,7 @@ async def download(
@router.post("/upload") @router.post("/upload")
def upload(file: UploadFile = File(...)): def upload(file: UploadFile = File(...)):
file_path = UPLOAD_DIR / file.filename file_path = f"{UPLOAD_DIR}/{file.filename}"
# Save file in chunks # Save file in chunks
with file_path.open("wb+") as f: with file_path.open("wb+") as f:
......
...@@ -15,6 +15,7 @@ from pathlib import Path ...@@ -15,6 +15,7 @@ from pathlib import Path
try: try:
from dotenv import load_dotenv, find_dotenv from dotenv import load_dotenv, find_dotenv
load_dotenv(find_dotenv("../.env")) load_dotenv(find_dotenv("../.env"))
except ImportError: except ImportError:
print("dotenv not installed, skipping...") print("dotenv not installed, skipping...")
...@@ -24,12 +25,12 @@ except ImportError: ...@@ -24,12 +25,12 @@ except ImportError:
# File Upload # File Upload
#################################### ####################################
DATA_DIR = Path(os.getenv("DATA_DIR", './data')).resolve() DATA_DIR = Path(os.getenv("DATA_DIR", "./data")).resolve()
UPLOAD_DIR = DATA_DIR / "uploads" UPLOAD_DIR = f"{DATA_DIR}/uploads"
UPLOAD_DIR.mkdir(parents=True, exist_ok=True) UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
WEB_DIR = Path(os.getenv("WEB_DIR", "../build")) FRONTEND_BUILD_DIR = Path(os.getenv("FRONTEND_BUILD_DIR", "../build"))
#################################### ####################################
# ENV (dev,test,prod) # ENV (dev,test,prod)
...@@ -84,7 +85,7 @@ if WEBUI_AUTH and WEBUI_JWT_SECRET_KEY == "": ...@@ -84,7 +85,7 @@ if WEBUI_AUTH and WEBUI_JWT_SECRET_KEY == "":
# RAG # RAG
#################################### ####################################
CHROMA_DATA_PATH = DATA_DIR / "vector_db" CHROMA_DATA_PATH = f"{DATA_DIR}/vector_db"
EMBED_MODEL = "all-MiniLM-L6-v2" EMBED_MODEL = "all-MiniLM-L6-v2"
CHROMA_CLIENT = chromadb.PersistentClient( CHROMA_CLIENT = chromadb.PersistentClient(
path=str(CHROMA_DATA_PATH), settings=Settings(allow_reset=True) path=str(CHROMA_DATA_PATH), settings=Settings(allow_reset=True)
......
...@@ -14,7 +14,7 @@ from apps.openai.main import app as openai_app ...@@ -14,7 +14,7 @@ from apps.openai.main import app as openai_app
from apps.web.main import app as webui_app from apps.web.main import app as webui_app
from apps.rag.main import app as rag_app from apps.rag.main import app as rag_app
from config import ENV, WEB_DIR from config import ENV, FRONTEND_BUILD_DIR
class SPAStaticFiles(StaticFiles): class SPAStaticFiles(StaticFiles):
...@@ -58,4 +58,8 @@ app.mount("/openai/api", openai_app) ...@@ -58,4 +58,8 @@ app.mount("/openai/api", openai_app)
app.mount("/rag/api/v1", rag_app) app.mount("/rag/api/v1", rag_app)
app.mount("/", SPAStaticFiles(directory=str(WEB_DIR), html=True), name="spa-static-files") app.mount(
"/",
SPAStaticFiles(directory=str(FRONTEND_BUILD_DIR), html=True),
name="spa-static-files",
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment