vellum_ai 0.3.15 → 0.14.74

data/lib/vellum_ai/types/json_input_request.rb

@@ -1,51 +1,69 @@
 # frozen_string_literal: true
-
+require "ostruct"
 require "json"
 
 module Vellum
-  # A user input representing a JSON object
+# A user input representing a JSON object
   class JsonInputRequest
-    attr_reader :name, :value, :additional_properties
+  # @return [String] The variable's name
+    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 name [String] The variable's name, as defined in the deployment.
-    # @param value [Hash{String => String}]
+    # @param name [String] The variable's name
+    # @param type [String] 
+    # @param value [Object] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [JsonInputRequest]
-    def initialize(name:, value:, additional_properties: nil)
-      # @type [String] The variable's name, as defined in the deployment.
+    # @return [Vellum::JsonInputRequest]
+    def initialize(name:, type:, value:, additional_properties: nil)
       @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 = { "name": name, "type": type, "value": value }
     end
-
-    # Deserialize a JSON object to an instance of JsonInputRequest
+# Deserialize a JSON object to an instance of JsonInputRequest
     #
-    # @param json_object [JSON]
-    # @return [JsonInputRequest]
+    # @param json_object [String] 
+    # @return [Vellum::JsonInputRequest]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      JSON.parse(json_object)
-      name = struct.name
-      value = struct.value
-      new(name: name, value: value, additional_properties: struct)
+      parsed_json = JSON.parse(json_object)
+      name = parsed_json["name"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]
+      new(
+        name: name,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of JsonInputRequest to a JSON object
+# Serialize an instance of JsonInputRequest to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "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.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/json_vellum_value.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+# A value representing a JSON object.
+  class JsonVellumValue
+  # @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 type [String] 
+    # @param value [Object] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::JsonVellumValue]
+    def initialize(type:, value:, additional_properties: nil)
+      @type = type
+      @value = value
+      @additional_properties = additional_properties
+      @_field_set = { "type": type, "value": value }
+    end
+# Deserialize a JSON object to an instance of JsonVellumValue
+    #
+    # @param json_object [String] 
+    # @return [Vellum::JsonVellumValue]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      type = parsed_json["type"]
+      value = parsed_json["value"]
+      new(
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of JsonVellumValue 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.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

data/lib/vellum_ai/types/json_vellum_value_request.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+# A value representing a JSON object.
+  class JsonVellumValueRequest
+  # @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 type [String] 
+    # @param value [Object] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::JsonVellumValueRequest]
+    def initialize(type:, value:, additional_properties: nil)
+      @type = type
+      @value = value
+      @additional_properties = additional_properties
+      @_field_set = { "type": type, "value": value }
+    end
+# Deserialize a JSON object to an instance of JsonVellumValueRequest
+    #
+    # @param json_object [String] 
+    # @return [Vellum::JsonVellumValueRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      type = parsed_json["type"]
+      value = parsed_json["value"]
+      new(
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of JsonVellumValueRequest 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.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

data/lib/vellum_ai/types/logical_operator.rb

@@ -1,25 +1,60 @@
 # frozen_string_literal: true
 
 module Vellum
-  # @type [LOGICAL_OPERATOR]
-  LOGICAL_OPERATOR = {
-    equals: "=",
-    does_not_equal: "!=",
-    less_than: "<",
-    greater_than: ">",
-    less_than_or_equal_to: "<=",
-    greater_than_or_equal_to: ">=",
-    contains: "contains",
-    begins_with: "beginsWith",
-    ends_with: "endsWith",
-    does_not_contain: "doesNotContain",
-    does_not_begin_with: "doesNotBeginWith",
-    does_not_end_with: "doesNotEndWith",
-    null: "null",
-    not_null: "notNull",
-    in_: "in",
-    not_in: "notIn",
-    between: "between",
-    not_between: "notBetween"
-  }.freeze
-end
+# * `=` - EQUALS
+#  * `!=` - DOES_NOT_EQUAL
+#  * `<` - LESS_THAN
+#  * `>` - GREATER_THAN
+#  * `<=` - LESS_THAN_OR_EQUAL_TO
+#  * `>=` - GREATER_THAN_OR_EQUAL_TO
+#  * `contains` - CONTAINS
+#  * `beginsWith` - BEGINS_WITH
+#  * `endsWith` - ENDS_WITH
+#  * `doesNotContain` - DOES_NOT_CONTAIN
+#  * `doesNotBeginWith` - DOES_NOT_BEGIN_WITH
+#  * `doesNotEndWith` - DOES_NOT_END_WITH
+#  * `null` - NULL
+#  * `notNull` - NOT_NULL
+#  * `in` - IN
+#  * `notIn` - NOT_IN
+#  * `between` - BETWEEN
+#  * `notBetween` - NOT_BETWEEN
+#  * `blank` - BLANK
+#  * `notBlank` - NOT_BLANK
+#  * `coalesce` - COALESCE
+#  * `accessField` - ACCESS_FIELD
+#  * `parseJson` - PARSE_JSON
+#  * `and` - AND
+#  * `or` - OR
+#  * `isError` - IS_ERROR
+  class LogicalOperator
+
+    EQUALS = "="
+    DOES_NOT_EQUAL = "!="
+    LESS_THAN = "<"
+    GREATER_THAN = ">"
+    LESS_THAN_OR_EQUAL_TO = "<="
+    GREATER_THAN_OR_EQUAL_TO = ">="
+    CONTAINS = "contains"
+    BEGINS_WITH = "beginsWith"
+    ENDS_WITH = "endsWith"
+    DOES_NOT_CONTAIN = "doesNotContain"
+    DOES_NOT_BEGIN_WITH = "doesNotBeginWith"
+    DOES_NOT_END_WITH = "doesNotEndWith"
+    NULL = "null"
+    NOT_NULL = "notNull"
+    IN = "in"
+    NOT_IN = "notIn"
+    BETWEEN = "between"
+    NOT_BETWEEN = "notBetween"
+    BLANK = "blank"
+    NOT_BLANK = "notBlank"
+    COALESCE = "coalesce"
+    ACCESS_FIELD = "accessField"
+    PARSE_JSON = "parseJson"
+    AND = "and"
+    OR = "or"
+    IS_ERROR = "isError"
+
+  end
+end

data/lib/vellum_ai/types/logprobs_enum.rb

@@ -1,6 +1,12 @@
 # frozen_string_literal: true
 
 module Vellum
-  # @type [LOGPROBS_ENUM]
-  LOGPROBS_ENUM = { all: "ALL", none: "NONE" }.freeze
-end
+# * `ALL` - ALL
+#  * `NONE` - NONE
+  class LogprobsEnum
+
+    ALL = "ALL"
+    NONE = "NONE"
+
+  end
+end

data/lib/vellum_ai/types/map_node_result.rb

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

data/lib/vellum_ai/types/map_node_result_data.rb

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

data/lib/vellum_ai/types/merge_node_result.rb

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

data/lib/vellum_ai/types/merge_node_result_data.rb

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