async-ollama 0.5.1 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0a1e184fa4a7d73adb1dda49cf25837a4190a1774a92b32289845d00bfe45c6f
-  data.tar.gz: 10de9be19b93771d181630058916310746be94155807375e8887d970da040566
+  metadata.gz: ea205e7dc0418f74e20575287b89c71e46e3a92322352043bea8ebb93134f629
+  data.tar.gz: e0254ec2f71b7e2539eee9315d0134716cd1a00494324ad3727d2f7b81129145
 SHA512:
-  metadata.gz: 1904a23e98d954b45803b8615de72eb87faaf91c45165a4a82679b75f99d9589a9db0b095f3a2a2ea8c4eec1b1a3447c84a2c80afa9f34be967155d9c4e71179
-  data.tar.gz: b82a1dfa1b2f1f51efaea9345065749c68546383f30ae026b27086981c2c7c4fb2e2494124bda2b0aa701ce3fb5a6a2e55e0ada92cc1c1c18f073a659108d38d
+  metadata.gz: 50c4ac988c537de057a6b87cf8183f40572f9fb9b3136686e75604643c4aa70e5fac398e3492053d2f10e2229d205f6ec9992fac15ac30bcc8716c0a3d429ce2
+  data.tar.gz: fc74eb93e1d69f1378571c47a95d9aafe53b1aa888da1091e293e915cbcfe48a138f434e00bbaf8941b8c4cb93d5ae057639b447bf6368adfe7bb5db781e7370

data/bake/async/ollama.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2024, by Samuel Williams.
+
+def initialize(...)
+	super
+	
+	require "async/ollama/client"
+end
+
+def models
+	Async::Ollama::Client.open do |client|
+		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

data/lib/async/ollama/chat.rb

@@ -15,6 +15,7 @@ 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]

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.1"
+		VERSION = "0.7.0"
 	end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: async-ollama
 version: !ruby/object:Gem::Version
-  version: 0.5.1
+  version: 0.7.0
 platform: ruby
 authors:
 - Samuel Williams
@@ -70,12 +70,14 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- bake/async/ollama.rb
 - lib/async/ollama.rb
 - lib/async/ollama/chat.rb
 - lib/async/ollama/client.rb
 - 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