phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

try passing tests

Browse source
This commit is contained in:
ishaan-jaff 2023-07-31 17:40:08 -07:00
parent 5b5bd59783
commit d2e29ea0ca
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/main.py
View file

@ -283,8 +283,8 @@ def completion(
], ],
} }
response = new_response response = new_response
else: else:
logging(model=model, input=messages, azure=azure, logger_fn=logger_fn) raise Exception(f"Model '{model}' not found. Please check your model name and try again.")
return response return response
except Exception as e: except Exception as e:
logging(model=model, input=messages, azure=azure, additional_args={"max_tokens": max_tokens}, logger_fn=logger_fn) logging(model=model, input=messages, azure=azure, additional_args={"max_tokens": max_tokens}, logger_fn=logger_fn)

4
litellm/tests/test_bad_params.py
View file

@ -23,6 +23,6 @@ model_val = "krrish is a model"
# test on empty # test on empty
try: try:
response = completion(model=model_val, messages=messages) response = completion(model=model_val, messages=messages)
except: except Exception as e:
print(f"error occurred: {traceback.format_exc()}") print(f"error occurred: {e}")
pass pass