vellum_ai 0.3.15 → 0.14.74

data/lib/vellum_ai/types/node_output_compiled_string_value.rb

@@ -1,50 +1,79 @@
 # frozen_string_literal: true
-
+require_relative "workflow_node_result_event_state"
+require "ostruct"
 require "json"
 
 module Vellum
+# An output returned by a node that is of type STRING.
   class NodeOutputCompiledStringValue
-    attr_reader :node_output_id, :value, :additional_properties
+  # @return [String] 
+    attr_reader :type
+  # @return [String] 
+    attr_reader :value
+  # @return [String] 
+    attr_reader :node_output_id
+  # @return [Vellum::WorkflowNodeResultEventState] 
+    attr_reader :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 node_output_id [String]
-    # @param value [String]
+    # @param type [String] 
+    # @param value [String] 
+    # @param node_output_id [String] 
+    # @param state [Vellum::WorkflowNodeResultEventState] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [NodeOutputCompiledStringValue]
-    def initialize(node_output_id:, value: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::NodeOutputCompiledStringValue]
+    def initialize(type:, value: OMIT, node_output_id:, state: OMIT, additional_properties: nil)
+      @type = type
+      @value = value if value != OMIT
       @node_output_id = node_output_id
-      # @type [String]
-      @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @state = state if state != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "type": type, "value": value, "node_output_id": node_output_id, "state": state }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of NodeOutputCompiledStringValue
+# Deserialize a JSON object to an instance of NodeOutputCompiledStringValue
     #
-    # @param json_object [JSON]
-    # @return [NodeOutputCompiledStringValue]
+    # @param json_object [String] 
+    # @return [Vellum::NodeOutputCompiledStringValue]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      JSON.parse(json_object)
-      node_output_id = struct.node_output_id
-      value = struct.value
-      new(node_output_id: node_output_id, value: value, additional_properties: struct)
+      parsed_json = JSON.parse(json_object)
+      type = parsed_json["type"]
+      value = parsed_json["value"]
+      node_output_id = parsed_json["node_output_id"]
+      state = parsed_json["state"]
+      new(
+        type: type,
+        value: value,
+        node_output_id: node_output_id,
+        state: state,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of NodeOutputCompiledStringValue to a JSON object
+# Serialize an instance of NodeOutputCompiledStringValue to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "node_output_id": @node_output_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.node_output_id.is_a?(String) != false || raise("Passed value for field obj.node_output_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?(String) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+      obj.node_output_id.is_a?(String) != false || raise("Passed value for field obj.node_output_id is not the expected type, validation failed.")
+      obj.state&.is_a?(Vellum::WorkflowNodeResultEventState) != false || raise("Passed value for field obj.state is not the expected type, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/node_output_compiled_value.rb

@@ -1,5 +1,4 @@
 # frozen_string_literal: true
-
 require "json"
 require_relative "node_output_compiled_string_value"
 require_relative "node_output_compiled_number_value"
@@ -8,161 +7,148 @@ require_relative "node_output_compiled_chat_history_value"
 require_relative "node_output_compiled_search_results_value"
 require_relative "node_output_compiled_error_value"
 require_relative "node_output_compiled_array_value"
-require_relative "node_output_compiled_function_value"
+require_relative "node_output_compiled_function_call_value"
 
 module Vellum
   class NodeOutputCompiledValue
-    attr_reader :member, :discriminant
 
-    private_class_method :new
-    alias kind_of? is_a?
-    # @param member [Object]
-    # @param discriminant [String]
-    # @return [NodeOutputCompiledValue]
-    def initialize(member:, discriminant:)
-      # @type [Object]
-      @member = member
-      # @type [String]
-      @discriminant = discriminant
-    end
 
-    # Deserialize a JSON object to an instance of NodeOutputCompiledValue
+# Deserialize a JSON object to an instance of NodeOutputCompiledValue
     #
-    # @param json_object [JSON]
-    # @return [NodeOutputCompiledValue]
+    # @param json_object [String] 
+    # @return [Vellum::NodeOutputCompiledValue]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      member = case struct.type
-               when "STRING"
-                 NodeOutputCompiledStringValue.from_json(json_object: json_object)
-               when "NUMBER"
-                 NodeOutputCompiledNumberValue.from_json(json_object: json_object)
-               when "JSON"
-                 NodeOutputCompiledJsonValue.from_json(json_object: json_object)
-               when "CHAT_HISTORY"
-                 NodeOutputCompiledChatHistoryValue.from_json(json_object: json_object)
-               when "SEARCH_RESULTS"
-                 NodeOutputCompiledSearchResultsValue.from_json(json_object: json_object)
-               when "ERROR"
-                 NodeOutputCompiledErrorValue.from_json(json_object: json_object)
-               when "ARRAY"
-                 NodeOutputCompiledArrayValue.from_json(json_object: json_object)
-               when "FUNCTION_CALL"
-                 NodeOutputCompiledFunctionValue.from_json(json_object: json_object)
-               else
-                 NodeOutputCompiledStringValue.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 "ERROR"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "ARRAY"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "FUNCTION_CALL"
-        { **@member.to_json, type: @discriminant }.to_json
-      else
-        { "type": @discriminant, value: @member }.to_json
-      end
-      @member.to_json
+      begin
+        Vellum::NodeOutputCompiledStringValue.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::NodeOutputCompiledStringValue.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::NodeOutputCompiledNumberValue.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::NodeOutputCompiledNumberValue.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::NodeOutputCompiledJsonValue.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::NodeOutputCompiledJsonValue.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::NodeOutputCompiledChatHistoryValue.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::NodeOutputCompiledChatHistoryValue.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::NodeOutputCompiledSearchResultsValue.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::NodeOutputCompiledSearchResultsValue.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::NodeOutputCompiledErrorValue.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::NodeOutputCompiledErrorValue.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::NodeOutputCompiledArrayValue.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::NodeOutputCompiledArrayValue.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::NodeOutputCompiledFunctionCallValue.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::NodeOutputCompiledFunctionCallValue.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"
-        NodeOutputCompiledStringValue.validate_raw(obj: obj)
-      when "NUMBER"
-        NodeOutputCompiledNumberValue.validate_raw(obj: obj)
-      when "JSON"
-        NodeOutputCompiledJsonValue.validate_raw(obj: obj)
-      when "CHAT_HISTORY"
-        NodeOutputCompiledChatHistoryValue.validate_raw(obj: obj)
-      when "SEARCH_RESULTS"
-        NodeOutputCompiledSearchResultsValue.validate_raw(obj: obj)
-      when "ERROR"
-        NodeOutputCompiledErrorValue.validate_raw(obj: obj)
-      when "ARRAY"
-        NodeOutputCompiledArrayValue.validate_raw(obj: obj)
-      when "FUNCTION_CALL"
-        NodeOutputCompiledFunctionValue.validate_raw(obj: obj)
-      else
-        raise("Passed value matched no type within the union, validation failed.")
+      begin
+        return Vellum::NodeOutputCompiledStringValue.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 [NodeOutputCompiledStringValue]
-    # @return [NodeOutputCompiledValue]
-    def self.string(member:)
-      new(member: member, discriminant: "STRING")
-    end
-
-    # @param member [NodeOutputCompiledNumberValue]
-    # @return [NodeOutputCompiledValue]
-    def self.number(member:)
-      new(member: member, discriminant: "NUMBER")
-    end
-
-    # @param member [NodeOutputCompiledJsonValue]
-    # @return [NodeOutputCompiledValue]
-    def self.json(member:)
-      new(member: member, discriminant: "JSON")
-    end
-
-    # @param member [NodeOutputCompiledChatHistoryValue]
-    # @return [NodeOutputCompiledValue]
-    def self.chat_history(member:)
-      new(member: member, discriminant: "CHAT_HISTORY")
-    end
-
-    # @param member [NodeOutputCompiledSearchResultsValue]
-    # @return [NodeOutputCompiledValue]
-    def self.search_results(member:)
-      new(member: member, discriminant: "SEARCH_RESULTS")
-    end
-
-    # @param member [NodeOutputCompiledErrorValue]
-    # @return [NodeOutputCompiledValue]
-    def self.error(member:)
-      new(member: member, discriminant: "ERROR")
-    end
-
-    # @param member [NodeOutputCompiledArrayValue]
-    # @return [NodeOutputCompiledValue]
-    def self.array(member:)
-      new(member: member, discriminant: "ARRAY")
-    end
-
-    # @param member [NodeOutputCompiledFunctionValue]
-    # @return [NodeOutputCompiledValue]
-    def self.function_call(member:)
-      new(member: member, discriminant: "FUNCTION_CALL")
+      begin
+        return Vellum::NodeOutputCompiledNumberValue.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::NodeOutputCompiledJsonValue.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::NodeOutputCompiledChatHistoryValue.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::NodeOutputCompiledSearchResultsValue.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::NodeOutputCompiledErrorValue.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::NodeOutputCompiledArrayValue.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::NodeOutputCompiledFunctionCallValue.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/node_parent_context.rb

@@ -0,0 +1,100 @@
+# frozen_string_literal: true
+require_relative "parent_context"
+require_relative "span_link"
+require_relative "vellum_code_resource_definition"
+require "ostruct"
+require "json"
+
+module Vellum
+  class NodeParentContext
+  # @return [Vellum::ParentContext] 
+    attr_reader :parent
+  # @return [Array<Vellum::SpanLink>] 
+    attr_reader :links
+  # @return [Vellum::VellumCodeResourceDefinition] 
+    attr_reader :node_definition
+  # @return [String] 
+    attr_reader :type
+  # @return [String] 
+    attr_reader :span_id
+  # @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 parent [Vellum::ParentContext] 
+    # @param links [Array<Vellum::SpanLink>] 
+    # @param node_definition [Vellum::VellumCodeResourceDefinition] 
+    # @param type [String] 
+    # @param span_id [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::NodeParentContext]
+    def initialize(parent: OMIT, links: OMIT, node_definition:, type:, span_id:, additional_properties: nil)
+      @parent = parent if parent != OMIT
+      @links = links if links != OMIT
+      @node_definition = node_definition
+      @type = type
+      @span_id = span_id
+      @additional_properties = additional_properties
+      @_field_set = { "parent": parent, "links": links, "node_definition": node_definition, "type": type, "span_id": span_id }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of NodeParentContext
+    #
+    # @param json_object [String] 
+    # @return [Vellum::NodeParentContext]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      unless parsed_json["parent"].nil?
+        parent = parsed_json["parent"].to_json
+        parent = Vellum::ParentContext.from_json(json_object: parent)
+      else
+        parent = nil
+      end
+      links = parsed_json["links"]&.map do | item |
+  item = item.to_json
+  Vellum::SpanLink.from_json(json_object: item)
+end
+      unless parsed_json["node_definition"].nil?
+        node_definition = parsed_json["node_definition"].to_json
+        node_definition = Vellum::VellumCodeResourceDefinition.from_json(json_object: node_definition)
+      else
+        node_definition = nil
+      end
+      type = parsed_json["type"]
+      span_id = parsed_json["span_id"]
+      new(
+        parent: parent,
+        links: links,
+        node_definition: node_definition,
+        type: type,
+        span_id: span_id,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of NodeParentContext 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.parent.nil? || Vellum::ParentContext.validate_raw(obj: obj.parent)
+      obj.links&.is_a?(Array) != false || raise("Passed value for field obj.links is not the expected type, validation failed.")
+      Vellum::VellumCodeResourceDefinition.validate_raw(obj: obj.node_definition)
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      obj.span_id.is_a?(String) != false || raise("Passed value for field obj.span_id is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/normalized_log_probs.rb

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