async-ollama 0.7.0 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ea205e7dc0418f74e20575287b89c71e46e3a92322352043bea8ebb93134f629
-  data.tar.gz: e0254ec2f71b7e2539eee9315d0134716cd1a00494324ad3727d2f7b81129145
+  metadata.gz: cf57f6eea0f76d479dd4775772fe20d43a85d70b3e06869ae842923fb9e7bb7f
+  data.tar.gz: 18d439e6c5d24af612d67deaf2221b83485b401c8a39f2e49fe96c4540fe0829
 SHA512:
-  metadata.gz: 50c4ac988c537de057a6b87cf8183f40572f9fb9b3136686e75604643c4aa70e5fac398e3492053d2f10e2229d205f6ec9992fac15ac30bcc8716c0a3d429ce2
-  data.tar.gz: fc74eb93e1d69f1378571c47a95d9aafe53b1aa888da1091e293e915cbcfe48a138f434e00bbaf8941b8c4cb93d5ae057639b447bf6368adfe7bb5db781e7370
+  metadata.gz: 9742560a8fc515088f6a9e15008892e001d84b0fb439af817a2a78885a9cfed4ef969ac73801e942e2bf6e830b2c844207f80a4a1af2889218207f0aa516ebf2
+  data.tar.gz: 8a9d307856bfdf9bf883222a036f505e2e43f1dda370b7231ad84ffd2d5e2f1db8e03c3ddf488198b295e2254a1979b712fe966d9e82774775ebe11c5e4e149f

data/bake/async/ollama.rb

@@ -17,11 +17,7 @@ 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
-	
+def pull(models: [Async::Ollama::MODEL])
 	Async::Ollama::Client.open do |client|
 		models.each do |model|
 			client.pull(model) do |response|

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.7.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

data.tar.gz.sig

@@ -1,2 +1,2 @@
-��)'�;sU���\"X
-{_Q2�q�?�##9F`t��+j E�����k����݅ �!=xMP��Z!gK{xO/I��4,�5ԁ}=8�Ѻ&�����F~B�|� ��z�����Y�r�>b�nR�u#�'>�ܦ4u�D����u��*��m�m��휊�*̪A.���6c&��V�q��W�#��
1���1��������L�4�؀lX�ǎ���������c儆�x�ZJh訙�	8%�Y,z굄QÔ^�=M��p�X�h���Y����QQTG����Y����>��jV6�ե��P�7E�Zn��5�%?�E�m�K8Udџd8>�u}������K���진�6�5�7"}�IT�&8��1��$�
+
���u2�Ɲ�H����\�?��ˑ���D
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: async-ollama
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.8.0
 platform: ruby
 authors:
 - Samuel Williams