xmengnet/go-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

refactor(workers): 优化AI和LSP工作者的错误处理和日志输出

Browse source
This commit is contained in:
liyp 2024-06-30 23:36:17 +08:00
parent be83757074
commit 21296b7878
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
workers/ai.go
View file

@ -83,7 +83,7 @@ func (a *AI) GetMsg() string {
log.Println("PROMRT 未配置") log.Println("PROMRT 未配置")
PROMPT = "" PROMPT = ""
} }
println("PROMPT:", PROMPT) // println("PROMPT:", PROMPT)
requestBody := map[string]interface{}{ requestBody := map[string]interface{}{
"model": MODEL, "model": MODEL,
"messages": []map[string]string{{ "messages": []map[string]string{{

5
workers/lsp.go
View file

@ -18,10 +18,13 @@ func (a *Lsp) GetMsg() string {
url := "https://api.lolicon.app/setu/v2?r18=0&size=small" url := "https://api.lolicon.app/setu/v2?r18=0&size=small"
resp, err := http.Get(url) resp, err := http.Get(url)
if err != nil { if err != nil {
return "获取失败" return "请求图片失败"
} }
defer resp.Body.Close() defer resp.Body.Close()
budy, err := io.ReadAll(resp.Body) budy, err := io.ReadAll(resp.Body)
if err != nil {
return "读取失败"
}
var res map[string]interface{} var res map[string]interface{}
err = json.Unmarshal(budy, &res) err = json.Unmarshal(budy, &res)
if err != nil { if err != nil {