adds tool_choice option

drnic · Apr 25, 2024 · e11e62b · e11e62b
1 parent 979601e
commit e11e62b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/groq/client.rb b/lib/groq/client.rb
@@ -23,7 +23,7 @@ def initialize(config = {}, &faraday_middleware)
   end
 
   # TODO: support stream: true; or &stream block
-  def chat(messages, model_id: nil, tools: nil, max_tokens: nil, temperature: nil, json: false)
+  def chat(messages, model_id: nil, tools: nil, tool_choice: nil, max_tokens: nil, temperature: nil, json: false)
     unless messages.is_a?(Array) || messages.is_a?(String)
       raise ArgumentError, "require messages to be an Array or String"
     end
@@ -38,6 +38,7 @@ def chat(messages, model_id: nil, tools: nil, max_tokens: nil, temperature: nil,
       model: model_id,
       messages: messages,
       tools: tools,
+      tool_choice: tool_choice,
       max_tokens: max_tokens || @max_tokens,
       temperature: temperature || @temperature,
       response_format: json ? {type: "json_object"} : nil