brainzlab 0.1.2 → 0.1.3

data/lib/brainzlab/vault/client.rb

@@ -1,22 +1,22 @@
 # frozen_string_literal: true
 
-require "net/http"
-require "json"
-require "uri"
+require 'net/http'
+require 'json'
+require 'uri'
 
 module BrainzLab
   module Vault
     class Client
       def initialize(config)
         @config = config
-        @base_url = config.vault_url || "https://vault.brainzlab.ai"
+        @base_url = config.vault_url || 'https://vault.brainzlab.ai'
       end
 
       def get(key, environment:)
         response = request(
           :get,
           "/api/v1/secrets/#{CGI.escape(key)}",
-          headers: { "X-Vault-Environment" => environment }
+          headers: { 'X-Vault-Environment' => environment }
         )
 
         return nil unless response.is_a?(Net::HTTPSuccess)
@@ -24,7 +24,7 @@ module BrainzLab
         data = JSON.parse(response.body, symbolize_names: true)
         data[:value]
       rescue StandardError => e
-        log_error("get", e)
+        log_error('get', e)
         nil
       end
 
@@ -38,22 +38,22 @@ module BrainzLab
 
         response = request(
           :post,
-          "/api/v1/secrets",
-          headers: { "X-Vault-Environment" => environment },
+          '/api/v1/secrets',
+          headers: { 'X-Vault-Environment' => environment },
           body: body
         )
 
         response.is_a?(Net::HTTPSuccess) || response.is_a?(Net::HTTPCreated)
       rescue StandardError => e
-        log_error("set", e)
+        log_error('set', e)
         false
       end
 
       def list(environment:)
         response = request(
           :get,
-          "/api/v1/secrets",
-          headers: { "X-Vault-Environment" => environment }
+          '/api/v1/secrets',
+          headers: { 'X-Vault-Environment' => environment }
         )
 
         return [] unless response.is_a?(Net::HTTPSuccess)
@@ -61,7 +61,7 @@ module BrainzLab
         data = JSON.parse(response.body, symbolize_names: true)
         data[:secrets] || []
       rescue StandardError => e
-        log_error("list", e)
+        log_error('list', e)
         []
       end
 
@@ -69,7 +69,7 @@ module BrainzLab
         response = request(:delete, "/api/v1/secrets/#{CGI.escape(key)}")
         response.is_a?(Net::HTTPSuccess) || response.is_a?(Net::HTTPNoContent)
       rescue StandardError => e
-        log_error("delete", e)
+        log_error('delete', e)
         false
       end
 
@@ -77,8 +77,8 @@ module BrainzLab
         params = { format: format }
         response = request(
           :get,
-          "/api/v1/sync/export",
-          headers: { "X-Vault-Environment" => environment },
+          '/api/v1/sync/export',
+          headers: { 'X-Vault-Environment' => environment },
           params: params
         )
 
@@ -92,28 +92,28 @@ module BrainzLab
           response.body
         end
       rescue StandardError => e
-        log_error("export", e)
-        format == :json ? {} : ""
+        log_error('export', e)
+        format == :json ? {} : ''
       end
 
       def provision(project_id:, app_name:)
         response = request(
           :post,
-          "/api/v1/projects/provision",
+          '/api/v1/projects/provision',
           body: { project_id: project_id, app_name: app_name },
           use_service_key: true
         )
 
         response.is_a?(Net::HTTPSuccess) || response.is_a?(Net::HTTPCreated)
       rescue StandardError => e
-        log_error("provision", e)
+        log_error('provision', e)
         false
       end
 
       # Get all provider keys for the current project
       # Returns a hash of provider => decrypted_key
       def get_provider_keys
-        response = request(:get, "/api/v1/provider_keys/bulk")
+        response = request(:get, '/api/v1/provider_keys/bulk')
 
         return {} unless response.is_a?(Net::HTTPSuccess)
 
@@ -125,15 +125,15 @@ module BrainzLab
         end
         keys
       rescue StandardError => e
-        log_error("get_provider_keys", e)
+        log_error('get_provider_keys', e)
         {}
       end
 
       # Get a specific provider key
-      def get_provider_key(provider:, model_type: "llm")
+      def get_provider_key(provider:, model_type: 'llm')
         response = request(
           :get,
-          "/api/v1/provider_keys/resolve",
+          '/api/v1/provider_keys/resolve',
           params: { provider: provider, model_type: model_type }
         )
 
@@ -142,7 +142,7 @@ module BrainzLab
         data = JSON.parse(response.body, symbolize_names: true)
         data[:key]
       rescue StandardError => e
