vellum_ai 0.3.10 → 0.3.12

data/lib/vellum_ai/types/prompt_template_block_properties_request.rb

@@ -7,11 +7,12 @@ require "json"
 
 module Vellum
   class PromptTemplateBlockPropertiesRequest
-    attr_reader :chat_role, :chat_message_unterminated, :template, :template_type, :function_name,
+    attr_reader :chat_role, :chat_message_unterminated, :chat_source, :template, :template_type, :function_name,
                 :function_description, :function_parameters, :function_forced, :blocks, :additional_properties
 
     # @param chat_role [CHAT_MESSAGE_ROLE]
     # @param chat_message_unterminated [Boolean]
+    # @param chat_source [String]
     # @param template [String]
     # @param template_type [VELLUM_VARIABLE_TYPE]
     # @param function_name [String]
@@ -21,13 +22,15 @@ module Vellum
     # @param blocks [Array<PromptTemplateBlockRequest>]
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [PromptTemplateBlockPropertiesRequest]
-    def initialize(chat_role: nil, chat_message_unterminated: nil, template: nil, template_type: nil,
+    def initialize(chat_role: nil, chat_message_unterminated: nil, chat_source: nil, template: nil, template_type: nil,
                    function_name: nil, function_description: nil, function_parameters: nil, function_forced: nil, blocks: nil, additional_properties: nil)
       # @type [CHAT_MESSAGE_ROLE]
       @chat_role = chat_role
       # @type [Boolean]
       @chat_message_unterminated = chat_message_unterminated
       # @type [String]
+      @chat_source = chat_source
+      # @type [String]
       @template = template
       # @type [VELLUM_VARIABLE_TYPE]
       @template_type = template_type
@@ -54,6 +57,7 @@ module Vellum
       parsed_json = JSON.parse(json_object)
       chat_role = CHAT_MESSAGE_ROLE.key(parsed_json["chat_role"]) || parsed_json["chat_role"]
       chat_message_unterminated = struct.chat_message_unterminated
+      chat_source = struct.chat_source
       template = struct.template
       template_type = VELLUM_VARIABLE_TYPE.key(parsed_json["template_type"]) || parsed_json["template_type"]
       function_name = struct.function_name
@@ -64,8 +68,8 @@ module Vellum
         v = v.to_json
         PromptTemplateBlockRequest.from_json(json_object: v)
       end
-      new(chat_role: chat_role, chat_message_unterminated: chat_message_unterminated, template: template,
-          template_type: template_type, function_name: function_name, function_description: function_description, function_parameters: function_parameters, function_forced: function_forced, blocks: blocks, additional_properties: struct)
+      new(chat_role: chat_role, chat_message_unterminated: chat_message_unterminated, chat_source: chat_source,
+          template: template, template_type: template_type, function_name: function_name, function_description: function_description, function_parameters: function_parameters, function_forced: function_forced, blocks: blocks, additional_properties: struct)
     end
 
     # Serialize an instance of PromptTemplateBlockPropertiesRequest to a JSON object
@@ -75,6 +79,7 @@ module Vellum
       {
         "chat_role": CHAT_MESSAGE_ROLE[@chat_role] || @chat_role,
         "chat_message_unterminated": @chat_message_unterminated,
+        "chat_source": @chat_source,
         "template": @template,
         "template_type": VELLUM_VARIABLE_TYPE[@template_type] || @template_type,
         "function_name": @function_name,
@@ -92,6 +97,7 @@ module Vellum
     def self.validate_raw(obj:)
       obj.chat_role&.is_a?(CHAT_MESSAGE_ROLE) != false || raise("Passed value for field obj.chat_role is not the expected type, validation failed.")
       obj.chat_message_unterminated&.is_a?(Boolean) != false || raise("Passed value for field obj.chat_message_unterminated is not the expected type, validation failed.")
+      obj.chat_source&.is_a?(String) != false || raise("Passed value for field obj.chat_source is not the expected type, validation failed.")
       obj.template&.is_a?(String) != false || raise("Passed value for field obj.template is not the expected type, validation failed.")
       obj.template_type&.is_a?(VELLUM_VARIABLE_TYPE) != false || raise("Passed value for field obj.template_type is not the expected type, validation failed.")
       obj.function_name&.is_a?(String) != false || raise("Passed value for field obj.function_name is not the expected type, validation failed.")

data/lib/vellum_ai/types/rejected_workflow_node_result_event.rb

@@ -8,17 +8,19 @@ require "json"
 module Vellum
   # An event that indicates that the node has rejected its execution.
   class RejectedWorkflowNodeResultEvent
-    attr_reader :id, :node_id, :node_result_id, :ts, :data, :error, :additional_properties
+    attr_reader :id, :node_id, :node_result_id, :ts, :data, :source_execution_id, :error, :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 error [WorkflowEventError]
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [RejectedWorkflowNodeResultEvent]
-    def initialize(id:, node_id:, node_result_id:, error:, ts: nil, data: nil, additional_properties: nil)
+    def initialize(id:, node_id:, node_result_id:, error:, ts: nil, data: nil, source_execution_id: 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 [WorkflowEventError]
       @error = error
       # @type [OpenStruct] Additional properties unmapped to the current class definition
@@ -52,14 +56,15 @@ module Vellum
         data = parsed_json["data"].to_json
         data = WorkflowNodeResultData.from_json(json_object: data)
       end
+      source_execution_id = struct.source_execution_id
       if parsed_json["error"].nil?
         error = nil
       else
         error = parsed_json["error"].to_json
         error = WorkflowEventError.from_json(json_object: error)
       end
-      new(id: id, node_id: node_id, node_result_id: node_result_id, ts: ts, data: data, error: error,
-          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, error: error, additional_properties: struct)
     end
 
     # Serialize an instance of RejectedWorkflowNodeResultEvent to a JSON object
@@ -72,6 +77,7 @@ module Vellum
         "node_result_id": @node_result_id,
         "ts": @ts,
         "data": @data,
+        "source_execution_id": @source_execution_id,
         "error": @error
       }.to_json
     end
@@ -86,6 +92,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.")
       WorkflowEventError.validate_raw(obj: obj.error)
     end
   end

data/lib/vellum_ai/types/streaming_workflow_node_result_event.rb

@@ -8,19 +8,21 @@ require "json"
 module Vellum
   # An event that indicates that the node has execution is in progress.
   class StreamingWorkflowNodeResultEvent
-    attr_reader :id, :node_id, :node_result_id, :ts, :data, :output, :output_index, :additional_properties
+    attr_reader :id, :node_id, :node_result_id, :ts, :data, :source_execution_id, :output, :output_index,
+                :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 [NodeOutputCompiledValue]
     # @param output_index [Integer]
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [StreamingWorkflowNodeResultEvent]
-    def initialize(id:, node_id:, node_result_id:, ts: nil, data: nil, output: nil, output_index: nil,
-                   additional_properties: nil)
+    def initialize(id:, node_id:, node_result_id:, ts: nil, data: nil, source_execution_id: nil, output: nil,
+                   output_index: nil, additional_properties: nil)
       # @type [String]
       @id = id
       # @type [String]
@@ -31,6 +33,8 @@ module Vellum
       @ts = ts
       # @type [WorkflowNodeResultData]
       @data = data
+      # @type [String]
+      @source_execution_id = source_execution_id
       # @type [NodeOutputCompiledValue]
       @output = output
       # @type [Integer]
@@ -56,6 +60,7 @@ module Vellum
         data = parsed_json["data"].to_json
         data = WorkflowNodeResultData.from_json(json_object: data)
       end
+      source_execution_id = struct.source_execution_id
       if parsed_json["output"].nil?
         output = nil
       else
@@ -63,8 +68,8 @@ module Vellum
         output = NodeOutputCompiledValue.from_json(json_object: output)
       end
       output_index = struct.output_index
-      new(id: id, node_id: node_id, node_result_id: node_result_id, ts: ts, data: data, output: output,
-          output_index: output_index, 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, output: output, output_index: output_index, additional_properties: struct)
     end
 
     # Serialize an instance of StreamingWorkflowNodeResultEvent to a JSON object
@@ -77,6 +82,7 @@ module Vellum
         "node_result_id": @node_result_id,
         "ts": @ts,
         "data": @data,
+        "source_execution_id": @source_execution_id,
         "output": @output,
         "output_index": @output_index
       }.to_json
@@ -92,6 +98,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.output.nil? || NodeOutputCompiledValue.validate_raw(obj: obj.output)
       obj.output_index&.is_a?(Integer) != false || raise("Passed value for field obj.output_index is not the expected type, validation failed.")
     end

data/lib/vellum_ai/types/workflow_execution_event_error_code.rb

@@ -4,6 +4,7 @@ module Vellum
   # @type [WORKFLOW_EXECUTION_EVENT_ERROR_CODE]
   WORKFLOW_EXECUTION_EVENT_ERROR_CODE = {
     workflow_initialization: "WORKFLOW_INITIALIZATION",
+    workflow_cancelled: "WORKFLOW_CANCELLED",
     node_execution_count_limit_reached: "NODE_EXECUTION_COUNT_LIMIT_REACHED",
     internal_server_error: "INTERNAL_SERVER_ERROR",
     node_execution: "NODE_EXECUTION",

data/lib/vellum_ai/types/workflow_result_event.rb

@@ -5,11 +5,12 @@ require "date"
 require_relative "workflow_result_event_output_data"
 require_relative "workflow_event_error"
 require_relative "workflow_output"
+require_relative "execution_vellum_value"
 require "json"
 
 module Vellum
   class WorkflowResultEvent
-    attr_reader :id, :state, :ts, :output, :error, :outputs, :additional_properties
+    attr_reader :id, :state, :ts, :output, :error, :outputs, :inputs, :additional_properties
 
     # @param id [String]
     # @param state [WORKFLOW_NODE_RESULT_EVENT_STATE]
@@ -17,9 +18,10 @@ module Vellum
     # @param output [WorkflowResultEventOutputData]
     # @param error [WorkflowEventError]
     # @param outputs [Array<WorkflowOutput>]
+    # @param inputs [Array<ExecutionVellumValue>]
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [WorkflowResultEvent]
-    def initialize(id:, state:, ts:, output: nil, error: nil, outputs: nil, additional_properties: nil)
+    def initialize(id:, state:, ts:, output: nil, error: nil, outputs: nil, inputs: nil, additional_properties: nil)
       # @type [String]
       @id = id
       # @type [WORKFLOW_NODE_RESULT_EVENT_STATE]
@@ -32,6 +34,8 @@ module Vellum
       @error = error
       # @type [Array<WorkflowOutput>]
       @outputs = outputs
+      # @type [Array<ExecutionVellumValue>]
+      @inputs = inputs
       # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
     end
@@ -62,7 +66,12 @@ module Vellum
         v = v.to_json
         WorkflowOutput.from_json(json_object: v)
       end
-      new(id: id, state: state, ts: ts, output: output, error: error, outputs: outputs, additional_properties: struct)
+      inputs = parsed_json["inputs"].map do |v|
+        v = v.to_json
+        ExecutionVellumValue.from_json(json_object: v)
+      end
+      new(id: id, state: state, ts: ts, output: output, error: error, outputs: outputs, inputs: inputs,
+          additional_properties: struct)
     end
 
     # Serialize an instance of WorkflowResultEvent to a JSON object
@@ -75,7 +84,8 @@ module Vellum
         "ts": @ts,
         "output": @output,
         "error": @error,
-        "outputs": @outputs
+        "outputs": @outputs,
+        "inputs": @inputs
       }.to_json
     end
 
@@ -90,6 +100,7 @@ module Vellum
       obj.output.nil? || WorkflowResultEventOutputData.validate_raw(obj: obj.output)
       obj.error.nil? || WorkflowEventError.validate_raw(obj: obj.error)
       obj.outputs&.is_a?(Array) != false || raise("Passed value for field obj.outputs is not the expected type, validation failed.")
+      obj.inputs&.is_a?(Array) != false || raise("Passed value for field obj.inputs is not the expected type, validation failed.")
     end
   end
 end

data/lib/vellum_ai.rb

@@ -96,11 +96,11 @@ module Vellum
 
     # Executes a deployed Workflow and returns its outputs.
     #
+    # @param inputs [Array<Hash>] The list of inputs defined in the Workflow's Deployment with their corresponding values.Request of type Array<WorkflowRequestInputRequest>, as a Hash
     # @param workflow_deployment_id [String] The ID of the Workflow Deployment. Must provide either this or workflow_deployment_name.
     # @param workflow_deployment_name [String] The name of the Workflow Deployment. Must provide either this or workflow_deployment_id.
     # @param release_tag [String] Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
-    # @param inputs [Array<Hash>] The list of inputs defined in the Workflow's Deployment with their corresponding values.Request of type Array<WorkflowRequestInputRequest>, as a Hash
-    # @param external_id [String] Optionally include a unique identifier for monitoring purposes. Must be unique for a given workflow deployment.
+    # @param external_id [String] Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
     # @param request_options [RequestOptions]
     # @return [ExecuteWorkflowResponse]
     def execute_workflow(inputs:, workflow_deployment_id: nil, workflow_deployment_name: nil, release_tag: nil,
@@ -111,10 +111,10 @@ module Vellum
         req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
         req.body = {
           **(request_options&.additional_body_parameters || {}),
+          inputs: inputs,
           workflow_deployment_id: workflow_deployment_id,
           workflow_deployment_name: workflow_deployment_name,
           release_tag: release_tag,
-          inputs: inputs,
           external_id: external_id
         }.compact
         req.url "#{@request_client.default_environment[:Predict]}/v1/execute-workflow"
@@ -318,11 +318,11 @@ module Vellum
 
     # Executes a deployed Workflow and returns its outputs.
     #
+    # @param inputs [Array<Hash>] The list of inputs defined in the Workflow's Deployment with their corresponding values.Request of type Array<WorkflowRequestInputRequest>, as a Hash
     # @param workflow_deployment_id [String] The ID of the Workflow Deployment. Must provide either this or workflow_deployment_name.
     # @param workflow_deployment_name [String] The name of the Workflow Deployment. Must provide either this or workflow_deployment_id.
     # @param release_tag [String] Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
-    # @param inputs [Array<Hash>] The list of inputs defined in the Workflow's Deployment with their corresponding values.Request of type Array<WorkflowRequestInputRequest>, as a Hash
-    # @param external_id [String] Optionally include a unique identifier for monitoring purposes. Must be unique for a given workflow deployment.
+    # @param external_id [String] Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
     # @param request_options [RequestOptions]
     # @return [ExecuteWorkflowResponse]
     def execute_workflow(inputs:, workflow_deployment_id: nil, workflow_deployment_name: nil, release_tag: nil,
@@ -333,10 +333,10 @@ module Vellum
         req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
         req.body = {
           **(request_options&.additional_body_parameters || {}),
+          inputs: inputs,
           workflow_deployment_id: workflow_deployment_id,
           workflow_deployment_name: workflow_deployment_name,
           release_tag: release_tag,
-          inputs: inputs,
           external_id: external_id
         }.compact
         req.url "#{@async_request_client.default_environment[:Predict]}/v1/execute-workflow"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vellum_ai
 version: !ruby/object:Gem::Version
-  version: 0.3.10
+  version: 0.3.12
 platform: ruby
 authors:
 - Vellum
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-03-07 00:00:00.000000000 Z
+date: 2024-03-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: async-http-faraday
@@ -148,6 +148,15 @@ files:
 - lib/vellum_ai/types/execute_workflow_response.rb
 - lib/vellum_ai/types/execute_workflow_stream_error_response.rb
 - lib/vellum_ai/types/execute_workflow_workflow_result_event.rb
+- lib/vellum_ai/types/execution_array_vellum_value.rb
+- lib/vellum_ai/types/execution_chat_history_vellum_value.rb
+- lib/vellum_ai/types/execution_error_vellum_value.rb
+- lib/vellum_ai/types/execution_function_call_vellum_value.rb
+- lib/vellum_ai/types/execution_json_vellum_value.rb
+- lib/vellum_ai/types/execution_number_vellum_value.rb
+- lib/vellum_ai/types/execution_search_results_vellum_value.rb
+- lib/vellum_ai/types/execution_string_vellum_value.rb
+- lib/vellum_ai/types/execution_vellum_value.rb
 - lib/vellum_ai/types/finish_reason_enum.rb
 - lib/vellum_ai/types/fulfilled_enum.rb
 - lib/vellum_ai/types/fulfilled_execute_prompt_event.rb
@@ -176,6 +185,7 @@ files:
 - lib/vellum_ai/types/image_chat_message_content.rb
 - lib/vellum_ai/types/image_chat_message_content_request.rb
 - lib/vellum_ai/types/image_enum.rb
+- lib/vellum_ai/types/image_variable_value.rb
 - lib/vellum_ai/types/indexing_state_enum.rb
 - lib/vellum_ai/types/initiated_enum.rb
 - lib/vellum_ai/types/initiated_execute_prompt_event.rb