Compare commits

...

1 commit

Author SHA1 Message Date
monosans 45613e0537
Replace only first occurrence of data: in response 2023-10-08 16:13:28 +03:00
2 changed files with 2 additions and 2 deletions

View file

@ -151,7 +151,7 @@ async def respond(
chunk = chunk.decode('utf8').strip()
if 'azure' in provider_name:
chunk = chunk.strip().replace('data: ', '')
chunk = chunk.strip().replace('data: ', '', 1)
if not chunk or 'prompt_filter_results' in chunk:
continue

View file

@ -100,7 +100,7 @@ async def test_chat_stream_gpt3() -> float:
async for chunk in response.aiter_text():
for subchunk in chunk.split('\n\n'):
chunk = subchunk.replace('data: ', '').strip()
chunk = subchunk.replace('data: ', '', 1).strip()
if chunk == '[DONE]':
break