vellum_ai 0.3.4

data/lib/vellum_ai/sandboxes/client.rb

@@ -0,0 +1,132 @@
+# frozen_string_literal: true
+
+require_relative "../../requests"
+require_relative "../types/scenario_input_request"
+require_relative "../types/sandbox_scenario"
+require "async"
+
+module Vellum
+  class SandboxesClient
+    attr_reader :request_client
+
+    # @param request_client [RequestClient]
+    # @return [SandboxesClient]
+    def initialize(request_client:)
+      # @type [RequestClient]
+      @request_client = request_client
+    end
+
+    # Upserts a new scenario for a sandbox, keying off of the optionally provided scenario id.
+    #
+    # If an id is provided and has a match, the scenario will be updated. If no id is provided or no match
+    # is found, a new scenario will be appended to the end.
+    #
+    # Note that a full replacement of the scenario is performed, so any fields not provided will be removed
+    # or overwritten with default values.
+    #
+    # @param id [String] A UUID string identifying this sandbox.
+    # @param label [String]
+    # @param inputs [Array<Hash>] The inputs for the scenarioRequest of type Array<ScenarioInputRequest>, as a Hash
+    #   * :key (String)
+    #   * :type (SCENARIO_INPUT_TYPE_ENUM)
+    #   * :value (String)
+    #   * :chat_history (Array<ChatMessageRequest>)
+    # @param scenario_id [String] The id of the scenario to update. If none is provided, an id will be generated and a new scenario will be appended.
+    # @param request_options [RequestOptions]
+    # @return [SandboxScenario]
+    def upsert_sandbox_scenario(id:, inputs:, label: nil, scenario_id: nil, request_options: nil)
+      response = @request_client.conn.post do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.body = {
+          **(request_options&.additional_body_parameters || {}),
+          label: label,
+          inputs: inputs,
+          scenario_id: scenario_id
+        }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/sandboxes/#{id}/scenarios"
+      end
+      SandboxScenario.from_json(json_object: response.body)
+    end
+
+    # Deletes an existing scenario from a sandbox, keying off of the provided scenario id.
+    #
+    # @param id [String] A UUID string identifying this sandbox.
+    # @param scenario_id [String] An id identifying the scenario that you'd like to delete
+    # @param request_options [RequestOptions]
+    # @return [Void]
+    def delete_sandbox_scenario(id:, scenario_id:, request_options: nil)
+      @request_client.conn.delete do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/sandboxes/#{id}/scenarios/#{scenario_id}"
+      end
+    end
+  end
+
+  class AsyncSandboxesClient
+    attr_reader :request_client
+
+    # @param request_client [AsyncRequestClient]
+    # @return [AsyncSandboxesClient]
+    def initialize(request_client:)
+      # @type [AsyncRequestClient]
+      @request_client = request_client
+    end
+
+    # Upserts a new scenario for a sandbox, keying off of the optionally provided scenario id.
+    #
+    # If an id is provided and has a match, the scenario will be updated. If no id is provided or no match
+    # is found, a new scenario will be appended to the end.
+    #
+    # Note that a full replacement of the scenario is performed, so any fields not provided will be removed
+    # or overwritten with default values.
+    #
+    # @param id [String] A UUID string identifying this sandbox.
+    # @param label [String]
+    # @param inputs [Array<Hash>] The inputs for the scenarioRequest of type Array<ScenarioInputRequest>, as a Hash
+    #   * :key (String)
+    #   * :type (SCENARIO_INPUT_TYPE_ENUM)
+    #   * :value (String)
+    #   * :chat_history (Array<ChatMessageRequest>)
+    # @param scenario_id [String] The id of the scenario to update. If none is provided, an id will be generated and a new scenario will be appended.
+    # @param request_options [RequestOptions]
+    # @return [SandboxScenario]
+    def upsert_sandbox_scenario(id:, inputs:, label: nil, scenario_id: nil, request_options: nil)
+      Async do
+        response = @request_client.conn.post do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.body = {
+            **(request_options&.additional_body_parameters || {}),
+            label: label,
+            inputs: inputs,
+            scenario_id: scenario_id
+          }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/sandboxes/#{id}/scenarios"
+        end
+        SandboxScenario.from_json(json_object: response.body)
+      end
+    end
+
+    # Deletes an existing scenario from a sandbox, keying off of the provided scenario id.
+    #
+    # @param id [String] A UUID string identifying this sandbox.
+    # @param scenario_id [String] An id identifying the scenario that you'd like to delete
+    # @param request_options [RequestOptions]
+    # @return [Void]
+    def delete_sandbox_scenario(id:, scenario_id:, request_options: nil)
+      Async do
+        @request_client.conn.delete do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/sandboxes/#{id}/scenarios/#{scenario_id}"
+        end
+      end
+    end
+  end
+end

data/lib/vellum_ai/test_suites/client.rb

@@ -0,0 +1,130 @@
+# frozen_string_literal: true
+
+require_relative "../../requests"
+require_relative "../types/named_test_case_variable_value_request"
+require_relative "../types/test_suite_test_case"
+require "async"
+
+module Vellum
+  class TestSuitesClient
+    attr_reader :request_client
+
+    # @param request_client [RequestClient]
+    # @return [TestSuitesClient]
+    def initialize(request_client:)
+      # @type [RequestClient]
+      @request_client = request_client
+    end
+
+    # Upserts a new test case for a test suite, keying off of the optionally provided test case id.
+    #
+    # If an id is provided and has a match, the test case will be updated. If no id is provided or no match
+    # is found, a new test case will be appended to the end.
+    #
+    # Note that a full replacement of the test case is performed, so any fields not provided will be removed
+    # or overwritten with default values.
+    #
+    # @param id [String] A UUID string identifying this test suite.
+    # @param upsert_test_suite_test_case_request_id [String]
+    # @param label [String]
+    # @param input_values [Array<Hash>] Request of type Array<NamedTestCaseVariableValueRequest>, as a Hash
+    # @param evaluation_values [Array<Hash>] Request of type Array<NamedTestCaseVariableValueRequest>, as a Hash
+    # @param request_options [RequestOptions]
+    # @return [TestSuiteTestCase]
+    def upsert_test_suite_test_case(id:, input_values:, evaluation_values:,
+                                    upsert_test_suite_test_case_request_id: nil, label: nil, request_options: nil)
+      response = @request_client.conn.post do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.body = {
+          **(request_options&.additional_body_parameters || {}),
+          id: upsert_test_suite_test_case_request_id,
+          label: label,
+          input_values: input_values,
+          evaluation_values: evaluation_values
+        }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/test-suites/#{id}/test-cases"
+      end
+      TestSuiteTestCase.from_json(json_object: response.body)
+    end
+
+    # Deletes an existing test case for a test suite, keying off of the test case id.
+    #
+    # @param id [String] A UUID string identifying this test suite.
+    # @param test_case_id [String] An id identifying the test case that you'd like to delete
+    # @param request_options [RequestOptions]
+    # @return [Void]
+    def delete_test_suite_test_case(id:, test_case_id:, request_options: nil)
+      @request_client.conn.delete do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/test-suites/#{id}/test-cases/#{test_case_id}"
+      end
+    end
+  end
+
+  class AsyncTestSuitesClient
+    attr_reader :request_client
+
+    # @param request_client [AsyncRequestClient]
+    # @return [AsyncTestSuitesClient]
+    def initialize(request_client:)
+      # @type [AsyncRequestClient]
+      @request_client = request_client
+    end
+
+    # Upserts a new test case for a test suite, keying off of the optionally provided test case id.
+    #
+    # If an id is provided and has a match, the test case will be updated. If no id is provided or no match
+    # is found, a new test case will be appended to the end.
+    #
+    # Note that a full replacement of the test case is performed, so any fields not provided will be removed
+    # or overwritten with default values.
+    #
+    # @param id [String] A UUID string identifying this test suite.
+    # @param upsert_test_suite_test_case_request_id [String]
+    # @param label [String]
+    # @param input_values [Array<Hash>] Request of type Array<NamedTestCaseVariableValueRequest>, as a Hash
+    # @param evaluation_values [Array<Hash>] Request of type Array<NamedTestCaseVariableValueRequest>, as a Hash
+    # @param request_options [RequestOptions]
+    # @return [TestSuiteTestCase]
+    def upsert_test_suite_test_case(id:, input_values:, evaluation_values:,
+                                    upsert_test_suite_test_case_request_id: nil, label: nil, request_options: nil)
+      Async do
+        response = @request_client.conn.post do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.body = {
+            **(request_options&.additional_body_parameters || {}),
+            id: upsert_test_suite_test_case_request_id,
+            label: label,
+            input_values: input_values,
+            evaluation_values: evaluation_values
+          }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/test-suites/#{id}/test-cases"
+        end
+        TestSuiteTestCase.from_json(json_object: response.body)
+      end
+    end
+
+    # Deletes an existing test case for a test suite, keying off of the test case id.
+    #
+    # @param id [String] A UUID string identifying this test suite.
+    # @param test_case_id [String] An id identifying the test case that you'd like to delete
+    # @param request_options [RequestOptions]
+    # @return [Void]
+    def delete_test_suite_test_case(id:, test_case_id:, request_options: nil)
+      Async do
+        @request_client.conn.delete do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/test-suites/#{id}/test-cases/#{test_case_id}"
+        end
+      end
+    end
+  end
+end

data/lib/vellum_ai/types/api_node_result.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require_relative "api_node_result_data"
+require "json"
+
+module Vellum
+  # A Node Result Event emitted from an API Node.
+  class ApiNodeResult
+    attr_reader :data, :additional_properties
+
+    # @param data [ApiNodeResultData]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [ApiNodeResult]
+    def initialize(data:, additional_properties: nil)
+      # @type [ApiNodeResultData]
+      @data = data
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of ApiNodeResult
+    #
+    # @param json_object [JSON]
+    # @return [ApiNodeResult]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      if parsed_json["data"].nil?
+        data = nil
+      else
+        data = parsed_json["data"].to_json
+        data = ApiNodeResultData.from_json(json_object: data)
+      end
+      new(data: data, additional_properties: struct)
+    end
+
+    # Serialize an instance of ApiNodeResult to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "data": @data }.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:)
+      ApiNodeResultData.validate_raw(obj: obj.data)
+    end
+  end
+end

data/lib/vellum_ai/types/api_node_result_data.rb

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

data/lib/vellum_ai/types/array_chat_message_content.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require_relative "array_chat_message_content_item"
+require "json"
+
+module Vellum
+  # A list of chat message content items.
+  class ArrayChatMessageContent
+    attr_reader :value, :additional_properties
+
+    # @param value [Array<ArrayChatMessageContentItem>]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [ArrayChatMessageContent]
+    def initialize(value:, additional_properties: nil)
+      # @type [Array<ArrayChatMessageContentItem>]
+      @value = value
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of ArrayChatMessageContent
+    #
+    # @param json_object [JSON]
+    # @return [ArrayChatMessageContent]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      value = parsed_json["value"].map do |v|
+        v = v.to_json
+        ArrayChatMessageContentItem.from_json(json_object: v)
+      end
+      new(value: value, additional_properties: struct)
+    end
+
+    # Serialize an instance of ArrayChatMessageContent to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "value": @value }.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?(Array) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/array_chat_message_content_item.rb

@@ -0,0 +1,103 @@
+# frozen_string_literal: true
+
+require "json"
+require_relative "string_chat_message_content"
+require_relative "function_call_chat_message_content"
+require_relative "image_chat_message_content"
+
+module Vellum
+  class ArrayChatMessageContentItem
+    attr_reader :member, :discriminant
+
+    private_class_method :new
+    alias kind_of? is_a?
+    # @param member [Object]
+    # @param discriminant [String]
+    # @return [ArrayChatMessageContentItem]
+    def initialize(member:, discriminant:)
+      # @type [Object]
+      @member = member
+      # @type [String]
+      @discriminant = discriminant
+    end
+
+    # Deserialize a JSON object to an instance of ArrayChatMessageContentItem
+    #
+    # @param json_object [JSON]
+    # @return [ArrayChatMessageContentItem]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      member = case struct.type
+               when "STRING"
+                 StringChatMessageContent.from_json(json_object: json_object)
+               when "FUNCTION_CALL"
+                 FunctionCallChatMessageContent.from_json(json_object: json_object)
+               when "IMAGE"
+                 ImageChatMessageContent.from_json(json_object: json_object)
+               else
+                 StringChatMessageContent.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 "STRING"
+        { **@member.to_json, type: @discriminant }.to_json
+      when "FUNCTION_CALL"
+        { **@member.to_json, type: @discriminant }.to_json
+      when "IMAGE"
+        { **@member.to_json, type: @discriminant }.to_json
+      else
+        { "type": @discriminant, value: @member }.to_json
+      end
+      @member.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:)
+      case obj.type
+      when "STRING"
+        StringChatMessageContent.validate_raw(obj: obj)
+      when "FUNCTION_CALL"
+        FunctionCallChatMessageContent.validate_raw(obj: obj)
+      when "IMAGE"
+        ImageChatMessageContent.validate_raw(obj: obj)
+      else
+        raise("Passed value matched no type within the union, validation failed.")
+      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 [StringChatMessageContent]
+    # @return [ArrayChatMessageContentItem]
+    def self.string(member:)
+      new(member: member, discriminant: "STRING")
+    end
+
+    # @param member [FunctionCallChatMessageContent]
+    # @return [ArrayChatMessageContentItem]
+    def self.function_call(member:)
+      new(member: member, discriminant: "FUNCTION_CALL")
+    end
+
+    # @param member [ImageChatMessageContent]
+    # @return [ArrayChatMessageContentItem]
+    def self.image(member:)
+      new(member: member, discriminant: "IMAGE")
+    end
+  end
+end

data/lib/vellum_ai/types/array_chat_message_content_item_request.rb

@@ -0,0 +1,103 @@
+# frozen_string_literal: true
+
+require "json"
+require_relative "string_chat_message_content_request"
+require_relative "function_call_chat_message_content_request"
+require_relative "image_chat_message_content_request"
+
+module Vellum
+  class ArrayChatMessageContentItemRequest
+    attr_reader :member, :discriminant
+
+    private_class_method :new
+    alias kind_of? is_a?
+    # @param member [Object]
+    # @param discriminant [String]
+    # @return [ArrayChatMessageContentItemRequest]
+    def initialize(member:, discriminant:)
+      # @type [Object]
+      @member = member
+      # @type [String]
+      @discriminant = discriminant
+    end
+
+    # Deserialize a JSON object to an instance of ArrayChatMessageContentItemRequest
+    #
+    # @param json_object [JSON]
+    # @return [ArrayChatMessageContentItemRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      member = case struct.type
+               when "STRING"
+                 StringChatMessageContentRequest.from_json(json_object: json_object)
+               when "FUNCTION_CALL"
+                 FunctionCallChatMessageContentRequest.from_json(json_object: json_object)
+               when "IMAGE"
+                 ImageChatMessageContentRequest.from_json(json_object: json_object)
+               else
+                 StringChatMessageContentRequest.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 "STRING"
+        { **@member.to_json, type: @discriminant }.to_json
+      when "FUNCTION_CALL"
+        { **@member.to_json, type: @discriminant }.to_json
+      when "IMAGE"
+        { **@member.to_json, type: @discriminant }.to_json
+      else
+        { "type": @discriminant, value: @member }.to_json
+      end
+      @member.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:)
+      case obj.type
+      when "STRING"
+        StringChatMessageContentRequest.validate_raw(obj: obj)
+      when "FUNCTION_CALL"
+        FunctionCallChatMessageContentRequest.validate_raw(obj: obj)
+      when "IMAGE"
+        ImageChatMessageContentRequest.validate_raw(obj: obj)
+      else
+        raise("Passed value matched no type within the union, validation failed.")
+      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 [StringChatMessageContentRequest]
+    # @return [ArrayChatMessageContentItemRequest]
+    def self.string(member:)
+      new(member: member, discriminant: "STRING")
+    end
+
+    # @param member [FunctionCallChatMessageContentRequest]
+    # @return [ArrayChatMessageContentItemRequest]
+    def self.function_call(member:)
+      new(member: member, discriminant: "FUNCTION_CALL")
+    end
+
+    # @param member [ImageChatMessageContentRequest]
+    # @return [ArrayChatMessageContentItemRequest]
+    def self.image(member:)
+      new(member: member, discriminant: "IMAGE")
+    end
+  end
+end