vellum_ai 0.3.15 → 0.14.74

data/lib/vellum_ai/types/rejected_execute_prompt_event.rb

@@ -1,68 +1,90 @@
 # frozen_string_literal: true
-
 require_relative "vellum_error"
 require_relative "rejected_prompt_execution_meta"
+require "ostruct"
 require "json"
 
 module Vellum
-  # The final data returned indicating an error occurred during the stream.
+# The final data returned indicating an error occurred during the stream.
   class RejectedExecutePromptEvent
-    attr_reader :error, :execution_id, :meta, :additional_properties
+  # @return [String] 
+    attr_reader :state
+  # @return [Vellum::VellumError] 
+    attr_reader :error
+  # @return [String] 
+    attr_reader :execution_id
+  # @return [Vellum::RejectedPromptExecutionMeta] 
+    attr_reader :meta
+  # @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 error [VellumError]
-    # @param execution_id [String]
-    # @param meta [RejectedPromptExecutionMeta]
+    # @param state [String] 
+    # @param error [Vellum::VellumError] 
+    # @param execution_id [String] 
+    # @param meta [Vellum::RejectedPromptExecutionMeta] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [RejectedExecutePromptEvent]
-    def initialize(error:, execution_id:, meta: nil, additional_properties: nil)
-      # @type [VellumError]
+    # @return [Vellum::RejectedExecutePromptEvent]
+    def initialize(state:, error:, execution_id:, meta: OMIT, additional_properties: nil)
+      @state = state
       @error = error
-      # @type [String]
       @execution_id = execution_id
-      # @type [RejectedPromptExecutionMeta]
-      @meta = meta
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @meta = meta if meta != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "state": state, "error": error, "execution_id": execution_id, "meta": meta }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of RejectedExecutePromptEvent
+# Deserialize a JSON object to an instance of RejectedExecutePromptEvent
     #
-    # @param json_object [JSON]
-    # @return [RejectedExecutePromptEvent]
+    # @param json_object [String] 
+    # @return [Vellum::RejectedExecutePromptEvent]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      if parsed_json["error"].nil?
-        error = nil
-      else
+      state = parsed_json["state"]
+      unless parsed_json["error"].nil?
         error = parsed_json["error"].to_json
-        error = VellumError.from_json(json_object: error)
-      end
-      execution_id = struct.execution_id
-      if parsed_json["meta"].nil?
-        meta = nil
+        error = Vellum::VellumError.from_json(json_object: error)
       else
+        error = nil
+      end
+      execution_id = parsed_json["execution_id"]
+      unless parsed_json["meta"].nil?
         meta = parsed_json["meta"].to_json
-        meta = RejectedPromptExecutionMeta.from_json(json_object: meta)
+        meta = Vellum::RejectedPromptExecutionMeta.from_json(json_object: meta)
+      else
+        meta = nil
       end
-      new(error: error, execution_id: execution_id, meta: meta, additional_properties: struct)
+      new(
+        state: state,
+        error: error,
+        execution_id: execution_id,
+        meta: meta,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of RejectedExecutePromptEvent to a JSON object
+# Serialize an instance of RejectedExecutePromptEvent to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "error": @error, "execution_id": @execution_id, "meta": @meta }.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:)
-      VellumError.validate_raw(obj: obj.error)
+      obj.state.is_a?(String) != false || raise("Passed value for field obj.state is not the expected type, validation failed.")
+      Vellum::VellumError.validate_raw(obj: obj.error)
       obj.execution_id.is_a?(String) != false || raise("Passed value for field obj.execution_id is not the expected type, validation failed.")
-      obj.meta.nil? || RejectedPromptExecutionMeta.validate_raw(obj: obj.meta)
+      obj.meta.nil? || Vellum::RejectedPromptExecutionMeta.validate_raw(obj: obj.meta)
     end
   end
-end
+end

data/lib/vellum_ai/types/rejected_execute_prompt_response.rb

@@ -1,73 +1,99 @@
 # frozen_string_literal: true
-
 require_relative "prompt_execution_meta"
 require_relative "vellum_error"
+require "ostruct"
 require "json"
 
 module Vellum
-  # The unsuccessful response from the model containing an error of what went wrong.
+# The unsuccessful response from the model containing an error of what went wrong.
   class RejectedExecutePromptResponse
-    attr_reader :meta, :raw, :execution_id, :error, :additional_properties
+  # @return [Vellum::PromptExecutionMeta] 
+    attr_reader :meta
+  # @return [Hash{String => Object}] The subset of the raw response from the model that the request opted into with
+#  `expand_raw`.
+    attr_reader :raw
+  # @return [String] The ID of the execution.
+    attr_reader :execution_id
+  # @return [String] 
+    attr_reader :state
+  # @return [Vellum::VellumError] 
+    attr_reader :error
+  # @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 meta [PromptExecutionMeta]
-    # @param raw [Hash{String => String}] The subset of the raw response from the model that the request opted into with `expand_raw`.
+    # @param meta [Vellum::PromptExecutionMeta] 
+    # @param raw [Hash{String => Object}] The subset of the raw response from the model that the request opted into with
+#  `expand_raw`.
     # @param execution_id [String] The ID of the execution.
-    # @param error [VellumError]
+    # @param state [String] 
+    # @param error [Vellum::VellumError] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [RejectedExecutePromptResponse]
-    def initialize(execution_id:, error:, meta: nil, raw: nil, additional_properties: nil)
-      # @type [PromptExecutionMeta]
-      @meta = meta
-      # @type [Hash{String => String}] The subset of the raw response from the model that the request opted into with `expand_raw`.
-      @raw = raw
-      # @type [String] The ID of the execution.
+    # @return [Vellum::RejectedExecutePromptResponse]
+    def initialize(meta: OMIT, raw: OMIT, execution_id:, state:, error:, additional_properties: nil)
+      @meta = meta if meta != OMIT
+      @raw = raw if raw != OMIT
       @execution_id = execution_id
-      # @type [VellumError]
+      @state = state
       @error = error
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "meta": meta, "raw": raw, "execution_id": execution_id, "state": state, "error": error }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of RejectedExecutePromptResponse
+# Deserialize a JSON object to an instance of RejectedExecutePromptResponse
     #
-    # @param json_object [JSON]
-    # @return [RejectedExecutePromptResponse]
+    # @param json_object [String] 
+    # @return [Vellum::RejectedExecutePromptResponse]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      if parsed_json["meta"].nil?
-        meta = nil
-      else
+      unless parsed_json["meta"].nil?
         meta = parsed_json["meta"].to_json
-        meta = PromptExecutionMeta.from_json(json_object: meta)
-      end
-      raw = struct.raw
-      execution_id = struct.execution_id
-      if parsed_json["error"].nil?
-        error = nil
+        meta = Vellum::PromptExecutionMeta.from_json(json_object: meta)
       else
+        meta = nil
+      end
+      raw = parsed_json["raw"]
+      execution_id = parsed_json["execution_id"]
+      state = parsed_json["state"]
+      unless parsed_json["error"].nil?
         error = parsed_json["error"].to_json
-        error = VellumError.from_json(json_object: error)
+        error = Vellum::VellumError.from_json(json_object: error)
+      else
+        error = nil
       end
-      new(meta: meta, raw: raw, execution_id: execution_id, error: error, additional_properties: struct)
+      new(
+        meta: meta,
+        raw: raw,
+        execution_id: execution_id,
+        state: state,
+        error: error,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of RejectedExecutePromptResponse to a JSON object
+# Serialize an instance of RejectedExecutePromptResponse to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "meta": @meta, "raw": @raw, "execution_id": @execution_id, "error": @error }.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.meta.nil? || PromptExecutionMeta.validate_raw(obj: obj.meta)
+      obj.meta.nil? || Vellum::PromptExecutionMeta.validate_raw(obj: obj.meta)
       obj.raw&.is_a?(Hash) != false || raise("Passed value for field obj.raw is not the expected type, validation failed.")
       obj.execution_id.is_a?(String) != false || raise("Passed value for field obj.execution_id is not the expected type, validation failed.")
-      VellumError.validate_raw(obj: obj.error)
+      obj.state.is_a?(String) != false || raise("Passed value for field obj.state is not the expected type, validation failed.")
+      Vellum::VellumError.validate_raw(obj: obj.error)
     end
   end
-end
+end

data/lib/vellum_ai/types/rejected_execute_workflow_workflow_result_event.rb

@@ -1,63 +1,90 @@
 # frozen_string_literal: true
-
 require "date"
 require_relative "workflow_event_error"
+require "ostruct"
 require "json"
 
 module Vellum
-  # The unsuccessful response from the Workflow execution containing an error specifying what went wrong.
+# The unsuccessful response from the Workflow execution containing an error
+#  specifying what went wrong.
   class RejectedExecuteWorkflowWorkflowResultEvent
-    attr_reader :id, :ts, :error, :additional_properties
+  # @return [String] 
+    attr_reader :id
+  # @return [String] 
+    attr_reader :state
+  # @return [DateTime] 
+    attr_reader :ts
+  # @return [Vellum::WorkflowEventError] 
+    attr_reader :error
+  # @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 ts [DateTime]
-    # @param error [WorkflowEventError]
+    # @param id [String] 
+    # @param state [String] 
+    # @param ts [DateTime] 
+    # @param error [Vellum::WorkflowEventError] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [RejectedExecuteWorkflowWorkflowResultEvent]
-    def initialize(id:, ts:, error:, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::RejectedExecuteWorkflowWorkflowResultEvent]
+    def initialize(id:, state:, ts:, error:, additional_properties: nil)
       @id = id
-      # @type [DateTime]
+      @state = state
       @ts = ts
-      # @type [WorkflowEventError]
       @error = error
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "id": id, "state": state, "ts": ts, "error": error }
     end
-
-    # Deserialize a JSON object to an instance of RejectedExecuteWorkflowWorkflowResultEvent
+# Deserialize a JSON object to an instance of
+#  RejectedExecuteWorkflowWorkflowResultEvent
     #
-    # @param json_object [JSON]
-    # @return [RejectedExecuteWorkflowWorkflowResultEvent]
+    # @param json_object [String] 
+    # @return [Vellum::RejectedExecuteWorkflowWorkflowResultEvent]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      id = struct.id
-      ts = DateTime.parse(parsed_json["ts"])
-      if parsed_json["error"].nil?
-        error = nil
-      else
+      id = parsed_json["id"]
+      state = parsed_json["state"]
+      ts = unless parsed_json["ts"].nil?
+  DateTime.parse(parsed_json["ts"])
+else
+  nil
+end
+      unless parsed_json["error"].nil?
         error = parsed_json["error"].to_json
-        error = WorkflowEventError.from_json(json_object: error)
+        error = Vellum::WorkflowEventError.from_json(json_object: error)
+      else
+        error = nil
       end
-      new(id: id, ts: ts, error: error, additional_properties: struct)
+      new(
+        id: id,
+        state: state,
+        ts: ts,
+        error: error,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of RejectedExecuteWorkflowWorkflowResultEvent to a JSON object
+# Serialize an instance of RejectedExecuteWorkflowWorkflowResultEvent to a JSON
+#  object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "id": @id, "ts": @ts, "error": @error }.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.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.")
-      WorkflowEventError.validate_raw(obj: obj.error)
+      Vellum::WorkflowEventError.validate_raw(obj: obj.error)
     end
   end
-end
+end

data/lib/vellum_ai/types/rejected_prompt_execution_meta.rb

@@ -1,52 +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`.
+# The subset of the metadata tracked by Vellum during prompt execution that the
+#  request opted into with `expand_meta`.
   class RejectedPromptExecutionMeta
-    attr_reader :latency, :finish_reason, :additional_properties
+  # @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 [FINISH_REASON_ENUM]
+    # @param latency [Integer] 
+    # @param finish_reason [Vellum::FinishReasonEnum] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [RejectedPromptExecutionMeta]
-    def initialize(latency: nil, finish_reason: nil, additional_properties: nil)
-      # @type [Integer]
-      @latency = latency
-      # @type [FINISH_REASON_ENUM]
-      @finish_reason = finish_reason
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::RejectedPromptExecutionMeta]
+    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 RejectedPromptExecutionMeta
+# Deserialize a JSON object to an instance of RejectedPromptExecutionMeta
     #
-    # @param json_object [JSON]
-    # @return [RejectedPromptExecutionMeta]
+    # @param json_object [String] 
+    # @return [Vellum::RejectedPromptExecutionMeta]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      latency = struct.latency
-      finish_reason = FINISH_REASON_ENUM.key(parsed_json["finish_reason"]) || parsed_json["finish_reason"]
-      new(latency: latency, finish_reason: finish_reason, additional_properties: struct)
+      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 RejectedPromptExecutionMeta to a JSON object
+# Serialize an instance of RejectedPromptExecutionMeta to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "latency": @latency, "finish_reason": FINISH_REASON_ENUM[@finish_reason] || @finish_reason }.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.latency&.is_a?(Integer) != false || raise("Passed value for field obj.latency is not the expected type, validation failed.")
-      obj.finish_reason&.is_a?(FINISH_REASON_ENUM) != false || raise("Passed value for field obj.finish_reason 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
+end