fix: 修复session维护过程的bug

2026-06-02 03:55:55 +00:00 · 2022-12-08 13:37:33 +08:00
parent b06754506a
commit c9547c32eb
2 changed files with 7 additions and 1 deletions
--- a/main.py
+++ b/main.py
@@ -53,5 +53,11 @@ if __name__ == '__main__':
        try:
            time.sleep(86400)
        except KeyboardInterrupt:
+            try:
+                for session in pkg.openai.session.sessions:
+                    pkg.openai.session.sessions[session].persistence()
+            except Exception as e:
+                if not isinstance(e, KeyboardInterrupt):
+                    raise e
            print("程序退出")
            break