Ibraaheem commited on
Commit
766dd24
·
1 Parent(s): fcb3282

Update private_gpt/launcher.py

Browse files
Files changed (1) hide show
  1. private_gpt/launcher.py +2 -2
private_gpt/launcher.py CHANGED
@@ -96,7 +96,7 @@ def create_app(root_injector: Injector) -> FastAPI:
96
 
97
  @model_router.post("/switch_model")
98
  async def switch_model(
99
- new_model: str, current_user: dict = Depends(get_current_user), app: FastAPI = Depends(get_app)
100
  ):
101
  # Check if the user has either "admin" or "user" role
102
  if "user" not in current_user.get("role", []):
@@ -116,7 +116,7 @@ def create_app(root_injector: Injector) -> FastAPI:
116
  # Switch the model using the LLMComponent
117
  llm_component = root_injector.get(LLMComponent)
118
  llm_component.switch_model(new_model, settings=settings)
119
- await app.reload()
120
 
121
  # Return a success message
122
  return {"message": f"Model switched to {new_model}"}
 
96
 
97
  @model_router.post("/switch_model")
98
  async def switch_model(
99
+ new_model: str, current_user: dict = Depends(get_current_user)
100
  ):
101
  # Check if the user has either "admin" or "user" role
102
  if "user" not in current_user.get("role", []):
 
116
  # Switch the model using the LLMComponent
117
  llm_component = root_injector.get(LLMComponent)
118
  llm_component.switch_model(new_model, settings=settings)
119
+
120
 
121
  # Return a success message
122
  return {"message": f"Model switched to {new_model}"}