chalk_ruby 0.4.1 → 0.4.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be8baa09bb22ab16fe18d5d82d89fa4d5ca75f2216f60daeb240ee185aecafc1
-  data.tar.gz: 07cfafd2ff5de2898f492697f17eba513e5cdea23df58ed8489b637c3a5b6b40
+  metadata.gz: 41aeb7dd875f71ac96acd9d74dc75f583d59bc21a8611002996de8915ef92c11
+  data.tar.gz: dd7ff83b7c46fdfc1dd9cddba6cf874113c3ad64a0126e3efbf5b0abd4b31da9
 SHA512:
-  metadata.gz: befc8666980d27db057207bcf38a97e8f347879ed0938a9f8a5fcd3cbb07fe06d00e458ec50b7a54c0f308ab1f8ad2c331abe5c63213acde8073e04954ae1442
-  data.tar.gz: b8fdb9a0157d03e68047bcc5da8ecd569441cbc60c93f9b42378c6a227a401b99b0a3eede639f5ec1bcf44891193fc619ea3ca7e5e64bc470e8cf40eeec3dc31
+  metadata.gz: fb9fc19e2b66f1ad829b3024d6d63a3aa28be882501689a56193424e447a3f98ebd49e795b0f3f8983052b7447afe6e767a4454923598f5625345c0361750d6e
+  data.tar.gz: e4f11dfeb743205a724e890335d6811f4f36eb93157da25874196f18bf8c26d52bb1149909f8760026414205a896ab3f6f42f78aa1096af02baa56dcec370adc

data/lib/chalk_ruby/client.rb

@@ -228,7 +228,7 @@ module ChalkRuby
     end
 
     # Warm up connections to the API and query servers by establishing
-    # TCP connections with lightweight HEAD requests
+    # connections without making actual requests
     def warm_connections
       # Get the query server host
       query_host = query_server_host
@@ -242,23 +242,13 @@ module ChalkRuby
 
     private
     
-    # Establish a connection to a host by making a lightweight request
+    # 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)
       
-      # Create the Faraday connection object
-      connection = requester.connection(host)
-      
-      # Make a HEAD request to "/" to actually establish the TCP connection
-      # HEAD is used as it's lightweight and most servers support it
-      begin
-        connection.head('/')
-      rescue => e
-        # Ignore errors - we're just warming the connection
-        # The server might not support HEAD on /, but the TCP connection
-        # will still be established
-      end
+      # This will create and cache the Faraday connection
+      requester.connection(host)
     end
 
     def api_server_request(method:, path:, body:, headers:)

data/lib/chalk_ruby/grpc_client.rb

@@ -248,7 +248,7 @@ module ChalkRuby
     end
 
     def get_token
-      response = @auth_stub.get_token(
+      response = auth_service.get_token(
         Chalk::Server::V1::GetTokenRequest.new(
           client_id: @config.client_id,
           client_secret: @config.client_secret
@@ -469,26 +469,55 @@ module ChalkRuby
     end
 
     def to_feather(input_hash)
-      unless red_arrow_available?
-        raise NotImplementedError, "to_feather requires the 'red-arrow' gem. Please add it to your Gemfile: gem 'red-arrow', '~> 18.0.0'"
-      end
+      raise NotImplementedError,
+            "Add `gem 'red-arrow', '~> 18.0'` to your Gemfile" unless red_arrow_available?
+
       require 'arrow'
 
-      # Ensure all values in the input hash are arrays
-      array_input_hash = input_hash.transform_values { |v| v.is_a?(Array) ? v : [v] }
+      # Normalise column data
+      array_input_hash = input_hash.transform_keys(&:to_s)
+                                   .transform_values { |v| v.is_a?(Array) ? v : [v] }
 
-      # Create a table from the transformed input hash
-      table = Arrow::Table.new(array_input_hash)
+      lengths = array_input_hash.values.map(&:length).uniq
+      raise ArgumentError, "Columns must be the same length (got #{lengths})" unless lengths.one?
 
-      # Write to feather format in memory
-      buffer = Arrow::ResizableBuffer.new(0)
+      table  = Arrow::Table.new(array_input_hash)
 
+      buffer = Arrow::ResizableBuffer.new(0)
       output = Arrow::BufferOutputStream.new(buffer)
 
+      # ──────────────────────────────────────────────────────────
       table.write_as_feather(output)
+      output.close                      # ← **critical**: flush footer
+      # ──────────────────────────────────────────────────────────
 
-      # Remove trailing null bytes from the resizable buffer; the buffer is 512 aligned
-      buffer.data.to_s.b.gsub(/ARROW1(.*)ARROW1.*/m) {"ARROW1#{$1}ARROW1"}
+      # The stream writes exactly `output.tell` bytes; slice to that.
+      bytes_written = output.tell
+      buffer.data.to_s.b.byteslice(0, bytes_written)
     end
     end
+
+    # def to_feather(input_hash)
+    #   unless red_arrow_available?
+    #     raise NotImplementedError, "to_feather requires the 'red-arrow' gem. Please add it to your Gemfile: gem 'red-arrow', '~> 18.0.0'"
+    #   end
+    #   require 'arrow'
+    #
+    #   # Ensure all values in the input hash are arrays
+    #   array_input_hash = input_hash.transform_values { |v| v.is_a?(Array) ? v : [v] }
+    #
+    #   # Create a table from the transformed input hash
+    #   table = Arrow::Table.new(array_input_hash)
+    #
+    #   # Write to feather format in memory
+    #   buffer = Arrow::ResizableBuffer.new(0)
+    #
+    #   output = Arrow::BufferOutputStream.new(buffer)
+    #
+    #   table.write_as_feather(output)
+    #
+    #   # Remove trailing null bytes from the resizable buffer; the buffer is 512 aligned
+    #   buffer.data.to_s.b.gsub(/ARROW1(.*)ARROW1.*/m) {"ARROW1#{$1}ARROW1"}
+    # end
+    # end
   end

data/lib/chalk_ruby/version.rb

@@ -1,3 +1,3 @@
 module ChalkRuby
-  VERSION = '0.4.1'.freeze
+  VERSION = '0.4.3'.freeze
 end

data/test/chalk_ruby/integration/grpc_client_test.rb

@@ -73,4 +73,32 @@ RSpec.describe ChalkRuby::GrpcClient do
       expect(response[:data][0]['user.id']).to eq(1)
     end
   end
+
+  describe '#get_token' do
+    let(:client) do
+      ChalkRuby::GrpcClient.new(
+        ChalkRuby::Config.new(
+          query_server: "standard-gke.chalk-develop.gcp.chalk.ai",
+          api_server: "api.staging.chalk.ai:443",
+          client_id: CLIENT_ID,
+          client_secret: CLIENT_SECRET,
+          environment: "tmnmc9beyujew"
+        )
+      )
+    end
+
+    it 'can get auth token successfully' do
+      token = client.get_token
+
+      expect(token).not_to be_nil
+      expect(token).to be_a(String)
+      expect(token.length).to be > 0
+    end
+
+    it 'uses auth_service instead of undefined @auth_stub' do
+      # This test ensures the fix for the NoMethodError is working
+      # The method should not raise an error about undefined @auth_stub
+      expect { client.get_token }.not_to raise_error(NoMethodError)
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chalk_ruby
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.4.3
 platform: ruby
 authors:
 - Chalk AI, Inc.
-autorequire:
+autorequire: 
 bindir: exe
 cert_chain: []
-date: 2025-07-08 00:00:00.000000000 Z
+date: 2025-08-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -376,7 +376,7 @@ metadata:
   bug_tracker_uri: https://github.com/chalk-ai/chalk-ruby/issues
   documentation_uri: https://docs.chalk.ai/docs
   source_code_uri: https://github.com/chalk-ai/chalk-ruby
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -391,8 +391,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.22
-signing_key:
+rubygems_version: 3.0.3.1
+signing_key: 
 specification_version: 4
 summary: A simple Ruby client for Chalk
 test_files: