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{
|
openaiModelsPrefixes := []string{
|
||||||
"glm",
|
"glm",
|
||||||
"gpt-5",
|
"gpt-5",
|
||||||
|
"gpt-6",
|
||||||
"o1",
|
"o1",
|
||||||
"o3",
|
"o3",
|
||||||
"o4",
|
"o4",
|
||||||
|
|
|
||||||
|
|
@ -12,6 +12,12 @@ import (
|
||||||
"github.com/stretchr/testify/assert"
|
"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) {
|
func TestBuildResponseRequestWithMaxTokens(t *testing.T) {
|
||||||
|
|
||||||
var msgs []*chat.ChatCompletionMessage
|
var msgs []*chat.ChatCompletionMessage
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue