vellum_ai 0.3.15 → 0.14.74

data/lib/vellum_ai/types/workflow_result_event_output_data_search_results.rb

@@ -1,83 +1,108 @@
 # frozen_string_literal: true
-
 require_relative "workflow_node_result_event_state"
 require_relative "search_result"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A Search Results output streamed from a Workflow execution.
+# A Search Results output streamed from a Workflow execution.
   class WorkflowResultEventOutputDataSearchResults
-    attr_reader :id, :name, :state, :node_id, :delta, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @return [String] 
+    attr_reader :name
+  # @return [Vellum::WorkflowNodeResultEventState] 
+    attr_reader :state
+  # @return [String] 
+    attr_reader :node_id
+  # @return [String] The newly output string value. Only relevant for string outputs with a state of
+#  STREAMING.
+    attr_reader :delta
+  # @return [String] 
+    attr_reader :type
+  # @return [Array<Vellum::SearchResult>] 
+    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 name [String]
-    # @param state [WORKFLOW_NODE_RESULT_EVENT_STATE]
-    # @param node_id [String]
-    # @param delta [String] The newly output string value. Only relevant for string outputs with a state of STREAMING.
-    # @param value [Array<SearchResult>]
+    # @param id [String] 
+    # @param name [String] 
+    # @param state [Vellum::WorkflowNodeResultEventState] 
+    # @param node_id [String] 
+    # @param delta [String] The newly output string value. Only relevant for string outputs with a state of
+#  STREAMING.
+    # @param type [String] 
+    # @param value [Array<Vellum::SearchResult>] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [WorkflowResultEventOutputDataSearchResults]
-    def initialize(name:, state:, node_id:, id: nil, delta: nil, value: nil, additional_properties: nil)
-      # @type [String]
-      @id = id
-      # @type [String]
+    # @return [Vellum::WorkflowResultEventOutputDataSearchResults]
+    def initialize(id: OMIT, name:, state:, node_id:, delta: OMIT, type:, value: OMIT, additional_properties: nil)
+      @id = id if id != OMIT
       @name = name
-      # @type [WORKFLOW_NODE_RESULT_EVENT_STATE]
       @state = state
-      # @type [String]
       @node_id = node_id
-      # @type [String] The newly output string value. Only relevant for string outputs with a state of STREAMING.
-      @delta = delta
-      # @type [Array<SearchResult>]
-      @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @delta = delta if delta != OMIT
+      @type = type
+      @value = value if value != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "id": id, "name": name, "state": state, "node_id": node_id, "delta": delta, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of WorkflowResultEventOutputDataSearchResults
+# Deserialize a JSON object to an instance of
+#  WorkflowResultEventOutputDataSearchResults
     #
-    # @param json_object [JSON]
-    # @return [WorkflowResultEventOutputDataSearchResults]
+    # @param json_object [String] 
+    # @return [Vellum::WorkflowResultEventOutputDataSearchResults]
     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
-      state = WORKFLOW_NODE_RESULT_EVENT_STATE.key(parsed_json["state"]) || parsed_json["state"]
-      node_id = struct.node_id
-      delta = struct.delta
-      value = parsed_json["value"].map do |v|
-        v = v.to_json
-        SearchResult.from_json(json_object: v)
-      end
-      new(id: id, name: name, state: state, node_id: node_id, delta: delta, value: value, additional_properties: struct)
+      id = parsed_json["id"]
+      name = parsed_json["name"]
+      state = parsed_json["state"]
+      node_id = parsed_json["node_id"]
+      delta = parsed_json["delta"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]&.map do | item |
+  item = item.to_json
+  Vellum::SearchResult.from_json(json_object: item)
+end
+      new(
+        id: id,
+        name: name,
+        state: state,
+        node_id: node_id,
+        delta: delta,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of WorkflowResultEventOutputDataSearchResults to a JSON object
+# Serialize an instance of WorkflowResultEventOutputDataSearchResults to a JSON
+#  object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      {
-        "id": @id,
-        "name": @name,
-        "state": WORKFLOW_NODE_RESULT_EVENT_STATE[@state] || @state,
-        "node_id": @node_id,
-        "delta": @delta,
-        "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.state.is_a?(WORKFLOW_NODE_RESULT_EVENT_STATE) != false || raise("Passed value for field obj.state 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.")
       obj.node_id.is_a?(String) != false || raise("Passed value for field obj.node_id is not the expected type, validation failed.")
       obj.delta&.is_a?(String) != false || raise("Passed value for field obj.delta 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_result_event_output_data_string.rb

@@ -1,79 +1,102 @@
 # frozen_string_literal: true
-
 require_relative "workflow_node_result_event_state"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A string output streamed from a Workflow execution.
+# A string output streamed from a Workflow execution.
   class WorkflowResultEventOutputDataString
-    attr_reader :id, :name, :state, :node_id, :delta, :value, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @return [String] 
+    attr_reader :name
+  # @return [Vellum::WorkflowNodeResultEventState] 
+    attr_reader :state
+  # @return [String] 
+    attr_reader :node_id
+  # @return [String] The newly output string value, meant to be concatenated with all previous. Will
+#  be non-null for events of state STREAMING.
+    attr_reader :delta
+  # @return [String] 
+    attr_reader :type
+  # @return [String] The entire string value. Will be non-null for events of state FULFILLED.
+    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 name [String]
-    # @param state [WORKFLOW_NODE_RESULT_EVENT_STATE]
-    # @param node_id [String]
-    # @param delta [String] The newly output string value, meant to be concatenated with all previous. Will be non-null for events of state STREAMING.
+    # @param id [String] 
+    # @param name [String] 
+    # @param state [Vellum::WorkflowNodeResultEventState] 
+    # @param node_id [String] 
+    # @param delta [String] The newly output string value, meant to be concatenated with all previous. Will
+#  be non-null for events of state STREAMING.
+    # @param type [String] 
     # @param value [String] The entire string value. Will be non-null for events of state FULFILLED.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [WorkflowResultEventOutputDataString]
-    def initialize(name:, state:, node_id:, id: nil, delta: nil, value: nil, additional_properties: nil)
-      # @type [String]
-      @id = id
-      # @type [String]
+    # @return [Vellum::WorkflowResultEventOutputDataString]
+    def initialize(id: OMIT, name:, state:, node_id:, delta: OMIT, type:, value: OMIT, additional_properties: nil)
+      @id = id if id != OMIT
       @name = name
-      # @type [WORKFLOW_NODE_RESULT_EVENT_STATE]
       @state = state
-      # @type [String]
       @node_id = node_id
-      # @type [String] The newly output string value, meant to be concatenated with all previous. Will be non-null for events of state STREAMING.
-      @delta = delta
-      # @type [String] The entire string value. Will be non-null for events of state FULFILLED.
-      @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @delta = delta if delta != OMIT
+      @type = type
+      @value = value if value != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "id": id, "name": name, "state": state, "node_id": node_id, "delta": delta, "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of WorkflowResultEventOutputDataString
+# Deserialize a JSON object to an instance of WorkflowResultEventOutputDataString
     #
-    # @param json_object [JSON]
-    # @return [WorkflowResultEventOutputDataString]
+    # @param json_object [String] 
+    # @return [Vellum::WorkflowResultEventOutputDataString]
     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
-      state = WORKFLOW_NODE_RESULT_EVENT_STATE.key(parsed_json["state"]) || parsed_json["state"]
-      node_id = struct.node_id
-      delta = struct.delta
-      value = struct.value
-      new(id: id, name: name, state: state, node_id: node_id, delta: delta, value: value, additional_properties: struct)
+      id = parsed_json["id"]
+      name = parsed_json["name"]
+      state = parsed_json["state"]
+      node_id = parsed_json["node_id"]
+      delta = parsed_json["delta"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]
+      new(
+        id: id,
+        name: name,
+        state: state,
+        node_id: node_id,
+        delta: delta,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of WorkflowResultEventOutputDataString to a JSON object
+# Serialize an instance of WorkflowResultEventOutputDataString to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      {
-        "id": @id,
-        "name": @name,
-        "state": WORKFLOW_NODE_RESULT_EVENT_STATE[@state] || @state,
-        "node_id": @node_id,
-        "delta": @delta,
-        "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.state.is_a?(WORKFLOW_NODE_RESULT_EVENT_STATE) != false || raise("Passed value for field obj.state 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.")
       obj.node_id.is_a?(String) != false || raise("Passed value for field obj.node_id is not the expected type, validation failed.")
       obj.delta&.is_a?(String) != false || raise("Passed value for field obj.delta 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.")
     end
   end
-end
+end

data/lib/vellum_ai/types/workflow_sandbox_example.rb

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

data/lib/vellum_ai/types/workflow_sandbox_parent_context.rb

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

data/lib/vellum_ai/types/workflow_stream_event.rb

@@ -1,90 +1,58 @@
 # frozen_string_literal: true
-
 require "json"
 require_relative "workflow_execution_workflow_result_event"
 require_relative "workflow_execution_node_result_event"
 
 module Vellum
   class WorkflowStreamEvent
-    attr_reader :member, :discriminant
 
-    private_class_method :new
-    alias kind_of? is_a?
-    # @param member [Object]
-    # @param discriminant [String]
-    # @return [WorkflowStreamEvent]
-    def initialize(member:, discriminant:)
-      # @type [Object]
-      @member = member
-      # @type [String]
-      @discriminant = discriminant
-    end
 
-    # Deserialize a JSON object to an instance of WorkflowStreamEvent
+# Deserialize a JSON object to an instance of WorkflowStreamEvent
     #
-    # @param json_object [JSON]
-    # @return [WorkflowStreamEvent]
+    # @param json_object [String] 
+    # @return [Vellum::WorkflowStreamEvent]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      member = case struct.type
-               when "WORKFLOW"
-                 WorkflowExecutionWorkflowResultEvent.from_json(json_object: json_object)
-               when "NODE"
-                 WorkflowExecutionNodeResultEvent.from_json(json_object: json_object)
-               else
-                 WorkflowExecutionWorkflowResultEvent.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 "WORKFLOW"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "NODE"
-        { **@member.to_json, type: @discriminant }.to_json
-      else
-        { "type": @discriminant, value: @member }.to_json
+      begin
+        Vellum::WorkflowExecutionWorkflowResultEvent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowExecutionWorkflowResultEvent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::WorkflowExecutionNodeResultEvent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::WorkflowExecutionNodeResultEvent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
       end
-      @member.to_json
+ 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 "WORKFLOW"
-        WorkflowExecutionWorkflowResultEvent.validate_raw(obj: obj)
-      when "NODE"
-        WorkflowExecutionNodeResultEvent.validate_raw(obj: obj)
-      else
-        raise("Passed value matched no type within the union, validation failed.")
+      begin
+        return Vellum::WorkflowExecutionWorkflowResultEvent.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 [WorkflowExecutionWorkflowResultEvent]
-    # @return [WorkflowStreamEvent]
-    def self.workflow(member:)
-      new(member: member, discriminant: "WORKFLOW")
-    end
-
-    # @param member [WorkflowExecutionNodeResultEvent]
-    # @return [WorkflowStreamEvent]
-    def self.node(member:)
-      new(member: member, discriminant: "NODE")
+      begin
+        return Vellum::WorkflowExecutionNodeResultEvent.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      raise("Passed value matched no type within the union, validation failed.")
     end
   end
-end
+end