async-ollama 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3365078c11cba990bad7d5c3b8d7edaac51bd1efd99bb2e73dd78d1fddf0778f
-  data.tar.gz: bdfb803fcd92ae54a6382b75e3008026939903eec083667bd2f1d93dd1b57504
+  metadata.gz: b063f95dd3bfc375203e52136932e15e7fdb060c5e8c2c88714285679d4ed535
+  data.tar.gz: a431d19a4817acf9d77ab225f22dfe523d0e04826f4ad87e615bd7c6d22d20f9
 SHA512:
-  metadata.gz: f523362fc9d63392e8fe0cd2704a04e768e3cbcc115af561bc500b6a9604d4aee02765d0625b91fba2bdede669f38e66aff70dbd1489fcd72389912d5a2cf252
-  data.tar.gz: 1fead04badd7785a75e54e5faa9e007cd132d8f6764588f8d3cf5cbc2e1fbbbfa475012bc546f83ae135ea2104745765337f143395243a459c769b89a2a41819
+  metadata.gz: ac83698f0c120ede1b595509da72e83ab81faffe50bed02e3aebf01e5b66579699d2329fbdaff0e0f087813677fc6f338029bccda9d62d0ed6726c26586e9973
+  data.tar.gz: 148408b29f5aafd3ce3d71c5ecd3f2eadbd496736893049e01701e52e214188912725a3ba13245b261d5c02e159725a29998493e7739aed6f8e0e0a08db0ad4b

data/lib/async/ollama/client.rb

@@ -4,7 +4,9 @@
 # Copyright, 2024, by Samuel Williams.
 
 require "async/rest/resource"
+
 require_relative "generate"
+require_relative "models"
 
 module Async
 	module Ollama
@@ -27,6 +29,10 @@ module Async
 					Generate.new(resource, value: response.read, metadata: response.headers)
 				end
 			end
+			
+			def models
+				Models.get(self.with(path: "/api/tags"))
+			end
 		end
 	end
 end

data/lib/async/ollama/models.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2024, by Samuel Williams.
+
+require "async/rest/representation"
+require_relative "wrapper"
+
+module Async
+	module Ollama
+		class Models < Async::REST::Representation[Wrapper]
+			def names
+				self.value[:models].map{|model| model[:name]}
+			end
+		end
+	end
+end

data/lib/async/ollama/version.rb

@@ -5,6 +5,6 @@
 
 module Async
 	module Ollama
-		VERSION = "0.3.0"
+		VERSION = "0.4.0"
 	end
 end

data/lib/async/ollama/wrapper.rb

@@ -85,7 +85,10 @@ module Async
 			end
 			
 			def parser_for(response)
-				case response.headers["content-type"]
+				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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: async-ollama
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Samuel Williams
@@ -37,7 +37,7 @@ cert_chain:
   Q2K9NVun/S785AP05vKkXZEFYxqG6EW012U4oLcFl5MySFajYXRYbuUpH6AY+HP8
   voD0MPg1DssDLKwXyt1eKD/+Fq0bFWhwVM/1XiAXL7lyYUyOq24KHgQ2Csg=
   -----END CERTIFICATE-----
-date: 2024-09-12 00:00:00.000000000 Z
+date: 2024-11-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: async
@@ -76,6 +76,7 @@ files:
 - lib/async/ollama.rb
 - lib/async/ollama/client.rb
 - lib/async/ollama/generate.rb
+- lib/async/ollama/models.rb
 - lib/async/ollama/version.rb
 - lib/async/ollama/wrapper.rb
 - license.md
@@ -101,7 +102,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.11
+rubygems_version: 3.5.22
 signing_key:
 specification_version: 4
 summary: A asynchronous interface to the ollama chat service