fix: hf hosted inference check (#1128)

2023-09-09 00:29:48 +08:00
parent 681eb1cfcc
commit c4d8bdc3db
3 changed files with 69 additions and 4 deletions
--- a/api/core/model_providers/providers/huggingface_hub_provider.py
+++ b/api/core/model_providers/providers/huggingface_hub_provider.py
@@ -89,7 +89,8 @@ class HuggingfaceHubProvider(BaseModelProvider):
                raise CredentialsValidateFailedError('Task Type must be provided.')

            if credentials['task_type'] not in ("text2text-generation", "text-generation", "summarization"):
-                raise CredentialsValidateFailedError('Task Type must be one of text2text-generation, text-generation, summarization.')
+                raise CredentialsValidateFailedError('Task Type must be one of text2text-generation, '
+                                                     'text-generation, summarization.')

            try:
                llm = HuggingFaceEndpointLLM(