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

added personality editor

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-02-07 00:36:55 +01:00
parent 33ec486144
commit 1b7245fe55
1 changed files with 50 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
lollms/server/endpoints/lollms_personalities_infos.py
View File

@ -232,6 +232,56 @@ def clear_personality_files_list():
# ------------------------------------------- Mounting/Unmounting/Remounting ------------------------------------------------
class PersonalityDataRequest(BaseModel):
category:str
name:str
@router.post("/get_personality_config")
def get_personality_config(data:PersonalityDataRequest):
print("- Recovering personality config")
category = data.category
name = data.name
package_path = f"{category}/{name}"
if category=="custom_personalities":
package_full_path = lollmsElfServer.lollms_paths.custom_personalities_path/f"{name}"
else:
package_full_path = lollmsElfServer.lollms_paths.personalities_zoo_path/package_path
config_file = package_full_path / "config.yaml"
if config_file.exists():
with open(config_file,"r") as f:
config = yaml.safe_load(f)
return {"status":True, "config":config}
else:
return {"status":False, "error":"Not found"}
class PersonalityDataRequest(BaseModel):
category:str
name:str
config:dict
@router.post("/set_personality_config")
def set_personality_config(data:PersonalityDataRequest):
print("- Recovering personality config")
category = data.category
name = data.name
config = data.config
package_path = f"{category}/{name}"
if category=="custom_personalities":
package_full_path = lollmsElfServer.lollms_paths.custom_personalities_path/f"{name}"
else:
package_full_path = lollmsElfServer.lollms_paths.personalities_zoo_path/package_path
config_file = package_full_path / "config.yaml"
if config_file.exists():
with open(config_file,"w") as f:
yaml.safe_dump(config, f)
return {"status":True}
else:
return {"status":False, "error":"Not found"}
class PersonalityMountingInfos(BaseModel):
category:str