brainzlab 0.1.0 → 0.1.2

data/lib/brainzlab/vision/provisioner.rb

@@ -0,0 +1,136 @@
+# frozen_string_literal: true
+
+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") }
+
+      def initialize(config)
+        @config = config
+      end
+
+      def ensure_project!
+        return unless should_provision?
+
+        # Try cached credentials first
+        if (cached = load_cached_credentials)
+          apply_credentials(cached)
+          return cached
+        end
+
+        # Provision new project
+        project = provision_project
+        return unless project
+
+        # Cache and apply credentials
+        cache_credentials(project)
+        apply_credentials(project)
+
+        project
+      end
+
+      private
+
+      def should_provision?
+        if @config.debug
+          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}")
+        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
+
+        log_debug("Will provision Vision project") if @config.debug
+        true
+      end
+
+      def log_debug(message)
+        if @config.logger
+          @config.logger.info("[BrainzLab::Debug] #{message}")
+        else
+          puts "[BrainzLab::Debug] #{message}"
+        end
+      end
+
+      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.body = JSON.generate({
+          name: @config.app_name,
+          environment: @config.environment
+        })
+
+        response = execute(uri, request)
+        return nil unless response.is_a?(Net::HTTPSuccess)
+
+        JSON.parse(response.body, symbolize_names: true)
+      rescue StandardError => e
+        log_error("Failed to provision Vision project: #{e.message}")
+        nil
+      end
+
+      def load_cached_credentials
+        path = cache_file_path
+        return nil unless File.exist?(path)
+
+        data = JSON.parse(File.read(path), symbolize_names: true)
+
+        # Validate cached data has required keys
+        return nil unless data[:ingest_key] || data[:api_key]
+
+        data
+      rescue StandardError => e
+        log_error("Failed to load cached Vision credentials: #{e.message}")
+        nil
+      end
+
+      def cache_credentials(project)
+        FileUtils.mkdir_p(CACHE_DIR)
+        File.write(cache_file_path, JSON.generate(project))
+      rescue StandardError => e
+        log_error("Failed to cache Vision credentials: #{e.message}")
+      end
+
+      def cache_file_path
+        File.join(CACHE_DIR, "#{@config.app_name}.vision.json")
+      end
+
+      def apply_credentials(project)
+        @config.vision_ingest_key = project[:ingest_key]
+        @config.vision_api_key = project[:api_key]
+
+        # Also set service name from app_name if not already set
+        @config.service ||= @config.app_name
+      end
+
+      def execute(uri, request)
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = uri.scheme == "https"
+        http.open_timeout = 5
+        http.read_timeout = 10
+        http.request(request)
+      end
+
+      def log_error(message)
+        return unless @config.logger
+
+        @config.logger.error("[BrainzLab] #{message}")
+      end
+    end
+  end
+end

data/lib/brainzlab/vision.rb

@@ -0,0 +1,157 @@
+# frozen_string_literal: true
+
+require_relative "vision/client"
+require_relative "vision/provisioner"
+
+module BrainzLab
+  module Vision
+    class << self
+      # Execute an autonomous AI task
+      # @param instruction [String] Natural language instruction for the AI
+      # @param start_url [String] URL to start from
+      # @param model [String] LLM model to use (default: claude-sonnet-4)
+      # @param browser_provider [String] Browser provider (default: local)
+      # @param max_steps [Integer] Maximum steps to execute (default: 50)
+      # @param timeout [Integer] Timeout in seconds (default: 300)
+      # @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
+
+        ensure_provisioned!
+        return { error: "Vision credentials not configured" } unless config.vision_valid?
+
+        client.execute_task(
+          instruction: instruction,
+          start_url: start_url,
+          model: model,
+          browser_provider: browser_provider,
+          max_steps: max_steps,
+          timeout: timeout
+        )
+      end
+
+      # Create a browser session
+      # @param url [String] Optional initial URL
+      # @param viewport [Hash] Viewport dimensions { width:, height: }
+      # @param browser_provider [String] Browser provider to use
+      # @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
+
+        ensure_provisioned!
+        return { error: "Vision credentials not configured" } unless config.vision_valid?
+
+        client.create_session(
+          url: url,
+          viewport: viewport,
+          browser_provider: browser_provider
+        )
+      end
+
+      # Perform an AI-powered action in a session
+      # @param session_id [String] Session ID
+      # @param instruction [String] Natural language instruction
+      # @param model [String] LLM model to use
+      # @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?
+
+        client.ai_action(
+          session_id: session_id,
+          instruction: instruction,
+          model: model
+        )
+      end
+
+      # Perform a direct browser action
+      # @param session_id [String] Session ID
+      # @param action [Symbol] Action type (:click, :type, :scroll, etc.)
+      # @param selector [String] Element selector
+      # @param value [String] Value for type/fill actions
+      # @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?
+
+        client.perform(
+          session_id: session_id,
+          action: action,
+          selector: selector,
+          value: value
+        )
+      end
+
+      # Extract structured data from the page
+      # @param session_id [String] Session ID
+      # @param schema [Hash] JSON schema for extraction
+      # @param instruction [String] Optional instruction for extraction
+      # @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?
+
+        client.extract(
+          session_id: session_id,
+          schema: schema,
+          instruction: instruction
+        )
+      end
+
+      # Close a browser session
+      # @param session_id [String] Session ID
+      # @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?
+
+        client.close_session(session_id: session_id)
+      end
+
+      # Take a screenshot
+      # @param session_id [String] Session ID
+      # @param full_page [Boolean] Capture full page (default: true)
+      # @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?
+
+        client.screenshot(session_id: session_id, full_page: full_page)
+      end
+
+      # Ensure project is auto-provisioned
+      def ensure_provisioned!
+        config = BrainzLab.configuration
+        if config.debug
+          puts "[BrainzLab::Debug] Vision.ensure_provisioned! called, @provisioned=#{@provisioned}"
+        end
+
+        return if @provisioned
+
+        @provisioned = true
+        provisioner.ensure_project!
+      end
+
+      def provisioner
+        @provisioner ||= Provisioner.new(BrainzLab.configuration)
+      end
+
+      def client
+        @client ||= Client.new(BrainzLab.configuration)
+      end
+
+      def reset!
+        @client = nil
+        @provisioner = nil
+        @provisioned = false
+      end
+    end
+  end
+end

