exa-ai 0.1.0

data/lib/exa/services/context.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    class Context
+      def initialize(connection, **params)
+        @connection = connection
+        @params = params
+      end
+
+      def call
+        response = @connection.post("/context", @params)
+        body = response.body
+
+        Resources::ContextResult.new(
+          request_id: body["requestId"],
+          query: body["query"],
+          response: body["response"],
+          results_count: body["resultsCount"],
+          cost_dollars: body["costDollars"],
+          search_time: body["searchTime"],
+          output_tokens: body["outputTokens"]
+        )
+      end
+    end
+  end
+end

data/lib/exa/services/find_similar.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require_relative "../resources/find_similar_result"
+
+module Exa
+  module Services
+    class FindSimilar
+      def initialize(connection, **params)
+        @connection = connection
+        @params = params
+      end
+
+      def call
+        response = @connection.post("/findSimilar", @params)
+        body = response.body
+
+        Resources::FindSimilarResult.new(
+          results: body["results"],
+          request_id: body["requestId"],
+          context: body["context"],
+          cost_dollars: body["costDollars"]
+        )
+      end
+    end
+  end
+end

data/lib/exa/services/get_contents.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    class GetContents
+      def initialize(connection, **params)
+        @connection = connection
+        @params = params
+      end
+
+      def call
+        response = @connection.post("/contents", @params)
+        body = response.body
+
+        Resources::ContentsResult.new(
+          results: body["results"],
+          request_id: body["requestId"],
+          context: body["context"],
+          statuses: body["statuses"],
+          cost_dollars: body["costDollars"]
+        )
+      end
+    end
+  end
+end

data/lib/exa/services/research_get.rb

@@ -0,0 +1,30 @@
+module Exa
+  module Services
+    class ResearchGet
+      def initialize(connection, research_id:, **params)
+        @connection = connection
+        @research_id = research_id
+        @params = params
+      end
+
+      def call
+        response = @connection.get("/research/v1/#{@research_id}", @params)
+        body = response.body
+
+        Resources::ResearchTask.new(
+          research_id: body["researchId"],
+          created_at: body["createdAt"],
+          status: body["status"],
+          instructions: body["instructions"],
+          model: body["model"],
+          output_schema: body["outputSchema"],
+          events: body["events"],
+          output: body["output"],
+          cost_dollars: body["costDollars"],
+          finished_at: body["finishedAt"],
+          error: body["error"]
+        )
+      end
+    end
+  end
+end

data/lib/exa/services/research_list.rb

@@ -0,0 +1,37 @@
+module Exa
+  module Services
+    class ResearchList
+      def initialize(connection, **params)
+        @connection = connection
+        @params = params
+      end
+
+      def call
+        response = @connection.get("/research/v1", @params)
+        body = response.body
+
+        data = body["data"].map do |task_data|
+          Resources::ResearchTask.new(
+            research_id: task_data["researchId"],
+            created_at: task_data["createdAt"],
+            status: task_data["status"],
+            instructions: task_data["instructions"],
+            model: task_data["model"],
+            output_schema: task_data["outputSchema"],
+            events: task_data["events"],
+            output: task_data["output"],
+            cost_dollars: task_data["costDollars"],
+            finished_at: task_data["finishedAt"],
+            error: task_data["error"]
+          )
+        end
+
+        Resources::ResearchList.new(
+          data: data,
+          has_more: body["hasMore"],
+          next_cursor: body["nextCursor"]
+        )
+      end
+    end
+  end
+end

data/lib/exa/services/research_start.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    class ResearchStart
+      def initialize(connection, **params)
+        @connection = connection
+        @params = params
+      end
+
+      def call
+        response = @connection.post("/research/v1", @params)
+        body = response.body
+
+        Resources::ResearchTask.new(
+          research_id: body["researchId"],
+          created_at: body["createdAt"],
+          status: body["status"],
+          instructions: body["instructions"],
+          model: body["model"],
+          output_schema: body["outputSchema"]
+        )
+      end
+    end
+  end
+end

data/lib/exa/services/search.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module Exa
+  module Services
+    class Search
+      def initialize(connection, **params)
+        @connection = connection
+        @params = params
+      end
+
+      def call
+        response = @connection.post("/search", @params)
+        body = response.body
+
+        Resources::SearchResult.new(
+          results: body["results"],
+          request_id: body["requestId"],
+          resolved_search_type: body["resolvedSearchType"],
+          search_type: body["searchType"],
+          context: body["context"],
+          cost_dollars: body["costDollars"]
+        )
+      end
+    end
+  end
+end

data/lib/exa/version.rb

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

data/lib/exa.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require_relative "exa/version"
+require_relative "exa/error"
+require_relative "exa/middleware/raise_error"
+require_relative "exa/connection"
+require_relative "exa/resources/search_result"
+require_relative "exa/resources/find_similar_result"
+require_relative "exa/resources/contents_result"
+require_relative "exa/resources/research_task"
+require_relative "exa/resources/research_list"
+require_relative "exa/resources/answer"
+require_relative "exa/resources/context_result"
+require_relative "exa/services/search"
+require_relative "exa/services/find_similar"
+require_relative "exa/services/get_contents"
+require_relative "exa/services/research_get"
+require_relative "exa/services/research_start"
+require_relative "exa/services/research_list"
+require_relative "exa/services/answer"
+require_relative "exa/services/context"
+require_relative "exa/client"
+require_relative "exa/cli/base"
+require_relative "exa/cli/polling"
+require_relative "exa/cli/error_handler"
+require_relative "exa/cli/formatters/search_formatter"
+require_relative "exa/cli/formatters/context_formatter"
+require_relative "exa/cli/formatters/contents_formatter"
+require_relative "exa/cli/formatters/research_formatter"
+
+module Exa
+  # Module-level configuration
+  class << self
+    attr_accessor :api_key, :base_url, :timeout
+
+    def configure
+      yield self
+    end
+
+    def reset
+      self.api_key = nil
+      self.base_url = DEFAULT_BASE_URL
+      self.timeout = DEFAULT_TIMEOUT
+    end
+  end
+
+  # Constants for default values
+  DEFAULT_BASE_URL = "https://api.exa.ai"
+  DEFAULT_TIMEOUT = 30
+
+  # Set defaults
+  self.base_url = DEFAULT_BASE_URL
+  self.timeout = DEFAULT_TIMEOUT
+end

metadata

@@ -0,0 +1,174 @@
+--- !ruby/object:Gem::Specification
+name: exa-ai
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Benjamin Jackson
+bindir: exe
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: vcr
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+- !ruby/object:Gem::Dependency
+  name: bundler-audit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+description: A Ruby gem for interacting with the Exa.ai search and discovery API
+email:
+- ben@exa.ai
+executables:
+- exa-ai
+- exa-ai-answer
+- exa-ai-context
+- exa-ai-get-contents
+- exa-ai-research-get
+- exa-ai-research-list
+- exa-ai-research-start
+- exa-ai-search
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- exe/exa-ai
+- exe/exa-ai-answer
+- exe/exa-ai-context
+- exe/exa-ai-get-contents
+- exe/exa-ai-research-get
+- exe/exa-ai-research-list
+- exe/exa-ai-research-start
+- exe/exa-ai-search
+- lib/exa.rb
+- lib/exa/cli/base.rb
+- lib/exa/cli/error_handler.rb
+- lib/exa/cli/formatters/answer_formatter.rb
+- lib/exa/cli/formatters/contents_formatter.rb
+- lib/exa/cli/formatters/context_formatter.rb
+- lib/exa/cli/formatters/research_formatter.rb
+- lib/exa/cli/formatters/search_formatter.rb
+- lib/exa/cli/polling.rb
+- lib/exa/client.rb
+- lib/exa/connection.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/research_list.rb
+- lib/exa/resources/research_task.rb
+- lib/exa/resources/search_result.rb
+- lib/exa/services/answer.rb
+- lib/exa/services/context.rb
+- lib/exa/services/find_similar.rb
+- lib/exa/services/get_contents.rb
+- lib/exa/services/research_get.rb
+- lib/exa/services/research_list.rb
+- lib/exa/services/research_start.rb
+- lib/exa/services/search.rb
+- lib/exa/version.rb
+homepage: https://github.com/benjaminjackson/exa-ruby
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/benjaminjackson/exa-ruby
+  source_code_uri: https://github.com/benjaminjackson/exa-ruby
+  changelog_uri: https://github.com/benjaminjackson/exa-ruby/blob/main/CHANGELOG.md
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.7.1
+specification_version: 4
+summary: Ruby client for the Exa.ai API
+test_files: []