scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 77

fix Llama models message to prompt conversion in for AWS Bedrock provider

Browse source
This commit is contained in:
aswny 2024-04-25 17:19:55 +00:00
parent 54e0acde35
commit 9ead717531
2 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/bedrock.py
View file

@ -653,6 +653,10 @@ def convert_messages_to_prompt(model, messages, provider, custom_prompt_dict):
prompt = prompt_factory(
model=model, messages=messages, custom_llm_provider="bedrock"
)
elif provider == "meta":
prompt = prompt_factory(
model=model, messages=messages, custom_llm_provider="bedrock"
)
else:
prompt = ""
for message in messages:

10
litellm/llms/prompt_templates/factory.py
View file

@ -1346,6 +1346,16 @@ def prompt_factory(
return anthropic_pt(messages=messages)
elif "mistral." in model:
return mistral_instruct_pt(messages=messages)
elif "llama2" in model:
return llama_2_chat_pt(messages=messages)
elif "llama3" in model:
return hf_chat_template(
model=model,
messages=messages,
chat_template=known_tokenizer_config[ # type: ignore
"meta-llama/Meta-Llama-3-8B-Instruct"
]["tokenizer"]["chat_template"],
)
elif custom_llm_provider == "perplexity":
for message in messages:
message.pop("name", None)