class Vellum::ExecutePromptEvent

def self.from_json(json_object:)
  struct = JSON.parse(json_object, object_class: OpenStruct)
  member = case struct.state
           when "INITIATED"
             InitiatedExecutePromptEvent.from_json(json_object: json_object)
           when "STREAMING"
             StreamingExecutePromptEvent.from_json(json_object: json_object)
           when "FULFILLED"
             FulfilledExecutePromptEvent.from_json(json_object: json_object)
           when "REJECTED"
             RejectedExecutePromptEvent.from_json(json_object: json_object)
           else
             InitiatedExecutePromptEvent.from_json(json_object: json_object)
           end
  new(member: member, discriminant: struct.state)
end

def self.fulfilled(member:)
  new(member: member, discriminant: "FULFILLED")
end

def self.initiated(member:)
  new(member: member, discriminant: "INITIATED")
end

def self.rejected(member:)
  new(member: member, discriminant: "REJECTED")
end

def self.streaming(member:)
  new(member: member, discriminant: "STREAMING")
end

def self.validate_raw(obj:)
  case obj.state
  when "INITIATED"
    InitiatedExecutePromptEvent.validate_raw(obj: obj)
  when "STREAMING"
    StreamingExecutePromptEvent.validate_raw(obj: obj)
  when "FULFILLED"
    FulfilledExecutePromptEvent.validate_raw(obj: obj)
  when "REJECTED"
    RejectedExecutePromptEvent.validate_raw(obj: obj)
  else
    raise("Passed value matched no type within the union, validation failed.")
  end
end

def initialize(member:, discriminant:)
  # @type [Object]
  @member = member
  # @type [String]
  @discriminant = discriminant
end

def is_a?(obj)
  @member.is_a?(obj)
end

def to_json(*_args)
  case @discriminant
  when "INITIATED"
    { **@member.to_json, state: @discriminant }.to_json
  when "STREAMING"
    { **@member.to_json, state: @discriminant }.to_json
  when "FULFILLED"
    { **@member.to_json, state: @discriminant }.to_json
  when "REJECTED"
    { **@member.to_json, state: @discriminant }.to_json
  else
    { "state": @discriminant, value: @member }.to_json
  end
  @member.to_json
end