cohere-ai 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6f22e3efb3ba5824ea31a317019c161357645909902224eaba56fc14ae74a612
-  data.tar.gz: 0bdabed7a0be20e8ab55fc5a5c931360b314734820a2606aab75f7e3596c6526
+  metadata.gz: a5326901fa2fcef2d71d74de376e3ca9df19d9adb65fe6cf5e1a6d0bd195b8d6
+  data.tar.gz: 78e162ac160c691b0060945511059a9ea185ee44105298bbf784f3a8e86fbf5a
 SHA512:
-  metadata.gz: c68ceb466d71170a84644d68d7be0d5f40c3f9ba3cf7e45b891f2c99460967f808a85402e92a9ccb67195ac76ae2776621e8f66d9538568b4bafe128a8006ee5
-  data.tar.gz: abccfbf233944338c738516610f398891ef76defb155142fa2e322ef22832b4817b786ba48e33f19d268ff2823822c191a04eaa21b26f60a710b3533430b9765
+  metadata.gz: 9cff2fb206e5daa899b440b27280c9b155b012364319da3536e28875a14cbebbb6c6f2a00e07409ec1edc501cd9c9a87154fbd31a28c084985bf8eccab627d28
+  data.tar.gz: 71453d884fd1bd3e76a9c78c5c9752dd6b695d887a1448aa12d3b0421a788f3da9306a6796beebbd0335537fa81a67ddb92dd68fd965e5e9b65d26169ff58640

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    cohere-ai (1.0.0)
+    cohere-ai (1.0.1)
       faraday (~> 2.8, >= 2.8.1)
 
 GEM

data/README.md

@@ -9,7 +9,7 @@ A Ruby gem for interacting with [Cohere AI](https://cohere.com).
 ## TL;DR and Quick Start
 
 ```ruby
-gem 'cohere-ai', '~> 1.0.0'
+gem 'cohere-ai', '~> 1.0.1'
 ```
 
 ```ruby
@@ -87,11 +87,11 @@ Result:
 ### Installing
 
 ```sh
-gem install cohere-ai -v 1.0.0
+gem install cohere-ai -v 1.0.1
 ```
 
 ```sh
-gem 'cohere-ai', '~> 1.0.0'
+gem 'cohere-ai', '~> 1.0.1'
 ```
 
 ### Credentials
@@ -694,6 +694,7 @@ CohereError
 
 MissingAPIKeyError
 BlockWithoutServerSentEventsError
+IncompleteJSONReceivedError
 
 RequestError
 ```
@@ -716,7 +717,7 @@ gem build cohere-ai.gemspec
 
 gem signin
 
-gem push cohere-ai-1.0.0.gem
+gem push cohere-ai-1.0.1.gem
 ```
 
 ### Updating the README

data/components/errors.rb

@@ -10,6 +10,7 @@ module Cohere
 
     class MissingAPIKeyError < CohereError; end
     class BlockWithoutServerSentEventsError < CohereError; end
+    class IncompleteJSONReceivedError < CohereError; end
 
     class RequestError < CohereError
       attr_reader :request, :payload

data/controllers/client.rb

@@ -88,6 +88,8 @@ module Cohere
 
         method_to_call = request_method.to_s.strip.downcase.to_sym
 
+        partial_json = ''
+
         response = Faraday.new(request: @request_options) do |faraday|
           faraday.response :raise_error
         end.send(method_to_call) do |request|
@@ -105,27 +107,43 @@ module Cohere
                 raise_error.on_complete(env.merge(body: chunk))
               end
 
-              result = { event: safe_parse_json(chunk), raw: { chunk:, bytes:, env: } }
+              partial_json += chunk
+
+              parsed_json = safe_parse_json(partial_json)
+
+              if parsed_json
+                result = { event: parsed_json, raw: { chunk:, bytes:, env: } }
 
-              callback.call(result[:event], result[:raw]) unless callback.nil?
+                callback.call(result[:event], result[:raw]) unless callback.nil?
 
-              results << result
+                results << result
+
+                partial_json = ''
+              end
             end
           end
         end
 
-        return safe_parse_json(response.body) unless server_sent_events_enabled
+        return safe_parse_json_with_fallback_to_raw(response.body) unless server_sent_events_enabled
+
+        raise IncompleteJSONReceivedError, partial_json if partial_json != ''
 
         results.map { |result| result[:event] }
       rescue Faraday::ServerError => e
         raise RequestError.new(e.message, request: e, payload:)
       end
 
-      def safe_parse_json(raw)
-        raw.start_with?('{', '[') ? JSON.parse(raw) : raw
+      def safe_parse_json_with_fallback_to_raw(raw)
+        raw.to_s.lstrip.start_with?('{', '[') ? JSON.parse(raw) : raw
       rescue JSON::ParserError
         raw
       end
+
+      def safe_parse_json(raw)
+        raw.to_s.lstrip.start_with?('{', '[') ? JSON.parse(raw) : nil
+      rescue JSON::ParserError
+        nil
+      end
     end
   end
 end

data/static/gem.rb

@@ -3,10 +3,10 @@
 module Cohere
   GEM = {
     name: 'cohere-ai',
-    version: '1.0.0',
+    version: '1.0.1',
     author: 'gbaptista',
     summary: 'Interact with Cohere AI.',
-    description: "A Ruby gem for interacting with Cohere AI platform.",
+    description: 'A Ruby gem for interacting with Cohere AI platform.',
     github: 'https://github.com/gbaptista/cohere-ai',
     gem_server: 'https://rubygems.org',
     license: 'MIT',

data/template.md

@@ -9,7 +9,7 @@ A Ruby gem for interacting with [Cohere AI](https://cohere.com).
 ## TL;DR and Quick Start
 
 ```ruby
-gem 'cohere-ai', '~> 1.0.0'
+gem 'cohere-ai', '~> 1.0.1'
 ```
 
 ```ruby
@@ -50,11 +50,11 @@ Result:
 ### Installing
 
 ```sh
-gem install cohere-ai -v 1.0.0
+gem install cohere-ai -v 1.0.1
 ```
 
 ```sh
-gem 'cohere-ai', '~> 1.0.0'
+gem 'cohere-ai', '~> 1.0.1'
 ```
 
 ### Credentials
@@ -657,6 +657,7 @@ CohereError
 
 MissingAPIKeyError
 BlockWithoutServerSentEventsError
+IncompleteJSONReceivedError
 
 RequestError
 ```
@@ -679,7 +680,7 @@ gem build cohere-ai.gemspec
 
 gem signin
 
-gem push cohere-ai-1.0.0.gem
+gem push cohere-ai-1.0.1.gem
 ```
 
 ### Updating the README

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cohere-ai
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - gbaptista
@@ -74,7 +74,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.22
+rubygems_version: 3.3.3
 signing_key: 
 specification_version: 4
 summary: Interact with Cohere AI.