vellum_ai 0.3.15 → 0.14.74

data/lib/vellum_ai/types/terminal_node_json_result.rb

@@ -1,55 +1,77 @@
 # frozen_string_literal: true
-
+require "ostruct"
 require "json"
 
 module Vellum
   class TerminalNodeJsonResult
-    attr_reader :id, :name, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @return [String] The unique name given to the terminal node that produced this output.
+    attr_reader :name
+  # @return [String] 
+    attr_reader :type
+  # @return [Object] 
+    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 unique name given to the terminal node that produced this output.
-    # @param value [Hash{String => String}]
+    # @param type [String] 
+    # @param value [Object] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [TerminalNodeJsonResult]
-    def initialize(name:, id: nil, value: nil, additional_properties: nil)
-      # @type [String]
-      @id = id
-      # @type [String] The unique name given to the terminal node that produced this output.
+    # @return [Vellum::TerminalNodeJsonResult]
+    def initialize(id: OMIT, name:, type:, value:, additional_properties: nil)
+      @id = id if id != OMIT
       @name = name
-      # @type [Hash{String => String}]
+      @type = type
       @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @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 TerminalNodeJsonResult
+# Deserialize a JSON object to an instance of TerminalNodeJsonResult
     #
-    # @param json_object [JSON]
-    # @return [TerminalNodeJsonResult]
+    # @param json_object [String] 
+    # @return [Vellum::TerminalNodeJsonResult]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      JSON.parse(json_object)
-      id = struct.id
-      name = struct.name
-      value = struct.value
-      new(id: id, name: name, value: value, additional_properties: struct)
+      parsed_json = JSON.parse(json_object)
+      id = parsed_json["id"]
+      name = parsed_json["name"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]
+      new(
+        id: id,
+        name: name,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of TerminalNodeJsonResult to a JSON object
+# Serialize an instance of TerminalNodeJsonResult 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&.is_a?(Hash) != 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.value.is_a?(Object) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/terminal_node_number_result.rb

@@ -1,55 +1,77 @@
 # frozen_string_literal: true
-
+require "ostruct"
 require "json"
 
 module Vellum
   class TerminalNodeNumberResult
-    attr_reader :id, :name, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @return [String] The unique name given to the terminal node that produced this output.
+    attr_reader :name
+  # @return [String] 
+    attr_reader :type
+  # @return [Float] 
+    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 unique name given to the terminal node that produced this output.
-    # @param value [Float]
+    # @param type [String] 
+    # @param value [Float] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [TerminalNodeNumberResult]
-    def initialize(name:, id: nil, value: nil, additional_properties: nil)
-      # @type [String]
-      @id = id
-      # @type [String] The unique name given to the terminal node that produced this output.
+    # @return [Vellum::TerminalNodeNumberResult]
+    def initialize(id: OMIT, name:, type:, value: OMIT, additional_properties: nil)
+      @id = id if id != OMIT
       @name = name
-      # @type [Float]
-      @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 TerminalNodeNumberResult
+# Deserialize a JSON object to an instance of TerminalNodeNumberResult
     #
-    # @param json_object [JSON]
-    # @return [TerminalNodeNumberResult]
+    # @param json_object [String] 
+    # @return [Vellum::TerminalNodeNumberResult]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      JSON.parse(json_object)
-      id = struct.id
-      name = struct.name
-      value = struct.value
-      new(id: id, name: name, value: value, additional_properties: struct)
+      parsed_json = JSON.parse(json_object)
+      id = parsed_json["id"]
+      name = parsed_json["name"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]
+      new(
+        id: id,
+        name: name,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of TerminalNodeNumberResult to a JSON object
+# Serialize an instance of TerminalNodeNumberResult 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?(Float) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/terminal_node_result.rb

@@ -1,52 +1,68 @@
 # frozen_string_literal: true
-
 require_relative "terminal_node_result_data"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A Node Result Event emitted from a Terminal Node.
+# A Node Result Event emitted from a Terminal Node.
   class TerminalNodeResult
-    attr_reader :data, :additional_properties
+  # @return [String] 
+    attr_reader :type
+  # @return [Vellum::TerminalNodeResultData] 
+    attr_reader :data
+  # @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 data [TerminalNodeResultData]
+    # @param type [String] 
+    # @param data [Vellum::TerminalNodeResultData] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [TerminalNodeResult]
-    def initialize(data:, additional_properties: nil)
-      # @type [TerminalNodeResultData]
+    # @return [Vellum::TerminalNodeResult]
+    def initialize(type:, data:, additional_properties: nil)
+      @type = type
       @data = data
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "type": type, "data": data }
     end
-
-    # Deserialize a JSON object to an instance of TerminalNodeResult
+# Deserialize a JSON object to an instance of TerminalNodeResult
     #
-    # @param json_object [JSON]
-    # @return [TerminalNodeResult]
+    # @param json_object [String] 
+    # @return [Vellum::TerminalNodeResult]
     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
+      type = parsed_json["type"]
+      unless parsed_json["data"].nil?
         data = parsed_json["data"].to_json
-        data = TerminalNodeResultData.from_json(json_object: data)
+        data = Vellum::TerminalNodeResultData.from_json(json_object: data)
+      else
+        data = nil
       end
-      new(data: data, additional_properties: struct)
+      new(
+        type: type,
+        data: data,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of TerminalNodeResult to a JSON object
+# Serialize an instance of TerminalNodeResult to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "data": @data }.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:)
-      TerminalNodeResultData.validate_raw(obj: obj.data)
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      Vellum::TerminalNodeResultData.validate_raw(obj: obj.data)
     end
   end
-end
+end

data/lib/vellum_ai/types/terminal_node_result_data.rb

@@ -1,51 +1,57 @@
 # frozen_string_literal: true
-
 require_relative "terminal_node_result_output"
+require "ostruct"
 require "json"
 
 module Vellum
   class TerminalNodeResultData
-    attr_reader :output, :additional_properties
+  # @return [Vellum::TerminalNodeResultOutput] 
+    attr_reader :output
+  # @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 output [TerminalNodeResultOutput]
+    # @param output [Vellum::TerminalNodeResultOutput] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [TerminalNodeResultData]
+    # @return [Vellum::TerminalNodeResultData]
     def initialize(output:, additional_properties: nil)
-      # @type [TerminalNodeResultOutput]
       @output = output
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "output": output }
     end
-
-    # Deserialize a JSON object to an instance of TerminalNodeResultData
+# Deserialize a JSON object to an instance of TerminalNodeResultData
     #
-    # @param json_object [JSON]
-    # @return [TerminalNodeResultData]
+    # @param json_object [String] 
+    # @return [Vellum::TerminalNodeResultData]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      if parsed_json["output"].nil?
-        output = nil
-      else
+      unless parsed_json["output"].nil?
         output = parsed_json["output"].to_json
-        output = TerminalNodeResultOutput.from_json(json_object: output)
+        output = Vellum::TerminalNodeResultOutput.from_json(json_object: output)
+      else
+        output = nil
       end
       new(output: output, additional_properties: struct)
     end
-
-    # Serialize an instance of TerminalNodeResultData to a JSON object
+# Serialize an instance of TerminalNodeResultData to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "output": @output }.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:)
-      TerminalNodeResultOutput.validate_raw(obj: obj.output)
+      Vellum::TerminalNodeResultOutput.validate_raw(obj: obj.output)
     end
   end
-end
+end