Skip to content

Commit 5fd8ed1

Browse files
Merge pull request #828 from lmoros-DB/DatabricksRM_bug_fix
fixing issue in DatabricksRM preventing indexed text from being returned
2 parents b9a8dad + 34f6d0d commit 5fd8ed1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

dspy/retrieve/databricks_rm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,8 @@ def forward(self, query: Union[str, List[float]], query_type: str = 'vector') ->
136136
else:
137137
doc_ids.append(str(val))
138138
text = val
139+
if col["name"] == self.text_column_name:
140+
text = val
139141
if col["name"] == 'score':
140142
score = val
141143
docs[text] += score

0 commit comments

Comments
 (0)