vellum_ai 0.3.4

data/lib/vellum_ai/types/prompt_template_block_properties.rb

@@ -0,0 +1,104 @@
+# frozen_string_literal: true
+
+require_relative "chat_message_role"
+require_relative "vellum_variable_type"
+require_relative "prompt_template_block"
+require "json"
+
+module Vellum
+  class PromptTemplateBlockProperties
+    attr_reader :chat_role, :chat_message_unterminated, :template, :template_type, :function_name,
+                :function_description, :function_parameters, :function_forced, :blocks, :additional_properties
+
+    # @param chat_role [CHAT_MESSAGE_ROLE]
+    # @param chat_message_unterminated [Boolean]
+    # @param template [String]
+    # @param template_type [VELLUM_VARIABLE_TYPE]
+    # @param function_name [String]
+    # @param function_description [String]
+    # @param function_parameters [Hash{String => String}]
+    # @param function_forced [Boolean]
+    # @param blocks [Array<PromptTemplateBlock>]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [PromptTemplateBlockProperties]
+    def initialize(chat_role: nil, chat_message_unterminated: nil, template: nil, template_type: nil,
+                   function_name: nil, function_description: nil, function_parameters: nil, function_forced: nil, blocks: nil, additional_properties: nil)
+      # @type [CHAT_MESSAGE_ROLE]
+      @chat_role = chat_role
+      # @type [Boolean]
+      @chat_message_unterminated = chat_message_unterminated
+      # @type [String]
+      @template = template
+      # @type [VELLUM_VARIABLE_TYPE]
+      @template_type = template_type
+      # @type [String]
+      @function_name = function_name
+      # @type [String]
+      @function_description = function_description
+      # @type [Hash{String => String}]
+      @function_parameters = function_parameters
+      # @type [Boolean]
+      @function_forced = function_forced
+      # @type [Array<PromptTemplateBlock>]
+      @blocks = blocks
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of PromptTemplateBlockProperties
+    #
+    # @param json_object [JSON]
+    # @return [PromptTemplateBlockProperties]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      chat_role = CHAT_MESSAGE_ROLE.key(parsed_json["chat_role"]) || parsed_json["chat_role"]
+      chat_message_unterminated = struct.chat_message_unterminated
+      template = struct.template
+      template_type = VELLUM_VARIABLE_TYPE.key(parsed_json["template_type"]) || parsed_json["template_type"]
+      function_name = struct.function_name
+      function_description = struct.function_description
+      function_parameters = struct.function_parameters
+      function_forced = struct.function_forced
+      blocks = parsed_json["blocks"].map do |v|
+        v = v.to_json
+        PromptTemplateBlock.from_json(json_object: v)
+      end
+      new(chat_role: chat_role, chat_message_unterminated: chat_message_unterminated, template: template,
+          template_type: template_type, function_name: function_name, function_description: function_description, function_parameters: function_parameters, function_forced: function_forced, blocks: blocks, additional_properties: struct)
+    end
+
+    # Serialize an instance of PromptTemplateBlockProperties to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      {
+        "chat_role": CHAT_MESSAGE_ROLE[@chat_role] || @chat_role,
+        "chat_message_unterminated": @chat_message_unterminated,
+        "template": @template,
+        "template_type": VELLUM_VARIABLE_TYPE[@template_type] || @template_type,
+        "function_name": @function_name,
+        "function_description": @function_description,
+        "function_parameters": @function_parameters,
+        "function_forced": @function_forced,
+        "blocks": @blocks
+      }.to_json
+    end
+
+    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+    #
+    # @param obj [Object]
+    # @return [Void]
+    def self.validate_raw(obj:)
+      obj.chat_role&.is_a?(CHAT_MESSAGE_ROLE) != false || raise("Passed value for field obj.chat_role is not the expected type, validation failed.")
+      obj.chat_message_unterminated&.is_a?(Boolean) != false || raise("Passed value for field obj.chat_message_unterminated is not the expected type, validation failed.")
+      obj.template&.is_a?(String) != false || raise("Passed value for field obj.template is not the expected type, validation failed.")
+      obj.template_type&.is_a?(VELLUM_VARIABLE_TYPE) != false || raise("Passed value for field obj.template_type is not the expected type, validation failed.")
+      obj.function_name&.is_a?(String) != false || raise("Passed value for field obj.function_name is not the expected type, validation failed.")
+      obj.function_description&.is_a?(String) != false || raise("Passed value for field obj.function_description is not the expected type, validation failed.")
+      obj.function_parameters&.is_a?(Hash) != false || raise("Passed value for field obj.function_parameters is not the expected type, validation failed.")
+      obj.function_forced&.is_a?(Boolean) != false || raise("Passed value for field obj.function_forced is not the expected type, validation failed.")
+      obj.blocks&.is_a?(Array) != false || raise("Passed value for field obj.blocks is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/prompt_template_block_properties_request.rb

@@ -0,0 +1,104 @@
+# frozen_string_literal: true
+
+require_relative "chat_message_role"
+require_relative "vellum_variable_type"
+require_relative "prompt_template_block_request"
+require "json"
+
+module Vellum
+  class PromptTemplateBlockPropertiesRequest
+    attr_reader :chat_role, :chat_message_unterminated, :template, :template_type, :function_name,
+                :function_description, :function_parameters, :function_forced, :blocks, :additional_properties
+
+    # @param chat_role [CHAT_MESSAGE_ROLE]
+    # @param chat_message_unterminated [Boolean]
+    # @param template [String]
+    # @param template_type [VELLUM_VARIABLE_TYPE]
+    # @param function_name [String]
+    # @param function_description [String]
+    # @param function_parameters [Hash{String => String}]
+    # @param function_forced [Boolean]
+    # @param blocks [Array<PromptTemplateBlockRequest>]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [PromptTemplateBlockPropertiesRequest]
+    def initialize(chat_role: nil, chat_message_unterminated: nil, template: nil, template_type: nil,
+                   function_name: nil, function_description: nil, function_parameters: nil, function_forced: nil, blocks: nil, additional_properties: nil)
+      # @type [CHAT_MESSAGE_ROLE]
+      @chat_role = chat_role
+      # @type [Boolean]
+      @chat_message_unterminated = chat_message_unterminated
+      # @type [String]
+      @template = template
+      # @type [VELLUM_VARIABLE_TYPE]
+      @template_type = template_type
+      # @type [String]
+      @function_name = function_name
+      # @type [String]
+      @function_description = function_description
+      # @type [Hash{String => String}]
+      @function_parameters = function_parameters
+      # @type [Boolean]
+      @function_forced = function_forced
+      # @type [Array<PromptTemplateBlockRequest>]
+      @blocks = blocks
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of PromptTemplateBlockPropertiesRequest
+    #
+    # @param json_object [JSON]
+    # @return [PromptTemplateBlockPropertiesRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      chat_role = CHAT_MESSAGE_ROLE.key(parsed_json["chat_role"]) || parsed_json["chat_role"]
+      chat_message_unterminated = struct.chat_message_unterminated
+      template = struct.template
+      template_type = VELLUM_VARIABLE_TYPE.key(parsed_json["template_type"]) || parsed_json["template_type"]
+      function_name = struct.function_name
+      function_description = struct.function_description
+      function_parameters = struct.function_parameters
+      function_forced = struct.function_forced
+      blocks = parsed_json["blocks"].map do |v|
+        v = v.to_json
+        PromptTemplateBlockRequest.from_json(json_object: v)
+      end
+      new(chat_role: chat_role, chat_message_unterminated: chat_message_unterminated, template: template,
+          template_type: template_type, function_name: function_name, function_description: function_description, function_parameters: function_parameters, function_forced: function_forced, blocks: blocks, additional_properties: struct)
+    end
+
+    # Serialize an instance of PromptTemplateBlockPropertiesRequest to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      {
+        "chat_role": CHAT_MESSAGE_ROLE[@chat_role] || @chat_role,
+        "chat_message_unterminated": @chat_message_unterminated,
+        "template": @template,
+        "template_type": VELLUM_VARIABLE_TYPE[@template_type] || @template_type,
+        "function_name": @function_name,
+        "function_description": @function_description,
+        "function_parameters": @function_parameters,
+        "function_forced": @function_forced,
+        "blocks": @blocks
+      }.to_json
+    end
+
+    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+    #
+    # @param obj [Object]
+    # @return [Void]
+    def self.validate_raw(obj:)
+      obj.chat_role&.is_a?(CHAT_MESSAGE_ROLE) != false || raise("Passed value for field obj.chat_role is not the expected type, validation failed.")
+      obj.chat_message_unterminated&.is_a?(Boolean) != false || raise("Passed value for field obj.chat_message_unterminated is not the expected type, validation failed.")
+      obj.template&.is_a?(String) != false || raise("Passed value for field obj.template is not the expected type, validation failed.")
+      obj.template_type&.is_a?(VELLUM_VARIABLE_TYPE) != false || raise("Passed value for field obj.template_type is not the expected type, validation failed.")
+      obj.function_name&.is_a?(String) != false || raise("Passed value for field obj.function_name is not the expected type, validation failed.")
+      obj.function_description&.is_a?(String) != false || raise("Passed value for field obj.function_description is not the expected type, validation failed.")
+      obj.function_parameters&.is_a?(Hash) != false || raise("Passed value for field obj.function_parameters is not the expected type, validation failed.")
+      obj.function_forced&.is_a?(Boolean) != false || raise("Passed value for field obj.function_forced is not the expected type, validation failed.")
+      obj.blocks&.is_a?(Array) != false || raise("Passed value for field obj.blocks is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/prompt_template_block_request.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require_relative "block_type_enum"
+require_relative "prompt_template_block_properties_request"
+require "json"
+
+module Vellum
+  class PromptTemplateBlockRequest
+    attr_reader :id, :block_type, :properties, :additional_properties
+
+    # @param id [String]
+    # @param block_type [BLOCK_TYPE_ENUM]
+    # @param properties [PromptTemplateBlockPropertiesRequest]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [PromptTemplateBlockRequest]
+    def initialize(id:, block_type:, properties:, additional_properties: nil)
+      # @type [String]
+      @id = id
+      # @type [BLOCK_TYPE_ENUM]
+      @block_type = block_type
+      # @type [PromptTemplateBlockPropertiesRequest]
+      @properties = properties
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of PromptTemplateBlockRequest
+    #
+    # @param json_object [JSON]
+    # @return [PromptTemplateBlockRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      id = struct.id
+      block_type = BLOCK_TYPE_ENUM.key(parsed_json["block_type"]) || parsed_json["block_type"]
+      if parsed_json["properties"].nil?
+        properties = nil
+      else
+        properties = parsed_json["properties"].to_json
+        properties = PromptTemplateBlockPropertiesRequest.from_json(json_object: properties)
+      end
+      new(id: id, block_type: block_type, properties: properties, additional_properties: struct)
+    end
+
+    # Serialize an instance of PromptTemplateBlockRequest to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "id": @id, "block_type": BLOCK_TYPE_ENUM[@block_type] || @block_type, "properties": @properties }.to_json
+    end
+
+    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+    #
+    # @param obj [Object]
+    # @return [Void]
+    def self.validate_raw(obj:)
+      obj.id.is_a?(String) != false || raise("Passed value for field obj.id is not the expected type, validation failed.")
+      obj.block_type.is_a?(BLOCK_TYPE_ENUM) != false || raise("Passed value for field obj.block_type is not the expected type, validation failed.")
+      PromptTemplateBlockPropertiesRequest.validate_raw(obj: obj.properties)
+    end
+  end
+end

data/lib/vellum_ai/types/provider_enum.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module Vellum
+  # @type [PROVIDER_ENUM]
+  PROVIDER_ENUM = {
+    anthropic: "ANTHROPIC",
+    aws_bedrock: "AWS_BEDROCK",
+    azure_openai: "AZURE_OPENAI",
+    cohere: "COHERE",
+    google: "GOOGLE",
+    hosted: "HOSTED",
+    mosaicml: "MOSAICML",
+    openai: "OPENAI",
+    fireworks_ai: "FIREWORKS_AI",
+    huggingface: "HUGGINGFACE",
+    mystic: "MYSTIC",
+    pyq: "PYQ",
+    replicate: "REPLICATE"
+  }.freeze
+end

data/lib/vellum_ai/types/raw_prompt_execution_overrides_request.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "json"
+
+module Vellum
+  class RawPromptExecutionOverridesRequest
+    attr_reader :body, :headers, :url, :additional_properties
+
+    # @param body [Hash{String => String}]
+    # @param headers [Hash{String => String}] The raw headers to send to the model host.
+    # @param url [String] The raw URL to send to the model host.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [RawPromptExecutionOverridesRequest]
+    def initialize(body: nil, headers: nil, url: nil, additional_properties: nil)
+      # @type [Hash{String => String}]
+      @body = body
+      # @type [Hash{String => String}] The raw headers to send to the model host.
+      @headers = headers
+      # @type [String] The raw URL to send to the model host.
+      @url = url
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of RawPromptExecutionOverridesRequest
+    #
+    # @param json_object [JSON]
+    # @return [RawPromptExecutionOverridesRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      JSON.parse(json_object)
+      body = struct.body
+      headers = struct.headers
+      url = struct.url
+      new(body: body, headers: headers, url: url, additional_properties: struct)
+    end
+
+    # Serialize an instance of RawPromptExecutionOverridesRequest to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "body": @body, "headers": @headers, "url": @url }.to_json
+    end
+
+    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+    #
+    # @param obj [Object]
+    # @return [Void]
+    def self.validate_raw(obj:)
+      obj.body&.is_a?(Hash) != false || raise("Passed value for field obj.body is not the expected type, validation failed.")
+      obj.headers&.is_a?(Hash) != false || raise("Passed value for field obj.headers is not the expected type, validation failed.")
+      obj.url&.is_a?(String) != false || raise("Passed value for field obj.url is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/register_prompt_error_response.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require "json"
+
+module Vellum
+  class RegisterPromptErrorResponse
+    attr_reader :detail, :additional_properties
+
+    # @param detail [String] Details about why the request failed.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [RegisterPromptErrorResponse]
+    def initialize(detail:, additional_properties: nil)
+      # @type [String] Details about why the request failed.
+      @detail = detail
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of RegisterPromptErrorResponse
+    #
+    # @param json_object [JSON]
+    # @return [RegisterPromptErrorResponse]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      JSON.parse(json_object)
+      detail = struct.detail
+      new(detail: detail, additional_properties: struct)
+    end
+
+    # Serialize an instance of RegisterPromptErrorResponse to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "detail": @detail }.to_json
+    end
+
+    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+    #
+    # @param obj [Object]
+    # @return [Void]
+    def self.validate_raw(obj:)
+      obj.detail.is_a?(String) != false || raise("Passed value for field obj.detail is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/register_prompt_model_parameters_request.rb

@@ -0,0 +1,98 @@
+# frozen_string_literal: true
+
+require "json"
+
+module Vellum
+  class RegisterPromptModelParametersRequest
+    attr_reader :temperature, :max_tokens, :stop, :top_p, :top_k, :frequency_penalty, :presence_penalty, :logit_bias,
+                :custom_parameters, :additional_properties
+
+    # @param temperature [Float]
+    # @param max_tokens [Integer]
+    # @param stop [Array<String>]
+    # @param top_p [Float]
+    # @param top_k [Integer]
+    # @param frequency_penalty [Float]
+    # @param presence_penalty [Float]
+    # @param logit_bias [Hash{String => String}]
+    # @param custom_parameters [Hash{String => String}]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [RegisterPromptModelParametersRequest]
+    def initialize(temperature:, max_tokens:, top_p:, frequency_penalty:, presence_penalty:, stop: nil, top_k: nil,
+                   logit_bias: nil, custom_parameters: nil, additional_properties: nil)
+      # @type [Float]
+      @temperature = temperature
+      # @type [Integer]
+      @max_tokens = max_tokens
+      # @type [Array<String>]
+      @stop = stop
+      # @type [Float]
+      @top_p = top_p
+      # @type [Integer]
+      @top_k = top_k
+      # @type [Float]
+      @frequency_penalty = frequency_penalty
+      # @type [Float]
+      @presence_penalty = presence_penalty
+      # @type [Hash{String => String}]
+      @logit_bias = logit_bias
+      # @type [Hash{String => String}]
+      @custom_parameters = custom_parameters
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of RegisterPromptModelParametersRequest
+    #
+    # @param json_object [JSON]
+    # @return [RegisterPromptModelParametersRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      JSON.parse(json_object)
+      temperature = struct.temperature
+      max_tokens = struct.max_tokens
+      stop = struct.stop
+      top_p = struct.top_p
+      top_k = struct.top_k
+      frequency_penalty = struct.frequency_penalty
+      presence_penalty = struct.presence_penalty
+      logit_bias = struct.logit_bias
+      custom_parameters = struct.custom_parameters
+      new(temperature: temperature, max_tokens: max_tokens, stop: stop, top_p: top_p, top_k: top_k,
+          frequency_penalty: frequency_penalty, presence_penalty: presence_penalty, logit_bias: logit_bias, custom_parameters: custom_parameters, additional_properties: struct)
+    end
+
+    # Serialize an instance of RegisterPromptModelParametersRequest to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      {
+        "temperature": @temperature,
+        "max_tokens": @max_tokens,
+        "stop": @stop,
+        "top_p": @top_p,
+        "top_k": @top_k,
+        "frequency_penalty": @frequency_penalty,
+        "presence_penalty": @presence_penalty,
+        "logit_bias": @logit_bias,
+        "custom_parameters": @custom_parameters
+      }.to_json
+    end
+
+    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+    #
+    # @param obj [Object]
+    # @return [Void]
+    def self.validate_raw(obj:)
+      obj.temperature.is_a?(Float) != false || raise("Passed value for field obj.temperature is not the expected type, validation failed.")
+      obj.max_tokens.is_a?(Integer) != false || raise("Passed value for field obj.max_tokens is not the expected type, validation failed.")
+      obj.stop&.is_a?(Array) != false || raise("Passed value for field obj.stop is not the expected type, validation failed.")
+      obj.top_p.is_a?(Float) != false || raise("Passed value for field obj.top_p is not the expected type, validation failed.")
+      obj.top_k&.is_a?(Integer) != false || raise("Passed value for field obj.top_k is not the expected type, validation failed.")
+      obj.frequency_penalty.is_a?(Float) != false || raise("Passed value for field obj.frequency_penalty is not the expected type, validation failed.")
+      obj.presence_penalty.is_a?(Float) != false || raise("Passed value for field obj.presence_penalty is not the expected type, validation failed.")
+      obj.logit_bias&.is_a?(Hash) != false || raise("Passed value for field obj.logit_bias is not the expected type, validation failed.")
+      obj.custom_parameters&.is_a?(Hash) != false || raise("Passed value for field obj.custom_parameters is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/register_prompt_prompt.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require "json"
+
+module Vellum
+  class RegisterPromptPrompt
+    attr_reader :id, :label, :additional_properties
+
+    # @param id [String] The ID of the generated prompt.
+    # @param label [String] A human-friendly label for the generated prompt.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [RegisterPromptPrompt]
+    def initialize(id:, label:, additional_properties: nil)
+      # @type [String] The ID of the generated prompt.
+      @id = id
+      # @type [String] A human-friendly label for the generated prompt.
+      @label = label
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of RegisterPromptPrompt
+    #
+    # @param json_object [JSON]
+    # @return [RegisterPromptPrompt]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      JSON.parse(json_object)
+      id = struct.id
+      label = struct.label
+      new(id: id, label: label, additional_properties: struct)
+    end
+
+    # Serialize an instance of RegisterPromptPrompt to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "id": @id, "label": @label }.to_json
+    end
+
+    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+    #
+    # @param obj [Object]
+    # @return [Void]
+    def self.validate_raw(obj:)
+      obj.id.is_a?(String) != false || raise("Passed value for field obj.id is not the expected type, validation failed.")
+      obj.label.is_a?(String) != false || raise("Passed value for field obj.label is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/register_prompt_prompt_info_request.rb

@@ -0,0 +1,60 @@
+# frozen_string_literal: true
+
+require_relative "prompt_template_block_data_request"
+require_relative "registered_prompt_input_variable_request"
+require "json"
+
+module Vellum
+  class RegisterPromptPromptInfoRequest
+    attr_reader :prompt_block_data, :input_variables, :additional_properties
+
+    # @param prompt_block_data [PromptTemplateBlockDataRequest]
+    # @param input_variables [Array<RegisteredPromptInputVariableRequest>] The input variables specified in the prompt template.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [RegisterPromptPromptInfoRequest]
+    def initialize(prompt_block_data:, input_variables:, additional_properties: nil)
+      # @type [PromptTemplateBlockDataRequest]
+      @prompt_block_data = prompt_block_data
+      # @type [Array<RegisteredPromptInputVariableRequest>] The input variables specified in the prompt template.
+      @input_variables = input_variables
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of RegisterPromptPromptInfoRequest
+    #
+    # @param json_object [JSON]
+    # @return [RegisterPromptPromptInfoRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      if parsed_json["prompt_block_data"].nil?
+        prompt_block_data = nil
+      else
+        prompt_block_data = parsed_json["prompt_block_data"].to_json
+        prompt_block_data = PromptTemplateBlockDataRequest.from_json(json_object: prompt_block_data)
+      end
+      input_variables = parsed_json["input_variables"].map do |v|
+        v = v.to_json
+        RegisteredPromptInputVariableRequest.from_json(json_object: v)
+      end
+      new(prompt_block_data: prompt_block_data, input_variables: input_variables, additional_properties: struct)
+    end
+
+    # Serialize an instance of RegisterPromptPromptInfoRequest to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "prompt_block_data": @prompt_block_data, "input_variables": @input_variables }.to_json
+    end
+
+    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+    #
+    # @param obj [Object]
+    # @return [Void]
+    def self.validate_raw(obj:)
+      PromptTemplateBlockDataRequest.validate_raw(obj: obj.prompt_block_data)
+      obj.input_variables.is_a?(Array) != false || raise("Passed value for field obj.input_variables is not the expected type, validation failed.")
+    end
+  end
+end