37 lines
890 B
Go
37 lines
890 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/ssgo/httpclient"
|
||
|
|
"github.com/ssgo/log"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Ans struct {
|
||
|
|
Choices []struct {
|
||
|
|
Message struct {
|
||
|
|
Content string
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func ask(text string) string {
|
||
|
|
log.DefaultLogger.Info("start ask", "text", text)
|
||
|
|
ansR := httpclient.GetClient(time.Second*60).Post("https://api.ppinfra.com/v3/openai/chat/completions", map[string]any{
|
||
|
|
"model": "deepseek/deepseek-r1/community",
|
||
|
|
"messages": []map[string]any{
|
||
|
|
{
|
||
|
|
"role": "user",
|
||
|
|
"content": text,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
"response_format": map[string]any{"type": "text"},
|
||
|
|
}, "Content-Type", "application/json", "Authorization", "Bearer sk_siUuVF2utu9Mg0-LM5aM1KteXNQand33myY2iTq7Pcw")
|
||
|
|
if err := ansR.Error; err != nil {
|
||
|
|
log.DefaultLogger.Error("ask error", "err", err)
|
||
|
|
}
|
||
|
|
ans := Ans{}
|
||
|
|
ansR.To(&ans)
|
||
|
|
log.DefaultLogger.Info("ask stopped", "ans", ans)
|
||
|
|
return ans.Choices[0].Message.Content
|
||
|
|
}
|