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

add gpt-3.5-instruct

Browse source
This commit is contained in:
ishaan-jaff 2023-09-18 12:14:51 -07:00
parent 205bc9bab6
commit 0f88b82c4f
2 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/__init__.py
View file

@ -93,6 +93,7 @@ open_ai_chat_completion_models: List = [
"gpt-3.5-turbo-16k-0613",
]
open_ai_text_completion_models: List = [
"gpt-3.5-turbo-instruct",
"text-davinci-003",
"text-curie-001",
"text-babbage-001",

9
litellm/tests/test_completion.py
View file

@ -150,7 +150,7 @@ def hf_test_completion():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
hf_test_completion()
# hf_test_completion()
def test_completion_cohere(): # commenting for now as the cohere endpoint is being flaky
try:
@ -219,6 +219,13 @@ def test_completion_text_openai():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_completion_gpt_instruct():
try:
response = completion(model="gpt-3.5-turbo-instruct", messages=messages)
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_gpt_instruct()
def test_completion_openai_with_optional_params():
try: