Skip to content

Commit 34f6d0d

Browse files
author
lmorosdb
committed
fixing issue in DatabricksRM preventing indexed text from being returned
- There were two missing code lines that prevent the "text" column from being returned when the "froward" method is executed
1 parent b9a8dad commit 34f6d0d

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)