diff --git a/docs/my-website/docs/providers/datarobot.md b/docs/my-website/docs/providers/datarobot.md new file mode 100644 index 00000000000..3f4a0f71ac4 --- /dev/null +++ b/docs/my-website/docs/providers/datarobot.md @@ -0,0 +1,43 @@ +import Tabs from '@theme/Tabs'; +import TabItem from '@theme/TabItem'; + +# DataRobot +LiteLLM supports all models from [DataRobot](https://datarobot.com). Select `datarobot` as the provider to route your request through the `datarobot` OpenAI-compatible endpoint using the upstream [official OpenAI Python API library](https://github.com/openai/openai-python/blob/main/README.md). + +## Usage + +### Environment variables +```python +import os +from litellm import completion +os.environ["DATAROBOT_API_KEY"] = "" +os.environ["DATAROBOT_API_BASE"] = "" # [OPTIONAL] defaults to https://app.datarobot.com + +response = completion( + model="datarobot/openai/gpt-4o-mini", + messages=messages, + ) + + +### Completion +```python +import litellm +import os + +response = litellm.completion( + model="datarobot/openai/gpt-4o-mini", # add `datarobot/` prefix to model so litellm knows to route through DataRobot + messages=[ + { + "role": "user", + "content": "Hey, how's it going?", + } + ], +) +print(response) +``` + +## DataRobot completion models + +🚨 LiteLLM supports _all_ DataRobot LLM gateway models. To get a list for your installation and user account, send the following CURL command: +`curl -X GET -H "Authorization: Bearer $DATAROBOT_API_TOKEN" "$DATAROBOT_ENDPOINT/genai/llmgw/catalog/" | jq | grep 'model":'DATAROBOT_ENDPOINT/genai/llmgw/catalog/` +