scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

(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
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_text_completion.py
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",