diff --git a/api/services/completion_service.py b/api/services/completion_service.py
index ac669997f7..6b8ee520eb 100644
--- a/api/services/completion_service.py
+++ b/api/services/completion_service.py
@@ -390,8 +390,8 @@ class CompletionService:
                         result = json.loads(result)
                         if result.get('error'):
                             cls.handle_error(result)
-
-                        return cls.get_message_response_data(result.get('data'))
+                        if '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
                     raise CompletionStoppedError()