starred/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2026-04-18 01:54:17 +02:00
Code Issues Packages Projects Releases Wiki Activity

cmd/config: fix cloud model limit lookups in integrations (#14650)

Browse Source
This commit is contained in:
Parth Sareen
2026-03-05 13:57:28 -08:00
committed by GitHub
parent 15732f0ea7
commit 9896e3627f
6 changed files with 79 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cmd/config/droid_test.go
View File

@@ -1277,7 +1277,8 @@ func TestDroidEdit_LocalModelDefaultMaxOutput(t *testing.T) {
func TestDroidEdit_CloudModelLimitsUsed(t *testing.T) {
// Verify that every cloud model in cloudModelLimits has a valid output
// value that would be used for maxOutputTokens when isCloudModel returns true.
// :cloud suffix stripping must also work since that's how users specify them.
// Cloud suffix normalization must also work since integrations may see either
// :cloud or -cloud model names.
for name, expected := range cloudModelLimits {
t.Run(name, func(t *testing.T) {
l, ok := lookupCloudModelLimit(name)
@@ -1296,6 +1297,15 @@ func TestDroidEdit_CloudModelLimitsUsed(t *testing.T) {
if l2.Output != expected.Output {
t.Errorf(":cloud output = %d, want %d", l2.Output, expected.Output)
}
// Also verify -cloud suffix lookup
dashCloudName := name + "-cloud"
l3, ok := lookupCloudModelLimit(dashCloudName)
if !ok {
t.Fatalf("lookupCloudModelLimit(%q) returned false", dashCloudName)
}
if l3.Output != expected.Output {
t.Errorf("-cloud output = %d, want %d", l3.Output, expected.Output)
}
})
}
}