Fix for #158 (#162)

Josh-XT · Apr 29, 2023 · 3a6a9d1 · 3a6a9d1
1 parent 6c2fced
commit 3a6a9d1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/provider/huggingchat.py b/provider/huggingchat.py
@@ -38,8 +38,10 @@ def instruct(self, prompt: str) -> str:
             },
             stream=True,
         )
-
-        assert res.status_code == 200, "Failed to send message"
+        if res.status_code != 200:
+            print(f"Status code: {res.status_code}")
+            print(f"Response content: {res.content}")
+            raise Exception("Failed to send prompt to AI")
 
         last_response = None
         for chunk in res.iter_content(chunk_size=None):