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

(test) AI21 test

Browse source
This commit is contained in:
ishaan-jaff 2023-11-14 15:01:50 -08:00
parent 75cf3b3555
commit d80e2310ed
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_completion.py
View file

@ -639,6 +639,7 @@ def test_azure_openai_ad_token():
def tester(
kwargs, # kwargs to completion
):
print(kwargs["additional_args"])
if kwargs["additional_args"]["headers"]["Authorization"] != 'Bearer gm':
pytest.fail("AZURE AD TOKEN Passed but not set in request header")
return
@ -656,9 +657,11 @@ def test_azure_openai_ad_token():
)
print("azure ad token respoonse\n")
print(response)
litellm.input_callback = []
except:
litellm.input_callback = []
pass
test_azure_openai_ad_token()
# test_azure_openai_ad_token()
# test_completion_azure()
@ -1357,6 +1360,8 @@ def test_mistral_anyscale_stream():
#### Test A121 ###################
def test_completion_ai21():
print("running ai21 j2light test")
litellm.set_verbose=True
model_name = "j2-light"
try:
response = completion(model=model_name, messages=messages, max_tokens=100, temperature=0.8)