phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(factory.py): fix prompt mapping

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-04 14:11:11 -08:00
parent 0c11827a53
commit 091cea1acb
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -829,7 +829,7 @@ def prompt_factory(
if custom_llm_provider == "ollama": if custom_llm_provider == "ollama":
return ollama_pt(model=model, messages=messages) return ollama_pt(model=model, messages=messages)
elif custom_llm_provider == "anthropic": elif custom_llm_provider == "anthropic":
if model == "claude-instant-1" or model == "claude-2.1": if model == "claude-instant-1" or model == "claude-2":
return anthropic_pt(messages=messages) return anthropic_pt(messages=messages)
return anthropic_messages_pt(messages=messages) return anthropic_messages_pt(messages=messages)
elif custom_llm_provider == "together_ai": elif custom_llm_provider == "together_ai":