fix: enable raw mode for GPT-6 models

This commit is contained in:
ctbaum 2026-09-06 12:57:33 +02:00
parent 6b0914d1bb
commit 8db08ad20f
2 changed files with 7 additions and 0 deletions

View file

@ -223,6 +223,7 @@ func (o *Client) NeedsRawMode(modelName string) bool {
openaiModelsPrefixes := []string{
"glm",
"gpt-5",
"gpt-6",
"o1",
"o3",
"o4",

View file

@ -12,6 +12,12 @@ import (
"github.com/stretchr/testify/assert"
)
func TestNeedsRawModeGPT6(t *testing.T) {
if !NewClient().NeedsRawMode("gpt-6-astra") {
t.Fatal("gpt-6 models should use raw mode")
}
}
func TestBuildResponseRequestWithMaxTokens(t *testing.T) {
var msgs []*chat.ChatCompletionMessage