From 8c410cfaf1be35545d21ffce9c80ae53c212c3cb Mon Sep 17 00:00:00 2001 From: Tanmay Garg Date: Tue, 9 Apr 2024 19:55:55 +0530 Subject: [PATCH] fix(qdrant_rm): forward can be called without specifying k --- dspy/retrieve/qdrant_rm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dspy/retrieve/qdrant_rm.py b/dspy/retrieve/qdrant_rm.py index 1ee63f2753..0ffd42dd80 100644 --- a/dspy/retrieve/qdrant_rm.py +++ b/dspy/retrieve/qdrant_rm.py @@ -53,7 +53,7 @@ def __init__( super().__init__(k=k) - def forward(self, query_or_queries: Union[str, List[str]], k: Optional[int],**kwargs) -> dspy.Prediction: + def forward(self, query_or_queries: Union[str, List[str]], k: Optional[int] = None,**kwargs) -> dspy.Prediction: """Search with Qdrant for self.k top passages for query Args: