vellum_ai 0.3.11 → 0.3.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a34112cfe303a361bb71b349a1b752f9af3a0aeab320d337423fff77fb2b2cac
-  data.tar.gz: 3de6e6dc8f28729679ae89cb99a7bd210520457564910e499c0ebdddd938a219
+  metadata.gz: ea4b18359c0ea4e08dbd7be1adb49eba513b4160627f89378704480d9d7dae3e
+  data.tar.gz: 7d1d699a392e57dee0a9dbcf84396c55def4b2e03e6fc53ed9076d2057733670
 SHA512:
-  metadata.gz: f728fd9611386219ed4871bd39a5fba81166c72f04e42d4c877f1f5e4539d5178d0f4b1a195b5090cbba8091e49fb4985dab61903ec5bbd6c89f0fc2bdac0bca
-  data.tar.gz: 6d9034005bbdfa8a321ddd061a6aacbfa64ab2421506069726eb56ca6bf07eef593ef55903aa354fe1b4e9fc1e72578f93700ace1c547374992738115a461fdb
+  metadata.gz: 8046d5a9c7f8073f96c6dc3d6c95da1d841617972dd0e7c52f2f3ccd8d4d66ec90ae5e0b5bea781434cba60b2745945fc9fc1911ed49d43f957f1ade65829a31
+  data.tar.gz: 830eb6ddb227723925dd888a40cc2a1e274df0fb63359b53c4d6634589f9f37b08b2bf9c2a9277fc13d2a58155b07ab81507ac7f0bb7b7e6f23fa6b4a57223cc

data/lib/requests.rb

@@ -20,7 +20,7 @@ module Vellum
       @headers = {
         "X-Fern-Language": "Ruby",
         "X-Fern-SDK-Name": "Vellum",
-        "X-Fern-SDK-Version": "0.3.11",
+        "X-Fern-SDK-Version": "0.3.13",
         "X_API_KEY": api_key.to_s
       }
       @conn = Faraday.new(headers: @headers) do |faraday|
@@ -46,7 +46,7 @@ module Vellum
       @headers = {
         "X-Fern-Language": "Ruby",
         "X-Fern-SDK-Name": "Vellum",
-        "X-Fern-SDK-Version": "0.3.11",
+        "X-Fern-SDK-Version": "0.3.13",
         "X_API_KEY": api_key.to_s
       }
       @conn = Faraday.new(headers: @headers) do |faraday|

data/lib/types_export.rb

@@ -86,6 +86,7 @@ require_relative "vellum_ai/types/generate_stream_result_data"
 require_relative "vellum_ai/types/image_chat_message_content"
 require_relative "vellum_ai/types/image_chat_message_content_request"
 require_relative "vellum_ai/types/image_enum"
+require_relative "vellum_ai/types/image_variable_value"
 require_relative "vellum_ai/types/indexing_state_enum"
 require_relative "vellum_ai/types/initiated_enum"
 require_relative "vellum_ai/types/initiated_execute_prompt_event"
@@ -137,6 +138,7 @@ require_relative "vellum_ai/types/paginated_document_index_read_list"
 require_relative "vellum_ai/types/paginated_slim_deployment_read_list"
 require_relative "vellum_ai/types/paginated_slim_document_list"
 require_relative "vellum_ai/types/paginated_slim_workflow_deployment_list"
+require_relative "vellum_ai/types/paginated_test_suite_run_execution_list"
 require_relative "vellum_ai/types/processing_failure_reason_enum"
 require_relative "vellum_ai/types/processing_state_enum"
 require_relative "vellum_ai/types/prompt_deployment_expand_meta_request_request"
@@ -232,6 +234,35 @@ require_relative "vellum_ai/types/test_case_number_variable_value"
 require_relative "vellum_ai/types/test_case_search_results_variable_value"
 require_relative "vellum_ai/types/test_case_string_variable_value"
 require_relative "vellum_ai/types/test_case_variable_value"
+require_relative "vellum_ai/types/test_suite_run_deployment_release_tag_exec_config"
+require_relative "vellum_ai/types/test_suite_run_deployment_release_tag_exec_config_data"
+require_relative "vellum_ai/types/test_suite_run_deployment_release_tag_exec_config_data_request"
+require_relative "vellum_ai/types/test_suite_run_deployment_release_tag_exec_config_request"
+require_relative "vellum_ai/types/test_suite_run_deployment_release_tag_exec_config_type_enum"
+require_relative "vellum_ai/types/test_suite_run_exec_config"
+require_relative "vellum_ai/types/test_suite_run_exec_config_request"
+require_relative "vellum_ai/types/test_suite_run_execution"
+require_relative "vellum_ai/types/test_suite_run_execution_chat_history_output"
+require_relative "vellum_ai/types/test_suite_run_execution_error_output"
+require_relative "vellum_ai/types/test_suite_run_execution_json_output"
+require_relative "vellum_ai/types/test_suite_run_execution_metric_result"
+require_relative "vellum_ai/types/test_suite_run_execution_number_output"
+require_relative "vellum_ai/types/test_suite_run_execution_output"
+require_relative "vellum_ai/types/test_suite_run_execution_search_results_output"
+require_relative "vellum_ai/types/test_suite_run_execution_string_output"
+require_relative "vellum_ai/types/test_suite_run_metric_error_output"
+require_relative "vellum_ai/types/test_suite_run_metric_error_output_type_enum"
+require_relative "vellum_ai/types/test_suite_run_metric_number_output"
+require_relative "vellum_ai/types/test_suite_run_metric_number_output_type_enum"
+require_relative "vellum_ai/types/test_suite_run_metric_output"
+require_relative "vellum_ai/types/test_suite_run_read"
+require_relative "vellum_ai/types/test_suite_run_state"
+require_relative "vellum_ai/types/test_suite_run_test_suite"
+require_relative "vellum_ai/types/test_suite_run_workflow_release_tag_exec_config"
+require_relative "vellum_ai/types/test_suite_run_workflow_release_tag_exec_config_data"
+require_relative "vellum_ai/types/test_suite_run_workflow_release_tag_exec_config_data_request"
+require_relative "vellum_ai/types/test_suite_run_workflow_release_tag_exec_config_request"
+require_relative "vellum_ai/types/test_suite_run_workflow_release_tag_exec_config_type_enum"
 require_relative "vellum_ai/types/test_suite_test_case"
 require_relative "vellum_ai/types/upload_document_error_response"
 require_relative "vellum_ai/types/upload_document_response"

data/lib/vellum_ai/document_indexes/client.rb

@@ -95,6 +95,78 @@ module Vellum
       end
       DocumentIndexRead.from_json(json_object: response.body)
     end
+
+    # Used to fully update a Document Index given its ID.
+    #
+    # @param id [String] A UUID string identifying this document index.
+    # @param label [String] A human-readable label for the document index
+    # @param status [ENTITY_STATUS] The current status of the document index
+    #   * `ACTIVE` - Active
+    #   * `ARCHIVED` - Archived
+    # @param environment [ENVIRONMENT_ENUM] The environment this document index is used in
+    #   * `DEVELOPMENT` - Development
+    #   * `STAGING` - Staging
+    #   * `PRODUCTION` - Production
+    # @param request_options [RequestOptions]
+    # @return [DocumentIndexRead]
+    def update(id:, label:, status: nil, environment: nil, request_options: nil)
+      response = @request_client.conn.put do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.body = {
+          **(request_options&.additional_body_parameters || {}),
+          label: label,
+          status: status,
+          environment: environment
+        }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/document-indexes/#{id}"
+      end
+      DocumentIndexRead.from_json(json_object: response.body)
+    end
+
+    # Used to delete a Document Index given its ID.
+    #
+    # @param id [String] A UUID string identifying this document index.
+    # @param request_options [RequestOptions]
+    # @return [Void]
+    def destroy(id:, request_options: nil)
+      @request_client.conn.delete do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/document-indexes/#{id}"
+      end
+    end
+
+    # Used to partial update a Document Index given its ID.
+    #
+    # @param id [String] A UUID string identifying this document index.
+    # @param label [String] A human-readable label for the document index
+    # @param status [ENTITY_STATUS] The current status of the document index
+    #   * `ACTIVE` - Active
+    #   * `ARCHIVED` - Archived
+    # @param environment [ENVIRONMENT_ENUM] The environment this document index is used in
+    #   * `DEVELOPMENT` - Development
+    #   * `STAGING` - Staging
+    #   * `PRODUCTION` - Production
+    # @param request_options [RequestOptions]
+    # @return [DocumentIndexRead]
+    def partial_update(id:, label: nil, status: nil, environment: nil, request_options: nil)
+      response = @request_client.conn.patch do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.body = {
+          **(request_options&.additional_body_parameters || {}),
+          label: label,
+          status: status,
+          environment: environment
+        }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/document-indexes/#{id}"
+      end
+      DocumentIndexRead.from_json(json_object: response.body)
+    end
   end
 
   class AsyncDocumentIndexesClient
