phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(types/utils.py): fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-18 20:18:29 -07:00
parent dae6659dbc
commit b79e21a81a
1 changed files with 2 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/types/utils.py
View file

@ -1,22 +1,13 @@
import json
import time
import uuid
import json
import time
import uuid
from enum import Enum
from typing import Dict, List, Literal, Optional, Tuple, Union
from typing import Dict, List, Literal, Optional, Tuple, Union
from openai._models import BaseModel as OpenAIObject
from pydantic import ConfigDict
from typing_extensions import Dict, Required, TypedDict, override
from ..litellm_core_utils.core_helpers import map_finish_reason
from .llms.openai import ChatCompletionToolCallChunk, ChatCompletionUsageBlock
from typing_extensions import Dict, Required, TypedDict, override
from ..litellm_core_utils.core_helpers import map_finish_reason
from .llms.openai import ChatCompletionToolCallChunk, ChatCompletionUsageBlock
@ -548,9 +539,9 @@ class ModelResponse(OpenAIObject):
new_choices = []
for choice in choices:
if isinstance(choice, Choices):
_new_choice = choice
_new_choice = choice # type: ignore
elif isinstance(choice, dict):
_new_choice = Choices(**choice)
_new_choice = Choices(**choice) # type: ignore
new_choices.append(_new_choice)
choices = new_choices
else: