fix(anthropic.py): bug fix

This commit is contained in:
Krrish Dholakia 2024-03-12 19:32:42 -07:00
parent e892fc99a3
commit d620b4dc5d
2 changed files with 3 additions and 3 deletions

View file

@ -270,8 +270,8 @@ def completion(
streaming_model_response.choices[0].finish_reason = model_response.choices[
0
].finish_reason
# streaming_model_response.choices = [litellm.utils.StreamingChoices()]
streaming_model_response.choices[0].index = model_response.choices[0].index
streaming_model_response.choices = [litellm.utils.StreamingChoices()]
_tool_calls = []
print_verbose(
f"type of model_response.choices[0]: {type(model_response.choices[0])}"

View file

@ -480,12 +480,12 @@ class ModelResponse(OpenAIObject):
object=None,
system_fingerprint=None,
usage=None,
stream=False,
stream=None,
response_ms=None,
hidden_params=None,
**params,
):
if stream:
if stream is not None and stream == True:
object = "chat.completion.chunk"
choices = [StreamingChoices()]
else: