mirror of
https://github.com/danielmiessler/fabric.git
synced 2026-09-10 07:36:44 -04:00
fix: enable raw mode for GPT-6 models
This commit is contained in:
parent
6b0914d1bb
commit
8db08ad20f
|
|
@ -223,6 +223,7 @@ func (o *Client) NeedsRawMode(modelName string) bool {
|
|||
openaiModelsPrefixes := []string{
|
||||
"glm",
|
||||
"gpt-5",
|
||||
"gpt-6",
|
||||
"o1",
|
||||
"o3",
|
||||
"o4",
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue