vellum_ai 0.3.15 → 0.14.75

data/lib/vellum_ai/types/workflow_output.rb

@@ -1,5 +1,4 @@
 # frozen_string_literal: true
-
 require "json"
 require_relative "workflow_output_string"
 require_relative "workflow_output_number"
@@ -13,169 +12,159 @@ require_relative "workflow_output_image"
 
 module Vellum
   class WorkflowOutput
-    attr_reader :member, :discriminant
 
-    private_class_method :new
-    alias kind_of? is_a?
-    # @param member [Object]
-    # @param discriminant [String]
-    # @return [WorkflowOutput]
-    def initialize(member:, discriminant:)
-      # @type [Object]
-      @member = member
-      # @type [String]
-      @discriminant = discriminant
-    end
 
-    # Deserialize a JSON object to an instance of WorkflowOutput
+# Deserialize a JSON object to an instance of WorkflowOutput
     #
-    # @param json_object [JSON]
-    # @return [WorkflowOutput]
+    # @param json_object [String] 
+    # @return [Vellum::WorkflowOutput]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      member = case struct.type
-               when "STRING"
-                 WorkflowOutputString.from_json(json_object: json_object)
-               when "NUMBER"
-                 WorkflowOutputNumber.from_json(json_object: json_object)
-               when "JSON"
-                 WorkflowOutputJson.from_json(json_object: json_object)
-               when "CHAT_HISTORY"
-                 WorkflowOutputChatHistory.from_json(json_object: json_object)
-               when "SEARCH_RESULTS"
-                 WorkflowOutputSearchResults.from_json(json_object: json_object)
-               when "ARRAY"
-                 WorkflowOutputArray.from_json(json_object: json_object)
-               when "ERROR"
-                 WorkflowOutputError.from_json(json_object: json_object)
-               when "FUNCTION_CALL"
-                 WorkflowOutputFunctionCall.from_json(json_object: json_object)
-               when "IMAGE"
-                 WorkflowOutputImage.from_json(json_object: json_object)
-               else
-                 WorkflowOutputString.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 "NUMBER"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "JSON"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "CHAT_HISTORY"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "SEARCH_RESULTS"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "ARRAY"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "ERROR"
-        { **@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
+      begin
+        Vellum::WorkflowOutputString.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowOutputString.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::WorkflowOutputNumber.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowOutputNumber.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::WorkflowOutputJson.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowOutputJson.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::WorkflowOutputChatHistory.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowOutputChatHistory.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::WorkflowOutputSearchResults.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowOutputSearchResults.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::WorkflowOutputArray.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowOutputArray.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::WorkflowOutputError.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowOutputError.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::WorkflowOutputFunctionCall.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowOutputFunctionCall.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::WorkflowOutputImage.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowOutputImage.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+ 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 "STRING"
-        WorkflowOutputString.validate_raw(obj: obj)
-      when "NUMBER"
-        WorkflowOutputNumber.validate_raw(obj: obj)
-      when "JSON"
-        WorkflowOutputJson.validate_raw(obj: obj)
-      when "CHAT_HISTORY"
-        WorkflowOutputChatHistory.validate_raw(obj: obj)
-      when "SEARCH_RESULTS"
-        WorkflowOutputSearchResults.validate_raw(obj: obj)
-      when "ARRAY"
-        WorkflowOutputArray.validate_raw(obj: obj)
-      when "ERROR"
-        WorkflowOutputError.validate_raw(obj: obj)
-      when "FUNCTION_CALL"
-        WorkflowOutputFunctionCall.validate_raw(obj: obj)
-      when "IMAGE"
-        WorkflowOutputImage.validate_raw(obj: obj)
-      else
-        raise("Passed value matched no type within the union, validation failed.")
+      begin
+        return Vellum::WorkflowOutputString.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 [WorkflowOutputString]
-    # @return [WorkflowOutput]
-    def self.string(member:)
-      new(member: member, discriminant: "STRING")
-    end
-
-    # @param member [WorkflowOutputNumber]
-    # @return [WorkflowOutput]
-    def self.number(member:)
-      new(member: member, discriminant: "NUMBER")
-    end
-
-    # @param member [WorkflowOutputJson]
-    # @return [WorkflowOutput]
-    def self.json(member:)
-      new(member: member, discriminant: "JSON")
-    end
-
-    # @param member [WorkflowOutputChatHistory]
-    # @return [WorkflowOutput]
-    def self.chat_history(member:)
-      new(member: member, discriminant: "CHAT_HISTORY")
-    end
-
-    # @param member [WorkflowOutputSearchResults]
-    # @return [WorkflowOutput]
-    def self.search_results(member:)
-      new(member: member, discriminant: "SEARCH_RESULTS")
-    end
-
-    # @param member [WorkflowOutputArray]
-    # @return [WorkflowOutput]
-    def self.array(member:)
-      new(member: member, discriminant: "ARRAY")
-    end
-
-    # @param member [WorkflowOutputError]
-    # @return [WorkflowOutput]
-    def self.error(member:)
-      new(member: member, discriminant: "ERROR")
-    end
-
-    # @param member [WorkflowOutputFunctionCall]
-    # @return [WorkflowOutput]
-    def self.function_call(member:)
-      new(member: member, discriminant: "FUNCTION_CALL")
-    end
-
-    # @param member [WorkflowOutputImage]
-    # @return [WorkflowOutput]
-    def self.image(member:)
-      new(member: member, discriminant: "IMAGE")
+      begin
+        return Vellum::WorkflowOutputNumber.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::WorkflowOutputJson.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::WorkflowOutputChatHistory.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::WorkflowOutputSearchResults.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::WorkflowOutputArray.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::WorkflowOutputError.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::WorkflowOutputFunctionCall.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::WorkflowOutputImage.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/workflow_output_array.rb

@@ -1,60 +1,82 @@
 # frozen_string_literal: true
-
-require_relative "array_variable_value_item"
+require_relative "vellum_value"
+require "ostruct"
 require "json"
 
 module Vellum
-  # An array output from a Workflow execution.
+# An array output from a Workflow execution.
   class WorkflowOutputArray
-    attr_reader :id, :name, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @return [String] The output's name, as defined in the workflow
+    attr_reader :name
+  # @return [String] 
+    attr_reader :type
+  # @return [Array<Vellum::VellumValue>] 
+    attr_reader :value
+  # @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 id [String]
+    # @param id [String] 
     # @param name [String] The output's name, as defined in the workflow
-    # @param value [Array<ArrayVariableValueItem>]
+    # @param type [String] 
+    # @param value [Array<Vellum::VellumValue>] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [WorkflowOutputArray]
-    def initialize(id:, name:, value: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::WorkflowOutputArray]
+    def initialize(id:, name:, type:, value: OMIT, additional_properties: nil)
       @id = id
-      # @type [String] The output's name, as defined in the workflow
       @name = name
-      # @type [Array<ArrayVariableValueItem>]
-      @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @type = type
+      @value = value if value != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "id": id, "name": name, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of WorkflowOutputArray
+# Deserialize a JSON object to an instance of WorkflowOutputArray
     #
-    # @param json_object [JSON]
-    # @return [WorkflowOutputArray]
+    # @param json_object [String] 
+    # @return [Vellum::WorkflowOutputArray]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      id = struct.id
-      name = struct.name
-      value = parsed_json["value"].map do |v|
-        v = v.to_json
-        ArrayVariableValueItem.from_json(json_object: v)
-      end
-      new(id: id, name: name, value: value, additional_properties: struct)
+      id = parsed_json["id"]
+      name = parsed_json["name"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]&.map do | item |
+  item = item.to_json
+  Vellum::VellumValue.from_json(json_object: item)
+end
+      new(
+        id: id,
+        name: name,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of WorkflowOutputArray to a JSON object
+# Serialize an instance of WorkflowOutputArray to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "id": @id, "name": @name, "value": @value }.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.id.is_a?(String) != false || raise("Passed value for field obj.id 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.")
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
       obj.value&.is_a?(Array) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/workflow_output_chat_history.rb

@@ -1,60 +1,82 @@
 # frozen_string_literal: true
-
 require_relative "chat_message"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A chat history output from a Workflow execution.
+# A chat history output from a Workflow execution.
   class WorkflowOutputChatHistory
-    attr_reader :id, :name, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @return [String] The output's name, as defined in the workflow
+    attr_reader :name
+  # @return [String] 
+    attr_reader :type
+  # @return [Array<Vellum::ChatMessage>] 
+    attr_reader :value
+  # @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 id [String]
+    # @param id [String] 
     # @param name [String] The output's name, as defined in the workflow
-    # @param value [Array<ChatMessage>]
+    # @param type [String] 
+    # @param value [Array<Vellum::ChatMessage>] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [WorkflowOutputChatHistory]
-    def initialize(id:, name:, value: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::WorkflowOutputChatHistory]
+    def initialize(id:, name:, type:, value: OMIT, additional_properties: nil)
       @id = id
-      # @type [String] The output's name, as defined in the workflow
       @name = name
-      # @type [Array<ChatMessage>]
-      @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @type = type
+      @value = value if value != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "id": id, "name": name, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of WorkflowOutputChatHistory
+# Deserialize a JSON object to an instance of WorkflowOutputChatHistory
     #
-    # @param json_object [JSON]
-    # @return [WorkflowOutputChatHistory]
+    # @param json_object [String] 
+    # @return [Vellum::WorkflowOutputChatHistory]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      id = struct.id
-      name = struct.name
-      value = parsed_json["value"].map do |v|
-        v = v.to_json
-        ChatMessage.from_json(json_object: v)
-      end
-      new(id: id, name: name, value: value, additional_properties: struct)
+      id = parsed_json["id"]
+      name = parsed_json["name"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]&.map do | item |
+  item = item.to_json
+  Vellum::ChatMessage.from_json(json_object: item)
+end
+      new(
+        id: id,
+        name: name,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of WorkflowOutputChatHistory to a JSON object
+# Serialize an instance of WorkflowOutputChatHistory to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "id": @id, "name": @name, "value": @value }.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.id.is_a?(String) != false || raise("Passed value for field obj.id 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.")
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
       obj.value&.is_a?(Array) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/workflow_output_error.rb

@@ -1,62 +1,84 @@
 # frozen_string_literal: true
-
 require_relative "vellum_error"
+require "ostruct"
 require "json"
 
 module Vellum
-  # An error output from a Workflow execution.
+# An error output from a Workflow execution.
   class WorkflowOutputError
-    attr_reader :id, :name, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @return [String] The output's name, as defined in the workflow
+    attr_reader :name
+  # @return [String] 
+    attr_reader :type
+  # @return [Vellum::VellumError] 
+    attr_reader :value
+  # @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 id [String]
+    # @param id [String] 
     # @param name [String] The output's name, as defined in the workflow
-    # @param value [VellumError]
+    # @param type [String] 
+    # @param value [Vellum::VellumError] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [WorkflowOutputError]
-    def initialize(id:, name:, value: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::WorkflowOutputError]
+    def initialize(id:, name:, type:, value: OMIT, additional_properties: nil)
       @id = id
-      # @type [String] The output's name, as defined in the workflow
       @name = name
-      # @type [VellumError]
-      @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @type = type
+      @value = value if value != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "id": id, "name": name, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of WorkflowOutputError
+# Deserialize a JSON object to an instance of WorkflowOutputError
     #
-    # @param json_object [JSON]
-    # @return [WorkflowOutputError]
+    # @param json_object [String] 
+    # @return [Vellum::WorkflowOutputError]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      id = struct.id
-      name = struct.name
-      if parsed_json["value"].nil?
-        value = nil
-      else
+      id = parsed_json["id"]
+      name = parsed_json["name"]
+      type = parsed_json["type"]
+      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
-      new(id: id, name: name, value: value, additional_properties: struct)
+      new(
+        id: id,
+        name: name,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of WorkflowOutputError to a JSON object
+# Serialize an instance of WorkflowOutputError to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "id": @id, "name": @name, "value": @value }.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.id.is_a?(String) != false || raise("Passed value for field obj.id 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.")
-      obj.value.nil? || 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.value.nil? || Vellum::VellumError.validate_raw(obj: obj.value)
     end
   end
-end
+end