|
@@ -2,6 +2,7 @@ package server
|
|
|
|
|
|
import (
|
|
|
"bytes"
|
|
|
+ "cmp"
|
|
|
"encoding/json"
|
|
|
"fmt"
|
|
|
"io"
|
|
@@ -53,6 +54,8 @@ func (t *responseRecorder) CloseNotify() <-chan bool {
|
|
|
|
|
|
func createRequest(t *testing.T, fn func(*gin.Context), body any) *httptest.ResponseRecorder {
|
|
|
t.Helper()
|
|
|
+ // if OLLAMA_MODELS is not set, set it to the temp directory
|
|
|
+ t.Setenv("OLLAMA_MODELS", cmp.Or(os.Getenv("OLLAMA_MODELS"), t.TempDir()))
|
|
|
|
|
|
w := NewRecorder()
|
|
|
c, _ := gin.CreateTestContext(w)
|