exa-ai 0.1.0

data/exe/exa-ai-answer

@@ -0,0 +1,131 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Set up load paths
+require "bundler/setup"
+require "exa"
+require "exa/cli/base"
+require "exa/cli/formatters/answer_formatter"
+
+# Parse command-line arguments
+def parse_args(argv)
+  args = {
+    output_format: "json",
+    api_key: nil,
+    text: false
+  }
+
+  # Extract query (first non-flag argument)
+  query_parts = []
+  i = 0
+  while i < argv.length
+    arg = argv[i]
+    case arg
+    when "--text"
+      args[:text] = true
+      i += 1
+    when "--output-schema"
+      args[:output_schema] = argv[i + 1]
+      i += 2
+    when "--api-key"
+      args[:api_key] = argv[i + 1]
+      i += 2
+    when "--output-format"
+      args[:output_format] = argv[i + 1]
+      i += 2
+    when "--help", "-h"
+      puts <<~HELP
+        Usage: exa-api answer QUERY [OPTIONS]
+
+        Generate an answer to a question using Exa AI
+
+        Arguments:
+          QUERY                 Question or query to answer (required)
+
+        Options:
+          --text                Include full text content from sources
+          --output-schema JSON  JSON schema for structured output
+          --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-api answer "What is the capital of France?"
+          exa-api answer "Latest AI breakthroughs" --text
+          exa-api answer "Ruby best practices" --output-format pretty
+          exa-api answer "What is the capital of France?" --output-schema '{"type":"object","properties":{"city":{"type":"string"},"state":{"type":"string"}}}'
+      HELP
+      exit 0
+    else
+      query_parts << arg
+      i += 1
+    end
+  end
+
+  args[:query] = query_parts.join(" ")
+  args
+end
+
+# Main execution
+begin
+  args = parse_args(ARGV)
+
+  # Validate query
+  if args[:query].nil? || args[:query].empty?
+    $stderr.puts "Error: Query is required"
+    $stderr.puts "Run 'exa-api answer --help' for usage information"
+    exit 1
+  end
+
+  # Resolve API key
+  api_key = Exa::CLI::Base.resolve_api_key(args[:api_key])
+
+  # Resolve output format
+  output_format = Exa::CLI::Base.resolve_output_format(args[:output_format])
+
+  # Build client
+  client = Exa::CLI::Base.build_client(api_key)
+
+  # Prepare answer parameters
+  answer_params = {}
+  answer_params[:text] = args[:text] if args[:text]
+
+  # Parse output_schema as JSON if provided
+  if args[:output_schema]
+    begin
+      answer_params[:output_schema] = JSON.parse(args[:output_schema])
+    rescue JSON::ParserError => e
+      $stderr.puts "Error: Invalid JSON in --output-schema: #{e.message}"
+      exit 1
+    end
+  end
+
+  # Execute answer
+  result = client.answer(args[:query], **answer_params)
+
+  # Format and output result
+  output = Exa::CLI::Formatters::AnswerFormatter.format(result, output_format)
+  puts output
+
+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-context

@@ -0,0 +1,104 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "bundler/setup"
+require "exa"
+require "exa/cli/base"
+require "exa/cli/formatters/context_formatter"
+
+# Parse command line arguments
+def parse_args(args)
+  query = nil
+  api_key = nil
+  tokens_num = "dynamic"
+  output_format = nil
+
+  i = 0
+  while i < args.length
+    arg = args[i]
+
+    case arg
+    when "--api-key"
+      i += 1
+      api_key = args[i]
+    when "--tokens-num"
+      i += 1
+      tokens_num = args[i]
+      # Convert to integer if it's not "dynamic"
+      tokens_num = tokens_num.to_i unless tokens_num.downcase == "dynamic"
+    when "--output-format"
+      i += 1
+      output_format = args[i]
+    when "--help", "-h"
+      print_help
+      exit 0
+    else
+      # First non-flag argument is the query
+      query = arg if query.nil?
+    end
+
+    i += 1
+  end
+
+  { query: query, api_key: api_key, tokens_num: tokens_num, output_format: output_format }
+end
+
+def print_help
+  puts "Exa Context - Get code context from repositories"
+  puts ""
+  puts "Usage: exa-api context <query> [options]"
+  puts ""
+  puts "Arguments:"
+  puts "  query                  Search query for code context (required)"
+  puts ""
+  puts "Options:"
+  puts "  --tokens-num NUM       Number of tokens for response (or 'dynamic', default: dynamic)"
+  puts "  --api-key KEY          Exa API key (or set EXA_API_KEY env var)"
+  puts "  --output-format FMT    Output format: json, pretty, or text (default: json)"
+  puts "  --help, -h             Show this help message"
+  puts ""
+  puts "Examples:"
+  puts "  exa-api context 'React hooks'"
+  puts "  exa-api context 'authentication with JWT in Ruby' --tokens-num 5000"
+  puts "  exa-api context 'React hooks' --output-format text"
+end
+
+begin
+  # Parse arguments
+  options = parse_args(ARGV)
+
+  # Validate query
+  unless options[:query]
+    puts "Error: Query argument required"
+    puts ""
+    puts "Run 'exa-api context --help' for usage information."
+    exit 1
+  end
+
+  # Resolve API key
+  api_key = Exa::CLI::Base.resolve_api_key(options[:api_key])
+
+  # Resolve output format
+  output_format = Exa::CLI::Base.resolve_output_format(options[:output_format])
+
+  # Build client
+  client = Exa::CLI::Base.build_client(api_key)
+
+  # Build request parameters
+  params = {}
+  params[:tokens_num] = options[:tokens_num]
+
+  # Call API
+  result = client.context(options[:query], **params)
+
+  # Format and output
+  output = Exa::CLI::Formatters::ContextFormatter.format(result, output_format)
+  puts output
+rescue Exa::Error => e
+  puts "Error: #{e.message}"
+  exit 1
+rescue StandardError => e
+  puts "Unexpected error: #{e.message}"
+  puts e.backtrace.first(5) if ENV["DEBUG"]
+  exit 1
+end

data/exe/exa-ai-get-contents

@@ -0,0 +1,114 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "bundler/setup"
+require "exa"
+require "exa/cli/base"
+require "exa/cli/formatters/contents_formatter"
+
+# Parse command line arguments
+def parse_args(args)
+  ids = []
+  api_key = nil
+  text = false
+  highlights = false
+  summary = false
+  output_format = nil
+
+  i = 0
+  while i < args.length
+    arg = args[i]
+
+    case arg
+    when "--api-key"
+      i += 1
+      api_key = args[i]
+    when "--text"
+      text = true
+    when "--highlights"
+      highlights = true
+    when "--summary"
+      summary = true
+    when "--output-format"
+      i += 1
+      output_format = args[i]
+    when "--help", "-h"
+      print_help
+      exit 0
+    else
+      # First non-flag argument is the IDs (comma-separated or single)
+      if ids.empty?
+        ids_arg = arg
+        ids = ids_arg.include?(",") ? ids_arg.split(",").map(&:strip) : [ids_arg]
+      end
+    end
+
+    i += 1
+  end
+
+  { ids: ids, api_key: api_key, text: text, highlights: highlights, summary: summary, output_format: output_format }
+end
+
+def print_help
+  puts "Exa Get-Contents - Retrieve page contents"
+  puts ""
+  puts "Usage: exa-api get-contents <ids> [options]"
+  puts ""
+  puts "Arguments:"
+  puts "  ids                    Comma-separated list of IDs or URLs (required)"
+  puts ""
+  puts "Options:"
+  puts "  --text                 Include page text in response"
+  puts "  --highlights           Include highlights in response"
+  puts "  --summary              Include summary in response"
+  puts "  --api-key KEY          Exa API key (or set EXA_API_KEY env var)"
+  puts "  --output-format FMT    Output format: json, pretty, or text (default: json)"
+  puts "  --help, -h             Show this help message"
+  puts ""
+  puts "Examples:"
+  puts "  exa-api get-contents 'https://example.com'"
+  puts "  exa-api get-contents 'id1,id2,id3' --text"
+  puts "  exa-api get-contents 'https://example.com' --highlights --output-format pretty"
+end
+
+begin
+  # Parse arguments
+  options = parse_args(ARGV)
+
+  # Validate IDs
+  if options[:ids].empty?
+    puts "Error: IDs argument required"
+    puts ""
+    puts "Run 'exa-api get-contents --help' for usage information."
+    exit 1
+  end
+
+  # Resolve API key
+  api_key = Exa::CLI::Base.resolve_api_key(options[:api_key])
+
+  # Resolve output format
+  output_format = Exa::CLI::Base.resolve_output_format(options[:output_format])
+
+  # Build client
+  client = Exa::CLI::Base.build_client(api_key)
+
+  # Build request parameters
+  params = {}
+  params[:text] = true if options[:text]
+  params[:highlights] = true if options[:highlights]
+  params[:summary] = true if options[:summary]
+
+  # Call API
+  result = client.get_contents(options[:ids], **params)
+
+  # Format and output
+  output = Exa::CLI::Formatters::ContentsFormatter.format(result, output_format)
+  puts output
+rescue Exa::Error => e
+  puts "Error: #{e.message}"
+  exit 1
+rescue StandardError => e
+  puts "Unexpected error: #{e.message}"
+  puts e.backtrace.first(5) if ENV["DEBUG"]
+  exit 1
+end

data/exe/exa-ai-research-get

