Fix: export chat save in db; botPersona switching back to assistant

1 files changed, 3 insertions, 0 deletions

diff --git a/session.go b/session.go
index ee03397..c978f53 100644
--- a/session.go
+++ b/session.go
@@ -47,6 +47,9 @@ func importChat(filename string) error {
 		return err
 	}
 	activeChatName = filepath.Base(filename)
+	if _, ok := chatMap[activeChatName]; !ok {
+		addNewChat(activeChatName)
+	}
 	chatBody.Messages = messages
 	cfg.AssistantRole = messages[1].Role
 	if cfg.AssistantRole == cfg.UserRole {