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

refactor(openai.py): more logging around failed openai calls

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-29 19:30:31 -08:00
parent 7d42330d50
commit 17e1485fbe
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/openai.py
View file

@ -334,10 +334,14 @@ class OpenAIChatCompletion(BaseLLM):
model_response_object=model_response, model_response_object=model_response,
) )
except Exception as e: except Exception as e:
if print_verbose is not None:
print_verbose(f"openai.py: Received openai error - {str(e)}")
if ( if (
"Conversation roles must alternate user/assistant" in str(e) "Conversation roles must alternate user/assistant" in str(e)
or "user and assistant roles should be alternating" in str(e) or "user and assistant roles should be alternating" in str(e)
) and messages is not None: ) and messages is not None:
if print_verbose is not None:
print_verbose("openai.py: REFORMATS THE MESSAGE!")
# reformat messages to ensure user/assistant are alternating, if there's either 2 consecutive 'user' messages or 2 consecutive 'assistant' message, add a blank 'user' or 'assistant' message to ensure compatibility # reformat messages to ensure user/assistant are alternating, if there's either 2 consecutive 'user' messages or 2 consecutive 'assistant' message, add a blank 'user' or 'assistant' message to ensure compatibility
new_messages = [] new_messages = []
for i in range(len(messages) - 1): # type: ignore for i in range(len(messages) - 1): # type: ignore