Overview
Databricks provides embedding models through their Foundation Model APIs.Quick Start
CLI Usage
Setup
Available Models
| Model | Dimensions |
|---|---|
databricks/databricks-bge-large-en | 1024 |
databricks/databricks-gte-large-en | 1024 |
Generate embeddings using Databricks Foundation Models
from praisonaiagents import embedding
result = embedding(
input="Hello world",
model="databricks/databricks-bge-large-en"
)
print(f"Dimensions: {len(result.embeddings[0])}")
praisonai embed "Hello world" --model databricks/databricks-bge-large-en
export DATABRICKS_API_KEY="your-databricks-token"
export DATABRICKS_API_BASE="https://your-workspace.cloud.databricks.com"
| Model | Dimensions |
|---|---|
databricks/databricks-bge-large-en | 1024 |
databricks/databricks-gte-large-en | 1024 |