Skip to content

Commit 8e066ce

Browse files
committed
fix: couple more issues
1 parent cbbc1b3 commit 8e066ce

File tree

1 file changed

+9
-17
lines changed

1 file changed

+9
-17
lines changed

deepgram/clients/agent/v1/websocket/options.py

Lines changed: 9 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -112,18 +112,12 @@ class Function(BaseResponse):
112112

113113
def __getitem__(self, key):
114114
_dict = self.to_dict()
115-
if "parameters" in _dict:
116-
_dict["parameters"] = [
117-
Parameters.from_dict(parameters) for parameters in _dict["parameters"]
118-
]
119-
if "headers" in _dict:
120-
_dict["headers"] = [
121-
Header.from_dict(headers) for headers in _dict["headers"]
122-
]
123-
if "endpoint" in _dict:
124-
_dict["endpoint"] = [
125-
Endpoint.from_dict(endpoint) for endpoint in _dict["endpoint"]
126-
]
115+
if "parameters" in _dict and isinstance(_dict["parameters"], dict):
116+
_dict["parameters"] = Parameters.from_dict(_dict["parameters"])
117+
if "headers" in _dict and isinstance(_dict["headers"], list):
118+
_dict["headers"] = [Header.from_dict(header) for header in _dict["headers"]]
119+
if "endpoint" in _dict and isinstance(_dict["endpoint"], dict):
120+
_dict["endpoint"] = Endpoint.from_dict(_dict["endpoint"])
127121
return _dict[key]
128122

129123

@@ -155,7 +149,7 @@ class ListenProvider(BaseResponse):
155149

156150
def __getitem__(self, key):
157151
_dict = self.to_dict()
158-
if "keyterms" in _dict:
152+
if "keyterms" in _dict and isinstance(_dict["keyterms"], list):
159153
_dict["keyterms"] = [str(keyterm) for keyterm in _dict["keyterms"]]
160154
return _dict[key]
161155

@@ -214,10 +208,8 @@ class SpeakProvider(BaseResponse):
214208

215209
def __getitem__(self, key):
216210
_dict = self.to_dict()
217-
if "voice" in _dict:
218-
_dict["voice"] = [
219-
CartesiaVoice.from_dict(voice) for voice in _dict["voice"]
220-
]
211+
if "voice" in _dict and isinstance(_dict["voice"], dict):
212+
_dict["voice"] = CartesiaVoice.from_dict(_dict["voice"])
221213
return _dict[key]
222214

223215

0 commit comments

Comments
 (0)