omniai 1.6.0 → 1.6.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b44fd75b138e1f46f517e7077367e02bdbcf1f798ee000087c5ba4c97bb7494b
-  data.tar.gz: 88002c5d7018fc5a940b30bfa644a93e93e9b93d0cbdd993306bf15943ea790d
+  metadata.gz: ee3875380bef07417246342465e621d52dcaf30bf7fff0beb99e341cb61d5258
+  data.tar.gz: 9bc2db4fd5dbdc456810753e0738c4da0481670c84d3c18f6f7ca310cd0d5514
 SHA512:
-  metadata.gz: f3bc56ecbeaaf50d776c87eb07ba571b85f2ab001b8181cff66875a9a282289fdec9bfa2c2a716637758f61a76d0c3ddf8b794abee6e3adba6117ad4b4fc4ccd
-  data.tar.gz: 233f99eb1daf1a75173559ad2649461aa1b1e1bc1f53eba65c5113690f38017258db79f4ba8d83e525c363adcd76caae9997fda49cf32c01198917c030b2b8f7
+  metadata.gz: d61725b7ed361b560d90a5887084dcaee519dca4b509fbf0a4335f10106647a577b6f852c51fd2341fa9f3be361e29d5341fd1572f068317f278651fbe6323e9
+  data.tar.gz: 7ed311f56483076821f7906f30c6ebc8ce21798f4a03bf2bfaab4d5bab66c56c5f9a61ffb95cc32f988e91bcac32d90fc1b15df0bca14a0e7bbfe858037b510b

data/lib/omniai/chat.rb

@@ -57,7 +57,11 @@ module OmniAI
     # @param stream [Proc, IO, nil] optional
     # @param tools [Array<OmniAI::Tool>] optional
     # @param format [Symbol, nil] optional - :json
+    #
     # @yield [prompt] optional
+    # @yieldparam prompt [OmniAI::Chat::Prompt]
+    #
+    # @return [OmniAi::Chat]
     def initialize(prompt = nil, client:, model:, temperature: nil, stream: nil, tools: nil, format: nil, &block)
       raise ArgumentError, 'prompt or block is required' if !prompt && !block
 

data/lib/omniai/client.rb

@@ -124,15 +124,18 @@ module OmniAI
 
     # @raise [OmniAI::Error]
     #
-    # @param messages [String, Array, Hash]
+    # @param messages [String] optional
     # @param model [String] optional
     # @param format [Symbol] optional :text or :json
     # @param temperature [Float, nil] optional
     # @param stream [Proc, nil] optional
     # @param tools [Array<OmniAI::Tool>] optional
     #
+    # @yield [prompt] optional
+    # @yieldparam prompt [OmniAI::Chat::Prompt]
+    #
     # @return [OmniAI::Chat::Completion]
-    def chat(messages, model:, temperature: nil, format: nil, stream: nil, tools: nil)
+    def chat(messages = nil, model:, temperature: nil, format: nil, stream: nil, tools: nil, &)
       raise NotImplementedError, "#{self.class.name}#chat undefined"
     end
 

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = '1.6.0'
+  VERSION = '1.6.2'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.6.0
+  version: 1.6.2
 platform: ruby
 authors:
 - Kevin Sylvestre