diff --git a/dspy/retrieve/databricks_rm.py b/dspy/retrieve/databricks_rm.py index ecddddb855..fdff1395b8 100644 --- a/dspy/retrieve/databricks_rm.py +++ b/dspy/retrieve/databricks_rm.py @@ -136,6 +136,8 @@ def forward(self, query: Union[str, List[float]], query_type: str = 'vector') -> else: doc_ids.append(str(val)) text = val + if col["name"] == self.text_column_name: + text = val if col["name"] == 'score': score = val docs[text] += score