From e4d0a9c325137b684d9c0bb02694e8f10197314a Mon Sep 17 00:00:00 2001
From: Michael Yang <mxyng@pm.me>
Date: Wed, 28 Aug 2024 14:07:48 -0700
Subject: [PATCH] fix(test): do not clobber models directory

---
 server/model_test.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/server/model_test.go b/server/model_test.go
index 7753c549..e1737a5b 100644
--- a/server/model_test.go
+++ b/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 {