vapi_server_sdk 0.1.0 → 0.3.0

data/lib/vapi_server_sdk/types/open_ai_function.rb

@@ -6,11 +6,20 @@ require "json"
 
 module Vapi
   class OpenAiFunction
+    # @return [Boolean] This is a boolean that controls whether to enable strict schema adherence when
+    #  generating the function call. If set to true, the model will follow the exact
+    #  schema defined in the parameters field. Only a subset of JSON Schema is
+    #  supported when strict is true. Learn more about Structured Outputs in the
+    #  [OpenAI
+    #  guide](https://openai.com/index/introducing-structured-outputs-in-the-api/).
+    #  @default false
+    attr_reader :strict
     # @return [String] This is the the name of the function to be called.
     #  Must be a-z, A-Z, 0-9, or contain underscores and dashes, with a maximum length
     #  of 64.
     attr_reader :name
-    # @return [String]
+    # @return [String] This is the description of what the function does, used by the AI to choose when
+    #  and how to call the function.
     attr_reader :description
     # @return [Vapi::OpenAiFunctionParameters] These are the parameters the functions accepts, described as a JSON Schema
     #  object.
@@ -28,10 +37,18 @@ module Vapi
 
     OMIT = Object.new
 
+    # @param strict [Boolean] This is a boolean that controls whether to enable strict schema adherence when
+    #  generating the function call. If set to true, the model will follow the exact
+    #  schema defined in the parameters field. Only a subset of JSON Schema is
+    #  supported when strict is true. Learn more about Structured Outputs in the
+    #  [OpenAI
+    #  guide](https://openai.com/index/introducing-structured-outputs-in-the-api/).
+    #  @default false
     # @param name [String] This is the the name of the function to be called.
     #  Must be a-z, A-Z, 0-9, or contain underscores and dashes, with a maximum length
     #  of 64.
-    # @param description [String]
+    # @param description [String] This is the description of what the function does, used by the AI to choose when
+    #  and how to call the function.
     # @param parameters [Vapi::OpenAiFunctionParameters] These are the parameters the functions accepts, described as a JSON Schema
     #  object.
     #  See the [OpenAI guide](https://platform.openai.com/docs/guides/function-calling)
@@ -41,12 +58,18 @@ module Vapi
     #  Omitting parameters defines a function with an empty parameter list.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::OpenAiFunction]
-    def initialize(name:, description: OMIT, parameters: OMIT, additional_properties: nil)
+    def initialize(name:, strict: OMIT, description: OMIT, parameters: OMIT, additional_properties: nil)
+      @strict = strict if strict != OMIT
       @name = name
       @description = description if description != OMIT
       @parameters = parameters if parameters != OMIT
       @additional_properties = additional_properties
-      @_field_set = { "name": name, "description": description, "parameters": parameters }.reject do |_k, v|
+      @_field_set = {
+        "strict": strict,
+        "name": name,
+        "description": description,
+        "parameters": parameters
+      }.reject do |_k, v|
         v == OMIT
       end
     end
@@ -58,6 +81,7 @@ module Vapi
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
+      strict = parsed_json["strict"]
       name = parsed_json["name"]
       description = parsed_json["description"]
       if parsed_json["parameters"].nil?
@@ -67,6 +91,7 @@ module Vapi
         parameters = Vapi::OpenAiFunctionParameters.from_json(json_object: parameters)
       end
       new(
+        strict: strict,
         name: name,
         description: description,
         parameters: parameters,
@@ -88,6 +113,7 @@ module Vapi
     # @param obj [Object]
     # @return [Void]
     def self.validate_raw(obj:)
+      obj.strict&.is_a?(Boolean) != false || raise("Passed value for field obj.strict 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.")
       obj.description&.is_a?(String) != false || raise("Passed value for field obj.description is not the expected type, validation failed.")
       obj.parameters.nil? || Vapi::OpenAiFunctionParameters.validate_raw(obj: obj.parameters)

data/lib/vapi_server_sdk/types/open_ai_model.rb

@@ -2,9 +2,9 @@
 
 require_relative "open_ai_message"
 require_relative "open_ai_model_tools_item"
+require_relative "create_custom_knowledge_base_dto"
 require_relative "open_ai_model_model"
 require_relative "open_ai_model_fallback_models_item"
-require_relative "knowledge_base"
 require "ostruct"
 require "json"
 
@@ -20,6 +20,10 @@ module Vapi
     #  tools, use `tools`.
     #  Both `tools` and `toolIds` can be used together.
     attr_reader :tool_ids
+    # @return [Vapi::CreateCustomKnowledgeBaseDto] These are the options for the knowledge base.
+    attr_reader :knowledge_base
+    # @return [String] This is the ID of the knowledge base the model will use.
+    attr_reader :knowledge_base_id
     # @return [Vapi::OpenAiModelModel] This is the OpenAI model that will be used.
     attr_reader :model
     # @return [Array<Vapi::OpenAiModelFallbackModelsItem>] These are the fallback models that will be used if the primary model fails. This
@@ -31,8 +35,6 @@ module Vapi
     # @return [Float] This is the temperature that will be used for calls. Default is 0 to leverage
     #  caching for lower latency.
     attr_reader :temperature
-    # @return [Vapi::KnowledgeBase] These are the options for the knowledge base.
-    attr_reader :knowledge_base
     # @return [Float] This is the max number of tokens that the assistant will be allowed to generate
     #  in each turn of the conversation. Default is 250.
     attr_reader :max_tokens
@@ -63,6 +65,8 @@ module Vapi
     # @param tool_ids [Array<String>] These are the tools that the assistant can use during the call. To use transient
     #  tools, use `tools`.
     #  Both `tools` and `toolIds` can be used together.
+    # @param knowledge_base [Vapi::CreateCustomKnowledgeBaseDto] These are the options for the knowledge base.
+    # @param knowledge_base_id [String] This is the ID of the knowledge base the model will use.
     # @param model [Vapi::OpenAiModelModel] This is the OpenAI model that will be used.
     # @param fallback_models [Array<Vapi::OpenAiModelFallbackModelsItem>] These are the fallback models that will be used if the primary model fails. This
     #  shouldn't be specified unless you have a specific reason to do so. Vapi will
@@ -70,7 +74,6 @@ module Vapi
     # @param semantic_caching_enabled [Boolean]
     # @param temperature [Float] This is the temperature that will be used for calls. Default is 0 to leverage
     #  caching for lower latency.
-    # @param knowledge_base [Vapi::KnowledgeBase] These are the options for the knowledge base.
     # @param max_tokens [Float] This is the max number of tokens that the assistant will be allowed to generate
     #  in each turn of the conversation. Default is 250.
     # @param emotion_recognition_enabled [Boolean] This determines whether we detect user's emotion while they speak and send it as
@@ -85,16 +88,17 @@ module Vapi
     #  @default 0
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::OpenAiModel]
-    def initialize(model:, messages: OMIT, tools: OMIT, tool_ids: OMIT, fallback_models: OMIT,
-                   semantic_caching_enabled: OMIT, temperature: OMIT, knowledge_base: OMIT, max_tokens: OMIT, emotion_recognition_enabled: OMIT, num_fast_turns: OMIT, additional_properties: nil)
+    def initialize(model:, messages: OMIT, tools: OMIT, tool_ids: OMIT, knowledge_base: OMIT, knowledge_base_id: OMIT,
+                   fallback_models: OMIT, semantic_caching_enabled: OMIT, temperature: OMIT, max_tokens: OMIT, emotion_recognition_enabled: OMIT, num_fast_turns: OMIT, additional_properties: nil)
       @messages = messages if messages != OMIT
       @tools = tools if tools != OMIT
       @tool_ids = tool_ids if tool_ids != OMIT
+      @knowledge_base = knowledge_base if knowledge_base != OMIT
+      @knowledge_base_id = knowledge_base_id if knowledge_base_id != OMIT
       @model = model
       @fallback_models = fallback_models if fallback_models != OMIT
       @semantic_caching_enabled = semantic_caching_enabled if semantic_caching_enabled != OMIT
       @temperature = temperature if temperature != OMIT
-      @knowledge_base = knowledge_base if knowledge_base != OMIT
       @max_tokens = max_tokens if max_tokens != OMIT
       @emotion_recognition_enabled = emotion_recognition_enabled if emotion_recognition_enabled != OMIT
       @num_fast_turns = num_fast_turns if num_fast_turns != OMIT
@@ -103,11 +107,12 @@ module Vapi
         "messages": messages,
         "tools": tools,
         "toolIds": tool_ids,
+        "knowledgeBase": knowledge_base,
+        "knowledgeBaseId": knowledge_base_id,
         "model": model,
         "fallbackModels": fallback_models,
         "semanticCachingEnabled": semantic_caching_enabled,
         "temperature": temperature,
-        "knowledgeBase": knowledge_base,
         "maxTokens": max_tokens,
         "emotionRecognitionEnabled": emotion_recognition_enabled,
         "numFastTurns": num_fast_turns
@@ -132,16 +137,17 @@ module Vapi
         Vapi::OpenAiModelToolsItem.from_json(json_object: item)
       end
       tool_ids = parsed_json["toolIds"]
-      model = parsed_json["model"]
-      fallback_models = parsed_json["fallbackModels"]
-      semantic_caching_enabled = parsed_json["semanticCachingEnabled"]
-      temperature = parsed_json["temperature"]
       if parsed_json["knowledgeBase"].nil?
         knowledge_base = nil
       else
         knowledge_base = parsed_json["knowledgeBase"].to_json
-        knowledge_base = Vapi::KnowledgeBase.from_json(json_object: knowledge_base)
+        knowledge_base = Vapi::CreateCustomKnowledgeBaseDto.from_json(json_object: knowledge_base)
       end
+      knowledge_base_id = parsed_json["knowledgeBaseId"]
+      model = parsed_json["model"]
+      fallback_models = parsed_json["fallbackModels"]
+      semantic_caching_enabled = parsed_json["semanticCachingEnabled"]
+      temperature = parsed_json["temperature"]
       max_tokens = parsed_json["maxTokens"]
       emotion_recognition_enabled = parsed_json["emotionRecognitionEnabled"]
       num_fast_turns = parsed_json["numFastTurns"]
@@ -149,11 +155,12 @@ module Vapi
         messages: messages,
         tools: tools,
         tool_ids: tool_ids,
+        knowledge_base: knowledge_base,
+        knowledge_base_id: knowledge_base_id,
         model: model,
         fallback_models: fallback_models,
         semantic_caching_enabled: semantic_caching_enabled,
         temperature: temperature,
-        knowledge_base: knowledge_base,
         max_tokens: max_tokens,
         emotion_recognition_enabled: emotion_recognition_enabled,
         num_fast_turns: num_fast_turns,
@@ -178,11 +185,12 @@ module Vapi
       obj.messages&.is_a?(Array) != false || raise("Passed value for field obj.messages is not the expected type, validation failed.")
       obj.tools&.is_a?(Array) != false || raise("Passed value for field obj.tools is not the expected type, validation failed.")
       obj.tool_ids&.is_a?(Array) != false || raise("Passed value for field obj.tool_ids is not the expected type, validation failed.")
+      obj.knowledge_base.nil? || Vapi::CreateCustomKnowledgeBaseDto.validate_raw(obj: obj.knowledge_base)
+      obj.knowledge_base_id&.is_a?(String) != false || raise("Passed value for field obj.knowledge_base_id is not the expected type, validation failed.")
       obj.model.is_a?(Vapi::OpenAiModelModel) != false || raise("Passed value for field obj.model is not the expected type, validation failed.")
       obj.fallback_models&.is_a?(Array) != false || raise("Passed value for field obj.fallback_models is not the expected type, validation failed.")
       obj.semantic_caching_enabled&.is_a?(Boolean) != false || raise("Passed value for field obj.semantic_caching_enabled is not the expected type, validation failed.")
       obj.temperature&.is_a?(Float) != false || raise("Passed value for field obj.temperature is not the expected type, validation failed.")
-      obj.knowledge_base.nil? || Vapi::KnowledgeBase.validate_raw(obj: obj.knowledge_base)
       obj.max_tokens&.is_a?(Float) != false || raise("Passed value for field obj.max_tokens is not the expected type, validation failed.")
       obj.emotion_recognition_enabled&.is_a?(Boolean) != false || raise("Passed value for field obj.emotion_recognition_enabled is not the expected type, validation failed.")
       obj.num_fast_turns&.is_a?(Float) != false || raise("Passed value for field obj.num_fast_turns is not the expected type, validation failed.")

