vellum_ai 0.3.15 → 0.14.74

data/lib/vellum_ai/types/chat_message_content.rb

@@ -1,116 +1,122 @@
 # frozen_string_literal: true
-
 require "json"
 require_relative "string_chat_message_content"
 require_relative "function_call_chat_message_content"
 require_relative "array_chat_message_content"
 require_relative "image_chat_message_content"
+require_relative "audio_chat_message_content"
+require_relative "document_chat_message_content"
 
 module Vellum
   class ChatMessageContent
-    attr_reader :member, :discriminant
 
-    private_class_method :new
-    alias kind_of? is_a?
-    # @param member [Object]
-    # @param discriminant [String]
-    # @return [ChatMessageContent]
-    def initialize(member:, discriminant:)
-      # @type [Object]
-      @member = member
-      # @type [String]
-      @discriminant = discriminant
-    end
 
-    # Deserialize a JSON object to an instance of ChatMessageContent
+# Deserialize a JSON object to an instance of ChatMessageContent
     #
-    # @param json_object [JSON]
-    # @return [ChatMessageContent]
+    # @param json_object [String] 
+    # @return [Vellum::ChatMessageContent]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      member = case struct.type
-               when "STRING"
-                 StringChatMessageContent.from_json(json_object: json_object)
-               when "FUNCTION_CALL"
-                 FunctionCallChatMessageContent.from_json(json_object: json_object)
-               when "ARRAY"
-                 ArrayChatMessageContent.from_json(json_object: json_object)
-               when "IMAGE"
-                 ImageChatMessageContent.from_json(json_object: json_object)
-               else
-                 StringChatMessageContent.from_json(json_object: json_object)
-               end
-      new(member: member, discriminant: struct.type)
-    end
-
-    # For Union Types, to_json functionality is delegated to the wrapped member.
-    #
-    # @return [JSON]
-    def to_json(*_args)
-      case @discriminant
-      when "STRING"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "FUNCTION_CALL"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "ARRAY"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "IMAGE"
-        { **@member.to_json, type: @discriminant }.to_json
-      else
-        { "type": @discriminant, value: @member }.to_json
+      begin
+        Vellum::StringChatMessageContent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::StringChatMessageContent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::FunctionCallChatMessageContent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::FunctionCallChatMessageContent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
       end
-      @member.to_json
+      begin
+        Vellum::ArrayChatMessageContent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::ArrayChatMessageContent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::ImageChatMessageContent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::ImageChatMessageContent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::AudioChatMessageContent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::AudioChatMessageContent.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::DocumentChatMessageContent.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::DocumentChatMessageContent.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.type
-      when "STRING"
-        StringChatMessageContent.validate_raw(obj: obj)
-      when "FUNCTION_CALL"
-        FunctionCallChatMessageContent.validate_raw(obj: obj)
-      when "ARRAY"
-        ArrayChatMessageContent.validate_raw(obj: obj)
-      when "IMAGE"
-        ImageChatMessageContent.validate_raw(obj: obj)
-      else
-        raise("Passed value matched no type within the union, validation failed.")
+      begin
+        return Vellum::StringChatMessageContent.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 [StringChatMessageContent]
-    # @return [ChatMessageContent]
-    def self.string(member:)
-      new(member: member, discriminant: "STRING")
-    end
-
-    # @param member [FunctionCallChatMessageContent]
-    # @return [ChatMessageContent]
-    def self.function_call(member:)
-      new(member: member, discriminant: "FUNCTION_CALL")
-    end
-
-    # @param member [ArrayChatMessageContent]
-    # @return [ChatMessageContent]
-    def self.array(member:)
-      new(member: member, discriminant: "ARRAY")
-    end
-
-    # @param member [ImageChatMessageContent]
-    # @return [ChatMessageContent]
-    def self.image(member:)
-      new(member: member, discriminant: "IMAGE")
+      begin
+        return Vellum::FunctionCallChatMessageContent.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::ArrayChatMessageContent.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::ImageChatMessageContent.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::AudioChatMessageContent.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::DocumentChatMessageContent.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/chat_message_content_request.rb

@@ -1,116 +1,122 @@
 # frozen_string_literal: true
-
 require "json"
 require_relative "string_chat_message_content_request"
 require_relative "function_call_chat_message_content_request"
 require_relative "array_chat_message_content_request"
 require_relative "image_chat_message_content_request"
+require_relative "audio_chat_message_content_request"
+require_relative "document_chat_message_content_request"
 
 module Vellum
   class ChatMessageContentRequest
-    attr_reader :member, :discriminant
 
-    private_class_method :new
-    alias kind_of? is_a?
-    # @param member [Object]
-    # @param discriminant [String]
-    # @return [ChatMessageContentRequest]
-    def initialize(member:, discriminant:)
-      # @type [Object]
-      @member = member
-      # @type [String]
-      @discriminant = discriminant
-    end
 
-    # Deserialize a JSON object to an instance of ChatMessageContentRequest
+# Deserialize a JSON object to an instance of ChatMessageContentRequest
     #
-    # @param json_object [JSON]
-    # @return [ChatMessageContentRequest]
+    # @param json_object [String] 
+    # @return [Vellum::ChatMessageContentRequest]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
-      member = case struct.type
-               when "STRING"
-                 StringChatMessageContentRequest.from_json(json_object: json_object)
-               when "FUNCTION_CALL"
-                 FunctionCallChatMessageContentRequest.from_json(json_object: json_object)
-               when "ARRAY"
-                 ArrayChatMessageContentRequest.from_json(json_object: json_object)
-               when "IMAGE"
-                 ImageChatMessageContentRequest.from_json(json_object: json_object)
-               else
-                 StringChatMessageContentRequest.from_json(json_object: json_object)
-               end
-      new(member: member, discriminant: struct.type)
-    end
-
-    # For Union Types, to_json functionality is delegated to the wrapped member.
-    #
-    # @return [JSON]
-    def to_json(*_args)
-      case @discriminant
-      when "STRING"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "FUNCTION_CALL"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "ARRAY"
-        { **@member.to_json, type: @discriminant }.to_json
-      when "IMAGE"
-        { **@member.to_json, type: @discriminant }.to_json
-      else
-        { "type": @discriminant, value: @member }.to_json
+      begin
+        Vellum::StringChatMessageContentRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::StringChatMessageContentRequest.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::FunctionCallChatMessageContentRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::FunctionCallChatMessageContentRequest.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
       end
-      @member.to_json
+      begin
+        Vellum::ArrayChatMessageContentRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::ArrayChatMessageContentRequest.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::ImageChatMessageContentRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::ImageChatMessageContentRequest.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::AudioChatMessageContentRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::AudioChatMessageContentRequest.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::DocumentChatMessageContentRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::DocumentChatMessageContentRequest.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.type
-      when "STRING"
-        StringChatMessageContentRequest.validate_raw(obj: obj)
-      when "FUNCTION_CALL"
-        FunctionCallChatMessageContentRequest.validate_raw(obj: obj)
-      when "ARRAY"
-        ArrayChatMessageContentRequest.validate_raw(obj: obj)
-      when "IMAGE"
-        ImageChatMessageContentRequest.validate_raw(obj: obj)
-      else
-        raise("Passed value matched no type within the union, validation failed.")
+      begin
+        return Vellum::StringChatMessageContentRequest.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 [StringChatMessageContentRequest]
-    # @return [ChatMessageContentRequest]
-    def self.string(member:)
-      new(member: member, discriminant: "STRING")
-    end
-
-    # @param member [FunctionCallChatMessageContentRequest]
-    # @return [ChatMessageContentRequest]
-    def self.function_call(member:)
-      new(member: member, discriminant: "FUNCTION_CALL")
-    end
-
-    # @param member [ArrayChatMessageContentRequest]
-    # @return [ChatMessageContentRequest]
-    def self.array(member:)
-      new(member: member, discriminant: "ARRAY")
-    end
-
-    # @param member [ImageChatMessageContentRequest]
-    # @return [ChatMessageContentRequest]
-    def self.image(member:)
-      new(member: member, discriminant: "IMAGE")
+      begin
+        return Vellum::FunctionCallChatMessageContentRequest.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::ArrayChatMessageContentRequest.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::ImageChatMessageContentRequest.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::AudioChatMessageContentRequest.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::DocumentChatMessageContentRequest.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/chat_message_prompt_block.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+require_relative "prompt_block_state"
+require_relative "ephemeral_prompt_cache_config"
+require_relative "chat_message_role"
+require_relative "prompt_block"
+require "ostruct"
+require "json"
+
+module Vellum
+# A block that represents a chat message in a prompt template.
+  class ChatMessagePromptBlock
+  # @return [String] 
+    attr_reader :block_type
+  # @return [Vellum::PromptBlockState] 
+    attr_reader :state
+  # @return [Vellum::EphemeralPromptCacheConfig] 
+    attr_reader :cache_config
+  # @return [Vellum::ChatMessageRole] 
+    attr_reader :chat_role
+  # @return [String] 
+    attr_reader :chat_source
+  # @return [Boolean] 
+    attr_reader :chat_message_unterminated
+  # @return [Array<Vellum::PromptBlock>] 
+    attr_reader :blocks
+  # @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 block_type [String] 
+    # @param state [Vellum::PromptBlockState] 
+    # @param cache_config [Vellum::EphemeralPromptCacheConfig] 
+    # @param chat_role [Vellum::ChatMessageRole] 
+    # @param chat_source [String] 
+    # @param chat_message_unterminated [Boolean] 
+    # @param blocks [Array<Vellum::PromptBlock>] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::ChatMessagePromptBlock]
+    def initialize(block_type:, state: OMIT, cache_config: OMIT, chat_role:, chat_source: OMIT, chat_message_unterminated: OMIT, blocks:, additional_properties: nil)
+      @block_type = block_type
+      @state = state if state != OMIT
+      @cache_config = cache_config if cache_config != OMIT
+      @chat_role = chat_role
+      @chat_source = chat_source if chat_source != OMIT
+      @chat_message_unterminated = chat_message_unterminated if chat_message_unterminated != OMIT
+      @blocks = blocks
+      @additional_properties = additional_properties
+      @_field_set = { "block_type": block_type, "state": state, "cache_config": cache_config, "chat_role": chat_role, "chat_source": chat_source, "chat_message_unterminated": chat_message_unterminated, "blocks": blocks }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of ChatMessagePromptBlock
+    #
+    # @param json_object [String] 
+    # @return [Vellum::ChatMessagePromptBlock]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      block_type = parsed_json["block_type"]
+      state = parsed_json["state"]
+      unless parsed_json["cache_config"].nil?
+        cache_config = parsed_json["cache_config"].to_json
+        cache_config = Vellum::EphemeralPromptCacheConfig.from_json(json_object: cache_config)
+      else
+        cache_config = nil
+      end
+      chat_role = parsed_json["chat_role"]
+      chat_source = parsed_json["chat_source"]
+      chat_message_unterminated = parsed_json["chat_message_unterminated"]
+      blocks = parsed_json["blocks"]&.map do | item |
+  item = item.to_json
+  Vellum::PromptBlock.from_json(json_object: item)
+end
+      new(
+        block_type: block_type,
+        state: state,
+        cache_config: cache_config,
+        chat_role: chat_role,
+        chat_source: chat_source,
+        chat_message_unterminated: chat_message_unterminated,
+        blocks: blocks,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of ChatMessagePromptBlock 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.block_type.is_a?(String) != false || raise("Passed value for field obj.block_type is not the expected type, validation failed.")
+      obj.state&.is_a?(Vellum::PromptBlockState) != false || raise("Passed value for field obj.state is not the expected type, validation failed.")
+      obj.cache_config.nil? || Vellum::EphemeralPromptCacheConfig.validate_raw(obj: obj.cache_config)
+      obj.chat_role.is_a?(Vellum::ChatMessageRole) != false || raise("Passed value for field obj.chat_role 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.chat_message_unterminated&.is_a?(Boolean) != false || raise("Passed value for field obj.chat_message_unterminated is not the expected type, validation failed.")
+      obj.blocks.is_a?(Array) != false || raise("Passed value for field obj.blocks is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/chat_message_request.rb

@@ -1,67 +1,84 @@
 # frozen_string_literal: true
-
 require_relative "chat_message_role"
 require_relative "chat_message_content_request"
+require "ostruct"
 require "json"
 
 module Vellum
   class ChatMessageRequest
-    attr_reader :text, :role, :content, :source, :additional_properties
+  # @return [String] 
+    attr_reader :text
+  # @return [Vellum::ChatMessageRole] 
+    attr_reader :role
+  # @return [Vellum::ChatMessageContentRequest] 
+    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 [ChatMessageContentRequest]
+    # @param text [String] 
+    # @param role [Vellum::ChatMessageRole] 
+    # @param content [Vellum::ChatMessageContentRequest] 
     # @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 [ChatMessageRequest]
-    def initialize(role:, text: nil, content: nil, source: nil, additional_properties: nil)
-      # @type [String]
-      @text = text
-      # @type [CHAT_MESSAGE_ROLE]
+    # @return [Vellum::ChatMessageRequest]
+    def initialize(text: OMIT, role:, content: OMIT, source: OMIT, additional_properties: nil)
+      @text = text if text != OMIT
       @role = role
-      # @type [ChatMessageContentRequest]
-      @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 ChatMessageRequest
+# Deserialize a JSON object to an instance of ChatMessageRequest
     #
-    # @param json_object [JSON]
-    # @return [ChatMessageRequest]
+    # @param json_object [String] 
+    # @return [Vellum::ChatMessageRequest]
     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 = ChatMessageContentRequest.from_json(json_object: content)
+        content = Vellum::ChatMessageContentRequest.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 ChatMessageRequest to a JSON object
+# Serialize an instance of ChatMessageRequest 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? || ChatMessageContentRequest.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::ChatMessageContentRequest.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