openaiext 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4622de30e5df0ef4fd41d49f329dd9ec64ca84f87bf8be59dbf3301e75b99429
-  data.tar.gz: fbd7a1741c752c1852f7d7962aaa44bb2dcbf641029de864963783d3b3d2f6a4
+  metadata.gz: 276d45b2a3e4b7e984a1fe2dbc7a05275500c8e15f0e331fe2a2f980db72dea0
+  data.tar.gz: 3366f551da7e14cfb6b0f273e352d79e9eb2c9eb159eb579b4e426f522d3e800
 SHA512:
-  metadata.gz: bd2d2f0ccda1be4e864414de652fbb628f16b172e786d6f8ba45266009829ceda9dacd9996a9af12bf2be63f2eb322363374942b5d6ddbac3edbca5bc61edbdb
-  data.tar.gz: 0ad5bb389487ccd355ae6cb0ac23080eb713b465434b189fe7c86ecedab1301ad7dbd7106ca24c2e5f26c8e2a1223b44a827774df8b14d6d007020fad428b1c4
+  metadata.gz: 38cd47294905c538f82f2a594148e4e8e64fb63caf9858be1f3efdd44613890654f941baab616e4f524f47cacaa5266809c23df9a0a87c360e4ebe70b4783d20
+  data.tar.gz: 6705205f99790956d259acc4793444d3c5740e656cf0e05427aa9f01bd1df0b8941363850793c146cf8bfe898f1cd267d08e7e052bb8d82e7fef20f5b9da5749

data/lib/openaiext/agent.rb

@@ -36,7 +36,7 @@ module OpenAIExt
 
       params[:additional_messages] = [{ role: :user, content: additional_message }] unless additional_message.nil?
 
-      params[:model] = model || @assistant['model']
+      params[:model] = OpenAIExt::Model.select(model) || @assistant['model']
 
       run_id = @openai_client.runs.create(thread_id: @thread['id'], parameters: params)['id']
 

data/lib/openaiext/model.rb

@@ -0,0 +1,24 @@
+module OpenAIExt
+  module Model
+    GPT_BASIC_MODEL = ENV.fetch('OPENAI_GPT_BASIC_MODEL', 'gpt-4o-mini')
+    GPT_ADVANCED_MODEL = ENV.fetch('OPENAI_GPT_ADVANCED_MODEL', 'gpt-4o-2024-08-06')
+
+    O1_BASIC_MODEL = ENV.fetch('OPENAI_O1_BASIC_MODEL', 'o1-mini')
+    O1_ADVANCED_MODEL = ENV.fetch('OPENAI_O1_ADVANCED_MODEL', 'o1-preview')
+
+    def self.select(model)
+      case model
+      when :gpt_basic
+        GPT_BASIC_MODEL
+      when :gpt_advanced
+        GPT_ADVANCED_MODEL
+      when :o1_basic
+        O1_BASIC_MODEL
+      when :o1_advanced
+        O1_ADVANCED_MODEL
+      else
+        model
+      end
+    end
+  end
+end

data/lib/openaiext.rb

@@ -1,16 +1,11 @@
 require 'openai'
 
+require 'openaiext/model'
 require 'openaiext/messages'
 require 'openaiext/response_extender'
 require 'openaiext/agent'
 
 module OpenAIExt
-  GPT_BASIC_MODEL = ENV.fetch('OPENAI_GPT_BASIC_MODEL', 'gpt-4o-mini')
-  GPT_ADVANCED_MODEL = ENV.fetch('OPENAI_GPT_ADVANCED_MODEL', 'gpt-4o-2024-08-06')
-
-  O1_BASIC_MODEL = ENV.fetch('OPENAI_O1_BASIC_MODEL', 'o1-mini')
-  O1_ADVANCED_MODEL = ENV.fetch('OPENAI_O1_ADVANCED_MODEL', 'o1-preview')
-  
   MAX_TOKENS = ENV.fetch('OPENAI_MAX_TOKENS', 16_383).to_i
 
   def self.embeddings(input, model: 'text-embedding-3-large')
@@ -33,7 +28,7 @@ module OpenAIExt
   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)
-    model = select_model(model)
+    model = OpenAIExt::Model.select(model)
     is_o1_model = model.start_with?('o1')
 
     messages = OpenAIExt::Messages.new(messages) unless messages.is_a?(OpenAIExt::Messages)
@@ -69,19 +64,4 @@ module OpenAIExt
   end
 
   def self.models = OpenAI::Client.new.models.list
-
-  def self.select_model(model)
-    case model
-    when :gpt_basic
-      GPT_BASIC_MODEL
-    when :gpt_advanced
-      GPT_ADVANCED_MODEL
-    when :o1_basic
-      O1_BASIC_MODEL
-    when :o1_advanced
-      O1_ADVANCED_MODEL
-    else
-      model
-    end
-  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: openaiext
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Gedean Dias
@@ -62,6 +62,7 @@ files:
 - lib/openaiext.rb
 - lib/openaiext/agent.rb
 - lib/openaiext/messages.rb
+- lib/openaiext/model.rb
 - lib/openaiext/response_extender.rb
 homepage: https://github.com/gedean/openaiext
 licenses: