perf: 在session自动重置时保留非default的prompt (#494)

This commit is contained in:
Rock Chin 2023-07-06 23:09:39 +08:00
parent 06037df607
commit b4fc1057d1

View File

@ -345,7 +345,8 @@ class Session:
if expired:
pkg.utils.context.get_database_manager().set_session_expired(self.name, self.create_timestamp)
self.default_prompt = self.get_default_prompt(use_prompt)
if use_prompt:
self.default_prompt = self.get_default_prompt(use_prompt)
self.prompt = []
self.token_counts = []
self.create_timestamp = int(time.time())