omniai 1.3.0 → 1.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 341ce12cf0950b167a43c298f3ccb1df821bbdc469bc6824b14219bef4e38ab5
-  data.tar.gz: 9badfe7e48d20fe84054d9206e8572a333f382407c3236a3715e1d85cf6b6a84
+  metadata.gz: 28e6e2a736d78cbc0135308eb28320a85a319b78a888cd96fa3320456c3c0222
+  data.tar.gz: '044786d21ba8f2316599836af153c6dec80d055a0cbcf1e843c000296d1df350'
 SHA512:
-  metadata.gz: f81164e402f5e1e4b2ccc2e627675d42631cfda985bf91469eb32735a789eecbe69c7f10bd9d2d5b4959f2d0f2484ae1bf7ced58f607a7cae822846c831ce96c
-  data.tar.gz: '0182828dad6fb6f6d98550608c3bd57ee980ff4acbcc6dfb3ee7d61a16200566f1c8965424537583d0b73225298ca5eba6ddafd3c51791ca847f53264a7654fd'
+  metadata.gz: c260f1b52fe5ffec70dc5c6755426c9798062bb234da303d76ec21319d450433e95f03ca787fcfa8cd966899235416d5a83580ffd11c24615d8a58df8d64efbc
+  data.tar.gz: 0acc9d0556d0bf8de430ca3a52977c88c6a43edfe017226d1831b4ccc034d2ba3bcde38fad112d61d8910bfc9a9ee440999354919a8d4c3cb614a0d0fc3e2314

data/README.md

@@ -87,40 +87,11 @@ client = OmniAI::Example::Client.new(logger:)
 ```
 
 ```
-I, [...]  INFO -- : > POST https://...
-D, [...] DEBUG -- : Authorization: Bearer ...
+[INFO]: POST https://...
+[INFO]: 200 OK
 ...
-{"messages":[{"role":"user","content":"Tell me a joke!"}],"model":"..."}
-I, [...]  INFO -- : < 200 OK
-D, [...] DEBUG -- : Date: ...
-...
-{
-  "id": "...",
-  "object": "...",
-  ...
-}
-```
-
-The level of the logger can be configured to either `INFO` and `DEBUG`:
-
-**INFO**:
-
-```ruby
-logger.level = Logger::INFO
 ```
 
-- Request: verb / URI
-- Response: status
-
-**DEBUG**:
-
-```ruby
-logger.level = Logger::DEBUG
-```
-
-- Request: verb / URI / headers / body
-- Response: status / headers / body
-
 #### Timeouts
 
 Timeouts are configurable by passing a `timeout` an integer duration for the request / response of any APIs using:

data/lib/omniai/chat.rb

@@ -59,6 +59,7 @@ module OmniAI
     # @raise [HTTPError]
     def process!
       response = request!
+
       raise HTTPError, response.flush unless response.status.ok?
 
       parse!(response:)

data/lib/omniai/client.rb

@@ -57,7 +57,7 @@ module OmniAI
     # @return [HTTP::Client]
     def connection
       http = HTTP.persistent(@host)
-      http = http.use(logging: { logger: @logger }) if @logger
+      http = http.use(instrumentation: { instrumenter: Instrumentation.new(logger: @logger) }) if @logger
       http = http.timeout(@timeout) if @timeout
       http
     end

data/lib/omniai/instrumentation.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module OmniAI
+  # Used for logging.
+  class Instrumentation
+    # @param logger [Logger]
+    def initialize(logger:)
+      @logger = logger
+    end
+
+    # @param name [String]
+    # @param payload [Hash]
+    # @option payload [Exception] :error
+    def instrument(name, payload = {})
+      error = payload[:error]
+      return unless error
+
+      @logger.error("#{name}: #{error.message}")
+    end
+
+    # @param name [String]
+    # @param payload [Hash]
+    # @option payload [HTTP::Request] :request
+    def start(_, payload)
+      request = payload[:request]
+      @logger.info("#{request.verb.upcase} #{request.uri}")
+    end
+
+    # @param name [String]
+    # @param payload [Hash]
+    # @option payload [HTTP::Response] :response
+    def finish(_, payload)
+      response = payload[:response]
+      @logger.info("#{response.status.code} #{response.status.reason}")
+    end
+  end
+end

data/lib/omniai/speak.rb

@@ -84,6 +84,7 @@ module OmniAI
     # @return [Tempfile]
     def process!(&block)
       response = request!
+
       raise HTTPError, response.flush unless response.status.ok?
 
       if block

data/lib/omniai/transcribe.rb

@@ -117,6 +117,7 @@ module OmniAI
     # @return [OmniAI::Transcribe::Transcription]
     def process!
       response = request!
+
       raise HTTPError, response.flush unless response.status.ok?
 
       text = @format.nil? || @format.eql?(Format::JSON) ? response.parse['text'] : String(response.body)

data/lib/omniai/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.3.1
 platform: ruby
 authors:
 - Kevin Sylvestre
@@ -80,6 +80,7 @@ files:
 - lib/omniai/chat/usage.rb
 - lib/omniai/client.rb
 - lib/omniai/config.rb
+- lib/omniai/instrumentation.rb
 - lib/omniai/speak.rb
 - lib/omniai/transcribe.rb
 - lib/omniai/transcribe/transcription.rb