ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2024-12-20 04:47:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed playground endpoint

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-02-25 10:50:44 +01:00
parent 513ce6c546
commit f96fb8f263
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
endpoints/lollms_playground.py
View File

@ -64,8 +64,7 @@ async def add_preset(preset_data: PresetData):
presets_folder.mkdir(exist_ok=True, parents=True)
# Ensure the name doesn't contain any path manipulation characters
if ".." in preset_data.name or "/" in preset_data.name:
raise HTTPException(status_code=400, detail="Invalid preset name")
sanitize_path_from_endpoint(preset_data.name,exception_text="Invalid preset name")
fn = preset_data.name.lower().replace(" ","_")
filename = presets_folder/f"{fn}.yaml"
@ -89,8 +88,7 @@ async def del_preset(preset_data: PresetData):
raise HTTPException(status_code=400, detail="Preset name is missing in the request")
# Ensure the name doesn't contain any path manipulation characters
if ".." in preset_data.name or "/" in preset_data.name:
raise HTTPException(status_code=400, detail="Invalid preset name")
sanitize_path_from_endpoint(preset_data.name,exception_text="Invalid preset name")
presets_file = lollmsElfServer.lollms_paths.personal_discussions_path/"lollms_playground_presets"/preset_data.name
try:
@ -116,9 +114,9 @@ async def save_presets(preset_data: PresetDataWithValue):
if preset_data.preset is None:
raise HTTPException(status_code=400, detail="Preset data is missing in the request")
# Ensure the name doesn't contain any path manipulation characters
if ".." in preset_data.name or "/" in preset_data.name:
raise HTTPException(status_code=400, detail="Invalid preset name")
sanitize_path_from_endpoint(preset_data.name,exception_text="Invalid preset name")
presets_file = lollmsElfServer.lollms_paths.personal_discussions_path/"presets.json"
# Save the JSON data to a file.

2
lollms_core

@ -1 +1 @@
Subproject commit a98fa14046b6e8a16e3d5bebcab29cae13ada7ba
Subproject commit b786cfaa837abbfd97a61f2d527576163c127d00