Skip to main content

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-en

Setup

export DATABRICKS_API_KEY="your-databricks-token"
export DATABRICKS_API_BASE="https://your-workspace.cloud.databricks.com"

Available Models

ModelDimensions
databricks/databricks-bge-large-en1024
databricks/databricks-gte-large-en1024