data/lib/brainzlab.rb

@@ -6,7 +6,18 @@ require_relative "brainzlab/context"
 require_relative "brainzlab/recall"
 require_relative "brainzlab/reflex"
 require_relative "brainzlab/pulse"
+require_relative "brainzlab/flux"
+require_relative "brainzlab/signal"
+require_relative "brainzlab/vault"
+require_relative "brainzlab/vision"
+require_relative "brainzlab/cortex"
+require_relative "brainzlab/beacon"
+require_relative "brainzlab/nerve"
+require_relative "brainzlab/dendrite"
+require_relative "brainzlab/sentinel"
+require_relative "brainzlab/synapse"
 require_relative "brainzlab/instrumentation"
+require_relative "brainzlab/utilities"
 
 module BrainzLab
   class << self
@@ -24,6 +35,16 @@ module BrainzLab
       Recall.reset!
       Reflex.reset!
       Pulse.reset!
+      Flux.reset!
+      Signal.reset!
+      Vault.reset!
+      Vision.reset!
+      Cortex.reset!
+      Beacon.reset!
+      Nerve.reset!
+      Dendrite.reset!
+      Sentinel.reset!
+      Synapse.reset!
     end
 
     # Context management
@@ -102,6 +123,86 @@ module BrainzLab
         )
       end
 
+      # Check Flux
+      if configuration.flux_enabled
+        results[:services][:flux] = check_service_health(
+          url: configuration.flux_url,
+          name: 'Flux'
+        )
+      end
+
+      # Check Signal
+      if configuration.signal_enabled
+        results[:services][:signal] = check_service_health(
+          url: configuration.signal_url,
+          name: 'Signal'
+        )
+      end
+
+      # Check Vault
+      if configuration.vault_enabled
+        results[:services][:vault] = check_service_health(
+          url: configuration.vault_url,
+          name: 'Vault'
+        )
+      end
+
+      # Check Vision
+      if configuration.vision_enabled
+        results[:services][:vision] = check_service_health(
+          url: configuration.vision_url,
+          name: 'Vision'
+        )
+      end
+
+      # Check Cortex
+      if configuration.cortex_enabled
+        results[:services][:cortex] = check_service_health(
+          url: configuration.cortex_url,
+          name: 'Cortex'
+        )
+      end
+
+      # Check Beacon
+      if configuration.beacon_enabled
+        results[:services][:beacon] = check_service_health(
+          url: configuration.beacon_url,
+          name: 'Beacon'
+        )
+      end
+
+      # Check Nerve
+      if configuration.nerve_enabled
+        results[:services][:nerve] = check_service_health(
+          url: configuration.nerve_url,
+          name: 'Nerve'
+        )
+      end
+
+      # Check Dendrite
+      if configuration.dendrite_enabled
+        results[:services][:dendrite] = check_service_health(
+          url: configuration.dendrite_url,
+          name: 'Dendrite'
+        )
+      end
+
+      # Check Sentinel
+      if configuration.sentinel_enabled
+        results[:services][:sentinel] = check_service_health(
+          url: configuration.sentinel_url,
+          name: 'Sentinel'
+        )
+      end
+
+      # Check Synapse
+      if configuration.synapse_enabled
+        results[:services][:synapse] = check_service_health(
+          url: configuration.synapse_url,
+          name: 'Synapse'
+        )
+      end
+
       # Overall status
       has_failure = results[:services].values.any? { |s| s[:status] == 'error' }
       results[:status] = has_failure ? 'degraded' : 'ok'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: brainzlab
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Brainz Lab
@@ -93,21 +93,59 @@ files:
 - README.md
 - lib/brainzlab-sdk.rb
 - lib/brainzlab.rb
+- lib/brainzlab/beacon.rb
+- lib/brainzlab/beacon/client.rb
+- lib/brainzlab/beacon/provisioner.rb
 - lib/brainzlab/configuration.rb
 - lib/brainzlab/context.rb
+- lib/brainzlab/cortex.rb
+- lib/brainzlab/cortex/cache.rb
+- lib/brainzlab/cortex/client.rb
+- lib/brainzlab/cortex/provisioner.rb
+- lib/brainzlab/dendrite.rb
+- lib/brainzlab/dendrite/client.rb
+- lib/brainzlab/dendrite/provisioner.rb
+- lib/brainzlab/devtools.rb
+- lib/brainzlab/devtools/assets/devtools.css
+- lib/brainzlab/devtools/assets/devtools.js
+- lib/brainzlab/devtools/assets/logo.svg
+- lib/brainzlab/devtools/assets/templates/debug_panel.html.erb
+- lib/brainzlab/devtools/assets/templates/error_page.html.erb
+- lib/brainzlab/devtools/data/collector.rb
+- lib/brainzlab/devtools/middleware/asset_server.rb
+- lib/brainzlab/devtools/middleware/database_handler.rb
+- lib/brainzlab/devtools/middleware/debug_panel.rb
+- lib/brainzlab/devtools/middleware/error_page.rb
+- lib/brainzlab/devtools/renderers/debug_panel_renderer.rb
+- lib/brainzlab/devtools/renderers/error_page_renderer.rb
+- lib/brainzlab/flux.rb
+- lib/brainzlab/flux/buffer.rb
+- lib/brainzlab/flux/client.rb
+- lib/brainzlab/flux/provisioner.rb
 - lib/brainzlab/instrumentation.rb
 - lib/brainzlab/instrumentation/action_mailer.rb
 - lib/brainzlab/instrumentation/active_record.rb
+- lib/brainzlab/instrumentation/aws.rb
+- lib/brainzlab/instrumentation/dalli.rb
 - lib/brainzlab/instrumentation/delayed_job.rb
 - lib/brainzlab/instrumentation/elasticsearch.rb
+- lib/brainzlab/instrumentation/excon.rb
 - lib/brainzlab/instrumentation/faraday.rb
+- lib/brainzlab/instrumentation/good_job.rb
 - lib/brainzlab/instrumentation/grape.rb
 - lib/brainzlab/instrumentation/graphql.rb
 - lib/brainzlab/instrumentation/httparty.rb
 - lib/brainzlab/instrumentation/mongodb.rb
 - lib/brainzlab/instrumentation/net_http.rb
 - lib/brainzlab/instrumentation/redis.rb
+- lib/brainzlab/instrumentation/resque.rb
 - lib/brainzlab/instrumentation/sidekiq.rb
+- lib/brainzlab/instrumentation/solid_queue.rb
+- lib/brainzlab/instrumentation/stripe.rb
+- lib/brainzlab/instrumentation/typhoeus.rb
+- lib/brainzlab/nerve.rb
+- lib/brainzlab/nerve/client.rb
+- lib/brainzlab/nerve/provisioner.rb
 - lib/brainzlab/pulse.rb
 - lib/brainzlab/pulse/client.rb
 - lib/brainzlab/pulse/instrumentation.rb
@@ -126,7 +164,28 @@ files:
 - lib/brainzlab/reflex/breadcrumbs.rb
 - lib/brainzlab/reflex/client.rb
 - lib/brainzlab/reflex/provisioner.rb
+- lib/brainzlab/sentinel.rb
+- lib/brainzlab/sentinel/client.rb
+- lib/brainzlab/sentinel/provisioner.rb
+- lib/brainzlab/signal.rb
+- lib/brainzlab/signal/client.rb
+- lib/brainzlab/signal/provisioner.rb
+- lib/brainzlab/synapse.rb
+- lib/brainzlab/synapse/client.rb
+- lib/brainzlab/synapse/provisioner.rb
+- lib/brainzlab/utilities.rb
+- lib/brainzlab/utilities/circuit_breaker.rb
+- lib/brainzlab/utilities/health_check.rb
+- lib/brainzlab/utilities/log_formatter.rb
+- lib/brainzlab/utilities/rate_limiter.rb
+- lib/brainzlab/vault.rb
+- lib/brainzlab/vault/cache.rb
+- lib/brainzlab/vault/client.rb
+- lib/brainzlab/vault/provisioner.rb
 - lib/brainzlab/version.rb
+- lib/brainzlab/vision.rb
+- lib/brainzlab/vision/client.rb
+- lib/brainzlab/vision/provisioner.rb
 - lib/generators/brainzlab/install/install_generator.rb
 - lib/generators/brainzlab/install/templates/brainzlab.rb.tt
 homepage: https://brainzlab.ai
@@ -138,6 +197,7 @@ metadata:
   changelog_uri: https://github.com/brainz-lab/brainzlab-ruby/blob/main/CHANGELOG.md
   documentation_uri: https://docs.brainzlab.ai/sdk/ruby
   rubygems_mfa_required: 'true'
+  github_repo: ssh://github.com/brainz-lab/brainzlab-ruby
 rdoc_options: []
 require_paths:
 - lib
@@ -145,7 +205,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.1.0
+      version: 3.2.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="