Fixes unit tests

Author: mrDzurb

tests/unitary/with_extras/aqua/test_model_handler.py

@@ -6,6 +6,7 @@
 from unittest import TestCase
 from unittest.mock import MagicMock, patch
 
+from aqua.model.constants import ModelTask
 import pytest
 from huggingface_hub.hf_api import HfApi, ModelInfo
 from huggingface_hub.utils import GatedRepoError
@@ -276,7 +277,8 @@ def test_post_negative(self, mock_uuid, mock_format_hf_custom_error_message):
                 '{"status": 400, "message": "Something went wrong with your request.", '
                 '"service_payload": {}, "reason": "Unsupported pipeline tag for the chosen '
                 "model: 'not-text-generation'. AQUA currently supports the following tasks only: "
-                'text-generation. Please select a model with a compatible pipeline tag.", "request_id": "###"}'
+                f'{", ".join(ModelTask.values())}. '
+                'Please select a model with a compatible pipeline tag.", "request_id": "###"}'
             )
         get_hf_model_info.cache_clear()