fix: guard completed API with message id (#23184)

This commit is contained in:
Shirasawa authored and GitHub committed 2026-04-01 05:39:34 -05:00
1 parent 16e6cb458f
commit d7a5a903a6
3 files changed
+17 -2

No files matched your search

+7
View File
@@ -316,6 +316,10 @@ async def chat_completed(request: Request, form_data: dict, user: Any):
models = request.app.state.MODELS
data = form_data
if not data.get('id'):
raise Exception('Missing message id')
model_id = data['model']
if model_id not in models:
raise Exception('Model not found')
@@ -327,6 +331,9 @@ async def chat_completed(request: Request, form_data: dict, user: Any):
except Exception as e:
raise Exception(f'Error: {e}')
if not data.get('id'):
raise Exception('Missing message id')
metadata = {
'chat_id': data['chat_id'],
'message_id': data['id'],