data/lib/vapi_server_sdk/types/open_ai_model_fallback_models_item.rb

@@ -2,11 +2,19 @@
 
 module Vapi
   class OpenAiModelFallbackModelsItem
+    O_1_PREVIEW = "o1-preview"
+    O_1_PREVIEW_20240912 = "o1-preview-2024-09-12"
+    O_1_MINI = "o1-mini"
+    O_1_MINI_20240912 = "o1-mini-2024-09-12"
+    GPT_4_O_REALTIME_PREVIEW_20241001 = "gpt-4o-realtime-preview-2024-10-01"
+    GPT_4_O_REALTIME_PREVIEW_20241217 = "gpt-4o-realtime-preview-2024-12-17"
+    GPT_4_O_MINI_REALTIME_PREVIEW_20241217 = "gpt-4o-mini-realtime-preview-2024-12-17"
     GPT_4_O_MINI = "gpt-4o-mini"
     GPT_4_O_MINI_20240718 = "gpt-4o-mini-2024-07-18"
     GPT_4_O = "gpt-4o"
     GPT_4_O_20240513 = "gpt-4o-2024-05-13"
     GPT_4_O_20240806 = "gpt-4o-2024-08-06"
+    GPT_4_O_20241120 = "gpt-4o-2024-11-20"
     GPT_4_TURBO = "gpt-4-turbo"
     GPT_4_TURBO_20240409 = "gpt-4-turbo-2024-04-09"
     GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"

data/lib/vapi_server_sdk/types/open_ai_model_model.rb

@@ -3,11 +3,19 @@
 module Vapi
   # This is the OpenAI model that will be used.
   class OpenAiModelModel
+    O_1_PREVIEW = "o1-preview"
+    O_1_PREVIEW_20240912 = "o1-preview-2024-09-12"
+    O_1_MINI = "o1-mini"
+    O_1_MINI_20240912 = "o1-mini-2024-09-12"
+    GPT_4_O_REALTIME_PREVIEW_20241001 = "gpt-4o-realtime-preview-2024-10-01"
+    GPT_4_O_REALTIME_PREVIEW_20241217 = "gpt-4o-realtime-preview-2024-12-17"
+    GPT_4_O_MINI_REALTIME_PREVIEW_20241217 = "gpt-4o-mini-realtime-preview-2024-12-17"
     GPT_4_O_MINI = "gpt-4o-mini"
     GPT_4_O_MINI_20240718 = "gpt-4o-mini-2024-07-18"
     GPT_4_O = "gpt-4o"
     GPT_4_O_20240513 = "gpt-4o-2024-05-13"
     GPT_4_O_20240806 = "gpt-4o-2024-08-06"
+    GPT_4_O_20241120 = "gpt-4o-2024-11-20"
     GPT_4_TURBO = "gpt-4-turbo"
     GPT_4_TURBO_20240409 = "gpt-4-turbo-2024-04-09"
     GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"

data/lib/vapi_server_sdk/types/open_ai_voice.rb

@@ -2,22 +2,24 @@
 
 require_relative "open_ai_voice_id"
 require_relative "chunk_plan"
+require_relative "fallback_plan"
 require "ostruct"
 require "json"
 
 module Vapi
   class OpenAiVoice
-    # @return [Boolean] This determines whether fillers are injected into the model output before
-    #  inputting it into the voice provider.
-    #  Default `false` because you can achieve better results with prompting the model.
-    attr_reader :filler_injection_enabled
     # @return [Vapi::OpenAiVoiceId] This is the provider-specific ID that will be used.
+    #  Please note that ash, ballad, coral, sage, and verse may only be used with
+    #  realtime models.
     attr_reader :voice_id
     # @return [Float] This is the speed multiplier that will be used.
     attr_reader :speed
     # @return [Vapi::ChunkPlan] This is the plan for chunking the model output before it is sent to the voice
     #  provider.
     attr_reader :chunk_plan
+    # @return [Vapi::FallbackPlan] This is the plan for voice provider fallbacks in the event that the primary
+    #  voice provider fails.
+    attr_reader :fallback_plan
     # @return [OpenStruct] Additional properties unmapped to the current class definition
     attr_reader :additional_properties
     # @return [Object]
@@ -26,26 +28,27 @@ module Vapi
 
     OMIT = Object.new
 
-    # @param filler_injection_enabled [Boolean] This determines whether fillers are injected into the model output before
-    #  inputting it into the voice provider.
-    #  Default `false` because you can achieve better results with prompting the model.
     # @param voice_id [Vapi::OpenAiVoiceId] This is the provider-specific ID that will be used.
+    #  Please note that ash, ballad, coral, sage, and verse may only be used with
+    #  realtime models.
     # @param speed [Float] This is the speed multiplier that will be used.
     # @param chunk_plan [Vapi::ChunkPlan] This is the plan for chunking the model output before it is sent to the voice
     #  provider.
+    # @param fallback_plan [Vapi::FallbackPlan] This is the plan for voice provider fallbacks in the event that the primary
+    #  voice provider fails.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::OpenAiVoice]
-    def initialize(voice_id:, filler_injection_enabled: OMIT, speed: OMIT, chunk_plan: OMIT, additional_properties: nil)
-      @filler_injection_enabled = filler_injection_enabled if filler_injection_enabled != OMIT
+    def initialize(voice_id:, speed: OMIT, chunk_plan: OMIT, fallback_plan: OMIT, additional_properties: nil)
       @voice_id = voice_id
       @speed = speed if speed != OMIT
       @chunk_plan = chunk_plan if chunk_plan != OMIT
+      @fallback_plan = fallback_plan if fallback_plan != OMIT
       @additional_properties = additional_properties
       @_field_set = {
-        "fillerInjectionEnabled": filler_injection_enabled,
         "voiceId": voice_id,
         "speed": speed,
-        "chunkPlan": chunk_plan
+        "chunkPlan": chunk_plan,
+        "fallbackPlan": fallback_plan
       }.reject do |_k, v|
         v == OMIT
       end
@@ -58,7 +61,6 @@ module Vapi
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
-      filler_injection_enabled = parsed_json["fillerInjectionEnabled"]
       voice_id = parsed_json["voiceId"]
       speed = parsed_json["speed"]
       if parsed_json["chunkPlan"].nil?
@@ -67,11 +69,17 @@ module Vapi
         chunk_plan = parsed_json["chunkPlan"].to_json
         chunk_plan = Vapi::ChunkPlan.from_json(json_object: chunk_plan)
       end
+      if parsed_json["fallbackPlan"].nil?
+        fallback_plan = nil
+      else
+        fallback_plan = parsed_json["fallbackPlan"].to_json
+        fallback_plan = Vapi::FallbackPlan.from_json(json_object: fallback_plan)
+      end
       new(
-        filler_injection_enabled: filler_injection_enabled,
         voice_id: voice_id,
         speed: speed,
         chunk_plan: chunk_plan,
+        fallback_plan: fallback_plan,
         additional_properties: struct
       )
     end
@@ -90,10 +98,10 @@ module Vapi
     # @param obj [Object]
     # @return [Void]
     def self.validate_raw(obj:)
-      obj.filler_injection_enabled&.is_a?(Boolean) != false || raise("Passed value for field obj.filler_injection_enabled is not the expected type, validation failed.")
       obj.voice_id.is_a?(Vapi::OpenAiVoiceId) != false || raise("Passed value for field obj.voice_id is not the expected type, validation failed.")
       obj.speed&.is_a?(Float) != false || raise("Passed value for field obj.speed is not the expected type, validation failed.")
       obj.chunk_plan.nil? || Vapi::ChunkPlan.validate_raw(obj: obj.chunk_plan)
+      obj.fallback_plan.nil? || Vapi::FallbackPlan.validate_raw(obj: obj.fallback_plan)
     end
   end
 end

data/lib/vapi_server_sdk/types/open_ai_voice_id.rb

@@ -2,6 +2,8 @@
 
 module Vapi
   # This is the provider-specific ID that will be used.
+  #  Please note that ash, ballad, coral, sage, and verse may only be used with
+  #  realtime models.
   class OpenAiVoiceId
     ALLOY = "alloy"
     ECHO = "echo"
