leolaish commited on
Commit
38ed5bc
·
verified ·
1 Parent(s): 0d4014c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -530,9 +530,10 @@ def parse_data_chunk(data_chunk):
530
  Returns:
531
  list: A list of individual data entries with whitespace stripped.
532
  """
 
 
 
533
  chunks = data_chunk.split("data:")
534
- stripped_chunks = map(lambda chunk: chunk.strip(), chunks)
535
- return [chunk for chunk in stripped_chunks if chunk]
536
 
537
 
538
  def generate(message, temperature):
 
530
  Returns:
531
  list: A list of individual data entries with whitespace stripped.
532
  """
533
+ def parse_data_chunk(data_chunk):
534
+ if isinstance(data_chunk, ChatCompletionStreamOutput):
535
+ data_chunk = data_chunk.text
536
  chunks = data_chunk.split("data:")
 
 
537
 
538
 
539
  def generate(message, temperature):