phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(openai-py): fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-10 21:56:14 -07:00
parent 68461f5863
commit dfd91c5d46
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/openai.py
View file

@ -210,8 +210,10 @@ class OpenAIChatCompletion(BaseLLM):
headers["Authorization"] = f"Bearer {api_key}" headers["Authorization"] = f"Bearer {api_key}"
return headers return headers
def convert_to_model_response_object(self, response_object: dict, model_response_object: ModelResponse): def convert_to_model_response_object(self, response_object: Optional[dict]=None, model_response_object: Optional[ModelResponse]=None):
try: try:
if response_object is None or model_response_object is None:
raise CustomOpenAIError(status_code=500, message="Error in response object format")
choice_list=[] choice_list=[]
for idx, choice in enumerate(response_object["choices"]): for idx, choice in enumerate(response_object["choices"]):
message = Message(content=choice["message"]["content"], role=choice["message"]["role"]) message = Message(content=choice["message"]["content"], role=choice["message"]["role"])

1
litellm/tests/test_exceptions.py
View file

@ -12,7 +12,6 @@ from litellm import (
embedding, embedding,
completion, completion,
# AuthenticationError, # AuthenticationError,
InvalidRequestError,
ContextWindowExceededError, ContextWindowExceededError,
# RateLimitError, # RateLimitError,
# ServiceUnavailableError, # ServiceUnavailableError,