mirror of
https://github.com/ollama/ollama.git
synced 2025-05-11 18:36:41 +02:00
2412adf42b
This commit introduces a new API implementation for handling interactions with the registry and the local model cache. The new API is located in server/internal/registry. The package name is "registry" and should be considered temporary; it is hidden and not bleeding outside of the server package. As the commits roll in, we'll start consuming more of the API and then let reverse osmosis take effect, at which point it will surface closer to the root level packages as much as needed.
93 lines
2.1 KiB
Go
93 lines
2.1 KiB
Go
package blob
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func isCaseSensitive(dir string) bool {
|
|
defer func() {
|
|
os.Remove(filepath.Join(dir, "_casecheck"))
|
|
}()
|
|
|
|
exists := func(file string) bool {
|
|
_, err := os.Stat(file)
|
|
return err == nil
|
|
}
|
|
|
|
file := filepath.Join(dir, "_casecheck")
|
|
FILE := filepath.Join(dir, "_CASECHECK")
|
|
if exists(file) || exists(FILE) {
|
|
panic(fmt.Sprintf("_casecheck already exists in %q; remove and try again.", dir))
|
|
}
|
|
|
|
err := os.WriteFile(file, nil, 0o666)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return !exists(FILE)
|
|
}
|
|
|
|
func isCI() bool {
|
|
return os.Getenv("CI") != ""
|
|
}
|
|
|
|
const volumeHint = `
|
|
|
|
Unable to locate case-insensitive TMPDIR on darwin.
|
|
|
|
To run tests, create the case-insensitive volume /Volumes/data:
|
|
|
|
$ sudo diskutil apfs addVolume disk1 APFSX data -mountpoint /Volumes/data
|
|
|
|
or run with:
|
|
|
|
CI=1 go test ./...
|
|
|
|
`
|
|
|
|
// useCaseInsensitiveTempDir sets TMPDIR to a case-insensitive directory
|
|
// can find one, otherwise it skips the test if the CI environment variable is
|
|
// set, or GOOS is not darwin.
|
|
func useCaseInsensitiveTempDir(t *testing.T) bool {
|
|
if isCaseSensitive(os.TempDir()) {
|
|
// Use the default temp dir if it is already case-sensitive.
|
|
return true
|
|
}
|
|
if runtime.GOOS == "darwin" {
|
|
// If darwin, check for the special case-sensitive volume and
|
|
// use it if available.
|
|
const volume = "/Volumes/data"
|
|
_, err := os.Stat(volume)
|
|
if err == nil {
|
|
tmpdir := filepath.Join(volume, "tmp")
|
|
os.MkdirAll(tmpdir, 0o700)
|
|
t.Setenv("TMPDIR", tmpdir)
|
|
return true
|
|
}
|
|
if isCI() {
|
|
// Special case darwin in CI; it is not case-sensitive
|
|
// by default, and we will be testing other platforms
|
|
// that are case-sensitive, so we'll have the test
|
|
// being skipped covered there.
|
|
t.Skip("Skipping test in CI for darwin; TMPDIR is not case-insensitive.")
|
|
}
|
|
}
|
|
|
|
if !isCI() {
|
|
// Require devs to always tests with a case-insensitive TMPDIR.
|
|
|
|
// TODO(bmizerany): Print platform-specific instructions or
|
|
// link to docs on that topic.
|
|
lines := strings.Split(volumeHint, "\n")
|
|
for _, line := range lines {
|
|
t.Skip(line)
|
|
}
|
|
}
|
|
return false
|
|
}
|