update hf testing

This commit is contained in:
Krrish Dholakia 2023-09-27 10:53:45 -07:00
parent 07c74838ff
commit e5ceed4334
2 changed files with 1 additions and 18 deletions

View file

@ -200,23 +200,6 @@ def hf_test_completion():
# hf_test_completion()
def hf_test_completion_task_none():
try:
litellm.set_verbose=True
user_message = "My name is Merve and my favorite"
messages = [{ "content": user_message,"role": "user"}]
model = "huggingface/cerebras/btlm-3b-8k-base"
response = completion(
model=model,
messages=messages,
task=None,
)
# Add any assertions here to check the response
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
hf_test_completion_task_none()
# this should throw an exception, to trigger https://logs.litellm.ai/
# def hf_test_error_logs():
# try:

View file

@ -1,6 +1,6 @@
[tool.poetry]
name = "litellm"
version = "0.1.783"
version = "0.1.784"
description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"]
license = "MIT License"