ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove junk hf stream print statement

Browse Source
This commit is contained in:
ishaan-jaff 2023-09-28 17:00:42 -07:00
parent 09b8c08cad
commit a604ca577b
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
litellm/tests/test_completion.py
View File

@ -73,7 +73,7 @@ def test_completion_claude_max_tokens():
litellm.set_verbose = False
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_claude_max_tokens()
# test_completion_claude_max_tokens()
# def test_completion_oobabooga():
# try:
@ -209,6 +209,22 @@ def test_get_hf_task_for_model():
# pytest.fail(f"Error occurred: {e}")
# hf_test_completion_tgi()
# def hf_test_completion_tgi_stream():
# try:
# response = litellm.completion(
# model="huggingface/glaiveai/glaive-coder-7b",
# messages=[{ "content": "Hello, how are you?","role": "user"}],
# api_base="https://wjiegasee9bmqke2.us-east-1.aws.endpoints.huggingface.cloud",
# stream=True
# )
# # Add any assertions here to check the response
# print(response)
# for chunk in response:
# print(chunk)
# except Exception as e:
# pytest.fail(f"Error occurred: {e}")
# hf_test_completion_tgi_stream()
# ################### Hugging Face Conversational models ########################
# def hf_test_completion_conv():
# try: