dexa-ai-sdk 0.3.5

data/lib/dexa-ai-sdk/tasks/client.rb

@@ -0,0 +1,190 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    class Client
+      # @param client [DexaAiSdk::Internal::Http::RawClient]
+      #
+      # @return [void]
+      def initialize(client:)
+        @client = client
+      end
+
+      # @param request_options [Hash]
+      # @param params [DexaAiSdk::Types::RunTasksRequest]
+      # @option request_options [String] :base_url
+      # @option request_options [Hash{String => Object}] :additional_headers
+      # @option request_options [Hash{String => Object}] :additional_query_parameters
+      # @option request_options [Hash{String => Object}] :additional_body_parameters
+      # @option request_options [Integer] :timeout_in_seconds
+      #
+      # @example
+      #   client.tasks.run_tasks(
+      #     model: "model",
+      #     tasks: []
+      #   )
+      #
+      # @return [DexaAiSdk::Types::RunTasksResponse]
+      def run_tasks(request_options: {}, **params)
+        params = DexaAiSdk::Internal::Types::Utils.normalize_keys(params)
+        request = DexaAiSdk::Internal::JSON::Request.new(
+          base_url: request_options[:base_url],
+          method: "POST",
+          path: "v1/tasks",
+          body: DexaAiSdk::Types::RunTasksRequest.new(params).to_h,
+          request_options: request_options
+        )
+        begin
+          response = @client.send(request)
+        rescue Net::HTTPRequestTimeout
+          raise DexaAiSdk::Errors::TimeoutError
+        end
+        code = response.code.to_i
+        if code.between?(200, 299)
+          (response.body.to_s.empty? ? nil : DexaAiSdk::Types::RunTasksResponse.load(response.body))
+        else
+          error_class = DexaAiSdk::Errors::ResponseError.subclass_for_code(code)
+          error_types = {
+            400 => DexaAiSdk::Types::ErrorResponse,
+            422 => DexaAiSdk::Types::ErrorResponse,
+            500 => DexaAiSdk::Types::ErrorResponse
+          }
+          error_body = DexaAiSdk::Errors::ResponseError.load_error_body(code, response.body, error_types)
+          raise error_class.new(response.body, code: code, body: error_body)
+        end
+      end
+
+      # @param request_options [Hash]
+      # @param params [DexaAiSdk::Types::RunTasksRequest]
+      # @option request_options [String] :base_url
+      # @option request_options [Hash{String => Object}] :additional_headers
+      # @option request_options [Hash{String => Object}] :additional_query_parameters
+      # @option request_options [Hash{String => Object}] :additional_body_parameters
+      # @option request_options [Integer] :timeout_in_seconds
+      #
+      # @example
+      #   client.tasks.decide_extract(
+      #     model: "model",
+      #     tasks: []
+      #   )
+      #
+      # @return [DexaAiSdk::Types::RunTasksResponse]
+      def decide_extract(request_options: {}, **params)
+        params = DexaAiSdk::Internal::Types::Utils.normalize_keys(params)
+        request = DexaAiSdk::Internal::JSON::Request.new(
+          base_url: request_options[:base_url],
+          method: "POST",
+          path: "v1/decide-extract",
+          body: DexaAiSdk::Types::RunTasksRequest.new(params).to_h,
+          request_options: request_options
+        )
+        begin
+          response = @client.send(request)
+        rescue Net::HTTPRequestTimeout
+          raise DexaAiSdk::Errors::TimeoutError
+        end
+        code = response.code.to_i
+        if code.between?(200, 299)
+          (response.body.to_s.empty? ? nil : DexaAiSdk::Types::RunTasksResponse.load(response.body))
+        else
+          error_class = DexaAiSdk::Errors::ResponseError.subclass_for_code(code)
+          error_types = {
+            400 => DexaAiSdk::Types::ErrorResponse,
+            422 => DexaAiSdk::Types::ErrorResponse,
+            500 => DexaAiSdk::Types::ErrorResponse
+          }
+          error_body = DexaAiSdk::Errors::ResponseError.load_error_body(code, response.body, error_types)
+          raise error_class.new(response.body, code: code, body: error_body)
+        end
+      end
+
+      # @param request_options [Hash]
+      # @param params [DexaAiSdk::Tasks::Types::DecideRequest]
+      # @option request_options [String] :base_url
+      # @option request_options [Hash{String => Object}] :additional_headers
+      # @option request_options [Hash{String => Object}] :additional_query_parameters
+      # @option request_options [Hash{String => Object}] :additional_body_parameters
+      # @option request_options [Integer] :timeout_in_seconds
+      #
+      # @example
+      #   client.tasks.decide(
+      #     model: "model",
+      #     question: "question"
+      #   )
+      #
+      # @return [DexaAiSdk::Types::DecideResponse]
+      def decide(request_options: {}, **params)
+        params = DexaAiSdk::Internal::Types::Utils.normalize_keys(params)
+        request = DexaAiSdk::Internal::JSON::Request.new(
+          base_url: request_options[:base_url],
+          method: "POST",
+          path: "decide",
+          body: DexaAiSdk::Tasks::Types::DecideRequest.new(params).to_h,
+          request_options: request_options
+        )
+        begin
+          response = @client.send(request)
+        rescue Net::HTTPRequestTimeout
+          raise DexaAiSdk::Errors::TimeoutError
+        end
+        code = response.code.to_i
+        if code.between?(200, 299)
+          (response.body.to_s.empty? ? nil : DexaAiSdk::Types::DecideResponse.load(response.body))
+        else
+          error_class = DexaAiSdk::Errors::ResponseError.subclass_for_code(code)
+          error_types = {
+            400 => DexaAiSdk::Types::ErrorResponse,
+            422 => DexaAiSdk::Types::ErrorResponse,
+            500 => DexaAiSdk::Types::ErrorResponse
+          }
+          error_body = DexaAiSdk::Errors::ResponseError.load_error_body(code, response.body, error_types)
+          raise error_class.new(response.body, code: code, body: error_body)
+        end
+      end
+
+      # @param request_options [Hash]
+      # @param params [DexaAiSdk::Tasks::Types::ExtractRequest]
+      # @option request_options [String] :base_url
+      # @option request_options [Hash{String => Object}] :additional_headers
+      # @option request_options [Hash{String => Object}] :additional_query_parameters
+      # @option request_options [Hash{String => Object}] :additional_body_parameters
+      # @option request_options [Integer] :timeout_in_seconds
+      #
+      # @example
+      #   client.tasks.extract(
+      #     model: "model",
+      #     instruction: "instruction"
+      #   )
+      #
+      # @return [DexaAiSdk::Types::ExtractResponse]
+      def extract(request_options: {}, **params)
+        params = DexaAiSdk::Internal::Types::Utils.normalize_keys(params)
+        request = DexaAiSdk::Internal::JSON::Request.new(
+          base_url: request_options[:base_url],
+          method: "POST",
+          path: "extract",
+          body: DexaAiSdk::Tasks::Types::ExtractRequest.new(params).to_h,
+          request_options: request_options
+        )
+        begin
+          response = @client.send(request)
+        rescue Net::HTTPRequestTimeout
+          raise DexaAiSdk::Errors::TimeoutError
+        end
+        code = response.code.to_i
+        if code.between?(200, 299)
+          (response.body.to_s.empty? ? nil : DexaAiSdk::Types::ExtractResponse.load(response.body))
+        else
+          error_class = DexaAiSdk::Errors::ResponseError.subclass_for_code(code)
+          error_types = {
+            400 => DexaAiSdk::Types::ErrorResponse,
+            422 => DexaAiSdk::Types::ErrorResponse,
+            500 => DexaAiSdk::Types::ErrorResponse
+          }
+          error_body = DexaAiSdk::Errors::ResponseError.load_error_body(code, response.body, error_types)
+          raise error_class.new(response.body, code: code, body: error_body)
+        end
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/decide_request.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      class DecideRequest < Internal::Types::Model
+        field :model, -> { String }, optional: false, nullable: false
+
+        field :context, -> { String }, optional: true, nullable: false
+
+        field :messages, -> { Internal::Types::Array[DexaAiSdk::Types::Message] }, optional: true, nullable: false
+
+        field :images, -> { Internal::Types::Array[DexaAiSdk::Types::ImageInput] }, optional: true, nullable: false
+
+        field :audio, -> { Internal::Types::Array[DexaAiSdk::Types::AudioInput] }, optional: true, nullable: false
+
+        field :question, -> { String }, optional: false, nullable: false
+
+        field :choices, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+
+        field :options, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+
+        field :instruction, -> { String }, optional: true, nullable: false
+
+        field :temperature, -> { Float }, optional: true, nullable: false
+
+        field :max_output_tokens, -> { Integer }, optional: true, nullable: false
+
+        field :response_format, -> { DexaAiSdk::Tasks::Types::DecideRequestResponseFormat }, optional: true, nullable: false
+
+        field :target_language, -> { DexaAiSdk::Tasks::Types::DecideRequestTargetLanguage }, optional: true, nullable: false
+
+        field :output_language, -> { DexaAiSdk::Tasks::Types::DecideRequestOutputLanguage }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/decide_request_output_language.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      module DecideRequestOutputLanguage
+        extend DexaAiSdk::Internal::Types::Enum
+
+        EN = "en"
+        ES = "es"
+        FR = "fr"
+        DE = "de"
+        IT = "it"
+        PT = "pt"
+        NL = "nl"
+        RU = "ru"
+        JA = "ja"
+        KO = "ko"
+        ZH = "zh"
+        AR = "ar"
+        HI = "hi"
+        TR = "tr"
+        PL = "pl"
+        SV = "sv"
+        DA = "da"
+        NO = "no"
+        FI = "fi"
+        CS = "cs"
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/decide_request_response_format.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      class DecideRequestResponseFormat < Internal::Types::Model
+        field :type, -> { DexaAiSdk::Tasks::Types::DecideRequestResponseFormatType }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/decide_request_response_format_type.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      module DecideRequestResponseFormatType
+        extend DexaAiSdk::Internal::Types::Enum
+
+        JSON_OBJECT = "json_object"
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/decide_request_target_language.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      module DecideRequestTargetLanguage
+        extend DexaAiSdk::Internal::Types::Enum
+
+        EN = "en"
+        ES = "es"
+        FR = "fr"
+        DE = "de"
+        IT = "it"
+        PT = "pt"
+        NL = "nl"
+        RU = "ru"
+        JA = "ja"
+        KO = "ko"
+        ZH = "zh"
+        AR = "ar"
+        HI = "hi"
+        TR = "tr"
+        PL = "pl"
+        SV = "sv"
+        DA = "da"
+        NO = "no"
+        FI = "fi"
+        CS = "cs"
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/extract_request.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      class ExtractRequest < Internal::Types::Model
+        field :model, -> { String }, optional: false, nullable: false
+
+        field :context, -> { String }, optional: true, nullable: false
+
+        field :messages, -> { Internal::Types::Array[DexaAiSdk::Types::Message] }, optional: true, nullable: false
+
+        field :images, -> { Internal::Types::Array[DexaAiSdk::Types::ImageInput] }, optional: true, nullable: false
+
+        field :audio, -> { Internal::Types::Array[DexaAiSdk::Types::AudioInput] }, optional: true, nullable: false
+
+        field :instruction, -> { String }, optional: false, nullable: false
+
+        field :fields, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+
+        field :keys, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+
+        field :schema, -> { Internal::Types::Hash[String, Object] }, optional: true, nullable: false
+
+        field :temperature, -> { Float }, optional: true, nullable: false
+
+        field :max_output_tokens, -> { Integer }, optional: true, nullable: false
+
+        field :response_format, -> { DexaAiSdk::Tasks::Types::ExtractRequestResponseFormat }, optional: true, nullable: false
+
+        field :target_language, -> { DexaAiSdk::Tasks::Types::ExtractRequestTargetLanguage }, optional: true, nullable: false
+
+        field :output_language, -> { DexaAiSdk::Tasks::Types::ExtractRequestOutputLanguage }, optional: true, nullable: false
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/extract_request_output_language.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      module ExtractRequestOutputLanguage
+        extend DexaAiSdk::Internal::Types::Enum
+
+        EN = "en"
+        ES = "es"
+        FR = "fr"
+        DE = "de"
+        IT = "it"
+        PT = "pt"
+        NL = "nl"
+        RU = "ru"
+        JA = "ja"
+        KO = "ko"
+        ZH = "zh"
+        AR = "ar"
+        HI = "hi"
+        TR = "tr"
+        PL = "pl"
+        SV = "sv"
+        DA = "da"
+        NO = "no"
+        FI = "fi"
+        CS = "cs"
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/extract_request_response_format.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      class ExtractRequestResponseFormat < Internal::Types::Model
+        field :type, -> { DexaAiSdk::Tasks::Types::ExtractRequestResponseFormatType }, optional: false, nullable: false
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/extract_request_response_format_type.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      module ExtractRequestResponseFormatType
+        extend DexaAiSdk::Internal::Types::Enum
+
+        JSON_OBJECT = "json_object"
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/tasks/types/extract_request_target_language.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Tasks
+    module Types
+      module ExtractRequestTargetLanguage
+        extend DexaAiSdk::Internal::Types::Enum
+
+        EN = "en"
+        ES = "es"
+        FR = "fr"
+        DE = "de"
+        IT = "it"
+        PT = "pt"
+        NL = "nl"
+        RU = "ru"
+        JA = "ja"
+        KO = "ko"
+        ZH = "zh"
+        AR = "ar"
+        HI = "hi"
+        TR = "tr"
+        PL = "pl"
+        SV = "sv"
+        DA = "da"
+        NO = "no"
+        FI = "fi"
+        CS = "cs"
+      end
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/audio_input.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    class AudioInput < Internal::Types::Model
+      field :type, -> { DexaAiSdk::Types::AudioInputType }, optional: true, nullable: false
+
+      field :url, -> { String }, optional: true, nullable: false
+
+      field :data, -> { String }, optional: true, nullable: false
+
+      field :mime_type, -> { String }, optional: true, nullable: false
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/audio_input_type.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    module AudioInputType
+      extend DexaAiSdk::Internal::Types::Enum
+
+      URL = "url"
+      BASE64 = "base64"
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/choice.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    class Choice < Internal::Types::Model
+      field :message, -> { DexaAiSdk::Types::ResponseMessage }, optional: false, nullable: false
+
+      field :finish_reason, -> { DexaAiSdk::Types::ChoiceFinishReason }, optional: false, nullable: false
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/choice_finish_reason.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    module ChoiceFinishReason
+      extend DexaAiSdk::Internal::Types::Enum
+
+      STOP = "stop"
+      LENGTH = "length"
+      CONTENT_FILTER = "content_filter"
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/content_part.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    class ContentPart < Internal::Types::Model
+      field :type, -> { DexaAiSdk::Types::ContentPartType }, optional: true, nullable: false
+
+      field :text, -> { String }, optional: true, nullable: false
+
+      field :language, -> { DexaAiSdk::Types::ContentPartLanguage }, optional: true, nullable: false
+
+      field :image_url, -> { DexaAiSdk::Types::ImageURL }, optional: true, nullable: false
+
+      field :image, -> { DexaAiSdk::Types::ImageInput }, optional: true, nullable: false
+
+      field :audio, -> { DexaAiSdk::Types::AudioInput }, optional: true, nullable: false
+
+      field :input_audio, -> { DexaAiSdk::Types::AudioInput }, optional: true, nullable: false
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/content_part_language.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    module ContentPartLanguage
+      extend DexaAiSdk::Internal::Types::Enum
+
+      EN = "en"
+      ES = "es"
+      FR = "fr"
+      DE = "de"
+      IT = "it"
+      PT = "pt"
+      NL = "nl"
+      RU = "ru"
+      JA = "ja"
+      KO = "ko"
+      ZH = "zh"
+      AR = "ar"
+      HI = "hi"
+      TR = "tr"
+      PL = "pl"
+      SV = "sv"
+      DA = "da"
+      NO = "no"
+      FI = "fi"
+      CS = "cs"
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/content_part_type.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    module ContentPartType
+      extend DexaAiSdk::Internal::Types::Enum
+
+      TEXT = "text"
+      IMAGE_URL = "image_url"
+      IMAGE = "image"
+      AUDIO = "audio"
+      INPUT_AUDIO = "input_audio"
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/decide_response.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    class DecideResponse < Internal::Types::Model
+      field :id, -> { String }, optional: false, nullable: false
+
+      field :model, -> { String }, optional: false, nullable: false
+
+      field :choices, -> { Internal::Types::Array[DexaAiSdk::Types::Choice] }, optional: false, nullable: false
+
+      field :result, -> { DexaAiSdk::Types::DecideResult }, optional: false, nullable: false
+
+      field :decision, -> { DexaAiSdk::Types::DecideResult }, optional: true, nullable: false
+
+      field :value, -> { String }, optional: true, nullable: false
+
+      field :usage, -> { DexaAiSdk::Types::Usage }, optional: false, nullable: false
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/decide_result.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    class DecideResult < Internal::Types::Model
+      field :value, -> { String }, optional: false, nullable: false
+
+      field :reason, -> { String }, optional: true, nullable: false
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/decide_task.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    class DecideTask < Internal::Types::Model
+      field :name, -> { String }, optional: false, nullable: false
+
+      field :id, -> { String }, optional: true, nullable: false
+
+      field :question, -> { String }, optional: false, nullable: false
+
+      field :choices, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+
+      field :options, -> { Internal::Types::Array[String] }, optional: true, nullable: false
+
+      field :instruction, -> { String }, optional: true, nullable: false
+
+      field :target_language, -> { DexaAiSdk::Types::DecideTaskTargetLanguage }, optional: true, nullable: false
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/decide_task_target_language.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    module DecideTaskTargetLanguage
+      extend DexaAiSdk::Internal::Types::Enum
+
+      EN = "en"
+      ES = "es"
+      FR = "fr"
+      DE = "de"
+      IT = "it"
+      PT = "pt"
+      NL = "nl"
+      RU = "ru"
+      JA = "ja"
+      KO = "ko"
+      ZH = "zh"
+      AR = "ar"
+      HI = "hi"
+      TR = "tr"
+      PL = "pl"
+      SV = "sv"
+      DA = "da"
+      NO = "no"
+      FI = "fi"
+      CS = "cs"
+    end
+  end
+end

data/lib/dexa-ai-sdk/types/error_response.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module DexaAiSdk
+  module Types
+    class ErrorResponse < Internal::Types::Model
+      field :error, -> { DexaAiSdk::Types::ErrorResponseError }, optional: false, nullable: false
+    end
+  end
+end