phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) add text_completion with prompt list

Browse source
This commit is contained in:
ishaan-jaff 2023-11-03 18:02:50 -07:00
parent df57e9247a
commit 34751d8562
1 changed files with 17 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

18
litellm/tests/test_text_completion.py
View file

@ -69,4 +69,20 @@ def test_completion_hf_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_hf_prompt_array() # test_completion_hf_prompt_array()
def test_completion_text_003_prompt_array():
try:
litellm.set_verbose=False
response = text_completion(
model="text-davinci-003",
prompt=token_prompt, # token prompt is a 2d list
)
print("\n\n response")
print(response)
# response_str = response["choices"][0]["text"]
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_text_003_prompt_array()