nextbillionai 0.1.0.pre.alpha.4

data/lib/nextbillionai.rb

@@ -0,0 +1,294 @@
+# frozen_string_literal: true
+
+# Standard libraries.
+# rubocop:disable Lint/RedundantRequireStatement
+require "English"
+require "cgi"
+require "date"
+require "erb"
+require "etc"
+require "json"
+require "net/http"
+require "pathname"
+require "rbconfig"
+require "securerandom"
+require "set"
+require "stringio"
+require "time"
+require "uri"
+# rubocop:enable Lint/RedundantRequireStatement
+
+# We already ship the preferred sorbet manifests in the package itself.
+# `tapioca` currently does not offer us a way to opt out of unnecessary compilation.
+if Object.const_defined?(:Tapioca) &&
+   caller.chain([$PROGRAM_NAME]).chain(ARGV).any?(/tapioca/) &&
+   ARGV.none?(/dsl/)
+  return
+end
+
+# Gems.
+require "connection_pool"
+
+# Package files.
+require_relative "nextbillionai/version"
+require_relative "nextbillionai/internal/util"
+require_relative "nextbillionai/internal/type/converter"
+require_relative "nextbillionai/internal/type/unknown"
+require_relative "nextbillionai/internal/type/boolean"
+require_relative "nextbillionai/internal/type/file_input"
+require_relative "nextbillionai/internal/type/enum"
+require_relative "nextbillionai/internal/type/union"
+require_relative "nextbillionai/internal/type/array_of"
+require_relative "nextbillionai/internal/type/hash_of"
+require_relative "nextbillionai/internal/type/base_model"
+require_relative "nextbillionai/internal/type/base_page"
+require_relative "nextbillionai/internal/type/request_parameters"
+require_relative "nextbillionai/internal"
+require_relative "nextbillionai/request_options"
+require_relative "nextbillionai/file_part"
+require_relative "nextbillionai/errors"
+require_relative "nextbillionai/internal/transport/base_client"
+require_relative "nextbillionai/internal/transport/pooled_net_requester"
+require_relative "nextbillionai/client"
+require_relative "nextbillionai/models/geofence_entity_create"
+require_relative "nextbillionai/models/rich_group_request"
+require_relative "nextbillionai/models/access"
+require_relative "nextbillionai/models/address"
+require_relative "nextbillionai/models/area_list_params"
+require_relative "nextbillionai/models/area_list_response"
+require_relative "nextbillionai/models/autocomplete_suggest_params"
+require_relative "nextbillionai/models/autocomplete_suggest_response"
+require_relative "nextbillionai/models/autosuggest_suggest_params"
+require_relative "nextbillionai/models/autosuggest_suggest_response"
+require_relative "nextbillionai/models/batch_create_params"
+require_relative "nextbillionai/models/batch_create_response"
+require_relative "nextbillionai/models/batch_retrieve_params"
+require_relative "nextbillionai/models/batch_retrieve_response"
+require_relative "nextbillionai/models/browse_search_params"
+require_relative "nextbillionai/models/browse_search_response"
+require_relative "nextbillionai/models/categories"
+require_relative "nextbillionai/models/contact_object"
+require_relative "nextbillionai/models/contacts"
+require_relative "nextbillionai/models/direction_compute_route_params"
+require_relative "nextbillionai/models/direction_compute_route_response"
+require_relative "nextbillionai/models/discover_retrieve_params"
+require_relative "nextbillionai/models/discover_retrieve_response"
+require_relative "nextbillionai/models/distance_matrix/json_create_params"
+require_relative "nextbillionai/models/distance_matrix/json_retrieve_params"
+require_relative "nextbillionai/models/distance_matrix/json_retrieve_response"
+require_relative "nextbillionai/models/fleetify/document_template_content_request"
+require_relative "nextbillionai/models/fleetify/document_template_content_response"
+require_relative "nextbillionai/models/fleetify/document_template_create_params"
+require_relative "nextbillionai/models/fleetify/document_template_create_response"
+require_relative "nextbillionai/models/fleetify/document_template_delete_params"
+require_relative "nextbillionai/models/fleetify/document_template_delete_response"
+require_relative "nextbillionai/models/fleetify/document_template_list_params"
+require_relative "nextbillionai/models/fleetify/document_template_list_response"
+require_relative "nextbillionai/models/fleetify/document_template_retrieve_params"
+require_relative "nextbillionai/models/fleetify/document_template_retrieve_response"
+require_relative "nextbillionai/models/fleetify/document_template_update_params"
+require_relative "nextbillionai/models/fleetify/document_template_update_response"
+require_relative "nextbillionai/models/fleetify/route_create_params"
+require_relative "nextbillionai/models/fleetify/route_create_response"
+require_relative "nextbillionai/models/fleetify/route_redispatch_params"
+require_relative "nextbillionai/models/fleetify/route_redispatch_response"
+require_relative "nextbillionai/models/fleetify/routes/document_submission"
+require_relative "nextbillionai/models/fleetify/routes/route_step_completion_mode"
+require_relative "nextbillionai/models/fleetify/routes/route_step_geofence_config"
+require_relative "nextbillionai/models/fleetify/routes/route_steps_request"
+require_relative "nextbillionai/models/fleetify/routes/route_steps_response"
+require_relative "nextbillionai/models/fleetify/routes/step_complete_params"
+require_relative "nextbillionai/models/fleetify/routes/step_create_params"
+require_relative "nextbillionai/models/fleetify/routes/step_create_response"
+require_relative "nextbillionai/models/fleetify/routes/step_delete_params"
+require_relative "nextbillionai/models/fleetify/routes/step_delete_response"
+require_relative "nextbillionai/models/fleetify/routes/step_update_params"
+require_relative "nextbillionai/models/fleetify/routes/step_update_response"
+require_relative "nextbillionai/models/fleetify/routing_response"
+require_relative "nextbillionai/models/geocode_batch_create_params"
+require_relative "nextbillionai/models/geocode_batch_create_response"
+require_relative "nextbillionai/models/geocode_retrieve_params"
+require_relative "nextbillionai/models/geocode_retrieve_response"
+require_relative "nextbillionai/models/geocode_structured_retrieve_params"
+require_relative "nextbillionai/models/geocode_structured_retrieve_response"
+require_relative "nextbillionai/models/geofence/batch_create_params"
+require_relative "nextbillionai/models/geofence/batch_create_response"
+require_relative "nextbillionai/models/geofence/batch_delete_params"
+require_relative "nextbillionai/models/geofence/batch_list_params"
+require_relative "nextbillionai/models/geofence/batch_list_response"
+require_relative "nextbillionai/models/geofence/console_preview_params"
+require_relative "nextbillionai/models/geofence/console_preview_response"
+require_relative "nextbillionai/models/geofence/console_search_params"
+require_relative "nextbillionai/models/geofence/console_search_response"
+require_relative "nextbillionai/models/geofence/polygon_geojson"
+require_relative "nextbillionai/models/geofence"
+require_relative "nextbillionai/models/geofence_contains_params"
+require_relative "nextbillionai/models/geofence_contains_response"
+require_relative "nextbillionai/models/geofence_create_params"
+require_relative "nextbillionai/models/geofence_create_response"
+require_relative "nextbillionai/models/geofence_delete_params"
+require_relative "nextbillionai/models/geofence_list_params"
+require_relative "nextbillionai/models/geofence_list_response"
+require_relative "nextbillionai/models/geofence_retrieve_params"
+require_relative "nextbillionai/models/geofence_retrieve_response"
+require_relative "nextbillionai/models/geofence_update_params"
+require_relative "nextbillionai/models/isochrone_compute_params"
+require_relative "nextbillionai/models/isochrone_compute_response"
+require_relative "nextbillionai/models/lookup_by_id_params"
+require_relative "nextbillionai/models/lookup_by_id_response"
+require_relative "nextbillionai/models/map_create_segment_params"
+require_relative "nextbillionai/models/map_view"
+require_relative "nextbillionai/models/mdm_create_distance_matrix_params"
+require_relative "nextbillionai/models/mdm_create_distance_matrix_response"
+require_relative "nextbillionai/models/mdm_get_distance_matrix_status_params"
+require_relative "nextbillionai/models/mdm_get_distance_matrix_status_response"
+require_relative "nextbillionai/models/multigeocode/place_create_params"
+require_relative "nextbillionai/models/multigeocode/place_create_response"
+require_relative "nextbillionai/models/multigeocode/place_delete_params"
+require_relative "nextbillionai/models/multigeocode/place_delete_response"
+require_relative "nextbillionai/models/multigeocode/place_item"
+require_relative "nextbillionai/models/multigeocode/place_retrieve_params"
+require_relative "nextbillionai/models/multigeocode/place_retrieve_response"
+require_relative "nextbillionai/models/multigeocode/place_update_params"
+require_relative "nextbillionai/models/multigeocode/place_update_response"
+require_relative "nextbillionai/models/multigeocode_search_params"
+require_relative "nextbillionai/models/multigeocode_search_response"
+require_relative "nextbillionai/models/navigation_retrieve_route_params"
+require_relative "nextbillionai/models/navigation_retrieve_route_response"
+require_relative "nextbillionai/models/optimization/driver_assignment_assign_params"
+require_relative "nextbillionai/models/optimization/driver_assignment_assign_response"
+require_relative "nextbillionai/models/optimization/job"
+require_relative "nextbillionai/models/optimization/location"
+require_relative "nextbillionai/models/optimization/shipment"
+require_relative "nextbillionai/models/optimization/v2_retrieve_result_params"
+require_relative "nextbillionai/models/optimization/v2_retrieve_result_response"
+require_relative "nextbillionai/models/optimization/v2_submit_params"
+require_relative "nextbillionai/models/optimization/vehicle"
+require_relative "nextbillionai/models/optimization_compute_params"
+require_relative "nextbillionai/models/optimization_compute_response"
+require_relative "nextbillionai/models/optimization_re_optimize_params"
+require_relative "nextbillionai/models/position"
+require_relative "nextbillionai/models/postalcode_retrieve_coordinates_params"
+require_relative "nextbillionai/models/postalcode_retrieve_coordinates_response"
+require_relative "nextbillionai/models/post_response"
+require_relative "nextbillionai/models/restriction_create_params"
+require_relative "nextbillionai/models/restriction_delete_params"
+require_relative "nextbillionai/models/restriction_delete_response"
+require_relative "nextbillionai/models/restriction_list_by_bbox_params"
+require_relative "nextbillionai/models/restriction_list_by_bbox_response"
+require_relative "nextbillionai/models/restriction_list_params"
+require_relative "nextbillionai/models/restriction_list_response"
+require_relative "nextbillionai/models/restriction_retrieve_params"
+require_relative "nextbillionai/models/restriction_set_state_params"
+require_relative "nextbillionai/models/restrictions_item_list_params"
+require_relative "nextbillionai/models/restrictions_item_list_response"
+require_relative "nextbillionai/models/restriction_update_params"
+require_relative "nextbillionai/models/revgeocode_retrieve_params"
+require_relative "nextbillionai/models/revgeocode_retrieve_response"
+require_relative "nextbillionai/models/rich_group_response"
+require_relative "nextbillionai/models/route_report_create_params"
+require_relative "nextbillionai/models/route_report_create_response"
+require_relative "nextbillionai/models/skynet/asset/event_list_params"
+require_relative "nextbillionai/models/skynet/asset/event_list_response"
+require_relative "nextbillionai/models/skynet/asset/location_get_last_params"
+require_relative "nextbillionai/models/skynet/asset/location_get_last_response"
+require_relative "nextbillionai/models/skynet/asset/location_list_params"
+require_relative "nextbillionai/models/skynet/asset/location_list_response"
+require_relative "nextbillionai/models/skynet/asset/track_location"
+require_relative "nextbillionai/models/skynet/asset_bind_params"
+require_relative "nextbillionai/models/skynet/asset_create_params"
+require_relative "nextbillionai/models/skynet/asset_create_response"
+require_relative "nextbillionai/models/skynet/asset_delete_params"
+require_relative "nextbillionai/models/skynet/asset_details"
+require_relative "nextbillionai/models/skynet/asset_list_params"
+require_relative "nextbillionai/models/skynet/asset_list_response"
+require_relative "nextbillionai/models/skynet/asset_retrieve_params"
+require_relative "nextbillionai/models/skynet/asset_retrieve_response"
+require_relative "nextbillionai/models/skynet/asset_track_params"
+require_relative "nextbillionai/models/skynet/asset_update_attributes_params"
+require_relative "nextbillionai/models/skynet/asset_update_params"
+require_relative "nextbillionai/models/skynet/config_retrieve_params"
+require_relative "nextbillionai/models/skynet/config_retrieve_response"
+require_relative "nextbillionai/models/skynet/config_test_webhook_params"
+require_relative "nextbillionai/models/skynet/config_test_webhook_response"
+require_relative "nextbillionai/models/skynet/config_update_params"
+require_relative "nextbillionai/models/skynet/metadata"
+require_relative "nextbillionai/models/skynet/meta_data"
+require_relative "nextbillionai/models/skynet/monitor_create_params"
+require_relative "nextbillionai/models/skynet/monitor_create_response"
+require_relative "nextbillionai/models/skynet/monitor_delete_params"
+require_relative "nextbillionai/models/skynet/monitor_list_params"
+require_relative "nextbillionai/models/skynet/monitor_list_response"
+require_relative "nextbillionai/models/skynet/monitor_retrieve_params"
+require_relative "nextbillionai/models/skynet/monitor_retrieve_response"
+require_relative "nextbillionai/models/skynet/monitor_update_params"
+require_relative "nextbillionai/models/skynet/namespaced_apikey_create_params"
+require_relative "nextbillionai/models/skynet/namespaced_apikey_create_response"
+require_relative "nextbillionai/models/skynet/namespaced_apikey_delete_params"
+require_relative "nextbillionai/models/skynet/namespaced_apikey_delete_response"
+require_relative "nextbillionai/models/skynet/pagination"
+require_relative "nextbillionai/models/skynet/search/polygon_create_params"
+require_relative "nextbillionai/models/skynet/search/polygon_get_params"
+require_relative "nextbillionai/models/skynet/search_around_params"
+require_relative "nextbillionai/models/skynet/search_bound_params"
+require_relative "nextbillionai/models/skynet/search_response"
+require_relative "nextbillionai/models/skynet/simple_resp"
+require_relative "nextbillionai/models/skynet/monitor"
+require_relative "nextbillionai/models/skynet/trip_delete_params"
+require_relative "nextbillionai/models/skynet/trip_end_params"
+require_relative "nextbillionai/models/skynet/trip_get_summary_params"
+require_relative "nextbillionai/models/skynet/trip_get_summary_response"
+require_relative "nextbillionai/models/skynet/trip_retrieve_params"
+require_relative "nextbillionai/models/skynet/trip_retrieve_response"
+require_relative "nextbillionai/models/skynet/trip_start_params"
+require_relative "nextbillionai/models/skynet/trip_start_response"
+require_relative "nextbillionai/models/skynet/trip_stop"
+require_relative "nextbillionai/models/skynet/trip_update_params"
+require_relative "nextbillionai/models/skynet_subscribe_params"
+require_relative "nextbillionai/models/skynet_subscribe_response"
+require_relative "nextbillionai/models/snap_to_road_snap_params"
+require_relative "nextbillionai/models/snap_to_road_snap_response"
+require_relative "nextbillionai/models"
+require_relative "nextbillionai/resources/areas"
+require_relative "nextbillionai/resources/autocomplete"
+require_relative "nextbillionai/resources/autosuggest"
+require_relative "nextbillionai/resources/batch"
+require_relative "nextbillionai/resources/browse"
+require_relative "nextbillionai/resources/directions"
+require_relative "nextbillionai/resources/discover"
+require_relative "nextbillionai/resources/distance_matrix"
+require_relative "nextbillionai/resources/distance_matrix/json"
+require_relative "nextbillionai/resources/fleetify"
+require_relative "nextbillionai/resources/fleetify/document_templates"
+require_relative "nextbillionai/resources/fleetify/routes"
+require_relative "nextbillionai/resources/fleetify/routes/steps"
+require_relative "nextbillionai/resources/geocode"
+require_relative "nextbillionai/resources/geofence"
+require_relative "nextbillionai/resources/geofence/batch"
+require_relative "nextbillionai/resources/geofence/console"
+require_relative "nextbillionai/resources/isochrone"
+require_relative "nextbillionai/resources/lookup"
+require_relative "nextbillionai/resources/map"
+require_relative "nextbillionai/resources/mdm"
+require_relative "nextbillionai/resources/multigeocode"
+require_relative "nextbillionai/resources/multigeocode/place"
+require_relative "nextbillionai/resources/navigation"
+require_relative "nextbillionai/resources/optimization"
+require_relative "nextbillionai/resources/optimization/driver_assignment"
+require_relative "nextbillionai/resources/optimization/v2"
+require_relative "nextbillionai/resources/postalcode"
+require_relative "nextbillionai/resources/restrictions"
+require_relative "nextbillionai/resources/restrictions_items"
+require_relative "nextbillionai/resources/revgeocode"
+require_relative "nextbillionai/resources/route_report"
+require_relative "nextbillionai/resources/skynet"
+require_relative "nextbillionai/resources/skynet/asset"
+require_relative "nextbillionai/resources/skynet/asset/event"
+require_relative "nextbillionai/resources/skynet/asset/location"
+require_relative "nextbillionai/resources/skynet/config"
+require_relative "nextbillionai/resources/skynet/monitor"
+require_relative "nextbillionai/resources/skynet/namespaced_apikeys"
+require_relative "nextbillionai/resources/skynet/search"
+require_relative "nextbillionai/resources/skynet/search/polygon"
+require_relative "nextbillionai/resources/skynet/trip"
+require_relative "nextbillionai/resources/snap_to_roads"

data/manifest.yaml

@@ -0,0 +1,15 @@
+dependencies:
+  - English
+  - cgi
+  - date
+  - erb
+  - etc
+  - json
+  - net/http
+  - pathname
+  - rbconfig
+  - securerandom
+  - set
+  - stringio
+  - time
+  - uri

data/rbi/nextbillionai/client.rbi

@@ -0,0 +1,121 @@
+# typed: strong
+
+module Nextbillionai
+  class Client < Nextbillionai::Internal::Transport::BaseClient
+    DEFAULT_MAX_RETRIES = 2
+
+    DEFAULT_TIMEOUT_IN_SECONDS = T.let(60.0, Float)
+
+    DEFAULT_INITIAL_RETRY_DELAY = T.let(0.5, Float)
+
+    DEFAULT_MAX_RETRY_DELAY = T.let(8.0, Float)
+
+    sig { returns(String) }
+    attr_reader :api_key
+
+    sig { returns(Nextbillionai::Resources::Fleetify) }
+    attr_reader :fleetify
+
+    sig { returns(Nextbillionai::Resources::Skynet) }
+    attr_reader :skynet
+
+    sig { returns(Nextbillionai::Resources::Geocode) }
+    attr_reader :geocode
+
+    sig { returns(Nextbillionai::Resources::Optimization) }
+    attr_reader :optimization
+
+    sig { returns(Nextbillionai::Resources::Geofence) }
+    attr_reader :geofence
+
+    sig { returns(Nextbillionai::Resources::Discover) }
+    attr_reader :discover
+
+    sig { returns(Nextbillionai::Resources::Browse) }
+    attr_reader :browse
+
+    sig { returns(Nextbillionai::Resources::Mdm) }
+    attr_reader :mdm
+
+    sig { returns(Nextbillionai::Resources::Isochrone) }
+    attr_reader :isochrone
+
+    sig { returns(Nextbillionai::Resources::Restrictions) }
+    attr_reader :restrictions
+
+    sig { returns(Nextbillionai::Resources::RestrictionsItems) }
+    attr_reader :restrictions_items
+
+    sig { returns(Nextbillionai::Resources::DistanceMatrix) }
+    attr_reader :distance_matrix
+
+    sig { returns(Nextbillionai::Resources::Autocomplete) }
+    attr_reader :autocomplete
+
+    sig { returns(Nextbillionai::Resources::Navigation) }
+    attr_reader :navigation
+
+    sig { returns(Nextbillionai::Resources::Map) }
+    attr_reader :map
+
+    sig { returns(Nextbillionai::Resources::Autosuggest) }
+    attr_reader :autosuggest
+
+    sig { returns(Nextbillionai::Resources::Directions) }
+    attr_reader :directions
+
+    sig { returns(Nextbillionai::Resources::Batch) }
+    attr_reader :batch
+
+    sig { returns(Nextbillionai::Resources::Multigeocode) }
+    attr_reader :multigeocode
+
+    sig { returns(Nextbillionai::Resources::Revgeocode) }
+    attr_reader :revgeocode
+
+    sig { returns(Nextbillionai::Resources::RouteReport) }
+    attr_reader :route_report
+
+    sig { returns(Nextbillionai::Resources::SnapToRoads) }
+    attr_reader :snap_to_roads
+
+    sig { returns(Nextbillionai::Resources::Postalcode) }
+    attr_reader :postalcode
+
+    sig { returns(Nextbillionai::Resources::Lookup) }
+    attr_reader :lookup
+
+    sig { returns(Nextbillionai::Resources::Areas) }
+    attr_reader :areas
+
+    # @api private
+    sig { override.returns(T::Hash[String, String]) }
+    private def auth_query
+    end
+
+    # Creates and returns a new client for interacting with the API.
+    sig do
+      params(
+        api_key: T.nilable(String),
+        base_url: T.nilable(String),
+        max_retries: Integer,
+        timeout: Float,
+        initial_retry_delay: Float,
+        max_retry_delay: Float
+      ).returns(T.attached_class)
+    end
+    def self.new(
+      # Defaults to `ENV["NEXTBILLION_SDK_API_KEY"]`
+      api_key: ENV["NEXTBILLION_SDK_API_KEY"],
+      # Override the default base URL for the API, e.g.,
+      # `"https://api.example.com/v2/"`. Defaults to `ENV["NEXTBILLION_SDK_BASE_URL"]`
+      base_url: ENV["NEXTBILLION_SDK_BASE_URL"],
+      # Max number of retries to attempt after a failed retryable request.
+      max_retries: Nextbillionai::Client::DEFAULT_MAX_RETRIES,
+      timeout: Nextbillionai::Client::DEFAULT_TIMEOUT_IN_SECONDS,
+      initial_retry_delay: Nextbillionai::Client::DEFAULT_INITIAL_RETRY_DELAY,
+      max_retry_delay: Nextbillionai::Client::DEFAULT_MAX_RETRY_DELAY
+    )
+    end
+  end
+end

