vellum_ai 0.3.15 → 0.14.75

data/lib/vellum_ai/types/ad_hoc_initiated_prompt_execution_meta.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+# The subset of the metadata tracked by Vellum during prompt execution that the
+#  request opted into with `expand_meta`.
+  class AdHocInitiatedPromptExecutionMeta
+  # @return [String] 
+    attr_reader :model_name
+  # @return [Integer] 
+    attr_reader :latency
+  # @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 model_name [String] 
+    # @param latency [Integer] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::AdHocInitiatedPromptExecutionMeta]
+    def initialize(model_name: OMIT, latency: OMIT, additional_properties: nil)
+      @model_name = model_name if model_name != OMIT
+      @latency = latency if latency != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "model_name": model_name, "latency": latency }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of AdHocInitiatedPromptExecutionMeta
+    #
+    # @param json_object [String] 
+    # @return [Vellum::AdHocInitiatedPromptExecutionMeta]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      model_name = parsed_json["model_name"]
+      latency = parsed_json["latency"]
+      new(
+        model_name: model_name,
+        latency: latency,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of AdHocInitiatedPromptExecutionMeta 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.model_name&.is_a?(String) != false || raise("Passed value for field obj.model_name is not the expected type, validation failed.")
+      obj.latency&.is_a?(Integer) != false || raise("Passed value for field obj.latency is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/ad_hoc_rejected_prompt_execution_meta.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+require_relative "finish_reason_enum"
+require "ostruct"
+require "json"
+
+module Vellum
+# The subset of the metadata tracked by Vellum during prompt execution that the
+#  request opted into with `expand_meta`.
+  class AdHocRejectedPromptExecutionMeta
+  # @return [Integer] 
+    attr_reader :latency
+  # @return [Vellum::FinishReasonEnum] 
+    attr_reader :finish_reason
+  # @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 latency [Integer] 
+    # @param finish_reason [Vellum::FinishReasonEnum] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::AdHocRejectedPromptExecutionMeta]
+    def initialize(latency: OMIT, finish_reason: OMIT, additional_properties: nil)
+      @latency = latency if latency != OMIT
+      @finish_reason = finish_reason if finish_reason != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "latency": latency, "finish_reason": finish_reason }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of AdHocRejectedPromptExecutionMeta
+    #
+    # @param json_object [String] 
+    # @return [Vellum::AdHocRejectedPromptExecutionMeta]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      latency = parsed_json["latency"]
+      finish_reason = parsed_json["finish_reason"]
+      new(
+        latency: latency,
+        finish_reason: finish_reason,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of AdHocRejectedPromptExecutionMeta 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.latency&.is_a?(Integer) != false || raise("Passed value for field obj.latency is not the expected type, validation failed.")
+      obj.finish_reason&.is_a?(Vellum::FinishReasonEnum) != false || raise("Passed value for field obj.finish_reason is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/ad_hoc_streaming_prompt_execution_meta.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+# The subset of the metadata tracked by Vellum during prompt execution that the
+#  request opted into with `expand_meta`.
+  class AdHocStreamingPromptExecutionMeta
+  # @return [Integer] 
+    attr_reader :latency
+  # @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 latency [Integer] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::AdHocStreamingPromptExecutionMeta]
+    def initialize(latency: OMIT, additional_properties: nil)
+      @latency = latency if latency != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "latency": latency }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of AdHocStreamingPromptExecutionMeta
+    #
+    # @param json_object [String] 
+    # @return [Vellum::AdHocStreamingPromptExecutionMeta]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      latency = parsed_json["latency"]
+      new(latency: latency, additional_properties: struct)
+    end
+# Serialize an instance of AdHocStreamingPromptExecutionMeta 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.latency&.is_a?(Integer) != false || raise("Passed value for field obj.latency is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/api_node_result.rb

@@ -1,52 +1,68 @@
 # frozen_string_literal: true
-
 require_relative "api_node_result_data"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A Node Result Event emitted from an API Node.
+# A Node Result Event emitted from an API Node.
   class ApiNodeResult
-    attr_reader :data, :additional_properties
+  # @return [String] 
+    attr_reader :type
+  # @return [Vellum::ApiNodeResultData] 
+    attr_reader :data
+  # @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 data [ApiNodeResultData]
+    # @param type [String] 
+    # @param data [Vellum::ApiNodeResultData] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [ApiNodeResult]
-    def initialize(data:, additional_properties: nil)
-      # @type [ApiNodeResultData]
+    # @return [Vellum::ApiNodeResult]
+    def initialize(type:, data:, additional_properties: nil)
+      @type = type
       @data = data
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "type": type, "data": data }
     end
-
-    # Deserialize a JSON object to an instance of ApiNodeResult
+# Deserialize a JSON object to an instance of ApiNodeResult
     #
-    # @param json_object [JSON]
-    # @return [ApiNodeResult]
+    # @param json_object [String] 
+    # @return [Vellum::ApiNodeResult]
     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
+      type = parsed_json["type"]
+      unless parsed_json["data"].nil?
         data = parsed_json["data"].to_json
-        data = ApiNodeResultData.from_json(json_object: data)
+        data = Vellum::ApiNodeResultData.from_json(json_object: data)
+      else
+        data = nil
       end
-      new(data: data, additional_properties: struct)
+      new(
+        type: type,
+        data: data,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of ApiNodeResult to a JSON object
+# Serialize an instance of ApiNodeResult to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "data": @data }.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:)
-      ApiNodeResultData.validate_raw(obj: obj.data)
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      Vellum::ApiNodeResultData.validate_raw(obj: obj.data)
     end
   end
-end
+end

data/lib/vellum_ai/types/api_node_result_data.rb

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

data/lib/vellum_ai/types/api_request_parent_context.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+require_relative "parent_context"
+require_relative "span_link"
+require "ostruct"
+require "json"
+
+module Vellum
+  class ApiRequestParentContext
+  # @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 [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 additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::ApiRequestParentContext]
+    def initialize(parent: OMIT, links: OMIT, type:, span_id:, additional_properties: nil)
+      @parent = parent if parent != OMIT
+      @links = links if links != OMIT
+      @type = type
+      @span_id = span_id
+      @additional_properties = additional_properties
+      @_field_set = { "parent": parent, "links": links, "type": type, "span_id": span_id }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of ApiRequestParentContext
+    #
+    # @param json_object [String] 
+    # @return [Vellum::ApiRequestParentContext]
+    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"]
+      new(
+        parent: parent,
+        links: links,
+        type: type,
+        span_id: span_id,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of ApiRequestParentContext 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.")
+    end
+  end
+end

data/lib/vellum_ai/types/{image_enum.rb → api_version_enum.rb}

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Vellum
-  IMAGE_ENUM = String
-end
+  API_VERSION_ENUM = String
+end

data/lib/vellum_ai/types/array_chat_message_content.rb

@@ -1,50 +1,66 @@
 # frozen_string_literal: true
-
 require_relative "array_chat_message_content_item"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A list of chat message content items.
+# A list of chat message content items.
   class ArrayChatMessageContent
-    attr_reader :value, :additional_properties
+  # @return [String] 
+    attr_reader :type
+  # @return [Array<Vellum::ArrayChatMessageContentItem>] 
+    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 [Array<ArrayChatMessageContentItem>]
+    # @param type [String] 
+    # @param value [Array<Vellum::ArrayChatMessageContentItem>] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [ArrayChatMessageContent]
-    def initialize(value:, additional_properties: nil)
-      # @type [Array<ArrayChatMessageContentItem>]
+    # @return [Vellum::ArrayChatMessageContent]
+    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 ArrayChatMessageContent
+# Deserialize a JSON object to an instance of ArrayChatMessageContent
     #
-    # @param json_object [JSON]
-    # @return [ArrayChatMessageContent]
+    # @param json_object [String] 
+    # @return [Vellum::ArrayChatMessageContent]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      value = parsed_json["value"].map do |v|
-        v = v.to_json
-        ArrayChatMessageContentItem.from_json(json_object: v)
-      end
-      new(value: value, additional_properties: struct)
+      type = parsed_json["type"]
+      value = parsed_json["value"]&.map do | item |
+  item = item.to_json
+  Vellum::ArrayChatMessageContentItem.from_json(json_object: item)
+end
+      new(
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of ArrayChatMessageContent to a JSON object
+# Serialize an instance of ArrayChatMessageContent 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:)
+      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