diff --git a/api/services/completion_service.py b/api/services/completion_service.py
index 3c05e99c9e..531d803f54 100644
--- a/api/services/completion_service.py
+++ b/api/services/completion_service.py
@@ -367,7 +367,7 @@ class CompletionService:
                         result = json.loads(result)
                         if result.get('error'):
                             cls.handle_error(result)
-                        if 'data' in result:
+                        if result['event'] == 'message' and 'data' in result:
                             return cls.get_message_response_data(result.get('data'))
             except ValueError as e:
                 if e.args[0] != "I/O operation on closed file.":  # ignore this error