Skip to content

Commit 5f8a02b

Browse files
committed
Fix Phind. Resolves #21
1 parent 8f98a52 commit 5f8a02b

File tree

1 file changed

+17
-17
lines changed

1 file changed

+17
-17
lines changed

src/pytgpt/phind/main.py

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)