Puller interface implementation for Ollama provider

Part of #283
symflower · Jul 25, 2024 · e31f76e · e31f76e
1 parent 98aeb3b
commit e31f76e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/provider/ollama/ollama.go b/provider/ollama/ollama.go
@@ -119,3 +119,8 @@ func (p *Provider) Load(modelIdentifier string) error {
 func (p *Provider) Unload(modelIdentifier string) error {
 	return tools.OllamaUnload(p.url, strings.TrimPrefix(modelIdentifier, p.ID()+provider.ProviderModelSeparator))
 }
+
+// Pull downloads the given model.
+func (p *Provider) Pull(logger *log.Logger, modelIdentifier string) error {
+	return tools.OllamaPull(logger, p.binaryPath, p.url, strings.TrimPrefix(modelIdentifier, p.ID()+provider.ProviderModelSeparator))
+}
diff --git a/provider/provider.go b/provider/provider.go
@@ -65,5 +65,5 @@ type Loader interface {
 // Puller is a provider that is capable of pulling models.
 type Puller interface {
 	// Pull downloads the given model.
-	Pull(logger *log.Logger, modelIdentifier string) error
+	Pull(logger *log.Logger, modelIdentifier string) (err error)
 }