vellum_ai 0.3.15 → 0.14.74

data/lib/vellum_ai/types/node_execution_resumed_event.rb

@@ -0,0 +1,134 @@
+# frozen_string_literal: true
+require_relative "parent_context"
+require_relative "span_link"
+require_relative "node_execution_resumed_body"
+require "date"
+require_relative "api_version_enum"
+require "ostruct"
+require "json"
+
+module Vellum
+  class NodeExecutionResumedEvent
+  # @return [Vellum::ParentContext] 
+    attr_reader :parent
+  # @return [Array<Vellum::SpanLink>] 
+    attr_reader :links
+  # @return [String] 
+    attr_reader :name
+  # @return [Vellum::NodeExecutionResumedBody] 
+    attr_reader :body
+  # @return [String] 
+    attr_reader :id
+  # @return [DateTime] 
+    attr_reader :timestamp
+  # @return [Vellum::API_VERSION_ENUM] 
+    attr_reader :api_version
+  # @return [String] 
+    attr_reader :trace_id
+  # @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 name [String] 
+    # @param body [Vellum::NodeExecutionResumedBody] 
+    # @param id [String] 
+    # @param timestamp [DateTime] 
+    # @param api_version [Vellum::API_VERSION_ENUM] 
+    # @param trace_id [String] 
+    # @param span_id [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::NodeExecutionResumedEvent]
+    def initialize(parent: OMIT, links: OMIT, name:, body:, id:, timestamp:, api_version: OMIT, trace_id:, span_id:, additional_properties: nil)
+      @parent = parent if parent != OMIT
+      @links = links if links != OMIT
+      @name = name
+      @body = body
+      @id = id
+      @timestamp = timestamp
+      @api_version = api_version if api_version != OMIT
+      @trace_id = trace_id
+      @span_id = span_id
+      @additional_properties = additional_properties
+      @_field_set = { "parent": parent, "links": links, "name": name, "body": body, "id": id, "timestamp": timestamp, "api_version": api_version, "trace_id": trace_id, "span_id": span_id }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of NodeExecutionResumedEvent
+    #
+    # @param json_object [String] 
+    # @return [Vellum::NodeExecutionResumedEvent]
+    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
+      name = parsed_json["name"]
+      unless parsed_json["body"].nil?
+        body = parsed_json["body"].to_json
+        body = Vellum::NodeExecutionResumedBody.from_json(json_object: body)
+      else
+        body = nil
+      end
+      id = parsed_json["id"]
+      timestamp = unless parsed_json["timestamp"].nil?
+  DateTime.parse(parsed_json["timestamp"])
+else
+  nil
+end
+      api_version = parsed_json["api_version"]
+      trace_id = parsed_json["trace_id"]
+      span_id = parsed_json["span_id"]
+      new(
+        parent: parent,
+        links: links,
+        name: name,
+        body: body,
+        id: id,
+        timestamp: timestamp,
+        api_version: api_version,
+        trace_id: trace_id,
+        span_id: span_id,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of NodeExecutionResumedEvent 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.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+      Vellum::NodeExecutionResumedBody.validate_raw(obj: obj.body)
+      obj.id.is_a?(String) != false || raise("Passed value for field obj.id is not the expected type, validation failed.")
+      obj.timestamp.is_a?(DateTime) != false || raise("Passed value for field obj.timestamp is not the expected type, validation failed.")
+      obj.api_version&.is_a?(String) != false || raise("Passed value for field obj.api_version is not the expected type, validation failed.")
+      obj.trace_id.is_a?(String) != false || raise("Passed value for field obj.trace_id 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/node_execution_span.rb

@@ -0,0 +1,118 @@
+# frozen_string_literal: true
+require_relative "vellum_node_execution_event"
+require_relative "node_execution_span_attributes"
+require "date"
+require "date"
+require "ostruct"
+require "json"
+
+module Vellum
+  class NodeExecutionSpan
+  # @return [String] 
+    attr_reader :name
+  # @return [Array<Vellum::VellumNodeExecutionEvent>] 
+    attr_reader :events
+  # @return [Vellum::NodeExecutionSpanAttributes] 
+    attr_reader :attributes
+  # @return [String] 
+    attr_reader :span_id
+  # @return [DateTime] 
+    attr_reader :start_ts
+  # @return [DateTime] 
+    attr_reader :end_ts
+  # @return [String] 
+    attr_reader :parent_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 name [String] 
+    # @param events [Array<Vellum::VellumNodeExecutionEvent>] 
+    # @param attributes [Vellum::NodeExecutionSpanAttributes] 
+    # @param span_id [String] 
+    # @param start_ts [DateTime] 
+    # @param end_ts [DateTime] 
+    # @param parent_span_id [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::NodeExecutionSpan]
+    def initialize(name:, events:, attributes:, span_id:, start_ts:, end_ts:, parent_span_id: OMIT, additional_properties: nil)
+      @name = name
+      @events = events
+      @attributes = attributes
+      @span_id = span_id
+      @start_ts = start_ts
+      @end_ts = end_ts
+      @parent_span_id = parent_span_id if parent_span_id != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "name": name, "events": events, "attributes": attributes, "span_id": span_id, "start_ts": start_ts, "end_ts": end_ts, "parent_span_id": parent_span_id }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of NodeExecutionSpan
+    #
+    # @param json_object [String] 
+    # @return [Vellum::NodeExecutionSpan]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      name = parsed_json["name"]
+      events = parsed_json["events"]&.map do | item |
+  item = item.to_json
+  Vellum::VellumNodeExecutionEvent.from_json(json_object: item)
+end
+      unless parsed_json["attributes"].nil?
+        attributes = parsed_json["attributes"].to_json
+        attributes = Vellum::NodeExecutionSpanAttributes.from_json(json_object: attributes)
+      else
+        attributes = nil
+      end
+      span_id = parsed_json["span_id"]
+      start_ts = unless parsed_json["start_ts"].nil?
+  DateTime.parse(parsed_json["start_ts"])
+else
+  nil
+end
+      end_ts = unless parsed_json["end_ts"].nil?
+  DateTime.parse(parsed_json["end_ts"])
+else
+  nil
+end
+      parent_span_id = parsed_json["parent_span_id"]
+      new(
+        name: name,
+        events: events,
+        attributes: attributes,
+        span_id: span_id,
+        start_ts: start_ts,
+        end_ts: end_ts,
+        parent_span_id: parent_span_id,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of NodeExecutionSpan 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.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+      obj.events.is_a?(Array) != false || raise("Passed value for field obj.events is not the expected type, validation failed.")
+      Vellum::NodeExecutionSpanAttributes.validate_raw(obj: obj.attributes)
+      obj.span_id.is_a?(String) != false || raise("Passed value for field obj.span_id is not the expected type, validation failed.")
+      obj.start_ts.is_a?(DateTime) != false || raise("Passed value for field obj.start_ts is not the expected type, validation failed.")
+      obj.end_ts.is_a?(DateTime) != false || raise("Passed value for field obj.end_ts is not the expected type, validation failed.")
+      obj.parent_span_id&.is_a?(String) != false || raise("Passed value for field obj.parent_span_id is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/node_execution_span_attributes.rb

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

data/lib/vellum_ai/types/node_execution_streaming_body.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+require_relative "vellum_code_resource_definition"
+require_relative "base_output"
+require "ostruct"
+require "json"
+
+module Vellum
+  class NodeExecutionStreamingBody
+  # @return [Vellum::VellumCodeResourceDefinition] 
+    attr_reader :node_definition
+  # @return [Vellum::BaseOutput] 
+    attr_reader :output
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param node_definition [Vellum::VellumCodeResourceDefinition] 
+    # @param output [Vellum::BaseOutput] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::NodeExecutionStreamingBody]
+    def initialize(node_definition:, output:, additional_properties: nil)
+      @node_definition = node_definition
+      @output = output
+      @additional_properties = additional_properties
+      @_field_set = { "node_definition": node_definition, "output": output }
+    end
+# Deserialize a JSON object to an instance of NodeExecutionStreamingBody
+    #
+    # @param json_object [String] 
+    # @return [Vellum::NodeExecutionStreamingBody]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      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
+      unless parsed_json["output"].nil?
+        output = parsed_json["output"].to_json
+        output = Vellum::BaseOutput.from_json(json_object: output)
+      else
+        output = nil
+      end
+      new(
+        node_definition: node_definition,
+        output: output,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of NodeExecutionStreamingBody 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:)
+      Vellum::VellumCodeResourceDefinition.validate_raw(obj: obj.node_definition)
+      Vellum::BaseOutput.validate_raw(obj: obj.output)
+    end
+  end
+end

data/lib/vellum_ai/types/node_execution_streaming_event.rb

@@ -0,0 +1,134 @@
+# frozen_string_literal: true
+require_relative "parent_context"
+require_relative "span_link"
+require_relative "node_execution_streaming_body"
+require "date"
+require_relative "api_version_enum"
+require "ostruct"
+require "json"
+
+module Vellum
+  class NodeExecutionStreamingEvent
+  # @return [Vellum::ParentContext] 
+    attr_reader :parent
+  # @return [Array<Vellum::SpanLink>] 
+    attr_reader :links
+  # @return [String] 
+    attr_reader :name
+  # @return [Vellum::NodeExecutionStreamingBody] 
+    attr_reader :body
+  # @return [String] 
+    attr_reader :id
+  # @return [DateTime] 
+    attr_reader :timestamp
+  # @return [Vellum::API_VERSION_ENUM] 
+    attr_reader :api_version
+  # @return [String] 
+    attr_reader :trace_id
+  # @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 name [String] 
+    # @param body [Vellum::NodeExecutionStreamingBody] 
+    # @param id [String] 
+    # @param timestamp [DateTime] 
+    # @param api_version [Vellum::API_VERSION_ENUM] 
+    # @param trace_id [String] 
+    # @param span_id [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::NodeExecutionStreamingEvent]
+    def initialize(parent: OMIT, links: OMIT, name:, body:, id:, timestamp:, api_version: OMIT, trace_id:, span_id:, additional_properties: nil)
+      @parent = parent if parent != OMIT
+      @links = links if links != OMIT
+      @name = name
+      @body = body
+      @id = id
+      @timestamp = timestamp
+      @api_version = api_version if api_version != OMIT
+      @trace_id = trace_id
+      @span_id = span_id
+      @additional_properties = additional_properties
+      @_field_set = { "parent": parent, "links": links, "name": name, "body": body, "id": id, "timestamp": timestamp, "api_version": api_version, "trace_id": trace_id, "span_id": span_id }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of NodeExecutionStreamingEvent
+    #
+    # @param json_object [String] 
+    # @return [Vellum::NodeExecutionStreamingEvent]
+    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
+      name = parsed_json["name"]
+      unless parsed_json["body"].nil?
+        body = parsed_json["body"].to_json
+        body = Vellum::NodeExecutionStreamingBody.from_json(json_object: body)
+      else
+        body = nil
+      end
+      id = parsed_json["id"]
+      timestamp = unless parsed_json["timestamp"].nil?
+  DateTime.parse(parsed_json["timestamp"])
+else
+  nil
+end
+      api_version = parsed_json["api_version"]
+      trace_id = parsed_json["trace_id"]
+      span_id = parsed_json["span_id"]
+      new(
+        parent: parent,
+        links: links,
+        name: name,
+        body: body,
+        id: id,
+        timestamp: timestamp,
+        api_version: api_version,
+        trace_id: trace_id,
+        span_id: span_id,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of NodeExecutionStreamingEvent 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.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+      Vellum::NodeExecutionStreamingBody.validate_raw(obj: obj.body)
+      obj.id.is_a?(String) != false || raise("Passed value for field obj.id is not the expected type, validation failed.")
+      obj.timestamp.is_a?(DateTime) != false || raise("Passed value for field obj.timestamp is not the expected type, validation failed.")
+      obj.api_version&.is_a?(String) != false || raise("Passed value for field obj.api_version is not the expected type, validation failed.")
+      obj.trace_id.is_a?(String) != false || raise("Passed value for field obj.trace_id 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/node_input_compiled_array_value.rb

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