omniai-openai 2.0.1 → 2.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d9aa6adb553c3bb4ee1c6463b1d5654d4e9deb01044ee3e5e3974cb9fc762e2c
-  data.tar.gz: 3d2347a9ca7a198bc96b93afc670b1283fd9f871ff73de35ea5c8d0a6c78d56b
+  metadata.gz: 41ef95afc5946c5c220a540871805830c94f04f6b6e890782a30e44b56af9033
+  data.tar.gz: cd627890f6206f48e7187bae2dc713d875fd274b0aa1b1d3643b4cb89a63912e
 SHA512:
-  metadata.gz: d671369f7115e0b63193aa47815bd4552c2a375b8719c7585f376cd01e79f5b8dcfc163d8e910e75426146c011a7e3d74d3ae1994a38b9120d2eda3ab866a9bb
-  data.tar.gz: 0124e71067ecde28627f775ff4c27781d72a223a081163404a2fa0c67d1c890455c409baf7793a629f4158c9d85bd828eb971000ca4d431a31c9f01165eb2a81
+  metadata.gz: 50fe670e52459ea9fa0365d0d8ba1c586fe020c0fe36dc2e0de99114925e56e2aa6d6c7527d09d48482cbd391900d98301b3316b67e4ccb9a53773eb62abf34c
+  data.tar.gz: b3baa12324acea0b5b1165e721a25670baf77ffba23839903c68cf6c2d09a78063e273ddae6f10a0a1c19c2ca42c7d63964a47c9c1f78306e2bf0c5cdd509aeb

data/lib/omniai/openai/chat.rb

@@ -13,6 +13,7 @@ module OmniAI
     #   completion.choice.message.content # '...'
     class Chat < OmniAI::Chat
       JSON_RESPONSE_FORMAT = { type: "json_object" }.freeze
+      DEFAULT_STREAM_OPTIONS = { include_usage: ENV.fetch("OMNIAI_STREAM_USAGE", "on").eql?("on") }.freeze
 
       module Model
         GPT_4O = "gpt-4o"
@@ -35,9 +36,10 @@ module OmniAI
         OmniAI::OpenAI.config.chat_options.merge({
           messages: @prompt.serialize,
           model: @model,
-          stream: @stream.nil? ? nil : !@stream.nil?,
-          temperature: @temperature,
           response_format: (JSON_RESPONSE_FORMAT if @format.eql?(:json)),
+          stream: stream? || nil,
+          stream_options: (DEFAULT_STREAM_OPTIONS if stream?),
+          temperature: @temperature,
           tools: (@tools.map(&:serialize) if @tools&.any?),
         }).compact
       end

data/lib/omniai/openai/client.rb

@@ -104,7 +104,7 @@ module OmniAI
       # @param format [Symbol] :text, :srt, :vtt, or :json (default)
       #
       # @return [OmniAI::Transcribe]
-      def transcribe(path, model: Transcribe::Model::WHISPER, language: nil, prompt: nil, temperature: nil, format: nil)
+      def transcribe(path, model: Transcribe::DEFAULT_MODEL, language: nil, prompt: nil, temperature: nil, format: nil)
         Transcribe.process!(path, model:, language:, prompt:, temperature:, format:, client: self)
       end
 
@@ -125,7 +125,7 @@ module OmniAI
       # @yield [output] optional
       #
       # @return [Tempfile``]
-      def speak(input, model: Speak::Model::TTS_1_HD, voice: Speak::Voice::ALLOY, speed: nil, format: nil, &)
+      def speak(input, model: Speak::DEFAULT_MODEL, voice: Speak::DEFAULT_VOICE, speed: nil, format: nil, &)
         Speak.process!(input, model:, voice:, speed:, format:, client: self, &)
       end
 

data/lib/omniai/openai/speak.rb

@@ -7,17 +7,25 @@ module OmniAI
       module Model
         TTS_1 = "tts-1"
         TTS_1_HD = "tts-1-hd"
+        GPT_4O_MINI_TTS = "gpt-4o-mini-tts"
       end
 
       module Voice
         ALLOY = "alloy" # https://platform.openai.com/docs/guides/text-to-speech/alloy
+        ASH = "ash" # https://platform.openai.com/docs/guides/text-to-speech/ash
+        BALLARD = "ballard" # https://platform.openai.com/docs/guides/text-to-speech/ballard
+        CORAL = "coral" # https://platform.openai.com/docs/guides/text-to-speech/coral
         ECHO = "echo" # https://platform.openai.com/docs/guides/text-to-speech/echo
         FABLE = "fable" # https://platform.openai.com/docs/guides/text-to-speech/fable
         NOVA = "nova" # https://platform.openai.com/docs/guides/text-to-speech/nova
         ONYX = "onyx" # https://platform.openai.com/docs/guides/text-to-speech/onyx
+        SAGE = "sage" # https://platform.openai.com/docs/guides/text-to-speech/sage
         SHIMMER = "shimmer" # https://platform.openai.com/docs/guides/text-to-speech/shimmer
       end
 
+      DEFAULT_MODEL = Model::GPT_4O_MINI_TTS
+      DEFAULT_VOICE = Voice::ALLOY
+
     protected
 
       # @return [Hash]

data/lib/omniai/openai/transcribe.rb

@@ -6,9 +6,13 @@ module OmniAI
     class Transcribe < OmniAI::Transcribe
       module Model
         WHISPER_1 = "whisper-1"
+        GPT_4O_TRANSCRIBE = "gpt-4o-transcribe"
+        GPT_4O_MINI_TRANSCRIBE = "gpt-4-0-mini-transcribe"
         WHISPER = WHISPER_1
       end
 
+      DEFAULT_MODEL = Model::WHISPER
+
     protected
 
       # @return [Hash]

data/lib/omniai/openai/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module OpenAI
-    VERSION = "2.0.1"
+    VERSION = "2.2.0"
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: omniai-openai
 version: !ruby/object:Gem::Version
-  version: 2.0.1
+  version: 2.2.0
 platform: ruby
 authors:
 - Kevin Sylvestre
 bindir: exe
 cert_chain: []
-date: 2025-03-10 00:00:00.000000000 Z
+date: 2025-03-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -29,14 +29,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: '2.2'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: '2.2'
 - !ruby/object:Gem::Dependency
   name: zeitwerk
   requirement: !ruby/object:Gem::Requirement