vapi_server_sdk 0.4.0 → 0.9.1

data/lib/vapi_server_sdk/types/transfer_plan.rb

@@ -3,6 +3,7 @@
 require_relative "transfer_plan_mode"
 require_relative "transfer_plan_message"
 require_relative "summary_plan"
+require_relative "transfer_fallback_plan"
 require "ostruct"
 require "json"
 
@@ -27,26 +28,84 @@ 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.
+    #  - `warm-transfer-experimental`: The assistant puts the customer on hold, dials
+    #  the destination, and if the destination answers (and is human), delivers a
+    #  message or summary before connecting the customer. If the destination is
+    #  unreachable or not human (e.g., with voicemail detection), the assistant
+    #  delivers the `fallbackMessage` to the customer and optionally ends the call.
     #  @default 'blind-transfer'
     attr_reader :mode
     # @return [Vapi::TransferPlanMessage] This is the message the assistant will deliver to the destination party before
     #  connecting the customer.
     #  Usage:
     #  - Used only when `mode` is `blind-transfer-add-summary-to-sip-header`,
-    #  `warm-transfer-say-message` or
-    #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-message`.
+    #  `warm-transfer-say-message`,
+    #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-message`, or
+    #  `warm-transfer-experimental`.
     attr_reader :message
+    # @return [Float] This is the timeout in seconds for the
+    #  warm-transfer-wait-for-operator-to-speak-first-and-then-say-message/summary
+    #  @default 60
+    attr_reader :timeout
     # @return [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
+    #  - 'dial': Uses SIP DIAL to transfer the call
     attr_reader :sip_verb
+    # @return [String] This is the URL to an audio file played while the customer is on hold during
+    #  transfer.
+    #  Usage:
+    #  - Used only when `mode` is `warm-transfer-experimental`.
+    #  - Used when transferring calls to play hold audio for the customer.
+    #  - Must be a publicly accessible URL to an audio file.
+    #  - Supported formats: MP3 and WAV.
+    #  - If not provided, the default hold audio will be used.
+    attr_reader :hold_audio_url
+    # @return [String] This is the URL to an audio file played after the warm transfer message or
+    #  summary is delivered to the destination party.
+    #  It can be used to play a custom sound like 'beep' to notify that the transfer is
+    #  complete.
+    #  Usage:
+    #  - Used only when `mode` is `warm-transfer-experimental`.
+    #  - Used when transferring calls to play hold audio for the destination party.
+    #  - Must be a publicly accessible URL to an audio file.
+    #  - Supported formats: MP3 and WAV.
+    attr_reader :transfer_complete_audio_url
+    # @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:
     #  - Used only when `mode` is `blind-transfer-add-summary-to-sip-header` or
     #  `warm-transfer-say-summary` or
-    #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-summary`.
+    #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-summary` or
+    #  `warm-transfer-experimental`.
     attr_reader :summary_plan
+    # @return [Boolean] This flag includes the sipHeaders from above in the refer to sip uri as url
+    #  encoded query params.
+    #  @default false
+    attr_reader :sip_headers_in_refer_to_enabled
+    # @return [Vapi::TransferFallbackPlan] This configures the fallback plan when the transfer fails (destination
+    #  unreachable, busy, or not human).
+    #  Usage:
+    #  - Used only when `mode` is `warm-transfer-experimental`.
+    #  - If not provided when using `warm-transfer-experimental`, a default message
+    #  will be used.
+    attr_reader :fallback_plan
     # @return [OpenStruct] Additional properties unmapped to the current class definition
     attr_reader :additional_properties
     # @return [Object]
@@ -74,35 +133,100 @@ 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.
+    #  - `warm-transfer-experimental`: The assistant puts the customer on hold, dials
+    #  the destination, and if the destination answers (and is human), delivers a
+    #  message or summary before connecting the customer. If the destination is
+    #  unreachable or not human (e.g., with voicemail detection), the assistant
+    #  delivers the `fallbackMessage` to the customer and optionally ends 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.
     #  Usage:
     #  - Used only when `mode` is `blind-transfer-add-summary-to-sip-header`,
-    #  `warm-transfer-say-message` or
-    #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-message`.
+    #  `warm-transfer-say-message`,
+    #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-message`, or
+    #  `warm-transfer-experimental`.
+    # @param timeout [Float] This is the timeout in seconds for the
+    #  warm-transfer-wait-for-operator-to-speak-first-and-then-say-message/summary
+    #  @default 60
     # @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
+    #  - 'dial': Uses SIP DIAL to transfer the call
+    # @param hold_audio_url [String] This is the URL to an audio file played while the customer is on hold during
+    #  transfer.
+    #  Usage:
+    #  - Used only when `mode` is `warm-transfer-experimental`.
+    #  - Used when transferring calls to play hold audio for the customer.
+    #  - Must be a publicly accessible URL to an audio file.
+    #  - Supported formats: MP3 and WAV.
+    #  - If not provided, the default hold audio will be used.
+    # @param transfer_complete_audio_url [String] This is the URL to an audio file played after the warm transfer message or
+    #  summary is delivered to the destination party.
+    #  It can be used to play a custom sound like 'beep' to notify that the transfer is
+    #  complete.
+    #  Usage:
+    #  - Used only when `mode` is `warm-transfer-experimental`.
+    #  - Used when transferring calls to play hold audio for the destination party.
+    #  - Must be a publicly accessible URL to an audio file.
+    #  - Supported formats: MP3 and WAV.
+    # @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:
     #  - Used only when `mode` is `blind-transfer-add-summary-to-sip-header` or
     #  `warm-transfer-say-summary` or
-    #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-summary`.
+    #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-summary` or
+    #  `warm-transfer-experimental`.
+    # @param sip_headers_in_refer_to_enabled [Boolean] This flag includes the sipHeaders from above in the refer to sip uri as url
+    #  encoded query params.
+    #  @default false
+    # @param fallback_plan [Vapi::TransferFallbackPlan] This configures the fallback plan when the transfer fails (destination
+    #  unreachable, busy, or not human).
+    #  Usage:
+    #  - Used only when `mode` is `warm-transfer-experimental`.
+    #  - If not provided when using `warm-transfer-experimental`, a default message
+    #  will be used.
     # @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, timeout: OMIT, sip_verb: OMIT, hold_audio_url: OMIT,
+                   transfer_complete_audio_url: OMIT, twiml: OMIT, summary_plan: OMIT, sip_headers_in_refer_to_enabled: OMIT, fallback_plan: OMIT, additional_properties: nil)
       @mode = mode
       @message = message if message != OMIT
+      @timeout = timeout if timeout != OMIT
       @sip_verb = sip_verb if sip_verb != OMIT
+      @hold_audio_url = hold_audio_url if hold_audio_url != OMIT
+      @transfer_complete_audio_url = transfer_complete_audio_url if transfer_complete_audio_url != OMIT
+      @twiml = twiml if twiml != OMIT
       @summary_plan = summary_plan if summary_plan != OMIT
+      @sip_headers_in_refer_to_enabled = sip_headers_in_refer_to_enabled if sip_headers_in_refer_to_enabled != OMIT
+      @fallback_plan = fallback_plan if fallback_plan != OMIT
       @additional_properties = additional_properties
       @_field_set = {
         "mode": mode,
         "message": message,
+        "timeout": timeout,
         "sipVerb": sip_verb,
-        "summaryPlan": summary_plan
+        "holdAudioUrl": hold_audio_url,
+        "transferCompleteAudioUrl": transfer_complete_audio_url,
+        "twiml": twiml,
+        "summaryPlan": summary_plan,
+        "sipHeadersInReferToEnabled": sip_headers_in_refer_to_enabled,
+        "fallbackPlan": fallback_plan
       }.reject do |_k, v|
         v == OMIT
       end
@@ -122,18 +246,35 @@ module Vapi
         message = parsed_json["message"].to_json
         message = Vapi::TransferPlanMessage.from_json(json_object: message)
       end
+      timeout = parsed_json["timeout"]
       sip_verb = parsed_json["sipVerb"]
+      hold_audio_url = parsed_json["holdAudioUrl"]
+      transfer_complete_audio_url = parsed_json["transferCompleteAudioUrl"]
+      twiml = parsed_json["twiml"]
       if parsed_json["summaryPlan"].nil?
         summary_plan = nil
       else
         summary_plan = parsed_json["summaryPlan"].to_json
         summary_plan = Vapi::SummaryPlan.from_json(json_object: summary_plan)
       end
+      sip_headers_in_refer_to_enabled = parsed_json["sipHeadersInReferToEnabled"]
+      if parsed_json["fallbackPlan"].nil?
+        fallback_plan = nil
+      else
+        fallback_plan = parsed_json["fallbackPlan"].to_json
+        fallback_plan = Vapi::TransferFallbackPlan.from_json(json_object: fallback_plan)
+      end
       new(
         mode: mode,
         message: message,
+        timeout: timeout,
         sip_verb: sip_verb,
+        hold_audio_url: hold_audio_url,
+        transfer_complete_audio_url: transfer_complete_audio_url,
+        twiml: twiml,
         summary_plan: summary_plan,
+        sip_headers_in_refer_to_enabled: sip_headers_in_refer_to_enabled,
+        fallback_plan: fallback_plan,
         additional_properties: struct
       )
     end
@@ -154,8 +295,14 @@ module Vapi
     def self.validate_raw(obj:)
       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.timeout&.is_a?(Float) != false || raise("Passed value for field obj.timeout is not the expected type, validation failed.")
       obj.sip_verb&.is_a?(Hash) != false || raise("Passed value for field obj.sip_verb is not the expected type, validation failed.")
+      obj.hold_audio_url&.is_a?(String) != false || raise("Passed value for field obj.hold_audio_url is not the expected type, validation failed.")
+      obj.transfer_complete_audio_url&.is_a?(String) != false || raise("Passed value for field obj.transfer_complete_audio_url 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)
+      obj.sip_headers_in_refer_to_enabled&.is_a?(Boolean) != false || raise("Passed value for field obj.sip_headers_in_refer_to_enabled is not the expected type, validation failed.")
+      obj.fallback_plan.nil? || Vapi::TransferFallbackPlan.validate_raw(obj: obj.fallback_plan)
     end
   end
 end

data/lib/vapi_server_sdk/types/transfer_plan_message.rb

@@ -8,8 +8,9 @@ module Vapi
   #  connecting the customer.
   #  Usage:
   #  - Used only when `mode` is `blind-transfer-add-summary-to-sip-header`,
-  #  `warm-transfer-say-message` or
-  #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-message`.
+  #  `warm-transfer-say-message`,
+  #  `warm-transfer-wait-for-operator-to-speak-first-and-then-say-message`, or
+  #  `warm-transfer-experimental`.
   class TransferPlanMessage
     # Deserialize a JSON object to an instance of TransferPlanMessage
     #

data/lib/vapi_server_sdk/types/transfer_plan_mode.rb

@@ -20,13 +20,23 @@ 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.
+  #  - `warm-transfer-experimental`: The assistant puts the customer on hold, dials
+  #  the destination, and if the destination answers (and is human), delivers a
+  #  message or summary before connecting the customer. If the destination is
+  #  unreachable or not human (e.g., with voicemail detection), the assistant
+  #  delivers the `fallbackMessage` to the customer and optionally ends 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"
+    WARM_TRANSFER_EXPERIMENTAL = "warm-transfer-experimental"
   end
 end

data/lib/vapi_server_sdk/types/trieve_credential.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+
+require "date"
+require "ostruct"
+require "json"
+
+module Vapi
+  class TrieveCredential
+    # @return [String]
+    attr_reader :provider
+    # @return [String] This is not returned in the API.
+    attr_reader :api_key
+    # @return [String] This is the unique identifier for the credential.
+    attr_reader :id
+    # @return [String] This is the unique identifier for the org that this credential belongs to.
+    attr_reader :org_id
+    # @return [DateTime] This is the ISO 8601 date-time string of when the credential was created.
+    attr_reader :created_at
+    # @return [DateTime] This is the ISO 8601 date-time string of when the assistant was last updated.
+    attr_reader :updated_at
+    # @return [String] This is the name of credential. This is just for your reference.
+    attr_reader :name
+    # @return [OpenStruct] Additional properties unmapped to the current class definition
+    attr_reader :additional_properties
+    # @return [Object]
+    attr_reader :_field_set
+    protected :_field_set
+
+    OMIT = Object.new
+
+    # @param provider [String]
+    # @param api_key [String] This is not returned in the API.
+    # @param id [String] This is the unique identifier for the credential.
+    # @param org_id [String] This is the unique identifier for the org that this credential belongs to.
+    # @param created_at [DateTime] This is the ISO 8601 date-time string of when the credential was created.
+    # @param updated_at [DateTime] This is the ISO 8601 date-time string of when the assistant was last updated.
+    # @param name [String] This is the name of credential. This is just for your reference.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vapi::TrieveCredential]
+    def initialize(provider:, api_key:, id:, org_id:, created_at:, updated_at:, name: OMIT, additional_properties: nil)
+      @provider = provider
+      @api_key = api_key
+      @id = id
+      @org_id = org_id
+      @created_at = created_at
+      @updated_at = updated_at
+      @name = name if name != OMIT
+      @additional_properties = additional_properties
+      @_field_set = {
+        "provider": provider,
+        "apiKey": api_key,
+        "id": id,
+        "orgId": org_id,
+        "createdAt": created_at,
+        "updatedAt": updated_at,
+        "name": name
+      }.reject do |_k, v|
+        v == OMIT
+      end
+    end
+
+    # Deserialize a JSON object to an instance of TrieveCredential
+    #
+    # @param json_object [String]
+    # @return [Vapi::TrieveCredential]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      provider = parsed_json["provider"]
+      api_key = parsed_json["apiKey"]
+      id = parsed_json["id"]
+      org_id = parsed_json["orgId"]
+      created_at = (DateTime.parse(parsed_json["createdAt"]) unless parsed_json["createdAt"].nil?)
+      updated_at = (DateTime.parse(parsed_json["updatedAt"]) unless parsed_json["updatedAt"].nil?)
+      name = parsed_json["name"]
+      new(
+        provider: provider,
+        api_key: api_key,
+        id: id,
+        org_id: org_id,
+        created_at: created_at,
+        updated_at: updated_at,
+        name: name,
+        additional_properties: struct
+      )
+    end
+
+    # Serialize an instance of TrieveCredential 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.is_a?(String) != false || raise("Passed value for field obj.provider is not the expected type, validation failed.")
+      obj.api_key.is_a?(String) != false || raise("Passed value for field obj.api_key is not the expected type, validation failed.")
+      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.")
+      obj.created_at.is_a?(DateTime) != false || raise("Passed value for field obj.created_at is not the expected type, validation failed.")
+      obj.updated_at.is_a?(DateTime) != false || raise("Passed value for field obj.updated_at is not the expected type, validation failed.")
+      obj.name&.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vapi_server_sdk/types/trieve_knowledge_base.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 require_relative "trieve_knowledge_base_search_plan"
-require_relative "trieve_knowledge_base_create_plan"
+require_relative "trieve_knowledge_base_import"
 require "ostruct"
 require "json"
 
@@ -15,7 +15,7 @@ module Vapi
     #  - 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
+    # @return [Vapi::TrieveKnowledgeBaseImport] 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.
@@ -36,7 +36,7 @@ module Vapi
     #  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
+    # @param create_plan [Vapi::TrieveKnowledgeBaseImport] 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.
@@ -78,7 +78,7 @@ module Vapi
         create_plan = nil
       else
         create_plan = parsed_json["createPlan"].to_json
-        create_plan = Vapi::TrieveKnowledgeBaseCreatePlan.from_json(json_object: create_plan)
+        create_plan = Vapi::TrieveKnowledgeBaseImport.from_json(json_object: create_plan)
       end
       id = parsed_json["id"]
       org_id = parsed_json["orgId"]
@@ -108,7 +108,7 @@ module Vapi
     def self.validate_raw(obj:)
       obj.name&.is_a?(String) != false || raise("Passed value for field obj.name is not the expected type, validation failed.")
       obj.search_plan.nil? || Vapi::TrieveKnowledgeBaseSearchPlan.validate_raw(obj: obj.search_plan)
-      obj.create_plan.nil? || Vapi::TrieveKnowledgeBaseCreatePlan.validate_raw(obj: obj.create_plan)
+      obj.create_plan.nil? || Vapi::TrieveKnowledgeBaseImport.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_create.rb

@@ -6,6 +6,8 @@ require "json"
 
 module Vapi
   class TrieveKnowledgeBaseCreate
+    # @return [String] This is to create a new dataset on Trieve.
+    attr_reader :type
     # @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
@@ -16,13 +18,15 @@ module Vapi
 
     OMIT = Object.new
 
+    # @param type [String] This is to create a new dataset on Trieve.
     # @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)
+    def initialize(type:, chunk_plans:, additional_properties: nil)
+      @type = type
       @chunk_plans = chunk_plans
       @additional_properties = additional_properties
-      @_field_set = { "chunkPlans": chunk_plans }
+      @_field_set = { "type": type, "chunkPlans": chunk_plans }
     end
 
     # Deserialize a JSON object to an instance of TrieveKnowledgeBaseCreate
@@ -32,11 +36,16 @@ module Vapi
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
+      type = parsed_json["type"]
       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)
