Преглед на файлове

Merge pull request #6546 from ollama/mxyng/fix-test

fix(test): do not clobber models directory
Michael Yang преди 8 месеца
родител
ревизия
47c2b947a9
променени са 1 файла, в които са добавени 2 реда и са изтрити 0 реда
  1. 2 0
      server/model_test.go

+ 2 - 0
server/model_test.go

@@ -139,6 +139,7 @@ The temperature in San Francisco, CA is 70°F and in Toronto, Canada is 20°C.`,
 
 func TestParseFromFileFromLayer(t *testing.T) {
 	tempModels := t.TempDir()
+	t.Setenv("OLLAMA_MODELS", tempModels)
 
 	file, err := os.CreateTemp(tempModels, "")
 	if err != nil {
@@ -189,6 +190,7 @@ func TestParseFromFileFromLayer(t *testing.T) {
 
 func TestParseLayerFromCopy(t *testing.T) {
 	tempModels := t.TempDir()
+	t.Setenv("OLLAMA_MODELS", tempModels)
 
 	file2, err := os.CreateTemp(tempModels, "")
 	if err != nil {