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

2026-04-01 05:39:34 -05:00
parent 16e6cb458f
commit d7a5a903a6
3 changed files with 17 additions and 2 deletions
@@ -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'],