feat: editable prompt suggestions integration

This commit is contained in:
Timothy J. Baek 2024-01-22 21:53:13 -08:00
parent 4e1b52e91b
commit c4a039326f
6 changed files with 172 additions and 52 deletions

View file

@ -49,14 +49,15 @@ async def set_global_default_models(
)
@router.post("/default/suggestions", response_model=str)
@router.post("/default/suggestions", response_model=List[PromptSuggestion])
async def set_global_default_suggestions(
request: Request,
form_data: SetDefaultSuggestionsForm,
user=Depends(get_current_user),
):
if user.role == "admin":
request.app.state.DEFAULT_PROMPT_SUGGESTIONS = form_data.suggestions
data = form_data.model_dump()
request.app.state.DEFAULT_PROMPT_SUGGESTIONS = data["suggestions"]
return request.app.state.DEFAULT_PROMPT_SUGGESTIONS
else:
raise HTTPException(