1 files changed, 18 insertions, 0 deletions
diff --git a/session.go b/session.go
index c3d826f..21b5b9e 100644
--- a/session.go
+++ b/session.go
@@ -79,6 +79,24 @@ func loadHistoryChat(chatName string) ([]models.RoleMsg, error) {
 	return chat.ToHistory()
 }
 
+func loadAgentsLastChat(agent string) ([]models.RoleMsg, error) {
+	chat, err := store.GetLastChatByAgent(agent)
+	if err != nil {
+		return nil, err
+	}
+	history, err := chat.ToHistory()
+	if err != nil {
+		return nil, err
+	}
+	if chat.Name == "" {
+		logger.Warn("empty chat name", "id", chat.ID)
+		chat.Name = fmt.Sprintf("%s_%d", chat.Agent, chat.ID)
+	}
+	chatMap[chat.Name] = chat
+	activeChatName = chat.Name
+	return history, nil
+}
+
 func loadOldChatOrGetNew() []models.RoleMsg {
 	newChat := &models.Chat{
 		ID:        0,