Update 02_data_pipeline.py

Changed default embeddings
databricks · Jan 17, 2025 · 109ff1f · 109ff1f
1 parent 4bf1ac7
commit 109ff1f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/agent_app_sample_code/02_data_pipeline.py b/agent_app_sample_code/02_data_pipeline.py
@@ -97,9 +97,7 @@
 
 # Embedding model endpoint. The list of off-the-shelf embeddings can be found here:
 # https://docs.databricks.com/en/machine-learning/foundation-models/index.html
-EMBEDDING_MODEL_ENDPOINT = "databricks-bge-large-en"
-# EMBEDDING_MODEL_ENDPOINT = "ep-embeddings-small"
-# EMBEDDING_MODEL_ENDPOINT = "bge-test"
+EMBEDDING_MODEL_ENDPOINT = "databricks-gte-large-en"
 
 # COMMAND ----------