Skip to content

Commit 06f4551

Browse files
Allow returning empty results when nothing is retrieved from DatabricksRM (#1995)
* raise error when databricks index is not ready * don't fail on non-existing data_array
1 parent 45a4d15 commit 06f4551

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

dspy/retrieve/databricks_rm.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -273,11 +273,12 @@ def forward(
273273

274274
# Extracting the results
275275
items = []
276-
for _, data_row in enumerate(results["result"]["data_array"]):
277-
item = {}
278-
for col_name, val in zip(col_names, data_row):
279-
item[col_name] = val
280-
items += [item]
276+
if "data_array" in results["result"]:
277+
for _, data_row in enumerate(results["result"]["data_array"]):
278+
item = {}
279+
for col_name, val in zip(col_names, data_row):
280+
item[col_name] = val
281+
items += [item]
281282

282283
# Sorting results by score in descending order
283284
sorted_docs = sorted(items, key=lambda x: x["score"], reverse=True)[: self.k]

0 commit comments

Comments
 (0)