utils.go

package main

import (
	"encoding/base64"
	"fmt"
	"os"
	"path/filepath"

	"github.com/freesrz93/ask-gpt/consts"
)

const (
	configFile    = "config.yaml"
	sessionDir    = "sessions"
	tempSession   = "temp"
	defaultRole   = "default"
	backendOpenai = "openai"
	defaultPrompt = "You are a polymath. Your role is to provide accurate and clear answer to the user's questions. Include necessary technical details and terminologies in your answer."

	AIPrefix   = "Assistant: "
	UserPrefix = "User: "
)

var CfgDir = func() string {
	home, err := os.UserHomeDir()
	if err != nil {
		panic(err)
	}
	return filepath.Join(home, ".config", consts.AppName)
}()

func safeName(raw string) string {
	return base64.URLEncoding.EncodeToString([]byte(raw))
}

func P(s string) {
	_, _ = os.Stdout.WriteString(s)
}

func Pln() {
	P("\n")
}

func PFatal(v any) {
	P("error: " + fmt.Sprint(v))
	os.Exit(1)
}