vapi_server_sdk 0.1.0 → 0.3.0

data/lib/vapi_server_sdk/types/update_workflow_block_dto.rb

@@ -0,0 +1,154 @@
+# frozen_string_literal: true
+
+require_relative "update_workflow_block_dto_messages_item"
+require_relative "json_schema"
+require_relative "update_workflow_block_dto_steps_item"
+require "ostruct"
+require "json"
+
+module Vapi
+  class UpdateWorkflowBlockDto
+    # @return [Array<Vapi::UpdateWorkflowBlockDtoMessagesItem>] These are the pre-configured messages that will be spoken to the user while the
+    #  block is running.
+    attr_reader :messages
+    # @return [Vapi::JsonSchema] This is the input schema for the block. This is the input the block needs to
+    #  run. It's given to the block as `steps[0].input`
+    #  These are accessible as variables:
+    #  - ({{input.propertyName}}) in context of the block execution (step)
+    #  - ({{stepName.input.propertyName}}) in context of the workflow
+    attr_reader :input_schema
+    # @return [Vapi::JsonSchema] This is the output schema for the block. This is the output the block will
+    #  return to the workflow (`{{stepName.output}}`).
+    #  These are accessible as variables:
+    #  - ({{output.propertyName}}) in context of the block execution (step)
+    #  - ({{stepName.output.propertyName}}) in context of the workflow (read caveat #1)
+    #  - ({{blockName.output.propertyName}}) in context of the workflow (read caveat
+    #  #2)
+    #  Caveats:
+    #  1. a workflow can execute a step multiple times. example, if a loop is used in
+    #  the graph. {{stepName.output.propertyName}} will reference the latest usage of
+    #  the step.
+    #  2. a workflow can execute a block multiple times. example, if a step is called
+    #  multiple times or if a block is used in multiple steps.
+    #  {{blockName.output.propertyName}} will reference the latest usage of the block.
+    #  this liquid variable is just provided for convenience when creating blocks
+    #  outside of a workflow with steps.
+    attr_reader :output_schema
+    # @return [Array<Vapi::UpdateWorkflowBlockDtoStepsItem>] These are the steps in the workflow.
+    attr_reader :steps
+    # @return [String] This is the name of the block. 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 messages [Array<Vapi::UpdateWorkflowBlockDtoMessagesItem>] These are the pre-configured messages that will be spoken to the user while the
+    #  block is running.
+    # @param input_schema [Vapi::JsonSchema] This is the input schema for the block. This is the input the block needs to
+    #  run. It's given to the block as `steps[0].input`
+    #  These are accessible as variables:
+    #  - ({{input.propertyName}}) in context of the block execution (step)
+    #  - ({{stepName.input.propertyName}}) in context of the workflow
+    # @param output_schema [Vapi::JsonSchema] This is the output schema for the block. This is the output the block will
+    #  return to the workflow (`{{stepName.output}}`).
+    #  These are accessible as variables:
+    #  - ({{output.propertyName}}) in context of the block execution (step)
+    #  - ({{stepName.output.propertyName}}) in context of the workflow (read caveat #1)
+    #  - ({{blockName.output.propertyName}}) in context of the workflow (read caveat
+    #  #2)
+    #  Caveats:
+    #  1. a workflow can execute a step multiple times. example, if a loop is used in
+    #  the graph. {{stepName.output.propertyName}} will reference the latest usage of
+    #  the step.
+    #  2. a workflow can execute a block multiple times. example, if a step is called
+    #  multiple times or if a block is used in multiple steps.
+    #  {{blockName.output.propertyName}} will reference the latest usage of the block.
+    #  this liquid variable is just provided for convenience when creating blocks
+    #  outside of a workflow with steps.
+    # @param steps [Array<Vapi::UpdateWorkflowBlockDtoStepsItem>] These are the steps in the workflow.
+    # @param name [String] This is the name of the block. This is just for your reference.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vapi::UpdateWorkflowBlockDto]
+    def initialize(messages: OMIT, input_schema: OMIT, output_schema: OMIT, steps: OMIT, name: OMIT,
+                   additional_properties: nil)
+      @messages = messages if messages != OMIT
+      @input_schema = input_schema if input_schema != OMIT
+      @output_schema = output_schema if output_schema != OMIT
+      @steps = steps if steps != OMIT
+      @name = name if name != OMIT
+      @additional_properties = additional_properties
+      @_field_set = {
+        "messages": messages,
+        "inputSchema": input_schema,
+        "outputSchema": output_schema,
+        "steps": steps,
+        "name": name
+      }.reject do |_k, v|
+        v == OMIT
+      end
+    end
+
+    # Deserialize a JSON object to an instance of UpdateWorkflowBlockDto
+    #
+    # @param json_object [String]
+    # @return [Vapi::UpdateWorkflowBlockDto]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      messages = parsed_json["messages"]&.map do |item|
+        item = item.to_json
+        Vapi::UpdateWorkflowBlockDtoMessagesItem.from_json(json_object: item)
+      end
+      if parsed_json["inputSchema"].nil?
+        input_schema = nil
+      else
+        input_schema = parsed_json["inputSchema"].to_json
+        input_schema = Vapi::JsonSchema.from_json(json_object: input_schema)
+      end
+      if parsed_json["outputSchema"].nil?
+        output_schema = nil
+      else
+        output_schema = parsed_json["outputSchema"].to_json
+        output_schema = Vapi::JsonSchema.from_json(json_object: output_schema)
+      end
+      steps = parsed_json["steps"]&.map do |item|
+        item = item.to_json
+        Vapi::UpdateWorkflowBlockDtoStepsItem.from_json(json_object: item)
+      end
+      name = parsed_json["name"]
+      new(
+        messages: messages,
+        input_schema: input_schema,
+        output_schema: output_schema,
+        steps: steps,
+        name: name,
+        additional_properties: struct
+      )
+    end
+
+    # Serialize an instance of UpdateWorkflowBlockDto 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.messages&.is_a?(Array) != false || raise("Passed value for field obj.messages is not the expected type, validation failed.")
+      obj.input_schema.nil? || Vapi::JsonSchema.validate_raw(obj: obj.input_schema)
+      obj.output_schema.nil? || Vapi::JsonSchema.validate_raw(obj: obj.output_schema)
+      obj.steps&.is_a?(Array) != false || raise("Passed value for field obj.steps 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/update_workflow_block_dto_messages_item.rb

@@ -0,0 +1,94 @@
+# frozen_string_literal: true
+
+require "json"
+require_relative "block_start_message"
+require_relative "block_complete_message"
+
+module Vapi
+  class UpdateWorkflowBlockDtoMessagesItem
+    # @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::UpdateWorkflowBlockDtoMessagesItem]
+    def initialize(member:, discriminant:)
+      @member = member
+      @discriminant = discriminant
+    end
+
+    # Deserialize a JSON object to an instance of UpdateWorkflowBlockDtoMessagesItem
+    #
+    # @param json_object [String]
+    # @return [Vapi::UpdateWorkflowBlockDtoMessagesItem]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      member = case struct.type
+               when "block-start"
+                 Vapi::BlockStartMessage.from_json(json_object: json_object)
+               when "block-complete"
+                 Vapi::BlockCompleteMessage.from_json(json_object: json_object)
+               else
+                 Vapi::BlockStartMessage.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 "block-start"
+        { **@member.to_json, type: @discriminant }.to_json
+      when "block-complete"
+        { **@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 "block-start"
+        Vapi::BlockStartMessage.validate_raw(obj: obj)
+      when "block-complete"
+        Vapi::BlockCompleteMessage.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::BlockStartMessage]
+    # @return [Vapi::UpdateWorkflowBlockDtoMessagesItem]
+    def self.block_start(member:)
+      new(member: member, discriminant: "block-start")
+    end
+
+    # @param member [Vapi::BlockCompleteMessage]
+    # @return [Vapi::UpdateWorkflowBlockDtoMessagesItem]
+    def self.block_complete(member:)
+      new(member: member, discriminant: "block-complete")
+    end
+  end
+end

data/lib/vapi_server_sdk/types/update_workflow_block_dto_steps_item.rb

@@ -0,0 +1,94 @@
+# frozen_string_literal: true
+
+require "json"
+require_relative "handoff_step"
+require_relative "callback_step"
+
+module Vapi
+  class UpdateWorkflowBlockDtoStepsItem
+    # @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::UpdateWorkflowBlockDtoStepsItem]
+    def initialize(member:, discriminant:)
+      @member = member
+      @discriminant = discriminant
+    end
+
+    # Deserialize a JSON object to an instance of UpdateWorkflowBlockDtoStepsItem
+    #
+    # @param json_object [String]
+    # @return [Vapi::UpdateWorkflowBlockDtoStepsItem]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      member = case struct.type
+               when "handoff"
+                 Vapi::HandoffStep.from_json(json_object: json_object)
+               when "callback"
+                 Vapi::CallbackStep.from_json(json_object: json_object)
+               else
+                 Vapi::HandoffStep.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 "handoff"
+        { **@member.to_json, type: @discriminant }.to_json
+      when "callback"
+        { **@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 "handoff"
+        Vapi::HandoffStep.validate_raw(obj: obj)
+      when "callback"
+        Vapi::CallbackStep.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::HandoffStep]
+    # @return [Vapi::UpdateWorkflowBlockDtoStepsItem]
+    def self.handoff(member:)
+      new(member: member, discriminant: "handoff")
+    end
+
+    # @param member [Vapi::CallbackStep]
+    # @return [Vapi::UpdateWorkflowBlockDtoStepsItem]
+    def self.callback(member:)
+      new(member: member, discriminant: "callback")
+    end
+  end
+end

data/lib/vapi_server_sdk/types/update_x_ai_credential_dto.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+require "ostruct"
+require "json"
+
+module Vapi
+  class UpdateXAiCredentialDto
+    # @return [String] This is not returned in the API.
+    attr_reader :api_key
+    # @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 api_key [String] This is not returned in the API.
+    # @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::UpdateXAiCredentialDto]
+    def initialize(api_key: OMIT, name: OMIT, additional_properties: nil)
+      @api_key = api_key if api_key != OMIT
+      @name = name if name != OMIT
+      @additional_properties = additional_properties
+      @_field_set = { "apiKey": api_key, "name": name }.reject do |_k, v|
+        v == OMIT
+      end
+    end
+
+    # Deserialize a JSON object to an instance of UpdateXAiCredentialDto
+    #
+    # @param json_object [String]
+    # @return [Vapi::UpdateXAiCredentialDto]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      api_key = parsed_json["apiKey"]
+      name = parsed_json["name"]
+      new(
+        api_key: api_key,
+        name: name,
+        additional_properties: struct
+      )
+    end
+
+    # Serialize an instance of UpdateXAiCredentialDto 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.api_key&.is_a?(String) != false || raise("Passed value for field obj.api_key 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/vapi_cost.rb

@@ -1,10 +1,13 @@
 # frozen_string_literal: true
 
+require_relative "vapi_cost_sub_type"
 require "ostruct"
 require "json"
 
 module Vapi
   class VapiCost
+    # @return [Vapi::VapiCostSubType] This is the sub type of the cost.
+    attr_reader :sub_type
     # @return [Float] This is the minutes of Vapi usage. This should match `call.endedAt` -
     #  `call.startedAt`.
     attr_reader :minutes
@@ -18,16 +21,18 @@ module Vapi
 
     OMIT = Object.new
 
+    # @param sub_type [Vapi::VapiCostSubType] This is the sub type of the cost.
     # @param minutes [Float] This is the minutes of Vapi usage. This should match `call.endedAt` -
     #  `call.startedAt`.
     # @param cost [Float] This is the cost of the component in USD.
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::VapiCost]
-    def initialize(minutes:, cost:, additional_properties: nil)
+    def initialize(sub_type:, minutes:, cost:, additional_properties: nil)
+      @sub_type = sub_type
       @minutes = minutes
       @cost = cost
       @additional_properties = additional_properties
