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

fix(anthropic.py-+-bedrock.py): anthropic prompt format

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-20 10:56:07 -07:00
parent 9347b25e37
commit 2f9e112c14
4 changed files with 60 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/main.py
View file

@ -586,7 +586,7 @@ def completion(
return response
response = model_response
elif model in litellm.anthropic_models:
elif custom_llm_provider=="anthropic":
anthropic_key = (
api_key or litellm.anthropic_key or os.environ.get("ANTHROPIC_API_KEY") or litellm.api_key
)
@ -600,6 +600,7 @@ def completion(
model=model,
messages=messages,
api_base=api_base,
custom_prompt_dict=litellm.custom_prompt_dict,
model_response=model_response,
print_verbose=print_verbose,
optional_params=optional_params,
@ -1036,6 +1037,7 @@ def completion(
model_response = bedrock.completion(
model=model,
messages=messages,
custom_prompt_dict=litellm.custom_prompt_dict,
model_response=model_response,
print_verbose=print_verbose,
optional_params=optional_params,