Add DEFAULT_MODEL constant.

socketry · Nov 26, 2024 · e4dbb96 · e4dbb96
1 parent 59ebaf3
commit e4dbb96
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/lib/async/ollama/client.rb b/lib/async/ollama/client.rb
@@ -19,7 +19,7 @@ class Client < Async::REST::Resource
 			# @parameter prompt [String] The prompt to generate a response from.
 			def generate(prompt, **options, &block)
 				options[:prompt] = prompt
-				options[:model] ||= "llama3"
+				options[:model] ||= DEFAULT_MODEL
 
 				Generate.post(self.with(path: "/api/generate"), options) do |resource, response|
 					if block_given?

diff --git a/lib/async/ollama/conversation.rb b/lib/async/ollama/conversation.rb
@@ -9,7 +9,7 @@
 module Async
 	module Ollama
 		class Conversation
-			def initialize(client, model: "llama3", context: nil)
+			def initialize(client, model: DEFAULT_MODEL, context: nil)
 				@client = client
 
 				@toolbox = Toolbox.new

diff --git a/lib/async/ollama/generate.rb b/lib/async/ollama/generate.rb
@@ -8,6 +8,8 @@
 
 module Async
 	module Ollama
+		DEFAULT_MODEL = "llama3"
+
 		class Generate < Async::REST::Representation[Wrapper]
 			# The response to the prompt.
 			def response