@@ -9,5 +11,10 @@ module Vapi
     ONYX = "onyx"
     NOVA = "nova"
     SHIMMER = "shimmer"
+    ASH = "ash"
+    BALLAD = "ballad"
+    CORAL = "coral"
+    SAGE = "sage"
+    VERSE = "verse"
   end
 end

data/lib/vapi_server_sdk/types/open_router_credential.rb

@@ -18,6 +18,8 @@ module Vapi
     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]
@@ -32,15 +34,17 @@ module Vapi
     # @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::OpenRouterCredential]
-    def initialize(provider:, api_key:, id:, org_id:, created_at:, updated_at:, additional_properties: nil)
+    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,
@@ -48,8 +52,11 @@ module Vapi
         "id": id,
         "orgId": org_id,
         "createdAt": created_at,
-        "updatedAt": updated_at
-      }
+        "updatedAt": updated_at,
+        "name": name
+      }.reject do |_k, v|
+        v == OMIT
+      end
     end
 
     # Deserialize a JSON object to an instance of OpenRouterCredential
@@ -65,6 +72,7 @@ module Vapi
       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,
@@ -72,6 +80,7 @@ module Vapi
         org_id: org_id,
         created_at: created_at,
         updated_at: updated_at,
+        name: name,
         additional_properties: struct
       )
     end
@@ -96,6 +105,7 @@ module Vapi
       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/open_router_model.rb

@@ -2,7 +2,7 @@
 
 require_relative "open_ai_message"
 require_relative "open_router_model_tools_item"
-require_relative "knowledge_base"
+require_relative "create_custom_knowledge_base_dto"
 require "ostruct"
 require "json"
 
@@ -18,13 +18,15 @@ module Vapi
     #  tools, use `tools`.
     #  Both `tools` and `toolIds` can be used together.
     attr_reader :tool_ids
+    # @return [Vapi::CreateCustomKnowledgeBaseDto] These are the options for the knowledge base.
+    attr_reader :knowledge_base
+    # @return [String] This is the ID of the knowledge base the model will use.
+    attr_reader :knowledge_base_id
     # @return [String] This is the name of the model. Ex. cognitivecomputations/dolphin-mixtral-8x7b
     attr_reader :model
     # @return [Float] This is the temperature that will be used for calls. Default is 0 to leverage
     #  caching for lower latency.
     attr_reader :temperature
-    # @return [Vapi::KnowledgeBase] These are the options for the knowledge base.
-    attr_reader :knowledge_base
     # @return [Float] This is the max number of tokens that the assistant will be allowed to generate
     #  in each turn of the conversation. Default is 250.
     attr_reader :max_tokens
@@ -55,10 +57,11 @@ module Vapi
     # @param tool_ids [Array<String>] These are the tools that the assistant can use during the call. To use transient
     #  tools, use `tools`.
     #  Both `tools` and `toolIds` can be used together.
+    # @param knowledge_base [Vapi::CreateCustomKnowledgeBaseDto] These are the options for the knowledge base.
+    # @param knowledge_base_id [String] This is the ID of the knowledge base the model will use.
     # @param model [String] This is the name of the model. Ex. cognitivecomputations/dolphin-mixtral-8x7b
     # @param temperature [Float] This is the temperature that will be used for calls. Default is 0 to leverage
     #  caching for lower latency.
-    # @param knowledge_base [Vapi::KnowledgeBase] These are the options for the knowledge base.
     # @param max_tokens [Float] This is the max number of tokens that the assistant will be allowed to generate
     #  in each turn of the conversation. Default is 250.
     # @param emotion_recognition_enabled [Boolean] This determines whether we detect user's emotion while they speak and send it as
@@ -73,14 +76,15 @@ module Vapi
     #  @default 0
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::OpenRouterModel]
-    def initialize(model:, messages: OMIT, tools: OMIT, tool_ids: OMIT, temperature: OMIT, knowledge_base: OMIT,
-                   max_tokens: OMIT, emotion_recognition_enabled: OMIT, num_fast_turns: OMIT, additional_properties: nil)
+    def initialize(model:, messages: OMIT, tools: OMIT, tool_ids: OMIT, knowledge_base: OMIT, knowledge_base_id: OMIT,
+                   temperature: OMIT, max_tokens: OMIT, emotion_recognition_enabled: OMIT, num_fast_turns: OMIT, additional_properties: nil)
       @messages = messages if messages != OMIT
       @tools = tools if tools != OMIT
       @tool_ids = tool_ids if tool_ids != OMIT
+      @knowledge_base = knowledge_base if knowledge_base != OMIT
+      @knowledge_base_id = knowledge_base_id if knowledge_base_id != OMIT
       @model = model
       @temperature = temperature if temperature != OMIT
-      @knowledge_base = knowledge_base if knowledge_base != OMIT
       @max_tokens = max_tokens if max_tokens != OMIT
       @emotion_recognition_enabled = emotion_recognition_enabled if emotion_recognition_enabled != OMIT
       @num_fast_turns = num_fast_turns if num_fast_turns != OMIT
@@ -89,9 +93,10 @@ module Vapi
         "messages": messages,
         "tools": tools,
         "toolIds": tool_ids,
+        "knowledgeBase": knowledge_base,
+        "knowledgeBaseId": knowledge_base_id,
         "model": model,
         "temperature": temperature,
-        "knowledgeBase": knowledge_base,
         "maxTokens": max_tokens,
         "emotionRecognitionEnabled": emotion_recognition_enabled,
         "numFastTurns": num_fast_turns
@@ -116,14 +121,15 @@ module Vapi
         Vapi::OpenRouterModelToolsItem.from_json(json_object: item)
       end
       tool_ids = parsed_json["toolIds"]
-      model = parsed_json["model"]
-      temperature = parsed_json["temperature"]
       if parsed_json["knowledgeBase"].nil?
         knowledge_base = nil
       else
         knowledge_base = parsed_json["knowledgeBase"].to_json
-        knowledge_base = Vapi::KnowledgeBase.from_json(json_object: knowledge_base)
+        knowledge_base = Vapi::CreateCustomKnowledgeBaseDto.from_json(json_object: knowledge_base)
       end
+      knowledge_base_id = parsed_json["knowledgeBaseId"]
+      model = parsed_json["model"]
+      temperature = parsed_json["temperature"]
       max_tokens = parsed_json["maxTokens"]
       emotion_recognition_enabled = parsed_json["emotionRecognitionEnabled"]
       num_fast_turns = parsed_json["numFastTurns"]
@@ -131,9 +137,10 @@ module Vapi
         messages: messages,
         tools: tools,
         tool_ids: tool_ids,
+        knowledge_base: knowledge_base,
+        knowledge_base_id: knowledge_base_id,
         model: model,
         temperature: temperature,
-        knowledge_base: knowledge_base,
         max_tokens: max_tokens,
         emotion_recognition_enabled: emotion_recognition_enabled,
         num_fast_turns: num_fast_turns,
@@ -158,9 +165,10 @@ module Vapi
       obj.messages&.is_a?(Array) != false || raise("Passed value for field obj.messages is not the expected type, validation failed.")
       obj.tools&.is_a?(Array) != false || raise("Passed value for field obj.tools is not the expected type, validation failed.")
       obj.tool_ids&.is_a?(Array) != false || raise("Passed value for field obj.tool_ids is not the expected type, validation failed.")
+      obj.knowledge_base.nil? || Vapi::CreateCustomKnowledgeBaseDto.validate_raw(obj: obj.knowledge_base)
+      obj.knowledge_base_id&.is_a?(String) != false || raise("Passed value for field obj.knowledge_base_id is not the expected type, validation failed.")
       obj.model.is_a?(String) != false || raise("Passed value for field obj.model is not the expected type, validation failed.")
       obj.temperature&.is_a?(Float) != false || raise("Passed value for field obj.temperature is not the expected type, validation failed.")
-      obj.knowledge_base.nil? || Vapi::KnowledgeBase.validate_raw(obj: obj.knowledge_base)
       obj.max_tokens&.is_a?(Float) != false || raise("Passed value for field obj.max_tokens is not the expected type, validation failed.")
       obj.emotion_recognition_enabled&.is_a?(Boolean) != false || raise("Passed value for field obj.emotion_recognition_enabled is not the expected type, validation failed.")
       obj.num_fast_turns&.is_a?(Float) != false || raise("Passed value for field obj.num_fast_turns is not the expected type, validation failed.")