diff --git a/dspy/retrieve/deeplake_rm.py b/dspy/retrieve/deeplake_rm.py index df38fbefb..ae110b6c8 100644 --- a/dspy/retrieve/deeplake_rm.py +++ b/dspy/retrieve/deeplake_rm.py @@ -78,7 +78,7 @@ def embedding_function(self, texts, model="text-embedding-ada-002"): return [data.embedding for data in openai.embeddings.create(input = texts, model=model).data] def forward( - self, query_or_queries: Union[str, List[str]], k: Optional[int], + self, query_or_queries: Union[str, List[str]], k: Optional[int],**kwargs ) -> dspy.Prediction: """Search with DeepLake for self.k top passages for query @@ -102,7 +102,7 @@ def forward( passages = defaultdict(float) #deeplake doesn't support batch querying, manually querying each query and storing them for query in queries: - results = self._deeplake_client.search(query, k=k) + results = self._deeplake_client.search(query, k=k, **kwargs) for score,text in zip(results.get('score',0.0),results.get('text',"")): passages[text] += score