[On-Prem] Extend on prem agent config (#701)

* Add kb, fix types

* Fix types

Author: lsquires

src/elevenlabs/conversational_ai/conversation.py

@@ -6,7 +6,7 @@
 import json
 import logging
 import threading
-from typing import Any, Awaitable, Callable, Dict, Literal, Optional, Tuple, Union
+from typing import Any, Awaitable, Callable, Dict, List, Literal, Optional, Tuple, Union
 import urllib.parse
 
 import websockets
@@ -322,15 +322,17 @@ def __init__(
         post_call_transcription_webhook_url: Optional[str] = None,
         post_call_audio_webhook_url: Optional[str] = None,
         agent_config_dict: Optional[dict] = None,
-        override_agent_config_list: Optional[dict] = None,
-        tools_config_list: Optional[dict] = None,
+        override_agent_config_list: Optional[List[dict]] = None,
+        tools_config_list: Optional[List[dict]] = None,
+        prompt_knowledge_base: Optional[List[str]] = None,
     ):
         self.on_prem_conversation_url = on_prem_conversation_url
         self.post_call_transcription_webhook_url = post_call_transcription_webhook_url
         self.post_call_audio_webhook_url = post_call_audio_webhook_url
-        self.agent_config_dict = agent_config_dict or {}
-        self.override_agent_config_list = override_agent_config_list or {}
-        self.tools_config_list = tools_config_list or {}
+        self.agent_config_dict = agent_config_dict
+        self.override_agent_config_list = override_agent_config_list
+        self.tools_config_list = tools_config_list
+        self.prompt_knowledge_base = prompt_knowledge_base
 
 class BaseConversation:
     """Base class for conversation implementations with shared parameters and logic."""
@@ -386,6 +388,7 @@ def _create_on_prem_initiation_message(self):
                 "tools_config_list": self.on_prem_config.tools_config_list,
                 "post_call_transcription_webhook_url": self.on_prem_config.post_call_transcription_webhook_url,
                 "post_call_audio_webhook_url": self.on_prem_config.post_call_audio_webhook_url,
+                "prompt_knowledge_base": self.on_prem_config.prompt_knowledge_base,
             }
         )