From 5403934c7be28ef295f985c15dadec9f4924c491 Mon Sep 17 00:00:00 2001 From: Edwin Zhang Date: Fri, 24 May 2024 17:57:16 -0400 Subject: [PATCH 1/2] Support multiple assistants in start --- vapi_python/vapi_python.py | 18 +++++++++++++++--- 1 file changed, 15 insertions(+), 3 deletions(-) diff --git a/vapi_python/vapi_python.py b/vapi_python/vapi_python.py index e6f5ac8..e081934 100644 --- a/vapi_python/vapi_python.py +++ b/vapi_python/vapi_python.py @@ -27,12 +27,24 @@ def __init__(self, *, api_key, api_url="https://api.vapi.ai"): self.api_key = api_key self.api_url = api_url - def start(self, *, assistant_id=None, assistant=None, assistant_overrides=None): + def start( + self, + *, + assistant_id=None, + assistant=None, + assistants=None, + assistant_override=None, + assistant_overrides=None, + ): # Start a new call if assistant_id: - assistant = {'assistantId': assistant_id, 'assistantOverrides': assistant_overrides} + assistant = {'assistantId': assistant_id, 'assistantOverride': assistant_override} elif assistant: - assistant = {'assistant': assistant, 'assistantOverrides': assistant_overrides} + assistant = {'assistant': assistant, 'assistantOverride': assistant_override} + elif assistants: + assistant = {'assistants': assistants, 'assistantOverride': assistant_overrides, 'assistantOverrides': assistant_overrides} + else: + raise Exception("Error: No assistant specified.") call_id, web_call_url = create_web_call( self.api_url, self.api_key, assistant) From 613cabd436bad0ff5d68a82f49b2dd7349cea8bb Mon Sep 17 00:00:00 2001 From: Edwin Zhang Date: Wed, 29 May 2024 17:20:15 -0400 Subject: [PATCH 2/2] Update for squads --- vapi_python/vapi_python.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/vapi_python/vapi_python.py b/vapi_python/vapi_python.py index e081934..e114948 100644 --- a/vapi_python/vapi_python.py +++ b/vapi_python/vapi_python.py @@ -6,13 +6,13 @@ CHANNELS = 1 -def create_web_call(api_url, api_key, assistant): +def create_web_call(api_url, api_key, payload): url = f"{api_url}/call/web" headers = { 'Authorization': 'Bearer ' + api_key, 'Content-Type': 'application/json' } - response = requests.post(url, headers=headers, json=assistant) + response = requests.post(url, headers=headers, json=payload) data = response.json() if response.status_code == 201: call_id = data.get('id') @@ -32,22 +32,24 @@ def start( *, assistant_id=None, assistant=None, - assistants=None, - assistant_override=None, assistant_overrides=None, + squad_id=None, + squad=None, ): # Start a new call if assistant_id: - assistant = {'assistantId': assistant_id, 'assistantOverride': assistant_override} + payload = {'assistantId': assistant_id, 'assistantOverrides': assistant_overrides} elif assistant: - assistant = {'assistant': assistant, 'assistantOverride': assistant_override} - elif assistants: - assistant = {'assistants': assistants, 'assistantOverride': assistant_overrides, 'assistantOverrides': assistant_overrides} + payload = {'assistant': assistant, 'assistantOverrides': assistant_overrides} + elif squad_id: + payload = {'squadId': squad_id} + elif squad: + payload = {'squad': squad} else: raise Exception("Error: No assistant specified.") call_id, web_call_url = create_web_call( - self.api_url, self.api_key, assistant) + self.api_url, self.api_key, payload) if not web_call_url: raise Exception("Error: Unable to create call.")