vellum_ai 0.3.14 → 0.14.74

data/lib/vellum_ai/types/fulfilled_workflow_node_result_event.rb

@@ -1,104 +1,128 @@
 # frozen_string_literal: true
-
 require "date"
 require_relative "workflow_node_result_data"
 require_relative "node_output_compiled_value"
+require "ostruct"
 require "json"
 
 module Vellum
-  # An event that indicates that the node has fulfilled its execution.
+# An event that indicates that the node has fulfilled its execution.
   class FulfilledWorkflowNodeResultEvent
-    attr_reader :id, :node_id, :node_result_id, :ts, :data, :source_execution_id, :output_values, :mocked,
-                :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @return [String] 
+    attr_reader :node_id
+  # @return [String] 
+    attr_reader :node_result_id
+  # @return [String] 
+    attr_reader :state
+  # @return [DateTime] 
+    attr_reader :ts
+  # @return [Vellum::WorkflowNodeResultData] 
+    attr_reader :data
+  # @return [String] 
+    attr_reader :source_execution_id
+  # @return [Array<Vellum::NodeOutputCompiledValue>] 
+    attr_reader :output_values
+  # @return [Boolean] 
+    attr_reader :mocked
+  # @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 node_id [String]
-    # @param node_result_id [String]
-    # @param ts [DateTime]
-    # @param data [WorkflowNodeResultData]
-    # @param source_execution_id [String]
-    # @param output_values [Array<NodeOutputCompiledValue>]
-    # @param mocked [Boolean]
+    # @param id [String] 
+    # @param node_id [String] 
+    # @param node_result_id [String] 
+    # @param state [String] 
+    # @param ts [DateTime] 
+    # @param data [Vellum::WorkflowNodeResultData] 
+    # @param source_execution_id [String] 
+    # @param output_values [Array<Vellum::NodeOutputCompiledValue>] 
+    # @param mocked [Boolean] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [FulfilledWorkflowNodeResultEvent]
-    def initialize(id:, node_id:, node_result_id:, ts: nil, data: nil, source_execution_id: nil, output_values: nil,
-                   mocked: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::FulfilledWorkflowNodeResultEvent]
+    def initialize(id:, node_id:, node_result_id:, state:, ts: OMIT, data: OMIT, source_execution_id: OMIT, output_values:, mocked: OMIT, additional_properties: nil)
       @id = id
-      # @type [String]
       @node_id = node_id
-      # @type [String]
       @node_result_id = node_result_id
-      # @type [DateTime]
-      @ts = ts
-      # @type [WorkflowNodeResultData]
-      @data = data
-      # @type [String]
-      @source_execution_id = source_execution_id
-      # @type [Array<NodeOutputCompiledValue>]
+      @state = state
+      @ts = ts if ts != OMIT
+      @data = data if data != OMIT
+      @source_execution_id = source_execution_id if source_execution_id != OMIT
       @output_values = output_values
-      # @type [Boolean]
-      @mocked = mocked
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @mocked = mocked if mocked != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "id": id, "node_id": node_id, "node_result_id": node_result_id, "state": state, "ts": ts, "data": data, "source_execution_id": source_execution_id, "output_values": output_values, "mocked": mocked }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of FulfilledWorkflowNodeResultEvent
+# Deserialize a JSON object to an instance of FulfilledWorkflowNodeResultEvent
     #
-    # @param json_object [JSON]
-    # @return [FulfilledWorkflowNodeResultEvent]
+    # @param json_object [String] 
+    # @return [Vellum::FulfilledWorkflowNodeResultEvent]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      id = struct.id
-      node_id = struct.node_id
-      node_result_id = struct.node_result_id
-      ts = DateTime.parse(parsed_json["ts"])
-      if parsed_json["data"].nil?
-        data = nil
-      else
+      id = parsed_json["id"]
+      node_id = parsed_json["node_id"]
+      node_result_id = parsed_json["node_result_id"]
+      state = parsed_json["state"]
+      ts = unless parsed_json["ts"].nil?
+  DateTime.parse(parsed_json["ts"])
+else
+  nil
+end
+      unless parsed_json["data"].nil?
         data = parsed_json["data"].to_json
-        data = WorkflowNodeResultData.from_json(json_object: data)
-      end
-      source_execution_id = struct.source_execution_id
-      output_values = parsed_json["output_values"].map do |v|
-        v = v.to_json
-        NodeOutputCompiledValue.from_json(json_object: v)
+        data = Vellum::WorkflowNodeResultData.from_json(json_object: data)
+      else
+        data = nil
       end
-      mocked = struct.mocked
-      new(id: id, node_id: node_id, node_result_id: node_result_id, ts: ts, data: data,
-          source_execution_id: source_execution_id, output_values: output_values, mocked: mocked, additional_properties: struct)
+      source_execution_id = parsed_json["source_execution_id"]
+      output_values = parsed_json["output_values"]&.map do | item |
+  item = item.to_json
+  Vellum::NodeOutputCompiledValue.from_json(json_object: item)
+end
+      mocked = parsed_json["mocked"]
+      new(
+        id: id,
+        node_id: node_id,
+        node_result_id: node_result_id,
+        state: state,
+        ts: ts,
+        data: data,
+        source_execution_id: source_execution_id,
+        output_values: output_values,
+        mocked: mocked,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of FulfilledWorkflowNodeResultEvent to a JSON object
+# Serialize an instance of FulfilledWorkflowNodeResultEvent to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      {
-        "id": @id,
-        "node_id": @node_id,
-        "node_result_id": @node_result_id,
-        "ts": @ts,
-        "data": @data,
-        "source_execution_id": @source_execution_id,
-        "output_values": @output_values,
-        "mocked": @mocked
-      }.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.node_id.is_a?(String) != false || raise("Passed value for field obj.node_id is not the expected type, validation failed.")
       obj.node_result_id.is_a?(String) != false || raise("Passed value for field obj.node_result_id is not the expected type, validation failed.")
+      obj.state.is_a?(String) != false || raise("Passed value for field obj.state is not the expected type, validation failed.")
       obj.ts&.is_a?(DateTime) != false || raise("Passed value for field obj.ts is not the expected type, validation failed.")
-      obj.data.nil? || WorkflowNodeResultData.validate_raw(obj: obj.data)
+      obj.data.nil? || Vellum::WorkflowNodeResultData.validate_raw(obj: obj.data)
       obj.source_execution_id&.is_a?(String) != false || raise("Passed value for field obj.source_execution_id is not the expected type, validation failed.")
-      obj.output_values&.is_a?(Array) != false || raise("Passed value for field obj.output_values is not the expected type, validation failed.")
+      obj.output_values.is_a?(Array) != false || raise("Passed value for field obj.output_values is not the expected type, validation failed.")
       obj.mocked&.is_a?(Boolean) != false || raise("Passed value for field obj.mocked is not the expected type, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/function_call.rb

@@ -1,90 +1,71 @@
 # frozen_string_literal: true
-
+require "ostruct"
 require "json"
-require_relative "fulfilled_function_call"
-require_relative "rejected_function_call"
 
 module Vellum
+# The final resolved function call value.
   class FunctionCall
-    attr_reader :member, :discriminant
+  # @return [Hash{String => Object}] 
+    attr_reader :arguments
+  # @return [String] 
+    attr_reader :id
+  # @return [String] 
+    attr_reader :name
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
 
-    private_class_method :new
-    alias kind_of? is_a?
-    # @param member [Object]
-    # @param discriminant [String]
-    # @return [FunctionCall]
-    def initialize(member:, discriminant:)
-      # @type [Object]
-      @member = member
-      # @type [String]
-      @discriminant = discriminant
-    end
+    OMIT = Object.new
 
-    # Deserialize a JSON object to an instance of FunctionCall
+    # @param arguments [Hash{String => Object}] 
+    # @param id [String] 
+    # @param name [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::FunctionCall]
+    def initialize(arguments:, id: OMIT, name:, additional_properties: nil)
+      @arguments = arguments
+      @id = id if id != OMIT
+      @name = name
+      @additional_properties = additional_properties
+      @_field_set = { "arguments": arguments, "id": id, "name": name }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of FunctionCall
     #
-    # @param json_object [JSON]
-    # @return [FunctionCall]
+    # @param json_object [String] 
+    # @return [Vellum::FunctionCall]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      member = case struct.state
-               when "FULFILLED"
-                 FulfilledFunctionCall.from_json(json_object: json_object)
-               when "REJECTED"
-                 RejectedFunctionCall.from_json(json_object: json_object)
-               else
-                 FulfilledFunctionCall.from_json(json_object: json_object)
-               end
-      new(member: member, discriminant: struct.state)
+      parsed_json = JSON.parse(json_object)
+      arguments = parsed_json["arguments"]
+      id = parsed_json["id"]
+      name = parsed_json["name"]
+      new(
+        arguments: arguments,
+        id: id,
+        name: name,
+        additional_properties: struct
+      )
     end
-
-    # For Union Types, to_json functionality is delegated to the wrapped member.
+# Serialize an instance of FunctionCall to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      case @discriminant
-      when "FULFILLED"
-        { **@member.to_json, state: @discriminant }.to_json
-      when "REJECTED"
-        { **@member.to_json, state: @discriminant }.to_json
-      else
-        { "state": @discriminant, value: @member }.to_json
-      end
-      @member.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:)
-      case obj.state
-      when "FULFILLED"
-        FulfilledFunctionCall.validate_raw(obj: obj)
-      when "REJECTED"
-        RejectedFunctionCall.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 [FulfilledFunctionCall]
-    # @return [FunctionCall]
-    def self.fulfilled(member:)
-      new(member: member, discriminant: "FULFILLED")
-    end
-
-    # @param member [RejectedFunctionCall]
-    # @return [FunctionCall]
-    def self.rejected(member:)
-      new(member: member, discriminant: "REJECTED")
+      obj.arguments.is_a?(Hash) != false || raise("Passed value for field obj.arguments is not the expected type, validation failed.")
+      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.")
     end
   end
-end
+end

data/lib/vellum_ai/types/function_call_chat_message_content.rb

@@ -1,52 +1,68 @@
 # frozen_string_literal: true
-
 require_relative "function_call_chat_message_content_value"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A function call value that is used in a chat message.
+# A function call value that is used in a chat message.
   class FunctionCallChatMessageContent
-    attr_reader :value, :additional_properties
+  # @return [String] 
+    attr_reader :type
+  # @return [Vellum::FunctionCallChatMessageContentValue] 
+    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 value [FunctionCallChatMessageContentValue]
+    # @param type [String] 
+    # @param value [Vellum::FunctionCallChatMessageContentValue] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [FunctionCallChatMessageContent]
-    def initialize(value:, additional_properties: nil)
-      # @type [FunctionCallChatMessageContentValue]
+    # @return [Vellum::FunctionCallChatMessageContent]
+    def initialize(type:, value:, additional_properties: nil)
+      @type = type
       @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "type": type, "value": value }
     end
-
-    # Deserialize a JSON object to an instance of FunctionCallChatMessageContent
+# Deserialize a JSON object to an instance of FunctionCallChatMessageContent
     #
-    # @param json_object [JSON]
-    # @return [FunctionCallChatMessageContent]
+    # @param json_object [String] 
+    # @return [Vellum::FunctionCallChatMessageContent]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      if parsed_json["value"].nil?
-        value = nil
-      else
+      type = parsed_json["type"]
+      unless parsed_json["value"].nil?
         value = parsed_json["value"].to_json
-        value = FunctionCallChatMessageContentValue.from_json(json_object: value)
+        value = Vellum::FunctionCallChatMessageContentValue.from_json(json_object: value)
+      else
+        value = nil
       end
-      new(value: value, additional_properties: struct)
+      new(
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of FunctionCallChatMessageContent to a JSON object
+# Serialize an instance of FunctionCallChatMessageContent to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "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:)
-      FunctionCallChatMessageContentValue.validate_raw(obj: obj.value)
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      Vellum::FunctionCallChatMessageContentValue.validate_raw(obj: obj.value)
     end
   end
-end
+end

data/lib/vellum_ai/types/function_call_chat_message_content_request.rb

@@ -1,52 +1,69 @@
 # frozen_string_literal: true
-
 require_relative "function_call_chat_message_content_value_request"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A function call value that is used in a chat message.
+# A function call value that is used in a chat message.
   class FunctionCallChatMessageContentRequest
-    attr_reader :value, :additional_properties
+  # @return [String] 
+    attr_reader :type
+  # @return [Vellum::FunctionCallChatMessageContentValueRequest] 
+    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 value [FunctionCallChatMessageContentValueRequest]
+    # @param type [String] 
+    # @param value [Vellum::FunctionCallChatMessageContentValueRequest] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [FunctionCallChatMessageContentRequest]
-    def initialize(value:, additional_properties: nil)
-      # @type [FunctionCallChatMessageContentValueRequest]
+    # @return [Vellum::FunctionCallChatMessageContentRequest]
+    def initialize(type:, value:, additional_properties: nil)
+      @type = type
       @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "type": type, "value": value }
     end
-
-    # Deserialize a JSON object to an instance of FunctionCallChatMessageContentRequest
+# Deserialize a JSON object to an instance of
+#  FunctionCallChatMessageContentRequest
     #
-    # @param json_object [JSON]
-    # @return [FunctionCallChatMessageContentRequest]
+    # @param json_object [String] 
+    # @return [Vellum::FunctionCallChatMessageContentRequest]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      if parsed_json["value"].nil?
-        value = nil
-      else
+      type = parsed_json["type"]
+      unless parsed_json["value"].nil?
         value = parsed_json["value"].to_json
-        value = FunctionCallChatMessageContentValueRequest.from_json(json_object: value)
+        value = Vellum::FunctionCallChatMessageContentValueRequest.from_json(json_object: value)
+      else
+        value = nil
       end
-      new(value: value, additional_properties: struct)
+      new(
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of FunctionCallChatMessageContentRequest to a JSON object
+# Serialize an instance of FunctionCallChatMessageContentRequest to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "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:)
-      FunctionCallChatMessageContentValueRequest.validate_raw(obj: obj.value)
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      Vellum::FunctionCallChatMessageContentValueRequest.validate_raw(obj: obj.value)
     end
   end
-end
+end

data/lib/vellum_ai/types/function_call_chat_message_content_value.rb

@@ -1,51 +1,66 @@
 # frozen_string_literal: true
-
+require "ostruct"
 require "json"
 
 module Vellum
-  # The final resolved function call value.
+# The final resolved function call value.
   class FunctionCallChatMessageContentValue
-    attr_reader :name, :arguments, :id, :additional_properties
+  # @return [String] 
+    attr_reader :name
+  # @return [Hash{String => Object}] 
+    attr_reader :arguments
+  # @return [String] 
+    attr_reader :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 arguments [Hash{String => String}]
-    # @param id [String]
+    # @param name [String] 
+    # @param arguments [Hash{String => Object}] 
+    # @param id [String] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [FunctionCallChatMessageContentValue]
-    def initialize(name:, arguments:, id: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::FunctionCallChatMessageContentValue]
+    def initialize(name:, arguments:, id: OMIT, additional_properties: nil)
       @name = name
-      # @type [Hash{String => String}]
       @arguments = arguments
-      # @type [String]
-      @id = id
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @id = id if id != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "name": name, "arguments": arguments, "id": id }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of FunctionCallChatMessageContentValue
+# Deserialize a JSON object to an instance of FunctionCallChatMessageContentValue
     #
-    # @param json_object [JSON]
-    # @return [FunctionCallChatMessageContentValue]
+    # @param json_object [String] 
+    # @return [Vellum::FunctionCallChatMessageContentValue]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      JSON.parse(json_object)
-      name = struct.name
-      arguments = struct.arguments
-      id = struct.id
-      new(name: name, arguments: arguments, id: id, additional_properties: struct)
+      parsed_json = JSON.parse(json_object)
+      name = parsed_json["name"]
+      arguments = parsed_json["arguments"]
+      id = parsed_json["id"]
+      new(
+        name: name,
+        arguments: arguments,
+        id: id,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of FunctionCallChatMessageContentValue to a JSON object
+# Serialize an instance of FunctionCallChatMessageContentValue to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "name": @name, "arguments": @arguments, "id": @id }.to_json
+    # @return [String]
+    def to_json
+      @_field_set&.to_json
     end
-
-    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+# Leveraged for Union-type generation, validate_raw attempts to parse the given
+#  hash and check each fields type against the current object's property
+#  definitions.
     #
-    # @param obj [Object]
+    # @param obj [Object] 
     # @return [Void]
     def self.validate_raw(obj:)
       obj.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
@@ -53,4 +68,4 @@ module Vellum
       obj.id&.is_a?(String) != false || raise("Passed value for field obj.id is not the expected type, validation failed.")
     end
   end
-end
+end