From a6968d06e6f72fe6b80a06ed5452ff1f02ef1eb7 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 9 Oct 2023 15:22:51 -0700 Subject: [PATCH] fix(anthropic.py): fix anthropic prompt --- litellm/llms/anthropic.py | 4 ++-- litellm/llms/bedrock.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/llms/anthropic.py b/litellm/llms/anthropic.py index 5b8c5f27f0..6dac6b0425 100644 --- a/litellm/llms/anthropic.py +++ b/litellm/llms/anthropic.py @@ -8,8 +8,8 @@ from litellm.utils import ModelResponse import litellm class AnthropicConstants(Enum): - HUMAN_PROMPT = "\n\nHuman:" - AI_PROMPT = "\n\nAssistant:" + HUMAN_PROMPT = "\n\nHuman: " + AI_PROMPT = "\n\nAssistant: " class AnthropicError(Exception): def __init__(self, status_code, message): diff --git a/litellm/llms/bedrock.py b/litellm/llms/bedrock.py index 2ccb6f9d8b..f6bad8acbf 100644 --- a/litellm/llms/bedrock.py +++ b/litellm/llms/bedrock.py @@ -164,8 +164,8 @@ class AmazonAI21Config(): and v is not None} class AnthropicConstants(Enum): - HUMAN_PROMPT = "\n\nHuman:" - AI_PROMPT = "\n\nAssistant:" + HUMAN_PROMPT = "\n\nHuman: " + AI_PROMPT = "\n\nAssistant: " def init_bedrock_client(