Skip to content

Commit c4583f3

Browse files
committed
Clarify var name
1 parent ec5267b commit c4583f3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

dspy/retrieve/pgvector_rm.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def forward(self, query: str, include_similarity: bool = False):
101101
# Embed query
102102
query_embedding = self._get_embeddings(query)
103103

104-
related_paragraphs = []
104+
retrieved_docs = []
105105

106106
fields = sql.SQL(',').join([
107107
sql.Identifier(f)
@@ -133,9 +133,9 @@ def forward(self, query: str, include_similarity: bool = False):
133133
columns = [descrip[0] for descrip in cur.description]
134134
for row in rows:
135135
data = dict(zip(columns, row))
136-
related_paragraphs.append(dspy.Example(**data))
136+
retrieved_docs.append(dspy.Example(**data))
137137
# Return Prediction
138-
return related_paragraphs
138+
return retrieved_docs
139139

140140
def _get_embeddings(self, query: str) -> list[float]:
141141
if self.openai_client is not None:

0 commit comments

Comments
 (0)