vellum_ai 0.3.14 → 0.14.74

data/lib/vellum_ai/types/generate_stream_result.rb

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

data/lib/vellum_ai/types/generate_stream_result_data.rb

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

data/lib/vellum_ai/types/google_vertex_ai_vectorizer_config.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+  class GoogleVertexAiVectorizerConfig
+  # @return [String] 
+    attr_reader :project_id
+  # @return [String] 
+    attr_reader :region
+  # @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 project_id [String] 
+    # @param region [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::GoogleVertexAiVectorizerConfig]
+    def initialize(project_id:, region:, additional_properties: nil)
+      @project_id = project_id
+      @region = region
+      @additional_properties = additional_properties
+      @_field_set = { "project_id": project_id, "region": region }
+    end
+# Deserialize a JSON object to an instance of GoogleVertexAiVectorizerConfig
+    #
+    # @param json_object [String] 
+    # @return [Vellum::GoogleVertexAiVectorizerConfig]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      project_id = parsed_json["project_id"]
+      region = parsed_json["region"]
+      new(
+        project_id: project_id,
+        region: region,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of GoogleVertexAiVectorizerConfig 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.project_id.is_a?(String) != false || raise("Passed value for field obj.project_id is not the expected type, validation failed.")
+      obj.region.is_a?(String) != false || raise("Passed value for field obj.region is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/google_vertex_ai_vectorizer_config_request.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+require "ostruct"
+require "json"
+
+module Vellum
+  class GoogleVertexAiVectorizerConfigRequest
+  # @return [String] 
+    attr_reader :project_id
+  # @return [String] 
+    attr_reader :region
+  # @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 project_id [String] 
+    # @param region [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::GoogleVertexAiVectorizerConfigRequest]
+    def initialize(project_id:, region:, additional_properties: nil)
+      @project_id = project_id
+      @region = region
+      @additional_properties = additional_properties
+      @_field_set = { "project_id": project_id, "region": region }
+    end
+# Deserialize a JSON object to an instance of
+#  GoogleVertexAiVectorizerConfigRequest
+    #
+    # @param json_object [String] 
+    # @return [Vellum::GoogleVertexAiVectorizerConfigRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      project_id = parsed_json["project_id"]
+      region = parsed_json["region"]
+      new(
+        project_id: project_id,
+        region: region,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of GoogleVertexAiVectorizerConfigRequest 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.project_id.is_a?(String) != false || raise("Passed value for field obj.project_id is not the expected type, validation failed.")
+      obj.region.is_a?(String) != false || raise("Passed value for field obj.region is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/google_vertex_ai_vectorizer_text_embedding_004.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+require_relative "google_vertex_ai_vectorizer_config"
+require "ostruct"
+require "json"
+
+module Vellum
+  class GoogleVertexAiVectorizerTextEmbedding004
+  # @return [String] 
+    attr_reader :model_name
+  # @return [Vellum::GoogleVertexAiVectorizerConfig] 
+    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 [Vellum::GoogleVertexAiVectorizerConfig] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::GoogleVertexAiVectorizerTextEmbedding004]
+    def initialize(model_name:, config:, additional_properties: nil)
+      @model_name = model_name
+      @config = config
+      @additional_properties = additional_properties
+      @_field_set = { "model_name": model_name, "config": config }
+    end
+# Deserialize a JSON object to an instance of
+#  GoogleVertexAiVectorizerTextEmbedding004
+    #
+    # @param json_object [String] 
+    # @return [Vellum::GoogleVertexAiVectorizerTextEmbedding004]
+    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"]
+      unless parsed_json["config"].nil?
+        config = parsed_json["config"].to_json
+        config = Vellum::GoogleVertexAiVectorizerConfig.from_json(json_object: config)
+      else
+        config = nil
+      end
+      new(
+        model_name: model_name,
+        config: config,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of GoogleVertexAiVectorizerTextEmbedding004 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.")
+      Vellum::GoogleVertexAiVectorizerConfig.validate_raw(obj: obj.config)
+    end
+  end
+end

data/lib/vellum_ai/types/google_vertex_ai_vectorizer_text_embedding_004_request.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+require_relative "google_vertex_ai_vectorizer_config_request"
+require "ostruct"
+require "json"
+
+module Vellum
+  class GoogleVertexAiVectorizerTextEmbedding004Request
+  # @return [String] 
+    attr_reader :model_name
+  # @return [Vellum::GoogleVertexAiVectorizerConfigRequest] 
+    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 [Vellum::GoogleVertexAiVectorizerConfigRequest] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::GoogleVertexAiVectorizerTextEmbedding004Request]
+    def initialize(model_name:, config:, additional_properties: nil)
+      @model_name = model_name
+      @config = config
+      @additional_properties = additional_properties
+      @_field_set = { "model_name": model_name, "config": config }
+    end
+# Deserialize a JSON object to an instance of
+#  GoogleVertexAiVectorizerTextEmbedding004Request
+    #
+    # @param json_object [String] 
+    # @return [Vellum::GoogleVertexAiVectorizerTextEmbedding004Request]
+    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"]
+      unless parsed_json["config"].nil?
+        config = parsed_json["config"].to_json
+        config = Vellum::GoogleVertexAiVectorizerConfigRequest.from_json(json_object: config)
+      else
+        config = nil
+      end
+      new(
+        model_name: model_name,
+        config: config,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of GoogleVertexAiVectorizerTextEmbedding004Request 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.")
+      Vellum::GoogleVertexAiVectorizerConfigRequest.validate_raw(obj: obj.config)
+    end
+  end
+end

data/lib/vellum_ai/types/google_vertex_ai_vectorizer_text_multilingual_embedding_002.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+require_relative "google_vertex_ai_vectorizer_config"
+require "ostruct"
+require "json"
+
+module Vellum
+  class GoogleVertexAiVectorizerTextMultilingualEmbedding002
+  # @return [String] 
+    attr_reader :model_name
+  # @return [Vellum::GoogleVertexAiVectorizerConfig] 
+    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 [Vellum::GoogleVertexAiVectorizerConfig] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002]
+    def initialize(model_name:, config:, additional_properties: nil)
+      @model_name = model_name
+      @config = config
+      @additional_properties = additional_properties
+      @_field_set = { "model_name": model_name, "config": config }
+    end
+# Deserialize a JSON object to an instance of
+#  GoogleVertexAiVectorizerTextMultilingualEmbedding002
+    #
+    # @param json_object [String] 
+    # @return [Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002]
+    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"]
+      unless parsed_json["config"].nil?
+        config = parsed_json["config"].to_json
+        config = Vellum::GoogleVertexAiVectorizerConfig.from_json(json_object: config)
+      else
+        config = nil
+      end
+      new(
+        model_name: model_name,
+        config: config,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of GoogleVertexAiVectorizerTextMultilingualEmbedding002 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.")
+      Vellum::GoogleVertexAiVectorizerConfig.validate_raw(obj: obj.config)
+    end
+  end
+end

data/lib/vellum_ai/types/google_vertex_ai_vectorizer_text_multilingual_embedding_002_request.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+require_relative "google_vertex_ai_vectorizer_config_request"
+require "ostruct"
+require "json"
+
+module Vellum
+  class GoogleVertexAiVectorizerTextMultilingualEmbedding002Request
+  # @return [String] 
+    attr_reader :model_name
+  # @return [Vellum::GoogleVertexAiVectorizerConfigRequest] 
+    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 [Vellum::GoogleVertexAiVectorizerConfigRequest] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002Request]
+    def initialize(model_name:, config:, additional_properties: nil)
+      @model_name = model_name
+      @config = config
+      @additional_properties = additional_properties
+      @_field_set = { "model_name": model_name, "config": config }
+    end
+# Deserialize a JSON object to an instance of
+#  GoogleVertexAiVectorizerTextMultilingualEmbedding002Request
+    #
+    # @param json_object [String] 
+    # @return [Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002Request]
+    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"]
+      unless parsed_json["config"].nil?
+        config = parsed_json["config"].to_json
+        config = Vellum::GoogleVertexAiVectorizerConfigRequest.from_json(json_object: config)
+      else
+        config = nil
+      end
+      new(
+        model_name: model_name,
+        config: config,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of
+#  GoogleVertexAiVectorizerTextMultilingualEmbedding002Request 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.")
+      Vellum::GoogleVertexAiVectorizerConfigRequest.validate_raw(obj: obj.config)
+    end
+  end
+end