vapi_server_sdk 0.3.0 → 0.5.0

data/lib/vapi_server_sdk/types/transfer_destination_assistant.rb

@@ -75,6 +75,29 @@ module Vapi
     #  user: Yes, please
     #  assistant: how can i help?
     #  user: i need help with my account
+    #  - `swap-system-message-in-history-and-remove-transfer-tool-messages`: This
+    #  replaces the original system message with the new assistant's system message on
+    #  transfer and removes transfer tool messages from conversation history sent to
+    #  the LLM.
+    #  Example:
+    #  Pre-transfer:
+    #  system: assistant1 system message
+    #  assistant: assistant1 first message
+    #  user: hey, good morning
+    #  assistant: how can i help?
+    #  user: i need help with my account
+    #  transfer-tool
+    #  transfer-tool-result
+    #  assistant: (destination.message)
+    #  Post-transfer:
+    #  system: assistant2 system message
+    #  assistant: assistant1 first message
+    #  user: hey, good morning
+    #  assistant: how can i help?
+    #  user: i need help with my account
+    #  assistant: (destination.message)
+    #  assistant: assistant2 first message (or model generated if firstMessageMode
+    #  is set to `assistant-speaks-first-with-model-generated-message`)
     #  @default 'rolling-history'
     attr_reader :transfer_mode
     # @return [String] This is the assistant to transfer the call to.
@@ -157,6 +180,29 @@ module Vapi
     #  user: Yes, please
     #  assistant: how can i help?
     #  user: i need help with my account
+    #  - `swap-system-message-in-history-and-remove-transfer-tool-messages`: This
+    #  replaces the original system message with the new assistant's system message on
+    #  transfer and removes transfer tool messages from conversation history sent to
+    #  the LLM.
+    #  Example:
+    #  Pre-transfer:
+    #  system: assistant1 system message
+    #  assistant: assistant1 first message
+    #  user: hey, good morning
+    #  assistant: how can i help?
+    #  user: i need help with my account
+    #  transfer-tool
+    #  transfer-tool-result
+    #  assistant: (destination.message)
+    #  Post-transfer:
+    #  system: assistant2 system message
+    #  assistant: assistant1 first message
+    #  user: hey, good morning
+    #  assistant: how can i help?
+    #  user: i need help with my account
+    #  assistant: (destination.message)
+    #  assistant: assistant2 first message (or model generated if firstMessageMode
+    #  is set to `assistant-speaks-first-with-model-generated-message`)
     #  @default 'rolling-history'
     # @param assistant_name [String] This is the assistant to transfer the call to.
     # @param description [String] This is the description of the destination, used by the AI to choose when and

data/lib/vapi_server_sdk/types/transfer_plan.rb

@@ -27,6 +27,9 @@ module Vapi
     #  - `warm-transfer-wait-for-operator-to-speak-first-and-then-say-summary`: The
     #  assistant dials the destination, waits for the operator to speak, provides a
     #  summary of the call to the destination party, and then connects the customer.
+    #  - `warm-transfer-twiml`: The assistant dials the destination, executes the twiml
+    #  instructions on the destination call leg, connects the customer, and leaves the
+    #  call.
     #  @default 'blind-transfer'
     attr_reader :mode
     # @return [Vapi::TransferPlanMessage] This is the message the assistant will deliver to the destination party before
@@ -40,6 +43,19 @@ module Vapi
     #  - 'refer': Uses SIP REFER to transfer the call (default)
     #  - 'bye': Ends current call with SIP BYE
     attr_reader :sip_verb
+    # @return [String] This is the TwiML instructions to execute on the destination call leg before
+    #  connecting the customer.
+    #  Usage:
+    #  - Used only when `mode` is `warm-transfer-twiml`.
+    #  - Supports only `Play`, `Say`, `Gather`, `Hangup` and `Pause` verbs.
+    #  - Maximum length is 4096 characters.
+    #  Example:
+    #  ```
+    #  <Say voice="alice" language="en-US">Hello, transferring a customer to you.</Say>
+    #  <Pause length="2"/>
+    #  <Say>They called about billing questions.</Say>
+    #  ```
+    attr_reader :twiml
     # @return [Vapi::SummaryPlan] This is the plan for generating a summary of the call to present to the
     #  destination party.
     #  Usage:
@@ -74,6 +90,9 @@ module Vapi
     #  - `warm-transfer-wait-for-operator-to-speak-first-and-then-say-summary`: The
     #  assistant dials the destination, waits for the operator to speak, provides a
     #  summary of the call to the destination party, and then connects the customer.
+    #  - `warm-transfer-twiml`: The assistant dials the destination, executes the twiml
+    #  instructions on the destination call leg, connects the customer, and leaves the
+    #  call.
     #  @default 'blind-transfer'
     # @param message [Vapi::TransferPlanMessage] This is the message the assistant will deliver to the destination party before
     #  connecting the customer.
@@ -84,6 +103,18 @@ module Vapi
     # @param sip_verb [Hash{String => Object}] This specifies the SIP verb to use while transferring the call.
     #  - 'refer': Uses SIP REFER to transfer the call (default)
     #  - 'bye': Ends current call with SIP BYE
+    # @param twiml [String] This is the TwiML instructions to execute on the destination call leg before
+    #  connecting the customer.
+    #  Usage:
+    #  - Used only when `mode` is `warm-transfer-twiml`.
+    #  - Supports only `Play`, `Say`, `Gather`, `Hangup` and `Pause` verbs.
+    #  - Maximum length is 4096 characters.
+    #  Example:
+    #  ```
+    #  <Say voice="alice" language="en-US">Hello, transferring a customer to you.</Say>
+    #  <Pause length="2"/>
+    #  <Say>They called about billing questions.</Say>
+    #  ```
     # @param summary_plan [Vapi::SummaryPlan] This is the plan for generating a summary of the call to present to the
     #  destination party.
     #  Usage:
@@ -92,16 +123,18 @@ module Vapi
     #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-summary`.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::TransferPlan]
-    def initialize(mode:, message: OMIT, sip_verb: OMIT, summary_plan: OMIT, additional_properties: nil)
+    def initialize(mode:, message: OMIT, sip_verb: OMIT, twiml: OMIT, summary_plan: OMIT, additional_properties: nil)
       @mode = mode
       @message = message if message != OMIT
       @sip_verb = sip_verb if sip_verb != OMIT
+      @twiml = twiml if twiml != OMIT
       @summary_plan = summary_plan if summary_plan != OMIT
       @additional_properties = additional_properties
       @_field_set = {
         "mode": mode,
         "message": message,
         "sipVerb": sip_verb,
+        "twiml": twiml,
         "summaryPlan": summary_plan
       }.reject do |_k, v|
         v == OMIT
@@ -123,6 +156,7 @@ module Vapi
         message = Vapi::TransferPlanMessage.from_json(json_object: message)
       end
       sip_verb = parsed_json["sipVerb"]
+      twiml = parsed_json["twiml"]
       if parsed_json["summaryPlan"].nil?
         summary_plan = nil
       else
@@ -133,6 +167,7 @@ module Vapi
         mode: mode,
         message: message,
         sip_verb: sip_verb,
+        twiml: twiml,
         summary_plan: summary_plan,
         additional_properties: struct
       )
@@ -155,6 +190,7 @@ module Vapi
       obj.mode.is_a?(Vapi::TransferPlanMode) != false || raise("Passed value for field obj.mode is not the expected type, validation failed.")
       obj.message.nil? || Vapi::TransferPlanMessage.validate_raw(obj: obj.message)
       obj.sip_verb&.is_a?(Hash) != false || raise("Passed value for field obj.sip_verb is not the expected type, validation failed.")
+      obj.twiml&.is_a?(String) != false || raise("Passed value for field obj.twiml is not the expected type, validation failed.")
       obj.summary_plan.nil? || Vapi::SummaryPlan.validate_raw(obj: obj.summary_plan)
     end
   end

data/lib/vapi_server_sdk/types/transfer_plan_mode.rb

@@ -20,12 +20,16 @@ module Vapi
   #  - `warm-transfer-wait-for-operator-to-speak-first-and-then-say-summary`: The
   #  assistant dials the destination, waits for the operator to speak, provides a
   #  summary of the call to the destination party, and then connects the customer.
+  #  - `warm-transfer-twiml`: The assistant dials the destination, executes the twiml
+  #  instructions on the destination call leg, connects the customer, and leaves the
+  #  call.
   #  @default 'blind-transfer'
   class TransferPlanMode
     BLIND_TRANSFER = "blind-transfer"
     BLIND_TRANSFER_ADD_SUMMARY_TO_SIP_HEADER = "blind-transfer-add-summary-to-sip-header"
     WARM_TRANSFER_SAY_MESSAGE = "warm-transfer-say-message"
     WARM_TRANSFER_SAY_SUMMARY = "warm-transfer-say-summary"
+    WARM_TRANSFER_TWIML = "warm-transfer-twiml"
     WARM_TRANSFER_WAIT_FOR_OPERATOR_TO_SPEAK_FIRST_AND_THEN_SAY_MESSAGE = "warm-transfer-wait-for-operator-to-speak-first-and-then-say-message"
     WARM_TRANSFER_WAIT_FOR_OPERATOR_TO_SPEAK_FIRST_AND_THEN_SAY_SUMMARY = "warm-transfer-wait-for-operator-to-speak-first-and-then-say-summary"
   end

data/lib/vapi_server_sdk/types/trieve_knowledge_base.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative "trieve_knowledge_base_vector_store_search_plan"
-require_relative "trieve_knowledge_base_vector_store_create_plan"
+require_relative "trieve_knowledge_base_search_plan"
+require_relative "trieve_knowledge_base_create_plan"
 require "ostruct"
 require "json"
 
@@ -9,17 +9,15 @@ module Vapi
   class TrieveKnowledgeBase
     # @return [String] This is the name of the knowledge base.
     attr_reader :name
-    # @return [Vapi::TrieveKnowledgeBaseVectorStoreSearchPlan] This is the plan on how to search the vector store while a call is going on.
-    attr_reader :vector_store_search_plan
-    # @return [Vapi::TrieveKnowledgeBaseVectorStoreCreatePlan] This is the plan if you want us to create a new vector store on your behalf. To
-    #  use an existing vector store from your account, use `vectoreStoreProviderId`
-    attr_reader :vector_store_create_plan
-    # @return [String] This is an vector store that you already have on your account with the provider.
-    #  To create a new vector store, use vectorStoreCreatePlan.
-    #  Usage:
-    #  - To bring your own vector store from Trieve, go to https://trieve.ai
-    #  - Create a dataset, and use the datasetId here.
-    attr_reader :vector_store_provider_id
+    # @return [Vapi::TrieveKnowledgeBaseSearchPlan] This is the searching plan used when searching for relevant chunks from the
+    #  vector store.
+    #  You should configure this if you're running into these issues:
+    #  - Too much unnecessary context is being fed as knowledge base context.
+    #  - Not enough relevant context is being fed as knowledge base context.
+    attr_reader :search_plan
+    # @return [Vapi::TrieveKnowledgeBaseCreatePlan] This is the plan if you want us to create/import a new vector store using
+    #  Trieve.
+    attr_reader :create_plan
     # @return [String] This is the id of the knowledge base.
     attr_reader :id
     # @return [String] This is the org id of the knowledge base.
@@ -33,32 +31,28 @@ module Vapi
     OMIT = Object.new
 
     # @param name [String] This is the name of the knowledge base.
-    # @param vector_store_search_plan [Vapi::TrieveKnowledgeBaseVectorStoreSearchPlan] This is the plan on how to search the vector store while a call is going on.
-    # @param vector_store_create_plan [Vapi::TrieveKnowledgeBaseVectorStoreCreatePlan] This is the plan if you want us to create a new vector store on your behalf. To
-    #  use an existing vector store from your account, use `vectoreStoreProviderId`
-    # @param vector_store_provider_id [String] This is an vector store that you already have on your account with the provider.
-    #  To create a new vector store, use vectorStoreCreatePlan.
-    #  Usage:
-    #  - To bring your own vector store from Trieve, go to https://trieve.ai
-    #  - Create a dataset, and use the datasetId here.
+    # @param search_plan [Vapi::TrieveKnowledgeBaseSearchPlan] This is the searching plan used when searching for relevant chunks from the
+    #  vector store.
+    #  You should configure this if you're running into these issues:
+    #  - Too much unnecessary context is being fed as knowledge base context.
+    #  - Not enough relevant context is being fed as knowledge base context.
+    # @param create_plan [Vapi::TrieveKnowledgeBaseCreatePlan] This is the plan if you want us to create/import a new vector store using
+    #  Trieve.
     # @param id [String] This is the id of the knowledge base.
     # @param org_id [String] This is the org id of the knowledge base.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::TrieveKnowledgeBase]
-    def initialize(vector_store_search_plan:, id:, org_id:, name: OMIT, vector_store_create_plan: OMIT,
-                   vector_store_provider_id: OMIT, additional_properties: nil)
+    def initialize(id:, org_id:, name: OMIT, search_plan: OMIT, create_plan: OMIT, additional_properties: nil)
       @name = name if name != OMIT
-      @vector_store_search_plan = vector_store_search_plan
-      @vector_store_create_plan = vector_store_create_plan if vector_store_create_plan != OMIT
-      @vector_store_provider_id = vector_store_provider_id if vector_store_provider_id != OMIT
+      @search_plan = search_plan if search_plan != OMIT
+      @create_plan = create_plan if create_plan != OMIT
       @id = id
       @org_id = org_id
       @additional_properties = additional_properties
       @_field_set = {
         "name": name,
-        "vectorStoreSearchPlan": vector_store_search_plan,
-        "vectorStoreCreatePlan": vector_store_create_plan,
-        "vectorStoreProviderId": vector_store_provider_id,
+        "searchPlan": search_plan,
+        "createPlan": create_plan,
         "id": id,
         "orgId": org_id
       }.reject do |_k, v|
@@ -74,26 +68,24 @@ module Vapi
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
       name = parsed_json["name"]
-      if parsed_json["vectorStoreSearchPlan"].nil?
-        vector_store_search_plan = nil
+      if parsed_json["searchPlan"].nil?
+        search_plan = nil
       else
-        vector_store_search_plan = parsed_json["vectorStoreSearchPlan"].to_json
-        vector_store_search_plan = Vapi::TrieveKnowledgeBaseVectorStoreSearchPlan.from_json(json_object: vector_store_search_plan)
+        search_plan = parsed_json["searchPlan"].to_json
+        search_plan = Vapi::TrieveKnowledgeBaseSearchPlan.from_json(json_object: search_plan)
       end
-      if parsed_json["vectorStoreCreatePlan"].nil?
-        vector_store_create_plan = nil
+      if parsed_json["createPlan"].nil?
+        create_plan = nil
       else
-        vector_store_create_plan = parsed_json["vectorStoreCreatePlan"].to_json
-        vector_store_create_plan = Vapi::TrieveKnowledgeBaseVectorStoreCreatePlan.from_json(json_object: vector_store_create_plan)
+        create_plan = parsed_json["createPlan"].to_json
+        create_plan = Vapi::TrieveKnowledgeBaseCreatePlan.from_json(json_object: create_plan)
       end
-      vector_store_provider_id = parsed_json["vectorStoreProviderId"]
       id = parsed_json["id"]
       org_id = parsed_json["orgId"]
       new(
         name: name,
-        vector_store_search_plan: vector_store_search_plan,
-        vector_store_create_plan: vector_store_create_plan,
-        vector_store_provider_id: vector_store_provider_id,
+        search_plan: search_plan,
+        create_plan: create_plan,
         id: id,
         org_id: org_id,
         additional_properties: struct
@@ -115,9 +107,8 @@ module Vapi
     # @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.")
-      Vapi::TrieveKnowledgeBaseVectorStoreSearchPlan.validate_raw(obj: obj.vector_store_search_plan)
-      obj.vector_store_create_plan.nil? || Vapi::TrieveKnowledgeBaseVectorStoreCreatePlan.validate_raw(obj: obj.vector_store_create_plan)
-      obj.vector_store_provider_id&.is_a?(String) != false || raise("Passed value for field obj.vector_store_provider_id is not the expected type, validation failed.")
+      obj.search_plan.nil? || Vapi::TrieveKnowledgeBaseSearchPlan.validate_raw(obj: obj.search_plan)
+      obj.create_plan.nil? || Vapi::TrieveKnowledgeBaseCreatePlan.validate_raw(obj: obj.create_plan)
       obj.id.is_a?(String) != false || raise("Passed value for field obj.id is not the expected type, validation failed.")
       obj.org_id.is_a?(String) != false || raise("Passed value for field obj.org_id is not the expected type, validation failed.")
     end

data/lib/vapi_server_sdk/types/{trieve_knowledge_base_vector_store_create_plan.rb → trieve_knowledge_base_chunk_plan.rb}

@@ -4,10 +4,12 @@ require "ostruct"
 require "json"
 
 module Vapi
-  class TrieveKnowledgeBaseVectorStoreCreatePlan
+  class TrieveKnowledgeBaseChunkPlan
     # @return [Array<String>] These are the file ids that will be used to create the vector store. To upload
     #  files, use the `POST /files` endpoint.
     attr_reader :file_ids
+    # @return [Array<String>] These are the websites that will be used to create the vector store.
+    attr_reader :websites
     # @return [Float] This is an optional field which allows you to specify the number of splits you
     #  want per chunk. If not specified, the default 20 is used. However, you may want
     #  to use a different number.
@@ -33,6 +35,7 @@ module Vapi
 
     # @param file_ids [Array<String>] These are the file ids that will be used to create the vector store. To upload
     #  files, use the `POST /files` endpoint.
+    # @param websites [Array<String>] These are the websites that will be used to create the vector store.
     # @param target_splits_per_chunk [Float] This is an optional field which allows you to specify the number of splits you
     #  want per chunk. If not specified, the default 20 is used. However, you may want
     #  to use a different number.
@@ -46,16 +49,18 @@ module Vapi
     #  such that 66 splits with a target_splits_per_chunk of 20 will result in 3 chunks
     #  with 22 splits each.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
-    # @return [Vapi::TrieveKnowledgeBaseVectorStoreCreatePlan]
-    def initialize(file_ids:, target_splits_per_chunk: OMIT, split_delimiters: OMIT, rebalance_chunks: OMIT,
-                   additional_properties: nil)
-      @file_ids = file_ids
+    # @return [Vapi::TrieveKnowledgeBaseChunkPlan]
+    def initialize(file_ids: OMIT, websites: OMIT, target_splits_per_chunk: OMIT, split_delimiters: OMIT,
+                   rebalance_chunks: OMIT, additional_properties: nil)
+      @file_ids = file_ids if file_ids != OMIT
+      @websites = websites if websites != OMIT
       @target_splits_per_chunk = target_splits_per_chunk if target_splits_per_chunk != OMIT
       @split_delimiters = split_delimiters if split_delimiters != OMIT
       @rebalance_chunks = rebalance_chunks if rebalance_chunks != OMIT
       @additional_properties = additional_properties
       @_field_set = {
         "fileIds": file_ids,
+        "websites": websites,
         "targetSplitsPerChunk": target_splits_per_chunk,
         "splitDelimiters": split_delimiters,
         "rebalanceChunks": rebalance_chunks
@@ -64,20 +69,21 @@ module Vapi
       end
     end
 
-    # Deserialize a JSON object to an instance of
-    #  TrieveKnowledgeBaseVectorStoreCreatePlan
+    # Deserialize a JSON object to an instance of TrieveKnowledgeBaseChunkPlan
     #
     # @param json_object [String]
-    # @return [Vapi::TrieveKnowledgeBaseVectorStoreCreatePlan]
+    # @return [Vapi::TrieveKnowledgeBaseChunkPlan]
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
       file_ids = parsed_json["fileIds"]
+      websites = parsed_json["websites"]
       target_splits_per_chunk = parsed_json["targetSplitsPerChunk"]
       split_delimiters = parsed_json["splitDelimiters"]
       rebalance_chunks = parsed_json["rebalanceChunks"]
       new(
         file_ids: file_ids,
+        websites: websites,
         target_splits_per_chunk: target_splits_per_chunk,
         split_delimiters: split_delimiters,
         rebalance_chunks: rebalance_chunks,
@@ -85,8 +91,7 @@ module Vapi
       )
     end
 
-    # Serialize an instance of TrieveKnowledgeBaseVectorStoreCreatePlan to a JSON
-    #  object
+    # Serialize an instance of TrieveKnowledgeBaseChunkPlan to a JSON object
     #
     # @return [String]
     def to_json(*_args)
@@ -100,7 +105,8 @@ module Vapi
     # @param obj [Object]
     # @return [Void]
     def self.validate_raw(obj:)
-      obj.file_ids.is_a?(Array) != false || raise("Passed value for field obj.file_ids is not the expected type, validation failed.")
+      obj.file_ids&.is_a?(Array) != false || raise("Passed value for field obj.file_ids is not the expected type, validation failed.")
+      obj.websites&.is_a?(Array) != false || raise("Passed value for field obj.websites is not the expected type, validation failed.")
       obj.target_splits_per_chunk&.is_a?(Float) != false || raise("Passed value for field obj.target_splits_per_chunk is not the expected type, validation failed.")
       obj.split_delimiters&.is_a?(Array) != false || raise("Passed value for field obj.split_delimiters is not the expected type, validation failed.")
       obj.rebalance_chunks&.is_a?(Boolean) != false || raise("Passed value for field obj.rebalance_chunks is not the expected type, validation failed.")

data/lib/vapi_server_sdk/types/trieve_knowledge_base_create.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require_relative "trieve_knowledge_base_chunk_plan"
+require "ostruct"
+require "json"
+
+module Vapi
+  class TrieveKnowledgeBaseCreate
+    # @return [Array<Vapi::TrieveKnowledgeBaseChunkPlan>] These are the chunk plans used to create the dataset.
+    attr_reader :chunk_plans
+    # @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 chunk_plans [Array<Vapi::TrieveKnowledgeBaseChunkPlan>] These are the chunk plans used to create the dataset.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vapi::TrieveKnowledgeBaseCreate]
+    def initialize(chunk_plans:, additional_properties: nil)
+      @chunk_plans = chunk_plans
+      @additional_properties = additional_properties
+      @_field_set = { "chunkPlans": chunk_plans }
+    end
+
+    # Deserialize a JSON object to an instance of TrieveKnowledgeBaseCreate
+    #
+    # @param json_object [String]
+    # @return [Vapi::TrieveKnowledgeBaseCreate]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      chunk_plans = parsed_json["chunkPlans"]&.map do |item|
+        item = item.to_json
+        Vapi::TrieveKnowledgeBaseChunkPlan.from_json(json_object: item)
+      end
+      new(chunk_plans: chunk_plans, additional_properties: struct)
+    end
+
+    # Serialize an instance of TrieveKnowledgeBaseCreate to a JSON object
+    #
+    # @return [String]
+    def to_json(*_args)
+      @_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.chunk_plans.is_a?(Array) != false || raise("Passed value for field obj.chunk_plans is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vapi_server_sdk/types/trieve_knowledge_base_create_plan.rb

@@ -0,0 +1,96 @@
+# frozen_string_literal: true
+
+require "json"
+require_relative "trieve_knowledge_base_create"
+require_relative "trieve_knowledge_base_import"
+
+module Vapi
+  # This is the plan if you want us to create/import a new vector store using
+  #  Trieve.
+  class TrieveKnowledgeBaseCreatePlan
+    # @return [Object]
+    attr_reader :member
+    # @return [String]
+    attr_reader :discriminant
+
+    private_class_method :new
+    alias kind_of? is_a?
+
+    # @param member [Object]
+    # @param discriminant [String]
+    # @return [Vapi::TrieveKnowledgeBaseCreatePlan]
+    def initialize(member:, discriminant:)
+      @member = member
+      @discriminant = discriminant
+    end
+
+    # Deserialize a JSON object to an instance of TrieveKnowledgeBaseCreatePlan
+    #
+    # @param json_object [String]
+    # @return [Vapi::TrieveKnowledgeBaseCreatePlan]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      member = case struct.type
+               when "create"
+                 Vapi::TrieveKnowledgeBaseCreate.from_json(json_object: json_object)
+               when "import"
+                 Vapi::TrieveKnowledgeBaseImport.from_json(json_object: json_object)
+               else
+                 Vapi::TrieveKnowledgeBaseCreate.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 [String]
+    def to_json(*_args)
+      case @discriminant
+      when "create"
+        { **@member.to_json, type: @discriminant }.to_json
+      when "import"
+        { **@member.to_json, type: @discriminant }.to_json
+      else
+        { "type": @discriminant, value: @member }.to_json
+      end
+      @member.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:)
+      case obj.type
+      when "create"
+        Vapi::TrieveKnowledgeBaseCreate.validate_raw(obj: obj)
+      when "import"
+        Vapi::TrieveKnowledgeBaseImport.validate_raw(obj: obj)
+      else
+        raise("Passed value matched no type within the union, validation failed.")
+      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 [Vapi::TrieveKnowledgeBaseCreate]
+    # @return [Vapi::TrieveKnowledgeBaseCreatePlan]
+    def self.create(member:)
+      new(member: member, discriminant: "create")
+    end
+
+    # @param member [Vapi::TrieveKnowledgeBaseImport]
+    # @return [Vapi::TrieveKnowledgeBaseCreatePlan]
+    def self.import(member:)
+      new(member: member, discriminant: "import")
+    end
+  end
+end

data/lib/vapi_server_sdk/types/trieve_knowledge_base_import.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "ostruct"
+require "json"
+
+module Vapi
+  class TrieveKnowledgeBaseImport
+    # @return [String] This is the `datasetId` of the dataset on your Trieve account.
+    attr_reader :provider_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 provider_id [String] This is the `datasetId` of the dataset on your Trieve account.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vapi::TrieveKnowledgeBaseImport]
+    def initialize(provider_id:, additional_properties: nil)
+      @provider_id = provider_id
+      @additional_properties = additional_properties
+      @_field_set = { "providerId": provider_id }
+    end
+
+    # Deserialize a JSON object to an instance of TrieveKnowledgeBaseImport
+    #
+    # @param json_object [String]
+    # @return [Vapi::TrieveKnowledgeBaseImport]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      provider_id = parsed_json["providerId"]
+      new(provider_id: provider_id, additional_properties: struct)
+    end
+
+    # Serialize an instance of TrieveKnowledgeBaseImport to a JSON object
+    #
+    # @return [String]
+    def to_json(*_args)
+      @_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.provider_id.is_a?(String) != false || raise("Passed value for field obj.provider_id is not the expected type, validation failed.")
+    end
+  end
+end