@@ -80,7 +80,7 @@ class BaseGenerationView(APIView):
             )
         service: type[SimpleService] = getattr(sys.modules['ml_model.services'], f'{model.slug.title()}')
         info = serializer.validated_data.pop('info', {})
-        input_message = Message(
+        input_message = Message.objects.create(
             **serializer.validated_data,
             info=info,
             content_object=store,