scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

add bedrock.anthropic support for system prompt using <admin> tag

Browse source
This commit is contained in:
canada4663 2023-09-30 21:22:46 -07:00
parent 0c640ab5ef
commit 07ed0abf82
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/llms/bedrock.py
View file

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