Skip to content

Commit 31b6cf1

Browse files
author
lmorosdb
committed
Setting the default value of the DatabricksRM.forward "query_type" param to "text"
It is recommended to use the fully managed approach for Vector Search which covers getting the embeddings from the of the indexed column during the ingestion and the embeddings of the text query during inference
1 parent b9a8dad commit 31b6cf1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dspy/retrieve/databricks_rm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def __init__(self, databricks_index_name = None, databricks_endpoint = None, dat
8787
self.docs_id_column_name = docs_id_column_name
8888
self.text_column_name = text_column_name
8989

90-
def forward(self, query: Union[str, List[float]], query_type: str = 'vector') -> dspy.Prediction:
90+
def forward(self, query: Union[str, List[float]], query_type: str = 'text') -> dspy.Prediction:
9191
"""Search with Databricks Vector Search Client for self.k top results for query
9292
9393
Args:

0 commit comments

Comments
 (0)