phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): fix linting error

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-09 20:47:12 -07:00
parent f899b828cf
commit 574f5056c8
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/types/utils.py
View file

@ -829,6 +829,9 @@ class StreamingChoices(OpenAIObject):
enhancements=None, enhancements=None,
**params, **params,
): ):
# Fix Perplexity return both delta and message cause OpenWebUI repect text
# https://github.com/BerriAI/litellm/issues/8455
params.pop("message", None)
super(StreamingChoices, self).__init__(**params) super(StreamingChoices, self).__init__(**params)
if finish_reason: if finish_reason:
self.finish_reason = map_finish_reason(finish_reason) self.finish_reason = map_finish_reason(finish_reason)
@ -836,10 +839,7 @@ class StreamingChoices(OpenAIObject):
self.finish_reason = None self.finish_reason = None
self.index = index self.index = index
if delta is not None: if delta is not None:
# Fix Perplexity return both delta and message cause OpenWebUI repect text
# https://github.com/BerriAI/litellm/issues/8455
if 'message' in params:
del self.message
if isinstance(delta, Delta): if isinstance(delta, Delta):
self.delta = delta self.delta = delta
elif isinstance(delta, dict): elif isinstance(delta, dict):