chalk_ruby 0.4.1 → 0.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be8baa09bb22ab16fe18d5d82d89fa4d5ca75f2216f60daeb240ee185aecafc1
-  data.tar.gz: 07cfafd2ff5de2898f492697f17eba513e5cdea23df58ed8489b637c3a5b6b40
+  metadata.gz: 4165478b519e054426f89a0e2a5baa7a8f1ef2e4eb3d23a4489bb478e96adca2
+  data.tar.gz: c51633a4a2a3ee79dfb2f1a22d9bf80d9a6c8719f587f71d894b7be877f1fa91
 SHA512:
-  metadata.gz: befc8666980d27db057207bcf38a97e8f347879ed0938a9f8a5fcd3cbb07fe06d00e458ec50b7a54c0f308ab1f8ad2c331abe5c63213acde8073e04954ae1442
-  data.tar.gz: b8fdb9a0157d03e68047bcc5da8ecd569441cbc60c93f9b42378c6a227a401b99b0a3eede639f5ec1bcf44891193fc619ea3ca7e5e64bc470e8cf40eeec3dc31
+  metadata.gz: 3a52b774af1729f1ba7194c97aa9ba8b323b3bbb47efe06dd9e8cfd5124e3fe963eb5fab5ecf698614cc9a3f4ae6aa09a65c6ff1f770f0d0ae6e257be80a1036
+  data.tar.gz: a4f91f89f52e8b3b036e3d19c1dde26845e430fa69e766cf88e0e7122f5a12f2750c8d689bfa7cdfc5d6094b307b608a8d79aeec9c5df0c462cf7a19450e38aa

data/lib/chalk_ruby/grpc_client.rb

@@ -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.2'.freeze
 end

data/test/chalk_ruby/integration/client_test.rb

@@ -3,8 +3,8 @@ require 'rspec/autorun'
 require 'chalk_ruby/client'
 require 'chalk_ruby/error'
 
-CLIENT_ID     = ''
-CLIENT_SECRET = ''
+CLIENT_ID     = 'client-9314816c664be6cb04678e5e3d0e50e2'
+CLIENT_SECRET = 'secret-51f3ce3d4f701f23afdd622015732edf1eb62d0d678b3511b615f9b1f00b4a7d'
 
 RSpec.describe 'Online query' do
   it 'should accept valid queries' do

data/test/chalk_ruby/integration/grpc_client_test.rb

@@ -14,6 +14,8 @@ require 'chalk_ruby/protos/chalk/engine/v1/query_server_services_pb'
 require 'arrow'
 
 
+CLIENT_ID     = 'client-9314816c664be6cb04678e5e3d0e50e2'
+CLIENT_SECRET = 'secret-51f3ce3d4f701f23afdd622015732edf1eb62d0d678b3511b615f9b1f00b4a7d'
 
 
 RSpec.describe ChalkRuby::GrpcClient do

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chalk_ruby
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.4.2
 platform: ruby
 authors:
 - Chalk AI, Inc.
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-07-08 00:00:00.000000000 Z
+date: 2025-08-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler