vellum_ai 0.3.14 → 0.14.74

data/lib/vellum_ai/types/chat_message_role.rb

@@ -1,6 +1,16 @@
 # frozen_string_literal: true
 
 module Vellum
-  # @type [CHAT_MESSAGE_ROLE]
-  CHAT_MESSAGE_ROLE = { system: "SYSTEM", assistant: "ASSISTANT", user: "USER", function: "FUNCTION" }.freeze
-end
+# * `SYSTEM` - System
+#  * `ASSISTANT` - Assistant
+#  * `USER` - User
+#  * `FUNCTION` - Function
+  class ChatMessageRole
+
+    SYSTEM = "SYSTEM"
+    ASSISTANT = "ASSISTANT"
+    USER = "USER"
+    FUNCTION = "FUNCTION"
+
+  end
+end

data/lib/vellum_ai/types/code_execution_node_array_result.rb

@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+require_relative "vellum_value"
+require "ostruct"
+require "json"
+
+module Vellum
+  class CodeExecutionNodeArrayResult
+  # @return [String] 
+    attr_reader :id
+  # @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 type [String] 
+    # @param value [Array<Vellum::VellumValue>] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::CodeExecutionNodeArrayResult]
+    def initialize(id:, type:, value: OMIT, additional_properties: nil)
+      @id = id
+      @type = type
+      @value = value if value != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "id": id, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of CodeExecutionNodeArrayResult
+    #
+    # @param json_object [String] 
+    # @return [Vellum::CodeExecutionNodeArrayResult]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      id = parsed_json["id"]
+      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,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of CodeExecutionNodeArrayResult 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.id.is_a?(String) != false || raise("Passed value for field obj.id 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

data/lib/vellum_ai/types/code_execution_node_chat_history_result.rb

@@ -1,54 +1,74 @@
 # frozen_string_literal: true
-
 require_relative "chat_message"
+require "ostruct"
 require "json"
 
 module Vellum
   class CodeExecutionNodeChatHistoryResult
-    attr_reader :id, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @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 value [Array<ChatMessage>]
+    # @param id [String] 
+    # @param type [String] 
+    # @param value [Array<Vellum::ChatMessage>] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [CodeExecutionNodeChatHistoryResult]
-    def initialize(id:, value: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::CodeExecutionNodeChatHistoryResult]
+    def initialize(id:, type:, value: OMIT, additional_properties: nil)
       @id = id
-      # @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, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of CodeExecutionNodeChatHistoryResult
+# Deserialize a JSON object to an instance of CodeExecutionNodeChatHistoryResult
     #
-    # @param json_object [JSON]
-    # @return [CodeExecutionNodeChatHistoryResult]
+    # @param json_object [String] 
+    # @return [Vellum::CodeExecutionNodeChatHistoryResult]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      id = struct.id
-      value = parsed_json["value"].map do |v|
-        v = v.to_json
-        ChatMessage.from_json(json_object: v)
-      end
-      new(id: id, value: value, additional_properties: struct)
+      id = parsed_json["id"]
+      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,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of CodeExecutionNodeChatHistoryResult to a JSON object
+# Serialize an instance of CodeExecutionNodeChatHistoryResult to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "id": @id, "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.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/code_execution_node_error_result.rb

@@ -1,56 +1,76 @@
 # frozen_string_literal: true
-
 require_relative "vellum_error"
+require "ostruct"
 require "json"
 
 module Vellum
   class CodeExecutionNodeErrorResult
-    attr_reader :id, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @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 value [VellumError]
+    # @param id [String] 
+    # @param type [String] 
+    # @param value [Vellum::VellumError] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [CodeExecutionNodeErrorResult]
-    def initialize(id:, value: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::CodeExecutionNodeErrorResult]
+    def initialize(id:, type:, value: OMIT, additional_properties: nil)
       @id = id
-      # @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, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of CodeExecutionNodeErrorResult
+# Deserialize a JSON object to an instance of CodeExecutionNodeErrorResult
     #
-    # @param json_object [JSON]
-    # @return [CodeExecutionNodeErrorResult]
+    # @param json_object [String] 
+    # @return [Vellum::CodeExecutionNodeErrorResult]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      id = struct.id
-      if parsed_json["value"].nil?
-        value = nil
-      else
+      id = parsed_json["id"]
+      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, value: value, additional_properties: struct)
+      new(
+        id: id,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of CodeExecutionNodeErrorResult to a JSON object
+# Serialize an instance of CodeExecutionNodeErrorResult to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "id": @id, "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.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

data/lib/vellum_ai/types/code_execution_node_function_call_result.rb

@@ -0,0 +1,76 @@
+# frozen_string_literal: true
+require_relative "function_call"
+require "ostruct"
+require "json"
+
+module Vellum
+  class CodeExecutionNodeFunctionCallResult
+  # @return [String] 
+    attr_reader :id
+  # @return [String] 
+    attr_reader :type
+  # @return [Vellum::FunctionCall] 
+    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 type [String] 
+    # @param value [Vellum::FunctionCall] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::CodeExecutionNodeFunctionCallResult]
+    def initialize(id:, type:, value: OMIT, additional_properties: nil)
+      @id = id
+      @type = type
+      @value = value if value != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "id": id, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of CodeExecutionNodeFunctionCallResult
+    #
+    # @param json_object [String] 
+    # @return [Vellum::CodeExecutionNodeFunctionCallResult]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      id = parsed_json["id"]
+      type = parsed_json["type"]
+      unless parsed_json["value"].nil?
+        value = parsed_json["value"].to_json
+        value = Vellum::FunctionCall.from_json(json_object: value)
+      else
+        value = nil
+      end
+      new(
+        id: id,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of CodeExecutionNodeFunctionCallResult 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.id.is_a?(String) != false || raise("Passed value for field obj.id 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.nil? || Vellum::FunctionCall.validate_raw(obj: obj.value)
+    end
+  end
+end

data/lib/vellum_ai/types/code_execution_node_json_result.rb

@@ -1,50 +1,68 @@
 # frozen_string_literal: true
-
+require "ostruct"
 require "json"
 
 module Vellum
   class CodeExecutionNodeJsonResult
-    attr_reader :id, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @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 value [Hash{String => String}]
+    # @param id [String] 
+    # @param type [String] 
+    # @param value [Object] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [CodeExecutionNodeJsonResult]
-    def initialize(id:, value: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::CodeExecutionNodeJsonResult]
+    def initialize(id:, type:, value:, additional_properties: nil)
       @id = id
-      # @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, "type": type, "value": value }
     end
-
-    # Deserialize a JSON object to an instance of CodeExecutionNodeJsonResult
+# Deserialize a JSON object to an instance of CodeExecutionNodeJsonResult
     #
-    # @param json_object [JSON]
-    # @return [CodeExecutionNodeJsonResult]
+    # @param json_object [String] 
+    # @return [Vellum::CodeExecutionNodeJsonResult]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      JSON.parse(json_object)
-      id = struct.id
-      value = struct.value
-      new(id: id, value: value, additional_properties: struct)
+      parsed_json = JSON.parse(json_object)
+      id = parsed_json["id"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]
+      new(
+        id: id,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of CodeExecutionNodeJsonResult to a JSON object
+# Serialize an instance of CodeExecutionNodeJsonResult to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "id": @id, "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.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/code_execution_node_number_result.rb

@@ -1,50 +1,70 @@
 # frozen_string_literal: true
-
+require "ostruct"
 require "json"
 
 module Vellum
   class CodeExecutionNodeNumberResult
-    attr_reader :id, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @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 value [Float]
+    # @param id [String] 
+    # @param type [String] 
+    # @param value [Float] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [CodeExecutionNodeNumberResult]
-    def initialize(id:, value: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::CodeExecutionNodeNumberResult]
+    def initialize(id:, type:, value: OMIT, additional_properties: nil)
       @id = id
-      # @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, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of CodeExecutionNodeNumberResult
+# Deserialize a JSON object to an instance of CodeExecutionNodeNumberResult
     #
-    # @param json_object [JSON]
-    # @return [CodeExecutionNodeNumberResult]
+    # @param json_object [String] 
+    # @return [Vellum::CodeExecutionNodeNumberResult]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      JSON.parse(json_object)
-      id = struct.id
-      value = struct.value
-      new(id: id, value: value, additional_properties: struct)
+      parsed_json = JSON.parse(json_object)
+      id = parsed_json["id"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]
+      new(
+        id: id,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of CodeExecutionNodeNumberResult to a JSON object
+# Serialize an instance of CodeExecutionNodeNumberResult to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "id": @id, "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.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/code_execution_node_result.rb

@@ -1,52 +1,68 @@
 # frozen_string_literal: true
-
 require_relative "code_execution_node_result_data"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A Node Result Event emitted from a Code Execution Node.
+# A Node Result Event emitted from a Code Execution Node.
   class CodeExecutionNodeResult
-    attr_reader :data, :additional_properties
+  # @return [String] 
+    attr_reader :type
+  # @return [Vellum::CodeExecutionNodeResultData] 
+    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 [CodeExecutionNodeResultData]
+    # @param type [String] 
+    # @param data [Vellum::CodeExecutionNodeResultData] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [CodeExecutionNodeResult]
-    def initialize(data:, additional_properties: nil)
-      # @type [CodeExecutionNodeResultData]
+    # @return [Vellum::CodeExecutionNodeResult]
+    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 CodeExecutionNodeResult
+# Deserialize a JSON object to an instance of CodeExecutionNodeResult
     #
-    # @param json_object [JSON]
-    # @return [CodeExecutionNodeResult]
+    # @param json_object [String] 
+    # @return [Vellum::CodeExecutionNodeResult]
     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 = CodeExecutionNodeResultData.from_json(json_object: data)
+        data = Vellum::CodeExecutionNodeResultData.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 CodeExecutionNodeResult to a JSON object
+# Serialize an instance of CodeExecutionNodeResult 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:)
-      CodeExecutionNodeResultData.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::CodeExecutionNodeResultData.validate_raw(obj: obj.data)
     end
   end
-end
+end