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

ai21 models

Browse source
This commit is contained in:
ishaan-jaff 2023-08-14 16:40:26 -07:00
parent ccec725be4
commit 06b5579ba6
4 changed files with 51 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_completion.py
View file

@ -213,6 +213,16 @@ def test_completion_together_ai_stream():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
#### Test A121 ###################
def test_completion_ai21():
model_name = "j2-light"
try:
response = completion(model=model_name, messages=messages)
# Add any assertions here to check the response
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test config file with completion #
# def test_completion_openai_config():
# try: