exa-ai 0.3.1 → 0.4.0

data/exe/exa-ai-enrichment-delete

@@ -0,0 +1,121 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "exa-ai"
+
+# Parse arguments
+webset_id = nil
+enrichment_id = nil
+force = false
+api_key = nil
+output_format = "json"
+
+args = ARGV.dup
+while args.any?
+  arg = args.shift
+  case arg
+  when "--force"
+    force = true
+  when "--api-key"
+    api_key = args.shift
+  when "--output-format"
+    output_format = args.shift
+  when "--help", "-h"
+    puts <<~HELP
+      Usage: exa-ai enrichment-delete <webset_id> <enrichment_id> [OPTIONS]
+
+      Delete an enrichment
+
+      Arguments:
+        webset_id              ID of the webset (required)
+        enrichment_id          ID of the enrichment to delete (required)
+
+      Options:
+        --force                Skip confirmation prompt
+        --api-key KEY          Exa API key (or set EXA_API_KEY env var)
+        --output-format FMT    Output format: json, pretty, or text (default: json)
+        --help, -h             Show this help message
+
+      Examples:
+        exa-ai enrichment-delete ws_123 enr_456
+        exa-ai enrichment-delete ws_123 enr_456 --force
+    HELP
+    exit 0
+  else
+    # First positional argument is webset_id, second is enrichment_id
+    if webset_id.nil?
+      webset_id = arg
+    elsif enrichment_id.nil?
+      enrichment_id = arg
+    else
+      $stderr.puts "Unknown option: #{arg}"
+      exit 1
+    end
+  end
+end
+
+# Validate
+if webset_id.nil?
+  $stderr.puts "Error: webset_id argument is required"
+  $stderr.puts "Usage: exa-ai enrichment-delete <webset_id> <enrichment_id> [OPTIONS]"
+  $stderr.puts "Try 'exa-ai enrichment-delete --help' for more information"
+  exit 1
+end
+
+if enrichment_id.nil?
+  $stderr.puts "Error: enrichment_id argument is required"
+  $stderr.puts "Usage: exa-ai enrichment-delete <webset_id> <enrichment_id> [OPTIONS]"
+  $stderr.puts "Try 'exa-ai enrichment-delete --help' for more information"
+  exit 1
+end
+
+begin
+  # Resolve API key and format
+  api_key = Exa::CLI::Base.resolve_api_key(api_key)
+  output_format = Exa::CLI::Base.resolve_output_format(output_format)
+
+  # Confirm deletion unless --force is used
+  unless force
+    $stderr.print "Delete enrichment #{enrichment_id} from webset #{webset_id}? (y/N): "
+    response = $stdin.gets.chomp.downcase
+    unless response == "y" || response == "yes"
+      $stderr.puts "Cancelled"
+      exit 0
+    end
+  end
+
+  # Build client
+  client = Exa::CLI::Base.build_client(api_key)
+
+  # Delete enrichment
+  result = client.delete_enrichment(webset_id: webset_id, id: enrichment_id)
+
+  # Format and output
+  output = Exa::CLI::Formatters::EnrichmentFormatter.format(result, output_format)
+  puts output
+
+rescue Exa::NotFound => e
+  $stderr.puts "Enrichment not found: #{e.message}"
+  exit 1
+rescue Exa::ConfigurationError => e
+  $stderr.puts "Configuration error: #{e.message}"
+  exit 1
+rescue Exa::Unauthorized => e
+  $stderr.puts "Authentication error: #{e.message}"
+  $stderr.puts "Check your API key (set EXA_API_KEY or use --api-key)"
+  exit 1
+rescue Exa::ClientError => e
+  $stderr.puts "Client error: #{e.message}"
+  exit 1
+rescue Exa::ServerError => e
+  $stderr.puts "Server error: #{e.message}"
+  $stderr.puts "The Exa API may be experiencing issues. Please try again later."
+  exit 1
+rescue Exa::Error => e
+  $stderr.puts "Error: #{e.message}"
+  exit 1
+rescue StandardError => e
+  $stderr.puts "Unexpected error: #{e.message}"
+  $stderr.puts e.backtrace.first(5) if ENV["DEBUG"]
+  exit 1
+end

data/exe/exa-ai-enrichment-get

@@ -0,0 +1,103 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "exa-ai"
+
+# Parse arguments
+webset_id = nil
+enrichment_id = nil
+api_key = nil
+output_format = "json"
+
+args = ARGV.dup
+while args.any?
+  arg = args.shift
+  case arg
+  when "--api-key"
+    api_key = args.shift
+  when "--output-format"
+    output_format = args.shift
+  when "--help", "-h"
+    puts <<~HELP
+      Usage: exa-ai enrichment-get <webset_id> <enrichment_id> [OPTIONS]
+
+      Retrieve details for an enrichment
+
+      Arguments:
+        webset_id              ID of the webset (required)
+        enrichment_id          ID of the enrichment to retrieve (required)
+
+      Options:
+        --api-key KEY          Exa API key (or set EXA_API_KEY env var)
+        --output-format FMT    Output format: json, pretty, or text (default: json)
+        --help, -h             Show this help message
+
+      Examples:
+        exa-ai enrichment-get ws_123 enr_456
+        exa-ai enrichment-get ws_123 enr_456 --output-format pretty
+    HELP
+    exit 0
+  else
+    # First positional argument is webset_id, second is enrichment_id
+    if webset_id.nil?
+      webset_id = arg
+    elsif enrichment_id.nil?
+      enrichment_id = arg
+    else
+      $stderr.puts "Unknown option: #{arg}"
+      exit 1
+    end
+  end
+end
+
+# Validate
+if webset_id.nil?
+  $stderr.puts "Error: webset_id argument is required"
+  $stderr.puts "Usage: exa-ai enrichment-get <webset_id> <enrichment_id> [OPTIONS]"
+  $stderr.puts "Try 'exa-ai enrichment-get --help' for more information"
+  exit 1
+end
+
+if enrichment_id.nil?
+  $stderr.puts "Error: enrichment_id argument is required"
+  $stderr.puts "Usage: exa-ai enrichment-get <webset_id> <enrichment_id> [OPTIONS]"
+  $stderr.puts "Try 'exa-ai enrichment-get --help' for more information"
+  exit 1
+end
+
+begin
+  # Resolve API key and format
+  api_key = Exa::CLI::Base.resolve_api_key(api_key)
+  output_format = Exa::CLI::Base.resolve_output_format(output_format)
+
+  # Build client
+  client = Exa::CLI::Base.build_client(api_key)
+
+  # Get enrichment
+  enrichment = client.get_enrichment(webset_id: webset_id, id: enrichment_id)
+
+  # Format and output
+  output = Exa::CLI::Formatters::EnrichmentFormatter.format(enrichment, output_format)
+  puts output
+
+rescue Exa::NotFound => e
+  $stderr.puts "Enrichment not found: #{e.message}"
+  exit 1
+rescue Exa::Unauthorized => e
+  $stderr.puts "Authentication error: #{e.message}"
+  $stderr.puts "Check your API key (set EXA_API_KEY or use --api-key)"
+  exit 1
+rescue Exa::ClientError => e
+  $stderr.puts "Client error: #{e.message}"
+  exit 1
+rescue Exa::ServerError => e
+  $stderr.puts "Server error: #{e.message}"
+  exit 1
+rescue Exa::Error => e
+  $stderr.puts "Error: #{e.message}"
+  exit 1
+rescue StandardError => e
+  $stderr.puts "Unexpected error: #{e.message}"
+  $stderr.puts e.backtrace.first(5) if ENV["DEBUG"]
+  exit 1
+end

data/exe/exa-ai-enrichment-list

@@ -0,0 +1,98 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "exa-ai"
+
+# Parse arguments
+webset_id = nil
+api_key = nil
+output_format = "json"
+
+args = ARGV.dup
+while args.any?
+  arg = args.shift
+  case arg
+  when "--api-key"
+    api_key = args.shift
+  when "--output-format"
+    output_format = args.shift
+  when "--help", "-h"
+    puts <<~HELP
+      Usage: exa-ai enrichment-list <webset_id> [OPTIONS]
+
+      List all enrichments for a webset
+
+      Arguments:
+        webset_id              ID of the webset (required)
+
+      Options:
+        --api-key KEY          Exa API key (or set EXA_API_KEY env var)
+        --output-format FMT    Output format: json, pretty, or text (default: json)
+        --help, -h             Show this help message
+
+      Examples:
+        exa-ai enrichment-list ws_123
+        exa-ai enrichment-list ws_123 --output-format text
+    HELP
+    exit 0
+  else
+    # First positional argument is webset_id
+    if webset_id.nil?
+      webset_id = arg
+    else
+      $stderr.puts "Unknown option: #{arg}"
+      exit 1
+    end
+  end
+end
+
+# Validate
+if webset_id.nil?
+  $stderr.puts "Error: webset_id argument is required"
+  $stderr.puts "Usage: exa-ai enrichment-list <webset_id> [OPTIONS]"
+  $stderr.puts "Try 'exa-ai enrichment-list --help' for more information"
+  exit 1
+end
+
+begin
+  # Resolve API key and format
+  api_key = Exa::CLI::Base.resolve_api_key(api_key)
+  output_format = Exa::CLI::Base.resolve_output_format(output_format)
+
+  # Build client
+  client = Exa::CLI::Base.build_client(api_key)
+
+  # Get webset to retrieve enrichments
+  webset = client.get_webset(webset_id)
+
+  # Extract enrichments array (or empty array if none)
+  enrichments_data = webset.enrichments || []
+
+  # Create collection
+  collection = Exa::Resources::WebsetEnrichmentCollection.new(data: enrichments_data)
+
+  # Format and output
+  output = Exa::CLI::Formatters::EnrichmentFormatter.format_collection(collection, output_format)
+  puts output
+
+rescue Exa::NotFound => e
+  $stderr.puts "Webset not found: #{e.message}"
+  exit 1
+rescue Exa::Unauthorized => e
+  $stderr.puts "Authentication error: #{e.message}"
+  $stderr.puts "Check your API key (set EXA_API_KEY or use --api-key)"
+  exit 1
+rescue Exa::ClientError => e
+  $stderr.puts "Client error: #{e.message}"
+  exit 1
+rescue Exa::ServerError => e
+  $stderr.puts "Server error: #{e.message}"
+  exit 1
+rescue Exa::Error => e
+  $stderr.puts "Error: #{e.message}"
+  exit 1
+rescue StandardError => e
+  $stderr.puts "Unexpected error: #{e.message}"
+  $stderr.puts e.backtrace.first(5) if ENV["DEBUG"]
+  exit 1
+end

data/exe/exa-ai-enrichment-update

@@ -0,0 +1,170 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "exa-ai"
+require "json"
+
+# Recursively convert hash keys from strings to symbols
+def deep_symbolize_keys(obj)
+  case obj
+  when Hash
+    obj.each_with_object({}) do |(key, value), result|
+      result[key.to_sym] = deep_symbolize_keys(value)
+    end
+  when Array
+    obj.map { |item| deep_symbolize_keys(item) }
+  else
+    obj
+  end
+end
+
+# Helper to parse JSON string or load from file
+def parse_json_or_file(value)
+  json_data = if value.start_with?("@")
+                file_path = value[1..]
+                JSON.parse(File.read(file_path))
+              else
+                JSON.parse(value)
+              end
+  deep_symbolize_keys(json_data)
+rescue JSON::ParserError => e
+  $stderr.puts "Error: Invalid JSON: #{e.message}"
+  exit 1
+rescue Errno::ENOENT => e
+  $stderr.puts "Error: File not found: #{e.message}"
+  exit 1
+end
+
+# Parse arguments
+webset_id = nil
+enrichment_id = nil
+description = nil
+format = nil
+options = nil
+metadata = nil
+api_key = nil
+output_format = "json"
+
+args = ARGV.dup
+while args.any?
+  arg = args.shift
+  case arg
+  when "--description"
+    description = args.shift
+  when "--format"
+    format = args.shift
+  when "--options"
+    options = parse_json_or_file(args.shift)
+  when "--metadata"
+    metadata = parse_json_or_file(args.shift)
+  when "--api-key"
+    api_key = args.shift
+  when "--output-format"
+    output_format = args.shift
+  when "--help", "-h"
+    puts <<~HELP
+      Usage: exa-ai enrichment-update <webset_id> <enrichment_id> [OPTIONS]
+
+      Update an enrichment's properties
+
+      Arguments:
+        webset_id              ID of the webset (required)
+        enrichment_id          ID of the enrichment to update (required)
+
+      Options:
+        --description TEXT     What to extract
+        --format TYPE          One of: text, url, options
+        --options JSON         Array of {label: "..."} (supports @file.json)
+        --metadata JSON        Custom metadata (supports @file.json)
+        --api-key KEY          Exa API key (or set EXA_API_KEY env var)
+        --output-format FMT    Output format: json, pretty, or text (default: json)
+        --help, -h             Show this help message
+
+      Examples:
+        exa-ai enrichment-update ws_123 enr_456 --description "New description"
+        exa-ai enrichment-update ws_123 enr_456 --metadata '{"priority":"high"}'
+        exa-ai enrichment-update ws_123 enr_456 --options options.json --format options
+    HELP
+    exit 0
+  else
+    # First positional argument is webset_id, second is enrichment_id
+    if webset_id.nil?
+      webset_id = arg
+    elsif enrichment_id.nil?
+      enrichment_id = arg
+    else
+      $stderr.puts "Unknown option: #{arg}"
+      exit 1
+    end
+  end
+end
+
+# Validate required arguments
+if webset_id.nil?
+  $stderr.puts "Error: webset_id argument is required"
+  $stderr.puts "Usage: exa-ai enrichment-update <webset_id> <enrichment_id> [OPTIONS]"
+  $stderr.puts "Try 'exa-ai enrichment-update --help' for more information"
+  exit 1
+end
+
+if enrichment_id.nil?
+  $stderr.puts "Error: enrichment_id argument is required"
+  $stderr.puts "Usage: exa-ai enrichment-update <webset_id> <enrichment_id> [OPTIONS]"
+  $stderr.puts "Try 'exa-ai enrichment-update --help' for more information"
+  exit 1
+end
+
+# Check that at least one update field is provided
+if description.nil? && format.nil? && options.nil? && metadata.nil?
+  $stderr.puts "Error: at least one update field is required (--description, --format, --options, or --metadata)"
+  $stderr.puts "Try 'exa-ai enrichment-update --help' for more information"
+  exit 1
+end
+
+begin
+  # Resolve API key and format
+  api_key = Exa::CLI::Base.resolve_api_key(api_key)
+  output_format = Exa::CLI::Base.resolve_output_format(output_format)
+
+  # Build client
+  client = Exa::CLI::Base.build_client(api_key)
+
+  # Build update params (only include provided fields)
+  update_params = {}
+  update_params[:description] = description if description
+  update_params[:format] = format if format
+  update_params[:options] = options if options
+  update_params[:metadata] = metadata if metadata
+
+  # Update enrichment
+  enrichment = client.update_enrichment(webset_id: webset_id, id: enrichment_id, **update_params)
+
+  # Format and output
+  output = Exa::CLI::Formatters::EnrichmentFormatter.format(enrichment, output_format)
+  puts output
+
+rescue Exa::NotFound => e
+  $stderr.puts "Enrichment not found: #{e.message}"
+  exit 1
+rescue Exa::ConfigurationError => e
+  $stderr.puts "Configuration error: #{e.message}"
+  exit 1
+rescue Exa::Unauthorized => e
+  $stderr.puts "Authentication error: #{e.message}"
+  $stderr.puts "Check your API key (set EXA_API_KEY or use --api-key)"
+  exit 1
+rescue Exa::ClientError => e
+  $stderr.puts "Client error: #{e.message}"
+  exit 1
+rescue Exa::ServerError => e
+  $stderr.puts "Server error: #{e.message}"
+  $stderr.puts "The Exa API may be experiencing issues. Please try again later."
+  exit 1
+rescue Exa::Error => e
+  $stderr.puts "Error: #{e.message}"
+  exit 1
+rescue StandardError => e
+  $stderr.puts "Unexpected error: #{e.message}"
+  $stderr.puts e.backtrace.first(5) if ENV["DEBUG"]
+  exit 1
+end