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

fixing claude exception-mapping + set key bug

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-07 12:07:33 -07:00
parent 1a4044091a
commit dd6b4db090
6 changed files with 24 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -164,7 +164,6 @@ def completion(
os.environ["REPLICATE_API_TOKEN"] = api_key
elif litellm.replicate_key:
os.environ["REPLICATE_API_TOKEN"] = litellm.replicate_key
prompt = " ".join([message["content"] for message in messages])
input = {"prompt": prompt}
if max_tokens != float('inf'):
@ -199,7 +198,7 @@ def completion(
if api_key:
os.environ["ANTHROPIC_API_KEY"] = api_key
elif litellm.anthropic_key:
os.environ["ANTHROPIC_API_TOKEN"] = litellm.anthropic_key
os.environ["ANTHROPIC_API_KEY"] = litellm.anthropic_key
prompt = f"{HUMAN_PROMPT}"
for message in messages:
if "role" in message: