fix: 在 OpenAI token 超出限制后丢弃历史会话(#193)

wzpan · Mar 26, 2023 · fee700c · fee700c
1 parent c00f454
commit fee700c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/robot/AI.py b/robot/AI.py
@@ -265,7 +265,10 @@ def chat(self, texts, parsed):
                 respond = response.choices[0].text
             logger.info(f"openai response: {respond}")
             return respond
-
+        except self.openai.error.InvalidRequestError:
+            logger.warning("token超出长度限制，丢弃历史会话")
+            self.context = []
+            self.chat(texts, parsed)
         except Exception:
             logger.critical(
                 "openai robot failed to response for %r", msg, exc_info=True