-        log_error("get_provider_key", e)
+        log_error('get_provider_key', e)
         nil
       end
 
@@ -151,35 +151,33 @@ module BrainzLab
       def request(method, path, headers: {}, body: nil, params: nil, use_service_key: false)
         uri = URI.parse("#{@base_url}#{path}")
 
-        if params
-          uri.query = URI.encode_www_form(params)
-        end
+        uri.query = URI.encode_www_form(params) if params
 
         http = Net::HTTP.new(uri.host, uri.port)
-        http.use_ssl = uri.scheme == "https"
+        http.use_ssl = uri.scheme == 'https'
         http.open_timeout = 10
         http.read_timeout = 30
 
         request = case method
-        when :get
-          Net::HTTP::Get.new(uri)
-        when :post
-          Net::HTTP::Post.new(uri)
-        when :put
-          Net::HTTP::Put.new(uri)
-        when :delete
-          Net::HTTP::Delete.new(uri)
-        end
+                  when :get
+                    Net::HTTP::Get.new(uri)
+                  when :post
+                    Net::HTTP::Post.new(uri)
+                  when :put
+                    Net::HTTP::Put.new(uri)
+                  when :delete
+                    Net::HTTP::Delete.new(uri)
+                  end
 
         # Set headers
-        request["Content-Type"] = "application/json"
-        request["Accept"] = "application/json"
+        request['Content-Type'] = 'application/json'
+        request['Accept'] = 'application/json'
 
         if use_service_key
-          request["X-Service-Key"] = @config.vault_master_key || @config.secret_key
+          request['X-Service-Key'] = @config.vault_master_key || @config.secret_key
         else
           auth_key = @config.vault_api_key || @config.secret_key
-          request["Authorization"] = "Bearer #{auth_key}" if auth_key
+          request['Authorization'] = "Bearer #{auth_key}" if auth_key
         end
 
         headers.each { |k, v| request[k] = v }

data/lib/brainzlab/vault/provisioner.rb

@@ -39,7 +39,7 @@ module BrainzLab
 
       def detect_project_id
         # Try environment variable first
-        return ENV["BRAINZLAB_PROJECT_ID"] if ENV["BRAINZLAB_PROJECT_ID"]
+        return ENV['BRAINZLAB_PROJECT_ID'] if ENV['BRAINZLAB_PROJECT_ID']
 
         # Could also detect from Platform API if we have a secret key
         nil

data/lib/brainzlab/vault.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require_relative "vault/client"
-require_relative "vault/cache"
-require_relative "vault/provisioner"
+require_relative 'vault/client'
+require_relative 'vault/cache'
+require_relative 'vault/provisioner'
 
 module BrainzLab
   module Vault
@@ -35,11 +35,11 @@ module BrainzLab
         secrets = export(environment: env, format: :json)
         secrets.each do |key, value|
           key_str = key.to_s
-          if overwrite || !ENV.key?(key_str)
-            ENV[key_str] = value.to_s
-            loaded[key_str] = value
-            BrainzLab.debug_log("[Vault] Loaded #{key_str}")
-          end
+          next unless overwrite || !ENV.key?(key_str)
+
+          ENV[key_str] = value.to_s
+          loaded[key_str] = value
+          BrainzLab.debug_log("[Vault] Loaded #{key_str}")
         end
 
         # Load provider keys (OpenAI, Anthropic, etc.)
@@ -67,11 +67,11 @@ module BrainzLab
 
         provider_keys.each do |provider, key|
           env_var = "#{provider.to_s.upcase}_API_KEY"
-          if overwrite || !ENV.key?(env_var)
-            ENV[env_var] = key
-            loaded[env_var] = key
-            BrainzLab.debug_log("[Vault] Loaded provider key: #{env_var}")
-          end
+          next unless overwrite || !ENV.key?(env_var)
+
+          ENV[env_var] = key
+          loaded[env_var] = key
+          BrainzLab.debug_log("[Vault] Loaded provider key: #{env_var}")
         end
 
         loaded
@@ -84,7 +84,7 @@ module BrainzLab
       # @param provider [String, Symbol] Provider name (openai, anthropic, etc.)
       # @param model_type [String] Model type (llm, embedding, etc.)
       # @return [String, nil] The API key
-      def provider_key(provider, model_type: "llm")
+      def provider_key(provider, model_type: 'llm')
         return nil unless enabled?
 
         ensure_provisioned!
@@ -113,9 +113,7 @@ module BrainzLab
         cache_key = "#{env}:#{key}"
 
         # Check cache first
-        if BrainzLab.configuration.vault_cache_enabled && cache.has?(cache_key)
-          return cache.get(cache_key)
-        end
+        return cache.get(cache_key) if BrainzLab.configuration.vault_cache_enabled && cache.has?(cache_key)
 
         value = client.get(key, environment: env)
 
@@ -144,9 +142,7 @@ module BrainzLab
         result = client.set(key, value, environment: env, description: description, note: note)
 
         # Invalidate cache
-        if result && BrainzLab.configuration.vault_cache_enabled
-          cache.delete("#{env}:#{key}")
-        end
+        cache.delete("#{env}:#{key}") if result && BrainzLab.configuration.vault_cache_enabled
 
         result
       end
@@ -176,9 +172,7 @@ module BrainzLab
         result = client.delete(key)
 
         # Invalidate all environment caches for this key
-        if result && BrainzLab.configuration.vault_cache_enabled
-          cache.delete_pattern("*:#{key}")
-        end
+        cache.delete_pattern("*:#{key}") if result && BrainzLab.configuration.vault_cache_enabled
 
         result
       end
@@ -207,9 +201,9 @@ module BrainzLab
 
         # Fall back to environment variable
         if env_var
-          ENV[env_var]
+          ENV.fetch(env_var, nil)
         else
-          ENV[key]
+          ENV.fetch(key, nil)
         end
       end
 

data/lib/brainzlab/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module BrainzLab
-  VERSION = "0.1.2"
+  VERSION = '0.1.3'
 end

data/lib/brainzlab/vision/client.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require "net/http"
-require "uri"
-require "json"
+require 'net/http'
+require 'uri'
+require 'json'
 
 module BrainzLab
   module Vision
@@ -22,7 +22,7 @@ module BrainzLab
         payload[:model] = model if model
         payload[:browser_provider] = browser_provider if browser_provider
 
-        post("/mcp/tools/vision_task", payload)
+        post('/mcp/tools/vision_task', payload)
       end
 
       # Create a browser session
@@ -32,7 +32,7 @@ module BrainzLab
         payload[:viewport] = viewport if viewport
         payload[:browser_provider] = browser_provider if browser_provider
 
-        post("/mcp/tools/vision_session_create", payload)
+        post('/mcp/tools/vision_session_create', payload)
       end
 
       # Perform an AI-powered action
@@ -43,7 +43,7 @@ module BrainzLab
         }
         payload[:model] = model if model
 
-        post("/mcp/tools/vision_ai_action", payload)
+        post('/mcp/tools/vision_ai_action', payload)
       end
 
       # Perform a direct browser action
@@ -55,7 +55,7 @@ module BrainzLab
         payload[:selector] = selector if selector
         payload[:value] = value if value
 
-        post("/mcp/tools/vision_perform", payload)
+        post('/mcp/tools/vision_perform', payload)
       end
 
       # Extract structured data
@@ -66,20 +66,20 @@ module BrainzLab
         }
         payload[:instruction] = instruction if instruction
 
-        post("/mcp/tools/vision_extract", payload)
+        post('/mcp/tools/vision_extract', payload)
       end
 
       # Close a session
       def close_session(session_id:)
-        post("/mcp/tools/vision_session_close", { session_id: session_id })
+        post('/mcp/tools/vision_session_close', { session_id: session_id })
       end
 
       # Take a screenshot
       def screenshot(session_id:, full_page: true)
-        post("/mcp/tools/vision_screenshot", {
-          session_id: session_id,
-          full_page: full_page
-        })
+        post('/mcp/tools/vision_screenshot', {
+               session_id: session_id,
+               full_page: full_page
+             })
       end
 
       private
@@ -87,9 +87,9 @@ module BrainzLab
       def post(path, payload)
         uri = URI.parse("#{@config.vision_url}#{path}")
         request = Net::HTTP::Post.new(uri)
-        request["Content-Type"] = "application/json"
-        request["Authorization"] = "Bearer #{auth_key}"
-        request["User-Agent"] = "brainzlab-sdk-ruby/#{BrainzLab::VERSION}"
+        request['Content-Type'] = 'application/json'
+        request['Authorization'] = "Bearer #{auth_key}"
+        request['User-Agent'] = "brainzlab-sdk-ruby/#{BrainzLab::VERSION}"
         request.body = JSON.generate(payload)
 
         response = execute(uri, request)
@@ -98,9 +98,9 @@ module BrainzLab
         when Net::HTTPSuccess
           JSON.parse(response.body, symbolize_names: true)
         when Net::HTTPUnauthorized
-          { error: "Unauthorized: Invalid API key" }
+          { error: 'Unauthorized: Invalid API key' }
         when Net::HTTPForbidden
