Merge pull request #2320 from BerriAI/litellm_claude_3_image_tool_calling_support

fix(anthropic.py): Support Claude 3 Vision + Tool Calling
This commit is contained in:
Krish Dholakia 2024-03-04 16:16:08 -08:00 committed by GitHub
commit a9133ae139
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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":