async-ollama 0.6.0 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3515f4f2edf781476a1e1f99287a04d9772fa67e8db6993bc0956727df55be08
-  data.tar.gz: e3ee2b84a8228d61154f24115679b9d9f18a8bfb79057c30cd0e3627fcc72bbc
+  metadata.gz: cf57f6eea0f76d479dd4775772fe20d43a85d70b3e06869ae842923fb9e7bb7f
+  data.tar.gz: 18d439e6c5d24af612d67deaf2221b83485b401c8a39f2e49fe96c4540fe0829
 SHA512:
-  metadata.gz: ca26d093fd6dd8a406937e5bb87eb2fe3fb248600bebe89616b2699be5a405c9d3531f2e81c651477e928c685f9c6fcb4372970af668dd98360810fa9bcd0419
-  data.tar.gz: 710ee1a718bf0002f7f7b846a967626f708572f353b86ccca58ca3592ab76783fae588a7223902ce5bcf62edf31ff8f5e096bc7a811e641c95f09e2d1e6aa575
+  metadata.gz: 9742560a8fc515088f6a9e15008892e001d84b0fb439af817a2a78885a9cfed4ef969ac73801e942e2bf6e830b2c844207f80a4a1af2889218207f0aa516ebf2
+  data.tar.gz: 8a9d307856bfdf9bf883222a036f505e2e43f1dda370b7231ad84ffd2d5e2f1db8e03c3ddf488198b295e2254a1979b712fe966d9e82774775ebe11c5e4e149f

data/bake/async/ollama.rb

@@ -0,0 +1,30 @@
+# 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: [Async::Ollama::MODEL])
+	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/generate.rb

@@ -9,7 +9,7 @@ require_relative "wrapper"
 module Async
 	module Ollama
 		# Represents a generated response from the Ollama API.
-		class Generate < Async::REST::Representation[Wrapper]
+		class Generate < Async::REST::Representation[GenerateWrapper]
 			# @returns [String | nil] The generated response, or nil if not present.
 			def response
 				self.value[:response]

data/lib/async/ollama/version.rb

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

data/lib/async/ollama/wrapper.rb

@@ -139,6 +139,24 @@ module Async
 				content_type = response.headers["content-type"]
 				media_type = content_type.split(";").first
 				
+				case media_type
+				when APPLICATION_JSON
+					return Async::REST::Wrapper::JSON::Parser
+				when APPLICATION_JSON_STREAM
+					return StreamingParser
+				end
+			end
+		end
+		
+		# Wraps generate-specific HTTP responses for the Ollama API, selecting the appropriate parser.
+		class GenerateWrapper < Wrapper
+			# Selects the appropriate parser for the generate HTTP response.
+			# @parameter response [Protocol::HTTP::Response] The HTTP response object.
+			# @returns [Class] The parser class to use.
+			def parser_for(response)
+				content_type = response.headers["content-type"]
+				media_type = content_type.split(";").first
+				
 				case media_type
 				when APPLICATION_JSON
 					return Async::REST::Wrapper::JSON::Parser

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: async-ollama
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.8.0
 platform: ruby
 authors:
 - Samuel Williams
@@ -70,6 +70,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- bake/async/ollama.rb
 - lib/async/ollama.rb
 - lib/async/ollama/chat.rb
 - lib/async/ollama/client.rb