chalk_ruby 0.2.3 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: de352344a0d00f988be587d33f4e1d181256c0a075832fa4daa804639af83f9d
-  data.tar.gz: d4bde660475858f5eeb7b6bdfb08bdabacacbc36c92962007e3f1acf8f929dff
+  metadata.gz: cfe57e2b3d1f09eaa9dece4b777788941b347319bca50b3242c52070287aa86f
+  data.tar.gz: '039220d201af61aff4e19fc73ea242a797b8681f87dc449e13187996eae653e6'
 SHA512:
-  metadata.gz: a811e70301fadceb6fd1e0822e51dd60f3a297f1ef1afe1e20919824b877a6bcd479dc3233b9a326f08b1b5028569cd42e42b9778ac29fde073349cb7f3d4e7a
-  data.tar.gz: f45d5ba28647765f57a527c1602053f7daa7aad04b72f17f594f68bf1ef0f901453a35d9a87525a66b99d1b10aace71cd14936b8699a4efe7943db4911045d58
+  metadata.gz: cc1d7cbfcdb2175ee61773918800270f963b136fea24573b06b97a8b22225f5488c9fed440755d964f32577822d61021122469c75c65b7e5271b19cccec03489
+  data.tar.gz: 79b8d14bbf904286f5589542dc85a0ff471b7e79ad675860893bd3e7fdaf9ceb3736a57773ed8bc78da9709a41df34dc6d4badcedad6d0c427a851dfb029aad0

data/lib/chalk_ruby/config.rb

@@ -16,6 +16,7 @@ module ChalkRuby
                   :query_timeout,
                   :api_timeout,
                   :connect_timeout,
+                  :query_service_root_ca_path,
                   :additional_headers
 
     # Creates a new ChalkRuby::Config object for use with ChalkRuby::Client.
@@ -45,18 +46,22 @@ module ChalkRuby
     # @option options [Float?] :connect_timeout
     #   The timeout for connect operations (in seconds).
     #
+    # @option options [String?] :query_service_root_ca_path
+    #   The root ca to use for trusting SSL verifying connections to the query server in grpc.
+    #
     # @option options [Hash<String, String>?] :additional_headers
     #  Additional headers to be sent with every request. Typically not required.
     #
     def initialize(opts = {})
-      @client_id          = opts[:client_id] || ENV['CHALK_CLIENT_ID']
-      @client_secret      = opts[:client_secret] || ENV['CHALK_CLIENT_SECRET']
-      @environment        = opts[:environment] || ENV['CHALK_ACTIVE_ENVIRONMENT']
-      @query_server       = opts[:query_server] || ENV['CHALK_QUERY_SERVER'] || Defaults::QUERY_SERVER
-      @api_server         = opts[:api_server] || ENV['CHALK_API_SERVER'] || Defaults::API_SERVER
-      @query_timeout      = opts[:query_timeout] || Defaults::QUERY_TIMEOUT
-      @api_timeout        = opts[:api_timeout] || Defaults::API_TIMEOUT
-      @connect_timeout    = opts[:connect_timeout] || Defaults::CONNECT_TIMEOUT
+      @client_id                        = opts[:client_id] || ENV['CHALK_CLIENT_ID']
+      @client_secret                    = opts[:client_secret] || ENV['CHALK_CLIENT_SECRET']
+      @environment                      = opts[:environment] || ENV['CHALK_ACTIVE_ENVIRONMENT']
+      @query_server                     = opts[:query_server] || ENV['CHALK_QUERY_SERVER'] || Defaults::QUERY_SERVER
+      @api_server                       = opts[:api_server] || ENV['CHALK_API_SERVER'] || Defaults::API_SERVER
+      @query_timeout                    = opts[:query_timeout] || Defaults::QUERY_TIMEOUT
+      @api_timeout                      = opts[:api_timeout] || Defaults::API_TIMEOUT
+      @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] || {}
 
       raise ChalkError, 'No Client ID provided, please set :client_id' if @client_id.nil?

data/lib/chalk_ruby/defaults.rb

@@ -32,6 +32,10 @@ module ChalkRuby
     API_TIMEOUT     = 300
     QUERY_TIMEOUT   = 1200
 
+    # GRPC
+    # ----------------------------------------
+    ROOT_CA_PATH = nil
+
     WAIT_TASK_DEFAULT_TIME_BEFORE_RETRY = 100
   end
 end

data/lib/chalk_ruby/grpc_client.rb

@@ -183,18 +183,14 @@ module ChalkRuby
     end
 
     def get_token
-      Token.new(
-        api_server_request(
-          method: :post,
-          path: '/v1/oauth/token',
-          body: {
-            client_id: @config.client_id,
-            client_secret: @config.client_secret,
-            grant_type: 'client_credentials'
-          },
-          headers: get_unauthenticated_headers
+      response = @auth_stub.get_token(
+        Chalk::Server::V1::GetTokenRequest.new(
+          client_id: @config.client_id,
+          client_secret: @config.client_secret
         )
       )
+
+      response.access_token
     end
 
     def get_unauthenticated_headers
@@ -280,7 +276,14 @@ module ChalkRuby
 
     def query_service
       if @query_service.nil?
-        @query_service = Chalk::Engine::V1::QueryService::Stub.new(query_server_host, GRPC::Core::ChannelCredentials.new(), interceptors: [engine_interceptor])
+        channel_creds =
+          if @config.query_service_root_ca_path.nil?
+            GRPC::Core::ChannelCredentials.new
+          else
+            GRPC::Core::ChannelCredentials.new(File.read(@config.query_service_root_ca_path))
+          end
+
+        @query_service = Chalk::Engine::V1::QueryService::Stub.new(query_server_host, channel_creds, interceptors: [engine_interceptor])
       end
 
       @query_service

data/lib/chalk_ruby/version.rb

@@ -1,3 +1,3 @@
 module ChalkRuby
-  VERSION = '0.2.3'.freeze
+  VERSION = '0.2.4'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chalk_ruby
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.2.4
 platform: ruby
 authors:
 - Chalk AI, Inc.
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-02-23 00:00:00.000000000 Z
+date: 2025-02-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler