fix(anthropic.py): fix anthropic prompt

This commit is contained in:
Krrish Dholakia 2023-10-09 15:22:51 -07:00
parent ba754a07a3
commit a6968d06e6
2 changed files with 4 additions and 4 deletions

View file

@ -8,8 +8,8 @@ from litellm.utils import ModelResponse
import litellm import litellm
class AnthropicConstants(Enum): class AnthropicConstants(Enum):
HUMAN_PROMPT = "\n\nHuman:" HUMAN_PROMPT = "\n\nHuman: "
AI_PROMPT = "\n\nAssistant:" AI_PROMPT = "\n\nAssistant: "
class AnthropicError(Exception): class AnthropicError(Exception):
def __init__(self, status_code, message): def __init__(self, status_code, message):

View file

@ -164,8 +164,8 @@ class AmazonAI21Config():
and v is not None} and v is not None}
class AnthropicConstants(Enum): class AnthropicConstants(Enum):
HUMAN_PROMPT = "\n\nHuman:" HUMAN_PROMPT = "\n\nHuman: "
AI_PROMPT = "\n\nAssistant:" AI_PROMPT = "\n\nAssistant: "
def init_bedrock_client( def init_bedrock_client(