vellum_ai 0.3.4

data/lib/vellum_ai/types/terminal_node_chat_history_result.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require_relative "chat_message"
+require "json"
+
+module Vellum
+  class TerminalNodeChatHistoryResult
+    attr_reader :id, :name, :value, :additional_properties
+
+    # @param id [String]
+    # @param name [String] The unique name given to the terminal node that produced this output.
+    # @param value [Array<ChatMessage>]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [TerminalNodeChatHistoryResult]
+    def initialize(name:, id: nil, value: nil, additional_properties: nil)
+      # @type [String]
+      @id = id
+      # @type [String] The unique name given to the terminal node that produced this output.
+      @name = name
+      # @type [Array<ChatMessage>]
+      @value = value
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of TerminalNodeChatHistoryResult
+    #
+    # @param json_object [JSON]
+    # @return [TerminalNodeChatHistoryResult]
+    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
+      value = parsed_json["value"].map do |v|
+        v = v.to_json
+        ChatMessage.from_json(json_object: v)
+      end
+      new(id: id, name: name, value: value, additional_properties: struct)
+    end
+
+    # Serialize an instance of TerminalNodeChatHistoryResult to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "id": @id, "name": @name, "value": @value }.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.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+      obj.value&.is_a?(Array) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/terminal_node_error_result.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+require_relative "vellum_error"
+require "json"
+
+module Vellum
+  class TerminalNodeErrorResult
+    attr_reader :id, :name, :value, :additional_properties
+
+    # @param id [String]
+    # @param name [String] The unique name given to the terminal node that produced this output.
+    # @param value [VellumError]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [TerminalNodeErrorResult]
+    def initialize(name:, id: nil, value: nil, additional_properties: nil)
+      # @type [String]
+      @id = id
+      # @type [String] The unique name given to the terminal node that produced this output.
+      @name = name
+      # @type [VellumError]
+      @value = value
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of TerminalNodeErrorResult
+    #
+    # @param json_object [JSON]
+    # @return [TerminalNodeErrorResult]
+    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
+      if parsed_json["value"].nil?
+        value = nil
+      else
+        value = parsed_json["value"].to_json
+        value = VellumError.from_json(json_object: value)
+      end
+      new(id: id, name: name, value: value, additional_properties: struct)
+    end
+
+    # Serialize an instance of TerminalNodeErrorResult to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "id": @id, "name": @name, "value": @value }.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.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+      obj.value.nil? || VellumError.validate_raw(obj: obj.value)
+    end
+  end
+end

data/lib/vellum_ai/types/terminal_node_json_result.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "json"
+
+module Vellum
+  class TerminalNodeJsonResult
+    attr_reader :id, :name, :value, :additional_properties
+
+    # @param id [String]
+    # @param name [String] The unique name given to the terminal node that produced this output.
+    # @param value [Hash{String => String}]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [TerminalNodeJsonResult]
+    def initialize(name:, id: nil, value: nil, additional_properties: nil)
+      # @type [String]
+      @id = id
+      # @type [String] The unique name given to the terminal node that produced this output.
+      @name = name
+      # @type [Hash{String => String}]
+      @value = value
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of TerminalNodeJsonResult
+    #
+    # @param json_object [JSON]
+    # @return [TerminalNodeJsonResult]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      JSON.parse(json_object)
+      id = struct.id
+      name = struct.name
+      value = struct.value
+      new(id: id, name: name, value: value, additional_properties: struct)
+    end
+
+    # Serialize an instance of TerminalNodeJsonResult to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "id": @id, "name": @name, "value": @value }.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.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.")
+    end
+  end
+end

data/lib/vellum_ai/types/terminal_node_number_result.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "json"
+
+module Vellum
+  class TerminalNodeNumberResult
+    attr_reader :id, :name, :value, :additional_properties
+
+    # @param id [String]
+    # @param name [String] The unique name given to the terminal node that produced this output.
+    # @param value [Float]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [TerminalNodeNumberResult]
+    def initialize(name:, id: nil, value: nil, additional_properties: nil)
+      # @type [String]
+      @id = id
+      # @type [String] The unique name given to the terminal node that produced this output.
+      @name = name
+      # @type [Float]
+      @value = value
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of TerminalNodeNumberResult
+    #
+    # @param json_object [JSON]
+    # @return [TerminalNodeNumberResult]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      JSON.parse(json_object)
+      id = struct.id
+      name = struct.name
+      value = struct.value
+      new(id: id, name: name, value: value, additional_properties: struct)
+    end
+
+    # Serialize an instance of TerminalNodeNumberResult to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "id": @id, "name": @name, "value": @value }.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.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+      obj.value&.is_a?(Float) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/terminal_node_result.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require_relative "terminal_node_result_data"
+require "json"
+
+module Vellum
+  # A Node Result Event emitted from a Terminal Node.
+  class TerminalNodeResult
+    attr_reader :data, :additional_properties
+
+    # @param data [TerminalNodeResultData]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [TerminalNodeResult]
+    def initialize(data:, additional_properties: nil)
+      # @type [TerminalNodeResultData]
+      @data = data
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of TerminalNodeResult
+    #
+    # @param json_object [JSON]
+    # @return [TerminalNodeResult]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      if parsed_json["data"].nil?
+        data = nil
+      else
+        data = parsed_json["data"].to_json
+        data = TerminalNodeResultData.from_json(json_object: data)
+      end
+      new(data: data, additional_properties: struct)
+    end
+
+    # Serialize an instance of TerminalNodeResult to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "data": @data }.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:)
+      TerminalNodeResultData.validate_raw(obj: obj.data)
+    end
+  end
+end

data/lib/vellum_ai/types/terminal_node_result_data.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+require_relative "terminal_node_result_output"
+require "json"
+
+module Vellum
+  class TerminalNodeResultData
+    attr_reader :output, :additional_properties
+
+    # @param output [TerminalNodeResultOutput]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [TerminalNodeResultData]
+    def initialize(output:, additional_properties: nil)
+      # @type [TerminalNodeResultOutput]
+      @output = output
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of TerminalNodeResultData
+    #
+    # @param json_object [JSON]
+    # @return [TerminalNodeResultData]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      if parsed_json["output"].nil?
+        output = nil
+      else
+        output = parsed_json["output"].to_json
+        output = TerminalNodeResultOutput.from_json(json_object: output)
+      end
+      new(output: output, additional_properties: struct)
+    end
+
+    # Serialize an instance of TerminalNodeResultData to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "output": @output }.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:)
+      TerminalNodeResultOutput.validate_raw(obj: obj.output)
+    end
+  end
+end

data/lib/vellum_ai/types/terminal_node_result_output.rb

@@ -0,0 +1,142 @@
+# frozen_string_literal: true
+
+require "json"
+require_relative "terminal_node_string_result"
+require_relative "terminal_node_number_result"
+require_relative "terminal_node_json_result"
+require_relative "terminal_node_chat_history_result"
+require_relative "terminal_node_search_results_result"
+require_relative "terminal_node_error_result"
+
+module Vellum
+  class TerminalNodeResultOutput
+    attr_reader :member, :discriminant
+
+    private_class_method :new
+    alias kind_of? is_a?
+    # @param member [Object]
+    # @param discriminant [String]
+    # @return [TerminalNodeResultOutput]
+    def initialize(member:, discriminant:)
+      # @type [Object]
+      @member = member
+      # @type [String]
+      @discriminant = discriminant
+    end
+
+    # Deserialize a JSON object to an instance of TerminalNodeResultOutput
+    #
+    # @param json_object [JSON]
+    # @return [TerminalNodeResultOutput]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      member = case struct.type
+               when "STRING"
+                 TerminalNodeStringResult.from_json(json_object: json_object)
+               when "NUMBER"
+                 TerminalNodeNumberResult.from_json(json_object: json_object)
+               when "JSON"
+                 TerminalNodeJsonResult.from_json(json_object: json_object)
+               when "CHAT_HISTORY"
+                 TerminalNodeChatHistoryResult.from_json(json_object: json_object)
+               when "SEARCH_RESULTS"
+                 TerminalNodeSearchResultsResult.from_json(json_object: json_object)
+               when "ERROR"
+                 TerminalNodeErrorResult.from_json(json_object: json_object)
+               else
+                 TerminalNodeStringResult.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
+      else
+        { "type": @discriminant, value: @member }.to_json
+      end
+      @member.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:)
+      case obj.type
+      when "STRING"
+        TerminalNodeStringResult.validate_raw(obj: obj)
+      when "NUMBER"
+        TerminalNodeNumberResult.validate_raw(obj: obj)
+      when "JSON"
+        TerminalNodeJsonResult.validate_raw(obj: obj)
+      when "CHAT_HISTORY"
+        TerminalNodeChatHistoryResult.validate_raw(obj: obj)
+      when "SEARCH_RESULTS"
+        TerminalNodeSearchResultsResult.validate_raw(obj: obj)
+      when "ERROR"
+        TerminalNodeErrorResult.validate_raw(obj: obj)
+      else
+        raise("Passed value matched no type within the union, validation failed.")
+      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 [TerminalNodeStringResult]
+    # @return [TerminalNodeResultOutput]
+    def self.string(member:)
+      new(member: member, discriminant: "STRING")
+    end
+
+    # @param member [TerminalNodeNumberResult]
+    # @return [TerminalNodeResultOutput]
+    def self.number(member:)
+      new(member: member, discriminant: "NUMBER")
+    end
+
+    # @param member [TerminalNodeJsonResult]
+    # @return [TerminalNodeResultOutput]
+    def self.json(member:)
+      new(member: member, discriminant: "JSON")
+    end
+
+    # @param member [TerminalNodeChatHistoryResult]
+    # @return [TerminalNodeResultOutput]
+    def self.chat_history(member:)
+      new(member: member, discriminant: "CHAT_HISTORY")
+    end
+
+    # @param member [TerminalNodeSearchResultsResult]
+    # @return [TerminalNodeResultOutput]
+    def self.search_results(member:)
+      new(member: member, discriminant: "SEARCH_RESULTS")
+    end
+
+    # @param member [TerminalNodeErrorResult]
+    # @return [TerminalNodeResultOutput]
+    def self.error(member:)
+      new(member: member, discriminant: "ERROR")
+    end
+  end
+end

data/lib/vellum_ai/types/terminal_node_search_results_result.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require_relative "search_result"
+require "json"
+
+module Vellum
+  class TerminalNodeSearchResultsResult
+    attr_reader :id, :name, :value, :additional_properties
+
+    # @param id [String]
+    # @param name [String] The unique name given to the terminal node that produced this output.
+    # @param value [Array<SearchResult>]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [TerminalNodeSearchResultsResult]
+    def initialize(name:, id: nil, value: nil, additional_properties: nil)
+      # @type [String]
+      @id = id
+      # @type [String] The unique name given to the terminal node that produced this output.
+      @name = name
+      # @type [Array<SearchResult>]
+      @value = value
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of TerminalNodeSearchResultsResult
+    #
+    # @param json_object [JSON]
+    # @return [TerminalNodeSearchResultsResult]
+    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
+      value = parsed_json["value"].map do |v|
+        v = v.to_json
+        SearchResult.from_json(json_object: v)
+      end
+      new(id: id, name: name, value: value, additional_properties: struct)
+    end
+
+    # Serialize an instance of TerminalNodeSearchResultsResult to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "id": @id, "name": @name, "value": @value }.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.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+      obj.value&.is_a?(Array) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/terminal_node_string_result.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "json"
+
+module Vellum
+  class TerminalNodeStringResult
+    attr_reader :id, :name, :value, :additional_properties
+
+    # @param id [String]
+    # @param name [String] The unique name given to the terminal node that produced this output.
+    # @param value [String]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [TerminalNodeStringResult]
+    def initialize(name:, id: nil, value: nil, additional_properties: nil)
+      # @type [String]
+      @id = id
+      # @type [String] The unique name given to the terminal node that produced this output.
+      @name = name
+      # @type [String]
+      @value = value
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of TerminalNodeStringResult
+    #
+    # @param json_object [JSON]
+    # @return [TerminalNodeStringResult]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      JSON.parse(json_object)
+      id = struct.id
+      name = struct.name
+      value = struct.value
+      new(id: id, name: name, value: value, additional_properties: struct)
+    end
+
+    # Serialize an instance of TerminalNodeStringResult to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "id": @id, "name": @name, "value": @value }.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.name.is_a?(String) != false || raise("Passed value for field obj.name 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

data/lib/vellum_ai/types/test_case_chat_history_variable_value.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require_relative "chat_message"
+require "json"
+
+module Vellum
+  class TestCaseChatHistoryVariableValue
+    attr_reader :variable_id, :value, :additional_properties
+
+    # @param variable_id [String]
+    # @param value [Array<ChatMessage>]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [TestCaseChatHistoryVariableValue]
+    def initialize(variable_id:, value: nil, additional_properties: nil)
+      # @type [String]
+      @variable_id = variable_id
+      # @type [Array<ChatMessage>]
+      @value = value
+      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @additional_properties = additional_properties
+    end
+
+    # Deserialize a JSON object to an instance of TestCaseChatHistoryVariableValue
+    #
+    # @param json_object [JSON]
+    # @return [TestCaseChatHistoryVariableValue]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      variable_id = struct.variable_id
+      value = parsed_json["value"].map do |v|
+        v = v.to_json
+        ChatMessage.from_json(json_object: v)
+      end
+      new(variable_id: variable_id, value: value, additional_properties: struct)
+    end
+
+    # Serialize an instance of TestCaseChatHistoryVariableValue to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "variable_id": @variable_id, "value": @value }.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.variable_id.is_a?(String) != false || raise("Passed value for field obj.variable_id 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