vapi_server_sdk 0.0.0.pre.alpha7 → 0.2.0

data/lib/vapi_server_sdk/tools/types/tools_get_response.rb

@@ -8,6 +8,9 @@ require_relative "../../types/ghl_tool"
 require_relative "../../types/make_tool"
 require_relative "../../types/transfer_call_tool"
 require_relative "../../types/output_tool"
+require_relative "../../types/bash_tool"
+require_relative "../../types/computer_tool"
+require_relative "../../types/text_editor_tool"
 
 module Vapi
   class Tools
@@ -49,6 +52,12 @@ module Vapi
                    Vapi::TransferCallTool.from_json(json_object: json_object)
                  when "output"
                    Vapi::OutputTool.from_json(json_object: json_object)
+                 when "bash"
+                   Vapi::BashTool.from_json(json_object: json_object)
+                 when "computer"
+                   Vapi::ComputerTool.from_json(json_object: json_object)
+                 when "textEditor"
+                   Vapi::TextEditorTool.from_json(json_object: json_object)
                  else
                    Vapi::DtmfTool.from_json(json_object: json_object)
                  end
@@ -74,6 +83,12 @@ module Vapi
           { **@member.to_json, type: @discriminant }.to_json
         when "output"
           { **@member.to_json, type: @discriminant }.to_json
+        when "bash"
+          { **@member.to_json, type: @discriminant }.to_json
+        when "computer"
+          { **@member.to_json, type: @discriminant }.to_json
+        when "textEditor"
+          { **@member.to_json, type: @discriminant }.to_json
         else
           { "type": @discriminant, value: @member }.to_json
         end
@@ -102,6 +117,12 @@ module Vapi
           Vapi::TransferCallTool.validate_raw(obj: obj)
         when "output"
           Vapi::OutputTool.validate_raw(obj: obj)
+        when "bash"
+          Vapi::BashTool.validate_raw(obj: obj)
+        when "computer"
+          Vapi::ComputerTool.validate_raw(obj: obj)
+        when "textEditor"
+          Vapi::TextEditorTool.validate_raw(obj: obj)
         else
           raise("Passed value matched no type within the union, validation failed.")
         end
@@ -156,6 +177,24 @@ module Vapi
       def self.output(member:)
         new(member: member, discriminant: "output")
       end
+
+      # @param member [Vapi::BashTool]
+      # @return [Vapi::Tools::ToolsGetResponse]
+      def self.bash(member:)
+        new(member: member, discriminant: "bash")
+      end
+
+      # @param member [Vapi::ComputerTool]
+      # @return [Vapi::Tools::ToolsGetResponse]
+      def self.computer(member:)
+        new(member: member, discriminant: "computer")
+      end
+
+      # @param member [Vapi::TextEditorTool]
+      # @return [Vapi::Tools::ToolsGetResponse]
+      def self.text_editor(member:)
+        new(member: member, discriminant: "textEditor")
+      end
     end
   end
 end

data/lib/vapi_server_sdk/tools/types/tools_list_response_item.rb

@@ -8,6 +8,9 @@ require_relative "../../types/ghl_tool"
 require_relative "../../types/make_tool"
 require_relative "../../types/transfer_call_tool"
 require_relative "../../types/output_tool"
+require_relative "../../types/bash_tool"
+require_relative "../../types/computer_tool"
+require_relative "../../types/text_editor_tool"
 
 module Vapi
   class Tools
@@ -49,6 +52,12 @@ module Vapi
                    Vapi::TransferCallTool.from_json(json_object: json_object)
                  when "output"
                    Vapi::OutputTool.from_json(json_object: json_object)
+                 when "bash"
+                   Vapi::BashTool.from_json(json_object: json_object)
+                 when "computer"
+                   Vapi::ComputerTool.from_json(json_object: json_object)
+                 when "textEditor"
+                   Vapi::TextEditorTool.from_json(json_object: json_object)
                  else
                    Vapi::DtmfTool.from_json(json_object: json_object)
                  end
