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

add import for co, anth

Browse source
This commit is contained in:
ishaan-jaff 2023-08-08 10:46:20 -07:00
parent b6f601bb86
commit a4bebe6815
3 changed files with 11 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -156,6 +156,7 @@ def completion(
elif "replicate" in model:
# import replicate/if it fails then pip install replicate
install_and_import("replicate")
import replicate
# replicate defaults to os.environ.get("REPLICATE_API_TOKEN")
# checking in case user set it to REPLICATE_API_KEY instead
if not get_secret("REPLICATE_API_TOKEN") and get_secret("REPLICATE_API_KEY"):
@ -246,6 +247,7 @@ def completion(
elif model in litellm.cohere_models:
# import cohere/if it fails then pip install cohere
install_and_import("cohere")
import cohere
if api_key:
cohere_key = api_key
elif litellm.cohere_key: