OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-09 07:56:42 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac: allow all params

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-05-29 02:56:37 +04:00
parent 551597b9cc
commit bb4115fa0e
2 changed files with 12 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
backend/open_webui/utils/middleware.py
View File

@ -667,29 +667,10 @@ def apply_params_to_form_data(form_data, model):
if "keep_alive" in params: if "keep_alive" in params:
form_data["keep_alive"] = params["keep_alive"] form_data["keep_alive"] = params["keep_alive"]
else: else:
if "seed" in params and params["seed"] is not None: if isinstance(params, dict):
form_data["seed"] = params["seed"] for key, value in params.items():
if value is not None:
if "stop" in params and params["stop"] is not None: form_data[key] = value
form_data["stop"] = params["stop"]
if "temperature" in params and params["temperature"] is not None:
form_data["temperature"] = params["temperature"]
if "max_tokens" in params and params["max_tokens"] is not None:
form_data["max_tokens"] = params["max_tokens"]
if "top_p" in params and params["top_p"] is not None:
form_data["top_p"] = params["top_p"]
if "frequency_penalty" in params and params["frequency_penalty"] is not None:
form_data["frequency_penalty"] = params["frequency_penalty"]
if "presence_penalty" in params and params["presence_penalty"] is not None:
form_data["presence_penalty"] = params["presence_penalty"]
if "reasoning_effort" in params and params["reasoning_effort"] is not None:
form_data["reasoning_effort"] = params["reasoning_effort"]
if "logit_bias" in params and params["logit_bias"] is not None: if "logit_bias" in params and params["logit_bias"] is not None:
try: try:
@ -703,7 +684,6 @@ def apply_params_to_form_data(form_data, model):
async def process_chat_payload(request, form_data, user, metadata, model): async def process_chat_payload(request, form_data, user, metadata, model):
form_data = apply_params_to_form_data(form_data, model) form_data = apply_params_to_form_data(form_data, model)
log.debug(f"form_data: {form_data}") log.debug(f"form_data: {form_data}")

11
backend/open_webui/utils/payload.py
View File

@ -45,9 +45,14 @@ def apply_model_params_to_body(
if not params: if not params:
return form_data return form_data
for key, cast_func in mappings.items(): for key, value in params.items():
if (value := params.get(key)) is not None: if value is not None:
form_data[key] = cast_func(value) if key in mappings:
cast_func = mappings[key]
if isinstance(cast_func, Callable):
form_data[key] = cast_func(value)
else:
form_data[key] = value
return form_data return form_data