|
@@ -2,10 +2,8 @@ package server
|
|
|
|
|
|
import (
|
|
import (
|
|
"bytes"
|
|
"bytes"
|
|
- "context"
|
|
|
|
"encoding/json"
|
|
"encoding/json"
|
|
"fmt"
|
|
"fmt"
|
|
- "io"
|
|
|
|
"os"
|
|
"os"
|
|
"path/filepath"
|
|
"path/filepath"
|
|
"testing"
|
|
"testing"
|
|
@@ -13,7 +11,6 @@ import (
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/google/go-cmp/cmp"
|
|
|
|
|
|
"github.com/ollama/ollama/api"
|
|
"github.com/ollama/ollama/api"
|
|
- "github.com/ollama/ollama/llm"
|
|
|
|
"github.com/ollama/ollama/template"
|
|
"github.com/ollama/ollama/template"
|
|
)
|
|
)
|
|
|
|
|
|
@@ -136,82 +133,3 @@ The temperature in San Francisco, CA is 70°F and in Toronto, Canada is 20°C.`,
|
|
})
|
|
})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
-
|
|
|
|
-func TestParseFromFileFromLayer(t *testing.T) {
|
|
|
|
- tempModels := t.TempDir()
|
|
|
|
-
|
|
|
|
- file, err := os.CreateTemp(tempModels, "")
|
|
|
|
- if err != nil {
|
|
|
|
- t.Fatalf("failed to open file: %v", err)
|
|
|
|
- }
|
|
|
|
- defer file.Close()
|
|
|
|
- if err := llm.WriteGGUF(file, llm.KV{"general.architecture": "gemma"}, []llm.Tensor{}); err != nil {
|
|
|
|
- t.Fatalf("failed to write gguf: %v", err)
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if _, err := file.Seek(0, io.SeekStart); err != nil {
|
|
|
|
- t.Fatalf("failed to seek to start: %v", err)
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- layers, err := parseFromFile(context.Background(), file, "", func(api.ProgressResponse) {})
|
|
|
|
- if err != nil {
|
|
|
|
- t.Fatalf("failed to parse from file: %v", err)
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if len(layers) != 1 {
|
|
|
|
- t.Fatalf("got %d != want 1", len(layers))
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if _, err := file.Seek(0, io.SeekStart); err != nil {
|
|
|
|
- t.Fatalf("failed to seek to start: %v", err)
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- layers2, err := parseFromFile(context.Background(), file, layers[0].Digest, func(api.ProgressResponse) {})
|
|
|
|
- if err != nil {
|
|
|
|
- t.Fatalf("failed to parse from file: %v", err)
|
|
|
|
- }
|
|
|
|
- if len(layers2) != 1 {
|
|
|
|
- t.Fatalf("got %d != want 1", len(layers2))
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if layers[0].Digest != layers2[0].Digest {
|
|
|
|
- t.Fatalf("got %s != want %s", layers[0].Digest, layers2[0].Digest)
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if layers[0].Size != layers2[0].Size {
|
|
|
|
- t.Fatalf("got %d != want %d", layers[0].Size, layers2[0].Size)
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if layers[0].MediaType != layers2[0].MediaType {
|
|
|
|
- t.Fatalf("got %v != want %v", layers[0].MediaType, layers2[0].MediaType)
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-func TestParseLayerFromCopy(t *testing.T) {
|
|
|
|
- tempModels := t.TempDir()
|
|
|
|
-
|
|
|
|
- file2, err := os.CreateTemp(tempModels, "")
|
|
|
|
- if err != nil {
|
|
|
|
- t.Fatalf("failed to open file: %v", err)
|
|
|
|
- }
|
|
|
|
- defer file2.Close()
|
|
|
|
-
|
|
|
|
- for range 5 {
|
|
|
|
- if err := llm.WriteGGUF(file2, llm.KV{"general.architecture": "gemma"}, []llm.Tensor{}); err != nil {
|
|
|
|
- t.Fatalf("failed to write gguf: %v", err)
|
|
|
|
- }
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if _, err := file2.Seek(0, io.SeekStart); err != nil {
|
|
|
|
- t.Fatalf("failed to seek to start: %v", err)
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- layers, err := parseFromFile(context.Background(), file2, "", func(api.ProgressResponse) {})
|
|
|
|
- if err != nil {
|
|
|
|
- t.Fatalf("failed to parse from file: %v", err)
|
|
|
|
- }
|
|
|
|
-
|
|
|
|
- if len(layers) != 5 {
|
|
|
|
- t.Fatalf("got %d != want 5", len(layers))
|
|
|
|
- }
|
|
|
|
-}
|
|
|