@@ -220,24 +220,24 @@ def get_message(self, response: dict) -> str:
220220 str: Message extracted
221221 """
222222 assert isinstance (response , dict ), "Response should be of dict data-type only"
223- delta : dict = response ["choices" ][0 ]["delta" ]
224-
223+ delta : dict = response ["choices" ][0 ]["delta" ]
224+
225225 if not delta :
226- return ''
227-
228- elif delta .get (' function_call' ):
229- function_call : dict = delta [' function_call' ]
230- if function_call .get (' name' ):
231- return function_call [' name' ]
232- elif function_call .get (' arguments' ):
233- return function_call .get (' arguments' )
234-
235- elif delta .get (' metadata' ):
236- return yaml .dump (delta [' metadata' ])
226+ return ""
227+
228+ elif delta .get (" function_call" ):
229+ function_call : dict = delta [" function_call" ]
230+ if function_call .get (" name" ):
231+ return function_call [" name" ]
232+ elif function_call .get (" arguments" ):
233+ return function_call .get (" arguments" )
234+
235+ elif delta .get (" metadata" ):
236+ return yaml .dump (delta [" metadata" ])
237237
238238 else :
239239 return (
240- response ["choices" ][0 ]["delta" ].get ("content" )
241- if response ["choices" ][0 ].get ("finish_reason" ) is None
242- else ""
243- )
240+ response ["choices" ][0 ]["delta" ].get ("content" )
241+ if response ["choices" ][0 ].get ("finish_reason" ) is None
242+ else ""
243+ )
0 commit comments