forked from phoenix/litellm-mirror
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()
|
# 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/
|
# this should throw an exception, to trigger https://logs.litellm.ai/
|
||||||
# def hf_test_error_logs():
|
# def hf_test_error_logs():
|
||||||
# try:
|
# try:
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
[tool.poetry]
|
[tool.poetry]
|
||||||
name = "litellm"
|
name = "litellm"
|
||||||
version = "0.1.783"
|
version = "0.1.784"
|
||||||
description = "Library to easily interface with LLM API providers"
|
description = "Library to easily interface with LLM API providers"
|
||||||
authors = ["BerriAI"]
|
authors = ["BerriAI"]
|
||||||
license = "MIT License"
|
license = "MIT License"
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue