Add bake task for pulling a model.

Author: ioquatix

bake/async/ollama.rb

@@ -14,3 +14,21 @@ def models
 		client.models.names
 	end
 end
+
+# Pulls the specified models from the Ollama API. If no models are specified, but there is a default model, it will pull that one.
+# @parameter models [Array(String)] The names of the models to pull.
+def pull(models)
+	if models.empty?
+		models = [Async::Ollama::Client.default_model]
+	end
+	
+	Async::Ollama::Client.open do |client|
+		models.each do |model|
+			client.pull(model) do |response|
+				response.each do |line|
+					puts line
+				end
+			end
+		end
+	end
+end

lib/async/ollama/client.rb

@@ -8,6 +8,7 @@
 require_relative "generate"
 require_relative "chat"
 require_relative "models"
+require_relative "pull"
 
 module Async
 	module Ollama
@@ -57,6 +58,16 @@ def chat(messages, **options, &block)
 			def models
 				Models.get(self.with(path: "/api/tags"))
 			end
+			
+			def pull(model)
+				Pull.post(self.with(path: "/api/pull"), model: model) do |resource, response|
+					if block_given?
+						yield response
+					end
+					
+					Pull.new(resource, value: response.read, metadata: response.headers)
+				end
+			end
 		end
 	end
 end

lib/async/ollama/pull.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2024-2025, by Samuel Williams.
+
+require "async/rest/representation"
+require_relative "wrapper"
+
+module Async
+	module Ollama
+		# Represents the response from pulling models from the Ollama API.
+		class Pull < Async::REST::Representation[Wrapper]
+		end
+	end
+end