vellum_ai 0.3.15 → 0.14.74

data/lib/vellum_ai/types/execute_api_request_headers_value.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+require "json"
+require_relative "vellum_secret"
+
+module Vellum
+  class ExecuteApiRequestHeadersValue
+
+
+# Deserialize a JSON object to an instance of ExecuteApiRequestHeadersValue
+    #
+    # @param json_object [String] 
+    # @return [Vellum::ExecuteApiRequestHeadersValue]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      begin
+        struct.is_a?(String) != false || raise("Passed value for field struct is not the expected type, validation failed.")
+        unless struct.nil?
+  return struct
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::VellumSecret.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::VellumSecret.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+ return struct
+    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:)
+      begin
+        return obj.is_a?(String) != false || raise("Passed value for field obj is not the expected type, validation failed.")
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::VellumSecret.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      raise("Passed value matched no type within the union, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/execute_api_response.rb

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

data/lib/vellum_ai/types/execute_api_response_json.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+require "json"
+
+module Vellum
+  class ExecuteApiResponseJson
+
+
+# Deserialize a JSON object to an instance of ExecuteApiResponseJson
+    #
+    # @param json_object [String] 
+    # @return [Vellum::ExecuteApiResponseJson]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      begin
+        struct.is_a?(Hash) != false || raise("Passed value for field struct is not the expected type, validation failed.")
+        unless struct.nil?
+  return struct
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        struct.is_a?(Array) != false || raise("Passed value for field struct is not the expected type, validation failed.")
+        unless struct.nil?
+  return struct
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+ return struct
+    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:)
+      begin
+        return obj.is_a?(Hash) != false || raise("Passed value for field obj is not the expected type, validation failed.")
+      rescue StandardError
+        # noop
+      end
+      begin
+        return obj.is_a?(Array) != false || raise("Passed value for field obj is not the expected type, validation failed.")
+      rescue StandardError
+        # noop
+      end
+      raise("Passed value matched no type within the union, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/execute_prompt_event.rb

@@ -1,5 +1,4 @@
 # frozen_string_literal: true
-
 require "json"
 require_relative "initiated_execute_prompt_event"
 require_relative "streaming_execute_prompt_event"
@@ -8,109 +7,84 @@ require_relative "rejected_execute_prompt_event"
 
 module Vellum
   class ExecutePromptEvent
-    attr_reader :member, :discriminant
 
-    private_class_method :new
-    alias kind_of? is_a?
-    # @param member [Object]
-    # @param discriminant [String]
-    # @return [ExecutePromptEvent]
-    def initialize(member:, discriminant:)
-      # @type [Object]
-      @member = member
-      # @type [String]
-      @discriminant = discriminant
-    end
 
-    # Deserialize a JSON object to an instance of ExecutePromptEvent
+# Deserialize a JSON object to an instance of ExecutePromptEvent
     #
-    # @param json_object [JSON]
-    # @return [ExecutePromptEvent]
+    # @param json_object [String] 
+    # @return [Vellum::ExecutePromptEvent]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      member = case struct.state
-               when "INITIATED"
-                 InitiatedExecutePromptEvent.from_json(json_object: json_object)
-               when "STREAMING"
-                 StreamingExecutePromptEvent.from_json(json_object: json_object)
-               when "FULFILLED"
-                 FulfilledExecutePromptEvent.from_json(json_object: json_object)
-               when "REJECTED"
-                 RejectedExecutePromptEvent.from_json(json_object: json_object)
-               else
-                 InitiatedExecutePromptEvent.from_json(json_object: json_object)
-               end
-      new(member: member, discriminant: struct.state)
-    end
-
-    # For Union Types, to_json functionality is delegated to the wrapped member.
-    #
-    # @return [JSON]
-    def to_json(*_args)
-      case @discriminant
-      when "INITIATED"
-        { **@member.to_json, state: @discriminant }.to_json
-      when "STREAMING"
-        { **@member.to_json, state: @discriminant }.to_json
-      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
+      begin
+        Vellum::InitiatedExecutePromptEvent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::InitiatedExecutePromptEvent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::StreamingExecutePromptEvent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::StreamingExecutePromptEvent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::FulfilledExecutePromptEvent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::FulfilledExecutePromptEvent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
       end
-      @member.to_json
+      begin
+        Vellum::RejectedExecutePromptEvent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::RejectedExecutePromptEvent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+ return struct
     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 "INITIATED"
-        InitiatedExecutePromptEvent.validate_raw(obj: obj)
-      when "STREAMING"
-        StreamingExecutePromptEvent.validate_raw(obj: obj)
-      when "FULFILLED"
-        FulfilledExecutePromptEvent.validate_raw(obj: obj)
-      when "REJECTED"
-        RejectedExecutePromptEvent.validate_raw(obj: obj)
-      else
-        raise("Passed value matched no type within the union, validation failed.")
+      begin
+        return Vellum::InitiatedExecutePromptEvent.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
       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 [InitiatedExecutePromptEvent]
-    # @return [ExecutePromptEvent]
-    def self.initiated(member:)
-      new(member: member, discriminant: "INITIATED")
-    end
-
-    # @param member [StreamingExecutePromptEvent]
-    # @return [ExecutePromptEvent]
-    def self.streaming(member:)
-      new(member: member, discriminant: "STREAMING")
-    end
-
-    # @param member [FulfilledExecutePromptEvent]
-    # @return [ExecutePromptEvent]
-    def self.fulfilled(member:)
-      new(member: member, discriminant: "FULFILLED")
-    end
-
-    # @param member [RejectedExecutePromptEvent]
-    # @return [ExecutePromptEvent]
-    def self.rejected(member:)
-      new(member: member, discriminant: "REJECTED")
+      begin
+        return Vellum::StreamingExecutePromptEvent.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::FulfilledExecutePromptEvent.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::RejectedExecutePromptEvent.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      raise("Passed value matched no type within the union, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/execute_prompt_response.rb

@@ -1,90 +1,58 @@
 # frozen_string_literal: true
-
 require "json"
 require_relative "fulfilled_execute_prompt_response"
 require_relative "rejected_execute_prompt_response"
 
 module Vellum
   class ExecutePromptResponse
-    attr_reader :member, :discriminant
 
-    private_class_method :new
-    alias kind_of? is_a?
-    # @param member [Object]
-    # @param discriminant [String]
-    # @return [ExecutePromptResponse]
-    def initialize(member:, discriminant:)
-      # @type [Object]
-      @member = member
-      # @type [String]
-      @discriminant = discriminant
-    end
 
-    # Deserialize a JSON object to an instance of ExecutePromptResponse
+# Deserialize a JSON object to an instance of ExecutePromptResponse
     #
-    # @param json_object [JSON]
-    # @return [ExecutePromptResponse]
+    # @param json_object [String] 
+    # @return [Vellum::ExecutePromptResponse]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      member = case struct.state
-               when "FULFILLED"
-                 FulfilledExecutePromptResponse.from_json(json_object: json_object)
-               when "REJECTED"
-                 RejectedExecutePromptResponse.from_json(json_object: json_object)
-               else
-                 FulfilledExecutePromptResponse.from_json(json_object: json_object)
-               end
-      new(member: member, discriminant: struct.state)
-    end
-
-    # For Union Types, to_json functionality is delegated to the wrapped member.
-    #
-    # @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
+      begin
+        Vellum::FulfilledExecutePromptResponse.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::FulfilledExecutePromptResponse.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::RejectedExecutePromptResponse.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::RejectedExecutePromptResponse.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
       end
-      @member.to_json
+ return struct
     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"
-        FulfilledExecutePromptResponse.validate_raw(obj: obj)
-      when "REJECTED"
-        RejectedExecutePromptResponse.validate_raw(obj: obj)
-      else
-        raise("Passed value matched no type within the union, validation failed.")
+      begin
+        return Vellum::FulfilledExecutePromptResponse.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
       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 [FulfilledExecutePromptResponse]
-    # @return [ExecutePromptResponse]
-    def self.fulfilled(member:)
-      new(member: member, discriminant: "FULFILLED")
-    end
-
-    # @param member [RejectedExecutePromptResponse]
-    # @return [ExecutePromptResponse]
-    def self.rejected(member:)
-      new(member: member, discriminant: "REJECTED")
+      begin
+        return Vellum::RejectedExecutePromptResponse.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      raise("Passed value matched no type within the union, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/execute_workflow_response.rb

@@ -1,67 +1,83 @@
 # frozen_string_literal: true
-
 require_relative "execute_workflow_workflow_result_event"
+require "ostruct"
 require "json"
 
 module Vellum
   class ExecuteWorkflowResponse
-    attr_reader :execution_id, :run_id, :external_id, :data, :additional_properties
+  # @return [String] 
+    attr_reader :execution_id
+  # @return [String] 
+    attr_reader :run_id
+  # @return [String] 
+    attr_reader :external_id
+  # @return [Vellum::ExecuteWorkflowWorkflowResultEvent] 
+    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 execution_id [String]
-    # @param run_id [String]
-    # @param external_id [String]
-    # @param data [ExecuteWorkflowWorkflowResultEvent]
+    # @param execution_id [String] 
+    # @param run_id [String] 
+    # @param external_id [String] 
+    # @param data [Vellum::ExecuteWorkflowWorkflowResultEvent] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [ExecuteWorkflowResponse]
-    def initialize(execution_id:, data:, run_id: nil, external_id: nil, additional_properties: nil)
-      # @type [String]
+    # @return [Vellum::ExecuteWorkflowResponse]
+    def initialize(execution_id:, run_id: OMIT, external_id: OMIT, data:, additional_properties: nil)
       @execution_id = execution_id
-      # @type [String]
-      @run_id = run_id
-      # @type [String]
-      @external_id = external_id
-      # @type [ExecuteWorkflowWorkflowResultEvent]
+      @run_id = run_id if run_id != OMIT
+      @external_id = external_id if external_id != OMIT
       @data = data
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "execution_id": execution_id, "run_id": run_id, "external_id": external_id, "data": data }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of ExecuteWorkflowResponse
+# Deserialize a JSON object to an instance of ExecuteWorkflowResponse
     #
-    # @param json_object [JSON]
-    # @return [ExecuteWorkflowResponse]
+    # @param json_object [String] 
+    # @return [Vellum::ExecuteWorkflowResponse]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      execution_id = struct.execution_id
-      run_id = struct.run_id
-      external_id = struct.external_id
-      if parsed_json["data"].nil?
-        data = nil
-      else
+      execution_id = parsed_json["execution_id"]
+      run_id = parsed_json["run_id"]
+      external_id = parsed_json["external_id"]
+      unless parsed_json["data"].nil?
         data = parsed_json["data"].to_json
-        data = ExecuteWorkflowWorkflowResultEvent.from_json(json_object: data)
+        data = Vellum::ExecuteWorkflowWorkflowResultEvent.from_json(json_object: data)
+      else
+        data = nil
       end
-      new(execution_id: execution_id, run_id: run_id, external_id: external_id, data: data,
-          additional_properties: struct)
+      new(
+        execution_id: execution_id,
+        run_id: run_id,
+        external_id: external_id,
+        data: data,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of ExecuteWorkflowResponse to a JSON object
+# Serialize an instance of ExecuteWorkflowResponse to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "execution_id": @execution_id, "run_id": @run_id, "external_id": @external_id, "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:)
       obj.execution_id.is_a?(String) != false || raise("Passed value for field obj.execution_id is not the expected type, validation failed.")
       obj.run_id&.is_a?(String) != false || raise("Passed value for field obj.run_id is not the expected type, validation failed.")
       obj.external_id&.is_a?(String) != false || raise("Passed value for field obj.external_id is not the expected type, validation failed.")
-      ExecuteWorkflowWorkflowResultEvent.validate_raw(obj: obj.data)
+      Vellum::ExecuteWorkflowWorkflowResultEvent.validate_raw(obj: obj.data)
     end
   end
-end
+end