UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-24 08:16:02 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Minor bugfixes

Browse Source
This commit is contained in:
Phil Szalay 2025-03-20 15:52:24 +01:00
parent a9dbeb79db
commit 1a2d7a1a51
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
backend/beyond_the_loop/routers/openai.py
View File

@ -560,8 +560,7 @@ async def generate_chat_completion(
payload = {**form_data} payload = {**form_data}
metadata = payload.pop("metadata", None) metadata = payload.pop("metadata", None)
model_id = form_data.get("model") model_info = Models.get_model_by_id(form_data.get("model"))
model_info = Models.get_model_by_id(model_id)
has_chat_id = "chat_id" in metadata and metadata["chat_id"] is not None has_chat_id = "chat_id" in metadata and metadata["chat_id"] is not None
@ -569,6 +568,14 @@ async def generate_chat_completion(
model_message_credit_cost = 0 model_message_credit_cost = 0
if has_chat_id: if has_chat_id:
if model_info is None:
raise HTTPException(
status_code=404,
detail="Model not found. Please check the model ID is correct.",
)
# Check for base_model_id first in case of user defined custom model
model_id = model_info.base_model_id if model_info.base_model_id else model_info.id
model_message_credit_cost = ModelMessageCreditCosts.get_cost_by_model(model_id) model_message_credit_cost = ModelMessageCreditCosts.get_cost_by_model(model_id)
# Get current credit balance # Get current credit balance