Skip to content

Commit 48663a2

Browse files
committed
feature(dspy): added Metadata for ChromaDB retrieval
1 parent fc664d5 commit 48663a2

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

dspy/retrieve/chromadb_rm.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,10 @@ def forward(
145145
query_embeddings=embeddings, n_results=k,
146146
)
147147

148-
passages = [dotdict({"long_text": x}) for x in results["documents"][0]]
149-
150-
return passages
148+
zipped_results = zip(
149+
results["ids"][0],
150+
results["distances"][0],
151+
results["documents"][0],
152+
results["metadatas"][0])
153+
results = [dotdict({"id": id, "score": dist, "long_text": doc, "metadatas": meta }) for id, dist, doc, meta in zipped_results]
154+
return results

0 commit comments

Comments
 (0)