diff --git a/litellm/main.py b/litellm/main.py index c5ab46423..e9148414f 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -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"): diff --git a/litellm/tests/test_bad_params.py b/litellm/tests/test_bad_params.py index 49018bd19..60dd734c9 100644 --- a/litellm/tests/test_bad_params.py +++ b/litellm/tests/test_bad_params.py @@ -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)