+      new(
+        type: type,
+        chunk_plans: chunk_plans,
+        additional_properties: struct
+      )
     end
 
     # Serialize an instance of TrieveKnowledgeBaseCreate to a JSON object
@@ -53,6 +62,7 @@ module Vapi
     # @param obj [Object]
     # @return [Void]
     def self.validate_raw(obj:)
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
       obj.chunk_plans.is_a?(Array) != false || raise("Passed value for field obj.chunk_plans is not the expected type, validation failed.")
     end
   end

data/lib/vapi_server_sdk/types/trieve_knowledge_base_import.rb

@@ -5,6 +5,8 @@ require "json"
 
 module Vapi
   class TrieveKnowledgeBaseImport
+    # @return [String] This is to import an existing dataset from Trieve.
+    attr_reader :type
     # @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
@@ -15,13 +17,15 @@ module Vapi
 
     OMIT = Object.new
 
+    # @param type [String] This is to import an existing dataset from Trieve.
     # @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)
+    def initialize(type:, provider_id:, additional_properties: nil)
+      @type = type
       @provider_id = provider_id
       @additional_properties = additional_properties
-      @_field_set = { "providerId": provider_id }
+      @_field_set = { "type": type, "providerId": provider_id }
     end
 
     # Deserialize a JSON object to an instance of TrieveKnowledgeBaseImport
