48 lines
1.6 KiB
Go
48 lines
1.6 KiB
Go
|
package openai
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func firstLine(s string) string {
|
|||
|
|
s = strings.TrimSpace(s)
|
|||
|
|
if i := strings.IndexByte(s, '\n'); i >= 0 {
|
|||
|
|
return s[:i]
|
|||
|
|
}
|
|||
|
|
return s
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CodexBaseInstructionsForModel 应按模型返回对应的真实 Codex base prompt。
|
|||
|
|
func TestCodexBaseInstructionsForModel(t *testing.T) {
|
|||
|
|
cases := []struct {
|
|||
|
|
model string
|
|||
|
|
wantHead string
|
|||
|
|
}{
|
|||
|
|
{"gpt-5-codex", "You are Codex, based on GPT-5"},
|
|||
|
|
{"gpt-5.3-codex", "You are Codex, based on GPT-5"},
|
|||
|
|
{"gpt-5.3-codex-spark", "You are Codex, based on GPT-5"},
|
|||
|
|
{"gpt-5.1-codex-max", "You are Codex, based on GPT-5"},
|
|||
|
|
{"gpt-5.2-codex", "You are Codex, based on GPT-5"},
|
|||
|
|
{"gpt-5.5", "You are Codex, a coding agent based on GPT-5"},
|
|||
|
|
{" GPT-5.5 ", "You are Codex, a coding agent based on GPT-5"},
|
|||
|
|
{"gpt-5.2", "You are GPT-5.2 running in the Codex CLI"},
|
|||
|
|
{"gpt-5.1", "You are GPT-5.1 running in the Codex CLI"},
|
|||
|
|
{"gpt-5", "You are Codex, a coding agent based on GPT-5"}, // 回退到最新（GPT-5.5）
|
|||
|
|
{"gpt-5.4", "You are Codex, a coding agent based on GPT-5"}, // 未单独维护 → 最新
|
|||
|
|
{"gpt-5.3", "You are Codex, a coding agent based on GPT-5"}, // 未单独维护 → 最新
|
|||
|
|
{"some-unknown-model", "You are Codex, a coding agent based on GPT-5"},
|
|||
|
|
{"", "You are Codex, a coding agent based on GPT-5"}, // 回退到最新
|
|||
|
|
}
|
|||
|
|
for _, c := range cases {
|
|||
|
|
got := strings.TrimSpace(CodexBaseInstructionsForModel(c.model))
|
|||
|
|
if got == "" {
|
|||
|
|
t.Errorf("model %q: got empty instructions", c.model)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if !strings.HasPrefix(got, c.wantHead) {
|
|||
|
|
t.Errorf("model %q: got prefix %q, want %q", c.model, firstLine(got), c.wantHead)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|