From 34f6d0df25979168780f8d3e5859163011bcd43b Mon Sep 17 00:00:00 2001
From: lmorosdb <lmorosdb>
Date: Mon, 15 Apr 2024 21:43:00 +0000
Subject: [PATCH] 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
---
 dspy/retrieve/databricks_rm.py | 2 ++
 1 file changed, 2 insertions(+)

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