Skip to content

Commit ec9d7a6

Browse files
committed
fix: fix context window size code
1 parent 6c0ad49 commit ec9d7a6

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

model/context_length_util.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,10 +59,12 @@ func getContextLength(typ string) int {
5959
return 32768
6060
}
6161
return 65536
62+
} else if strings.Contains(typ, "v4") {
63+
return 1000000
6264
} else if strings.Contains(typ, "v2.5") {
6365
return 8192
6466
} else if strings.Contains(typ, "v3") || strings.Contains(typ, "chat") || strings.Contains(typ, "reasoner") {
65-
return 65536
67+
return 1000000
6668
}
6769
} else if strings.Contains(typ, "qwen") {
6870
if strings.Contains(typ, "long") || strings.Contains(typ, "turbo") {

0 commit comments

Comments
 (0)