vellum_ai 0.3.14 → 0.14.74

data/lib/vellum_ai/types/indexing_config_vectorizer.rb

@@ -0,0 +1,186 @@
+# frozen_string_literal: true
+require "json"
+require_relative "open_ai_vectorizer_text_embedding_3_small"
+require_relative "open_ai_vectorizer_text_embedding_3_large"
+require_relative "open_ai_vectorizer_text_embedding_ada_002"
+require_relative "basic_vectorizer_intfloat_multilingual_e_5_large"
+require_relative "basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1"
+require_relative "basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1"
+require_relative "hkunlp_instructor_xl_vectorizer"
+require_relative "google_vertex_ai_vectorizer_text_embedding_004"
+require_relative "google_vertex_ai_vectorizer_text_multilingual_embedding_002"
+require_relative "fast_embed_vectorizer_baai_bge_small_en_v_15"
+
+module Vellum
+  class IndexingConfigVectorizer
+
+
+# Deserialize a JSON object to an instance of IndexingConfigVectorizer
+    #
+    # @param json_object [String] 
+    # @return [Vellum::IndexingConfigVectorizer]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      begin
+        Vellum::OpenAiVectorizerTextEmbedding3Small.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::OpenAiVectorizerTextEmbedding3Small.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::OpenAiVectorizerTextEmbedding3Large.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::OpenAiVectorizerTextEmbedding3Large.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::OpenAiVectorizerTextEmbeddingAda002.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::OpenAiVectorizerTextEmbeddingAda002.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::BasicVectorizerIntfloatMultilingualE5Large.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::BasicVectorizerIntfloatMultilingualE5Large.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::HkunlpInstructorXlVectorizer.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::HkunlpInstructorXlVectorizer.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::GoogleVertexAiVectorizerTextEmbedding004.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::GoogleVertexAiVectorizerTextEmbedding004.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::FastEmbedVectorizerBaaiBgeSmallEnV15.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::FastEmbedVectorizerBaaiBgeSmallEnV15.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.
+    #
+    # @param obj [Object] 
+    # @return [Void]
+    def self.validate_raw(obj:)
+      begin
+        return Vellum::OpenAiVectorizerTextEmbedding3Small.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::OpenAiVectorizerTextEmbedding3Large.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::OpenAiVectorizerTextEmbeddingAda002.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::BasicVectorizerIntfloatMultilingualE5Large.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::HkunlpInstructorXlVectorizer.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::GoogleVertexAiVectorizerTextEmbedding004.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::FastEmbedVectorizerBaaiBgeSmallEnV15.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      raise("Passed value matched no type within the union, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/indexing_config_vectorizer_request.rb

@@ -0,0 +1,186 @@
+# frozen_string_literal: true
+require "json"
+require_relative "open_ai_vectorizer_text_embedding_3_small_request"
+require_relative "open_ai_vectorizer_text_embedding_3_large_request"
+require_relative "open_ai_vectorizer_text_embedding_ada_002_request"
+require_relative "basic_vectorizer_intfloat_multilingual_e_5_large_request"
+require_relative "basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1_request"
+require_relative "basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request"
+require_relative "hkunlp_instructor_xl_vectorizer_request"
+require_relative "google_vertex_ai_vectorizer_text_embedding_004_request"
+require_relative "google_vertex_ai_vectorizer_text_multilingual_embedding_002_request"
+require_relative "fast_embed_vectorizer_baai_bge_small_en_v_15_request"
+
+module Vellum
+  class IndexingConfigVectorizerRequest
+
+
+# Deserialize a JSON object to an instance of IndexingConfigVectorizerRequest
+    #
+    # @param json_object [String] 
+    # @return [Vellum::IndexingConfigVectorizerRequest]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      begin
+        Vellum::OpenAiVectorizerTextEmbedding3SmallRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::OpenAiVectorizerTextEmbedding3SmallRequest.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::OpenAiVectorizerTextEmbedding3LargeRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::OpenAiVectorizerTextEmbedding3LargeRequest.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::OpenAiVectorizerTextEmbeddingAda002Request.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::OpenAiVectorizerTextEmbeddingAda002Request.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::BasicVectorizerIntfloatMultilingualE5LargeRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::BasicVectorizerIntfloatMultilingualE5LargeRequest.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::HkunlpInstructorXlVectorizerRequest.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::HkunlpInstructorXlVectorizerRequest.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::GoogleVertexAiVectorizerTextEmbedding004Request.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::GoogleVertexAiVectorizerTextEmbedding004Request.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002Request.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002Request.from_json(json_object: struct)
+else
+  return nil
+end
+      rescue StandardError
+        # noop
+      end
+      begin
+        Vellum::FastEmbedVectorizerBaaiBgeSmallEnV15Request.validate_raw(obj: struct)
+        unless struct.nil?
+  return Vellum::FastEmbedVectorizerBaaiBgeSmallEnV15Request.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.
+    #
+    # @param obj [Object] 
+    # @return [Void]
+    def self.validate_raw(obj:)
+      begin
+        return Vellum::OpenAiVectorizerTextEmbedding3SmallRequest.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::OpenAiVectorizerTextEmbedding3LargeRequest.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::OpenAiVectorizerTextEmbeddingAda002Request.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::BasicVectorizerIntfloatMultilingualE5LargeRequest.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::HkunlpInstructorXlVectorizerRequest.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::GoogleVertexAiVectorizerTextEmbedding004Request.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::GoogleVertexAiVectorizerTextMultilingualEmbedding002Request.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      begin
+        return Vellum::FastEmbedVectorizerBaaiBgeSmallEnV15Request.validate_raw(obj: obj)
+      rescue StandardError
+        # noop
+      end
+      raise("Passed value matched no type within the union, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/indexing_state_enum.rb

@@ -1,12 +1,18 @@
 # frozen_string_literal: true
 
 module Vellum
-  # @type [INDEXING_STATE_ENUM]
-  INDEXING_STATE_ENUM = {
-    awaiting_processing: "AWAITING_PROCESSING",
-    queued: "QUEUED",
-    indexing: "INDEXING",
-    indexed: "INDEXED",
-    failed: "FAILED"
-  }.freeze
-end
+# * `AWAITING_PROCESSING` - Awaiting Processing
+#  * `QUEUED` - Queued
+#  * `INDEXING` - Indexing
+#  * `INDEXED` - Indexed
+#  * `FAILED` - Failed
+  class IndexingStateEnum
+
+    AWAITING_PROCESSING = "AWAITING_PROCESSING"
+    QUEUED = "QUEUED"
+    INDEXING = "INDEXING"
+    INDEXED = "INDEXED"
+    FAILED = "FAILED"
+
+  end
+end

data/lib/vellum_ai/types/initiated_ad_hoc_execute_prompt_event.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+require_relative "ad_hoc_initiated_prompt_execution_meta"
+require "ostruct"
+require "json"
+
+module Vellum
+# The initial data returned indicating that the response from the model has
+#  returned and begun streaming.
+  class InitiatedAdHocExecutePromptEvent
+  # @return [String] 
+    attr_reader :state
+  # @return [Vellum::AdHocInitiatedPromptExecutionMeta] 
+    attr_reader :meta
+  # @return [String] 
+    attr_reader :execution_id
+  # @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 state [String] 
+    # @param meta [Vellum::AdHocInitiatedPromptExecutionMeta] 
+    # @param execution_id [String] 
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vellum::InitiatedAdHocExecutePromptEvent]
+    def initialize(state:, meta: OMIT, execution_id:, additional_properties: nil)
+      @state = state
+      @meta = meta if meta != OMIT
+      @execution_id = execution_id
+      @additional_properties = additional_properties
+      @_field_set = { "state": state, "meta": meta, "execution_id": execution_id }.reject do | _k, v |
+  v == OMIT
+end
+    end
+# Deserialize a JSON object to an instance of InitiatedAdHocExecutePromptEvent
+    #
+    # @param json_object [String] 
+    # @return [Vellum::InitiatedAdHocExecutePromptEvent]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      state = parsed_json["state"]
+      unless parsed_json["meta"].nil?
+        meta = parsed_json["meta"].to_json
+        meta = Vellum::AdHocInitiatedPromptExecutionMeta.from_json(json_object: meta)
+      else
+        meta = nil
+      end
+      execution_id = parsed_json["execution_id"]
+      new(
+        state: state,
+        meta: meta,
+        execution_id: execution_id,
+        additional_properties: struct
+      )
+    end
+# Serialize an instance of InitiatedAdHocExecutePromptEvent 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.state.is_a?(String) != false || raise("Passed value for field obj.state is not the expected type, validation failed.")
+      obj.meta.nil? || Vellum::AdHocInitiatedPromptExecutionMeta.validate_raw(obj: obj.meta)
+      obj.execution_id.is_a?(String) != false || raise("Passed value for field obj.execution_id is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vellum_ai/types/initiated_execute_prompt_event.rb

@@ -1,57 +1,78 @@
 # frozen_string_literal: true
-
 require_relative "initiated_prompt_execution_meta"
+require "ostruct"
 require "json"
 
 module Vellum
-  # The initial data returned indicating that the response from the model has returned and begun streaming.
+# The initial data returned indicating that the response from the model has
+#  returned and begun streaming.
   class InitiatedExecutePromptEvent
-    attr_reader :meta, :execution_id, :additional_properties
+  # @return [String] 
+    attr_reader :state
+  # @return [Vellum::InitiatedPromptExecutionMeta] 
+    attr_reader :meta
+  # @return [String] 
+    attr_reader :execution_id
+  # @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 meta [InitiatedPromptExecutionMeta]
-    # @param execution_id [String]
+    # @param state [String] 
+    # @param meta [Vellum::InitiatedPromptExecutionMeta] 
+    # @param execution_id [String] 
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [InitiatedExecutePromptEvent]
-    def initialize(execution_id:, meta: nil, additional_properties: nil)
-      # @type [InitiatedPromptExecutionMeta]
-      @meta = meta
-      # @type [String]
+    # @return [Vellum::InitiatedExecutePromptEvent]
+    def initialize(state:, meta: OMIT, execution_id:, additional_properties: nil)
+      @state = state
+      @meta = meta if meta != OMIT
       @execution_id = execution_id
-      # @type [OpenStruct] Additional properties unmapped to the current class definition
       @additional_properties = additional_properties
+      @_field_set = { "state": state, "meta": meta, "execution_id": execution_id }.reject do | _k, v |
+  v == OMIT
+end
     end
-
-    # Deserialize a JSON object to an instance of InitiatedExecutePromptEvent
+# Deserialize a JSON object to an instance of InitiatedExecutePromptEvent
     #
-    # @param json_object [JSON]
-    # @return [InitiatedExecutePromptEvent]
+    # @param json_object [String] 
+    # @return [Vellum::InitiatedExecutePromptEvent]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      if parsed_json["meta"].nil?
-        meta = nil
-      else
+      state = parsed_json["state"]
+      unless parsed_json["meta"].nil?
         meta = parsed_json["meta"].to_json
-        meta = InitiatedPromptExecutionMeta.from_json(json_object: meta)
+        meta = Vellum::InitiatedPromptExecutionMeta.from_json(json_object: meta)
+      else
+        meta = nil
       end
-      execution_id = struct.execution_id
-      new(meta: meta, execution_id: execution_id, additional_properties: struct)
+      execution_id = parsed_json["execution_id"]
+      new(
+        state: state,
+        meta: meta,
+        execution_id: execution_id,
+        additional_properties: struct
+      )
     end
-
-    # Serialize an instance of InitiatedExecutePromptEvent to a JSON object
+# Serialize an instance of InitiatedExecutePromptEvent to a JSON object
     #
-    # @return [JSON]
-    def to_json(*_args)
-      { "meta": @meta, "execution_id": @execution_id }.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.meta.nil? || InitiatedPromptExecutionMeta.validate_raw(obj: obj.meta)
+      obj.state.is_a?(String) != false || raise("Passed value for field obj.state is not the expected type, validation failed.")
+      obj.meta.nil? || Vellum::InitiatedPromptExecutionMeta.validate_raw(obj: obj.meta)
       obj.execution_id.is_a?(String) != false || raise("Passed value for field obj.execution_id is not the expected type, validation failed.")
     end
   end
-end
+end