Add error handling for empty system prompt in litellm_ai_handler and type conversion in utils.py

This commit is contained in:
mrT23
2024-08-13 16:26:32 +03:00
parent 38638bd1c4
commit e7e3970874
2 changed files with 5 additions and 1 deletions

View File

@ -106,6 +106,10 @@ class LiteLLMAIHandler(BaseAiHandler):
deployment_id = self.deployment_id
if self.azure:
model = 'azure/' + model
if 'claude' in model and not system:
system = "\n"
get_logger().warning(
"Empty system prompt for claude model. Adding a newline character to prevent OpenAI API error.")
messages = [{"role": "system", "content": system}, {"role": "user", "content": user}]
if img_path:
try: