mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 18:24:20 +00:00
update hf testing
This commit is contained in:
parent
07c74838ff
commit
e5ceed4334
2 changed files with 1 additions and 18 deletions
|
@ -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:
|
||||
|
|
|
@ -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"
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue