models: better error handling

manubot · Jan 5, 2023 · 473c764 · 473c764
1 parent c45d0a5
commit 473c764
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/manubot/ai_editor/models.py b/libs/manubot/ai_editor/models.py
@@ -446,7 +446,7 @@ def revise_paragraph(self, paragraph_text, section_name):
                 # if the error message suggests to sample again, let's do that
                 if "Please sample again" in error_message:
                     pass
-                elif "server is overloaded" in error_message:
+                elif "overloaded" in error_message:
                     time.sleep(5)
                 else:
                     # if the error mesaage suggests to reduce the number of tokens,