瀏覽代碼

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) {
 func TestParseFromFileFromLayer(t *testing.T) {
 	tempModels := t.TempDir()
 	tempModels := t.TempDir()
+	t.Setenv("OLLAMA_MODELS", tempModels)
 
 
 	file, err := os.CreateTemp(tempModels, "")
 	file, err := os.CreateTemp(tempModels, "")
 	if err != nil {
 	if err != nil {
@@ -189,6 +190,7 @@ func TestParseFromFileFromLayer(t *testing.T) {
 
 
 func TestParseLayerFromCopy(t *testing.T) {
 func TestParseLayerFromCopy(t *testing.T) {
 	tempModels := t.TempDir()
 	tempModels := t.TempDir()
+	t.Setenv("OLLAMA_MODELS", tempModels)
 
 
 	file2, err := os.CreateTemp(tempModels, "")
 	file2, err := os.CreateTemp(tempModels, "")
 	if err != nil {
 	if err != nil {