From 73209e31259ef5dfd6d7807010ff5a096fbec479 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 1 Aug 2024 16:52:52 -0700 Subject: [PATCH] docs dbrx --- docs/my-website/docs/providers/databricks.md | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/docs/my-website/docs/providers/databricks.md b/docs/my-website/docs/providers/databricks.md index c24a6220e2..395a544db4 100644 --- a/docs/my-website/docs/providers/databricks.md +++ b/docs/my-website/docs/providers/databricks.md @@ -5,6 +5,11 @@ import TabItem from '@theme/TabItem'; LiteLLM supports all models on Databricks +:::tip + +**We support ALL Databricks models, just set `model=databricks/` as a prefix when sending litellm requests** + +::: ## Usage @@ -185,6 +190,13 @@ response = litellm.embedding( ## Supported Databricks Chat Completion Models +:::tip + +**We support ALL Databricks models, just set `model=databricks/` as a prefix when sending litellm requests** + +::: + + | Model Name | Command | |----------------------------|------------------------------------------------------------------| | databricks-meta-llama-3-1-70b-instruct | `completion(model='databricks/databricks-meta-llama-3-1-70b-instruct', messages=messages)` | @@ -198,6 +210,13 @@ response = litellm.embedding( ## Supported Databricks Embedding Models +:::tip + +**We support ALL Databricks models, just set `model=databricks/` as a prefix when sending litellm requests** + +::: + + | Model Name | Command | |----------------------------|------------------------------------------------------------------| | databricks-bge-large-en | `embedding(model='databricks/databricks-bge-large-en', messages=messages)` |