tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-21 06:27:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(factory.py): fix prompt mapping

Browse Source
This commit is contained in:
Krrish Dholakia
2024-03-04 17:15:47 -08:00
parent 78efe027b2
commit 5e93bad4af
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/llms/prompt_templates/factory.py
+1 -1
View File
@@ -829,7 +829,7 @@ def prompt_factory(
if custom_llm_provider == "ollama":
return ollama_pt(model=model, messages=messages)
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_messages_pt(messages=messages)
elif custom_llm_provider == "together_ai":