chalk_ruby 0.3.1 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5078b236fa2532b078ba26509838b574bcb72c624ac4a27eb41f7fb34b9fb617
-  data.tar.gz: af01a9060816db57ff9198129a8d3ac2d0c7cce683e0ea59da1f9f6ed7ead09a
+  metadata.gz: 394c3d1735d1ee6d9ecc4df5d6ba9627a7323bc42a0c89df4879aadfc7a9b923
+  data.tar.gz: 62bca659a3b33122657df4c6bfccc2a420de7d815d8a4da09e7f3b01b13ef683
 SHA512:
-  metadata.gz: f0f5a7f1ca26cefac916b76bf8f83b77b94ecc64344985d7d5adf6fe77aaebc61b811b6c9f5cbc3dd1cc78dc7b1fe519a3d7b9e45656294c0bc021b42cafe9cf
-  data.tar.gz: 910e8c846f446f44bb556f34c430d4eec7e24482ff3e13f29ad6874c635f48f5539e0502c25dfb8a9d6d48227a538beedbcd598474d6bc3c059136746f9194b6
+  metadata.gz: 5027aaea4f0bc4c65b3f51d8552992c4831137f4a118f82f308332c9277381b2c4d86c71d4858aaf6f53a9518f6654918fec701188abb030dd48891b01f0b917
+  data.tar.gz: 90555aee1fde8a17db6624ea7d9ed8049e5ffee4f4ce4b2bdb1171edf91aae422cae8366958223ffe343bc9c2c69bdbacdb796079aca318519ba0503c56661a4

data/.github/workflows/release.yml

@@ -0,0 +1,41 @@
+name: Release Gem
+
+on:
+  push:
+    tags:
+      - 'v*'
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+    steps:
+      - uses: actions/checkout@v4
+      
+      - name: Set up Ruby
+        uses: ruby/setup-ruby@v1
+        with:
+          ruby-version: '3.0'
+          bundler-cache: true
+          
+      - name: Extract version from tag
+        id: get_version
+        run: echo "VERSION=${GITHUB_REF#refs/tags/v}" >> $GITHUB_ENV
+        
+      - name: Update version.rb
+        run: |
+          sed -i "s/VERSION = '.*'/VERSION = '${{ env.VERSION }}'/g" lib/chalk_ruby/version.rb
+          
+      - name: Build gem
+        run: gem build chalk_ruby.gemspec
+        
+      - name: Publish to RubyGems
+        run: |
+          mkdir -p $HOME/.gem
+          touch $HOME/.gem/credentials
+          chmod 0600 $HOME/.gem/credentials
+          printf -- "---\n:rubygems_api_key: ${GEM_HOST_API_KEY}\n" > $HOME/.gem/credentials
+          gem push "chalk_ruby-${{ env.VERSION }}.gem"
+        env:
+          GEM_HOST_API_KEY: ${{ secrets.GEM_HOST_API_KEY }}

data/.github/workflows/test.yml

@@ -0,0 +1,33 @@
+name: Run Tests
+
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    
+    strategy:
+      matrix:
+        ruby-version: ['2.7', '3.0', '3.1']
+    
+    steps:
+      - uses: actions/checkout@v4
+      
+      - name: Set up Ruby ${{ matrix.ruby-version }}
+        uses: ruby/setup-ruby@v1
+        with:
+          ruby-version: ${{ matrix.ruby-version }}
+          bundler-cache: true
+      
+      - name: Install dependencies
+        run: bundle install
+      
+      - name: Run linter
+        run: bundle exec rake rubocop
+        
+      - name: Run tests
+        run: bundle exec rake test:all

data/lib/chalk_ruby/client.rb

@@ -29,6 +29,11 @@ module ChalkRuby
     # @param additional_headers [Hash[String, String]]
     #   Additional headers to be sent with every request. Typically not required.
     #
+    # @param eagerly_initialize_connection_pool [Boolean]
+    #   Whether to eagerly initialize connections to the query and API servers.
+    #   When true, connections will be pre-warmed during client initialization.
+    #   Default is false (lazy initialization).
+    #
     # @return self
     #
     def self.create(
@@ -37,7 +42,8 @@ module ChalkRuby
       environment = nil,
       query_server = nil,
       api_server = nil,
-      additional_headers = {}
+      additional_headers = {},
+      eagerly_initialize_connection_pool = false
     )
       config = Config.new(
         client_id: client_id,
@@ -45,7 +51,8 @@ module ChalkRuby
         environment: environment,
         query_server: query_server,
         api_server: api_server,
-        additional_headers: additional_headers
+        additional_headers: additional_headers,
+        eagerly_initialize_connection_pool: eagerly_initialize_connection_pool
       )
       create_with_config(config)
     end
@@ -213,9 +220,36 @@ module ChalkRuby
       logger       = opts[:logger] || LoggerHelper.create
       requester    = opts[:http_requester] || Defaults::REQUESTER_CLASS.new(adapter: adapter, logger: logger)
       @transporter = Http::HttpRequesterChalk.new(requester: requester)
+      
+      # Eagerly initialize connections if configured
+      if @config.eagerly_initialize_connection_pool
+        warm_connections
+      end
+    end
+
+    # Warm up connections to the API and query servers by establishing
+    # connections without making actual requests
+    def warm_connections
+      # Get the query server host
+      query_host = query_server_host
+      
+      # Warm up API server connection
+      warm_connection(@config.api_server)
+      
+      # Warm up query server connection
+      warm_connection(query_host)
     end
 
     private
+    
+    # Establish a connection to a host without making a request
+    def warm_connection(host)
+      # Access the underlying HTTP requester
+      requester = @transporter.instance_variable_get(:@http_requester)
+      
+      # This will create and cache the Faraday connection
+      requester.connection(host)
+    end
 
     def api_server_request(method:, path:, body:, headers:)
       @transporter.send_request(

data/lib/chalk_ruby/config.rb

@@ -17,7 +17,8 @@ module ChalkRuby
                   :api_timeout,
                   :connect_timeout,
                   :query_service_root_ca_path,
-                  :additional_headers
+                  :additional_headers,
+                  :eagerly_initialize_connection_pool
 
     # Creates a new ChalkRuby::Config object for use with ChalkRuby::Client.
     #
@@ -52,6 +53,11 @@ module ChalkRuby
     # @option options [Hash<String, String>?] :additional_headers
     #  Additional headers to be sent with every request. Typically not required.
     #
+    # @option options [Boolean?] :eagerly_initialize_connection_pool
+    #  Whether to eagerly initialize connections to the query and API servers.
+    #  When true, connections will be pre-warmed during client initialization.
+    #  Default is false (lazy initialization).
+    #
     def initialize(opts = {})
       @client_id                        = opts[:client_id] || ENV['CHALK_CLIENT_ID']
       @client_secret                    = opts[:client_secret] || ENV['CHALK_CLIENT_SECRET']
@@ -63,6 +69,7 @@ module ChalkRuby
       @connect_timeout                  = opts[:connect_timeout] || Defaults::CONNECT_TIMEOUT
       @query_service_root_ca_path       = opts[:query_service_root_ca_path] || Defaults::ROOT_CA_PATH
       @additional_headers = opts[:additional_headers] || {}
+      @eagerly_initialize_connection_pool = opts[:eagerly_initialize_connection_pool] || false
 
       raise ChalkError, 'No Client ID provided, please set :client_id' if @client_id.nil?
       raise ChalkError, 'No Client Secret provided, please set :client_secret' if @client_secret.nil?

data/lib/chalk_ruby/version.rb

@@ -1,3 +1,3 @@
 module ChalkRuby
-  VERSION = '0.3.1'.freeze
+  VERSION = '0.4.0'.freeze
 end

data/test/chalk_ruby/integration/client_test.rb

@@ -8,7 +8,7 @@ CLIENT_SECRET = ''
 
 RSpec.describe 'Online query' do
   it 'should accept valid queries' do
-    client   = ChalkRuby::Client.create(CLIENT_ID, CLIENT_SECRET)
+    client   = ChalkRuby::Client.create(CLIENT_ID, CLIENT_SECRET, "tmnmrbyb8x53f")
     response = client.query(
       input: { 'user.id': 3454 },
       output: %w(user.id),

data/test/chalk_ruby/integration/connection_warming_test.rb

@@ -0,0 +1,35 @@
+require 'minitest/autorun'
+require 'chalk_ruby/client'
+require 'chalk_ruby/config'
+
+describe 'Connection warming' do
+  it 'should accept eagerly_initialize_connection_pool parameter in config' do
+    config = ChalkRuby::Config.new(
+      client_id: 'test_id',
+      client_secret: 'test_secret',
+      eagerly_initialize_connection_pool: true
+    )
+    
+    assert_equal true, config.eagerly_initialize_connection_pool
+  end
+  
+  it 'should default eagerly_initialize_connection_pool to false' do
+    config = ChalkRuby::Config.new(
+      client_id: 'test_id',
+      client_secret: 'test_secret'
+    )
+    
+    assert_equal false, config.eagerly_initialize_connection_pool
+  end
+  
+  it 'should accept eagerly_initialize_connection_pool parameter in create method' do
+    # This test verifies the parameter is accepted
+    params = ChalkRuby::Client.method(:create).parameters.map(&:last)
+    assert params.include?(:eagerly_initialize_connection_pool)
+  end
+  
+  it 'should have warm_connections method' do
+    # Test that the method exists on the class
+    assert ChalkRuby::Client.public_instance_methods.include?(:warm_connections)
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chalk_ruby
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.0
 platform: ruby
 authors:
 - Chalk AI, Inc.
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-05-05 00:00:00.000000000 Z
+date: 2025-07-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -234,6 +234,8 @@ files:
 - ".github/CODEOWNERS"
 - ".github/ISSUE_TEMPLATE/bug_report.md"
 - ".github/ISSUE_TEMPLATE/feature_request.md"
+- ".github/workflows/release.yml"
+- ".github/workflows/test.yml"
 - ".gitignore"
 - ".rubocop.yml"
 - ".rubocop_todo.yml"
@@ -364,6 +366,7 @@ files:
 - sig/chalk_ruby/versions.rbs
 - test/chalk_ruby/integration/arrow_conversion_test.rb
 - test/chalk_ruby/integration/client_test.rb
+- test/chalk_ruby/integration/connection_warming_test.rb
 - test/chalk_ruby/integration/grpc_client_test.rb
 - test/chalk_ruby/test_helper.rb
 homepage: https://github.com/chalk-ai/chalk-ruby
@@ -395,5 +398,6 @@ summary: A simple Ruby client for Chalk
 test_files:
 - test/chalk_ruby/integration/arrow_conversion_test.rb
 - test/chalk_ruby/integration/client_test.rb
+- test/chalk_ruby/integration/connection_warming_test.rb
 - test/chalk_ruby/integration/grpc_client_test.rb
 - test/chalk_ruby/test_helper.rb