(fix) streaming logprobs=None

This commit is contained in:
ishaan-jaff 2023-12-26 15:42:41 +05:30
parent 2070a785a4
commit ef074c707a

View file

@ -6705,6 +6705,7 @@ class CustomStreamWrapper:
text = "" text = ""
is_finished = False is_finished = False
finish_reason = None finish_reason = None
logprobs = None
original_chunk = None # this is used for function/tool calling original_chunk = None # this is used for function/tool calling
if len(str_line.choices) > 0: if len(str_line.choices) > 0:
if str_line.choices[0].delta.content is not None: if str_line.choices[0].delta.content is not None: