@@ -123,7 +123,7 @@ class Recraft(SimpleService):
         }
         data = {'input': payload}
         response = requests.post(
-            url=f'{self.generate_url}{payload['version']}/predictions',
+            url=f'{self.generate_url}{payload.get('version', 'recraft-v3')}/predictions',
             headers=headers,
             json=data,
         )
@@ -186,7 +186,7 @@ class Recraft(SimpleService):
             'линогравюра': 'linocut'
         }
         start_time = time.time()
-        extension = '.svg' if input_message.info['version'] == self.versions[1].slug else '.png'
+        extension = '.svg' if input_message.info.get('version', 'recraft-v3') == self.versions[1].slug else '.png'
         size = self._get_size(input_message.info.pop('width', 1024), input_message.info.pop('height', 1024))
         callback_data = dict(
             {