phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

docs: update ai21 docs

Browse source
This commit is contained in:
Miri Bar 2024-09-11 13:35:40 +03:00
parent ce5182710f
commit ebf42d6764
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
docs/my-website/docs/providers/ai21.md
View file

@ -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)
```