@@ -74,6 +83,12 @@ module Vapi
           { **@member.to_json, type: @discriminant }.to_json
         when "output"
           { **@member.to_json, type: @discriminant }.to_json
+        when "bash"
+          { **@member.to_json, type: @discriminant }.to_json
+        when "computer"
+          { **@member.to_json, type: @discriminant }.to_json
+        when "textEditor"
+          { **@member.to_json, type: @discriminant }.to_json
         else
           { "type": @discriminant, value: @member }.to_json
         end
@@ -102,6 +117,12 @@ module Vapi
           Vapi::TransferCallTool.validate_raw(obj: obj)
         when "output"
           Vapi::OutputTool.validate_raw(obj: obj)
+        when "bash"
+          Vapi::BashTool.validate_raw(obj: obj)
+        when "computer"
+          Vapi::ComputerTool.validate_raw(obj: obj)
+        when "textEditor"
+          Vapi::TextEditorTool.validate_raw(obj: obj)
         else
           raise("Passed value matched no type within the union, validation failed.")
         end
@@ -156,6 +177,24 @@ module Vapi
       def self.output(member:)
         new(member: member, discriminant: "output")
       end
+
+      # @param member [Vapi::BashTool]
+      # @return [Vapi::Tools::ToolsListResponseItem]
+      def self.bash(member:)
+        new(member: member, discriminant: "bash")
+      end
+
+      # @param member [Vapi::ComputerTool]
+      # @return [Vapi::Tools::ToolsListResponseItem]
+      def self.computer(member:)
+        new(member: member, discriminant: "computer")
+      end
+
+      # @param member [Vapi::TextEditorTool]
+      # @return [Vapi::Tools::ToolsListResponseItem]
+      def self.text_editor(member:)
+        new(member: member, discriminant: "textEditor")
+      end
     end
   end
 end

data/lib/vapi_server_sdk/tools/types/tools_update_response.rb

@@ -8,6 +8,9 @@ require_relative "../../types/ghl_tool"
 require_relative "../../types/make_tool"
 require_relative "../../types/transfer_call_tool"
 require_relative "../../types/output_tool"
+require_relative "../../types/bash_tool"
+require_relative "../../types/computer_tool"
+require_relative "../../types/text_editor_tool"
 
 module Vapi
   class Tools
@@ -49,6 +52,12 @@ module Vapi
                    Vapi::TransferCallTool.from_json(json_object: json_object)
                  when "output"
                    Vapi::OutputTool.from_json(json_object: json_object)
+                 when "bash"
+                   Vapi::BashTool.from_json(json_object: json_object)
+                 when "computer"
+                   Vapi::ComputerTool.from_json(json_object: json_object)
+                 when "textEditor"
+                   Vapi::TextEditorTool.from_json(json_object: json_object)
                  else
                    Vapi::DtmfTool.from_json(json_object: json_object)
                  end
@@ -74,6 +83,12 @@ module Vapi
           { **@member.to_json, type: @discriminant }.to_json
         when "output"
           { **@member.to_json, type: @discriminant }.to_json
+        when "bash"
+          { **@member.to_json, type: @discriminant }.to_json
+        when "computer"
+          { **@member.to_json, type: @discriminant }.to_json
+        when "textEditor"
+          { **@member.to_json, type: @discriminant }.to_json
         else
           { "type": @discriminant, value: @member }.to_json
         end
@@ -102,6 +117,12 @@ module Vapi
           Vapi::TransferCallTool.validate_raw(obj: obj)
         when "output"
           Vapi::OutputTool.validate_raw(obj: obj)
+        when "bash"
+          Vapi::BashTool.validate_raw(obj: obj)
+        when "computer"
+          Vapi::ComputerTool.validate_raw(obj: obj)
+        when "textEditor"
+          Vapi::TextEditorTool.validate_raw(obj: obj)
         else
           raise("Passed value matched no type within the union, validation failed.")
         end
@@ -156,6 +177,24 @@ module Vapi
       def self.output(member:)
         new(member: member, discriminant: "output")
       end
