async-ollama 0.5.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6ccd3b807e6fb1863176fe45a44a41176cc122d72c9650ef0e96eb748151c2a9
-  data.tar.gz: 28a590bb9f9a62e12419a6c5e6c1c6b1661cee0a0b1068a795c3ebcfb9422282
+  metadata.gz: 3515f4f2edf781476a1e1f99287a04d9772fa67e8db6993bc0956727df55be08
+  data.tar.gz: e3ee2b84a8228d61154f24115679b9d9f18a8bfb79057c30cd0e3627fcc72bbc
 SHA512:
-  metadata.gz: 68119787b00471de152fb5d99a3d94c6a86c20216925cb36cb8ed9c88bb694861d0b293dee7594ad411b65894a03ecf2e18c775708db9465c4341008fee3615e
-  data.tar.gz: 1dca4d6db7fd5cc6136f6a65ef5b61777ed6c720d9d7ec445d7d7443c1a4d6769c552939a02cc4fa306de93038c0cfff6ef7b328061130716c5280cfbb836fef
+  metadata.gz: ca26d093fd6dd8a406937e5bb87eb2fe3fb248600bebe89616b2699be5a405c9d3531f2e81c651477e928c685f9c6fcb4372970af668dd98360810fa9bcd0419
+  data.tar.gz: 710ee1a718bf0002f7f7b846a967626f708572f353b86ccca58ca3592ab76783fae588a7223902ce5bcf62edf31ff8f5e096bc7a811e641c95f09e2d1e6aa575

data/lib/async/ollama/chat.rb

@@ -15,6 +15,13 @@ module Async
 				self.value[:message]
 			end
 			
+			# @returns [String | nil] The content of the message, or nil if not present.
+			def response
+				if message = self.message
+					message[:content]
+				end
+			end
+			
 			# @returns [String | nil] The error message, or nil if not present.
 			def error
 				self.value[:error]

data/lib/async/ollama/client.rb

@@ -8,6 +8,7 @@ require "async/rest/resource"
 require_relative "generate"
 require_relative "chat"
 require_relative "models"
+require_relative "pull"
 
 module Async
 	module Ollama
@@ -57,6 +58,16 @@ module Async
 			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

data/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

data/lib/async/ollama/version.rb

@@ -5,6 +5,6 @@
 
 module Async
 	module Ollama
-		VERSION = "0.5.0"
+		VERSION = "0.6.0"
 	end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: async-ollama
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Samuel Williams
@@ -76,6 +76,7 @@ files:
 - lib/async/ollama/conversation.rb
 - lib/async/ollama/generate.rb
 - lib/async/ollama/models.rb
+- lib/async/ollama/pull.rb
 - lib/async/ollama/toolbox.rb
 - lib/async/ollama/version.rb
 - lib/async/ollama/wrapper.rb