ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
Show Stats 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 += (
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']}"

1
litellm/llms/bedrock.py
View File

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