@@ -0,0 +1,110 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "bundler/setup"
+require "exa"
+
+# Parse command-line arguments
+api_key = nil
+research_id = nil
+events = false
+stream = false
+output_format = "json"
+
+args = ARGV.dup
+while args.any?
+  arg = args.shift
+  case arg
+  when "--api-key"
+    api_key = args.shift
+  when "--events"
+    events = true
+  when "--stream"
+    stream = true
+  when "--output-format"
+    output_format = args.shift
+  when "--help", "-h"
+    puts <<~HELP
+      Usage: exa-api research-get <research_id> [options]
+
+      Get the status and results of a research task.
+
+      Arguments:
+        research_id             ID of the research task to retrieve
+
+      Options:
+        --api-key KEY           Exa API key (or use EXA_API_KEY env var)
+        --events                Include task execution events in response
+        --stream                Stream the response (for real-time updates)
+        --output-format FORMAT  Output format: json, pretty, or text (default: json)
+        --help, -h              Show this help message
+
+      Examples:
+        exa-api research-get research_123
+        exa-api research-get research_123 --events
+        exa-api research-get research_123 --stream
+        exa-api research-get research_123 --output-format pretty
+    HELP
+    exit 0
+  else
+    # First positional argument is the research_id
+    if research_id.nil?
+      research_id = arg
+    else
+      warn "Unknown option: #{arg}"
+      exit 1
+    end
+  end
+end
+
+# Validate required arguments
+if research_id.nil?
+  warn "Error: Research ID argument required"
+  warn "Usage: exa-api research-get <research_id> [options]"
+  warn "Try 'exa-api research-get --help' for more information"
+  exit 1
+end
+
+begin
+  # Resolve API key
+  api_key = Exa::CLI::Base.resolve_api_key(api_key)
+
+  # Build client
+  client = Exa::CLI::Base.build_client(api_key)
+
+  # Build parameters
+  params = {}
+  params[:events] = events if events
+  params[:stream] = stream if stream
+
+  # Call API
+  result = client.research_get(research_id, **params)
+
+  # Format output
+  formatted = Exa::CLI::Formatters::ResearchFormatter.format_task(
+    result,
+    output_format,
+    show_events: events
+  )
+  puts formatted
+
+rescue Exa::NotFound => e
+  warn "Research task not found: #{e.message}"
+  exit 1
+rescue Exa::Unauthorized => e
+  warn "Authentication failed: #{e.message}"
+  warn "Please provide a valid API key via --api-key or EXA_API_KEY environment variable"
+  exit 1
+rescue Exa::ClientError => e
+  warn "Client error: #{e.message}"
+  exit 1
+rescue Exa::ServerError => e
+  warn "Server error: #{e.message}"
+  exit 1
+rescue Exa::Error => e
+  warn "Error: #{e.message}"
+  exit 1
+rescue StandardError => e
+  warn "Unexpected error: #{e.message}"
+  exit 1
+end

data/exe/exa-ai-research-list

@@ -0,0 +1,95 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require "bundler/setup"
+require "exa"
+
+# Parse command-line arguments
+api_key = nil
+cursor = nil
+limit = 10
+output_format = "json"
+
+args = ARGV.dup
+while args.any?
+  arg = args.shift
+  case arg
+  when "--api-key"
+    api_key = args.shift
+  when "--cursor"
+    cursor = args.shift
+  when "--limit"
+    limit = args.shift.to_i
+  when "--output-format"
+    output_format = args.shift
+  when "--help", "-h"
+    puts <<~HELP
+      Usage: exa-api research-list [options]
+
+      List research tasks with cursor-based pagination.
+
+      Options:
+        --api-key KEY           Exa API key (or use EXA_API_KEY env var)
+        --cursor CURSOR         Pagination cursor for next page
+        --limit LIMIT           Number of results per page (default: 10)
+        --output-format FORMAT  Output format: json, pretty, or text (default: json)
+        --help, -h              Show this help message
+
+      Examples:
+        exa-api research-list
+        exa-api research-list --limit 20
+        exa-api research-list --cursor next_page_cursor
+        exa-api research-list --output-format pretty
+    HELP
+    exit 0
+  else
+    warn "Unknown option: #{arg}"
+    exit 1
+  end
+end
+
+begin
+  # Resolve API key
+  api_key = Exa::CLI::Base.resolve_api_key(api_key)
+
+  # Build client
+  client = Exa::CLI::Base.build_client(api_key)
+
+  # Build parameters
+  params = {limit: limit}
+  params[:cursor] = cursor if cursor
+
+  # Call API
+  result = client.research_list(**params)
+
+  # Format output
+  formatted = Exa::CLI::Formatters::ResearchFormatter.format_list(result, output_format)
+  puts formatted
+
+  # Show pagination info if there are more results
+  if result.has_more && result.next_cursor
+    if output_format == "pretty"
+      puts "\n" + "=" * 80
+      puts "More results available. Use --cursor #{result.next_cursor} to get next page."
+    else
+      warn "More results available. Use --cursor #{result.next_cursor} to get next page."
+    end
+  end
+
+rescue Exa::Unauthorized => e
+  warn "Authentication failed: #{e.message}"
+  warn "Please provide a valid API key via --api-key or EXA_API_KEY environment variable"
+  exit 1
+rescue Exa::ClientError => e
+  warn "Client error: #{e.message}"
+  exit 1
+rescue Exa::ServerError => e
+  warn "Server error: #{e.message}"
+  exit 1
+rescue Exa::Error => e
+  warn "Error: #{e.message}"
+  exit 1
+rescue StandardError => e
+  warn "Unexpected error: #{e.message}"
+  exit 1
+end