-          { error: "Forbidden: Vision is not enabled for this project" }
+          { error: 'Forbidden: Vision is not enabled for this project' }
         when Net::HTTPNotFound
           { error: "Not found: #{path}" }
         else
@@ -118,9 +118,9 @@ module BrainzLab
 
       def execute(uri, request)
         http = Net::HTTP.new(uri.host, uri.port)
-        http.use_ssl = uri.scheme == "https"
+        http.use_ssl = uri.scheme == 'https'
         http.open_timeout = 10
-        http.read_timeout = 300  # Long timeout for AI tasks
+        http.read_timeout = 300 # Long timeout for AI tasks
         http.request(request)
       end
     end

data/lib/brainzlab/vision/provisioner.rb

@@ -1,14 +1,14 @@
 # frozen_string_literal: true
 
-require "net/http"
-require "uri"
-require "json"
-require "fileutils"
+require 'net/http'
+require 'uri'
+require 'json'
+require 'fileutils'
 
 module BrainzLab
   module Vision
     class Provisioner
-      CACHE_DIR = ENV.fetch("BRAINZLAB_CACHE_DIR") { File.join(Dir.home, ".brainzlab") }
+      CACHE_DIR = ENV.fetch('BRAINZLAB_CACHE_DIR') { File.join(Dir.home, '.brainzlab') }
 
       def initialize(config)
         @config = config
@@ -38,21 +38,21 @@ module BrainzLab
 
       def should_provision?
         if @config.debug
-          log_debug("Checking Vision provision conditions:")
+          log_debug('Checking Vision provision conditions:')
           log_debug("  vision_auto_provision: #{@config.vision_auto_provision}")
           log_debug("  app_name: '#{@config.app_name}'")
-          log_debug("  vision_api_key set: #{@config.vision_api_key.to_s.strip.length > 0}")
-          log_debug("  vision_ingest_key set: #{@config.vision_ingest_key.to_s.strip.length > 0}")
-          log_debug("  vision_master_key set: #{@config.vision_master_key.to_s.strip.length > 0}")
+          log_debug("  vision_api_key set: #{@config.vision_api_key.to_s.strip.length.positive?}")
+          log_debug("  vision_ingest_key set: #{@config.vision_ingest_key.to_s.strip.length.positive?}")
+          log_debug("  vision_master_key set: #{@config.vision_master_key.to_s.strip.length.positive?}")
         end
 
         return false unless @config.vision_auto_provision
-        return false unless @config.app_name.to_s.strip.length > 0
-        return false if @config.vision_api_key.to_s.strip.length > 0
-        return false if @config.vision_ingest_key.to_s.strip.length > 0
-        return false unless @config.vision_master_key.to_s.strip.length > 0
+        return false unless @config.app_name.to_s.strip.length.positive?
+        return false if @config.vision_api_key.to_s.strip.length.positive?
+        return false if @config.vision_ingest_key.to_s.strip.length.positive?
+        return false unless @config.vision_master_key.to_s.strip.length.positive?
 
-        log_debug("Will provision Vision project") if @config.debug
+        log_debug('Will provision Vision project') if @config.debug
         true
       end
 
@@ -67,13 +67,13 @@ module BrainzLab
       def provision_project
         uri = URI.parse("#{@config.vision_url}/api/v1/projects/provision")
         request = Net::HTTP::Post.new(uri)
-        request["Content-Type"] = "application/json"
-        request["X-Master-Key"] = @config.vision_master_key
-        request["User-Agent"] = "brainzlab-sdk-ruby/#{BrainzLab::VERSION}"
+        request['Content-Type'] = 'application/json'
+        request['X-Master-Key'] = @config.vision_master_key
+        request['User-Agent'] = "brainzlab-sdk-ruby/#{BrainzLab::VERSION}"
         request.body = JSON.generate({
-          name: @config.app_name,
-          environment: @config.environment
-        })
+                                       name: @config.app_name,
+                                       environment: @config.environment
+                                     })
 
         response = execute(uri, request)
         return nil unless response.is_a?(Net::HTTPSuccess)
@@ -120,7 +120,7 @@ module BrainzLab
 
       def execute(uri, request)
         http = Net::HTTP.new(uri.host, uri.port)
-        http.use_ssl = uri.scheme == "https"
+        http.use_ssl = uri.scheme == 'https'
         http.open_timeout = 5
         http.read_timeout = 10
         http.request(request)

data/lib/brainzlab/vision.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative "vision/client"
-require_relative "vision/provisioner"
+require_relative 'vision/client'
+require_relative 'vision/provisioner'
 
 module BrainzLab
   module Vision
