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

Update databricks.md

Browse source 
updates some references to predibase to refer to Databricks
This commit is contained in:
Daniel Liden 2024-06-27 09:11:09 -04:00 • committed by Ishaan Jaff
parent 77b78f6630
commit a975486df0
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
docs/my-website/docs/providers/databricks.md
View file

@ -27,7 +27,7 @@ import os
os.environ["DATABRICKS_API_KEY"] = "databricks key" os.environ["DATABRICKS_API_KEY"] = "databricks key"
os.environ["DATABRICKS_API_BASE"] = "databricks base url" # e.g.: https://adb-3064715882934586.6.azuredatabricks.net/serving-endpoints os.environ["DATABRICKS_API_BASE"] = "databricks base url" # e.g.: https://adb-3064715882934586.6.azuredatabricks.net/serving-endpoints
# predibase llama-3 call # Databricks dbrx-instruct call
response = completion( response = completion(
model="databricks/databricks-dbrx-instruct", model="databricks/databricks-dbrx-instruct",
messages = [{ "content": "Hello, how are you?","role": "user"}] messages = [{ "content": "Hello, how are you?","role": "user"}]
@ -143,8 +143,8 @@ response = completion(
model_list: model_list:
- model_name: llama-3 - model_name: llama-3
litellm_params: litellm_params:
model: predibase/llama-3-8b-instruct model: databricks/databricks-dbrx-instruct
api_key: os.environ/PREDIBASE_API_KEY api_key: os.environ/DATABRICKS_API_KEY
max_tokens: 20 max_tokens: 20
temperature: 0.5 temperature: 0.5
``` ```
@ -162,7 +162,7 @@ import os
os.environ["DATABRICKS_API_KEY"] = "databricks key" os.environ["DATABRICKS_API_KEY"] = "databricks key"
os.environ["DATABRICKS_API_BASE"] = "databricks url" os.environ["DATABRICKS_API_BASE"] = "databricks url"
# predibase llama3 call # Databricks bge-large-en call
response = litellm.embedding( response = litellm.embedding(
model="databricks/databricks-bge-large-en", model="databricks/databricks-bge-large-en",
input=["good morning from litellm"], input=["good morning from litellm"],