UniqAI
/
open-webui
mirror of https://github.com/open-webui/open-webui
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: cache request body

This commit is contained in:
Timothy J. Baek 2024-02-05 22:57:24 -08:00
parent 0b8df52c97
commit 1b040143eb
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/apps/openai/main.py
View File

@ -81,10 +81,15 @@ async def speech(request: Request, user=Depends(get_current_user)):
body = await request.body()
filename = hashlib.sha256(body).hexdigest() + ".mp3"
print(body)
print(type(body))
name = hashlib.sha256(body).hexdigest()
SPEECH_CACHE_DIR = Path(CACHE_DIR).joinpath("./audio/speech/")
SPEECH_CACHE_DIR.mkdir(parents=True, exist_ok=True)
file_path = SPEECH_CACHE_DIR.joinpath(filename)
file_path = SPEECH_CACHE_DIR.joinpath(f"{name}.mp3")
file_body_path = SPEECH_CACHE_DIR.joinpath(f"{name}.json")
print(file_path)
@ -114,6 +119,9 @@ async def speech(request: Request, user=Depends(get_current_user)):
for chunk in r.iter_content(chunk_size=8192):
f.write(chunk)
with open(file_body_path, "w") as f:
json.dump(json.loads(body.decode("utf-8")), f)
# Return the saved file
return FileResponse(file_path)