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

added ram usage endpoint

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-06-06 20:23:33 +02:00
parent 5f2b53e309
commit afd8d2a4ef
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
app.py
View File

@ -90,6 +90,10 @@ class LoLLMsWebUI(LoLLMsAPPI):
"/disk_usage", "disk_usage", self.disk_usage, methods=["GET"]
)
self.add_endpoint(
"/ram_usage", "ram_usage", self.ram_usage, methods=["GET"]
)
self.add_endpoint(
"/list_bindings", "list_bindings", self.list_bindings, methods=["GET"]
@ -456,7 +460,19 @@ class LoLLMsWebUI(LoLLMsAPPI):
print("Set config results:")
print(result)
return jsonify(result)
def ram_usage(self):
"""
Returns the RAM usage in bytes.
"""
ram = psutil.virtual_memory()
return jsonify({
"total_space":ram.total,
"available_space":ram.free,
"percent_usage":ram.percent,
"binding_models_usage": ram.used
})
def disk_usage(self):
current_drive = Path.cwd().anchor
drive_disk_usage = psutil.disk_usage(current_drive)