From ebf42d6764972f8c5d3d4a48a3729fb14f69d9ae Mon Sep 17 00:00:00 2001 From: Miri Bar Date: Wed, 11 Sep 2024 13:35:40 +0300 Subject: [PATCH] docs: update ai21 docs --- docs/my-website/docs/providers/ai21.md | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/docs/my-website/docs/providers/ai21.md b/docs/my-website/docs/providers/ai21.md index f331fcff9..90e69bd29 100644 --- a/docs/my-website/docs/providers/ai21.md +++ b/docs/my-website/docs/providers/ai21.md @@ -3,7 +3,12 @@ import TabItem from '@theme/TabItem'; # AI21 -LiteLLM supports j2-light, j2-mid and j2-ultra from [AI21](https://www.ai21.com/studio/pricing) +LiteLLM supports the following [AI21](https://www.ai21.com/studio/pricing) models: +* `jamba-1.5-mini` +* `jamba-1.5-large` +* `j2-light` +* `j2-mid` +* `j2-ultra` :::tip @@ -30,7 +35,7 @@ os.environ["AI21_API_KEY"] = "your-api-key" messages = [{"role": "user", "content": "Write me a poem about the blue sky"}] -completion(model="j2-light", messages=messages) +completion(model="ai21/jamba-1.5-mini", messages=messages) ```