mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 11:42:44 +08:00
fix: 修复session维护过程的bug
This commit is contained in:
parent
b06754506a
commit
c9547c32eb
6
main.py
6
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
|
||||
|
|
|
@ -35,8 +35,8 @@ class DatabaseManager:
|
|||
|
||||
def heartbeat(self):
|
||||
while True:
|
||||
self.conn.ping(reconnect=True)
|
||||
time.sleep(30)
|
||||
self.conn.ping(reconnect=True)
|
||||
|
||||
def reconnect(self):
|
||||
self.conn = pymysql.connect(host=self.host, port=self.port, user=self.user, password=self.password,
|
||||
|
|
Loading…
Reference in New Issue
Block a user