Test fixes

This commit is contained in:
Ashwin Bharambe 2024-10-08 16:57:52 -07:00 committed by Ashwin Bharambe
parent 8eee5b9adc
commit f40cd62306
3 changed files with 1 additions and 3 deletions

View file

@ -84,7 +84,7 @@ class InferenceClient(Inference):
response.raise_for_status()
j = response.json()
yield ChatCompletionResponse(**j)
return ChatCompletionResponse(**j)
async def _stream_chat_completion(
self, request: ChatCompletionRequest