omniai 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: eb51f64b3514f196e07be3b77f2ca538878602af864f5c2c49e5e2506f05bfaf
-  data.tar.gz: f4069c9f19fbcc6f2caf90ab28e32ed83c8700b2ea8c686059afc612997b1ea6
+  metadata.gz: 53ccc1363252abefa6089ff068b11b3ac15b5817339dc935ae9b78cf1d146b20
+  data.tar.gz: 1b4fe76443ef245f7198c1d55edab1fa6d985afbef1638705caf3d40a1b88c05
 SHA512:
-  metadata.gz: 06ece9d4bd7db17c57bff05f3323ba2ddd5e10803a5e5b460d8f5c66ffad932880230270d5a3e5e3653e46c6aaeec0926cff740414c9bb8d0e799062e4ce3139
-  data.tar.gz: 8443711ec0a65da572ce47a2da284ef63e277bdffc64663ab0577c4ded90105faaa666819f09b9d8ac2d2e457db8f4c07303b987ebb96e719648dd8624e419c0
+  metadata.gz: f687cd00e0cbed7994d8a172c657d135741e8ca5afdc8f4bcff53ffb6ec6746f553f26cfeeddcf08a81a261e72e7dcd0b4aee06e0b9e336016a1843189bb4a29
+  data.tar.gz: 6d89697cc18301852e6648b2a323634a59df9bf90032146d35837ead4a4746bad13527aeef8c67af8cd0290bdaae3177de50d29c220c8fff86f844f081484970

data/lib/omniai/chat.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module OmniAI
-  # An abstract class that provides an interface for chatting for various vendors (e.g. OpenAI::ChatGPT).
+  # An abstract class that provides an interface for chatting for various vendors (e.g. OpenAI’s ChatGPT).
   #
   # Usage:
   #
@@ -23,13 +23,14 @@ module OmniAI
 
     # @raise [OmniAI::Error]
     #
-    # @param messages [String]
-    # @param model [String]
-    # @param format [Symbol] either :text or :json
-    # @param temperature [Float]
+    # @param messages [String, Array, Hash, OmnniAI::Chat::Message]
+    # @param model [String] optional
+    # @param format [Symbol] optional :text or :json
+    # @param temperature [Float, nil] optional
+    # @param stream [Proc, nil] optional
     #
     # @return [OmniAI::Chat::Request]
-    def completion(messages, model:, temperature: 0.0, format: :text)
+    def completion(messages, model:, temperature: nil, format: nil, stream: nil)
       raise NotImplementedError, "#{self.class.name}#completion undefined"
     end
   end

data/lib/omniai/client.rb

@@ -6,7 +6,7 @@ module OmniAI
   # Usage:
   #
   #   class OmniAI::OpenAI::Client < OmniAI::Client
-  #     def initialize(api_key: ENV.fetch('OPENAI_API_KEY'))
+  #     def initialize(api_key: ENV.fetch('OPENAI_API_KEY'), logger: nil)
   #       super
   #     end
   #
@@ -21,13 +21,15 @@ module OmniAI
     attr_accessor :api_key
 
     # @param api_key [String]
-    def initialize(api_key:)
+    # @param logger [Logger]
+    def initialize(api_key:, logger: nil)
       @api_key = api_key
+      @logger = logger
     end
 
     # @return [HTTP::Client]
     def connection
-      HTTP.auth("Bearer #{api_key}")
+      raise NotImplementedError, "#{self.class.name}#connection undefined"
     end
 
     # @return [OmniAI::Chat] an instance of OmniAI::Chat

data/lib/omniai/version.rb

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

metadata

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