openaiext 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fa387e4a2018bbe15a88f28a24d55aefe1b69d722fe060abf4700f85fa2c1e1b
-  data.tar.gz: c2ae11b8a9287fdabfebde0e97f5d6c20ffc0fc91728bea6bdfcc0ed0ac7c1c1
+  metadata.gz: 41b05586226d777a40b9a92f3df343a8c4dabbf3308879b3dbf333707148ee06
+  data.tar.gz: 932e68b7d141a9082a703a1e848b620a40ea56005d510c843b619b370f54f1ea
 SHA512:
-  metadata.gz: 1d8bd813c1f19c6f3940216a800cf9647df431c198516bbbf31d17f2e784b55f28df1b7d33b92cbdae0271e75f194b623fffe0e2967a9c9f669b988690883bbb
-  data.tar.gz: 705d4a98839780cd4ae56f419c241c0e85645620c385634250d69685f861631950f12ba4bbf6f584510534b3b32965dc8adc663a0f77d30762fdb9a499e78674
+  metadata.gz: 2e7e8fb1032594f7eeaa01ecce877932bb661a890ec2388ba97488fdc36795f168a36a30e4fa207294cfbd3bcffb89fd85fe91a35da775a4f6fd035dffc4ea79
+  data.tar.gz: 16fb70150afa2ca8d8ec1616f487d247bea38a356ac3cb9e89923a032867b1bde70d565846edcdf0b282919ab607f1f028cedbba249754a40a13dd96e0981015

data/lib/openaiext.rb

@@ -19,15 +19,15 @@ module OpenAIExt
     chat(messages: [{ role: :user, content: message_content }], model:, response_format:, max_tokens:, store:, tools:, auto_run_functions:, function_context:)    
   end  
 
-  def self.single_prompt(prompt:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil, temperature: nil, top_p: nil, frequency_penalty: nil, presence_penalty: nil)
-    chat(messages: [{ user: prompt }], model:, response_format:, max_tokens:, store:, tools:, auto_run_functions:, function_context:, temperature:, top_p:, frequency_penalty:, presence_penalty:)
+  def self.single_prompt(prompt:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil, temperature: nil, top_p: nil, frequency_penalty: nil, presence_penalty: nil, prediction: nil)
+    chat(messages: [{ user: prompt }], model:, response_format:, max_tokens:, store:, tools:, auto_run_functions:, function_context:, temperature:, top_p:, frequency_penalty:, presence_penalty:, prediction:)
   end
 
-  def self.single_chat(system:, user:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil, temperature: nil, top_p: nil, frequency_penalty: nil, presence_penalty: nil)
-    chat(messages: [{ system: }, { user: }], model:, response_format:, max_tokens:, store:, tools:, auto_run_functions:, function_context:, temperature:, top_p:, frequency_penalty:, presence_penalty:)
+  def self.single_chat(system:, user:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil, temperature: nil, top_p: nil, frequency_penalty: nil, presence_penalty: nil, prediction: nil)
+    chat(messages: [{ system: }, { user: }], model:, response_format:, max_tokens:, store:, tools:, auto_run_functions:, function_context:, temperature:, top_p:, frequency_penalty:, presence_penalty:, prediction:)
   end
 
-  def self.chat(messages:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil, temperature: nil, top_p: nil, frequency_penalty: nil, presence_penalty: nil)
+  def self.chat(messages:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil, temperature: nil, top_p: nil, frequency_penalty: nil, presence_penalty: nil, prediction: nil)
     model = OpenAIExt::Model.select(model)
     is_o1_model = model.start_with?('o1')
 
@@ -47,6 +47,7 @@ module OpenAIExt
     parameters[:top_p] = top_p if top_p
     parameters[:frequency_penalty] = frequency_penalty if frequency_penalty
     parameters[:presence_penalty] = presence_penalty if presence_penalty
+    parameters[:prediction] = prediction if prediction
 
     begin
       response = OpenAI::Client.new.chat(parameters:)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: openaiext
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-10-13 00:00:00.000000000 Z
+date: 2024-11-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai