exa-ai 0.4.1 → 0.6.0

data/exe/exa-ai-research-start

@@ -39,7 +39,7 @@ def parse_args(argv)
       i += 2
     when "--help", "-h"
       puts <<~HELP
-        Usage: exa-api research-start --instructions "TEXT" [OPTIONS]
+        Usage: exa-ai research-start --instructions "TEXT" [OPTIONS]
 
         Start a research task using Exa AI
 
@@ -54,10 +54,10 @@ def parse_args(argv)
           --help, -h            Show this help message
 
         Examples:
-          exa-api research-start --instructions "Find Ruby performance tips"
-          exa-api research-start --instructions "Analyze AI trends" --wait --events
-          exa-api research-start --instructions "Summarize papers" --model exa-research-pro --wait
-          exa-api research-start --instructions "Find stats" --output-schema '{"type":"object"}'
+          exa-ai research-start --instructions "Find Ruby performance tips"
+          exa-ai research-start --instructions "Analyze AI trends" --wait --events
+          exa-ai research-start --instructions "Summarize papers" --model exa-research-pro --wait
+          exa-ai research-start --instructions "Find stats" --output-schema '{"type":"object"}'
       HELP
       exit 0
     else
@@ -75,7 +75,7 @@ begin
   # Validate instructions
   if args[:instructions].nil? || args[:instructions].empty?
     $stderr.puts "Error: --instructions flag is required"
-    $stderr.puts "Run 'exa-api research-start --help' for usage information"
+    $stderr.puts "Run 'exa-ai research-start --help' for usage information"
     exit 1
   end
 
@@ -133,6 +133,7 @@ begin
       # Format and output final result
       output = Exa::CLI::Formatters::ResearchFormatter.format_task(final_task, output_format, show_events: args[:events])
       puts output
+      $stdout.flush
 
       # Exit with error code if task failed
       exit 1 if final_task.failed?
@@ -140,7 +141,7 @@ begin
     rescue Exa::CLI::Polling::TimeoutError => e
       $stderr.puts "\nError: Task did not complete within timeout period (5 minutes)"
       $stderr.puts "Task ID: #{task.research_id}"
-      $stderr.puts "You can check the status later with: exa-api research-get #{task.research_id}"
+      $stderr.puts "You can check the status later with: exa-ai research-get #{task.research_id}"
       exit 1
     end
   else

data/exe/exa-ai-search

@@ -20,7 +20,13 @@ def parse_args(argv)
       args[:num_results] = argv[i + 1].to_i
       i += 2
     when "--type"
-      args[:type] = argv[i + 1]
+      search_type = argv[i + 1]
+      valid_types = ["fast", "deep", "keyword", "auto"]
+      unless valid_types.include?(search_type)
+        $stderr.puts "Error: Search type must be one of: #{valid_types.join(', ')}"
+        exit 1
+      end
+      args[:type] = search_type
       i += 2
     when "--category"
       category = argv[i + 1]
@@ -116,7 +122,7 @@ def parse_args(argv)
       i += 2
     when "--help", "-h"
       puts <<~HELP
-        Usage: exa-api search QUERY [OPTIONS]
+        Usage: exa-ai search QUERY [OPTIONS]
 
         Search the web using Exa AI
 
@@ -125,7 +131,7 @@ def parse_args(argv)
 
         Options:
           --num-results N              Number of results to return (default: 10)
-          --type TYPE                  Search type: keyword, neural, fast, or auto (default: auto)
+          --type TYPE                  Search type: fast, deep, keyword, or auto (default: fast)
           --category CAT               Focus on specific data category
                                        Options: "company", "research paper", "news", "pdf",
                                        "github", "tweet", "personal site", "linkedin profile",
@@ -160,15 +166,15 @@ def parse_args(argv)
           --help, -h                   Show this help message
 
         Examples:
-          exa-api search "ruby programming"
-          exa-api search "machine learning" --num-results 5 --type keyword
-          exa-api search "Latest LLM research" --category "research paper"
-          exa-api search "AI startups" --category company
-          exa-api search "Anthropic" --linkedin company
-          exa-api search "Dario Amodei" --linkedin person
-          exa-api search "AI" --linkedin all
-          exa-api search "AI research" --include-domains arxiv.org,scholar.google.com
-          exa-api search "tutorials" --output-format pretty
+          exa-ai search "ruby programming"
+          exa-ai search "machine learning" --num-results 5 --type deep
+          exa-ai search "Latest LLM research" --category "research paper"
+          exa-ai search "AI startups" --category company
+          exa-ai search "Anthropic" --linkedin company
+          exa-ai search "Dario Amodei" --linkedin person
+          exa-ai search "AI" --linkedin all
+          exa-ai search "AI research" --include-domains arxiv.org,scholar.google.com
+          exa-ai search "tutorials" --output-format pretty
       HELP
       exit 0
     else
