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

fix bad test

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
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"):

3
litellm/tests/test_bad_params.py
View file

@ -10,7 +10,6 @@ parent_dir = os.path.join(current_dir, '..')
# Add the parent directory to the system path
sys.path.append(parent_dir)
import main
from main import embedding, completion
main.success_callback = ["posthog"]
@ -20,7 +19,7 @@ main.set_verbose = True
user_message = "Hello, how are you?"
messages = [{ "content": user_message,"role": "user"}]
model_val = None
model_val = "krrish is a model"
# test on empty
try:
response = completion(model=model_val, messages=messages)