add OLLAMA_MODELS to envconfig (#5029)

This commit is contained in:
Patrick Devine
2024-06-13 15:52:03 -04:00
committed by GitHub
parent 1fd236d177
commit 94618b2365
8 changed files with 47 additions and 11 deletions

View File

@@ -7,6 +7,8 @@ import (
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/ollama/ollama/envconfig"
)
func TestGetBlobsPath(t *testing.T) {
@@ -61,6 +63,7 @@ func TestGetBlobsPath(t *testing.T) {
for _, tc := range tests {
t.Run(tc.name, func(t *testing.T) {
t.Setenv("OLLAMA_MODELS", dir)
envconfig.LoadConfig()
got, err := GetBlobsPath(tc.digest)