phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(test) text_completion

Browse source
This commit is contained in:
ishaan-jaff 2023-11-06 13:17:49 -08:00
parent b4797bec3b
commit 07761ac93f
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_text_completion.py
View file

@ -72,7 +72,8 @@ def test_text_completion_basic():
# print("\n testing hf mistral\n") # print("\n testing hf mistral\n")
# response = text_completion( # response = text_completion(
# model="huggingface/mistralai/Mistral-7B-v0.1", # model="huggingface/mistralai/Mistral-7B-v0.1",
# prompt=token_prompt, # token prompt is a 2d list # prompt=token_prompt, # token prompt is a 2d list,
# max_tokens=0.0
# ) # )
# print("\n\n response") # print("\n\n response")
@ -98,7 +99,7 @@ def test_completion_text_003_prompt_array():
# response_str = response["choices"][0]["text"] # response_str = response["choices"][0]["text"]
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
test_completion_text_003_prompt_array() # test_completion_text_003_prompt_array()
# not including this in our ci cd pipeline, since we don't want to fail tests due to an unstable replit # not including this in our ci cd pipeline, since we don't want to fail tests due to an unstable replit