diff --git a/server/images.go b/server/images.go index 63ab2a68..cc07bd0f 100644 --- a/server/images.go +++ b/server/images.go @@ -502,6 +502,12 @@ func embeddingLayers(e EmbeddingParams) ([]*LayerReader, error) { } }() + // this will be used to check if we already have embeddings for a file + modelDigest, err := modelDigest(e.model) + if err != nil { + return nil, fmt.Errorf("model digest: %w", err) + } + addedFiles := make(map[string]bool) // keep track of files that have already been added for _, filePattern := range e.files { matchingFiles, err := filepath.Glob(filePattern) @@ -515,7 +521,7 @@ func embeddingLayers(e EmbeddingParams) ([]*LayerReader, error) { } addedFiles[filePath] = true // check if we already have embeddings for this file path - layerIdentifier := fmt.Sprintf("%s:%s", filePath, e.model) + layerIdentifier := fmt.Sprintf("%s:%s", filePath, modelDigest) digest, _ := GetSHA256Digest(strings.NewReader(layerIdentifier)) existing, err := existingFileEmbeddings(digest) if err != nil { @@ -585,6 +591,17 @@ func embeddingLayers(e EmbeddingParams) ([]*LayerReader, error) { return layers, nil } +func modelDigest(modelPath string) (string, error) { + modelFile, err := os.Open(modelPath) + if err != nil { + return "", fmt.Errorf("could not open model blob: %w", err) + } + defer modelFile.Close() + + digest, _ := GetSHA256Digest(modelFile) + return digest, nil +} + // existingFileEmbeddings checks if we already have embeddings for a file and loads them into a look-up map func existingFileEmbeddings(digest string) (map[string][]float64, error) { path, err := GetBlobsPath(digest)