Skip to content

Commit 205aba6

Browse files
authored
chore: limit the temperature and top_p parameter value range to (0.0, 1) for zhipu (songquanpeng#1091)
1 parent 8dac3af commit 205aba6

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

relay/channel/zhipu/adaptor.go

+8-3
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ import (
99
"github.com/songquanpeng/one-api/relay/model"
1010
"github.com/songquanpeng/one-api/relay/util"
1111
"io"
12+
"math"
1213
"net/http"
1314
"strings"
1415
)
@@ -52,9 +53,13 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.G
5253
if request == nil {
5354
return nil, errors.New("request is nil")
5455
}
55-
if request.TopP >= 1 {
56-
request.TopP = 0.99
57-
}
56+
// TopP (0.0, 1.0)
57+
request.TopP = math.Min(0.99, request.TopP)
58+
request.TopP = math.Max(0.01, request.TopP)
59+
60+
// Temperature (0.0, 1.0)
61+
request.Temperature = math.Min(0.99, request.Temperature)
62+
request.Temperature = math.Max(0.01, request.Temperature)
5863
a.SetVersionByModeName(request.Model)
5964
if a.APIVersion == "v4" {
6065
return request, nil

0 commit comments

Comments
 (0)