From 5c73b173874ec281a0cec6b17ed5c2820cb397d1 Mon Sep 17 00:00:00 2001 From: mjnitz02 Date: Tue, 12 Aug 2025 11:43:38 -0400 Subject: [PATCH 1/3] Add initial docs --- docs/my-website/docs/providers/datarobot.md | 55 +++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 docs/my-website/docs/providers/datarobot.md diff --git a/docs/my-website/docs/providers/datarobot.md b/docs/my-website/docs/providers/datarobot.md new file mode 100644 index 00000000000..a5534fa57b6 --- /dev/null +++ b/docs/my-website/docs/providers/datarobot.md @@ -0,0 +1,55 @@ +import Tabs from '@theme/Tabs'; +import TabItem from '@theme/TabItem'; + +# DataRobot +LiteLLM supports all models from [DataRobot](https://datarobot.com). Selecting `datarobot` as the provider routes 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, + ) +``` + + +## Usage - 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 + api_key="sk-1234", # api key to your openai compatible endpoint + api_base="https://app.datarobot.com", # set API Base of your DataRobot Endpoint + messages=[ + { + "role": "user", + "content": "Hey, how's it going?", + } + ], +) +print(response) +``` + +## OpenRouter Completion Models + +🚨 LiteLLM supports ALL OpenRouter models, send `model=openrouter/` to send it to open router. See all openrouter models [here](https://openrouter.ai/models) + +| Model Name | Function Call | +|---------------------------|-----------------------------------------------------| +| openrouter/openai/gpt-3.5-turbo | `completion('openrouter/openai/gpt-3.5-turbo', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | +| openrouter/openai/gpt-3.5-turbo-16k | `completion('openrouter/openai/gpt-3.5-turbo-16k', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | +| openrouter/openai/gpt-4 | `completion('openrouter/openai/gpt-4', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | +| openrouter/openai/gpt-4-32k | `completion('openrouter/openai/gpt-4-32k', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | +| openrouter/anthropic/claude-2 | `completion('openrouter/anthropic/claude-2', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | +| openrouter/anthropic/claude-instant-v1 | `completion('openrouter/anthropic/claude-instant-v1', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | +| openrouter/google/palm-2-chat-bison | `completion('openrouter/google/palm-2-chat-bison', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | +| openrouter/google/palm-2-codechat-bison | `completion('openrouter/google/palm-2-codechat-bison', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | +| openrouter/meta-llama/llama-2-13b-chat | `completion('openrouter/meta-llama/llama-2-13b-chat', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | +| openrouter/meta-llama/llama-2-70b-chat | `completion('openrouter/meta-llama/llama-2-70b-chat', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | From bf55bf507278e4d7fa7b23774290436d51569dba Mon Sep 17 00:00:00 2001 From: Carson Gee Date: Mon, 18 Aug 2025 09:06:31 -0600 Subject: [PATCH 2/3] Update datarobot.md --- docs/my-website/docs/providers/datarobot.md | 19 +++---------------- 1 file changed, 3 insertions(+), 16 deletions(-) diff --git a/docs/my-website/docs/providers/datarobot.md b/docs/my-website/docs/providers/datarobot.md index a5534fa57b6..1149d11bd2d 100644 --- a/docs/my-website/docs/providers/datarobot.md +++ b/docs/my-website/docs/providers/datarobot.md @@ -25,8 +25,6 @@ import os response = litellm.completion( model="datarobot/openai/gpt-4o-mini", # add `datarobot/` prefix to model so litellm knows to route through DataRobot - api_key="sk-1234", # api key to your openai compatible endpoint - api_base="https://app.datarobot.com", # set API Base of your DataRobot Endpoint messages=[ { "role": "user", @@ -37,19 +35,8 @@ response = litellm.completion( print(response) ``` -## OpenRouter Completion Models +## DataRobot Completion Models -🚨 LiteLLM supports ALL OpenRouter models, send `model=openrouter/` to send it to open router. See all openrouter models [here](https://openrouter.ai/models) +🚨 LiteLLM supports _all_ DataRobot LLM gateway models, to get a list for your installation and user account, you can send this simple CURL command: +`curl -X GET -H "Authorization: Bearer $DATAROBOT_API_TOKEN" "$DATAROBOT_ENDPOINT/genai/llmgw/catalog/" | jq | grep 'model":'DATAROBOT_ENDPOINT/genai/llmgw/catalog/` -| Model Name | Function Call | -|---------------------------|-----------------------------------------------------| -| openrouter/openai/gpt-3.5-turbo | `completion('openrouter/openai/gpt-3.5-turbo', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | -| openrouter/openai/gpt-3.5-turbo-16k | `completion('openrouter/openai/gpt-3.5-turbo-16k', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | -| openrouter/openai/gpt-4 | `completion('openrouter/openai/gpt-4', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | -| openrouter/openai/gpt-4-32k | `completion('openrouter/openai/gpt-4-32k', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | -| openrouter/anthropic/claude-2 | `completion('openrouter/anthropic/claude-2', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | -| openrouter/anthropic/claude-instant-v1 | `completion('openrouter/anthropic/claude-instant-v1', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | -| openrouter/google/palm-2-chat-bison | `completion('openrouter/google/palm-2-chat-bison', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | -| openrouter/google/palm-2-codechat-bison | `completion('openrouter/google/palm-2-codechat-bison', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | -| openrouter/meta-llama/llama-2-13b-chat | `completion('openrouter/meta-llama/llama-2-13b-chat', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | -| openrouter/meta-llama/llama-2-70b-chat | `completion('openrouter/meta-llama/llama-2-70b-chat', messages)` | `os.environ['OR_SITE_URL']`,`os.environ['OR_APP_NAME']`,`os.environ['OPENROUTER_API_KEY']` | From f90072b226ea86b8b8009348b50b18e66f78e1e7 Mon Sep 17 00:00:00 2001 From: Carson Gee Date: Wed, 27 Aug 2025 17:17:05 -0600 Subject: [PATCH 3/3] Apply suggestions from code review Co-authored-by: Nate Goudreault <50588929+nate-goudreault@users.noreply.github.com> --- docs/my-website/docs/providers/datarobot.md | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/docs/my-website/docs/providers/datarobot.md b/docs/my-website/docs/providers/datarobot.md index 1149d11bd2d..3f4a0f71ac4 100644 --- a/docs/my-website/docs/providers/datarobot.md +++ b/docs/my-website/docs/providers/datarobot.md @@ -2,9 +2,11 @@ import Tabs from '@theme/Tabs'; import TabItem from '@theme/TabItem'; # DataRobot -LiteLLM supports all models from [DataRobot](https://datarobot.com). Selecting `datarobot` as the provider routes 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). +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 +## Usage + +### Environment variables ```python import os from litellm import completion @@ -15,10 +17,9 @@ response = completion( model="datarobot/openai/gpt-4o-mini", messages=messages, ) -``` -## Usage - completion +### Completion ```python import litellm import os @@ -35,8 +36,8 @@ response = litellm.completion( print(response) ``` -## DataRobot Completion Models +## DataRobot completion models -🚨 LiteLLM supports _all_ DataRobot LLM gateway models, to get a list for your installation and user account, you can send this simple CURL command: +🚨 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/`