vellum_ai 0.3.14 → 0.14.74

data/lib/vellum_ai/types/test_suite_run_metric_error_output.rb

@@ -1,57 +1,75 @@
 # frozen_string_literal: true
-
 require_relative "vellum_error"
+require "ostruct"
 require "json"
 
 module Vellum
-  # Output for a test suite run metric that is of type ERROR
+# Output for a test suite run metric that is of type ERROR
   class TestSuiteRunMetricErrorOutput
-    attr_reader :value, :name, :additional_properties
+  # @return [Vellum::VellumError] 
+    attr_reader :value
+  # @return [String] 
+    attr_reader :type
+  # @return [String] 
+    attr_reader :name
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
 
-    # @param value [VellumError]
-    # @param name [String]
+    # @param value [Vellum::VellumError] 
+    # @param type [String] 
+    # @param name [String] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [TestSuiteRunMetricErrorOutput]
-    def initialize(value:, name:, additional_properties: nil)
-      # @type [VellumError]
+    # @return [Vellum::TestSuiteRunMetricErrorOutput]
+    def initialize(value:, type:, name:, additional_properties: nil)
       @value = value
-      # @type [String]
+      @type = type
       @name = name
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "value": value, "type": type, "name": name }
     end
-
-    # Deserialize a JSON object to an instance of TestSuiteRunMetricErrorOutput
+# Deserialize a JSON object to an instance of TestSuiteRunMetricErrorOutput
     #
-    # @param json_object [JSON]
-    # @return [TestSuiteRunMetricErrorOutput]
+    # @param json_object [String] 
+    # @return [Vellum::TestSuiteRunMetricErrorOutput]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      if parsed_json["value"].nil?
-        value = nil
-      else
+      unless parsed_json["value"].nil?
         value = parsed_json["value"].to_json
-        value = VellumError.from_json(json_object: value)
+        value = Vellum::VellumError.from_json(json_object: value)
+      else
+        value = nil
       end
-      name = struct.name
-      new(value: value, name: name, additional_properties: struct)
+      type = parsed_json["type"]
+      name = parsed_json["name"]
+      new(
+        value: value,
+        type: type,
+        name: name,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of TestSuiteRunMetricErrorOutput to a JSON object
+# Serialize an instance of TestSuiteRunMetricErrorOutput to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "value": @value, "name": @name }.to_json
+    # @return [String]
+    def to_json
+      @_field_set&.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.
+# 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]
+    # @param obj [Object] 
     # @return [Void]
     def self.validate_raw(obj:)
-      VellumError.validate_raw(obj: obj.value)
+      Vellum::VellumError.validate_raw(obj: obj.value)
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
       obj.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/test_suite_run_metric_json_output.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+# Output for a test suite run metric that is of type NUMBER
+  class TestSuiteRunMetricJsonOutput
+  # @return [Object] 
+    attr_reader :value
+  # @return [String] 
+    attr_reader :type
+  # @return [String] 
+    attr_reader :name
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param value [Object] 
+    # @param type [String] 
+    # @param name [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::TestSuiteRunMetricJsonOutput]
+    def initialize(value:, type:, name:, additional_properties: nil)
+      @value = value
+      @type = type
+      @name = name
+      @additional_properties = additional_properties
+      @_field_set = { "value": value, "type": type, "name": name }
+    end
+# Deserialize a JSON object to an instance of TestSuiteRunMetricJsonOutput
+    #
+    # @param json_object [String] 
+    # @return [Vellum::TestSuiteRunMetricJsonOutput]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      value = parsed_json["value"]
+      type = parsed_json["type"]
+      name = parsed_json["name"]
+      new(
+        value: value,
+        type: type,
+        name: name,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of TestSuiteRunMetricJsonOutput to a JSON object
+    #
+    # @return [String]
+    def to_json
+      @_field_set&.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.value.is_a?(Object) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      obj.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/test_suite_run_metric_number_output.rb

@@ -1,51 +1,71 @@
 # frozen_string_literal: true
-
+require "ostruct"
 require "json"
 
 module Vellum
-  # Output for a test suite run metric that is of type NUMBER
+# Output for a test suite run metric that is of type NUMBER
   class TestSuiteRunMetricNumberOutput
-    attr_reader :value, :name, :additional_properties
+  # @return [Float] 
+    attr_reader :value
+  # @return [String] 
+    attr_reader :type
+  # @return [String] 
+    attr_reader :name
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
 
-    # @param value [Float]
-    # @param name [String]
+    # @param value [Float] 
+    # @param type [String] 
+    # @param name [String] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [TestSuiteRunMetricNumberOutput]
-    def initialize(value:, name:, additional_properties: nil)
-      # @type [Float]
-      @value = value
-      # @type [String]
+    # @return [Vellum::TestSuiteRunMetricNumberOutput]
+    def initialize(value: OMIT, type:, name:, additional_properties: nil)
+      @value = value if value != OMIT
+      @type = type
       @name = name
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "value": value, "type": type, "name": name }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of TestSuiteRunMetricNumberOutput
+# Deserialize a JSON object to an instance of TestSuiteRunMetricNumberOutput
     #
-    # @param json_object [JSON]
-    # @return [TestSuiteRunMetricNumberOutput]
+    # @param json_object [String] 
+    # @return [Vellum::TestSuiteRunMetricNumberOutput]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      JSON.parse(json_object)
-      value = struct.value
-      name = struct.name
-      new(value: value, name: name, additional_properties: struct)
+      parsed_json = JSON.parse(json_object)
+      value = parsed_json["value"]
+      type = parsed_json["type"]
+      name = parsed_json["name"]
+      new(
+        value: value,
+        type: type,
+        name: name,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of TestSuiteRunMetricNumberOutput to a JSON object
+# Serialize an instance of TestSuiteRunMetricNumberOutput to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "value": @value, "name": @name }.to_json
+    # @return [String]
+    def to_json
+      @_field_set&.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.
+# 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]
+    # @param obj [Object] 
     # @return [Void]
     def self.validate_raw(obj:)
-      obj.value.is_a?(Float) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+      obj.value&.is_a?(Float) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
       obj.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/test_suite_run_metric_output.rb

@@ -1,90 +1,106 @@
 # frozen_string_literal: true
-
 require "json"
+require_relative "test_suite_run_metric_string_output"
 require_relative "test_suite_run_metric_number_output"
+require_relative "test_suite_run_metric_json_output"
 require_relative "test_suite_run_metric_error_output"
+require_relative "test_suite_run_metric_array_output"
 
 module Vellum
   class TestSuiteRunMetricOutput
-    attr_reader :member, :discriminant
 
-    private_class_method :new
-    alias kind_of? is_a?
-    # @param member [Object]
-    # @param discriminant [String]
-    # @return [TestSuiteRunMetricOutput]
-    def initialize(member:, discriminant:)
-      # @type [Object]
-      @member = member
-      # @type [String]
-      @discriminant = discriminant
-    end
 
-    # Deserialize a JSON object to an instance of TestSuiteRunMetricOutput
+# Deserialize a JSON object to an instance of TestSuiteRunMetricOutput
     #
-    # @param json_object [JSON]
-    # @return [TestSuiteRunMetricOutput]
+    # @param json_object [String] 
+    # @return [Vellum::TestSuiteRunMetricOutput]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      member = case struct.type
-               when "NUMBER"
-                 TestSuiteRunMetricNumberOutput.from_json(json_object: json_object)
-               when "ERROR"
-                 TestSuiteRunMetricErrorOutput.from_json(json_object: json_object)
-               else
-                 TestSuiteRunMetricNumberOutput.from_json(json_object: json_object)
-               end
-      new(member: member, discriminant: struct.type)
-    end
-
-    # For Union Types, to_json functionality is delegated to the wrapped member.
-    #
-    # @return [JSON]
-    def to_json(*_args)
-      case @discriminant
-      when "NUMBER"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "ERROR"
-        { **@member.to_json, type: @discriminant }.to_json
-      else
-        { "type": @discriminant, value: @member }.to_json
+      begin
+        Vellum::TestSuiteRunMetricStringOutput.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::TestSuiteRunMetricStringOutput.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::TestSuiteRunMetricNumberOutput.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::TestSuiteRunMetricNumberOutput.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::TestSuiteRunMetricJsonOutput.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::TestSuiteRunMetricJsonOutput.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::TestSuiteRunMetricErrorOutput.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::TestSuiteRunMetricErrorOutput.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::TestSuiteRunMetricArrayOutput.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::TestSuiteRunMetricArrayOutput.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
       end
-      @member.to_json
+ return struct
     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.
+# 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]
+    # @param obj [Object] 
     # @return [Void]
     def self.validate_raw(obj:)
-      case obj.type
-      when "NUMBER"
-        TestSuiteRunMetricNumberOutput.validate_raw(obj: obj)
-      when "ERROR"
-        TestSuiteRunMetricErrorOutput.validate_raw(obj: obj)
-      else
-        raise("Passed value matched no type within the union, validation failed.")
+      begin
+        return Vellum::TestSuiteRunMetricStringOutput.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
       end
-    end
-
-    # For Union Types, is_a? functionality is delegated to the wrapped member.
-    #
-    # @param obj [Object]
-    # @return [Boolean]
-    def is_a?(obj)
-      @member.is_a?(obj)
-    end
-
-    # @param member [TestSuiteRunMetricNumberOutput]
-    # @return [TestSuiteRunMetricOutput]
-    def self.number(member:)
-      new(member: member, discriminant: "NUMBER")
-    end
-
-    # @param member [TestSuiteRunMetricErrorOutput]
-    # @return [TestSuiteRunMetricOutput]
-    def self.error(member:)
-      new(member: member, discriminant: "ERROR")
+      begin
+        return Vellum::TestSuiteRunMetricNumberOutput.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::TestSuiteRunMetricJsonOutput.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::TestSuiteRunMetricErrorOutput.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::TestSuiteRunMetricArrayOutput.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      raise("Passed value matched no type within the union, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/test_suite_run_metric_string_output.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+# Output for a test suite run metric that is of type STRING
+  class TestSuiteRunMetricStringOutput
+  # @return [String] 
+    attr_reader :value
+  # @return [String] 
+    attr_reader :type
+  # @return [String] 
+    attr_reader :name
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param value [String] 
+    # @param type [String] 
+    # @param name [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::TestSuiteRunMetricStringOutput]
+    def initialize(value: OMIT, type:, name:, additional_properties: nil)
+      @value = value if value != OMIT
+      @type = type
+      @name = name
+      @additional_properties = additional_properties
+      @_field_set = { "value": value, "type": type, "name": name }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of TestSuiteRunMetricStringOutput
+    #
+    # @param json_object [String] 
+    # @return [Vellum::TestSuiteRunMetricStringOutput]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      value = parsed_json["value"]
+      type = parsed_json["type"]
+      name = parsed_json["name"]
+      new(
+        value: value,
+        type: type,
+        name: name,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of TestSuiteRunMetricStringOutput to a JSON object
+    #
+    # @return [String]
+    def to_json
+      @_field_set&.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.value&.is_a?(String) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      obj.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/test_suite_run_progress.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+  class TestSuiteRunProgress
+  # @return [Integer] 
+    attr_reader :number_of_requested_test_cases
+  # @return [Integer] 
+    attr_reader :number_of_completed_test_cases
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param number_of_requested_test_cases [Integer] 
+    # @param number_of_completed_test_cases [Integer] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::TestSuiteRunProgress]
+    def initialize(number_of_requested_test_cases:, number_of_completed_test_cases:, additional_properties: nil)
+      @number_of_requested_test_cases = number_of_requested_test_cases
+      @number_of_completed_test_cases = number_of_completed_test_cases
+      @additional_properties = additional_properties
+      @_field_set = { "number_of_requested_test_cases": number_of_requested_test_cases, "number_of_completed_test_cases": number_of_completed_test_cases }
+    end
+# Deserialize a JSON object to an instance of TestSuiteRunProgress
+    #
+    # @param json_object [String] 
+    # @return [Vellum::TestSuiteRunProgress]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      number_of_requested_test_cases = parsed_json["number_of_requested_test_cases"]
+      number_of_completed_test_cases = parsed_json["number_of_completed_test_cases"]
+      new(
+        number_of_requested_test_cases: number_of_requested_test_cases,
+        number_of_completed_test_cases: number_of_completed_test_cases,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of TestSuiteRunProgress to a JSON object
+    #
+    # @return [String]
+    def to_json
+      @_field_set&.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.number_of_requested_test_cases.is_a?(Integer) != false || raise("Passed value for field obj.number_of_requested_test_cases is not the expected type, validation failed.")
+      obj.number_of_completed_test_cases.is_a?(Integer) != false || raise("Passed value for field obj.number_of_completed_test_cases is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/test_suite_run_prompt_sandbox_exec_config_data_request.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+  class TestSuiteRunPromptSandboxExecConfigDataRequest
+  # @return [String] The ID of the Prompt Sandbox to run the Test Suite against.
+    attr_reader :prompt_sandbox_id
+  # @return [String] The ID of the Prompt Variant within the Prompt Sandbox that you'd like to run
+#  the Test Suite against.
+    attr_reader :prompt_variant_id
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param prompt_sandbox_id [String] The ID of the Prompt Sandbox to run the Test Suite against.
+    # @param prompt_variant_id [String] The ID of the Prompt Variant within the Prompt Sandbox that you'd like to run
+#  the Test Suite against.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::TestSuiteRunPromptSandboxExecConfigDataRequest]
+    def initialize(prompt_sandbox_id:, prompt_variant_id:, additional_properties: nil)
+      @prompt_sandbox_id = prompt_sandbox_id
+      @prompt_variant_id = prompt_variant_id
+      @additional_properties = additional_properties
+      @_field_set = { "prompt_sandbox_id": prompt_sandbox_id, "prompt_variant_id": prompt_variant_id }
+    end
+# Deserialize a JSON object to an instance of
+#  TestSuiteRunPromptSandboxExecConfigDataRequest
+    #
+    # @param json_object [String] 
+    # @return [Vellum::TestSuiteRunPromptSandboxExecConfigDataRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      prompt_sandbox_id = parsed_json["prompt_sandbox_id"]
+      prompt_variant_id = parsed_json["prompt_variant_id"]
+      new(
+        prompt_sandbox_id: prompt_sandbox_id,
+        prompt_variant_id: prompt_variant_id,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of TestSuiteRunPromptSandboxExecConfigDataRequest to a
+#  JSON object
+    #
+    # @return [String]
+    def to_json
+      @_field_set&.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.prompt_sandbox_id.is_a?(String) != false || raise("Passed value for field obj.prompt_sandbox_id is not the expected type, validation failed.")
+      obj.prompt_variant_id.is_a?(String) != false || raise("Passed value for field obj.prompt_variant_id is not the expected type, validation failed.")
+    end
+  end
+end