exa-ai 0.5.0 → 0.6.0

data/lib/exa/services/websets/import_create.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      class CreateImport
+        def initialize(connection, **params)
+          @connection = connection
+          @params = params
+        end
+
+        def call
+          response = @connection.post(
+            "/websets/v0/imports",
+            @params
+          )
+          body = response.body
+
+          Resources::Import.new(
+            id: body["id"],
+            object: body["object"],
+            status: body["status"],
+            format: body["format"],
+            entity: body["entity"],
+            title: body["title"],
+            count: body["count"],
+            metadata: body["metadata"],
+            failed_reason: body["failedReason"],
+            failed_at: body["failedAt"],
+            failed_message: body["failedMessage"],
+            created_at: body["createdAt"],
+            updated_at: body["updatedAt"],
+            upload_url: body["uploadUrl"],
+            upload_valid_until: body["uploadValidUntil"]
+          )
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/import_delete.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      class DeleteImport
+        def initialize(connection, id:)
+          @connection = connection
+          @id = id
+        end
+
+        def call
+          response = @connection.delete("/websets/v0/imports/#{@id}")
+          body = response.body
+
+          Resources::Import.new(
+            id: body["id"],
+            object: body["object"],
+            status: body["status"],
+            format: body["format"],
+            entity: body["entity"],
+            title: body["title"],
+            count: body["count"],
+            metadata: body["metadata"],
+            failed_reason: body["failedReason"],
+            failed_at: body["failedAt"],
+            failed_message: body["failedMessage"],
+            created_at: body["createdAt"],
+            updated_at: body["updatedAt"],
+            upload_url: body["uploadUrl"],
+            upload_valid_until: body["uploadValidUntil"]
+          )
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/import_get.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      class GetImport
+        def initialize(connection, id:)
+          @connection = connection
+          @id = id
+        end
+
+        def call
+          response = @connection.get("/websets/v0/imports/#{@id}")
+          body = response.body
+
+          Resources::Import.new(
+            id: body["id"],
+            object: body["object"],
+            status: body["status"],
+            format: body["format"],
+            entity: body["entity"],
+            title: body["title"],
+            count: body["count"],
+            metadata: body["metadata"],
+            failed_reason: body["failedReason"],
+            failed_at: body["failedAt"],
+            failed_message: body["failedMessage"],
+            created_at: body["createdAt"],
+            updated_at: body["updatedAt"],
+            upload_url: body["uploadUrl"],
+            upload_valid_until: body["uploadValidUntil"]
+          )
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/import_list.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      class ListImports
+        def initialize(connection, **params)
+          @connection = connection
+          @params = params
+        end
+
+        def call
+          response = @connection.get("/websets/v0/imports", @params)
+          body = response.body
+
+          Resources::ImportCollection.new(
+            data: body["data"],
+            has_more: body["hasMore"],
+            next_cursor: body["nextCursor"]
+          )
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/import_update.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      class UpdateImport
+        def initialize(connection, id:, **params)
+          @connection = connection
+          @id = id
+          @params = params
+        end
+
+        def call
+          response = @connection.patch("/websets/v0/imports/#{@id}", @params)
+          body = response.body
+
+          Resources::Import.new(
+            id: body["id"],
+            object: body["object"],
+            status: body["status"],
+            format: body["format"],
+            entity: body["entity"],
+            title: body["title"],
+            count: body["count"],
+            metadata: body["metadata"],
+            failed_reason: body["failedReason"],
+            failed_at: body["failedAt"],
+            failed_message: body["failedMessage"],
+            created_at: body["createdAt"],
+            updated_at: body["updatedAt"],
+            upload_url: body["uploadUrl"],
+            upload_valid_until: body["uploadValidUntil"]
+          )
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/import_upload.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+require "faraday"
+
+module Exa
+  module Services
+    module Websets
+      class UploadImport
+        def initialize(connection, file_path:, **params)
+          @connection = connection
+          @file_path = file_path
+          @params = params
+        end
+
+        def call
+          validate_file!
+
+          # Infer file size
+          file_size = File.size(@file_path)
+
+          # Create import with inferred size
+          import = CreateImport.new(@connection, size: file_size, **@params).call
+
+          # Upload file to the upload URL
+          upload_file(import.upload_url)
+
+          # Return the import resource
+          import
+        end
+
+        private
+
+        def validate_file!
+          unless File.exist?(@file_path)
+            raise Error, "File not found: #{@file_path}"
+          end
+
+          unless File.readable?(@file_path)
+            raise Error, "Permission denied: file is not readable: #{@file_path}"
+          end
+        end
+
+        def upload_file(upload_url)
+          file_content = File.read(@file_path)
+
+          upload_connection = Faraday.new do |f|
+            f.adapter Faraday.default_adapter
+          end
+
+          upload_connection.put(upload_url) do |req|
+            req.headers["Content-Type"] = "text/csv"
+            req.body = file_content
+          end
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/monitors/create.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      module Monitors
+        class Create
+          def initialize(connection, webset_id:, cadence:, behavior:, **params)
+            @connection = connection
+            @webset_id = webset_id
+            @cadence = cadence
+            @behavior = behavior
+            @params = params
+          end
+
+          def call
+            response = @connection.post(
+              "/websets/v0/monitors",
+              {
+                websetId: @webset_id,
+                cadence: @cadence,
+                behavior: @behavior
+              }.merge(@params)
+            )
+            body = response.body
+
+            Resources::Monitor.new(
+              id: body["id"],
+              object: body["object"],
+              status: body["status"],
+              webset_id: body["websetId"],
+              cadence: body["cadence"],
+              behavior: body["behavior"],
+              created_at: body["createdAt"],
+              updated_at: body["updatedAt"]
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/monitors/delete.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      module Monitors
+        class Delete
+          def initialize(connection, id:)
+            @connection = connection
+            @id = id
+          end
+
+          def call
+            response = @connection.delete("/websets/v0/monitors/#{@id}")
+            body = response.body
+
+            Resources::Monitor.new(
+              id: body["id"],
+              object: body["object"],
+              status: body["status"],
+              webset_id: body["websetId"],
+              cadence: body["cadence"],
+              behavior: body["behavior"],
+              created_at: body["createdAt"],
+              updated_at: body["updatedAt"]
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/monitors/get.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      module Monitors
+        class Get
+          def initialize(connection, id:, **params)
+            @connection = connection
+            @id = id
+            @params = params
+          end
+
+          def call
+            response = @connection.get("/websets/v0/monitors/#{@id}", @params)
+            body = response.body
+
+            Resources::Monitor.new(
+              id: body["id"],
+              object: body["object"],
+              status: body["status"],
+              webset_id: body["websetId"],
+              cadence: body["cadence"],
+              behavior: body["behavior"],
+              created_at: body["createdAt"],
+              updated_at: body["updatedAt"]
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/monitors/list.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      module Monitors
+        class List
+          def initialize(connection, **params)
+            @connection = connection
+            @params = params
+          end
+
+          def call
+            response = @connection.get("/websets/v0/monitors", @params)
+            body = response.body
+
+            Resources::MonitorCollection.new(
+              data: body["data"],
+              has_more: body["hasMore"],
+              next_cursor: body["nextCursor"]
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/monitors/runs/get.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      module Monitors
+        module Runs
+          class Get
+            def initialize(connection, monitor_id:, id:, **params)
+              @connection = connection
+              @monitor_id = monitor_id
+              @id = id
+              @params = params
+            end
+
+            def call
+              response = @connection.get("/websets/v0/monitors/#{@monitor_id}/runs/#{@id}", @params)
+              body = response.body
+
+              Resources::MonitorRun.new(
+                id: body["id"],
+                object: body["object"],
+                monitor_id: body["monitorId"],
+                status: body["status"],
+                created_at: body["createdAt"],
+                updated_at: body["updatedAt"],
+                completed_at: body["completedAt"],
+                failed_at: body["failedAt"],
+                failed_reason: body["failedReason"]
+              )
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/monitors/runs/list.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      module Monitors
+        module Runs
+          class List
+            def initialize(connection, monitor_id:, **params)
+              @connection = connection
+              @monitor_id = monitor_id
+              @params = params
+            end
+
+            def call
+              response = @connection.get("/websets/v0/monitors/#{@monitor_id}/runs", @params)
+              body = response.body
+
+              Resources::MonitorRunCollection.new(
+                data: body["data"],
+                has_more: body["hasMore"],
+                next_cursor: body["nextCursor"]
+              )
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/exa/services/websets/monitors/update.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    module Websets
+      module Monitors
+        class Update
+          def initialize(connection, id:, **params)
+            @connection = connection
+            @id = id
+            @params = params
+          end
+
+          def call
+            response = @connection.patch("/websets/v0/monitors/#{@id}", @params)
+            body = response.body
+
+            Resources::Monitor.new(
+              id: body["id"],
+              object: body["object"],
+              status: body["status"],
+              webset_id: body["websetId"],
+              cadence: body["cadence"],
+              behavior: body["behavior"],
+              created_at: body["createdAt"],
+              updated_at: body["updatedAt"]
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/exa/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Exa
-  VERSION = "0.5.0"
+  VERSION = "0.6.0"
 end

data/lib/exa.rb

@@ -2,6 +2,7 @@
 
 require_relative "exa/version"
 require_relative "exa/error"
+require_relative "exa/constants/websets"
 require_relative "exa/middleware/raise_error"
 require_relative "exa/connection"
 require_relative "exa/resources/search_result"
@@ -16,6 +17,12 @@ require_relative "exa/resources/webset"
 require_relative "exa/resources/webset_search"
 require_relative "exa/resources/webset_enrichment"
 require_relative "exa/resources/webset_enrichment_collection"
+require_relative "exa/resources/import"
+require_relative "exa/resources/import_collection"
+require_relative "exa/resources/monitor"
+require_relative "exa/resources/monitor_collection"
+require_relative "exa/resources/monitor_run"
+require_relative "exa/resources/monitor_run_collection"
 require_relative "exa/services/search"
 require_relative "exa/services/find_similar"
 require_relative "exa/services/get_contents"
@@ -42,6 +49,19 @@ require_relative "exa/services/websets/cancel_enrichment"
 require_relative "exa/services/websets/get_item"
 require_relative "exa/services/websets/delete_item"
 require_relative "exa/services/websets/list_items"
+require_relative "exa/services/websets/import_create"
+require_relative "exa/services/websets/import_upload"
+require_relative "exa/services/websets/import_list"
+require_relative "exa/services/websets/import_get"
+require_relative "exa/services/websets/import_update"
+require_relative "exa/services/websets/import_delete"
+require_relative "exa/services/websets/monitors/create"
+require_relative "exa/services/websets/monitors/list"
+require_relative "exa/services/websets/monitors/get"
+require_relative "exa/services/websets/monitors/update"
+require_relative "exa/services/websets/monitors/delete"
+require_relative "exa/services/websets/monitors/runs/list"
+require_relative "exa/services/websets/monitors/runs/get"
 require_relative "exa/client"
 require_relative "exa/cli/base"
 require_relative "exa/cli/polling"
@@ -54,6 +74,9 @@ require_relative "exa/cli/formatters/answer_formatter"
 require_relative "exa/cli/formatters/webset_formatter"
 require_relative "exa/cli/formatters/webset_item_formatter"
 require_relative "exa/cli/formatters/enrichment_formatter"
+require_relative "exa/cli/formatters/import_formatter"
+require_relative "exa/cli/formatters/monitor_formatter"
+require_relative "exa/cli/formatters/monitor_run_formatter"
 
 module Exa
   # Module-level configuration

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: exa-ai
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Benjamin Jackson
@@ -37,6 +37,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: toon-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.1'
 - !ruby/object:Gem::Dependency
   name: minitest
   requirement: !ruby/object:Gem::Requirement
@@ -149,6 +163,18 @@ files:
 - exe/exa-ai-enrichment-update
 - exe/exa-ai-find-similar
 - exe/exa-ai-get-contents
+- exe/exa-ai-import-create
+- exe/exa-ai-import-delete
+- exe/exa-ai-import-get
+- exe/exa-ai-import-list
+- exe/exa-ai-import-update
+- exe/exa-ai-monitor-create
+- exe/exa-ai-monitor-delete
+- exe/exa-ai-monitor-get
+- exe/exa-ai-monitor-list
+- exe/exa-ai-monitor-runs-get
+- exe/exa-ai-monitor-runs-list
+- exe/exa-ai-monitor-update
 - exe/exa-ai-research-get
 - exe/exa-ai-research-list
 - exe/exa-ai-research-start
@@ -173,6 +199,9 @@ files:
 - lib/exa/cli/formatters/contents_formatter.rb
 - lib/exa/cli/formatters/context_formatter.rb
 - lib/exa/cli/formatters/enrichment_formatter.rb
+- lib/exa/cli/formatters/import_formatter.rb
+- lib/exa/cli/formatters/monitor_formatter.rb
+- lib/exa/cli/formatters/monitor_run_formatter.rb
 - lib/exa/cli/formatters/research_formatter.rb
 - lib/exa/cli/formatters/search_formatter.rb
 - lib/exa/cli/formatters/webset_formatter.rb
@@ -180,12 +209,19 @@ files:
 - lib/exa/cli/polling.rb
 - lib/exa/client.rb
 - lib/exa/connection.rb
+- lib/exa/constants/websets.rb
 - lib/exa/error.rb
 - lib/exa/middleware/raise_error.rb
 - lib/exa/resources/answer.rb
 - lib/exa/resources/contents_result.rb
 - lib/exa/resources/context_result.rb
 - lib/exa/resources/find_similar_result.rb
+- lib/exa/resources/import.rb
+- lib/exa/resources/import_collection.rb
+- lib/exa/resources/monitor.rb
+- lib/exa/resources/monitor_collection.rb
+- lib/exa/resources/monitor_run.rb
+- lib/exa/resources/monitor_run_collection.rb
 - lib/exa/resources/research_list.rb
 - lib/exa/resources/research_task.rb
 - lib/exa/resources/search_result.rb
@@ -217,8 +253,21 @@ files:
 - lib/exa/services/websets/delete_item.rb
 - lib/exa/services/websets/get_item.rb
 - lib/exa/services/websets/get_search.rb
+- lib/exa/services/websets/import_create.rb
+- lib/exa/services/websets/import_delete.rb
+- lib/exa/services/websets/import_get.rb
+- lib/exa/services/websets/import_list.rb
+- lib/exa/services/websets/import_update.rb
+- lib/exa/services/websets/import_upload.rb
 - lib/exa/services/websets/list.rb
 - lib/exa/services/websets/list_items.rb
+- lib/exa/services/websets/monitors/create.rb
+- lib/exa/services/websets/monitors/delete.rb
+- lib/exa/services/websets/monitors/get.rb
+- lib/exa/services/websets/monitors/list.rb
+- lib/exa/services/websets/monitors/runs/get.rb
+- lib/exa/services/websets/monitors/runs/list.rb
+- lib/exa/services/websets/monitors/update.rb
 - lib/exa/services/websets/retrieve.rb
 - lib/exa/services/websets/retrieve_enrichment.rb
 - lib/exa/services/websets/update.rb