Expose models.

bake/async/ollama.rb

@@ -0,0 +1,11 @@
+def initialize(...)
+	super
+
+	require 'async/ollama/client'
+end
+
+def models
+	Async::Ollama::Client.open do |client|
+		client.models.names
+	end
+end

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 @@ def generate(prompt, **options, &block)
 					Generate.new(resource, value: response.read, metadata: response.headers)
 				end
 			end
+
+			def models
+				Models.get(self.with(path: "/api/tags"))
+			end
 		end
 	end
 end

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

lib/async/ollama/wrapper.rb

@@ -85,7 +85,10 @@ def join
 			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