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

Merge pull request #5631 from miri-bar/update_ai21_docs

Browse source 
docs: update ai21 docs
This commit is contained in:
Ishaan Jaff 2024-09-11 09:27:35 -07:00 • committed by GitHub
commit ccda616f2f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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 # 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 :::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"}] 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)
``` ```