tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-24 02:25:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

(test) regular hf tests

Browse Source
This commit is contained in:
ishaan-jaff
2023-11-16 12:00:49 -08:00
parent da8c2f4a4a
commit 3285113d2d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_text_completion.py
+3 -3
View File
@@ -46,7 +46,7 @@ def test_completion_openai_engine_and_model():
#print(response.choices[0].text)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_openai_engine_and_model()
# test_completion_openai_engine_and_model()
def test_completion_openai_engine():
try:
@@ -61,7 +61,7 @@ def test_completion_openai_engine():
#print(response.choices[0].text)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_openai_engine()
# test_completion_openai_engine()
def test_completion_chatgpt_prompt():
@@ -135,7 +135,7 @@ def test_completion_text_003_prompt_array():
##### hugging face tests
def test_completion_hf_prompt_array():
try:
litellm.set_verbose=False
litellm.set_verbose=True
print("\n testing hf mistral\n")
response = text_completion(
model="huggingface/mistralai/Mistral-7B-v0.1",