diff --git a/weaviate/collections/queries/bm25/query/async_.pyi b/weaviate/collections/queries/bm25/query/async_.pyi index 484f826ef..be5da211b 100644 --- a/weaviate/collections/queries/bm25/query/async_.pyi +++ b/weaviate/collections/queries/bm25/query/async_.pyi @@ -229,3 +229,20 @@ class _BM25QueryAsync( return_properties: Type[TProperties], return_references: Type[TReferences] ) -> GroupByReturn[TProperties, TReferences]: ... + @overload + async def bm25( + self, + query: Optional[str], + *, + query_properties: Optional[List[str]] = None, + limit: Optional[int] = None, + offset: Optional[int] = None, + auto_limit: Optional[int] = None, + filters: Optional[_Filters] = None, + group_by: Optional[GroupBy] = None, + rerank: Optional[Rerank] = None, + include_vector: INCLUDE_VECTOR = False, + return_metadata: Optional[METADATA] = None, + return_properties: Optional[ReturnProperties[TProperties]] = None, + return_references: Optional[ReturnReferences[TReferences]] = None + ) -> QuerySearchReturnType[Properties, References, TProperties, TReferences]: ... diff --git a/weaviate/collections/queries/bm25/query/executor.py b/weaviate/collections/queries/bm25/query/executor.py index 45bfb6867..b8da2f0ba 100644 --- a/weaviate/collections/queries/bm25/query/executor.py +++ b/weaviate/collections/queries/bm25/query/executor.py @@ -243,6 +243,26 @@ def bm25( return_references: Type[TReferences], ) -> executor.Result[GroupByReturn[TProperties, TReferences]]: ... + @overload + def bm25( + self, + query: Optional[str], + *, + query_properties: Optional[List[str]] = None, + limit: Optional[int] = None, + offset: Optional[int] = None, + auto_limit: Optional[int] = None, + filters: Optional[_Filters] = None, + group_by: Optional[GroupBy] = None, + rerank: Optional[Rerank] = None, + include_vector: INCLUDE_VECTOR = False, + return_metadata: Optional[METADATA] = None, + return_properties: Optional[ReturnProperties[TProperties]] = None, + return_references: Optional[ReturnReferences[TReferences]] = None, + ) -> executor.Result[ + QuerySearchReturnType[Properties, References, TProperties, TReferences] + ]: ... + def bm25( self, query: Optional[str], diff --git a/weaviate/collections/queries/bm25/query/sync.pyi b/weaviate/collections/queries/bm25/query/sync.pyi index 388981cae..09571b972 100644 --- a/weaviate/collections/queries/bm25/query/sync.pyi +++ b/weaviate/collections/queries/bm25/query/sync.pyi @@ -229,3 +229,20 @@ class _BM25Query( return_properties: Type[TProperties], return_references: Type[TReferences] ) -> GroupByReturn[TProperties, TReferences]: ... + @overload + def bm25( + self, + query: Optional[str], + *, + query_properties: Optional[List[str]] = None, + limit: Optional[int] = None, + offset: Optional[int] = None, + auto_limit: Optional[int] = None, + filters: Optional[_Filters] = None, + group_by: Optional[GroupBy] = None, + rerank: Optional[Rerank] = None, + include_vector: INCLUDE_VECTOR = False, + return_metadata: Optional[METADATA] = None, + return_properties: Optional[ReturnProperties[TProperties]] = None, + return_references: Optional[ReturnReferences[TReferences]] = None + ) -> QuerySearchReturnType[Properties, References, TProperties, TReferences]: ...