fix bad test

This commit is contained in:
ishaan-jaff 2023-07-31 17:02:35 -07:00
parent 929b6959a0
commit 8e6d344db2
2 changed files with 2 additions and 3 deletions

View file

@ -186,7 +186,7 @@ def completion(
model=model,
prompt = prompt
)
elif "replicate" in model:
elif "replicate" in model:
# replicate defaults to os.environ.get("REPLICATE_API_TOKEN")
# checking in case user set it to REPLICATE_API_KEY instead
if not os.environ.get("REPLICATE_API_TOKEN") and os.environ.get("REPLICATE_API_KEY"):