From 0970475a3bfb9fcf9d7258b7100b3b60f6b1d70a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 2 Sep 2024 11:47:48 -0700 Subject: [PATCH] test ai21 --- litellm/tests/test_completion.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 8fdf722f06..da6e33a96e 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -4474,3 +4474,13 @@ async def test_dynamic_azure_params(stream, sync_mode): except Exception as e: traceback.print_stack() raise e + + +@pytest.mark.asyncio() +async def test_completion_ai21(): + litellm.set_verbose = True + response = await litellm.acompletion( + model="ai21_chat/jamba-1.5-large", + messages=[{"role": "user", "content": "hi my name is ishaan"}], + ) + pass