-      @_field_set = { "minutes": minutes, "cost": cost }
+      @_field_set = { "subType": sub_type, "minutes": minutes, "cost": cost }
     end
 
     # Deserialize a JSON object to an instance of VapiCost
@@ -37,9 +42,11 @@ module Vapi
     def self.from_json(json_object:)
       struct = JSON.parse(json_object, object_class: OpenStruct)
       parsed_json = JSON.parse(json_object)
+      sub_type = parsed_json["subType"]
       minutes = parsed_json["minutes"]
       cost = parsed_json["cost"]
       new(
+        sub_type: sub_type,
         minutes: minutes,
         cost: cost,
         additional_properties: struct
@@ -60,6 +67,7 @@ module Vapi
     # @param obj [Object]
     # @return [Void]
     def self.validate_raw(obj:)
+      obj.sub_type.is_a?(Vapi::VapiCostSubType) != false || raise("Passed value for field obj.sub_type is not the expected type, validation failed.")
       obj.minutes.is_a?(Float) != false || raise("Passed value for field obj.minutes is not the expected type, validation failed.")
       obj.cost.is_a?(Float) != false || raise("Passed value for field obj.cost is not the expected type, validation failed.")
     end

data/lib/vapi_server_sdk/types/vapi_cost_sub_type.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module Vapi
+  # This is the sub type of the cost.
+  class VapiCostSubType
+    NORMAL = "normal"
+    OVERAGE = "overage"
+  end
+end

data/lib/vapi_server_sdk/types/vapi_model.rb

@@ -2,8 +2,8 @@
 
 require_relative "open_ai_message"
 require_relative "vapi_model_tools_item"
+require_relative "create_custom_knowledge_base_dto"
 require_relative "vapi_model_steps_item"
-require_relative "knowledge_base"
 require "ostruct"
 require "json"
 
@@ -19,6 +19,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 [Array<Vapi::VapiModelStepsItem>]
     attr_reader :steps
     # @return [String] This is the name of the model. Ex. cognitivecomputations/dolphin-mixtral-8x7b
@@ -26,8 +30,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
@@ -58,11 +60,12 @@ 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 steps [Array<Vapi::VapiModelStepsItem>]
     # @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
@@ -77,15 +80,16 @@ module Vapi
     #  @default 0
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::VapiModel]
-    def initialize(model:, messages: OMIT, tools: OMIT, tool_ids: OMIT, steps: 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,
+                   steps: 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
       @steps = steps if steps != 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
@@ -94,10 +98,11 @@ module Vapi
         "messages": messages,
         "tools": tools,
         "toolIds": tool_ids,
+        "knowledgeBase": knowledge_base,
+        "knowledgeBaseId": knowledge_base_id,
         "steps": steps,
         "model": model,
         "temperature": temperature,
-        "knowledgeBase": knowledge_base,
         "maxTokens": max_tokens,
         "emotionRecognitionEnabled": emotion_recognition_enabled,
         "numFastTurns": num_fast_turns
@@ -122,18 +127,19 @@ module Vapi
         Vapi::VapiModelToolsItem.from_json(json_object: item)
       end
       tool_ids = parsed_json["toolIds"]
+      if parsed_json["knowledgeBase"].nil?
+        knowledge_base = nil
+      else
+        knowledge_base = parsed_json["knowledgeBase"].to_json
+        knowledge_base = Vapi::CreateCustomKnowledgeBaseDto.from_json(json_object: knowledge_base)
+      end
+      knowledge_base_id = parsed_json["knowledgeBaseId"]
       steps = parsed_json["steps"]&.map do |item|
         item = item.to_json
         Vapi::VapiModelStepsItem.from_json(json_object: item)
       end
       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)
-      end
       max_tokens = parsed_json["maxTokens"]
       emotion_recognition_enabled = parsed_json["emotionRecognitionEnabled"]
       num_fast_turns = parsed_json["numFastTurns"]
@@ -141,10 +147,11 @@ module Vapi
         messages: messages,
         tools: tools,
         tool_ids: tool_ids,
+        knowledge_base: knowledge_base,
+        knowledge_base_id: knowledge_base_id,
         steps: steps,
         model: model,
         temperature: temperature,
-        knowledge_base: knowledge_base,
         max_tokens: max_tokens,
         emotion_recognition_enabled: emotion_recognition_enabled,
         num_fast_turns: num_fast_turns,
@@ -169,10 +176,11 @@ 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.steps&.is_a?(Array) != false || raise("Passed value for field obj.steps 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.")