Commit e69f3126 authored by Jun Siang Cheah's avatar Jun Siang Cheah
Browse files

fix: remove unused API for setting model config

parent e76a444e
...@@ -35,19 +35,6 @@ class SetDefaultSuggestionsForm(BaseModel): ...@@ -35,19 +35,6 @@ class SetDefaultSuggestionsForm(BaseModel):
suggestions: List[PromptSuggestion] suggestions: List[PromptSuggestion]
class ModelConfig(BaseModel):
id: str
name: str
description: str
vision_capable: bool
class SetModelConfigForm(BaseModel):
ollama: List[ModelConfig]
litellm: List[ModelConfig]
openai: List[ModelConfig]
############################ ############################
# SetDefaultModels # SetDefaultModels
############################ ############################
...@@ -70,14 +57,3 @@ async def set_global_default_suggestions( ...@@ -70,14 +57,3 @@ async def set_global_default_suggestions(
data = form_data.model_dump() data = form_data.model_dump()
request.app.state.DEFAULT_PROMPT_SUGGESTIONS = data["suggestions"] request.app.state.DEFAULT_PROMPT_SUGGESTIONS = data["suggestions"]
return request.app.state.DEFAULT_PROMPT_SUGGESTIONS return request.app.state.DEFAULT_PROMPT_SUGGESTIONS
@router.post("/models", response_model=SetModelConfigForm)
async def set_global_default_suggestions(
request: Request,
form_data: SetModelConfigForm,
user=Depends(get_admin_user),
):
data = form_data.model_dump()
request.app.state.MODEL_CONFIG = data
return request.app.state.MODEL_CONFIG
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment