Unverified Commit 114c3f22 authored by Daniel Hiltgen's avatar Daniel Hiltgen Committed by GitHub
Browse files

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

Also wires up support to override the default "smol" model
parent f2e9c9af
......@@ -28,7 +28,7 @@ import (
"github.com/stretchr/testify/require"
)
const (
var (
smol = "llama3.2:1b"
)
......@@ -37,6 +37,7 @@ var (
// Note: add newer models at the top of the list to test them first
ollamaEngineChatModels = []string{
"gpt-oss:20b",
"gemma3n:e2b",
"mistral-small3.2:latest",
"deepseek-r1:1.5b",
......@@ -126,6 +127,7 @@ var (
"gemma3n",
"glm4",
"goliath",
"gpt-oss:20b",
"granite-code",
"granite3-dense",
"granite3-guardian",
......@@ -255,8 +257,13 @@ var (
}
)
func Init() {
func init() {
lifecycle.InitLogging()
custom := os.Getenv("OLLAMA_TEST_SMOL_MODEL")
if custom != "" {
slog.Info("setting smol test model to " + custom)
smol = custom
}
}
func FindPort() string {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment