rasti-ai 1.2.0 → 1.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f62b3decd7ee67906392eff6218e29fd8a66bee1f5dc41729e4412193f7ff4d0
-  data.tar.gz: 3368912d5ee86e2ba10408aa4795c97820b8f313a6d17d693deca2f0454ad9e8
+  metadata.gz: 6e8aa13644a19b41a31445f268339fb14a9521ce85538493768bc5ee985b2a40
+  data.tar.gz: 82bc0cf208f412df7958bb8b95f5bdfa2b5deec0b892834afe82d0675543179e
 SHA512:
-  metadata.gz: da0b274e1ed8e4d4c0d9d4f7867ca546f47bc13c48246e111a71f465719237abf11985384699097f19d61822a33575fdd2a3398c63b2bb12a4612b11e563cec6
-  data.tar.gz: 48618afd26f383ff8c38d7527c3aac4f44015c9812677c93a893c7d9ccbddc0ed8cd4f6ebb6309acac3c334effaffec6138fb72715f036a51083839129f44cab
+  metadata.gz: 93490666739ff2b3dbc01a023149b947ee88e8ba503224194f968892b0070595bdc463332a22be3a0d77dbdae6ced218dbe62f6589fddb590def2ccc6a7e6fd2
+  data.tar.gz: 0c0ca23c2a085fdba7549ffa0696c8ffe8e492a626b5c123bf7259310a4088f6177acab70e14c7b856f8cb6d2a0743c5722fa18edc1e7f9bd7c84e1d4b2b45cd

data/lib/rasti/ai/open_ai/client.rb

@@ -5,12 +5,17 @@ module Rasti
 
         BASE_URL = 'https://api.openai.com/v1'.freeze
 
-        def initialize(api_key:nil, logger:nil)
+        RETRYABLE_STATUS_CODES = [502, 503, 504].freeze
+
+        def initialize(api_key:nil, logger:nil, http_connect_timeout:nil, http_read_timeout:nil, http_max_retries:nil)
           @api_key = api_key || Rasti::AI.openai_api_key
           @logger = logger || Rasti::AI.logger
+          @http_connect_timeout = http_connect_timeout || Rasti::AI.http_connect_timeout
+          @http_read_timeout = http_read_timeout || Rasti::AI.http_read_timeout
+          @http_max_retries = http_max_retries || Rasti::AI.http_max_retries
         end
 
-        def chat_completions(messages:, model:nil, tools:[], response_format: nil)
+        def chat_completions(messages:, model:nil, tools:[], response_format:nil)
           body = {
             model: model || Rasti::AI.openai_default_model,
             messages: messages,
@@ -25,35 +30,54 @@ module Rasti
 
         private
 
-        attr_reader :api_key, :logger
+        attr_reader :api_key, :logger, :http_connect_timeout, :http_read_timeout, :http_max_retries
 
         def post(relative_url, body)
-          url = "#{BASE_URL}#{relative_url}"
-          uri = URI(url)
+          max_retries = http_max_retries
+          retry_count = 0
+
+          begin
+            url = "#{BASE_URL}#{relative_url}"
+            uri = URI.parse url
+
+            logger.info(self.class) { "POST #{url}" }
+            logger.debug(self.class) { JSON.pretty_generate(body) }
+
+            request = Net::HTTP::Post.new uri
+            request['Authorization'] = "Bearer #{api_key}"
+            request['Content-Type'] = 'application/json'
+            request.body = JSON.dump body
 
-          logger.info(self.class) { "POST #{url}" }
-          logger.debug(self.class) { JSON.pretty_generate(body) }
+            http = Net::HTTP.new uri.host, uri.port
+            http.use_ssl = (uri.scheme == 'https')
 
-          request = Net::HTTP::Post.new uri
-          request['Authorization'] = "Bearer #{api_key}"
-          request['Content-Type'] = 'application/json'
-          request.body = JSON.dump(body)
+            http.open_timeout = http_connect_timeout
+            http.read_timeout = http_read_timeout
 
-          http = Net::HTTP.new uri.host, uri.port
-          http.use_ssl = uri.scheme == 'https'
+            response = http.request request
 
-          response = http.request request
+            logger.info(self.class) { "Response #{response.code}" }
+            logger.debug(self.class) { response.body }
 
-          logger.info(self.class) { "Response #{response.code}" }
-          logger.debug(self.class) { response.body }
+            if !response.is_a?(Net::HTTPSuccess) || RETRYABLE_STATUS_CODES.include?(response.code.to_i)
+              raise Errors::RequestFail.new(url, body, response)
+            end
 
-          raise Errors::RequestFail.new(url, body, response) unless response.is_a? Net::HTTPSuccess
+            JSON.parse response.body
 
-          JSON.parse response.body
+          rescue SocketError, Net::OpenTimeout, Net::ReadTimeout, Errors::RequestFail => e
+            if retry_count < max_retries
+              retry_count += 1
+              logger.warn(self.class) { "#{e.class.name}: #{e.message} (#{retry_count}/#{max_retries})" }
+              sleep retry_count
+              retry
+            end
+            raise
+          end
         end
 
       end
     end
 
   end
-end
+end

data/lib/rasti/ai/version.rb

@@ -1,5 +1,5 @@
 module Rasti
   module AI
-    VERSION = '1.2.0'
+    VERSION = '1.2.1'
   end
 end

data/lib/rasti/ai.rb

@@ -17,6 +17,10 @@ module Rasti
 
     attr_config :logger, Logger.new(STDOUT)
 
+    attr_config :http_connect_timeout, 60
+    attr_config :http_read_timeout, 60
+    attr_config :http_max_retries, 3
+
     attr_config :openai_api_key, ENV['OPENAI_API_KEY']
     attr_config :openai_default_model, ENV['OPENAI_DEFAULT_MODEL']
 

data/spec/minitest_helper.rb

@@ -15,6 +15,8 @@ require_relative 'support/helpers/resources'
 Rasti::AI.configure do |config|
   config.logger.level = Logger::FATAL
 
+  config.http_max_retries = 1
+
   config.openai_api_key = 'test_api_key'
   config.openai_default_model = 'gpt-test'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rasti-ai
 version: !ruby/object:Gem::Version
-  version: 1.2.0
+  version: 1.2.1
 platform: ruby
 authors:
 - Gabriel Naiman
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-11-18 00:00:00.000000000 Z
+date: 2025-12-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: multi_require