Building the Chat Manager in Go

Welcome back! In the previous lesson, we explored the importance of a robust system prompt and how it guides the behavior of our chatbot. Now, we will delve into the next step of our journey: building the chat manager. The ChatManager struct will be responsible for managing chat data effectively. By the end of this lesson, you will understand how to create, manage, and retrieve chat data using the ChatManager struct.

The ChatManager struct is designed to handle the storage and management of chat data. It serves as the backbone of our chatbot's data management system. We'll start by setting up the struct and then gradually add methods to handle chat creation, message addition, and conversation retrieval.

Let's begin by defining the ChatManager struct and its constructor. The constructor initializes an empty map, chats, which will store all chat data.

package main

import (
    "github.com/openai/openai-go"
)

type ChatManager struct {
    chats map[string]map[string]openai.ChatCompletionNewParams // user_id -> chat_id -> chat_data
}

func NewChatManager() *ChatManager {
    return &ChatManager{
        chats: make(map[string]map[string]openai.ChatCompletionNewParams),
    }
}

In this setup, chats is a nested map where the first key is the user_id, and the second key is the chat_id. This structure allows us to efficiently manage multiple chats for different users.

Next, we'll add the CreateChat method. This method is responsible for creating a new chat entry for a user. It takes three parameters: userID, chatID, and systemPrompt.

func (cm *ChatManager) CreateChat(userID, chatID, systemPrompt string) {
    if _, exists := cm.chats[userID]; !exists {
        cm.chats[userID] = make(map[string]openai.ChatCompletionNewParams)
    }

    cm.chats[userID][chatID] = openai.ChatCompletionNewParams{
        Messages: []openai.ChatCompletionMessageParamUnion{
            openai.SystemMessage(systemPrompt),
        },
    }
}

The CreateChat method checks if the userID exists in chats. If not, it creates a new entry. Then, it initializes the chat with the provided systemPrompt as a system message.