data/rbi/nextbillionai/errors.rbi

@@ -0,0 +1,178 @@
+# typed: strong
+
+module Nextbillionai
+  module Errors
+    class Error < StandardError
+      sig { returns(T.nilable(StandardError)) }
+      attr_accessor :cause
+    end
+
+    class ConversionError < Nextbillionai::Errors::Error
+      sig { returns(T.nilable(StandardError)) }
+      def cause
+      end
+
+      # @api private
+      sig do
+        params(
+          on: T::Class[StandardError],
+          method: Symbol,
+          target: T.anything,
+          value: T.anything,
+          cause: T.nilable(StandardError)
+        ).returns(T.attached_class)
+      end
+      def self.new(on:, method:, target:, value:, cause: nil)
+      end
+    end
+
+    class APIError < Nextbillionai::Errors::Error
+      sig { returns(URI::Generic) }
+      attr_accessor :url
+
+      sig { returns(T.nilable(Integer)) }
+      attr_accessor :status
+
+      sig { returns(T.nilable(T.anything)) }
+      attr_accessor :body
+
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: T.nilable(Integer),
+          body: T.nilable(Object),
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: nil
+      )
+      end
+    end
+
+    class APIConnectionError < Nextbillionai::Errors::APIError
+      sig { void }
+      attr_accessor :status
+
+      sig { void }
+      attr_accessor :body
+
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: NilClass,
+          body: NilClass,
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: "Connection error."
+      )
+      end
+    end
+
+    class APITimeoutError < Nextbillionai::Errors::APIConnectionError
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: NilClass,
+          body: NilClass,
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: "Request timed out."
+      )
+      end
+    end
+
+    class APIStatusError < Nextbillionai::Errors::APIError
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: Integer,
+          body: T.nilable(Object),
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.for(url:, status:, body:, request:, response:, message: nil)
+      end
+
+      sig { returns(Integer) }
+      attr_accessor :status
+
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: Integer,
+          body: T.nilable(Object),
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(url:, status:, body:, request:, response:, message: nil)
+      end
+    end
+
+    class BadRequestError < Nextbillionai::Errors::APIStatusError
+      HTTP_STATUS = 400
+    end
+
+    class AuthenticationError < Nextbillionai::Errors::APIStatusError
+      HTTP_STATUS = 401
+    end
+
+    class PermissionDeniedError < Nextbillionai::Errors::APIStatusError
+      HTTP_STATUS = 403
+    end
+
+    class NotFoundError < Nextbillionai::Errors::APIStatusError
+      HTTP_STATUS = 404
+    end
+
+    class ConflictError < Nextbillionai::Errors::APIStatusError
+      HTTP_STATUS = 409
+    end
+
+    class UnprocessableEntityError < Nextbillionai::Errors::APIStatusError
+      HTTP_STATUS = 422
+    end
+
+    class RateLimitError < Nextbillionai::Errors::APIStatusError
+      HTTP_STATUS = 429
+    end
+
+    class InternalServerError < Nextbillionai::Errors::APIStatusError
+      HTTP_STATUS = T.let((500..), T::Range[Integer])
+    end
+  end
+end

data/rbi/nextbillionai/file_part.rbi

@@ -0,0 +1,37 @@
+# typed: strong
+
+module Nextbillionai
+  class FilePart
+    sig { returns(T.any(Pathname, StringIO, IO, String)) }
+    attr_reader :content
+
+    sig { returns(T.nilable(String)) }
+    attr_reader :content_type
+
+    sig { returns(T.nilable(String)) }
+    attr_reader :filename
+
+    # @api private
+    sig { returns(String) }
+    private def read
+    end
+
+    sig { params(a: T.anything).returns(String) }
+    def to_json(*a)
+    end
+
+    sig { params(a: T.anything).returns(String) }
+    def to_yaml(*a)
+    end
+
+    sig do
+      params(
+        content: T.any(Pathname, StringIO, IO, String),
+        filename: T.nilable(String),
+        content_type: T.nilable(String)
+      ).returns(T.attached_class)
+    end
+    def self.new(content, filename: nil, content_type: nil)
+    end
+  end
+end