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

added similar changes to anthropic.py based on recommendation from @krrishdholakia

Browse source
This commit is contained in:
canada4663 2023-10-01 20:54:36 -07:00
parent 07ed0abf82
commit 2896df2645
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/anthropic.py
View file

@ -52,6 +52,10 @@ def completion(
prompt += ( prompt += (
f"{AnthropicConstants.HUMAN_PROMPT.value}{message['content']}" f"{AnthropicConstants.HUMAN_PROMPT.value}{message['content']}"
) )
elif message["role"] == "system":
prompt += (
f"{AnthropicConstants.HUMAN_PROMPT.value}<admin>{message['content']}</admin>"
)
else: else:
prompt += ( prompt += (
f"{AnthropicConstants.AI_PROMPT.value}{message['content']}" f"{AnthropicConstants.AI_PROMPT.value}{message['content']}"

1
litellm/llms/bedrock.py
View file

@ -44,7 +44,6 @@ def convert_messages_to_prompt(messages, provider):
prompt += ( prompt += (
f"{AnthropicConstants.HUMAN_PROMPT.value}<admin>{message['content']}</admin>" f"{AnthropicConstants.HUMAN_PROMPT.value}<admin>{message['content']}</admin>"
) )
else: else:
prompt += ( prompt += (
f"{AnthropicConstants.AI_PROMPT.value}{message['content']}" f"{AnthropicConstants.AI_PROMPT.value}{message['content']}"