@@ -189,5 +261,83 @@ module Vellum
         DocumentIndexRead.from_json(json_object: response.body)
       end
     end
+
+    # Used to fully update a Document Index given its ID.
+    #
+    # @param id [String] A UUID string identifying this document index.
+    # @param label [String] A human-readable label for the document index
+    # @param status [ENTITY_STATUS] The current status of the document index
+    #   * `ACTIVE` - Active
+    #   * `ARCHIVED` - Archived
+    # @param environment [ENVIRONMENT_ENUM] The environment this document index is used in
+    #   * `DEVELOPMENT` - Development
+    #   * `STAGING` - Staging
+    #   * `PRODUCTION` - Production
+    # @param request_options [RequestOptions]
+    # @return [DocumentIndexRead]
+    def update(id:, label:, status: nil, environment: nil, request_options: nil)
+      Async do
+        response = @request_client.conn.put do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.body = {
+            **(request_options&.additional_body_parameters || {}),
+            label: label,
+            status: status,
+            environment: environment
+          }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/document-indexes/#{id}"
+        end
+        DocumentIndexRead.from_json(json_object: response.body)
+      end
+    end
+
+    # Used to delete a Document Index given its ID.
+    #
+    # @param id [String] A UUID string identifying this document index.
+    # @param request_options [RequestOptions]
+    # @return [Void]
+    def destroy(id:, request_options: nil)
+      Async do
+        @request_client.conn.delete do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/document-indexes/#{id}"
+        end
+      end
+    end
+
+    # Used to partial update a Document Index given its ID.
+    #
+    # @param id [String] A UUID string identifying this document index.
+    # @param label [String] A human-readable label for the document index
+    # @param status [ENTITY_STATUS] The current status of the document index
+    #   * `ACTIVE` - Active
+    #   * `ARCHIVED` - Archived
+    # @param environment [ENVIRONMENT_ENUM] The environment this document index is used in
+    #   * `DEVELOPMENT` - Development
+    #   * `STAGING` - Staging
+    #   * `PRODUCTION` - Production
+    # @param request_options [RequestOptions]
+    # @return [DocumentIndexRead]
+    def partial_update(id:, label: nil, status: nil, environment: nil, request_options: nil)
+      Async do
+        response = @request_client.conn.patch do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.body = {
+            **(request_options&.additional_body_parameters || {}),
+            label: label,
+            status: status,
+            environment: environment
+          }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/document-indexes/#{id}"
+        end
+        DocumentIndexRead.from_json(json_object: response.body)
+      end
+    end
   end
 end

data/lib/vellum_ai/test_suite_runs/client.rb

@@ -0,0 +1,149 @@
+# frozen_string_literal: true
+
+require_relative "../../requests"
+require_relative "../types/test_suite_run_exec_config_request"
+require_relative "../types/test_suite_run_read"
+require_relative "../types/paginated_test_suite_run_execution_list"
+require "async"
+
+module Vellum
+  class TestSuiteRunsClient
+    attr_reader :request_client
+
+    # @param request_client [RequestClient]
+    # @return [TestSuiteRunsClient]
+    def initialize(request_client:)
+      # @type [RequestClient]
+      @request_client = request_client
+    end
+
+    # Trigger a Test Suite and create a new Test Suite Run
+    #
+    # @param test_suite_id [String] The ID of the Test Suite to run
+    # @param exec_config [Hash] Configuration that defines how the Test Suite should be runRequest of type TestSuiteRunExecConfigRequest, as a Hash
+    # @param request_options [RequestOptions]
+    # @return [TestSuiteRunRead]
+    def create(exec_config:, test_suite_id: nil, request_options: nil)
+      response = @request_client.conn.post do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.body = {
+          **(request_options&.additional_body_parameters || {}),
+          test_suite_id: test_suite_id,
+          exec_config: exec_config
+        }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/test-suite-runs"
+      end
+      TestSuiteRunRead.from_json(json_object: response.body)
+    end
+
+    # Retrieve a specific Test Suite Run by ID
+    #
+    # @param id [String] A UUID string identifying this test suite run.
+    # @param request_options [RequestOptions]
+    # @return [TestSuiteRunRead]
+    def retrieve(id:, request_options: nil)
+      response = @request_client.conn.get do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/test-suite-runs/#{id}"
+      end
+      TestSuiteRunRead.from_json(json_object: response.body)
+    end
+
+    # @param id [String] A UUID string identifying this test suite run.
+    # @param limit [Integer] Number of results to return per page.
+    # @param offset [Integer] The initial index from which to return the results.
+    # @param request_options [RequestOptions]
+    # @return [PaginatedTestSuiteRunExecutionList]
+    def list_test_suite_run_executions(id:, limit: nil, offset: nil, request_options: nil)
+      response = @request_client.conn.get do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.params = {
+          **(request_options&.additional_query_parameters || {}),
+          "limit": limit,
+          "offset": offset
+        }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/test-suite-runs/#{id}/executions"
+      end
+      PaginatedTestSuiteRunExecutionList.from_json(json_object: response.body)
+    end
+  end
+
+  class AsyncTestSuiteRunsClient
+    attr_reader :request_client
+
+    # @param request_client [AsyncRequestClient]
+    # @return [AsyncTestSuiteRunsClient]
+    def initialize(request_client:)
+      # @type [AsyncRequestClient]
+      @request_client = request_client
+    end
+
+    # Trigger a Test Suite and create a new Test Suite Run
+    #
+    # @param test_suite_id [String] The ID of the Test Suite to run
+    # @param exec_config [Hash] Configuration that defines how the Test Suite should be runRequest of type TestSuiteRunExecConfigRequest, as a Hash
+    # @param request_options [RequestOptions]
+    # @return [TestSuiteRunRead]
+    def create(exec_config:, test_suite_id: nil, request_options: nil)
+      Async do
+        response = @request_client.conn.post do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.body = {
+            **(request_options&.additional_body_parameters || {}),
+            test_suite_id: test_suite_id,
+            exec_config: exec_config
+          }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/test-suite-runs"
+        end
+        TestSuiteRunRead.from_json(json_object: response.body)
+      end
+    end
+
+    # Retrieve a specific Test Suite Run by ID
+    #
+    # @param id [String] A UUID string identifying this test suite run.
+    # @param request_options [RequestOptions]
+    # @return [TestSuiteRunRead]
+    def retrieve(id:, request_options: nil)
+      Async do
+        response = @request_client.conn.get do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/test-suite-runs/#{id}"
+        end
+        TestSuiteRunRead.from_json(json_object: response.body)
+      end
+    end
+
+    # @param id [String] A UUID string identifying this test suite run.
+    # @param limit [Integer] Number of results to return per page.
+    # @param offset [Integer] The initial index from which to return the results.
+    # @param request_options [RequestOptions]
+    # @return [PaginatedTestSuiteRunExecutionList]
+    def list_test_suite_run_executions(id:, limit: nil, offset: nil, request_options: nil)
+      Async do
+        response = @request_client.conn.get do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.params = {
+            **(request_options&.additional_query_parameters || {}),
+            "limit": limit,
+            "offset": offset
+          }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/test-suite-runs/#{id}/executions"
+        end
+        PaginatedTestSuiteRunExecutionList.from_json(json_object: response.body)
+      end
+    end
+  end
+end

data/lib/vellum_ai/types/array_variable_value_item.rb

@@ -8,6 +8,7 @@ require_relative "chat_history_variable_value"
 require_relative "search_results_variable_value"
 require_relative "error_variable_value"
 require_relative "function_call_variable_value"
+require_relative "image_variable_value"
 
 module Vellum
   class ArrayVariableValueItem
@@ -46,6 +47,8 @@ module Vellum
                  ErrorVariableValue.from_json(json_object: json_object)
                when "FUNCTION_CALL"
                  FunctionCallVariableValue.from_json(json_object: json_object)
+               when "IMAGE"
+                 ImageVariableValue.from_json(json_object: json_object)
                else
                  StringVariableValue.from_json(json_object: json_object)
                end
@@ -71,6 +74,8 @@ module Vellum
         { **@member.to_json, type: @discriminant }.to_json
       when "FUNCTION_CALL"
         { **@member.to_json, type: @discriminant }.to_json
+      when "IMAGE"
+        { **@member.to_json, type: @discriminant }.to_json
       else
         { "type": @discriminant, value: @member }.to_json
       end
@@ -97,6 +102,8 @@ module Vellum
         ErrorVariableValue.validate_raw(obj: obj)
       when "FUNCTION_CALL"
         FunctionCallVariableValue.validate_raw(obj: obj)
+      when "IMAGE"
+        ImageVariableValue.validate_raw(obj: obj)
       else
         raise("Passed value matched no type within the union, validation failed.")
       end
@@ -151,5 +158,11 @@ module Vellum
     def self.function_call(member:)
       new(member: member, discriminant: "FUNCTION_CALL")
     end
+
+    # @param member [ImageVariableValue]
+    # @return [ArrayVariableValueItem]
+    def self.image(member:)
+      new(member: member, discriminant: "IMAGE")
+    end
   end
 end

data/lib/vellum_ai/types/fulfilled_workflow_node_result_event.rb

@@ -8,17 +8,21 @@ require "json"
 module Vellum
   # An event that indicates that the node has fulfilled its execution.
   class FulfilledWorkflowNodeResultEvent
-    attr_reader :id, :node_id, :node_result_id, :ts, :data, :output_values, :additional_properties
+    attr_reader :id, :node_id, :node_result_id, :ts, :data, :source_execution_id, :output_values, :mocked,
+                :additional_properties
 
     # @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 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, output_values: nil, additional_properties: nil)
+    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]
       @id = id
       # @type [String]
@@ -29,8 +33,12 @@ module Vellum
       @ts = ts
       # @type [WorkflowNodeResultData]
       @data = data
+      # @type [String]
+      @source_execution_id = source_execution_id
       # @type [Array<NodeOutputCompiledValue>]
       @output_values = output_values
+      # @type [Boolean]
+      @mocked = mocked
       # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
     end
@@ -52,12 +60,14 @@ module Vellum
         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)
       end
-      new(id: id, node_id: node_id, node_result_id: node_result_id, ts: ts, data: data, output_values: output_values,
-          additional_properties: struct)
+      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)
     end
 
     # Serialize an instance of FulfilledWorkflowNodeResultEvent to a JSON object
@@ -70,7 +80,9 @@ module Vellum
         "node_result_id": @node_result_id,
         "ts": @ts,
         "data": @data,
-        "output_values": @output_values
+        "source_execution_id": @source_execution_id,
+        "output_values": @output_values,
+        "mocked": @mocked
       }.to_json
     end
 
@@ -84,7 +96,9 @@ module Vellum
       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.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.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.mocked&.is_a?(Boolean) != false || raise("Passed value for field obj.mocked is not the expected type, validation failed.")
     end
   end
 end

data/lib/vellum_ai/types/generate_request.rb

@@ -8,14 +8,14 @@ module Vellum
     attr_reader :input_values, :chat_history, :external_ids, :additional_properties
 
     # @param input_values [Hash{String => String}] Key/value pairs for each template variable defined in the deployment's prompt.
-    # @param chat_history [Array<ChatMessageRequest>] Optionally provide a list of chat messages that'll be used in place of the special {$chat_history} variable, if included in the prompt.
+    # @param chat_history [Array<ChatMessageRequest>] Optionally provide a list of chat messages that'll be used in place of the special chat_history variable, if included in the prompt.
     # @param external_ids [Array<String>] Optionally include a unique identifier for each generation, as represented outside of Vellum. Note that this should generally be a list of length one.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [GenerateRequest]
     def initialize(input_values:, chat_history: nil, external_ids: nil, additional_properties: nil)
       # @type [Hash{String => String}] Key/value pairs for each template variable defined in the deployment's prompt.
       @input_values = input_values
-      # @type [Array<ChatMessageRequest>] Optionally provide a list of chat messages that'll be used in place of the special {$chat_history} variable, if included in the prompt.
+      # @type [Array<ChatMessageRequest>] Optionally provide a list of chat messages that'll be used in place of the special chat_history variable, if included in the prompt.
       @chat_history = chat_history
       # @type [Array<String>] Optionally include a unique identifier for each generation, as represented outside of Vellum. Note that this should generally be a list of length one.
       @external_ids = external_ids

data/lib/vellum_ai/types/image_variable_value.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require_relative "vellum_image"
+require "json"
+
+module Vellum
+  # A base Vellum primitive value representing an image.
+  class ImageVariableValue
+    attr_reader :value, :additional_properties
+
+    # @param value [VellumImage]
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [ImageVariableValue]
+    def initialize(value: nil, additional_properties: nil)
+      # @type [VellumImage]
+      @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 ImageVariableValue
+    #
+    # @param json_object [JSON]
+    # @return [ImageVariableValue]
+    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
+        value = parsed_json["value"].to_json
+        value = VellumImage.from_json(json_object: value)
+      end
+      new(value: value, additional_properties: struct)
+    end
+
+    # Serialize an instance of ImageVariableValue to a JSON object
+    #
+    # @return [JSON]
+    def to_json(*_args)
+      { "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.value.nil? || VellumImage.validate_raw(obj: obj.value)
+    end
+  end
+end

data/lib/vellum_ai/types/initiated_workflow_node_result_event.rb

@@ -8,17 +8,19 @@ require "json"
 module Vellum
   # An event that indicates that the node has initiated its execution.
   class InitiatedWorkflowNodeResultEvent
-    attr_reader :id, :node_id, :node_result_id, :ts, :data, :input_values, :additional_properties
+    attr_reader :id, :node_id, :node_result_id, :ts, :data, :source_execution_id, :input_values, :additional_properties
 
     # @param id [String]
     # @param node_id [String]
     # @param node_result_id [String]
     # @param ts [DateTime]
     # @param data [WorkflowNodeResultData]
+    # @param source_execution_id [String]
     # @param input_values [Array<NodeInputVariableCompiledValue>]
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [InitiatedWorkflowNodeResultEvent]
-    def initialize(id:, node_id:, node_result_id:, ts: nil, data: nil, input_values: nil, additional_properties: nil)
+    def initialize(id:, node_id:, node_result_id:, ts: nil, data: nil, source_execution_id: nil, input_values: nil,
+                   additional_properties: nil)
       # @type [String]
       @id = id
       # @type [String]
@@ -29,6 +31,8 @@ module Vellum
       @ts = ts
       # @type [WorkflowNodeResultData]
       @data = data
+      # @type [String]
+      @source_execution_id = source_execution_id
       # @type [Array<NodeInputVariableCompiledValue>]
       @input_values = input_values
       # @type [OpenStruct] Additional properties unmapped to the current class definition
@@ -52,12 +56,13 @@ module Vellum
         data = parsed_json["data"].to_json
         data = WorkflowNodeResultData.from_json(json_object: data)
       end
+      source_execution_id = struct.source_execution_id
       input_values = parsed_json["input_values"].map do |v|
         v = v.to_json
         NodeInputVariableCompiledValue.from_json(json_object: v)
       end
-      new(id: id, node_id: node_id, node_result_id: node_result_id, ts: ts, data: data, input_values: input_values,
-          additional_properties: struct)
+      new(id: id, node_id: node_id, node_result_id: node_result_id, ts: ts, data: data,
+          source_execution_id: source_execution_id, input_values: input_values, additional_properties: struct)
     end
 
     # Serialize an instance of InitiatedWorkflowNodeResultEvent to a JSON object
@@ -70,6 +75,7 @@ module Vellum
         "node_result_id": @node_result_id,
         "ts": @ts,
         "data": @data,
+        "source_execution_id": @source_execution_id,
         "input_values": @input_values
       }.to_json
     end
@@ -84,6 +90,7 @@ module Vellum
       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.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.source_execution_id&.is_a?(String) != false || raise("Passed value for field obj.source_execution_id is not the expected type, validation failed.")
       obj.input_values&.is_a?(Array) != false || raise("Passed value for field obj.input_values is not the expected type, validation failed.")
     end
   end