@@ -31,8 +35,13 @@ module Vapi
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
+      type = parsed_json["type"]
       provider_id = parsed_json["providerId"]
-      new(provider_id: provider_id, additional_properties: struct)
+      new(
+        type: type,
+        provider_id: provider_id,
+        additional_properties: struct
+      )
     end
 
     # Serialize an instance of TrieveKnowledgeBaseImport to a JSON object
@@ -49,6 +58,7 @@ module Vapi
     # @param obj [Object]
     # @return [Void]
     def self.validate_raw(obj:)
+      obj.type.is_a?(String) != false || raise("Passed value for field obj.type is not the expected type, validation failed.")
       obj.provider_id.is_a?(String) != false || raise("Passed value for field obj.provider_id is not the expected type, validation failed.")
     end
   end

data/lib/vapi_server_sdk/types/trieve_knowledge_base_search_plan.rb

@@ -6,6 +6,9 @@ require "json"
 
 module Vapi
   class TrieveKnowledgeBaseSearchPlan
+    # @return [Float] Specifies the number of top chunks to return. This corresponds to the
+    #  `page_size` parameter in Trieve.
+    attr_reader :top_k
     # @return [Boolean] If true, stop words (specified in server/src/stop-words.txt in the git repo)
     #  will be removed. This will preserve queries that are entirely stop words.
     attr_reader :remove_stop_words
@@ -27,6 +30,8 @@ module Vapi
 
     OMIT = Object.new
 
+    # @param top_k [Float] Specifies the number of top chunks to return. This corresponds to the
+    #  `page_size` parameter in Trieve.
     # @param remove_stop_words [Boolean] If true, stop words (specified in server/src/stop-words.txt in the git repo)
     #  will be removed. This will preserve queries that are entirely stop words.
     # @param score_threshold [Float] This is the score threshold to filter out chunks with a score below the
@@ -39,12 +44,15 @@ module Vapi
     #  vector store.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::TrieveKnowledgeBaseSearchPlan]
-    def initialize(search_type:, remove_stop_words: OMIT, score_threshold: OMIT, additional_properties: nil)
+    def initialize(search_type:, top_k: OMIT, remove_stop_words: OMIT, score_threshold: OMIT,
+                   additional_properties: nil)
+      @top_k = top_k if top_k != OMIT
       @remove_stop_words = remove_stop_words if remove_stop_words != OMIT
       @score_threshold = score_threshold if score_threshold != OMIT
       @search_type = search_type
       @additional_properties = additional_properties
       @_field_set = {
+        "topK": top_k,
         "removeStopWords": remove_stop_words,
         "scoreThreshold": score_threshold,
         "searchType": search_type
@@ -60,10 +68,12 @@ module Vapi
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
+      top_k = parsed_json["topK"]
       remove_stop_words = parsed_json["removeStopWords"]
       score_threshold = parsed_json["scoreThreshold"]
       search_type = parsed_json["searchType"]
       new(
+        top_k: top_k,
         remove_stop_words: remove_stop_words,
         score_threshold: score_threshold,
         search_type: search_type,
@@ -85,6 +95,7 @@ module Vapi
     # @param obj [Object]
     # @return [Void]
     def self.validate_raw(obj:)
+      obj.top_k&.is_a?(Float) != false || raise("Passed value for field obj.top_k is not the expected type, validation failed.")
       obj.remove_stop_words&.is_a?(Boolean) != false || raise("Passed value for field obj.remove_stop_words is not the expected type, validation failed.")
       obj.score_threshold&.is_a?(Float) != false || raise("Passed value for field obj.score_threshold is not the expected type, validation failed.")
       obj.search_type.is_a?(Vapi::TrieveKnowledgeBaseSearchPlanSearchType) != false || raise("Passed value for field obj.search_type is not the expected type, validation failed.")