+
+      # @param member [Vapi::BashTool]
+      # @return [Vapi::Tools::ToolsUpdateResponse]
+      def self.bash(member:)
+        new(member: member, discriminant: "bash")
+      end
+
+      # @param member [Vapi::ComputerTool]
+      # @return [Vapi::Tools::ToolsUpdateResponse]
+      def self.computer(member:)
+        new(member: member, discriminant: "computer")
+      end
+
+      # @param member [Vapi::TextEditorTool]
+      # @return [Vapi::Tools::ToolsUpdateResponse]
+      def self.text_editor(member:)
+        new(member: member, discriminant: "textEditor")
+      end
     end
   end
 end

data/lib/vapi_server_sdk/types/analytics_query_dto.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require_relative "analytics_query"
+require "ostruct"
+require "json"
+
+module Vapi
+  class AnalyticsQueryDto
+    # @return [Array<Vapi::AnalyticsQuery>] This is the list of metric queries you want to perform.
+    attr_reader :queries
+    # @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 queries [Array<Vapi::AnalyticsQuery>] This is the list of metric queries you want to perform.
+    # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
+    # @return [Vapi::AnalyticsQueryDto]
+    def initialize(queries:, additional_properties: nil)
+      @queries = queries
+      @additional_properties = additional_properties
+      @_field_set = { "queries": queries }
+    end
+
+    # Deserialize a JSON object to an instance of AnalyticsQueryDto
+    #
+    # @param json_object [String]
+    # @return [Vapi::AnalyticsQueryDto]
+    def self.from_json(json_object:)
+      struct = JSON.parse(json_object, object_class: OpenStruct)
+      parsed_json = JSON.parse(json_object)
+      queries = parsed_json["queries"]&.map do |item|
+        item = item.to_json
+        Vapi::AnalyticsQuery.from_json(json_object: item)
+      end
+      new(queries: queries, additional_properties: struct)
+    end
+
+    # Serialize an instance of AnalyticsQueryDto 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.queries.is_a?(Array) != false || raise("Passed value for field obj.queries is not the expected type, validation failed.")
+    end
+  end
+end

data/lib/vapi_server_sdk/types/anthropic_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::AnthropicCredential]
-    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 AnthropicCredential
@@ -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/anthropic_model.rb

@@ -2,8 +2,8 @@
 
 require_relative "open_ai_message"
 require_relative "anthropic_model_tools_item"
+require_relative "create_custom_knowledge_base_dto"
 require_relative "anthropic_model_model"
-require_relative "knowledge_base"
 require "ostruct"
 require "json"
 
@@ -19,13 +19,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 [Vapi::AnthropicModelModel] This is the Anthropic/Claude models that will be used.
     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
@@ -56,10 +58,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 [Vapi::AnthropicModelModel] This is the Anthropic/Claude models that will be used.
     # @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
@@ -74,14 +77,15 @@ module Vapi
     #  @default 0
     # @param additional_properties [OpenStruct] Additional properties unmapped to the current class definition
     # @return [Vapi::AnthropicModel]
-    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
@@ -90,9 +94,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
@@ -117,14 +122,15 @@ module Vapi
         Vapi::AnthropicModelToolsItem.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"]
@@ -132,9 +138,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,
@@ -159,9 +166,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?(Vapi::AnthropicModelModel) != 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.")

data/lib/vapi_server_sdk/types/anthropic_model_model.rb

@@ -7,5 +7,7 @@ module Vapi
     CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
     CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
     CLAUDE_35_SONNET_20240620 = "claude-3-5-sonnet-20240620"
+    CLAUDE_35_SONNET_20241022 = "claude-3-5-sonnet-20241022"
+    CLAUDE_35_HAIKU_20241022 = "claude-3-5-haiku-20241022"
   end
 end

data/lib/vapi_server_sdk/types/anyscale_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::AnyscaleCredential]
-    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 AnyscaleCredential
@@ -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/anyscale_model.rb

@@ -2,7 +2,7 @@
 
 require_relative "open_ai_message"
 require_relative "anyscale_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::AnyscaleModel]
-    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::AnyscaleModelToolsItem.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.")