@@ -237,7 +243,7 @@ begin
   # Validate query
   if args[:query].nil? || args[:query].empty?
     $stderr.puts "Error: Query is required"
-    $stderr.puts "Run 'exa-api search --help' for usage information"
+    $stderr.puts "Run 'exa-ai search --help' for usage information"
     exit 1
   end
 
@@ -281,6 +287,7 @@ begin
   # Format and output result
   output = Exa::CLI::Formatters::SearchFormatter.format(result, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::ConfigurationError => e
   $stderr.puts "Configuration error: #{e.message}"

data/exe/exa-ai-webset-cancel

@@ -68,6 +68,7 @@ begin
   # Format and output
   output = Exa::CLI::Formatters::WebsetFormatter.format(webset, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::NotFound => e
   $stderr.puts "Webset not found: #{e.message}"

data/exe/exa-ai-webset-create

@@ -46,6 +46,9 @@ def parse_args(argv)
   while i < argv.length
     arg = argv[i]
     case arg
+    when "--import"
+      args[:import] = argv[i + 1]
+      i += 2
     when "--search"
       args[:search] = parse_json_or_file(argv[i + 1])
       i += 2
@@ -72,12 +75,14 @@ def parse_args(argv)
       i += 2
     when "--help", "-h"
       puts <<~HELP
-        Usage: exa-ai webset-create --search JSON [OPTIONS]
+        Usage: exa-ai webset-create (--search JSON | --import ID) [OPTIONS]
 
-        Create a new webset with search criteria
+        Create a new webset from search criteria or an import
 
-        Required:
+        Required (choose one):
           --search JSON              Search configuration (supports @file.json)
+          --import ID                Import or webset ID to create webset from
+                                     (accepts import_* or webset_* IDs)
 
         Options:
           --enrichments JSON         Array of enrichment configs (supports @file.json)
@@ -90,9 +95,17 @@ def parse_args(argv)
           --help, -h                 Show this help message
 
         Examples:
+          # Create webset from search
           exa-ai webset-create --search '{"query":"AI startups","count":10}'
           exa-ai webset-create --search @search.json --enrichments @enrichments.json
           exa-ai webset-create --search @search.json --wait
+
+          # Create webset from import
+          exa-ai webset-create --import import_abc123
+          exa-ai webset-create --import import_def456 --enrichments @enrichments.json
+
+          # Create webset from existing webset
+          exa-ai webset-create --import webset_xyz789
       HELP
       exit 0
     else
@@ -109,8 +122,14 @@ begin
   args = parse_args(ARGV)
 
   # Validate required parameters
-  unless args[:search]
-    $stderr.puts "Error: --search is required"
+  if args[:search] && args[:import]
+    $stderr.puts "Error: Cannot specify both --search and --import"
+    $stderr.puts "Run 'exa-ai webset-create --help' for usage information"
+    exit 1
+  end
+
+  unless args[:search] || args[:import]
+    $stderr.puts "Error: Either --search or --import is required"
     $stderr.puts "Run 'exa-ai webset-create --help' for usage information"
     exit 1
   end
@@ -125,7 +144,25 @@ begin
   client = Exa::CLI::Base.build_client(api_key)
 
   # Prepare webset parameters
-  webset_params = { search: args[:search] }
+  webset_params = {}
+
+  if args[:search]
+    webset_params[:search] = args[:search]
+  elsif args[:import]
+    # Detect source type from ID prefix and convert to API format
+    import_id = args[:import]
+    source = if import_id.start_with?("import_")
+               "import"
+             elsif import_id.start_with?("webset_")
+               "webset"
+             else
+               $stderr.puts "Error: Import ID must start with 'import_' or 'webset_'"
+               exit 1
+             end
+
+    webset_params[:import] = [{ source: source, id: import_id }]
+  end
+
   webset_params[:enrichments] = args[:enrichments] if args[:enrichments]
   webset_params[:exclude] = args[:exclude] if args[:exclude]
   webset_params[:externalId] = args[:external_id] if args[:external_id]
@@ -167,6 +204,7 @@ begin
   # Format and output result
   output = Exa::CLI::Formatters::WebsetFormatter.format(webset, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::ConfigurationError => e
   $stderr.puts "Configuration error: #{e.message}"

data/exe/exa-ai-webset-delete

@@ -82,6 +82,7 @@ begin
   # Format and output
   output = Exa::CLI::Formatters::WebsetFormatter.format(result, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::NotFound => e
   $stderr.puts "Webset not found: #{e.message}"

data/exe/exa-ai-webset-get

@@ -68,6 +68,7 @@ begin
   # Format and output
   output = Exa::CLI::Formatters::WebsetFormatter.format(webset, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::NotFound => e
   $stderr.puts "Webset not found: #{e.message}"

data/exe/exa-ai-webset-item-get

@@ -80,6 +80,7 @@ begin
   # Format and output
   output = Exa::CLI::Formatters::WebsetItemFormatter.format(item, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::NotFound => e
   $stderr.puts "Item not found: #{e.message}"

data/exe/exa-ai-webset-item-list

@@ -69,6 +69,7 @@ begin
   # Format and output
   output = Exa::CLI::Formatters::WebsetItemFormatter.format_collection(items, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::NotFound => e
   $stderr.puts "Webset not found: #{e.message}"

data/exe/exa-ai-webset-list

@@ -65,6 +65,7 @@ begin
   # Format and output
   output = Exa::CLI::Formatters::WebsetFormatter.format_collection(collection, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::ConfigurationError => e
   $stderr.puts "Configuration error: #{e.message}"

data/exe/exa-ai-webset-search-cancel

@@ -79,6 +79,7 @@ begin
   # Format and output
   output = Exa::CLI::Formatters::SearchFormatter.format(search, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::NotFound => e
   $stderr.puts "Search not found: #{e.message}"

data/exe/exa-ai-webset-search-create

@@ -205,6 +205,7 @@ begin
   # Format and output result
   output = Exa::CLI::Formatters::SearchFormatter.format(search, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::ConfigurationError => e
   $stderr.puts "Configuration error: #{e.message}"

data/exe/exa-ai-webset-search-get

@@ -80,6 +80,7 @@ begin
   # Format and output
   output = Exa::CLI::Formatters::SearchFormatter.format(search, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::NotFound => e
   $stderr.puts "Search not found: #{e.message}"

data/exe/exa-ai-webset-update

@@ -111,6 +111,7 @@ begin
   # Format and output
   output = Exa::CLI::Formatters::WebsetFormatter.format(webset, output_format)
   puts output
+  $stdout.flush
 
 rescue Exa::NotFound => e
   $stderr.puts "Webset not found: #{e.message}"

data/lib/exa/cli/base.rb

@@ -16,11 +16,11 @@ module Exa
       end
 
       # Resolve and validate output format
-      # Valid formats: json, pretty, text
+      # Valid formats: json, pretty, text, toon
       # Defaults to json
       def self.resolve_output_format(flag_value)
         format = (flag_value || "json").downcase
-        valid_formats = %w[json pretty text]
+        valid_formats = %w[json pretty text toon]
 
         return format if valid_formats.include?(format)
 
@@ -46,6 +46,12 @@ module Exa
           data.to_s
         end
       end
+
+      # Encode data as TOON format
+      def self.encode_as_toon(data)
+        require "toon" unless defined?(Toon)
+        Toon.encode(data)
+      end
     end
   end
 end

data/lib/exa/cli/formatters/answer_formatter.rb

@@ -12,6 +12,8 @@ module Exa
             format_pretty(result)
           when "text"
             format_text(result)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(result.to_h)
           else
             JSON.pretty_generate(result.to_h)
           end

data/lib/exa/cli/formatters/contents_formatter.rb

@@ -12,6 +12,8 @@ module Exa
             format_pretty(result)
           when "text"
             format_text(result)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(result.to_h)
           else
             JSON.pretty_generate(result.to_h)
           end

data/lib/exa/cli/formatters/context_formatter.rb

@@ -14,6 +14,8 @@ module Exa
             format_pretty(result)
           when "text"
             format_text(result)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(result.to_h)
           else
             JSON.pretty_generate(result.to_h)
           end

data/lib/exa/cli/formatters/enrichment_formatter.rb

@@ -12,6 +12,8 @@ module Exa
             JSON.pretty_generate(enrichment.to_h)
           when "text"
             format_as_text(enrichment)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(enrichment.to_h)
           else
             raise ArgumentError, "Unknown output format: #{output_format}"
           end
@@ -25,6 +27,8 @@ module Exa
             JSON.pretty_generate(collection.to_h)
           when "text"
             format_collection_as_text(collection)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(collection.to_h)
           else
             raise ArgumentError, "Unknown output format: #{output_format}"
           end

data/lib/exa/cli/formatters/import_formatter.rb

@@ -0,0 +1,85 @@
+# frozen_string_literal: true
+
+module Exa
+  module CLI
+    module Formatters
+      class ImportFormatter
+        def self.format(import, output_format)
+          case output_format
+          when "json"
+            JSON.generate(import.to_h)
+          when "pretty"
+            JSON.pretty_generate(import.to_h)
+          when "text"
+            format_as_text(import)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(import.to_h)
+          else
+            raise ArgumentError, "Unknown output format: #{output_format}"
+          end
+        end
+
+        def self.format_collection(collection, output_format)
+          case output_format
+          when "json"
+            JSON.generate(collection.to_h)
+          when "pretty"
+            JSON.pretty_generate(collection.to_h)
+          when "text"
+            format_collection_as_text(collection)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(collection.to_h)
+          else
+            raise ArgumentError, "Unknown output format: #{output_format}"
+          end
+        end
+
+        def self.format_as_text(import)
+          lines = []
+          lines << "Import: #{import.id}"
+          lines << "Status: #{import.status}"
+          lines << "Title: #{import.title}" if import.title
+          lines << "Format: #{import.format}" if import.format
+
+          if import.entity
+            entity_type = import.entity['type'] || import.entity[:type]
+            lines << "Entity Type: #{entity_type}" if entity_type
+          end
+
+          lines << "Count: #{import.count}" if import.count
+
+          if import.failed?
+            lines << "\nFailure Details:"
+            lines << "  Reason: #{import.failed_reason}" if import.failed_reason
+            lines << "  Message: #{import.failed_message}" if import.failed_message
+            lines << "  Failed At: #{import.failed_at}" if import.failed_at
+          end
+
+          if import.upload_url
+            lines << "\nUpload:"
+            lines << "  URL: #{import.upload_url}"
+            lines << "  Valid Until: #{import.upload_valid_until}" if import.upload_valid_until
+          end
+
+          lines << "\nCreated: #{import.created_at}" if import.created_at
+          lines << "Updated: #{import.updated_at}" if import.updated_at
+
+          lines.join("\n")
+        end
+        private_class_method :format_as_text
+
+        def self.format_collection_as_text(collection)
+          lines = ["Imports (#{collection.data.length} items):"]
+          collection.data.each do |imp|
+            lines << "\n  #{imp.id}"
+            lines << "  Status: #{imp.status}"
+            lines << "  Title: #{imp.title}" if imp.title
+            lines << "  Count: #{imp.count}" if imp.count
+          end
+          lines.join("\n")
+        end
+        private_class_method :format_collection_as_text
+      end
+    end
+  end
+end

data/lib/exa/cli/formatters/monitor_formatter.rb

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+module Exa
+  module CLI
+    module Formatters
+      class MonitorFormatter
+        def self.format(monitor, output_format)
+          case output_format
+          when "json"
+            JSON.generate(monitor.to_h)
+          when "pretty"
+            JSON.pretty_generate(monitor.to_h)
+          when "text"
+            format_as_text(monitor)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(monitor.to_h)
+          else
+            raise ArgumentError, "Unknown output format: #{output_format}"
+          end
+        end
+
+        def self.format_collection(collection, output_format)
+          case output_format
+          when "json"
+            JSON.generate(collection.to_h)
+          when "pretty"
+            JSON.pretty_generate(collection.to_h)
+          when "text"
+            format_collection_as_text(collection)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(collection.to_h)
+          else
+            raise ArgumentError, "Unknown output format: #{output_format}"
+          end
+        end
+
+        def self.format_as_text(monitor)
+          lines = []
+          lines << "Monitor: #{monitor.id}"
+          lines << "Webset: #{monitor.webset_id}" if monitor.webset_id
+          lines << "Status: #{monitor.status}"
+
+          if monitor.cadence
+            lines << "\nCadence:"
+            lines << "  Cron: #{monitor.cadence['cron']}" if monitor.cadence['cron']
+            lines << "  Timezone: #{monitor.cadence['timezone']}" if monitor.cadence['timezone']
+          end
+
+          if monitor.behavior
+            lines << "\nBehavior:"
+            lines << "  Type: #{monitor.behavior['type']}" if monitor.behavior['type']
+            lines << "  Query: #{monitor.behavior['query']}" if monitor.behavior['query']
+            lines << "  Count: #{monitor.behavior['count']}" if monitor.behavior['count']
+          end
+
+          lines << "\nCreated: #{monitor.created_at}" if monitor.created_at
+          lines << "Updated: #{monitor.updated_at}" if monitor.updated_at
+
+          lines.join("\n")
+        end
+        private_class_method :format_as_text
+
+        def self.format_collection_as_text(collection)
+          lines = ["Monitors (#{collection.data.length} items):"]
+          collection.data.each do |mon|
+            lines << "\n  #{mon['id']}"
+            lines << "  Status: #{mon['status']}"
+            lines << "  Webset: #{mon['websetId']}" if mon['websetId']
+          end
+
+          if collection.has_more
+            lines << "\nMore available (cursor: #{collection.next_cursor})"
+          end
+
+          lines.join("\n")
+        end
+        private_class_method :format_collection_as_text
+      end
+    end
+  end
+end

data/lib/exa/cli/formatters/monitor_run_formatter.rb

@@ -0,0 +1,75 @@
+# frozen_string_literal: true
+
+module Exa
+  module CLI
+    module Formatters
+      class MonitorRunFormatter
+        def self.format(monitor_run, output_format)
+          case output_format
+          when "json"
+            JSON.generate(monitor_run.to_h)
+          when "pretty"
+            JSON.pretty_generate(monitor_run.to_h)
+          when "text"
+            format_as_text(monitor_run)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(monitor_run.to_h)
+          else
+            raise ArgumentError, "Unknown output format: #{output_format}"
+          end
+        end
+
+        def self.format_collection(collection, output_format)
+          case output_format
+          when "json"
+            JSON.generate(collection.to_h)
+          when "pretty"
+            JSON.pretty_generate(collection.to_h)
+          when "text"
+            format_collection_as_text(collection)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(collection.to_h)
+          else
+            raise ArgumentError, "Unknown output format: #{output_format}"
+          end
+        end
+
+        def self.format_as_text(monitor_run)
+          lines = []
+          lines << "Monitor Run: #{monitor_run.id}"
+          lines << "Monitor: #{monitor_run.monitor_id}" if monitor_run.monitor_id
+          lines << "Status: #{monitor_run.status}"
+
+          lines << "\nCreated: #{monitor_run.created_at}" if monitor_run.created_at
+          lines << "Updated: #{monitor_run.updated_at}" if monitor_run.updated_at
+          lines << "Completed: #{monitor_run.completed_at}" if monitor_run.completed_at
+
+          if monitor_run.failed?
+            lines << "Failed: #{monitor_run.failed_at}" if monitor_run.failed_at
+            lines << "Reason: #{monitor_run.failed_reason}" if monitor_run.failed_reason
+          end
+
+          lines.join("\n")
+        end
+        private_class_method :format_as_text
+
+        def self.format_collection_as_text(collection)
+          lines = ["Monitor Runs (#{collection.data.length} items):"]
+          collection.data.each do |run|
+            lines << "\n  #{run['id']}"
+            lines << "  Status: #{run['status']}"
+            lines << "  Completed: #{run['completedAt']}" if run['completedAt']
+            lines << "  Failed: #{run['failedReason']}" if run['failedReason']
+          end
+
+          if collection.has_more
+            lines << "\nMore available (cursor: #{collection.next_cursor})"
+          end
+
+          lines.join("\n")
+        end
+        private_class_method :format_collection_as_text
+      end
+    end
+  end
+end

data/lib/exa/cli/formatters/research_formatter.rb

@@ -10,6 +10,8 @@ module Exa
             format_task_pretty(task, show_events: show_events)
           when "text"
             format_task_text(task, show_events: show_events)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(task.to_h)
           else
             JSON.pretty_generate(task.to_h)
           end
@@ -23,6 +25,8 @@ module Exa
             format_list_pretty(list)
           when "text"
             format_list_text(list)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(list.to_h)
           else
             JSON.pretty_generate(list.to_h)
           end

data/lib/exa/cli/formatters/search_formatter.rb

@@ -12,6 +12,8 @@ module Exa
             format_pretty(result)
           when "text"
             format_text(result)
+          when "toon"
+            Exa::CLI::Base.encode_as_toon(result.to_h)
           else
             JSON.pretty_generate(result.to_h)
           end