Skip to content

Commit

Permalink
Merge pull request stanfordnlp#829 from lmoros-DB/defaulting_Databric…
Browse files Browse the repository at this point in the history
…ksRM_query_type_to_text

Setting the default value of the DatabricksRM.forward  "query_type"
  • Loading branch information
arnavsinghvi11 authored Apr 15, 2024
2 parents 5fd8ed1 + 31b6cf1 commit eab1324
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dspy/retrieve/databricks_rm.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def __init__(self, databricks_index_name = None, databricks_endpoint = None, dat
self.docs_id_column_name = docs_id_column_name
self.text_column_name = text_column_name

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

0 comments on commit eab1324

Please sign in to comment.