feat: save prompt in power log for dalle-3

This commit is contained in:
RockYang
2024-03-21 15:55:39 +08:00
parent ec6186596d
commit bf5e72b7e0
3 changed files with 18 additions and 10 deletions

View File

@@ -4,7 +4,7 @@ import (
"encoding/json"
"fmt"
"math/rand"
"strconv"
"strings"
"time"
"golang.org/x/crypto/sha3"
@@ -94,10 +94,14 @@ func InterfaceToString(value interface{}) string {
return JsonEncode(value)
}
func Str2Float(str string) float64 {
num, err := strconv.ParseFloat(str, 64)
if err != nil {
return 0
func CutWords(str string, num int) string {
// 按空格分割字符串为单词切片
words := strings.Fields(str)
// 如果单词数量超过指定数量,则裁剪单词;否则保持原样
if len(words) > num {
return strings.Join(words[:num], " ") + " ..."
} else {
return str
}
return num
}