scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

test

Browse source
This commit is contained in:
ishaan-jaff 2023-09-16 09:27:39 -07:00
parent d2574e67b7
commit 93fbe4a733
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_completion.py
View file

@ -664,12 +664,13 @@ def test_completion_bedrock_titan():
def test_completion_bedrock_ai21():
try:
litellm.set_verbose = False
response = completion(
model="bedrock/ai21.j2-mid",
messages=messages,
temperature=0.2,
max_tokens=20,
logger_fn=logger_fn
top_p=0.2,
max_tokens=20
)
# Add any assertions here to check the response
print(response)