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