omniai 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5d90d000047c670a17478146fa35fca9c993127067b01b11113cc46e739d15e2
-  data.tar.gz: d025855eb6ea4ae79bc7c126a2853fff7a6a0e21d597fa86b2efb6b8d1c3ddf1
+  metadata.gz: 4e387dddb26068bab8b4aa35a8e990d6db7aed7f034ae4593105b680282e5d43
+  data.tar.gz: 20f9b49d315dea539937521507058a16b8d6f8b0aaeea26f1b6be2cacf07122c
 SHA512:
-  metadata.gz: 77247cd74f88fdaa527b377940f32cfaad187c26f8ae4c6f373c708c866ce5c956c0436194786acb474ae07f423ea83511df6bbd5029c91247aff4b5d8ee2891
-  data.tar.gz: c7d3a40a6c21fe1827b162e94eae0dd0d2f66014797e4d878fa6c7aacf17c443412fcefa2f4bc0ad008f7b64b41ab4e2d583109990d6418d77451f1ba67b4fbd
+  metadata.gz: 8c3f503fe81921c2d690f169bc22d3d55b00b590920dc31c47c625a93dd142c995e369a9ffa3675c82bf7d04d3861e3b6e09b98b6e3afbd72bfed91b606e8473
+  data.tar.gz: 443d264b0f4c39058fd759b379e604cb631435334fff5fe7528b06cf8502af40c2fbfdca86300e5135029e90ab6104bb0c2b86e5198bd69fcd9c8071e6fbc07a

data/lib/omniai/chat/request.rb

@@ -6,7 +6,7 @@ module OmniAI
     #
     # Usage:
     #
-    #    class OmniAI::OpenAI::ChatGPT::Completion < OmniAI::Chat::Completion
+    #    class OmniAI::OpenAI::ChatGPT::Request < OmniAI::Chat::Request
     #      module Model
     #        CHAT = "davinci"
     #      end
@@ -23,7 +23,7 @@ module OmniAI
       # @param model [String] required
       # @param temperature [Float, nil] optional
       # @param stream [Proc, nil] optional
-      # @param format [Symbol, nil] optional - :text or :json
+      # @param format [Symbol, nil] optional - :json
       def initialize(client:, messages:, model:, temperature: nil, stream: nil, format: nil)
         @client = client
         @messages = messages
@@ -44,12 +44,12 @@ module OmniAI
       protected
 
       # @param response [HTTP::Response]
-      # @return [OmniAI::Chat::Completion::Response]
+      # @return [OmniAI::Chat::Completion]
       def parse!(response:)
         if @stream
           stream!(response:)
         else
-          OmniAI::OpenAI::Chat::Completion.new(data: response.parse)
+          OmniAI::Chat::Completion.new(data: response.parse)
         end
       end
 
@@ -62,7 +62,7 @@ module OmniAI
           parser.feed(chunk) do |_, data|
             break if data.eql?('[DONE]')
 
-            chunk = OmniAI::OpenAI::Chat::Chunk.new(data: JSON.parse(data))
+            chunk = OmniAI::Chat::Chunk.new(data: JSON.parse(data))
             @stream.call(chunk)
           end
         end

data/lib/omniai/config.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module OmniAI
   # A configuration for each agent w/ `api_key` / `host` / `logger`.
   class Config

data/lib/omniai/version.rb

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

metadata

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