Databricks 嵌入¶

如果您在 colab 上打开此 Notebook，则可能需要安装 LlamaIndex 🦙。

In [ ]

%pip install llama-index
%pip install llama-index-embeddings-databricks

%pip install llama-index %pip install llama-index-embeddings-databricks

In [ ]

import os
from llama_index.core import Settings
from llama_index.embeddings.databricks import DatabricksEmbedding

import os from llama_index.core import Settings from llama_index.embeddings.databricks import DatabricksEmbedding

In [ ]

# Set up the DatabricksEmbedding class with the required model, API key and serving endpoint
os.environ["DATABRICKS_TOKEN"] = "<MY TOKEN>"
os.environ["DATABRICKS_SERVING_ENDPOINT"] = "<MY ENDPOINT>"
embed_model = DatabricksEmbedding(model="databricks-bge-large-en")
Settings.embed_model = embed_model

# 使用所需的模型、API 密钥和服务端点设置 DatabricksEmbedding 类 os.environ["DATABRICKS_TOKEN"] = "" os.environ["DATABRICKS_SERVING_ENDPOINT"] = "" embed_model = DatabricksEmbedding(model="databricks-bge-large-en") Settings.embed_model = embed_model

In [ ]

# Embed some text
embeddings = embed_model.get_text_embedding(
    "The DatabricksEmbedding integration works great."
)

# 嵌入一些文本 embeddings = embed_model.get_text_embedding( "The DatabricksEmbedding integration works great." )