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

(fix) test - deprecated textdavinci003

Browse source
This commit is contained in:
ishaan-jaff 2024-01-09 10:55:35 +05:30
parent 9c7a4fde87
commit f46fa2b8a8
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_text_completion.py
View file

@ -2721,7 +2721,9 @@ def test_completion_openai_engine():
print("\n text 003 test\n") print("\n text 003 test\n")
litellm.set_verbose = True litellm.set_verbose = True
response = text_completion( response = text_completion(
engine="gpt-3.5-turbo-instruct", prompt="What's the weather in SF?", max_tokens=5 engine="gpt-3.5-turbo-instruct",
prompt="What's the weather in SF?",
max_tokens=5,
) )
print(response) print(response)
response_str = response["choices"][0]["text"] response_str = response["choices"][0]["text"]
@ -2761,7 +2763,6 @@ def test_text_completion_basic():
prompt="good morning", prompt="good morning",
max_tokens=10, max_tokens=10,
logprobs=10, logprobs=10,
echo=True,
) )
print(response) print(response)
print(response.choices) print(response.choices)