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

(test) text completion add logprobs to test

Browse Source
This commit is contained in:
ishaan-jaff 2023-10-31 17:15:34 -07:00
parent 9223f7cc7a
commit e180bba541
1 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
litellm/tests/test_completion.py
View File

@ -56,7 +56,7 @@ def test_completion_claude():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_claude()
# test_completion_claude()
# def test_completion_oobabooga():
# try:
@ -402,13 +402,28 @@ def test_completion_openai_prompt_array():
try:
litellm.set_verbose=False
response = text_completion(
model="text-davinci-003", prompt=[[3152, 833]], max_tokens=10, logprobs=10
model="text-davinci-003", prompt=[[3152, 833]], max_tokens=10, logprobs=10, echo=True
)
print(response)
response_str = response["choices"][0]["text"]
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_openai_prompt_array()
test_completion_openai_prompt_array()
def test_completion_hf_prompt_array():
try:
litellm.set_verbose=False
response = text_completion(
model="huggingface/bigcode/starcoder",
prompt="good morning",
max_tokens=10, logprobs=10,
echo=True
)
print(response)
response_str = response["choices"][0]["text"]
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_hf_prompt_array()
def test_completion_text_openai():
try: