Puller interface implementation for Ollama provider

Part of #283
symflower · Jul 23, 2024 · 8502bd9 · 8502bd9
1 parent b8e0c34
commit 8502bd9
Showing 1 changed file with 5 additions and 0 deletions.
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))
+}