Files
chatgpt-dingtalk/config/prompt.go
2023-03-20 21:13:05 +08:00

30 lines
415 B
Go

package config
import (
"io/ioutil"
"log"
"gopkg.in/yaml.v2"
)
type Prompt struct {
Title string `yaml:"title"`
Content string `yaml:"content"`
}
var prompTmp *[]Prompt
// LoadPrompt 加载Prompt
func LoadPrompt() *[]Prompt {
data, err := ioutil.ReadFile("prompt.yml")
if err != nil {
log.Fatal(err)
}
err = yaml.Unmarshal(data, &prompTmp)
if err != nil {
log.Fatal(err)
}
return prompTmp
}