Enha: chat management table

1 files changed, 2 insertions, 0 deletions

diff --git a/session.go b/session.go
index e34c98c..ee38843 100644
--- a/session.go
+++ b/session.go
@@ -76,6 +76,8 @@ func loadHistoryChat(chatName string) ([]models.RoleMsg, error) {
 		return nil, err
 	}
 	activeChatName = chatName
+	cfg.AssistantRole = chat.Agent
+	cfg.AssistantIcon = "<" + chat.Agent + ">: "
 	return chat.ToHistory()
 }