Personalizing Your Tutor with System Prompts in Go

package main

import (
    "context"
    "fmt"
    "log"
    "os"
    "strings"
    "github.com/openai/openai-go"
    "github.com/openai/openai-go/option"
)

// Message struct for session history
type Message struct {
    Role    string
    Content string
}

// SendQuery sends the session history to DeepSeek and returns the AI's reply.
func SendQuery(client *openai.Client, session []Message) (string, error) {
    var messages []openai.ChatCompletionMessageParamUnion
    for _, m := range session {
        switch m.Role {
        case "user":
            messages = append(messages, openai.UserMessage(m.Content))
        case "assistant":
            messages = append(messages, openai.AssistantMessage(m.Content))
        case "system":
            messages = append(messages, openai.SystemMessage(m.Content))
        }
    }

    chatCompletion, err := client.Chat.Completions.New(
        context.Background(),
        openai.ChatCompletionNewParams{
            Messages: messages,
            Model:    "deepseek-ai/DeepSeek-V3",
        },
    )
    if err != nil {
        return "", err
    }
    if len(chatCompletion.Choices) == 0 {
        return "", nil
    }
    return chatCompletion.Choices[0].Message.Content, nil
}

// Helper function to capitalize the role
func capitalize(s string) string {
    if len(s) == 0 {
        return s
    }
    return strings.ToUpper(string(s[0])) + s[1:]
}

func main() {
    // Read OpenAI credentials from environment variables
    apiKey := os.Getenv("OPENAI_API_KEY")
    baseURL := os.Getenv("OPENAI_BASE_URL")

    if apiKey == "" {
        log.Fatal("Please set the OPENAI_API_KEY environment variable.")
    }


    client := openai.NewClient(
        option.WithAPIKey(apiKey),
        option.WithBaseURL(baseURL),
    )

    // Define the tutor's personality and expertise
    systemPrompt := "You are an experienced and patient tutor specialized in math and science, providing clear and concise explanations."

    // Structure the tutoring session with a system prompt and student query
    session := []Message{
        {Role: "system", Content: systemPrompt},
        {Role: "user", Content: "Can you help me understand the concept of derivatives in calculus?"},
    }

    // Get the tutor's response
    reply, err := SendQuery(&client, session)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("Tutor Response:", reply)
}