vellum_ai 0.3.14 → 0.14.74

data/lib/vellum_ai/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1_request.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+# Basic vectorizer for sentence-transformers/multi-qa-mpnet-base-cos-v1.
+  class BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request
+  # @return [Hash{String => Object}] 
+    attr_reader :config
+  # @return [String] 
+    attr_reader :model_name
+  # @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 config [Hash{String => Object}] 
+    # @param model_name [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request]
+    def initialize(config: OMIT, model_name:, additional_properties: nil)
+      @config = config if config != OMIT
+      @model_name = model_name
+      @additional_properties = additional_properties
+      @_field_set = { "config": config, "model_name": model_name }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of
+#  BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request
+    #
+    # @param json_object [String] 
+    # @return [Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      config = parsed_json["config"]
+      model_name = parsed_json["model_name"]
+      new(
+        config: config,
+        model_name: model_name,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of
+#  BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request 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.config&.is_a?(Hash) != false || raise("Passed value for field obj.config is not the expected type, validation failed.")
+      obj.model_name.is_a?(String) != false || raise("Passed value for field obj.model_name is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+# Basic vectorizer for sentence-transformers/multi-qa-mpnet-base-dot-v1.
+  class BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1
+  # @return [String] 
+    attr_reader :model_name
+  # @return [Hash{String => Object}] 
+    attr_reader :config
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param model_name [String] 
+    # @param config [Hash{String => Object}] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1]
+    def initialize(model_name:, config: OMIT, additional_properties: nil)
+      @model_name = model_name
+      @config = config if config != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "model_name": model_name, "config": config }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of
+#  BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1
+    #
+    # @param json_object [String] 
+    # @return [Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      model_name = parsed_json["model_name"]
+      config = parsed_json["config"]
+      new(
+        model_name: model_name,
+        config: config,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of
+#  BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1 to a JSON object
+    #
+    # @return [String]
+    def to_json
+      @_field_set&.to_json
+    end
+# Leveraged for Union-type generation, validate_raw attempts to parse the given
+#  hash and check each fields type against the current object's property
+#  definitions.
+    #
+    # @param obj [Object] 
+    # @return [Void]
+    def self.validate_raw(obj:)
+      obj.model_name.is_a?(String) != false || raise("Passed value for field obj.model_name is not the expected type, validation failed.")
+      obj.config&.is_a?(Hash) != false || raise("Passed value for field obj.config is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+# Basic vectorizer for sentence-transformers/multi-qa-mpnet-base-dot-v1.
+  class BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request
+  # @return [String] 
+    attr_reader :model_name
+  # @return [Hash{String => Object}] 
+    attr_reader :config
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param model_name [String] 
+    # @param config [Hash{String => Object}] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request]
+    def initialize(model_name:, config: OMIT, additional_properties: nil)
+      @model_name = model_name
+      @config = config if config != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "model_name": model_name, "config": config }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of
+#  BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request
+    #
+    # @param json_object [String] 
+    # @return [Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      model_name = parsed_json["model_name"]
+      config = parsed_json["config"]
+      new(
+        model_name: model_name,
+        config: config,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of
+#  BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request to a JSON object
+    #
+    # @return [String]
+    def to_json
+      @_field_set&.to_json
+    end
+# Leveraged for Union-type generation, validate_raw attempts to parse the given
+#  hash and check each fields type against the current object's property
+#  definitions.
+    #
+    # @param obj [Object] 
+    # @return [Void]
+    def self.validate_raw(obj:)
+      obj.model_name.is_a?(String) != false || raise("Passed value for field obj.model_name is not the expected type, validation failed.")
+      obj.config&.is_a?(Hash) != false || raise("Passed value for field obj.config is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/build_status_enum.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Vellum
+# * `QUEUED` - Queued
+#  * `BUILDING` - Building
+#  * `AVAILABLE` - Available
+#  * `FAILED` - Failed
+#  * `UNKNOWN` - Unknown
+  class BuildStatusEnum
+
+    QUEUED = "QUEUED"
+    BUILDING = "BUILDING"
+    AVAILABLE = "AVAILABLE"
+    FAILED = "FAILED"
+    UNKNOWN = "UNKNOWN"
+
+  end
+end

data/lib/vellum_ai/types/chat_history_input.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+require_relative "chat_message"
+require "ostruct"
+require "json"
+
+module Vellum
+# A user input representing a list of chat messages
+  class ChatHistoryInput
+  # @return [String] The variable's name, as defined in the deployment.
+    attr_reader :name
+  # @return [String] 
+    attr_reader :type
+  # @return [Array<Vellum::ChatMessage>] 
+    attr_reader :value
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param name [String] The variable's name, as defined in the deployment.
+    # @param type [String] 
+    # @param value [Array<Vellum::ChatMessage>] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::ChatHistoryInput]
+    def initialize(name:, type:, value:, additional_properties: nil)
+      @name = name
+      @type = type
+      @value = value
+      @additional_properties = additional_properties
+      @_field_set = { "name": name, "type": type, "value": value }
+    end
+# Deserialize a JSON object to an instance of ChatHistoryInput
+    #
+    # @param json_object [String] 
+    # @return [Vellum::ChatHistoryInput]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      name = parsed_json["name"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]&.map do | item |
+  item = item.to_json
+  Vellum::ChatMessage.from_json(json_object: item)
+end
+      new(
+        name: name,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of ChatHistoryInput 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.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      obj.value.is_a?(Array) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/chat_history_input_request.rb

@@ -1,55 +1,73 @@
 # frozen_string_literal: true
-
 require_relative "chat_message_request"
+require "ostruct"
 require "json"
 
 module Vellum
-  # A user input representing a list of chat messages
+# A user input representing a list of chat messages
   class ChatHistoryInputRequest
-    attr_reader :name, :value, :additional_properties
+  # @return [String] The variable's name, as defined in the deployment.
+    attr_reader :name
+  # @return [String] 
+    attr_reader :type
+  # @return [Array<Vellum::ChatMessageRequest>] 
+    attr_reader :value
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
 
     # @param name [String] The variable's name, as defined in the deployment.
-    # @param value [Array<ChatMessageRequest>]
+    # @param type [String] 
+    # @param value [Array<Vellum::ChatMessageRequest>] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [ChatHistoryInputRequest]
-    def initialize(name:, value:, additional_properties: nil)
-      # @type [String] The variable's name, as defined in the deployment.
+    # @return [Vellum::ChatHistoryInputRequest]
+    def initialize(name:, type:, value:, additional_properties: nil)
       @name = name
-      # @type [Array<ChatMessageRequest>]
+      @type = type
       @value = value
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "name": name, "type": type, "value": value }
     end
-
-    # Deserialize a JSON object to an instance of ChatHistoryInputRequest
+# Deserialize a JSON object to an instance of ChatHistoryInputRequest
     #
-    # @param json_object [JSON]
-    # @return [ChatHistoryInputRequest]
+    # @param json_object [String] 
+    # @return [Vellum::ChatHistoryInputRequest]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      name = struct.name
-      value = parsed_json["value"].map do |v|
-        v = v.to_json
-        ChatMessageRequest.from_json(json_object: v)
-      end
-      new(name: name, value: value, additional_properties: struct)
+      name = parsed_json["name"]
+      type = parsed_json["type"]
+      value = parsed_json["value"]&.map do | item |
+  item = item.to_json
+  Vellum::ChatMessageRequest.from_json(json_object: item)
+end
+      new(
+        name: name,
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of ChatHistoryInputRequest to a JSON object
+# Serialize an instance of ChatHistoryInputRequest to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "name": @name, "value": @value }.to_json
+    # @return [String]
+    def to_json
+      @_field_set&.to_json
     end
-
-    # Leveraged for Union-type generation, validate_raw attempts to parse the given hash and check each fields type against the current object's property definitions.
+# Leveraged for Union-type generation, validate_raw attempts to parse the given
+#  hash and check each fields type against the current object's property
+#  definitions.
     #
-    # @param obj [Object]
+    # @param obj [Object] 
     # @return [Void]
     def self.validate_raw(obj:)
       obj.name.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
       obj.value.is_a?(Array) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
     end
   end
-end
+end

data/lib/vellum_ai/types/chat_history_vellum_value.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+require_relative "chat_message"
+require "ostruct"
+require "json"
+
+module Vellum
+# A value representing Chat History.
+  class ChatHistoryVellumValue
+  # @return [String] 
+    attr_reader :type
+  # @return [Array<Vellum::ChatMessage>] 
+    attr_reader :value
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param type [String] 
+    # @param value [Array<Vellum::ChatMessage>] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::ChatHistoryVellumValue]
+    def initialize(type:, value: OMIT, additional_properties: nil)
+      @type = type
+      @value = value if value != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of ChatHistoryVellumValue
+    #
+    # @param json_object [String] 
+    # @return [Vellum::ChatHistoryVellumValue]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      type = parsed_json["type"]
+      value = parsed_json["value"]&.map do | item |
+  item = item.to_json
+  Vellum::ChatMessage.from_json(json_object: item)
+end
+      new(
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of ChatHistoryVellumValue 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.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      obj.value&.is_a?(Array) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/chat_history_vellum_value_request.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+require_relative "chat_message_request"
+require "ostruct"
+require "json"
+
+module Vellum
+# A value representing Chat History.
+  class ChatHistoryVellumValueRequest
+  # @return [String] 
+    attr_reader :type
+  # @return [Array<Vellum::ChatMessageRequest>] 
+    attr_reader :value
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param type [String] 
+    # @param value [Array<Vellum::ChatMessageRequest>] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::ChatHistoryVellumValueRequest]
+    def initialize(type:, value: OMIT, additional_properties: nil)
+      @type = type
+      @value = value if value != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "type": type, "value": value }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of ChatHistoryVellumValueRequest
+    #
+    # @param json_object [String] 
+    # @return [Vellum::ChatHistoryVellumValueRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      type = parsed_json["type"]
+      value = parsed_json["value"]&.map do | item |
+  item = item.to_json
+  Vellum::ChatMessageRequest.from_json(json_object: item)
+end
+      new(
+        type: type,
+        value: value,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of ChatHistoryVellumValueRequest 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.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
+      obj.value&.is_a?(Array) != false || raise("Passed value for field obj.value is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/chat_message.rb

@@ -1,67 +1,84 @@
 # frozen_string_literal: true
-
 require_relative "chat_message_role"
 require_relative "chat_message_content"
+require "ostruct"
 require "json"
 
 module Vellum
   class ChatMessage
-    attr_reader :text, :role, :content, :source, :additional_properties
+  # @return [String] 
+    attr_reader :text
+  # @return [Vellum::ChatMessageRole] 
+    attr_reader :role
+  # @return [Vellum::ChatMessageContent] 
+    attr_reader :content
+  # @return [String] An optional identifier representing who or what generated this message.
+    attr_reader :source
+  # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+  # @return [Object] 
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
 
-    # @param text [String]
-    # @param role [CHAT_MESSAGE_ROLE]
-    # @param content [ChatMessageContent]
+    # @param text [String] 
+    # @param role [Vellum::ChatMessageRole] 
+    # @param content [Vellum::ChatMessageContent] 
     # @param source [String] An optional identifier representing who or what generated this message.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [ChatMessage]
-    def initialize(role:, text: nil, content: nil, source: nil, additional_properties: nil)
-      # @type [String]
-      @text = text
-      # @type [CHAT_MESSAGE_ROLE]
+    # @return [Vellum::ChatMessage]
+    def initialize(text: OMIT, role:, content: OMIT, source: OMIT, additional_properties: nil)
+      @text = text if text != OMIT
       @role = role
-      # @type [ChatMessageContent]
-      @content = content
-      # @type [String] An optional identifier representing who or what generated this message.
-      @source = source
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
+      @content = content if content != OMIT
+      @source = source if source != OMIT
       @additional_properties = additional_properties
+      @_field_set = { "text": text, "role": role, "content": content, "source": source }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of ChatMessage
+# Deserialize a JSON object to an instance of ChatMessage
     #
-    # @param json_object [JSON]
-    # @return [ChatMessage]
+    # @param json_object [String] 
+    # @return [Vellum::ChatMessage]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      text = struct.text
-      role = CHAT_MESSAGE_ROLE.key(parsed_json["role"]) || parsed_json["role"]
-      if parsed_json["content"].nil?
-        content = nil
-      else
+      text = parsed_json["text"]
+      role = parsed_json["role"]
+      unless parsed_json["content"].nil?
         content = parsed_json["content"].to_json
-        content = ChatMessageContent.from_json(json_object: content)
+        content = Vellum::ChatMessageContent.from_json(json_object: content)
+      else
+        content = nil
       end
-      source = struct.source
-      new(text: text, role: role, content: content, source: source, additional_properties: struct)
+      source = parsed_json["source"]
+      new(
+        text: text,
+        role: role,
+        content: content,
+        source: source,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of ChatMessage to a JSON object
+# Serialize an instance of ChatMessage to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "text": @text, "role": CHAT_MESSAGE_ROLE[@role] || @role, "content": @content, "source": @source }.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.text&.is_a?(String) != false || raise("Passed value for field obj.text is not the expected type, validation failed.")
-      obj.role.is_a?(CHAT_MESSAGE_ROLE) != false || raise("Passed value for field obj.role is not the expected type, validation failed.")
-      obj.content.nil? || ChatMessageContent.validate_raw(obj: obj.content)
+      obj.role.is_a?(Vellum::ChatMessageRole) != false || raise("Passed value for field obj.role is not the expected type, validation failed.")
+      obj.content.nil? || Vellum::ChatMessageContent.validate_raw(obj: obj.content)
       obj.source&.is_a?(String) != false || raise("Passed value for field obj.source is not the expected type, validation failed.")
     end
   end
-end
+end