Skip to content

Commit 0ae493a

Browse files
authored
Merge pull request #1656 from weaviate/fix/missing-bm25-stub
Add missing default stub to `query.bm25`
2 parents b1395c5 + fed23c3 commit 0ae493a

3 files changed

Lines changed: 54 additions & 0 deletions

File tree

weaviate/collections/queries/bm25/query/async_.pyi

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,3 +229,20 @@ class _BM25QueryAsync(
229229
return_properties: Type[TProperties],
230230
return_references: Type[TReferences]
231231
) -> GroupByReturn[TProperties, TReferences]: ...
232+
@overload
233+
async def bm25(
234+
self,
235+
query: Optional[str],
236+
*,
237+
query_properties: Optional[List[str]] = None,
238+
limit: Optional[int] = None,
239+
offset: Optional[int] = None,
240+
auto_limit: Optional[int] = None,
241+
filters: Optional[_Filters] = None,
242+
group_by: Optional[GroupBy] = None,
243+
rerank: Optional[Rerank] = None,
244+
include_vector: INCLUDE_VECTOR = False,
245+
return_metadata: Optional[METADATA] = None,
246+
return_properties: Optional[ReturnProperties[TProperties]] = None,
247+
return_references: Optional[ReturnReferences[TReferences]] = None
248+
) -> QuerySearchReturnType[Properties, References, TProperties, TReferences]: ...

weaviate/collections/queries/bm25/query/executor.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -243,6 +243,26 @@ def bm25(
243243
return_references: Type[TReferences],
244244
) -> executor.Result[GroupByReturn[TProperties, TReferences]]: ...
245245

246+
@overload
247+
def bm25(
248+
self,
249+
query: Optional[str],
250+
*,
251+
query_properties: Optional[List[str]] = None,
252+
limit: Optional[int] = None,
253+
offset: Optional[int] = None,
254+
auto_limit: Optional[int] = None,
255+
filters: Optional[_Filters] = None,
256+
group_by: Optional[GroupBy] = None,
257+
rerank: Optional[Rerank] = None,
258+
include_vector: INCLUDE_VECTOR = False,
259+
return_metadata: Optional[METADATA] = None,
260+
return_properties: Optional[ReturnProperties[TProperties]] = None,
261+
return_references: Optional[ReturnReferences[TReferences]] = None,
262+
) -> executor.Result[
263+
QuerySearchReturnType[Properties, References, TProperties, TReferences]
264+
]: ...
265+
246266
def bm25(
247267
self,
248268
query: Optional[str],

weaviate/collections/queries/bm25/query/sync.pyi

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,3 +229,20 @@ class _BM25Query(
229229
return_properties: Type[TProperties],
230230
return_references: Type[TReferences]
231231
) -> GroupByReturn[TProperties, TReferences]: ...
232+
@overload
233+
def bm25(
234+
self,
235+
query: Optional[str],
236+
*,
237+
query_properties: Optional[List[str]] = None,
238+
limit: Optional[int] = None,
239+
offset: Optional[int] = None,
240+
auto_limit: Optional[int] = None,
241+
filters: Optional[_Filters] = None,
242+
group_by: Optional[GroupBy] = None,
243+
rerank: Optional[Rerank] = None,
244+
include_vector: INCLUDE_VECTOR = False,
245+
return_metadata: Optional[METADATA] = None,
246+
return_properties: Optional[ReturnProperties[TProperties]] = None,
247+
return_references: Optional[ReturnReferences[TReferences]] = None
248+
) -> QuerySearchReturnType[Properties, References, TProperties, TReferences]: ...

0 commit comments

Comments
 (0)