highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-08-29 12:44:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

tests: add integration coverage for oss-gpt (#11696)

Browse Source 
Also wires up support to override the default "smol" model
This commit is contained in:
Daniel Hiltgen
2025-08-07 15:06:57 -07:00
committed by GitHub
parent f2e9c9aff5
commit 114c3f2265
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
integration/utils_test.go
View File

@@ -28,7 +28,7 @@ import (
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
) )
const ( var (
smol = "llama3.2:1b" smol = "llama3.2:1b"
) )
@@ -37,6 +37,7 @@ var (
// Note: add newer models at the top of the list to test them first // Note: add newer models at the top of the list to test them first
ollamaEngineChatModels = []string{ ollamaEngineChatModels = []string{
"gpt-oss:20b",
"gemma3n:e2b", "gemma3n:e2b",
"mistral-small3.2:latest", "mistral-small3.2:latest",
"deepseek-r1:1.5b", "deepseek-r1:1.5b",
@@ -126,6 +127,7 @@ var (
"gemma3n", "gemma3n",
"glm4", "glm4",
"goliath", "goliath",
"gpt-oss:20b",
"granite-code", "granite-code",
"granite3-dense", "granite3-dense",
"granite3-guardian", "granite3-guardian",
@@ -255,8 +257,13 @@ var (
} }
) )
func Init() { func init() {
lifecycle.InitLogging() lifecycle.InitLogging()
custom := os.Getenv("OLLAMA_TEST_SMOL_MODEL")
if custom != "" {
slog.Info("setting smol test model to " + custom)
smol = custom
}
} }
func FindPort() string { func FindPort() string {