diff --git a/src/server.py b/src/server.py index d1cd6f1..c2b91e8 100644 --- a/src/server.py +++ b/src/server.py @@ -104,11 +104,12 @@ def get_cpu_info(): except Exception as e: return f"Error retrieving CPU info: {e}" -def system_info(): +def system_info(model): return { "Installed RAM": get_installed_ram(), "Graphics Card": get_graphics_cards(), - "CPU": get_cpu_info() + "CPU": get_cpu_info(), + "LLM Model": model } # %% @@ -471,7 +472,7 @@ class WebServer: @self.app.get('/api/system-info/{context_id}') async def get_system_info(context_id: str): - return JSONResponse(system_info()) + return JSONResponse(system_info(self.model)) @self.app.post('/api/chat/{context_id}') async def chat_endpoint(context_id: str, request: Request):