From 8feb2281bc32d26617d2da510d2c878ab2fd8aa8 Mon Sep 17 00:00:00 2001
From: 0xMRTT <0xMRTT@proton.me>
Date: Thu, 3 Aug 2023 13:22:00 +0200
Subject: [PATCH] fix: hfprovider

---
 src/providers/hfbasechat.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/providers/hfbasechat.py b/src/providers/hfbasechat.py
index 0cc408b..ad01356 100644
--- a/src/providers/hfbasechat.py
+++ b/src/providers/hfbasechat.py
@@ -24,5 +24,10 @@ class BaseHFChatProvider(BaseProvider):
         })
 
         print(output)
-        
-        return output['generated_text']
+
+        if 'generated_text' in output:
+            return output['generated_text']
+        elif 'error' in output:
+            match output['error']:
+                case "Rate limit reached. Please log in or use your apiToken":
+                    return _("You've reached the rate limit! Please add a token to the preferences. You can get the token by following this [guide](https://bavarder.codeberg.page/help/huggingface/)")