@@ -16,10 +16,10 @@ module BrainzLab
       # @return [Hash] Task result with extracted data
       def execute_task(instruction:, start_url:, model: nil, browser_provider: nil, max_steps: 50, timeout: 300)
         config = BrainzLab.configuration
-        return { error: "Vision is not enabled" } unless config.vision_enabled
+        return { error: 'Vision is not enabled' } unless config.vision_enabled
 
         ensure_provisioned!
-        return { error: "Vision credentials not configured" } unless config.vision_valid?
+        return { error: 'Vision credentials not configured' } unless config.vision_valid?
 
         client.execute_task(
           instruction: instruction,
@@ -38,10 +38,10 @@ module BrainzLab
       # @return [Hash] Session info with session_id
       def create_session(url: nil, viewport: nil, browser_provider: nil)
         config = BrainzLab.configuration
-        return { error: "Vision is not enabled" } unless config.vision_enabled
+        return { error: 'Vision is not enabled' } unless config.vision_enabled
 
         ensure_provisioned!
-        return { error: "Vision credentials not configured" } unless config.vision_valid?
+        return { error: 'Vision credentials not configured' } unless config.vision_valid?
 
         client.create_session(
           url: url,
@@ -57,8 +57,8 @@ module BrainzLab
       # @return [Hash] Action result
       def ai_action(session_id:, instruction:, model: nil)
         config = BrainzLab.configuration
-        return { error: "Vision is not enabled" } unless config.vision_enabled
-        return { error: "Vision credentials not configured" } unless config.vision_valid?
+        return { error: 'Vision is not enabled' } unless config.vision_enabled
+        return { error: 'Vision credentials not configured' } unless config.vision_valid?
 
         client.ai_action(
           session_id: session_id,
@@ -75,8 +75,8 @@ module BrainzLab
       # @return [Hash] Action result
       def perform(session_id:, action:, selector: nil, value: nil)
         config = BrainzLab.configuration
-        return { error: "Vision is not enabled" } unless config.vision_enabled
-        return { error: "Vision credentials not configured" } unless config.vision_valid?
+        return { error: 'Vision is not enabled' } unless config.vision_enabled
+        return { error: 'Vision credentials not configured' } unless config.vision_valid?
 
         client.perform(
           session_id: session_id,
@@ -93,8 +93,8 @@ module BrainzLab
       # @return [Hash] Extracted data
       def extract(session_id:, schema:, instruction: nil)
         config = BrainzLab.configuration
-        return { error: "Vision is not enabled" } unless config.vision_enabled
-        return { error: "Vision credentials not configured" } unless config.vision_valid?
+        return { error: 'Vision is not enabled' } unless config.vision_enabled
+        return { error: 'Vision credentials not configured' } unless config.vision_valid?
 
         client.extract(
           session_id: session_id,
@@ -108,8 +108,8 @@ module BrainzLab
       # @return [Hash] Close result
       def close_session(session_id:)
         config = BrainzLab.configuration
-        return { error: "Vision is not enabled" } unless config.vision_enabled
-        return { error: "Vision credentials not configured" } unless config.vision_valid?
+        return { error: 'Vision is not enabled' } unless config.vision_enabled
+        return { error: 'Vision credentials not configured' } unless config.vision_valid?
 
         client.close_session(session_id: session_id)
       end
@@ -120,8 +120,8 @@ module BrainzLab
       # @return [Hash] Screenshot data
       def screenshot(session_id:, full_page: true)
         config = BrainzLab.configuration
-        return { error: "Vision is not enabled" } unless config.vision_enabled
-        return { error: "Vision credentials not configured" } unless config.vision_valid?
+        return { error: 'Vision is not enabled' } unless config.vision_enabled
+        return { error: 'Vision credentials not configured' } unless config.vision_valid?
 
         client.screenshot(session_id: session_id, full_page: full_page)
       end
@@ -129,9 +129,7 @@ module BrainzLab
       # Ensure project is auto-provisioned
       def ensure_provisioned!
         config = BrainzLab.configuration
-        if config.debug
-          puts "[BrainzLab::Debug] Vision.ensure_provisioned! called, @provisioned=#{@provisioned}"
-        end
+        puts "[BrainzLab::Debug] Vision.ensure_provisioned! called, @provisioned=#{@provisioned}" if config.debug
 
         return if @provisioned
 

data/lib/brainzlab-sdk.rb

@@ -1,3 +1,3 @@
 # frozen_string_literal: true
 
-require_relative "brainzlab"
+require_relative 'brainzlab'