Skip to content

Commit eab1324

Browse files
Merge pull request #829 from lmoros-DB/defaulting_DatabricksRM_query_type_to_text
Setting the default value of the DatabricksRM.forward "query_type"
2 parents 5fd8ed1 + 31b6cf1 commit eab1324

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)