From 8759c5b14fe5a2ad0ae6043cfcc33f2a5b8505fc Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Thu, 21 May 2026 14:09:42 +1000 Subject: [PATCH 01/11] Add Boost API --- integration/test_collection_boost.py | 264 ++++++++++++++++++ weaviate/classes/query.py | 2 + weaviate/collections/classes/grpc.py | 252 +++++++++++++++++ weaviate/collections/grpc/query.py | 81 ++++++ .../queries/bm25/generate/async_.pyi | 14 + .../queries/bm25/generate/executor.py | 16 ++ .../queries/bm25/generate/sync.pyi | 14 + .../collections/queries/bm25/query/async_.pyi | 14 + .../queries/bm25/query/executor.py | 16 ++ .../collections/queries/bm25/query/sync.pyi | 14 + .../queries/hybrid/generate/async_.pyi | 14 + .../queries/hybrid/generate/executor.py | 16 ++ .../queries/hybrid/generate/sync.pyi | 14 + .../queries/hybrid/query/async_.pyi | 14 + .../queries/hybrid/query/executor.py | 16 ++ .../collections/queries/hybrid/query/sync.pyi | 14 + .../queries/near_image/generate/async_.pyi | 14 + .../queries/near_image/generate/executor.py | 16 ++ .../queries/near_image/generate/sync.pyi | 14 + .../queries/near_image/query/async_.pyi | 14 + .../queries/near_image/query/executor.py | 16 ++ .../queries/near_image/query/sync.pyi | 14 + .../queries/near_media/generate/async_.pyi | 14 + .../queries/near_media/generate/executor.py | 16 ++ .../queries/near_media/generate/sync.pyi | 14 + .../queries/near_media/query/async_.pyi | 14 + .../queries/near_media/query/executor.py | 16 ++ .../queries/near_media/query/sync.pyi | 14 + .../queries/near_object/generate/async_.pyi | 14 + .../queries/near_object/generate/executor.py | 16 ++ .../queries/near_object/generate/sync.pyi | 14 + .../queries/near_object/query/async_.pyi | 14 + .../queries/near_object/query/executor.py | 16 ++ .../queries/near_object/query/sync.pyi | 14 + .../queries/near_text/generate/async_.pyi | 14 + .../queries/near_text/generate/executor.py | 16 ++ .../queries/near_text/generate/sync.pyi | 14 + .../queries/near_text/query/async_.pyi | 14 + .../queries/near_text/query/executor.py | 16 ++ .../queries/near_text/query/sync.pyi | 14 + .../queries/near_vector/generate/async_.pyi | 14 + .../queries/near_vector/generate/executor.py | 16 ++ .../queries/near_vector/generate/sync.pyi | 14 + .../queries/near_vector/query/async_.pyi | 14 + .../queries/near_vector/query/executor.py | 16 ++ .../queries/near_vector/query/sync.pyi | 14 + weaviate/proto/v1/v4216/v1/base_search_pb2.py | 4 +- weaviate/proto/v1/v4216/v1/search_get_pb2.py | 94 ++++--- weaviate/proto/v1/v4216/v1/search_get_pb2.pyi | 85 +++++- weaviate/proto/v1/v5261/v1/search_get_pb2.py | 94 ++++--- weaviate/proto/v1/v5261/v1/search_get_pb2.pyi | 85 +++++- weaviate/proto/v1/v6300/v1/search_get_pb2.py | 94 ++++--- weaviate/proto/v1/v6300/v1/search_get_pb2.pyi | 85 +++++- 53 files changed, 1628 insertions(+), 128 deletions(-) create mode 100644 integration/test_collection_boost.py diff --git a/integration/test_collection_boost.py b/integration/test_collection_boost.py new file mode 100644 index 000000000..ff3b3ee3a --- /dev/null +++ b/integration/test_collection_boost.py @@ -0,0 +1,264 @@ +import pytest + +from integration.conftest import CollectionFactory +from weaviate.classes.query import Boost, Filter, MetadataQuery +from weaviate.collections.classes.config import Configure, DataType, Property +from weaviate.collections.classes.data import DataObject + + +def _create_collection(collection_factory: CollectionFactory): + """Create a collection with numeric and date properties for boost testing.""" + collection = collection_factory( + properties=[ + Property(name="text", data_type=DataType.TEXT), + Property(name="price", data_type=DataType.NUMBER), + Property(name="rating", data_type=DataType.NUMBER), + Property(name="count", data_type=DataType.INT), + Property(name="created", data_type=DataType.DATE), + ], + vectorizer_config=Configure.Vectorizer.none(), + vector_index_config=Configure.VectorIndex.flat(), + ) + if collection._connection._weaviate_version.is_lower_than(1, 38, 0): + pytest.skip("Boost requires Weaviate >= 1.38.0") + collection.data.insert_many( + [ + DataObject( + properties={ + "text": "cheap good", + "price": 10.0, + "rating": 4.9, + "count": 1000, + "created": "2024-01-01T00:00:00Z", + }, + vector=[1.0, 0.0, 0.0], + ), + DataObject( + properties={ + "text": "cheap bad", + "price": 10.0, + "rating": 2.0, + "count": 5, + "created": "2020-01-01T00:00:00Z", + }, + vector=[0.9, 0.1, 0.0], + ), + DataObject( + properties={ + "text": "expensive good", + "price": 500.0, + "rating": 4.8, + "count": 500, + "created": "2023-06-01T00:00:00Z", + }, + vector=[0.0, 1.0, 0.0], + ), + DataObject( + properties={ + "text": "expensive bad", + "price": 500.0, + "rating": 1.5, + "count": 2, + "created": "2019-01-01T00:00:00Z", + }, + vector=[0.0, 0.9, 0.1], + ), + DataObject( + properties={ + "text": "mid range", + "price": 50.0, + "rating": 3.5, + "count": 100, + "created": "2022-01-01T00:00:00Z", + }, + vector=[0.0, 0.0, 1.0], + ), + ] + ) + return collection + + +def test_boost_filter(collection_factory: CollectionFactory) -> None: + """Boost results matching a filter — boosted items should score higher.""" + collection = _create_collection(collection_factory) + + baseline = collection.query.near_vector( + near_vector=[1.0, 0.0, 0.0], + limit=5, + return_metadata=MetadataQuery(distance=True), + ).objects + + boosted = collection.query.near_vector( + near_vector=[1.0, 0.0, 0.0], + limit=5, + boost=Boost.filter( + Filter.by_property("rating").greater_or_equal(4.0), + weight=1.0, + ), + return_metadata=MetadataQuery(distance=True), + ).objects + + assert len(boosted) == 5 + # The boost should change the ordering compared to baseline + assert [o.uuid for o in baseline] != [o.uuid for o in boosted] + + +def test_boost_numeric_decay(collection_factory: CollectionFactory) -> None: + """Numeric decay: prefer items with price near the origin.""" + collection = _create_collection(collection_factory) + + result = collection.query.near_vector( + near_vector=[1.0, 0.0, 0.0], + limit=5, + boost=Boost.numeric_decay( + "price", + origin=50.0, + scale=20.0, + curve=Boost.Curve.LINEAR, + weight=1.0, + ), + return_metadata=MetadataQuery(distance=True), + ).objects + + assert len(result) == 5 + + +def test_boost_time_decay(collection_factory: CollectionFactory) -> None: + """Time decay: prefer items with dates closer to origin.""" + collection = _create_collection(collection_factory) + + result = collection.query.near_vector( + near_vector=[1.0, 0.0, 0.0], + limit=5, + boost=Boost.time_decay( + "created", + origin="2024-01-01T00:00:00Z", + scale="365d", + curve=Boost.Curve.EXPONENTIAL, + weight=1.0, + ), + return_metadata=MetadataQuery(distance=True), + ).objects + + assert len(result) == 5 + + +def test_boost_property_value(collection_factory: CollectionFactory) -> None: + """Property value boost: rank by a numeric property directly.""" + collection = _create_collection(collection_factory) + + result = collection.query.near_vector( + near_vector=[1.0, 0.0, 0.0], + limit=5, + boost=Boost.property( + "count", + modifier=Boost.Modifier.LOG1P, + weight=1.0, + ), + return_metadata=MetadataQuery(distance=True), + ).objects + + assert len(result) == 5 + + +def test_boost_blend(collection_factory: CollectionFactory) -> None: + """Blend multiple boost conditions together.""" + collection = _create_collection(collection_factory) + + result = collection.query.near_vector( + near_vector=[1.0, 0.0, 0.0], + limit=5, + boost=Boost.blend( + Boost.filter( + Filter.by_property("rating").greater_or_equal(4.0), + weight=2.0, + ), + Boost.numeric_decay( + "price", + origin=30.0, + scale=100.0, + curve=Boost.Curve.EXPONENTIAL, + ), + weight=0.8, + ), + return_metadata=MetadataQuery(distance=True), + ).objects + + assert len(result) == 5 + + +def test_boost_with_depth(collection_factory: CollectionFactory) -> None: + """Boost with explicit depth parameter.""" + collection = _create_collection(collection_factory) + + result = collection.query.near_vector( + near_vector=[1.0, 0.0, 0.0], + limit=5, + boost=Boost.filter( + Filter.by_property("rating").greater_or_equal(4.0), + weight=1.0, + depth=100, + ), + return_metadata=MetadataQuery(distance=True), + ).objects + + assert len(result) == 5 + + +def test_boost_bm25(collection_factory: CollectionFactory) -> None: + """Boost works with BM25 keyword search.""" + collection = _create_collection(collection_factory) + + result = collection.query.bm25( + query="cheap", + limit=5, + boost=Boost.filter( + Filter.by_property("rating").greater_or_equal(4.0), + weight=1.0, + ), + return_metadata=MetadataQuery(score=True), + ).objects + + assert len(result) >= 1 + + +def test_boost_hybrid(collection_factory: CollectionFactory) -> None: + """Boost works with hybrid search.""" + collection = _create_collection(collection_factory) + + result = collection.query.hybrid( + query="cheap", + vector=[1.0, 0.0, 0.0], + limit=5, + boost=Boost.filter( + Filter.by_property("price").less_than(100.0), + weight=0.6, + ), + return_metadata=MetadataQuery(score=True), + ).objects + + assert len(result) >= 1 + + +def test_boost_api_surface() -> None: + """Test the public API surface: factory guard + static methods.""" + with pytest.raises(TypeError): + Boost() + + # Static methods produce _Boost instances + b = Boost.filter( + Filter.by_property("x").equal("y"), + weight=0.5, + ) + assert len(b.conditions) == 1 + assert b.weight == 0.5 + + b = Boost.blend( + Boost.filter(Filter.by_property("x").equal("y"), weight=1.0), + Boost.property("z", modifier=Boost.Modifier.LOG1P), + weight=0.8, + depth=200, + ) + assert len(b.conditions) == 2 + assert b.weight == 0.8 + assert b.depth == 200 diff --git a/weaviate/classes/query.py b/weaviate/classes/query.py index 4b03030e9..2217fd90c 100644 --- a/weaviate/classes/query.py +++ b/weaviate/classes/query.py @@ -5,6 +5,7 @@ BM25OperatorFactory as BM25Operator, ) from weaviate.collections.classes.grpc import ( + Boost, Diversity, GroupBy, HybridFusion, @@ -38,6 +39,7 @@ "QueryNested", "QueryReference", "NearVector", + "Boost", "Rerank", "Sort", "TargetVectors", diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index bea10d845..ec72f02d3 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -1,4 +1,5 @@ from dataclasses import dataclass +from datetime import datetime, timedelta from enum import Enum, auto from typing import ( Any, @@ -268,6 +269,257 @@ class Rerank(_WeaviateInput): query: Optional[str] = Field(default=None) +@dataclass +class _TimeDecayFunction: + property: str # noqa: A003 + origin: str + scale: str + offset: Optional[str] = None + curve: Optional[str] = None + decay_value: Optional[float] = None + + +@dataclass +class _NumericDecayFunction: + property: str # noqa: A003 + origin: float + scale: float + offset: Optional[float] = None + curve: Optional[str] = None + decay_value: Optional[float] = None + + +@dataclass +class _PropertyValueFunction: + property: str # noqa: A003 + modifier: Optional[str] = None + + +@dataclass +class _BoostCondition: + filter: Optional[Any] = None # noqa: A003 # FilterReturn + time_decay: Optional[_TimeDecayFunction] = None + numeric_decay: Optional[_NumericDecayFunction] = None + property_value: Optional[_PropertyValueFunction] = None + weight: Optional[float] = None + + +@dataclass +class _Boost: + conditions: List[_BoostCondition] + weight: Optional[float] = None + depth: Optional[int] = None + + +def _decay_value_to_str(val: Union[str, int, float, timedelta, datetime]) -> str: + """Convert a decay parameter value to the string format expected by the server.""" + if isinstance(val, timedelta): + total_seconds = val.total_seconds() + if total_seconds >= 86400 and total_seconds % 86400 == 0: + return f"{int(total_seconds // 86400)}d" + if total_seconds >= 3600 and total_seconds % 3600 == 0: + return f"{int(total_seconds // 3600)}h" + if total_seconds >= 60 and total_seconds % 60 == 0: + return f"{int(total_seconds // 60)}m" + if total_seconds == int(total_seconds): + return f"{int(total_seconds)}s" + return f"{total_seconds}s" + if isinstance(val, datetime): + return val.isoformat() + return str(val) + + +class _BoostCurve(str, BaseEnum): + """Decay curve type for distance-based rank scoring.""" + + EXPONENTIAL = "exp" + GAUSSIAN = "gauss" + LINEAR = "linear" + + +class _BoostModifier(str, BaseEnum): + """Score modifier for property-value rank scoring.""" + + NONE = "none" + LOG1P = "log1p" + SQRT = "sqrt" + + +class Boost: + """Define soft-ranking conditions to boost or demote matching documents without excluding them. + + Use the static methods `boost()`, `decay()`, and `blend()` to create rank configurations. + """ + + Curve = _BoostCurve + Modifier = _BoostModifier + + def __init__(self) -> None: + raise TypeError("Boost cannot be instantiated. Use the static methods to create a rank.") + + @staticmethod + def filter( # noqa: A003 + filter: Any, # noqa: A002 + *, + weight: Optional[float] = None, + depth: Optional[int] = None, + ) -> _Boost: + """Boost or demote results matching a filter condition. + + Args: + filter: The filter condition (same as used in `filters=` parameter). + weight: Blending weight [0,1] controlling how much the rank affects final scores. + depth: Number of results to rescore (default 100, max 10000). Higher values improve accuracy at the cost of performance. + """ + return _Boost(conditions=[_BoostCondition(filter=filter)], weight=weight, depth=depth) + + @staticmethod + def time_decay( + property: str, # noqa: A002 + *, + origin: Optional[Union[str, datetime]] = None, + scale: Union[str, timedelta], + offset: Optional[Union[str, timedelta]] = None, + curve: Optional[Union[_BoostCurve, str]] = None, + decay_value: Optional[float] = None, + weight: Optional[float] = None, + depth: Optional[int] = None, + ) -> _Boost: + """Apply time-based decay scoring from an origin date. + + Args: + property: The date property name to compute distance from. + origin: The origin point. Use "now" for current time or a datetime for a specific time. + Defaults to "now". + scale: Distance from origin where score equals decay_value. Use timedelta + (e.g. timedelta(days=7)) or a string shorthand like "7d", "24h". + offset: Documents within this distance from origin get full score (default "0"). + Accepts the same types as scale. + curve: Decay curve type: `Boost.Curve.EXPONENTIAL` (default), `Boost.Curve.GAUSSIAN`, or `Boost.Curve.LINEAR`. + decay_value: Score at scale distance from origin (default 0.5). + weight: Blending weight [0,1] controlling how much the rank affects final scores. + depth: Number of results to rescore (default 100, max 10000). + """ + return _Boost( + conditions=[ + _BoostCondition( + time_decay=_TimeDecayFunction( + property=property, + origin=_decay_value_to_str(origin) if origin is not None else "", + scale=_decay_value_to_str(scale), + offset=_decay_value_to_str(offset) if offset is not None else None, + curve=curve.value if isinstance(curve, _BoostCurve) else curve, + decay_value=decay_value, + ) + ) + ], + weight=weight, + depth=depth, + ) + + @staticmethod + def numeric_decay( + property: str, # noqa: A002 + *, + origin: float, + scale: float, + offset: Optional[float] = None, + curve: Optional[Union[_BoostCurve, str]] = None, + decay_value: Optional[float] = None, + weight: Optional[float] = None, + depth: Optional[int] = None, + ) -> _Boost: + """Apply numeric distance-based decay scoring from an origin value. + + Args: + property: The numeric property name to compute distance from. + origin: The origin point (numeric value). + scale: Distance from origin where score equals decay_value. + offset: Documents within this distance from origin get full score (default 0). + curve: Decay curve type: `Boost.Curve.EXPONENTIAL` (default), `Boost.Curve.GAUSSIAN`, or `Boost.Curve.LINEAR`. + decay_value: Score at scale distance from origin (default 0.5). + weight: Blending weight [0,1] controlling how much the rank affects final scores. + depth: Number of results to rescore (default 100, max 10000). + """ + return _Boost( + conditions=[ + _BoostCondition( + numeric_decay=_NumericDecayFunction( + property=property, + origin=float(origin), + scale=float(scale), + offset=float(offset) if offset is not None else None, + curve=curve.value if isinstance(curve, _BoostCurve) else curve, + decay_value=decay_value, + ) + ) + ], + weight=weight, + depth=depth, + ) + + @staticmethod + def property( # noqa: A003 + name: str, + *, + modifier: Optional[Union[_BoostModifier, str]] = None, + weight: Optional[float] = None, + depth: Optional[int] = None, + ) -> _Boost: + """Rank by a numeric property's value directly. + + Args: + name: The property name to use as a ranking signal. + modifier: Score modifier: `Boost.Modifier.NONE` (default), `Boost.Modifier.LOG1P`, or `Boost.Modifier.SQRT`. + weight: Blending weight [0,1] controlling how much the rank affects final scores. + depth: Number of results to rescore (default 100, max 10000). + """ + return _Boost( + conditions=[ + _BoostCondition( + property_value=_PropertyValueFunction( + property=name, + modifier=modifier.value + if isinstance(modifier, _BoostModifier) + else modifier, + ) + ) + ], + weight=weight, + depth=depth, + ) + + @staticmethod + def blend( + *ranks: _Boost, + weight: Optional[float] = None, + depth: Optional[int] = None, + ) -> _Boost: + """Combine multiple rank conditions with individual weights. + + When blending, each sub-rank's weight becomes a per-condition weight, + and the `weight` parameter here controls the overall blending strength. + + Args: + *ranks: Rank objects created via `Boost.filter()`, `Boost.time_decay()`, `Boost.numeric_decay()`, or `Boost.property()`. + weight: Overall blending weight [0,1] for combining primary search and rank scores. + depth: Number of results to rescore (default 100, max 10000). Higher values improve accuracy at the cost of performance. + """ + conditions: List[_BoostCondition] = [] + for r in ranks: + for cond in r.conditions: + if cond.weight is None and r.weight is not None: + cond = _BoostCondition( + filter=cond.filter, + time_decay=cond.time_decay, + numeric_decay=cond.numeric_decay, + property_value=cond.property_value, + weight=r.weight, + ) + conditions.append(cond) + return _Boost(conditions=conditions, weight=weight, depth=depth) + + @dataclass class MMR: """Define MMR (Maximal Marginal Relevance) diversity selection. diff --git a/weaviate/collections/grpc/query.py b/weaviate/collections/grpc/query.py index 69f6ef8ff..c112d5574 100644 --- a/weaviate/collections/grpc/query.py +++ b/weaviate/collections/grpc/query.py @@ -32,6 +32,7 @@ QueryNested, Rerank, TargetVectorJoinType, + _Boost, _MetadataQuery, _QueryReference, _QueryReferenceMultiTarget, @@ -122,6 +123,7 @@ def get( return_references: Optional[REFERENCES] = None, generative: Optional[_Generative] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, ) -> search_get_pb2.SearchRequest: if self._validate_arguments: _validate_input(_ValidateArgument([_Sorting, None], "sort", sort)) @@ -144,6 +146,7 @@ def get( return_references=return_references, generative=generative, rerank=rerank, + boost=boost, sort_by=sort_by, ) @@ -167,6 +170,7 @@ def hybrid( return_references: Optional[REFERENCES] = None, generative: Optional[_Generative] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, ) -> search_get_pb2.SearchRequest: return self.__create_request( @@ -179,6 +183,7 @@ def hybrid( return_references=return_references, generative=generative, rerank=rerank, + boost=boost, autocut=autocut, hybrid_search=self._parse_hybrid( query, @@ -208,6 +213,7 @@ def bm25( return_references: Optional[REFERENCES] = None, generative: Optional[_Generative] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, ) -> search_get_pb2.SearchRequest: if self._validate_arguments: _validate_input( @@ -227,6 +233,7 @@ def bm25( return_references=return_references, generative=generative, rerank=rerank, + boost=boost, autocut=autocut, bm25=( base_search_pb2.BM25( @@ -259,6 +266,7 @@ def near_vector( group_by: Optional[_GroupBy] = None, generative: Optional[_Generative] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, return_metadata: Optional[_MetadataQuery] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -274,6 +282,7 @@ def near_vector( return_references=return_references, generative=generative, rerank=rerank, + boost=boost, autocut=autocut, group_by=group_by, near_vector=self._parse_near_vector( @@ -298,6 +307,7 @@ def near_object( group_by: Optional[_GroupBy] = None, generative: Optional[_Generative] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, return_metadata: Optional[_MetadataQuery] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -313,6 +323,7 @@ def near_object( return_references=return_references, generative=generative, rerank=rerank, + boost=boost, autocut=autocut, group_by=group_by, near_object=self._parse_near_object( @@ -339,6 +350,7 @@ def near_text( group_by: Optional[_GroupBy] = None, generative: Optional[_Generative] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, return_metadata: Optional[_MetadataQuery] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -354,6 +366,7 @@ def near_text( return_references=return_references, generative=generative, rerank=rerank, + boost=boost, autocut=autocut, group_by=group_by, near_text=self._parse_near_text( @@ -381,6 +394,7 @@ def near_media( group_by: Optional[_GroupBy] = None, generative: Optional[_Generative] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, return_metadata: Optional[_MetadataQuery] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -396,6 +410,7 @@ def near_media( return_references=return_references, generative=generative, rerank=rerank, + boost=boost, autocut=autocut, group_by=group_by, **self._parse_media( @@ -419,6 +434,7 @@ def __create_request( return_references: Optional[REFERENCES] = None, generative: Optional[_Generative] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, autocut: Optional[int] = None, group_by: Optional[_GroupBy] = None, near_vector: Optional[base_search_pb2.NearVector] = None, @@ -512,6 +528,7 @@ def __create_request( if rerank is not None else None ), + boost=self.__boost_to_grpc(boost), near_vector=near_vector, sort_by=sort_by, hybrid_search=hybrid_search, @@ -541,6 +558,70 @@ def _metadata_to_grpc(self, metadata: _MetadataQuery) -> search_get_pb2.Metadata query_profile=metadata.query_profile, ) + _Boost_pb2 = search_get_pb2.Boost + + _CURVE_TO_PROTO = { + "exp": _Boost_pb2.DECAY_CURVE_EXPONENTIAL, + "gauss": _Boost_pb2.DECAY_CURVE_GAUSS, + "linear": _Boost_pb2.DECAY_CURVE_LINEAR, + } + + _MODIFIER_TO_PROTO = { + "none": _Boost_pb2.PROPERTY_VALUE_MODIFIER_UNSPECIFIED, + "log1p": _Boost_pb2.PROPERTY_VALUE_MODIFIER_LOG1P, + "sqrt": _Boost_pb2.PROPERTY_VALUE_MODIFIER_SQRT, + } + + def __resolve_curve(self, curve: Optional[str]) -> "search_get_pb2.Boost.DecayCurve": + if curve is None: + return self._Boost_pb2.DECAY_CURVE_EXPONENTIAL + return self._CURVE_TO_PROTO.get(curve, self._Boost_pb2.DECAY_CURVE_EXPONENTIAL) + + def __boost_to_grpc(self, boost: Optional[_Boost]) -> Optional[search_get_pb2.Boost]: + if boost is None: + return None + _B = self._Boost_pb2 + conditions = [] + for cond in boost.conditions: + grpc_cond = _B.Condition(weight=cond.weight) + if cond.filter is not None: + grpc_cond.filter.CopyFrom(_FilterToGRPC.convert(cond.filter)) + elif cond.time_decay is not None: + grpc_cond.time_decay.CopyFrom( + _B.TimeDecayFunction( + property=cond.time_decay.property, + origin=cond.time_decay.origin, + scale=cond.time_decay.scale, + offset=cond.time_decay.offset, + curve=self.__resolve_curve(cond.time_decay.curve), + decay_value=cond.time_decay.decay_value, + ) + ) + elif cond.numeric_decay is not None: + grpc_cond.numeric_decay.CopyFrom( + _B.NumericDecayFunction( + property=cond.numeric_decay.property, + origin=cond.numeric_decay.origin, + scale=cond.numeric_decay.scale, + offset=cond.numeric_decay.offset, + curve=self.__resolve_curve(cond.numeric_decay.curve), + decay_value=cond.numeric_decay.decay_value, + ) + ) + elif cond.property_value is not None: + grpc_cond.property_value.CopyFrom( + _B.PropertyValueFunction( + property=cond.property_value.property, + modifier=self._MODIFIER_TO_PROTO.get( + cond.property_value.modifier, _B.PROPERTY_VALUE_MODIFIER_UNSPECIFIED + ) + if cond.property_value.modifier is not None + else _B.PROPERTY_VALUE_MODIFIER_UNSPECIFIED, + ) + ) + conditions.append(grpc_cond) + return search_get_pb2.Boost(conditions=conditions, weight=boost.weight, depth=boost.depth) + def __resolve_property(self, prop: QueryNested) -> search_get_pb2.ObjectPropertiesRequest: props = prop.properties if isinstance(prop.properties, list) else [prop.properties] return search_get_pb2.ObjectPropertiesRequest( diff --git a/weaviate/collections/queries/bm25/generate/async_.pyi b/weaviate/collections/queries/bm25/generate/async_.pyi index 36dae24a8..8f3a34fcd 100644 --- a/weaviate/collections/queries/bm25/generate/async_.pyi +++ b/weaviate/collections/queries/bm25/generate/async_.pyi @@ -8,6 +8,7 @@ from weaviate.collections.classes.grpc import ( BM25OperatorOptions, GroupBy, Rerank, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -46,6 +47,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -68,6 +70,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -90,6 +93,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -112,6 +116,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -134,6 +139,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -156,6 +162,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -178,6 +185,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -200,6 +208,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -222,6 +231,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -244,6 +254,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -266,6 +277,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -288,6 +300,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -310,6 +323,7 @@ class _BM25GenerateAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Optional[ReturnProperties[TProperties]] = None, diff --git a/weaviate/collections/queries/bm25/generate/executor.py b/weaviate/collections/queries/bm25/generate/executor.py index 437c3197a..2f226b5a5 100644 --- a/weaviate/collections/queries/bm25/generate/executor.py +++ b/weaviate/collections/queries/bm25/generate/executor.py @@ -8,6 +8,7 @@ BM25OperatorOptions, GroupBy, Rerank, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -56,6 +57,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -79,6 +81,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -102,6 +105,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -125,6 +129,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -148,6 +153,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -171,6 +177,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -196,6 +203,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -219,6 +227,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -242,6 +251,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -265,6 +275,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -288,6 +299,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -311,6 +323,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -334,6 +347,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Optional[ReturnProperties[TProperties]] = None, @@ -358,6 +372,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Optional[ReturnProperties[TProperties]] = None, @@ -429,6 +444,7 @@ def resp( filters=filters, group_by=_GroupBy.from_input(group_by), rerank=rerank, + boost=boost, return_metadata=self._parse_return_metadata(return_metadata, include_vector), return_properties=self._parse_return_properties(return_properties), return_references=self._parse_return_references(return_references), diff --git a/weaviate/collections/queries/bm25/generate/sync.pyi b/weaviate/collections/queries/bm25/generate/sync.pyi index ca5c0d303..9717e1980 100644 --- a/weaviate/collections/queries/bm25/generate/sync.pyi +++ b/weaviate/collections/queries/bm25/generate/sync.pyi @@ -8,6 +8,7 @@ from weaviate.collections.classes.grpc import ( BM25OperatorOptions, GroupBy, Rerank, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -46,6 +47,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -68,6 +70,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -90,6 +93,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -112,6 +116,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -134,6 +139,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -156,6 +162,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -178,6 +185,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -200,6 +208,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -222,6 +231,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -244,6 +254,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -266,6 +277,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -288,6 +300,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -310,6 +323,7 @@ class _BM25Generate( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Optional[ReturnProperties[TProperties]] = None, diff --git a/weaviate/collections/queries/bm25/query/async_.pyi b/weaviate/collections/queries/bm25/query/async_.pyi index 3b844e58b..e0fe02a47 100644 --- a/weaviate/collections/queries/bm25/query/async_.pyi +++ b/weaviate/collections/queries/bm25/query/async_.pyi @@ -8,6 +8,7 @@ from weaviate.collections.classes.grpc import ( BM25OperatorOptions, GroupBy, Rerank, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -39,6 +40,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -57,6 +59,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -75,6 +78,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -93,6 +97,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -111,6 +116,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -129,6 +135,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -147,6 +154,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -165,6 +173,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -183,6 +192,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -201,6 +211,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -219,6 +230,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -237,6 +249,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -255,6 +268,7 @@ class _BM25QueryAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Optional[ReturnProperties[TProperties]] = None, diff --git a/weaviate/collections/queries/bm25/query/executor.py b/weaviate/collections/queries/bm25/query/executor.py index 774339641..c81ca1fbe 100644 --- a/weaviate/collections/queries/bm25/query/executor.py +++ b/weaviate/collections/queries/bm25/query/executor.py @@ -8,6 +8,7 @@ BM25OperatorOptions, GroupBy, Rerank, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -48,6 +49,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -67,6 +69,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -86,6 +89,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -105,6 +109,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -124,6 +129,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -143,6 +149,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -164,6 +171,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -183,6 +191,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -202,6 +211,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -221,6 +231,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -240,6 +251,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -259,6 +271,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -278,6 +291,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Optional[ReturnProperties[TProperties]] = None, @@ -298,6 +312,7 @@ def bm25( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Optional[ReturnProperties[TProperties]] = None, @@ -363,6 +378,7 @@ def resp( filters=filters, group_by=_GroupBy.from_input(group_by), rerank=rerank, + boost=boost, return_metadata=self._parse_return_metadata(return_metadata, include_vector), return_properties=self._parse_return_properties(return_properties), return_references=self._parse_return_references(cast(Any, return_references)), diff --git a/weaviate/collections/queries/bm25/query/sync.pyi b/weaviate/collections/queries/bm25/query/sync.pyi index a4e6f7ae6..721c4a39c 100644 --- a/weaviate/collections/queries/bm25/query/sync.pyi +++ b/weaviate/collections/queries/bm25/query/sync.pyi @@ -8,6 +8,7 @@ from weaviate.collections.classes.grpc import ( BM25OperatorOptions, GroupBy, Rerank, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -39,6 +40,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -57,6 +59,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -75,6 +78,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -93,6 +97,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -111,6 +116,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -129,6 +135,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -147,6 +154,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -165,6 +173,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -183,6 +192,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Union[PROPERTIES, bool, None] = None, @@ -201,6 +211,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -219,6 +230,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -237,6 +249,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Type[TProperties], @@ -255,6 +268,7 @@ class _BM25Query( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, return_properties: Optional[ReturnProperties[TProperties]] = None, diff --git a/weaviate/collections/queries/hybrid/generate/async_.pyi b/weaviate/collections/queries/hybrid/generate/async_.pyi index 8c9faec33..eafac01c9 100644 --- a/weaviate/collections/queries/hybrid/generate/async_.pyi +++ b/weaviate/collections/queries/hybrid/generate/async_.pyi @@ -11,6 +11,7 @@ from weaviate.collections.classes.grpc import ( HybridVectorType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -54,6 +55,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -81,6 +83,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -108,6 +111,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -135,6 +139,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -162,6 +167,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -189,6 +195,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -216,6 +223,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -243,6 +251,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -270,6 +279,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -297,6 +307,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -324,6 +335,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -351,6 +363,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -378,6 +391,7 @@ class _HybridGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/hybrid/generate/executor.py b/weaviate/collections/queries/hybrid/generate/executor.py index 37bfb1b69..04250a1cd 100644 --- a/weaviate/collections/queries/hybrid/generate/executor.py +++ b/weaviate/collections/queries/hybrid/generate/executor.py @@ -13,6 +13,7 @@ HybridVectorType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -65,6 +66,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -93,6 +95,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -121,6 +124,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -149,6 +153,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -177,6 +182,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -205,6 +211,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -235,6 +242,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -263,6 +271,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -291,6 +300,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -319,6 +329,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -347,6 +358,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -375,6 +387,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -404,6 +417,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -433,6 +447,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -513,6 +528,7 @@ def resp( filters=filters, group_by=_GroupBy.from_input(group_by), rerank=rerank, + boost=boost, target_vector=target_vector, return_metadata=self._parse_return_metadata(return_metadata, include_vector), return_properties=self._parse_return_properties(return_properties), diff --git a/weaviate/collections/queries/hybrid/generate/sync.pyi b/weaviate/collections/queries/hybrid/generate/sync.pyi index f48966fb5..e4192ac05 100644 --- a/weaviate/collections/queries/hybrid/generate/sync.pyi +++ b/weaviate/collections/queries/hybrid/generate/sync.pyi @@ -11,6 +11,7 @@ from weaviate.collections.classes.grpc import ( HybridVectorType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -53,6 +54,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -80,6 +82,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -107,6 +110,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -134,6 +138,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -161,6 +166,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -188,6 +194,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -215,6 +222,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -242,6 +250,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -269,6 +278,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -296,6 +306,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -323,6 +334,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -350,6 +362,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -377,6 +390,7 @@ class _HybridGenerate( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/hybrid/query/async_.pyi b/weaviate/collections/queries/hybrid/query/async_.pyi index 9336f6ffe..9e700051b 100644 --- a/weaviate/collections/queries/hybrid/query/async_.pyi +++ b/weaviate/collections/queries/hybrid/query/async_.pyi @@ -11,6 +11,7 @@ from weaviate.collections.classes.grpc import ( HybridVectorType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -46,6 +47,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -69,6 +71,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -92,6 +95,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -115,6 +119,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -138,6 +143,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -161,6 +167,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -184,6 +191,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -207,6 +215,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -230,6 +239,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -253,6 +263,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -276,6 +287,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -299,6 +311,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -322,6 +335,7 @@ class _HybridQueryAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/hybrid/query/executor.py b/weaviate/collections/queries/hybrid/query/executor.py index 213d8b6e0..797ea8729 100644 --- a/weaviate/collections/queries/hybrid/query/executor.py +++ b/weaviate/collections/queries/hybrid/query/executor.py @@ -13,6 +13,7 @@ HybridVectorType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -57,6 +58,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -81,6 +83,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -105,6 +108,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -129,6 +133,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -153,6 +158,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -177,6 +183,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -203,6 +210,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -227,6 +235,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -251,6 +260,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -275,6 +285,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -299,6 +310,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -323,6 +335,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -348,6 +361,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -373,6 +387,7 @@ def hybrid( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -449,6 +464,7 @@ def resp( filters=filters, group_by=_GroupBy.from_input(group_by), rerank=rerank, + boost=boost, target_vector=target_vector, return_metadata=self._parse_return_metadata(return_metadata, include_vector), return_properties=self._parse_return_properties(return_properties), diff --git a/weaviate/collections/queries/hybrid/query/sync.pyi b/weaviate/collections/queries/hybrid/query/sync.pyi index 01b4a84f8..20d884276 100644 --- a/weaviate/collections/queries/hybrid/query/sync.pyi +++ b/weaviate/collections/queries/hybrid/query/sync.pyi @@ -11,6 +11,7 @@ from weaviate.collections.classes.grpc import ( HybridVectorType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -46,6 +47,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -69,6 +71,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -92,6 +95,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -115,6 +119,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -138,6 +143,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -161,6 +167,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -184,6 +191,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -207,6 +215,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -230,6 +239,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -253,6 +263,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -276,6 +287,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -299,6 +311,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -322,6 +335,7 @@ class _HybridQuery( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_image/generate/async_.pyi b/weaviate/collections/queries/near_image/generate/async_.pyi index f0845b218..ae3102924 100644 --- a/weaviate/collections/queries/near_image/generate/async_.pyi +++ b/weaviate/collections/queries/near_image/generate/async_.pyi @@ -9,6 +9,7 @@ from weaviate.collections.classes.grpc import ( GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -48,6 +49,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -72,6 +74,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -96,6 +99,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -120,6 +124,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -144,6 +149,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -168,6 +174,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -192,6 +199,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -216,6 +224,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -240,6 +249,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -264,6 +274,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -288,6 +299,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -312,6 +324,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -336,6 +349,7 @@ class _NearImageGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_image/generate/executor.py b/weaviate/collections/queries/near_image/generate/executor.py index 3f40122da..85e0c7bfb 100644 --- a/weaviate/collections/queries/near_image/generate/executor.py +++ b/weaviate/collections/queries/near_image/generate/executor.py @@ -12,6 +12,7 @@ NearMediaType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -61,6 +62,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -86,6 +88,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -111,6 +114,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -136,6 +140,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -161,6 +166,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -186,6 +192,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -212,6 +219,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -237,6 +245,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -262,6 +271,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -287,6 +297,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -312,6 +323,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -337,6 +349,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -363,6 +376,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -389,6 +403,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -463,6 +478,7 @@ def resp( group_by=_GroupBy.from_input(group_by), rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, generative=_Generative( single=single_prompt, diff --git a/weaviate/collections/queries/near_image/generate/sync.pyi b/weaviate/collections/queries/near_image/generate/sync.pyi index cabbfb821..8a9a1e16c 100644 --- a/weaviate/collections/queries/near_image/generate/sync.pyi +++ b/weaviate/collections/queries/near_image/generate/sync.pyi @@ -9,6 +9,7 @@ from weaviate.collections.classes.grpc import ( GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -48,6 +49,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -72,6 +74,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -96,6 +99,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -120,6 +124,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -144,6 +149,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -168,6 +174,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -192,6 +199,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -216,6 +224,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -240,6 +249,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -264,6 +274,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -288,6 +299,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -312,6 +324,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -336,6 +349,7 @@ class _NearImageGenerate( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_image/query/async_.pyi b/weaviate/collections/queries/near_image/query/async_.pyi index acd625795..d7d649d51 100644 --- a/weaviate/collections/queries/near_image/query/async_.pyi +++ b/weaviate/collections/queries/near_image/query/async_.pyi @@ -9,6 +9,7 @@ from weaviate.collections.classes.grpc import ( GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -41,6 +42,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -61,6 +63,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -81,6 +84,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -101,6 +105,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -121,6 +126,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -141,6 +147,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -161,6 +168,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -181,6 +189,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -201,6 +210,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -221,6 +231,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -241,6 +252,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -261,6 +273,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -281,6 +294,7 @@ class _NearImageQueryAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_image/query/executor.py b/weaviate/collections/queries/near_image/query/executor.py index fde831767..423f50ebd 100644 --- a/weaviate/collections/queries/near_image/query/executor.py +++ b/weaviate/collections/queries/near_image/query/executor.py @@ -12,6 +12,7 @@ NearMediaType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -53,6 +54,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -74,6 +76,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -95,6 +98,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -116,6 +120,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -137,6 +142,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -158,6 +164,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -181,6 +188,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -202,6 +210,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -223,6 +232,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -244,6 +254,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -265,6 +276,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -286,6 +298,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -308,6 +321,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -330,6 +344,7 @@ def near_image( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -402,6 +417,7 @@ def resp( group_by=_GroupBy.from_input(group_by), rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, limit=limit, offset=offset, diff --git a/weaviate/collections/queries/near_image/query/sync.pyi b/weaviate/collections/queries/near_image/query/sync.pyi index 01c5dadbf..268c57464 100644 --- a/weaviate/collections/queries/near_image/query/sync.pyi +++ b/weaviate/collections/queries/near_image/query/sync.pyi @@ -9,6 +9,7 @@ from weaviate.collections.classes.grpc import ( GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -40,6 +41,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -60,6 +62,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -80,6 +83,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -100,6 +104,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -120,6 +125,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -140,6 +146,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -160,6 +167,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -180,6 +188,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -200,6 +209,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -220,6 +230,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -240,6 +251,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -260,6 +272,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -280,6 +293,7 @@ class _NearImageQuery( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_media/generate/async_.pyi b/weaviate/collections/queries/near_media/generate/async_.pyi index b52e82108..d2c5694bf 100644 --- a/weaviate/collections/queries/near_media/generate/async_.pyi +++ b/weaviate/collections/queries/near_media/generate/async_.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( NearMediaType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -50,6 +51,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -75,6 +77,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -100,6 +103,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -125,6 +129,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -150,6 +155,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -175,6 +181,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -200,6 +207,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -225,6 +233,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -250,6 +259,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -275,6 +285,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -300,6 +311,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -325,6 +337,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -350,6 +363,7 @@ class _NearMediaGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_media/generate/executor.py b/weaviate/collections/queries/near_media/generate/executor.py index 5cb20160c..835d207c3 100644 --- a/weaviate/collections/queries/near_media/generate/executor.py +++ b/weaviate/collections/queries/near_media/generate/executor.py @@ -12,6 +12,7 @@ NearMediaType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -62,6 +63,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -88,6 +90,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -114,6 +117,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -140,6 +144,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -166,6 +171,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -192,6 +198,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -219,6 +226,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -245,6 +253,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -271,6 +280,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -297,6 +307,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -323,6 +334,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -349,6 +361,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -376,6 +389,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -403,6 +417,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -478,6 +493,7 @@ def resp( group_by=_GroupBy.from_input(group_by), rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, generative=_Generative( single=single_prompt, diff --git a/weaviate/collections/queries/near_media/generate/sync.pyi b/weaviate/collections/queries/near_media/generate/sync.pyi index 505b3edce..a57132c22 100644 --- a/weaviate/collections/queries/near_media/generate/sync.pyi +++ b/weaviate/collections/queries/near_media/generate/sync.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( NearMediaType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -50,6 +51,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -75,6 +77,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -100,6 +103,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -125,6 +129,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -150,6 +155,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -175,6 +181,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -200,6 +207,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -225,6 +233,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -250,6 +259,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -275,6 +285,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -300,6 +311,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -325,6 +337,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -350,6 +363,7 @@ class _NearMediaGenerate( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_media/query/async_.pyi b/weaviate/collections/queries/near_media/query/async_.pyi index 49b7923d6..fdf21de7f 100644 --- a/weaviate/collections/queries/near_media/query/async_.pyi +++ b/weaviate/collections/queries/near_media/query/async_.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( NearMediaType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -43,6 +44,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -64,6 +66,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -85,6 +88,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -106,6 +110,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -127,6 +132,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -148,6 +154,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -169,6 +176,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -190,6 +198,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -211,6 +220,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -232,6 +242,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -253,6 +264,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -274,6 +286,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -295,6 +308,7 @@ class _NearMediaQueryAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_media/query/executor.py b/weaviate/collections/queries/near_media/query/executor.py index 71e0dd09a..29f7f3276 100644 --- a/weaviate/collections/queries/near_media/query/executor.py +++ b/weaviate/collections/queries/near_media/query/executor.py @@ -12,6 +12,7 @@ NearMediaType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -54,6 +55,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -76,6 +78,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -98,6 +101,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -120,6 +124,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -142,6 +147,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -164,6 +170,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -188,6 +195,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -210,6 +218,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -232,6 +241,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -254,6 +264,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -276,6 +287,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -298,6 +310,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -321,6 +334,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -344,6 +358,7 @@ def near_media( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -417,6 +432,7 @@ def resp( group_by=_GroupBy.from_input(group_by), rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, limit=limit, offset=offset, diff --git a/weaviate/collections/queries/near_media/query/sync.pyi b/weaviate/collections/queries/near_media/query/sync.pyi index 71ba12e96..5a829638c 100644 --- a/weaviate/collections/queries/near_media/query/sync.pyi +++ b/weaviate/collections/queries/near_media/query/sync.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( NearMediaType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -42,6 +43,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -63,6 +65,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -84,6 +87,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -105,6 +109,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -126,6 +131,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -147,6 +153,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -168,6 +175,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -189,6 +197,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -210,6 +219,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -231,6 +241,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -252,6 +263,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -273,6 +285,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -294,6 +307,7 @@ class _NearMediaQuery( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_object/generate/async_.pyi b/weaviate/collections/queries/near_object/generate/async_.pyi index a66cc24d2..406a59233 100644 --- a/weaviate/collections/queries/near_object/generate/async_.pyi +++ b/weaviate/collections/queries/near_object/generate/async_.pyi @@ -9,6 +9,7 @@ from weaviate.collections.classes.grpc import ( GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -48,6 +49,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -72,6 +74,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -96,6 +99,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -120,6 +124,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -144,6 +149,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -168,6 +174,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -192,6 +199,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -216,6 +224,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -240,6 +249,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -264,6 +274,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -288,6 +299,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -312,6 +324,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -336,6 +349,7 @@ class _NearObjectGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_object/generate/executor.py b/weaviate/collections/queries/near_object/generate/executor.py index 152a9c0b4..c2c218783 100644 --- a/weaviate/collections/queries/near_object/generate/executor.py +++ b/weaviate/collections/queries/near_object/generate/executor.py @@ -11,6 +11,7 @@ GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -59,6 +60,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -84,6 +86,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -109,6 +112,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -134,6 +138,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -159,6 +164,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -184,6 +190,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -210,6 +217,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -235,6 +243,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -260,6 +269,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -285,6 +295,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -310,6 +321,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -335,6 +347,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -362,6 +375,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -388,6 +402,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -461,6 +476,7 @@ def resp( group_by=_GroupBy.from_input(group_by), rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, generative=_Generative( single=single_prompt, diff --git a/weaviate/collections/queries/near_object/generate/sync.pyi b/weaviate/collections/queries/near_object/generate/sync.pyi index 6aaa7ea16..b25a6ce50 100644 --- a/weaviate/collections/queries/near_object/generate/sync.pyi +++ b/weaviate/collections/queries/near_object/generate/sync.pyi @@ -9,6 +9,7 @@ from weaviate.collections.classes.grpc import ( GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -48,6 +49,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -72,6 +74,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -96,6 +99,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -120,6 +124,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -144,6 +149,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -168,6 +174,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -192,6 +199,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -216,6 +224,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -240,6 +249,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -264,6 +274,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -288,6 +299,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -312,6 +324,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -336,6 +349,7 @@ class _NearObjectGenerate( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_object/query/async_.pyi b/weaviate/collections/queries/near_object/query/async_.pyi index a121a992d..da11bac04 100644 --- a/weaviate/collections/queries/near_object/query/async_.pyi +++ b/weaviate/collections/queries/near_object/query/async_.pyi @@ -9,6 +9,7 @@ from weaviate.collections.classes.grpc import ( GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -41,6 +42,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -61,6 +63,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -81,6 +84,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -101,6 +105,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -121,6 +126,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -141,6 +147,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -161,6 +168,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -181,6 +189,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -201,6 +210,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -221,6 +231,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -241,6 +252,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -261,6 +273,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -281,6 +294,7 @@ class _NearObjectQueryAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_object/query/executor.py b/weaviate/collections/queries/near_object/query/executor.py index 02ea47778..6b9a8db40 100644 --- a/weaviate/collections/queries/near_object/query/executor.py +++ b/weaviate/collections/queries/near_object/query/executor.py @@ -11,6 +11,7 @@ GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -52,6 +53,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -73,6 +75,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -94,6 +97,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -115,6 +119,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -136,6 +141,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -157,6 +163,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -180,6 +187,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -201,6 +209,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -222,6 +231,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -243,6 +253,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -264,6 +275,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -285,6 +297,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -308,6 +321,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -330,6 +344,7 @@ def near_object( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -403,6 +418,7 @@ def resp( group_by=_GroupBy.from_input(group_by), rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, return_metadata=self._parse_return_metadata(return_metadata, include_vector), return_properties=self._parse_return_properties(return_properties), diff --git a/weaviate/collections/queries/near_object/query/sync.pyi b/weaviate/collections/queries/near_object/query/sync.pyi index cd17ab461..601be456b 100644 --- a/weaviate/collections/queries/near_object/query/sync.pyi +++ b/weaviate/collections/queries/near_object/query/sync.pyi @@ -9,6 +9,7 @@ from weaviate.collections.classes.grpc import ( GroupBy, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -41,6 +42,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -61,6 +63,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -81,6 +84,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -101,6 +105,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -121,6 +126,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -141,6 +147,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -161,6 +168,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -181,6 +189,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -201,6 +210,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -221,6 +231,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -241,6 +252,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -261,6 +273,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -281,6 +294,7 @@ class _NearObjectQuery( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_text/generate/async_.pyi b/weaviate/collections/queries/near_text/generate/async_.pyi index f609127b7..eb5d6343d 100644 --- a/weaviate/collections/queries/near_text/generate/async_.pyi +++ b/weaviate/collections/queries/near_text/generate/async_.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( Move, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -51,6 +52,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -77,6 +79,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -103,6 +106,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -129,6 +133,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -155,6 +160,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -181,6 +187,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -207,6 +214,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -233,6 +241,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -259,6 +268,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -285,6 +295,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -311,6 +322,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -337,6 +349,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -363,6 +376,7 @@ class _NearTextGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_text/generate/executor.py b/weaviate/collections/queries/near_text/generate/executor.py index eb1850ee6..811edb7e8 100644 --- a/weaviate/collections/queries/near_text/generate/executor.py +++ b/weaviate/collections/queries/near_text/generate/executor.py @@ -12,6 +12,7 @@ Move, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -62,6 +63,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -89,6 +91,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -116,6 +119,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -143,6 +147,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -170,6 +175,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -197,6 +203,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -225,6 +232,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -252,6 +260,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -279,6 +288,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -306,6 +316,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -333,6 +344,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -360,6 +372,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -388,6 +401,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -416,6 +430,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -494,6 +509,7 @@ def resp( group_by=_GroupBy.from_input(group_by), rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, generative=_Generative( single=single_prompt, diff --git a/weaviate/collections/queries/near_text/generate/sync.pyi b/weaviate/collections/queries/near_text/generate/sync.pyi index 2601f58cb..13d39e8e7 100644 --- a/weaviate/collections/queries/near_text/generate/sync.pyi +++ b/weaviate/collections/queries/near_text/generate/sync.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( Move, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -51,6 +52,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -77,6 +79,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -103,6 +106,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -129,6 +133,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -155,6 +160,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -181,6 +187,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -207,6 +214,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -233,6 +241,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -259,6 +268,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -285,6 +295,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -311,6 +322,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -337,6 +349,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -363,6 +376,7 @@ class _NearTextGenerate( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_text/query/async_.pyi b/weaviate/collections/queries/near_text/query/async_.pyi index 5d652b559..ed27b44be 100644 --- a/weaviate/collections/queries/near_text/query/async_.pyi +++ b/weaviate/collections/queries/near_text/query/async_.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( Move, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -43,6 +44,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -65,6 +67,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -87,6 +90,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -109,6 +113,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -131,6 +136,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -153,6 +159,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -175,6 +182,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -197,6 +205,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -219,6 +228,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -241,6 +251,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -263,6 +274,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -285,6 +297,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -307,6 +320,7 @@ class _NearTextQueryAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_text/query/executor.py b/weaviate/collections/queries/near_text/query/executor.py index e04ba4db2..d109a3ab2 100644 --- a/weaviate/collections/queries/near_text/query/executor.py +++ b/weaviate/collections/queries/near_text/query/executor.py @@ -12,6 +12,7 @@ Move, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -54,6 +55,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -77,6 +79,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -100,6 +103,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -123,6 +127,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -146,6 +151,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -169,6 +175,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -194,6 +201,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -217,6 +225,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -240,6 +249,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -263,6 +273,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -286,6 +297,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -309,6 +321,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -333,6 +346,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -357,6 +371,7 @@ def near_text( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -433,6 +448,7 @@ def resp( group_by=_GroupBy.from_input(group_by), rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, return_metadata=self._parse_return_metadata(return_metadata, include_vector), return_properties=self._parse_return_properties(return_properties), diff --git a/weaviate/collections/queries/near_text/query/sync.pyi b/weaviate/collections/queries/near_text/query/sync.pyi index 6847ecb33..f8ded7b28 100644 --- a/weaviate/collections/queries/near_text/query/sync.pyi +++ b/weaviate/collections/queries/near_text/query/sync.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( Move, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -43,6 +44,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -65,6 +67,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -87,6 +90,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -109,6 +113,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -131,6 +136,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -153,6 +159,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -175,6 +182,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -197,6 +205,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -219,6 +228,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -241,6 +251,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -263,6 +274,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -285,6 +297,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -307,6 +320,7 @@ class _NearTextQuery( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_vector/generate/async_.pyi b/weaviate/collections/queries/near_vector/generate/async_.pyi index 528d6a00b..56115d850 100644 --- a/weaviate/collections/queries/near_vector/generate/async_.pyi +++ b/weaviate/collections/queries/near_vector/generate/async_.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( NearVectorInputType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -49,6 +50,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -73,6 +75,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -97,6 +100,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -121,6 +125,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -145,6 +150,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -169,6 +175,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -193,6 +200,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -217,6 +225,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -241,6 +250,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -265,6 +275,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -289,6 +300,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -313,6 +325,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -337,6 +350,7 @@ class _NearVectorGenerateAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_vector/generate/executor.py b/weaviate/collections/queries/near_vector/generate/executor.py index 881076eb0..7341d5391 100644 --- a/weaviate/collections/queries/near_vector/generate/executor.py +++ b/weaviate/collections/queries/near_vector/generate/executor.py @@ -12,6 +12,7 @@ NearVectorInputType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -60,6 +61,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -85,6 +87,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -110,6 +113,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -135,6 +139,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -160,6 +165,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -185,6 +191,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -211,6 +218,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -236,6 +244,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -261,6 +270,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -286,6 +296,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -311,6 +322,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -336,6 +348,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -362,6 +375,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -388,6 +402,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -467,6 +482,7 @@ def resp( autocut=auto_limit, rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, return_metadata=self._parse_return_metadata(return_metadata, include_vector), return_properties=self._parse_return_properties(return_properties), diff --git a/weaviate/collections/queries/near_vector/generate/sync.pyi b/weaviate/collections/queries/near_vector/generate/sync.pyi index 9d82b3dee..b5fc3eb9e 100644 --- a/weaviate/collections/queries/near_vector/generate/sync.pyi +++ b/weaviate/collections/queries/near_vector/generate/sync.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( NearVectorInputType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -49,6 +50,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -73,6 +75,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -97,6 +100,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -121,6 +125,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -145,6 +150,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -169,6 +175,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -193,6 +200,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -217,6 +225,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -241,6 +250,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -265,6 +275,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -289,6 +300,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -313,6 +325,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -337,6 +350,7 @@ class _NearVectorGenerate( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_vector/query/async_.pyi b/weaviate/collections/queries/near_vector/query/async_.pyi index d0e8fe0d8..105abb45c 100644 --- a/weaviate/collections/queries/near_vector/query/async_.pyi +++ b/weaviate/collections/queries/near_vector/query/async_.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( NearVectorInputType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -42,6 +43,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -62,6 +64,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -82,6 +85,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -102,6 +106,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -122,6 +127,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -142,6 +148,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -162,6 +169,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -182,6 +190,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -202,6 +211,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -222,6 +232,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -242,6 +253,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -262,6 +274,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -282,6 +295,7 @@ class _NearVectorQueryAsync( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/collections/queries/near_vector/query/executor.py b/weaviate/collections/queries/near_vector/query/executor.py index 7e4f33f00..ed867fe93 100644 --- a/weaviate/collections/queries/near_vector/query/executor.py +++ b/weaviate/collections/queries/near_vector/query/executor.py @@ -12,6 +12,7 @@ NearVectorInputType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -52,6 +53,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -73,6 +75,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -94,6 +97,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -115,6 +119,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -136,6 +141,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -157,6 +163,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -180,6 +187,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -201,6 +209,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -222,6 +231,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -243,6 +253,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -264,6 +275,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -285,6 +297,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -307,6 +320,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -329,6 +343,7 @@ def near_vector( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -400,6 +415,7 @@ def resp( autocut=auto_limit, rerank=rerank, diversity_selection=diversity_selection, + boost=boost, target_vector=target_vector, return_metadata=self._parse_return_metadata(return_metadata, include_vector), return_properties=self._parse_return_properties(return_properties), diff --git a/weaviate/collections/queries/near_vector/query/sync.pyi b/weaviate/collections/queries/near_vector/query/sync.pyi index 3fd866ab1..31dcc1129 100644 --- a/weaviate/collections/queries/near_vector/query/sync.pyi +++ b/weaviate/collections/queries/near_vector/query/sync.pyi @@ -10,6 +10,7 @@ from weaviate.collections.classes.grpc import ( NearVectorInputType, Rerank, TargetVectorJoinType, + _Boost, ) from weaviate.collections.classes.internal import ( CrossReferences, @@ -42,6 +43,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -62,6 +64,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -82,6 +85,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -102,6 +106,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -122,6 +127,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -142,6 +148,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: Literal[None] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -162,6 +169,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -182,6 +190,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -202,6 +211,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -222,6 +232,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -242,6 +253,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -262,6 +274,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: GroupBy, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, @@ -282,6 +295,7 @@ class _NearVectorQuery( filters: Optional[FilterReturn] = None, group_by: Optional[GroupBy] = None, rerank: Optional[Rerank] = None, + boost: Optional[_Boost] = None, target_vector: Optional[TargetVectorJoinType] = None, include_vector: INCLUDE_VECTOR = False, return_metadata: Optional[METADATA] = None, diff --git a/weaviate/proto/v1/v4216/v1/base_search_pb2.py b/weaviate/proto/v1/v4216/v1/base_search_pb2.py index 306f01b4f..aa4fdf57d 100644 --- a/weaviate/proto/v1/v4216/v1/base_search_pb2.py +++ b/weaviate/proto/v1/v4216/v1/base_search_pb2.py @@ -32,8 +32,8 @@ _HYBRID.fields_by_name['vector_bytes']._serialized_options = b'\030\001' _HYBRID.fields_by_name['target_vectors']._options = None _HYBRID.fields_by_name['target_vectors']._serialized_options = b'\030\001' - _globals['_NEARVECTOR_VECTORPERTARGETENTRY']._options = None - _globals['_NEARVECTOR_VECTORPERTARGETENTRY']._serialized_options = b'8\001' + _NEARVECTOR_VECTORPERTARGETENTRY._options = None + _NEARVECTOR_VECTORPERTARGETENTRY._serialized_options = b'8\001' _NEARVECTOR.fields_by_name['vector']._options = None _NEARVECTOR.fields_by_name['vector']._serialized_options = b'\030\001' _NEARVECTOR.fields_by_name['vector_bytes']._options = None diff --git a/weaviate/proto/v1/v4216/v1/search_get_pb2.py b/weaviate/proto/v1/v4216/v1/search_get_pb2.py index 78f20076e..95eef9170 100644 --- a/weaviate/proto/v1/v4216/v1/search_get_pb2.py +++ b/weaviate/proto/v1/v4216/v1/search_get_pb2.py @@ -17,7 +17,7 @@ from weaviate.proto.v1.v4216.v1 import properties_pb2 as v1_dot_properties__pb2 -DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13v1/search_get.proto\x12\x0bweaviate.v1\x1a\rv1/base.proto\x1a\x14v1/base_search.proto\x1a\x13v1/generative.proto\x1a\x13v1/properties.proto\"\x9c\x0b\n\rSearchRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\x0e\n\x06tenant\x18\n \x01(\t\x12=\n\x11\x63onsistency_level\x18\x0b \x01(\x0e\x32\x1d.weaviate.v1.ConsistencyLevelH\x00\x88\x01\x01\x12\x37\n\nproperties\x18\x14 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequestH\x01\x88\x01\x01\x12\x33\n\x08metadata\x18\x15 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequestH\x02\x88\x01\x01\x12+\n\x08group_by\x18\x16 \x01(\x0b\x32\x14.weaviate.v1.GroupByH\x03\x88\x01\x01\x12\r\n\x05limit\x18\x1e \x01(\r\x12\x0e\n\x06offset\x18\x1f \x01(\r\x12\x0f\n\x07\x61utocut\x18 \x01(\r\x12\r\n\x05\x61\x66ter\x18! \x01(\t\x12$\n\x07sort_by\x18\" \x03(\x0b\x32\x13.weaviate.v1.SortBy\x12*\n\x07\x66ilters\x18( \x01(\x0b\x32\x14.weaviate.v1.FiltersH\x04\x88\x01\x01\x12/\n\rhybrid_search\x18) \x01(\x0b\x32\x13.weaviate.v1.HybridH\x05\x88\x01\x01\x12+\n\x0b\x62m25_search\x18* \x01(\x0b\x32\x11.weaviate.v1.BM25H\x06\x88\x01\x01\x12\x31\n\x0bnear_vector\x18+ \x01(\x0b\x32\x17.weaviate.v1.NearVectorH\x07\x88\x01\x01\x12\x31\n\x0bnear_object\x18, \x01(\x0b\x32\x17.weaviate.v1.NearObjectH\x08\x88\x01\x01\x12\x33\n\tnear_text\x18- \x01(\x0b\x32\x1b.weaviate.v1.NearTextSearchH\t\x88\x01\x01\x12\x35\n\nnear_image\x18. \x01(\x0b\x32\x1c.weaviate.v1.NearImageSearchH\n\x88\x01\x01\x12\x35\n\nnear_audio\x18/ \x01(\x0b\x32\x1c.weaviate.v1.NearAudioSearchH\x0b\x88\x01\x01\x12\x35\n\nnear_video\x18\x30 \x01(\x0b\x32\x1c.weaviate.v1.NearVideoSearchH\x0c\x88\x01\x01\x12\x35\n\nnear_depth\x18\x31 \x01(\x0b\x32\x1c.weaviate.v1.NearDepthSearchH\r\x88\x01\x01\x12\x39\n\x0cnear_thermal\x18\x32 \x01(\x0b\x32\x1e.weaviate.v1.NearThermalSearchH\x0e\x88\x01\x01\x12\x31\n\x08near_imu\x18\x33 \x01(\x0b\x32\x1a.weaviate.v1.NearIMUSearchH\x0f\x88\x01\x01\x12\x36\n\ngenerative\x18< \x01(\x0b\x32\x1d.weaviate.v1.GenerativeSearchH\x10\x88\x01\x01\x12(\n\x06rerank\x18= \x01(\x0b\x32\x13.weaviate.v1.RerankH\x11\x88\x01\x01\x12\x18\n\x0cuses_123_api\x18\x64 \x01(\x08\x42\x02\x18\x01\x12\x18\n\x0cuses_125_api\x18\x65 \x01(\x08\x42\x02\x18\x01\x12\x14\n\x0cuses_127_api\x18\x66 \x01(\x08\x42\x14\n\x12_consistency_levelB\r\n\x0b_propertiesB\x0b\n\t_metadataB\x0b\n\t_group_byB\n\n\x08_filtersB\x10\n\x0e_hybrid_searchB\x0e\n\x0c_bm25_searchB\x0e\n\x0c_near_vectorB\x0e\n\x0c_near_objectB\x0c\n\n_near_textB\r\n\x0b_near_imageB\r\n\x0b_near_audioB\r\n\x0b_near_videoB\r\n\x0b_near_depthB\x0f\n\r_near_thermalB\x0b\n\t_near_imuB\r\n\x0b_generativeB\t\n\x07_rerank\"L\n\x07GroupBy\x12\x0c\n\x04path\x18\x01 \x03(\t\x12\x18\n\x10number_of_groups\x18\x02 \x01(\x05\x12\x19\n\x11objects_per_group\x18\x03 \x01(\x05\")\n\x06SortBy\x12\x11\n\tascending\x18\x01 \x01(\x08\x12\x0c\n\x04path\x18\x02 \x03(\t\"\xf4\x01\n\x0fMetadataRequest\x12\x0c\n\x04uuid\x18\x01 \x01(\x08\x12\x0e\n\x06vector\x18\x02 \x01(\x08\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x04 \x01(\x08\x12\x10\n\x08\x64istance\x18\x05 \x01(\x08\x12\x11\n\tcertainty\x18\x06 \x01(\x08\x12\r\n\x05score\x18\x07 \x01(\x08\x12\x15\n\rexplain_score\x18\x08 \x01(\x08\x12\x15\n\ris_consistent\x18\t \x01(\x08\x12\x0f\n\x07vectors\x18\n \x03(\t\x12\x15\n\rquery_profile\x18\x0b \x01(\x08\"\xd1\x01\n\x11PropertiesRequest\x12\x1a\n\x12non_ref_properties\x18\x01 \x03(\t\x12\x39\n\x0eref_properties\x18\x02 \x03(\x0b\x32!.weaviate.v1.RefPropertiesRequest\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\x12$\n\x1creturn_all_nonref_properties\x18\x0b \x01(\x08\"\x8b\x01\n\x17ObjectPropertiesRequest\x12\x11\n\tprop_name\x18\x01 \x01(\t\x12\x1c\n\x14primitive_properties\x18\x02 \x03(\t\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\"\xb1\x01\n\x14RefPropertiesRequest\x12\x1a\n\x12reference_property\x18\x01 \x01(\t\x12\x32\n\nproperties\x18\x02 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequest\x12.\n\x08metadata\x18\x03 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequest\x12\x19\n\x11target_collection\x18\x04 \x01(\t\"8\n\x06Rerank\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x12\n\x05query\x18\x02 \x01(\tH\x00\x88\x01\x01\x42\x08\n\x06_query\"\xf7\x02\n\x0bSearchReply\x12\x0c\n\x04took\x18\x01 \x01(\x02\x12*\n\x07results\x18\x02 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12*\n\x19generative_grouped_result\x18\x03 \x01(\tB\x02\x18\x01H\x00\x88\x01\x01\x12\x34\n\x10group_by_results\x18\x04 \x03(\x0b\x32\x1a.weaviate.v1.GroupByResult\x12\x46\n\x1agenerative_grouped_results\x18\x05 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x01\x88\x01\x01\x12\x35\n\rquery_profile\x18\x06 \x01(\x0b\x32\x19.weaviate.v1.QueryProfileH\x02\x88\x01\x01\x42\x1c\n\x1a_generative_grouped_resultB\x1d\n\x1b_generative_grouped_resultsB\x10\n\x0e_query_profile\"\x9e\x03\n\x0cQueryProfile\x12\x36\n\x06shards\x18\x01 \x03(\x0b\x32&.weaviate.v1.QueryProfile.ShardProfile\x1a\x86\x01\n\rSearchProfile\x12\x45\n\x07\x64\x65tails\x18\x01 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.SearchProfile.DetailsEntry\x1a.\n\x0c\x44\x65tailsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\xcc\x01\n\x0cShardProfile\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04node\x18\x02 \x01(\t\x12\x46\n\x08searches\x18\x03 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.ShardProfile.SearchesEntry\x1aX\n\rSearchesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x36\n\x05value\x18\x02 \x01(\x0b\x32\'.weaviate.v1.QueryProfile.SearchProfile:\x02\x38\x01\"\x1c\n\x0bRerankReply\x12\r\n\x05score\x18\x01 \x01(\x01\"\xe9\x02\n\rGroupByResult\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\x0cmin_distance\x18\x02 \x01(\x02\x12\x14\n\x0cmax_distance\x18\x03 \x01(\x02\x12\x19\n\x11number_of_objects\x18\x04 \x01(\x03\x12*\n\x07objects\x18\x05 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12-\n\x06rerank\x18\x06 \x01(\x0b\x32\x18.weaviate.v1.RerankReplyH\x00\x88\x01\x01\x12\x39\n\ngenerative\x18\x07 \x01(\x0b\x32\x1c.weaviate.v1.GenerativeReplyB\x02\x18\x01H\x01\x88\x01\x01\x12=\n\x11generative_result\x18\x08 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x02\x88\x01\x01\x42\t\n\x07_rerankB\r\n\x0b_generativeB\x14\n\x12_generative_result\"\xb7\x01\n\x0cSearchResult\x12\x31\n\nproperties\x18\x01 \x01(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12-\n\x08metadata\x18\x02 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12\x36\n\ngenerative\x18\x03 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x00\x88\x01\x01\x42\r\n\x0b_generative\"\xf7\x04\n\x0eMetadataResult\x12\n\n\x02id\x18\x01 \x01(\t\x12\x12\n\x06vector\x18\x02 \x03(\x02\x42\x02\x18\x01\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x03\x12\"\n\x1a\x63reation_time_unix_present\x18\x04 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x05 \x01(\x03\x12%\n\x1dlast_update_time_unix_present\x18\x06 \x01(\x08\x12\x10\n\x08\x64istance\x18\x07 \x01(\x02\x12\x18\n\x10\x64istance_present\x18\x08 \x01(\x08\x12\x11\n\tcertainty\x18\t \x01(\x02\x12\x19\n\x11\x63\x65rtainty_present\x18\n \x01(\x08\x12\r\n\x05score\x18\x0b \x01(\x02\x12\x15\n\rscore_present\x18\x0c \x01(\x08\x12\x15\n\rexplain_score\x18\r \x01(\t\x12\x1d\n\x15\x65xplain_score_present\x18\x0e \x01(\x08\x12\x1a\n\ris_consistent\x18\x0f \x01(\x08H\x00\x88\x01\x01\x12\x16\n\ngenerative\x18\x10 \x01(\tB\x02\x18\x01\x12\x1e\n\x12generative_present\x18\x11 \x01(\x08\x42\x02\x18\x01\x12\x1d\n\x15is_consistent_present\x18\x12 \x01(\x08\x12\x14\n\x0cvector_bytes\x18\x13 \x01(\x0c\x12\x13\n\x0bid_as_bytes\x18\x14 \x01(\x0c\x12\x14\n\x0crerank_score\x18\x15 \x01(\x01\x12\x1c\n\x14rerank_score_present\x18\x16 \x01(\x08\x12%\n\x07vectors\x18\x17 \x03(\x0b\x32\x14.weaviate.v1.VectorsB\x10\n\x0e_is_consistent\"\x88\x02\n\x10PropertiesResult\x12\x33\n\tref_props\x18\x02 \x03(\x0b\x32 .weaviate.v1.RefPropertiesResult\x12\x19\n\x11target_collection\x18\x03 \x01(\t\x12-\n\x08metadata\x18\x04 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12.\n\rnon_ref_props\x18\x0b \x01(\x0b\x32\x17.weaviate.v1.Properties\x12\x1b\n\x13ref_props_requested\x18\x0c \x01(\x08J\x04\x08\x01\x10\x02J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\nJ\x04\x08\n\x10\x0b\"[\n\x13RefPropertiesResult\x12\x31\n\nproperties\x18\x01 \x03(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12\x11\n\tprop_name\x18\x02 \x01(\tBs\n#io.weaviate.client.grpc.protocol.v1B\x16WeaviateProtoSearchGetZ4github.com/weaviate/weaviate/grpc/generated;protocolb\x06proto3') +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13v1/search_get.proto\x12\x0bweaviate.v1\x1a\rv1/base.proto\x1a\x14v1/base_search.proto\x1a\x13v1/generative.proto\x1a\x13v1/properties.proto\"\xce\x0b\n\rSearchRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\x0e\n\x06tenant\x18\n \x01(\t\x12=\n\x11\x63onsistency_level\x18\x0b \x01(\x0e\x32\x1d.weaviate.v1.ConsistencyLevelH\x00\x88\x01\x01\x12\x37\n\nproperties\x18\x14 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequestH\x01\x88\x01\x01\x12\x33\n\x08metadata\x18\x15 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequestH\x02\x88\x01\x01\x12+\n\x08group_by\x18\x16 \x01(\x0b\x32\x14.weaviate.v1.GroupByH\x03\x88\x01\x01\x12\r\n\x05limit\x18\x1e \x01(\r\x12\x0e\n\x06offset\x18\x1f \x01(\r\x12\x0f\n\x07\x61utocut\x18 \x01(\r\x12\r\n\x05\x61\x66ter\x18! \x01(\t\x12$\n\x07sort_by\x18\" \x03(\x0b\x32\x13.weaviate.v1.SortBy\x12*\n\x07\x66ilters\x18( \x01(\x0b\x32\x14.weaviate.v1.FiltersH\x04\x88\x01\x01\x12/\n\rhybrid_search\x18) \x01(\x0b\x32\x13.weaviate.v1.HybridH\x05\x88\x01\x01\x12+\n\x0b\x62m25_search\x18* \x01(\x0b\x32\x11.weaviate.v1.BM25H\x06\x88\x01\x01\x12\x31\n\x0bnear_vector\x18+ \x01(\x0b\x32\x17.weaviate.v1.NearVectorH\x07\x88\x01\x01\x12\x31\n\x0bnear_object\x18, \x01(\x0b\x32\x17.weaviate.v1.NearObjectH\x08\x88\x01\x01\x12\x33\n\tnear_text\x18- \x01(\x0b\x32\x1b.weaviate.v1.NearTextSearchH\t\x88\x01\x01\x12\x35\n\nnear_image\x18. \x01(\x0b\x32\x1c.weaviate.v1.NearImageSearchH\n\x88\x01\x01\x12\x35\n\nnear_audio\x18/ \x01(\x0b\x32\x1c.weaviate.v1.NearAudioSearchH\x0b\x88\x01\x01\x12\x35\n\nnear_video\x18\x30 \x01(\x0b\x32\x1c.weaviate.v1.NearVideoSearchH\x0c\x88\x01\x01\x12\x35\n\nnear_depth\x18\x31 \x01(\x0b\x32\x1c.weaviate.v1.NearDepthSearchH\r\x88\x01\x01\x12\x39\n\x0cnear_thermal\x18\x32 \x01(\x0b\x32\x1e.weaviate.v1.NearThermalSearchH\x0e\x88\x01\x01\x12\x31\n\x08near_imu\x18\x33 \x01(\x0b\x32\x1a.weaviate.v1.NearIMUSearchH\x0f\x88\x01\x01\x12\x36\n\ngenerative\x18< \x01(\x0b\x32\x1d.weaviate.v1.GenerativeSearchH\x10\x88\x01\x01\x12(\n\x06rerank\x18= \x01(\x0b\x32\x13.weaviate.v1.RerankH\x11\x88\x01\x01\x12&\n\x05\x62oost\x18> \x01(\x0b\x32\x12.weaviate.v1.BoostH\x12\x88\x01\x01\x12\x18\n\x0cuses_123_api\x18\x64 \x01(\x08\x42\x02\x18\x01\x12\x18\n\x0cuses_125_api\x18\x65 \x01(\x08\x42\x02\x18\x01\x12\x14\n\x0cuses_127_api\x18\x66 \x01(\x08\x42\x14\n\x12_consistency_levelB\r\n\x0b_propertiesB\x0b\n\t_metadataB\x0b\n\t_group_byB\n\n\x08_filtersB\x10\n\x0e_hybrid_searchB\x0e\n\x0c_bm25_searchB\x0e\n\x0c_near_vectorB\x0e\n\x0c_near_objectB\x0c\n\n_near_textB\r\n\x0b_near_imageB\r\n\x0b_near_audioB\r\n\x0b_near_videoB\r\n\x0b_near_depthB\x0f\n\r_near_thermalB\x0b\n\t_near_imuB\r\n\x0b_generativeB\t\n\x07_rerankB\x08\n\x06_boost\"L\n\x07GroupBy\x12\x0c\n\x04path\x18\x01 \x03(\t\x12\x18\n\x10number_of_groups\x18\x02 \x01(\x05\x12\x19\n\x11objects_per_group\x18\x03 \x01(\x05\")\n\x06SortBy\x12\x11\n\tascending\x18\x01 \x01(\x08\x12\x0c\n\x04path\x18\x02 \x03(\t\"\xf4\x01\n\x0fMetadataRequest\x12\x0c\n\x04uuid\x18\x01 \x01(\x08\x12\x0e\n\x06vector\x18\x02 \x01(\x08\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x04 \x01(\x08\x12\x10\n\x08\x64istance\x18\x05 \x01(\x08\x12\x11\n\tcertainty\x18\x06 \x01(\x08\x12\r\n\x05score\x18\x07 \x01(\x08\x12\x15\n\rexplain_score\x18\x08 \x01(\x08\x12\x15\n\ris_consistent\x18\t \x01(\x08\x12\x0f\n\x07vectors\x18\n \x03(\t\x12\x15\n\rquery_profile\x18\x0b \x01(\x08\"\xd1\x01\n\x11PropertiesRequest\x12\x1a\n\x12non_ref_properties\x18\x01 \x03(\t\x12\x39\n\x0eref_properties\x18\x02 \x03(\x0b\x32!.weaviate.v1.RefPropertiesRequest\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\x12$\n\x1creturn_all_nonref_properties\x18\x0b \x01(\x08\"\x8b\x01\n\x17ObjectPropertiesRequest\x12\x11\n\tprop_name\x18\x01 \x01(\t\x12\x1c\n\x14primitive_properties\x18\x02 \x03(\t\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\"\xb1\x01\n\x14RefPropertiesRequest\x12\x1a\n\x12reference_property\x18\x01 \x01(\t\x12\x32\n\nproperties\x18\x02 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequest\x12.\n\x08metadata\x18\x03 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequest\x12\x19\n\x11target_collection\x18\x04 \x01(\t\"8\n\x06Rerank\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x12\n\x05query\x18\x02 \x01(\tH\x00\x88\x01\x01\x42\x08\n\x06_query\"\xf7\x02\n\x0bSearchReply\x12\x0c\n\x04took\x18\x01 \x01(\x02\x12*\n\x07results\x18\x02 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12*\n\x19generative_grouped_result\x18\x03 \x01(\tB\x02\x18\x01H\x00\x88\x01\x01\x12\x34\n\x10group_by_results\x18\x04 \x03(\x0b\x32\x1a.weaviate.v1.GroupByResult\x12\x46\n\x1agenerative_grouped_results\x18\x05 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x01\x88\x01\x01\x12\x35\n\rquery_profile\x18\x06 \x01(\x0b\x32\x19.weaviate.v1.QueryProfileH\x02\x88\x01\x01\x42\x1c\n\x1a_generative_grouped_resultB\x1d\n\x1b_generative_grouped_resultsB\x10\n\x0e_query_profile\"\x9e\x03\n\x0cQueryProfile\x12\x36\n\x06shards\x18\x01 \x03(\x0b\x32&.weaviate.v1.QueryProfile.ShardProfile\x1a\x86\x01\n\rSearchProfile\x12\x45\n\x07\x64\x65tails\x18\x01 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.SearchProfile.DetailsEntry\x1a.\n\x0c\x44\x65tailsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\xcc\x01\n\x0cShardProfile\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04node\x18\x02 \x01(\t\x12\x46\n\x08searches\x18\x03 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.ShardProfile.SearchesEntry\x1aX\n\rSearchesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x36\n\x05value\x18\x02 \x01(\x0b\x32\'.weaviate.v1.QueryProfile.SearchProfile:\x02\x38\x01\"\x1c\n\x0bRerankReply\x12\r\n\x05score\x18\x01 \x01(\x01\"\xe9\x02\n\rGroupByResult\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\x0cmin_distance\x18\x02 \x01(\x02\x12\x14\n\x0cmax_distance\x18\x03 \x01(\x02\x12\x19\n\x11number_of_objects\x18\x04 \x01(\x03\x12*\n\x07objects\x18\x05 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12-\n\x06rerank\x18\x06 \x01(\x0b\x32\x18.weaviate.v1.RerankReplyH\x00\x88\x01\x01\x12\x39\n\ngenerative\x18\x07 \x01(\x0b\x32\x1c.weaviate.v1.GenerativeReplyB\x02\x18\x01H\x01\x88\x01\x01\x12=\n\x11generative_result\x18\x08 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x02\x88\x01\x01\x42\t\n\x07_rerankB\r\n\x0b_generativeB\x14\n\x12_generative_result\"\xb7\x01\n\x0cSearchResult\x12\x31\n\nproperties\x18\x01 \x01(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12-\n\x08metadata\x18\x02 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12\x36\n\ngenerative\x18\x03 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x00\x88\x01\x01\x42\r\n\x0b_generative\"\xf7\x04\n\x0eMetadataResult\x12\n\n\x02id\x18\x01 \x01(\t\x12\x12\n\x06vector\x18\x02 \x03(\x02\x42\x02\x18\x01\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x03\x12\"\n\x1a\x63reation_time_unix_present\x18\x04 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x05 \x01(\x03\x12%\n\x1dlast_update_time_unix_present\x18\x06 \x01(\x08\x12\x10\n\x08\x64istance\x18\x07 \x01(\x02\x12\x18\n\x10\x64istance_present\x18\x08 \x01(\x08\x12\x11\n\tcertainty\x18\t \x01(\x02\x12\x19\n\x11\x63\x65rtainty_present\x18\n \x01(\x08\x12\r\n\x05score\x18\x0b \x01(\x02\x12\x15\n\rscore_present\x18\x0c \x01(\x08\x12\x15\n\rexplain_score\x18\r \x01(\t\x12\x1d\n\x15\x65xplain_score_present\x18\x0e \x01(\x08\x12\x1a\n\ris_consistent\x18\x0f \x01(\x08H\x00\x88\x01\x01\x12\x16\n\ngenerative\x18\x10 \x01(\tB\x02\x18\x01\x12\x1e\n\x12generative_present\x18\x11 \x01(\x08\x42\x02\x18\x01\x12\x1d\n\x15is_consistent_present\x18\x12 \x01(\x08\x12\x14\n\x0cvector_bytes\x18\x13 \x01(\x0c\x12\x13\n\x0bid_as_bytes\x18\x14 \x01(\x0c\x12\x14\n\x0crerank_score\x18\x15 \x01(\x01\x12\x1c\n\x14rerank_score_present\x18\x16 \x01(\x08\x12%\n\x07vectors\x18\x17 \x03(\x0b\x32\x14.weaviate.v1.VectorsB\x10\n\x0e_is_consistent\"\x88\x02\n\x10PropertiesResult\x12\x33\n\tref_props\x18\x02 \x03(\x0b\x32 .weaviate.v1.RefPropertiesResult\x12\x19\n\x11target_collection\x18\x03 \x01(\t\x12-\n\x08metadata\x18\x04 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12.\n\rnon_ref_props\x18\x0b \x01(\x0b\x32\x17.weaviate.v1.Properties\x12\x1b\n\x13ref_props_requested\x18\x0c \x01(\x08J\x04\x08\x01\x10\x02J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\nJ\x04\x08\n\x10\x0b\"[\n\x13RefPropertiesResult\x12\x31\n\nproperties\x18\x01 \x03(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12\x11\n\tprop_name\x18\x02 \x01(\t\"\xb3\t\n\x05\x42oost\x12\x30\n\nconditions\x18\x01 \x03(\x0b\x32\x1c.weaviate.v1.Boost.Condition\x12\x13\n\x06weight\x18\x02 \x01(\x02H\x00\x88\x01\x01\x12\x12\n\x05\x64\x65pth\x18\x03 \x01(\rH\x01\x88\x01\x01\x1aw\n\x15PropertyValueFunction\x12\x10\n\x08property\x18\x01 \x01(\t\x12?\n\x08modifier\x18\x02 \x01(\x0e\x32(.weaviate.v1.Boost.PropertyValueModifierH\x00\x88\x01\x01\x42\x0b\n\t_modifier\x1a\xcb\x01\n\x11TimeDecayFunction\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x0e\n\x06origin\x18\x02 \x01(\t\x12\r\n\x05scale\x18\x03 \x01(\t\x12\x13\n\x06offset\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x31\n\x05\x63urve\x18\x05 \x01(\x0e\x32\x1d.weaviate.v1.Boost.DecayCurveH\x01\x88\x01\x01\x12\x18\n\x0b\x64\x65\x63\x61y_value\x18\x06 \x01(\x02H\x02\x88\x01\x01\x42\t\n\x07_offsetB\x08\n\x06_curveB\x0e\n\x0c_decay_value\x1a\xce\x01\n\x14NumericDecayFunction\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x0e\n\x06origin\x18\x02 \x01(\x01\x12\r\n\x05scale\x18\x03 \x01(\x01\x12\x13\n\x06offset\x18\x04 \x01(\x01H\x00\x88\x01\x01\x12\x31\n\x05\x63urve\x18\x05 \x01(\x0e\x32\x1d.weaviate.v1.Boost.DecayCurveH\x01\x88\x01\x01\x12\x18\n\x0b\x64\x65\x63\x61y_value\x18\x06 \x01(\x02H\x02\x88\x01\x01\x42\t\n\x07_offsetB\x08\n\x06_curveB\x0e\n\x0c_decay_value\x1a\xa2\x02\n\tCondition\x12&\n\x06\x66ilter\x18\x01 \x01(\x0b\x32\x14.weaviate.v1.FiltersH\x00\x12:\n\ntime_decay\x18\x02 \x01(\x0b\x32$.weaviate.v1.Boost.TimeDecayFunctionH\x00\x12\x42\n\x0eproperty_value\x18\x03 \x01(\x0b\x32(.weaviate.v1.Boost.PropertyValueFunctionH\x00\x12@\n\rnumeric_decay\x18\x04 \x01(\x0b\x32\'.weaviate.v1.Boost.NumericDecayFunctionH\x00\x12\x13\n\x06weight\x18\x05 \x01(\x02H\x01\x88\x01\x01\x42\x0b\n\tconditionB\t\n\x07_weight\"\x85\x01\n\x15PropertyValueModifier\x12\'\n#PROPERTY_VALUE_MODIFIER_UNSPECIFIED\x10\x00\x12!\n\x1dPROPERTY_VALUE_MODIFIER_LOG1P\x10\x01\x12 \n\x1cPROPERTY_VALUE_MODIFIER_SQRT\x10\x02\"u\n\nDecayCurve\x12\x1b\n\x17\x44\x45\x43\x41Y_CURVE_UNSPECIFIED\x10\x00\x12\x15\n\x11\x44\x45\x43\x41Y_CURVE_GAUSS\x10\x01\x12\x16\n\x12\x44\x45\x43\x41Y_CURVE_LINEAR\x10\x02\x12\x1b\n\x17\x44\x45\x43\x41Y_CURVE_EXPONENTIAL\x10\x03\x42\t\n\x07_weightB\x08\n\x06_depthBs\n#io.weaviate.client.grpc.protocol.v1B\x16WeaviateProtoSearchGetZ4github.com/weaviate/weaviate/grpc/generated;protocolb\x06proto3') _globals = globals() _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals) @@ -44,43 +44,57 @@ _METADATARESULT.fields_by_name['generative_present']._options = None _METADATARESULT.fields_by_name['generative_present']._serialized_options = b'\030\001' _globals['_SEARCHREQUEST']._serialized_start=116 - _globals['_SEARCHREQUEST']._serialized_end=1552 - _globals['_GROUPBY']._serialized_start=1554 - _globals['_GROUPBY']._serialized_end=1630 - _globals['_SORTBY']._serialized_start=1632 - _globals['_SORTBY']._serialized_end=1673 - _globals['_METADATAREQUEST']._serialized_start=1676 - _globals['_METADATAREQUEST']._serialized_end=1920 - _globals['_PROPERTIESREQUEST']._serialized_start=1923 - _globals['_PROPERTIESREQUEST']._serialized_end=2132 - _globals['_OBJECTPROPERTIESREQUEST']._serialized_start=2135 - _globals['_OBJECTPROPERTIESREQUEST']._serialized_end=2274 - _globals['_REFPROPERTIESREQUEST']._serialized_start=2277 - _globals['_REFPROPERTIESREQUEST']._serialized_end=2454 - _globals['_RERANK']._serialized_start=2456 - _globals['_RERANK']._serialized_end=2512 - _globals['_SEARCHREPLY']._serialized_start=2515 - _globals['_SEARCHREPLY']._serialized_end=2890 - _globals['_QUERYPROFILE']._serialized_start=2893 - _globals['_QUERYPROFILE']._serialized_end=3307 - _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_start=2966 - _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_end=3100 - _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_start=3054 - _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_end=3100 - _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_start=3103 - _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_end=3307 - _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_start=3219 - _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_end=3307 - _globals['_RERANKREPLY']._serialized_start=3309 - _globals['_RERANKREPLY']._serialized_end=3337 - _globals['_GROUPBYRESULT']._serialized_start=3340 - _globals['_GROUPBYRESULT']._serialized_end=3701 - _globals['_SEARCHRESULT']._serialized_start=3704 - _globals['_SEARCHRESULT']._serialized_end=3887 - _globals['_METADATARESULT']._serialized_start=3890 - _globals['_METADATARESULT']._serialized_end=4521 - _globals['_PROPERTIESRESULT']._serialized_start=4524 - _globals['_PROPERTIESRESULT']._serialized_end=4788 - _globals['_REFPROPERTIESRESULT']._serialized_start=4790 - _globals['_REFPROPERTIESRESULT']._serialized_end=4881 + _globals['_SEARCHREQUEST']._serialized_end=1602 + _globals['_GROUPBY']._serialized_start=1604 + _globals['_GROUPBY']._serialized_end=1680 + _globals['_SORTBY']._serialized_start=1682 + _globals['_SORTBY']._serialized_end=1723 + _globals['_METADATAREQUEST']._serialized_start=1726 + _globals['_METADATAREQUEST']._serialized_end=1970 + _globals['_PROPERTIESREQUEST']._serialized_start=1973 + _globals['_PROPERTIESREQUEST']._serialized_end=2182 + _globals['_OBJECTPROPERTIESREQUEST']._serialized_start=2185 + _globals['_OBJECTPROPERTIESREQUEST']._serialized_end=2324 + _globals['_REFPROPERTIESREQUEST']._serialized_start=2327 + _globals['_REFPROPERTIESREQUEST']._serialized_end=2504 + _globals['_RERANK']._serialized_start=2506 + _globals['_RERANK']._serialized_end=2562 + _globals['_SEARCHREPLY']._serialized_start=2565 + _globals['_SEARCHREPLY']._serialized_end=2940 + _globals['_QUERYPROFILE']._serialized_start=2943 + _globals['_QUERYPROFILE']._serialized_end=3357 + _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_start=3016 + _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_end=3150 + _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_start=3104 + _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_end=3150 + _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_start=3153 + _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_end=3357 + _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_start=3269 + _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_end=3357 + _globals['_RERANKREPLY']._serialized_start=3359 + _globals['_RERANKREPLY']._serialized_end=3387 + _globals['_GROUPBYRESULT']._serialized_start=3390 + _globals['_GROUPBYRESULT']._serialized_end=3751 + _globals['_SEARCHRESULT']._serialized_start=3754 + _globals['_SEARCHRESULT']._serialized_end=3937 + _globals['_METADATARESULT']._serialized_start=3940 + _globals['_METADATARESULT']._serialized_end=4571 + _globals['_PROPERTIESRESULT']._serialized_start=4574 + _globals['_PROPERTIESRESULT']._serialized_end=4838 + _globals['_REFPROPERTIESRESULT']._serialized_start=4840 + _globals['_REFPROPERTIESRESULT']._serialized_end=4931 + _globals['_BOOST']._serialized_start=4934 + _globals['_BOOST']._serialized_end=6137 + _globals['_BOOST_PROPERTYVALUEFUNCTION']._serialized_start=5034 + _globals['_BOOST_PROPERTYVALUEFUNCTION']._serialized_end=5153 + _globals['_BOOST_TIMEDECAYFUNCTION']._serialized_start=5156 + _globals['_BOOST_TIMEDECAYFUNCTION']._serialized_end=5359 + _globals['_BOOST_NUMERICDECAYFUNCTION']._serialized_start=5362 + _globals['_BOOST_NUMERICDECAYFUNCTION']._serialized_end=5568 + _globals['_BOOST_CONDITION']._serialized_start=5571 + _globals['_BOOST_CONDITION']._serialized_end=5861 + _globals['_BOOST_PROPERTYVALUEMODIFIER']._serialized_start=5864 + _globals['_BOOST_PROPERTYVALUEMODIFIER']._serialized_end=5997 + _globals['_BOOST_DECAYCURVE']._serialized_start=5999 + _globals['_BOOST_DECAYCURVE']._serialized_end=6116 # @@protoc_insertion_point(module_scope) diff --git a/weaviate/proto/v1/v4216/v1/search_get_pb2.pyi b/weaviate/proto/v1/v4216/v1/search_get_pb2.pyi index a0300e672..c65735cb9 100644 --- a/weaviate/proto/v1/v4216/v1/search_get_pb2.pyi +++ b/weaviate/proto/v1/v4216/v1/search_get_pb2.pyi @@ -3,6 +3,7 @@ from weaviate.proto.v1.v4216.v1 import base_search_pb2 as _base_search_pb2 from weaviate.proto.v1.v4216.v1 import generative_pb2 as _generative_pb2 from weaviate.proto.v1.v4216.v1 import properties_pb2 as _properties_pb2 from google.protobuf.internal import containers as _containers +from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper from google.protobuf import descriptor as _descriptor from google.protobuf import message as _message from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union @@ -10,7 +11,7 @@ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Map DESCRIPTOR: _descriptor.FileDescriptor class SearchRequest(_message.Message): - __slots__ = ["collection", "tenant", "consistency_level", "properties", "metadata", "group_by", "limit", "offset", "autocut", "after", "sort_by", "filters", "hybrid_search", "bm25_search", "near_vector", "near_object", "near_text", "near_image", "near_audio", "near_video", "near_depth", "near_thermal", "near_imu", "generative", "rerank", "uses_123_api", "uses_125_api", "uses_127_api"] + __slots__ = ["collection", "tenant", "consistency_level", "properties", "metadata", "group_by", "limit", "offset", "autocut", "after", "sort_by", "filters", "hybrid_search", "bm25_search", "near_vector", "near_object", "near_text", "near_image", "near_audio", "near_video", "near_depth", "near_thermal", "near_imu", "generative", "rerank", "boost", "uses_123_api", "uses_125_api", "uses_127_api"] COLLECTION_FIELD_NUMBER: _ClassVar[int] TENANT_FIELD_NUMBER: _ClassVar[int] CONSISTENCY_LEVEL_FIELD_NUMBER: _ClassVar[int] @@ -36,6 +37,7 @@ class SearchRequest(_message.Message): NEAR_IMU_FIELD_NUMBER: _ClassVar[int] GENERATIVE_FIELD_NUMBER: _ClassVar[int] RERANK_FIELD_NUMBER: _ClassVar[int] + BOOST_FIELD_NUMBER: _ClassVar[int] USES_123_API_FIELD_NUMBER: _ClassVar[int] USES_125_API_FIELD_NUMBER: _ClassVar[int] USES_127_API_FIELD_NUMBER: _ClassVar[int] @@ -64,10 +66,11 @@ class SearchRequest(_message.Message): near_imu: _base_search_pb2.NearIMUSearch generative: _generative_pb2.GenerativeSearch rerank: Rerank + boost: Boost uses_123_api: bool uses_125_api: bool uses_127_api: bool - def __init__(self, collection: _Optional[str] = ..., tenant: _Optional[str] = ..., consistency_level: _Optional[_Union[_base_pb2.ConsistencyLevel, str]] = ..., properties: _Optional[_Union[PropertiesRequest, _Mapping]] = ..., metadata: _Optional[_Union[MetadataRequest, _Mapping]] = ..., group_by: _Optional[_Union[GroupBy, _Mapping]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., autocut: _Optional[int] = ..., after: _Optional[str] = ..., sort_by: _Optional[_Iterable[_Union[SortBy, _Mapping]]] = ..., filters: _Optional[_Union[_base_pb2.Filters, _Mapping]] = ..., hybrid_search: _Optional[_Union[_base_search_pb2.Hybrid, _Mapping]] = ..., bm25_search: _Optional[_Union[_base_search_pb2.BM25, _Mapping]] = ..., near_vector: _Optional[_Union[_base_search_pb2.NearVector, _Mapping]] = ..., near_object: _Optional[_Union[_base_search_pb2.NearObject, _Mapping]] = ..., near_text: _Optional[_Union[_base_search_pb2.NearTextSearch, _Mapping]] = ..., near_image: _Optional[_Union[_base_search_pb2.NearImageSearch, _Mapping]] = ..., near_audio: _Optional[_Union[_base_search_pb2.NearAudioSearch, _Mapping]] = ..., near_video: _Optional[_Union[_base_search_pb2.NearVideoSearch, _Mapping]] = ..., near_depth: _Optional[_Union[_base_search_pb2.NearDepthSearch, _Mapping]] = ..., near_thermal: _Optional[_Union[_base_search_pb2.NearThermalSearch, _Mapping]] = ..., near_imu: _Optional[_Union[_base_search_pb2.NearIMUSearch, _Mapping]] = ..., generative: _Optional[_Union[_generative_pb2.GenerativeSearch, _Mapping]] = ..., rerank: _Optional[_Union[Rerank, _Mapping]] = ..., uses_123_api: bool = ..., uses_125_api: bool = ..., uses_127_api: bool = ...) -> None: ... + def __init__(self, collection: _Optional[str] = ..., tenant: _Optional[str] = ..., consistency_level: _Optional[_Union[_base_pb2.ConsistencyLevel, str]] = ..., properties: _Optional[_Union[PropertiesRequest, _Mapping]] = ..., metadata: _Optional[_Union[MetadataRequest, _Mapping]] = ..., group_by: _Optional[_Union[GroupBy, _Mapping]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., autocut: _Optional[int] = ..., after: _Optional[str] = ..., sort_by: _Optional[_Iterable[_Union[SortBy, _Mapping]]] = ..., filters: _Optional[_Union[_base_pb2.Filters, _Mapping]] = ..., hybrid_search: _Optional[_Union[_base_search_pb2.Hybrid, _Mapping]] = ..., bm25_search: _Optional[_Union[_base_search_pb2.BM25, _Mapping]] = ..., near_vector: _Optional[_Union[_base_search_pb2.NearVector, _Mapping]] = ..., near_object: _Optional[_Union[_base_search_pb2.NearObject, _Mapping]] = ..., near_text: _Optional[_Union[_base_search_pb2.NearTextSearch, _Mapping]] = ..., near_image: _Optional[_Union[_base_search_pb2.NearImageSearch, _Mapping]] = ..., near_audio: _Optional[_Union[_base_search_pb2.NearAudioSearch, _Mapping]] = ..., near_video: _Optional[_Union[_base_search_pb2.NearVideoSearch, _Mapping]] = ..., near_depth: _Optional[_Union[_base_search_pb2.NearDepthSearch, _Mapping]] = ..., near_thermal: _Optional[_Union[_base_search_pb2.NearThermalSearch, _Mapping]] = ..., near_imu: _Optional[_Union[_base_search_pb2.NearIMUSearch, _Mapping]] = ..., generative: _Optional[_Union[_generative_pb2.GenerativeSearch, _Mapping]] = ..., rerank: _Optional[_Union[Rerank, _Mapping]] = ..., boost: _Optional[_Union[Boost, _Mapping]] = ..., uses_123_api: bool = ..., uses_125_api: bool = ..., uses_127_api: bool = ...) -> None: ... class GroupBy(_message.Message): __slots__ = ["path", "number_of_groups", "objects_per_group"] @@ -312,3 +315,81 @@ class RefPropertiesResult(_message.Message): properties: _containers.RepeatedCompositeFieldContainer[PropertiesResult] prop_name: str def __init__(self, properties: _Optional[_Iterable[_Union[PropertiesResult, _Mapping]]] = ..., prop_name: _Optional[str] = ...) -> None: ... + +class Boost(_message.Message): + __slots__ = ["conditions", "weight", "depth"] + class PropertyValueModifier(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): + __slots__ = [] + PROPERTY_VALUE_MODIFIER_UNSPECIFIED: _ClassVar[Boost.PropertyValueModifier] + PROPERTY_VALUE_MODIFIER_LOG1P: _ClassVar[Boost.PropertyValueModifier] + PROPERTY_VALUE_MODIFIER_SQRT: _ClassVar[Boost.PropertyValueModifier] + PROPERTY_VALUE_MODIFIER_UNSPECIFIED: Boost.PropertyValueModifier + PROPERTY_VALUE_MODIFIER_LOG1P: Boost.PropertyValueModifier + PROPERTY_VALUE_MODIFIER_SQRT: Boost.PropertyValueModifier + class DecayCurve(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): + __slots__ = [] + DECAY_CURVE_UNSPECIFIED: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_GAUSS: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_LINEAR: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_EXPONENTIAL: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_UNSPECIFIED: Boost.DecayCurve + DECAY_CURVE_GAUSS: Boost.DecayCurve + DECAY_CURVE_LINEAR: Boost.DecayCurve + DECAY_CURVE_EXPONENTIAL: Boost.DecayCurve + class PropertyValueFunction(_message.Message): + __slots__ = ["property", "modifier"] + PROPERTY_FIELD_NUMBER: _ClassVar[int] + MODIFIER_FIELD_NUMBER: _ClassVar[int] + property: str + modifier: Boost.PropertyValueModifier + def __init__(self, property: _Optional[str] = ..., modifier: _Optional[_Union[Boost.PropertyValueModifier, str]] = ...) -> None: ... + class TimeDecayFunction(_message.Message): + __slots__ = ["property", "origin", "scale", "offset", "curve", "decay_value"] + PROPERTY_FIELD_NUMBER: _ClassVar[int] + ORIGIN_FIELD_NUMBER: _ClassVar[int] + SCALE_FIELD_NUMBER: _ClassVar[int] + OFFSET_FIELD_NUMBER: _ClassVar[int] + CURVE_FIELD_NUMBER: _ClassVar[int] + DECAY_VALUE_FIELD_NUMBER: _ClassVar[int] + property: str + origin: str + scale: str + offset: str + curve: Boost.DecayCurve + decay_value: float + def __init__(self, property: _Optional[str] = ..., origin: _Optional[str] = ..., scale: _Optional[str] = ..., offset: _Optional[str] = ..., curve: _Optional[_Union[Boost.DecayCurve, str]] = ..., decay_value: _Optional[float] = ...) -> None: ... + class NumericDecayFunction(_message.Message): + __slots__ = ["property", "origin", "scale", "offset", "curve", "decay_value"] + PROPERTY_FIELD_NUMBER: _ClassVar[int] + ORIGIN_FIELD_NUMBER: _ClassVar[int] + SCALE_FIELD_NUMBER: _ClassVar[int] + OFFSET_FIELD_NUMBER: _ClassVar[int] + CURVE_FIELD_NUMBER: _ClassVar[int] + DECAY_VALUE_FIELD_NUMBER: _ClassVar[int] + property: str + origin: float + scale: float + offset: float + curve: Boost.DecayCurve + decay_value: float + def __init__(self, property: _Optional[str] = ..., origin: _Optional[float] = ..., scale: _Optional[float] = ..., offset: _Optional[float] = ..., curve: _Optional[_Union[Boost.DecayCurve, str]] = ..., decay_value: _Optional[float] = ...) -> None: ... + class Condition(_message.Message): + __slots__ = ["filter", "time_decay", "property_value", "numeric_decay", "weight"] + FILTER_FIELD_NUMBER: _ClassVar[int] + TIME_DECAY_FIELD_NUMBER: _ClassVar[int] + PROPERTY_VALUE_FIELD_NUMBER: _ClassVar[int] + NUMERIC_DECAY_FIELD_NUMBER: _ClassVar[int] + WEIGHT_FIELD_NUMBER: _ClassVar[int] + filter: _base_pb2.Filters + time_decay: Boost.TimeDecayFunction + property_value: Boost.PropertyValueFunction + numeric_decay: Boost.NumericDecayFunction + weight: float + def __init__(self, filter: _Optional[_Union[_base_pb2.Filters, _Mapping]] = ..., time_decay: _Optional[_Union[Boost.TimeDecayFunction, _Mapping]] = ..., property_value: _Optional[_Union[Boost.PropertyValueFunction, _Mapping]] = ..., numeric_decay: _Optional[_Union[Boost.NumericDecayFunction, _Mapping]] = ..., weight: _Optional[float] = ...) -> None: ... + CONDITIONS_FIELD_NUMBER: _ClassVar[int] + WEIGHT_FIELD_NUMBER: _ClassVar[int] + DEPTH_FIELD_NUMBER: _ClassVar[int] + conditions: _containers.RepeatedCompositeFieldContainer[Boost.Condition] + weight: float + depth: int + def __init__(self, conditions: _Optional[_Iterable[_Union[Boost.Condition, _Mapping]]] = ..., weight: _Optional[float] = ..., depth: _Optional[int] = ...) -> None: ... diff --git a/weaviate/proto/v1/v5261/v1/search_get_pb2.py b/weaviate/proto/v1/v5261/v1/search_get_pb2.py index 7686973af..e3bb860c4 100644 --- a/weaviate/proto/v1/v5261/v1/search_get_pb2.py +++ b/weaviate/proto/v1/v5261/v1/search_get_pb2.py @@ -18,7 +18,7 @@ from weaviate.proto.v1.v5261.v1 import properties_pb2 as v1_dot_properties__pb2 -DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13v1/search_get.proto\x12\x0bweaviate.v1\x1a\rv1/base.proto\x1a\x14v1/base_search.proto\x1a\x13v1/generative.proto\x1a\x13v1/properties.proto\"\x9c\x0b\n\rSearchRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\x0e\n\x06tenant\x18\n \x01(\t\x12=\n\x11\x63onsistency_level\x18\x0b \x01(\x0e\x32\x1d.weaviate.v1.ConsistencyLevelH\x00\x88\x01\x01\x12\x37\n\nproperties\x18\x14 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequestH\x01\x88\x01\x01\x12\x33\n\x08metadata\x18\x15 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequestH\x02\x88\x01\x01\x12+\n\x08group_by\x18\x16 \x01(\x0b\x32\x14.weaviate.v1.GroupByH\x03\x88\x01\x01\x12\r\n\x05limit\x18\x1e \x01(\r\x12\x0e\n\x06offset\x18\x1f \x01(\r\x12\x0f\n\x07\x61utocut\x18 \x01(\r\x12\r\n\x05\x61\x66ter\x18! \x01(\t\x12$\n\x07sort_by\x18\" \x03(\x0b\x32\x13.weaviate.v1.SortBy\x12*\n\x07\x66ilters\x18( \x01(\x0b\x32\x14.weaviate.v1.FiltersH\x04\x88\x01\x01\x12/\n\rhybrid_search\x18) \x01(\x0b\x32\x13.weaviate.v1.HybridH\x05\x88\x01\x01\x12+\n\x0b\x62m25_search\x18* \x01(\x0b\x32\x11.weaviate.v1.BM25H\x06\x88\x01\x01\x12\x31\n\x0bnear_vector\x18+ \x01(\x0b\x32\x17.weaviate.v1.NearVectorH\x07\x88\x01\x01\x12\x31\n\x0bnear_object\x18, \x01(\x0b\x32\x17.weaviate.v1.NearObjectH\x08\x88\x01\x01\x12\x33\n\tnear_text\x18- \x01(\x0b\x32\x1b.weaviate.v1.NearTextSearchH\t\x88\x01\x01\x12\x35\n\nnear_image\x18. \x01(\x0b\x32\x1c.weaviate.v1.NearImageSearchH\n\x88\x01\x01\x12\x35\n\nnear_audio\x18/ \x01(\x0b\x32\x1c.weaviate.v1.NearAudioSearchH\x0b\x88\x01\x01\x12\x35\n\nnear_video\x18\x30 \x01(\x0b\x32\x1c.weaviate.v1.NearVideoSearchH\x0c\x88\x01\x01\x12\x35\n\nnear_depth\x18\x31 \x01(\x0b\x32\x1c.weaviate.v1.NearDepthSearchH\r\x88\x01\x01\x12\x39\n\x0cnear_thermal\x18\x32 \x01(\x0b\x32\x1e.weaviate.v1.NearThermalSearchH\x0e\x88\x01\x01\x12\x31\n\x08near_imu\x18\x33 \x01(\x0b\x32\x1a.weaviate.v1.NearIMUSearchH\x0f\x88\x01\x01\x12\x36\n\ngenerative\x18< \x01(\x0b\x32\x1d.weaviate.v1.GenerativeSearchH\x10\x88\x01\x01\x12(\n\x06rerank\x18= \x01(\x0b\x32\x13.weaviate.v1.RerankH\x11\x88\x01\x01\x12\x18\n\x0cuses_123_api\x18\x64 \x01(\x08\x42\x02\x18\x01\x12\x18\n\x0cuses_125_api\x18\x65 \x01(\x08\x42\x02\x18\x01\x12\x14\n\x0cuses_127_api\x18\x66 \x01(\x08\x42\x14\n\x12_consistency_levelB\r\n\x0b_propertiesB\x0b\n\t_metadataB\x0b\n\t_group_byB\n\n\x08_filtersB\x10\n\x0e_hybrid_searchB\x0e\n\x0c_bm25_searchB\x0e\n\x0c_near_vectorB\x0e\n\x0c_near_objectB\x0c\n\n_near_textB\r\n\x0b_near_imageB\r\n\x0b_near_audioB\r\n\x0b_near_videoB\r\n\x0b_near_depthB\x0f\n\r_near_thermalB\x0b\n\t_near_imuB\r\n\x0b_generativeB\t\n\x07_rerank\"L\n\x07GroupBy\x12\x0c\n\x04path\x18\x01 \x03(\t\x12\x18\n\x10number_of_groups\x18\x02 \x01(\x05\x12\x19\n\x11objects_per_group\x18\x03 \x01(\x05\")\n\x06SortBy\x12\x11\n\tascending\x18\x01 \x01(\x08\x12\x0c\n\x04path\x18\x02 \x03(\t\"\xf4\x01\n\x0fMetadataRequest\x12\x0c\n\x04uuid\x18\x01 \x01(\x08\x12\x0e\n\x06vector\x18\x02 \x01(\x08\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x04 \x01(\x08\x12\x10\n\x08\x64istance\x18\x05 \x01(\x08\x12\x11\n\tcertainty\x18\x06 \x01(\x08\x12\r\n\x05score\x18\x07 \x01(\x08\x12\x15\n\rexplain_score\x18\x08 \x01(\x08\x12\x15\n\ris_consistent\x18\t \x01(\x08\x12\x0f\n\x07vectors\x18\n \x03(\t\x12\x15\n\rquery_profile\x18\x0b \x01(\x08\"\xd1\x01\n\x11PropertiesRequest\x12\x1a\n\x12non_ref_properties\x18\x01 \x03(\t\x12\x39\n\x0eref_properties\x18\x02 \x03(\x0b\x32!.weaviate.v1.RefPropertiesRequest\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\x12$\n\x1creturn_all_nonref_properties\x18\x0b \x01(\x08\"\x8b\x01\n\x17ObjectPropertiesRequest\x12\x11\n\tprop_name\x18\x01 \x01(\t\x12\x1c\n\x14primitive_properties\x18\x02 \x03(\t\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\"\xb1\x01\n\x14RefPropertiesRequest\x12\x1a\n\x12reference_property\x18\x01 \x01(\t\x12\x32\n\nproperties\x18\x02 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequest\x12.\n\x08metadata\x18\x03 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequest\x12\x19\n\x11target_collection\x18\x04 \x01(\t\"8\n\x06Rerank\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x12\n\x05query\x18\x02 \x01(\tH\x00\x88\x01\x01\x42\x08\n\x06_query\"\xf7\x02\n\x0bSearchReply\x12\x0c\n\x04took\x18\x01 \x01(\x02\x12*\n\x07results\x18\x02 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12*\n\x19generative_grouped_result\x18\x03 \x01(\tB\x02\x18\x01H\x00\x88\x01\x01\x12\x34\n\x10group_by_results\x18\x04 \x03(\x0b\x32\x1a.weaviate.v1.GroupByResult\x12\x46\n\x1agenerative_grouped_results\x18\x05 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x01\x88\x01\x01\x12\x35\n\rquery_profile\x18\x06 \x01(\x0b\x32\x19.weaviate.v1.QueryProfileH\x02\x88\x01\x01\x42\x1c\n\x1a_generative_grouped_resultB\x1d\n\x1b_generative_grouped_resultsB\x10\n\x0e_query_profile\"\x9e\x03\n\x0cQueryProfile\x12\x36\n\x06shards\x18\x01 \x03(\x0b\x32&.weaviate.v1.QueryProfile.ShardProfile\x1a\x86\x01\n\rSearchProfile\x12\x45\n\x07\x64\x65tails\x18\x01 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.SearchProfile.DetailsEntry\x1a.\n\x0c\x44\x65tailsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\xcc\x01\n\x0cShardProfile\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04node\x18\x02 \x01(\t\x12\x46\n\x08searches\x18\x03 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.ShardProfile.SearchesEntry\x1aX\n\rSearchesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x36\n\x05value\x18\x02 \x01(\x0b\x32\'.weaviate.v1.QueryProfile.SearchProfile:\x02\x38\x01\"\x1c\n\x0bRerankReply\x12\r\n\x05score\x18\x01 \x01(\x01\"\xe9\x02\n\rGroupByResult\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\x0cmin_distance\x18\x02 \x01(\x02\x12\x14\n\x0cmax_distance\x18\x03 \x01(\x02\x12\x19\n\x11number_of_objects\x18\x04 \x01(\x03\x12*\n\x07objects\x18\x05 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12-\n\x06rerank\x18\x06 \x01(\x0b\x32\x18.weaviate.v1.RerankReplyH\x00\x88\x01\x01\x12\x39\n\ngenerative\x18\x07 \x01(\x0b\x32\x1c.weaviate.v1.GenerativeReplyB\x02\x18\x01H\x01\x88\x01\x01\x12=\n\x11generative_result\x18\x08 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x02\x88\x01\x01\x42\t\n\x07_rerankB\r\n\x0b_generativeB\x14\n\x12_generative_result\"\xb7\x01\n\x0cSearchResult\x12\x31\n\nproperties\x18\x01 \x01(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12-\n\x08metadata\x18\x02 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12\x36\n\ngenerative\x18\x03 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x00\x88\x01\x01\x42\r\n\x0b_generative\"\xf7\x04\n\x0eMetadataResult\x12\n\n\x02id\x18\x01 \x01(\t\x12\x12\n\x06vector\x18\x02 \x03(\x02\x42\x02\x18\x01\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x03\x12\"\n\x1a\x63reation_time_unix_present\x18\x04 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x05 \x01(\x03\x12%\n\x1dlast_update_time_unix_present\x18\x06 \x01(\x08\x12\x10\n\x08\x64istance\x18\x07 \x01(\x02\x12\x18\n\x10\x64istance_present\x18\x08 \x01(\x08\x12\x11\n\tcertainty\x18\t \x01(\x02\x12\x19\n\x11\x63\x65rtainty_present\x18\n \x01(\x08\x12\r\n\x05score\x18\x0b \x01(\x02\x12\x15\n\rscore_present\x18\x0c \x01(\x08\x12\x15\n\rexplain_score\x18\r \x01(\t\x12\x1d\n\x15\x65xplain_score_present\x18\x0e \x01(\x08\x12\x1a\n\ris_consistent\x18\x0f \x01(\x08H\x00\x88\x01\x01\x12\x16\n\ngenerative\x18\x10 \x01(\tB\x02\x18\x01\x12\x1e\n\x12generative_present\x18\x11 \x01(\x08\x42\x02\x18\x01\x12\x1d\n\x15is_consistent_present\x18\x12 \x01(\x08\x12\x14\n\x0cvector_bytes\x18\x13 \x01(\x0c\x12\x13\n\x0bid_as_bytes\x18\x14 \x01(\x0c\x12\x14\n\x0crerank_score\x18\x15 \x01(\x01\x12\x1c\n\x14rerank_score_present\x18\x16 \x01(\x08\x12%\n\x07vectors\x18\x17 \x03(\x0b\x32\x14.weaviate.v1.VectorsB\x10\n\x0e_is_consistent\"\x88\x02\n\x10PropertiesResult\x12\x33\n\tref_props\x18\x02 \x03(\x0b\x32 .weaviate.v1.RefPropertiesResult\x12\x19\n\x11target_collection\x18\x03 \x01(\t\x12-\n\x08metadata\x18\x04 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12.\n\rnon_ref_props\x18\x0b \x01(\x0b\x32\x17.weaviate.v1.Properties\x12\x1b\n\x13ref_props_requested\x18\x0c \x01(\x08J\x04\x08\x01\x10\x02J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\nJ\x04\x08\n\x10\x0b\"[\n\x13RefPropertiesResult\x12\x31\n\nproperties\x18\x01 \x03(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12\x11\n\tprop_name\x18\x02 \x01(\tBs\n#io.weaviate.client.grpc.protocol.v1B\x16WeaviateProtoSearchGetZ4github.com/weaviate/weaviate/grpc/generated;protocolb\x06proto3') +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13v1/search_get.proto\x12\x0bweaviate.v1\x1a\rv1/base.proto\x1a\x14v1/base_search.proto\x1a\x13v1/generative.proto\x1a\x13v1/properties.proto\"\xce\x0b\n\rSearchRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\x0e\n\x06tenant\x18\n \x01(\t\x12=\n\x11\x63onsistency_level\x18\x0b \x01(\x0e\x32\x1d.weaviate.v1.ConsistencyLevelH\x00\x88\x01\x01\x12\x37\n\nproperties\x18\x14 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequestH\x01\x88\x01\x01\x12\x33\n\x08metadata\x18\x15 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequestH\x02\x88\x01\x01\x12+\n\x08group_by\x18\x16 \x01(\x0b\x32\x14.weaviate.v1.GroupByH\x03\x88\x01\x01\x12\r\n\x05limit\x18\x1e \x01(\r\x12\x0e\n\x06offset\x18\x1f \x01(\r\x12\x0f\n\x07\x61utocut\x18 \x01(\r\x12\r\n\x05\x61\x66ter\x18! \x01(\t\x12$\n\x07sort_by\x18\" \x03(\x0b\x32\x13.weaviate.v1.SortBy\x12*\n\x07\x66ilters\x18( \x01(\x0b\x32\x14.weaviate.v1.FiltersH\x04\x88\x01\x01\x12/\n\rhybrid_search\x18) \x01(\x0b\x32\x13.weaviate.v1.HybridH\x05\x88\x01\x01\x12+\n\x0b\x62m25_search\x18* \x01(\x0b\x32\x11.weaviate.v1.BM25H\x06\x88\x01\x01\x12\x31\n\x0bnear_vector\x18+ \x01(\x0b\x32\x17.weaviate.v1.NearVectorH\x07\x88\x01\x01\x12\x31\n\x0bnear_object\x18, \x01(\x0b\x32\x17.weaviate.v1.NearObjectH\x08\x88\x01\x01\x12\x33\n\tnear_text\x18- \x01(\x0b\x32\x1b.weaviate.v1.NearTextSearchH\t\x88\x01\x01\x12\x35\n\nnear_image\x18. \x01(\x0b\x32\x1c.weaviate.v1.NearImageSearchH\n\x88\x01\x01\x12\x35\n\nnear_audio\x18/ \x01(\x0b\x32\x1c.weaviate.v1.NearAudioSearchH\x0b\x88\x01\x01\x12\x35\n\nnear_video\x18\x30 \x01(\x0b\x32\x1c.weaviate.v1.NearVideoSearchH\x0c\x88\x01\x01\x12\x35\n\nnear_depth\x18\x31 \x01(\x0b\x32\x1c.weaviate.v1.NearDepthSearchH\r\x88\x01\x01\x12\x39\n\x0cnear_thermal\x18\x32 \x01(\x0b\x32\x1e.weaviate.v1.NearThermalSearchH\x0e\x88\x01\x01\x12\x31\n\x08near_imu\x18\x33 \x01(\x0b\x32\x1a.weaviate.v1.NearIMUSearchH\x0f\x88\x01\x01\x12\x36\n\ngenerative\x18< \x01(\x0b\x32\x1d.weaviate.v1.GenerativeSearchH\x10\x88\x01\x01\x12(\n\x06rerank\x18= \x01(\x0b\x32\x13.weaviate.v1.RerankH\x11\x88\x01\x01\x12&\n\x05\x62oost\x18> \x01(\x0b\x32\x12.weaviate.v1.BoostH\x12\x88\x01\x01\x12\x18\n\x0cuses_123_api\x18\x64 \x01(\x08\x42\x02\x18\x01\x12\x18\n\x0cuses_125_api\x18\x65 \x01(\x08\x42\x02\x18\x01\x12\x14\n\x0cuses_127_api\x18\x66 \x01(\x08\x42\x14\n\x12_consistency_levelB\r\n\x0b_propertiesB\x0b\n\t_metadataB\x0b\n\t_group_byB\n\n\x08_filtersB\x10\n\x0e_hybrid_searchB\x0e\n\x0c_bm25_searchB\x0e\n\x0c_near_vectorB\x0e\n\x0c_near_objectB\x0c\n\n_near_textB\r\n\x0b_near_imageB\r\n\x0b_near_audioB\r\n\x0b_near_videoB\r\n\x0b_near_depthB\x0f\n\r_near_thermalB\x0b\n\t_near_imuB\r\n\x0b_generativeB\t\n\x07_rerankB\x08\n\x06_boost\"L\n\x07GroupBy\x12\x0c\n\x04path\x18\x01 \x03(\t\x12\x18\n\x10number_of_groups\x18\x02 \x01(\x05\x12\x19\n\x11objects_per_group\x18\x03 \x01(\x05\")\n\x06SortBy\x12\x11\n\tascending\x18\x01 \x01(\x08\x12\x0c\n\x04path\x18\x02 \x03(\t\"\xf4\x01\n\x0fMetadataRequest\x12\x0c\n\x04uuid\x18\x01 \x01(\x08\x12\x0e\n\x06vector\x18\x02 \x01(\x08\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x04 \x01(\x08\x12\x10\n\x08\x64istance\x18\x05 \x01(\x08\x12\x11\n\tcertainty\x18\x06 \x01(\x08\x12\r\n\x05score\x18\x07 \x01(\x08\x12\x15\n\rexplain_score\x18\x08 \x01(\x08\x12\x15\n\ris_consistent\x18\t \x01(\x08\x12\x0f\n\x07vectors\x18\n \x03(\t\x12\x15\n\rquery_profile\x18\x0b \x01(\x08\"\xd1\x01\n\x11PropertiesRequest\x12\x1a\n\x12non_ref_properties\x18\x01 \x03(\t\x12\x39\n\x0eref_properties\x18\x02 \x03(\x0b\x32!.weaviate.v1.RefPropertiesRequest\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\x12$\n\x1creturn_all_nonref_properties\x18\x0b \x01(\x08\"\x8b\x01\n\x17ObjectPropertiesRequest\x12\x11\n\tprop_name\x18\x01 \x01(\t\x12\x1c\n\x14primitive_properties\x18\x02 \x03(\t\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\"\xb1\x01\n\x14RefPropertiesRequest\x12\x1a\n\x12reference_property\x18\x01 \x01(\t\x12\x32\n\nproperties\x18\x02 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequest\x12.\n\x08metadata\x18\x03 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequest\x12\x19\n\x11target_collection\x18\x04 \x01(\t\"8\n\x06Rerank\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x12\n\x05query\x18\x02 \x01(\tH\x00\x88\x01\x01\x42\x08\n\x06_query\"\xf7\x02\n\x0bSearchReply\x12\x0c\n\x04took\x18\x01 \x01(\x02\x12*\n\x07results\x18\x02 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12*\n\x19generative_grouped_result\x18\x03 \x01(\tB\x02\x18\x01H\x00\x88\x01\x01\x12\x34\n\x10group_by_results\x18\x04 \x03(\x0b\x32\x1a.weaviate.v1.GroupByResult\x12\x46\n\x1agenerative_grouped_results\x18\x05 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x01\x88\x01\x01\x12\x35\n\rquery_profile\x18\x06 \x01(\x0b\x32\x19.weaviate.v1.QueryProfileH\x02\x88\x01\x01\x42\x1c\n\x1a_generative_grouped_resultB\x1d\n\x1b_generative_grouped_resultsB\x10\n\x0e_query_profile\"\x9e\x03\n\x0cQueryProfile\x12\x36\n\x06shards\x18\x01 \x03(\x0b\x32&.weaviate.v1.QueryProfile.ShardProfile\x1a\x86\x01\n\rSearchProfile\x12\x45\n\x07\x64\x65tails\x18\x01 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.SearchProfile.DetailsEntry\x1a.\n\x0c\x44\x65tailsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\xcc\x01\n\x0cShardProfile\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04node\x18\x02 \x01(\t\x12\x46\n\x08searches\x18\x03 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.ShardProfile.SearchesEntry\x1aX\n\rSearchesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x36\n\x05value\x18\x02 \x01(\x0b\x32\'.weaviate.v1.QueryProfile.SearchProfile:\x02\x38\x01\"\x1c\n\x0bRerankReply\x12\r\n\x05score\x18\x01 \x01(\x01\"\xe9\x02\n\rGroupByResult\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\x0cmin_distance\x18\x02 \x01(\x02\x12\x14\n\x0cmax_distance\x18\x03 \x01(\x02\x12\x19\n\x11number_of_objects\x18\x04 \x01(\x03\x12*\n\x07objects\x18\x05 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12-\n\x06rerank\x18\x06 \x01(\x0b\x32\x18.weaviate.v1.RerankReplyH\x00\x88\x01\x01\x12\x39\n\ngenerative\x18\x07 \x01(\x0b\x32\x1c.weaviate.v1.GenerativeReplyB\x02\x18\x01H\x01\x88\x01\x01\x12=\n\x11generative_result\x18\x08 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x02\x88\x01\x01\x42\t\n\x07_rerankB\r\n\x0b_generativeB\x14\n\x12_generative_result\"\xb7\x01\n\x0cSearchResult\x12\x31\n\nproperties\x18\x01 \x01(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12-\n\x08metadata\x18\x02 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12\x36\n\ngenerative\x18\x03 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x00\x88\x01\x01\x42\r\n\x0b_generative\"\xf7\x04\n\x0eMetadataResult\x12\n\n\x02id\x18\x01 \x01(\t\x12\x12\n\x06vector\x18\x02 \x03(\x02\x42\x02\x18\x01\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x03\x12\"\n\x1a\x63reation_time_unix_present\x18\x04 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x05 \x01(\x03\x12%\n\x1dlast_update_time_unix_present\x18\x06 \x01(\x08\x12\x10\n\x08\x64istance\x18\x07 \x01(\x02\x12\x18\n\x10\x64istance_present\x18\x08 \x01(\x08\x12\x11\n\tcertainty\x18\t \x01(\x02\x12\x19\n\x11\x63\x65rtainty_present\x18\n \x01(\x08\x12\r\n\x05score\x18\x0b \x01(\x02\x12\x15\n\rscore_present\x18\x0c \x01(\x08\x12\x15\n\rexplain_score\x18\r \x01(\t\x12\x1d\n\x15\x65xplain_score_present\x18\x0e \x01(\x08\x12\x1a\n\ris_consistent\x18\x0f \x01(\x08H\x00\x88\x01\x01\x12\x16\n\ngenerative\x18\x10 \x01(\tB\x02\x18\x01\x12\x1e\n\x12generative_present\x18\x11 \x01(\x08\x42\x02\x18\x01\x12\x1d\n\x15is_consistent_present\x18\x12 \x01(\x08\x12\x14\n\x0cvector_bytes\x18\x13 \x01(\x0c\x12\x13\n\x0bid_as_bytes\x18\x14 \x01(\x0c\x12\x14\n\x0crerank_score\x18\x15 \x01(\x01\x12\x1c\n\x14rerank_score_present\x18\x16 \x01(\x08\x12%\n\x07vectors\x18\x17 \x03(\x0b\x32\x14.weaviate.v1.VectorsB\x10\n\x0e_is_consistent\"\x88\x02\n\x10PropertiesResult\x12\x33\n\tref_props\x18\x02 \x03(\x0b\x32 .weaviate.v1.RefPropertiesResult\x12\x19\n\x11target_collection\x18\x03 \x01(\t\x12-\n\x08metadata\x18\x04 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12.\n\rnon_ref_props\x18\x0b \x01(\x0b\x32\x17.weaviate.v1.Properties\x12\x1b\n\x13ref_props_requested\x18\x0c \x01(\x08J\x04\x08\x01\x10\x02J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\nJ\x04\x08\n\x10\x0b\"[\n\x13RefPropertiesResult\x12\x31\n\nproperties\x18\x01 \x03(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12\x11\n\tprop_name\x18\x02 \x01(\t\"\xb3\t\n\x05\x42oost\x12\x30\n\nconditions\x18\x01 \x03(\x0b\x32\x1c.weaviate.v1.Boost.Condition\x12\x13\n\x06weight\x18\x02 \x01(\x02H\x00\x88\x01\x01\x12\x12\n\x05\x64\x65pth\x18\x03 \x01(\rH\x01\x88\x01\x01\x1aw\n\x15PropertyValueFunction\x12\x10\n\x08property\x18\x01 \x01(\t\x12?\n\x08modifier\x18\x02 \x01(\x0e\x32(.weaviate.v1.Boost.PropertyValueModifierH\x00\x88\x01\x01\x42\x0b\n\t_modifier\x1a\xcb\x01\n\x11TimeDecayFunction\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x0e\n\x06origin\x18\x02 \x01(\t\x12\r\n\x05scale\x18\x03 \x01(\t\x12\x13\n\x06offset\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x31\n\x05\x63urve\x18\x05 \x01(\x0e\x32\x1d.weaviate.v1.Boost.DecayCurveH\x01\x88\x01\x01\x12\x18\n\x0b\x64\x65\x63\x61y_value\x18\x06 \x01(\x02H\x02\x88\x01\x01\x42\t\n\x07_offsetB\x08\n\x06_curveB\x0e\n\x0c_decay_value\x1a\xce\x01\n\x14NumericDecayFunction\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x0e\n\x06origin\x18\x02 \x01(\x01\x12\r\n\x05scale\x18\x03 \x01(\x01\x12\x13\n\x06offset\x18\x04 \x01(\x01H\x00\x88\x01\x01\x12\x31\n\x05\x63urve\x18\x05 \x01(\x0e\x32\x1d.weaviate.v1.Boost.DecayCurveH\x01\x88\x01\x01\x12\x18\n\x0b\x64\x65\x63\x61y_value\x18\x06 \x01(\x02H\x02\x88\x01\x01\x42\t\n\x07_offsetB\x08\n\x06_curveB\x0e\n\x0c_decay_value\x1a\xa2\x02\n\tCondition\x12&\n\x06\x66ilter\x18\x01 \x01(\x0b\x32\x14.weaviate.v1.FiltersH\x00\x12:\n\ntime_decay\x18\x02 \x01(\x0b\x32$.weaviate.v1.Boost.TimeDecayFunctionH\x00\x12\x42\n\x0eproperty_value\x18\x03 \x01(\x0b\x32(.weaviate.v1.Boost.PropertyValueFunctionH\x00\x12@\n\rnumeric_decay\x18\x04 \x01(\x0b\x32\'.weaviate.v1.Boost.NumericDecayFunctionH\x00\x12\x13\n\x06weight\x18\x05 \x01(\x02H\x01\x88\x01\x01\x42\x0b\n\tconditionB\t\n\x07_weight\"\x85\x01\n\x15PropertyValueModifier\x12\'\n#PROPERTY_VALUE_MODIFIER_UNSPECIFIED\x10\x00\x12!\n\x1dPROPERTY_VALUE_MODIFIER_LOG1P\x10\x01\x12 \n\x1cPROPERTY_VALUE_MODIFIER_SQRT\x10\x02\"u\n\nDecayCurve\x12\x1b\n\x17\x44\x45\x43\x41Y_CURVE_UNSPECIFIED\x10\x00\x12\x15\n\x11\x44\x45\x43\x41Y_CURVE_GAUSS\x10\x01\x12\x16\n\x12\x44\x45\x43\x41Y_CURVE_LINEAR\x10\x02\x12\x1b\n\x17\x44\x45\x43\x41Y_CURVE_EXPONENTIAL\x10\x03\x42\t\n\x07_weightB\x08\n\x06_depthBs\n#io.weaviate.client.grpc.protocol.v1B\x16WeaviateProtoSearchGetZ4github.com/weaviate/weaviate/grpc/generated;protocolb\x06proto3') _globals = globals() _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals) @@ -45,43 +45,57 @@ _globals['_METADATARESULT'].fields_by_name['generative_present']._loaded_options = None _globals['_METADATARESULT'].fields_by_name['generative_present']._serialized_options = b'\030\001' _globals['_SEARCHREQUEST']._serialized_start=116 - _globals['_SEARCHREQUEST']._serialized_end=1552 - _globals['_GROUPBY']._serialized_start=1554 - _globals['_GROUPBY']._serialized_end=1630 - _globals['_SORTBY']._serialized_start=1632 - _globals['_SORTBY']._serialized_end=1673 - _globals['_METADATAREQUEST']._serialized_start=1676 - _globals['_METADATAREQUEST']._serialized_end=1920 - _globals['_PROPERTIESREQUEST']._serialized_start=1923 - _globals['_PROPERTIESREQUEST']._serialized_end=2132 - _globals['_OBJECTPROPERTIESREQUEST']._serialized_start=2135 - _globals['_OBJECTPROPERTIESREQUEST']._serialized_end=2274 - _globals['_REFPROPERTIESREQUEST']._serialized_start=2277 - _globals['_REFPROPERTIESREQUEST']._serialized_end=2454 - _globals['_RERANK']._serialized_start=2456 - _globals['_RERANK']._serialized_end=2512 - _globals['_SEARCHREPLY']._serialized_start=2515 - _globals['_SEARCHREPLY']._serialized_end=2890 - _globals['_QUERYPROFILE']._serialized_start=2893 - _globals['_QUERYPROFILE']._serialized_end=3307 - _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_start=2966 - _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_end=3100 - _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_start=3054 - _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_end=3100 - _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_start=3103 - _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_end=3307 - _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_start=3219 - _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_end=3307 - _globals['_RERANKREPLY']._serialized_start=3309 - _globals['_RERANKREPLY']._serialized_end=3337 - _globals['_GROUPBYRESULT']._serialized_start=3340 - _globals['_GROUPBYRESULT']._serialized_end=3701 - _globals['_SEARCHRESULT']._serialized_start=3704 - _globals['_SEARCHRESULT']._serialized_end=3887 - _globals['_METADATARESULT']._serialized_start=3890 - _globals['_METADATARESULT']._serialized_end=4521 - _globals['_PROPERTIESRESULT']._serialized_start=4524 - _globals['_PROPERTIESRESULT']._serialized_end=4788 - _globals['_REFPROPERTIESRESULT']._serialized_start=4790 - _globals['_REFPROPERTIESRESULT']._serialized_end=4881 + _globals['_SEARCHREQUEST']._serialized_end=1602 + _globals['_GROUPBY']._serialized_start=1604 + _globals['_GROUPBY']._serialized_end=1680 + _globals['_SORTBY']._serialized_start=1682 + _globals['_SORTBY']._serialized_end=1723 + _globals['_METADATAREQUEST']._serialized_start=1726 + _globals['_METADATAREQUEST']._serialized_end=1970 + _globals['_PROPERTIESREQUEST']._serialized_start=1973 + _globals['_PROPERTIESREQUEST']._serialized_end=2182 + _globals['_OBJECTPROPERTIESREQUEST']._serialized_start=2185 + _globals['_OBJECTPROPERTIESREQUEST']._serialized_end=2324 + _globals['_REFPROPERTIESREQUEST']._serialized_start=2327 + _globals['_REFPROPERTIESREQUEST']._serialized_end=2504 + _globals['_RERANK']._serialized_start=2506 + _globals['_RERANK']._serialized_end=2562 + _globals['_SEARCHREPLY']._serialized_start=2565 + _globals['_SEARCHREPLY']._serialized_end=2940 + _globals['_QUERYPROFILE']._serialized_start=2943 + _globals['_QUERYPROFILE']._serialized_end=3357 + _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_start=3016 + _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_end=3150 + _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_start=3104 + _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_end=3150 + _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_start=3153 + _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_end=3357 + _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_start=3269 + _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_end=3357 + _globals['_RERANKREPLY']._serialized_start=3359 + _globals['_RERANKREPLY']._serialized_end=3387 + _globals['_GROUPBYRESULT']._serialized_start=3390 + _globals['_GROUPBYRESULT']._serialized_end=3751 + _globals['_SEARCHRESULT']._serialized_start=3754 + _globals['_SEARCHRESULT']._serialized_end=3937 + _globals['_METADATARESULT']._serialized_start=3940 + _globals['_METADATARESULT']._serialized_end=4571 + _globals['_PROPERTIESRESULT']._serialized_start=4574 + _globals['_PROPERTIESRESULT']._serialized_end=4838 + _globals['_REFPROPERTIESRESULT']._serialized_start=4840 + _globals['_REFPROPERTIESRESULT']._serialized_end=4931 + _globals['_BOOST']._serialized_start=4934 + _globals['_BOOST']._serialized_end=6137 + _globals['_BOOST_PROPERTYVALUEFUNCTION']._serialized_start=5034 + _globals['_BOOST_PROPERTYVALUEFUNCTION']._serialized_end=5153 + _globals['_BOOST_TIMEDECAYFUNCTION']._serialized_start=5156 + _globals['_BOOST_TIMEDECAYFUNCTION']._serialized_end=5359 + _globals['_BOOST_NUMERICDECAYFUNCTION']._serialized_start=5362 + _globals['_BOOST_NUMERICDECAYFUNCTION']._serialized_end=5568 + _globals['_BOOST_CONDITION']._serialized_start=5571 + _globals['_BOOST_CONDITION']._serialized_end=5861 + _globals['_BOOST_PROPERTYVALUEMODIFIER']._serialized_start=5864 + _globals['_BOOST_PROPERTYVALUEMODIFIER']._serialized_end=5997 + _globals['_BOOST_DECAYCURVE']._serialized_start=5999 + _globals['_BOOST_DECAYCURVE']._serialized_end=6116 # @@protoc_insertion_point(module_scope) diff --git a/weaviate/proto/v1/v5261/v1/search_get_pb2.pyi b/weaviate/proto/v1/v5261/v1/search_get_pb2.pyi index 783ca3564..5e5c6f9f4 100644 --- a/weaviate/proto/v1/v5261/v1/search_get_pb2.pyi +++ b/weaviate/proto/v1/v5261/v1/search_get_pb2.pyi @@ -3,6 +3,7 @@ from weaviate.proto.v1.v5261.v1 import base_search_pb2 as _base_search_pb2 from weaviate.proto.v1.v5261.v1 import generative_pb2 as _generative_pb2 from weaviate.proto.v1.v5261.v1 import properties_pb2 as _properties_pb2 from google.protobuf.internal import containers as _containers +from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper from google.protobuf import descriptor as _descriptor from google.protobuf import message as _message from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union @@ -10,7 +11,7 @@ from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Map DESCRIPTOR: _descriptor.FileDescriptor class SearchRequest(_message.Message): - __slots__ = ("collection", "tenant", "consistency_level", "properties", "metadata", "group_by", "limit", "offset", "autocut", "after", "sort_by", "filters", "hybrid_search", "bm25_search", "near_vector", "near_object", "near_text", "near_image", "near_audio", "near_video", "near_depth", "near_thermal", "near_imu", "generative", "rerank", "uses_123_api", "uses_125_api", "uses_127_api") + __slots__ = ("collection", "tenant", "consistency_level", "properties", "metadata", "group_by", "limit", "offset", "autocut", "after", "sort_by", "filters", "hybrid_search", "bm25_search", "near_vector", "near_object", "near_text", "near_image", "near_audio", "near_video", "near_depth", "near_thermal", "near_imu", "generative", "rerank", "boost", "uses_123_api", "uses_125_api", "uses_127_api") COLLECTION_FIELD_NUMBER: _ClassVar[int] TENANT_FIELD_NUMBER: _ClassVar[int] CONSISTENCY_LEVEL_FIELD_NUMBER: _ClassVar[int] @@ -36,6 +37,7 @@ class SearchRequest(_message.Message): NEAR_IMU_FIELD_NUMBER: _ClassVar[int] GENERATIVE_FIELD_NUMBER: _ClassVar[int] RERANK_FIELD_NUMBER: _ClassVar[int] + BOOST_FIELD_NUMBER: _ClassVar[int] USES_123_API_FIELD_NUMBER: _ClassVar[int] USES_125_API_FIELD_NUMBER: _ClassVar[int] USES_127_API_FIELD_NUMBER: _ClassVar[int] @@ -64,10 +66,11 @@ class SearchRequest(_message.Message): near_imu: _base_search_pb2.NearIMUSearch generative: _generative_pb2.GenerativeSearch rerank: Rerank + boost: Boost uses_123_api: bool uses_125_api: bool uses_127_api: bool - def __init__(self, collection: _Optional[str] = ..., tenant: _Optional[str] = ..., consistency_level: _Optional[_Union[_base_pb2.ConsistencyLevel, str]] = ..., properties: _Optional[_Union[PropertiesRequest, _Mapping]] = ..., metadata: _Optional[_Union[MetadataRequest, _Mapping]] = ..., group_by: _Optional[_Union[GroupBy, _Mapping]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., autocut: _Optional[int] = ..., after: _Optional[str] = ..., sort_by: _Optional[_Iterable[_Union[SortBy, _Mapping]]] = ..., filters: _Optional[_Union[_base_pb2.Filters, _Mapping]] = ..., hybrid_search: _Optional[_Union[_base_search_pb2.Hybrid, _Mapping]] = ..., bm25_search: _Optional[_Union[_base_search_pb2.BM25, _Mapping]] = ..., near_vector: _Optional[_Union[_base_search_pb2.NearVector, _Mapping]] = ..., near_object: _Optional[_Union[_base_search_pb2.NearObject, _Mapping]] = ..., near_text: _Optional[_Union[_base_search_pb2.NearTextSearch, _Mapping]] = ..., near_image: _Optional[_Union[_base_search_pb2.NearImageSearch, _Mapping]] = ..., near_audio: _Optional[_Union[_base_search_pb2.NearAudioSearch, _Mapping]] = ..., near_video: _Optional[_Union[_base_search_pb2.NearVideoSearch, _Mapping]] = ..., near_depth: _Optional[_Union[_base_search_pb2.NearDepthSearch, _Mapping]] = ..., near_thermal: _Optional[_Union[_base_search_pb2.NearThermalSearch, _Mapping]] = ..., near_imu: _Optional[_Union[_base_search_pb2.NearIMUSearch, _Mapping]] = ..., generative: _Optional[_Union[_generative_pb2.GenerativeSearch, _Mapping]] = ..., rerank: _Optional[_Union[Rerank, _Mapping]] = ..., uses_123_api: bool = ..., uses_125_api: bool = ..., uses_127_api: bool = ...) -> None: ... + def __init__(self, collection: _Optional[str] = ..., tenant: _Optional[str] = ..., consistency_level: _Optional[_Union[_base_pb2.ConsistencyLevel, str]] = ..., properties: _Optional[_Union[PropertiesRequest, _Mapping]] = ..., metadata: _Optional[_Union[MetadataRequest, _Mapping]] = ..., group_by: _Optional[_Union[GroupBy, _Mapping]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., autocut: _Optional[int] = ..., after: _Optional[str] = ..., sort_by: _Optional[_Iterable[_Union[SortBy, _Mapping]]] = ..., filters: _Optional[_Union[_base_pb2.Filters, _Mapping]] = ..., hybrid_search: _Optional[_Union[_base_search_pb2.Hybrid, _Mapping]] = ..., bm25_search: _Optional[_Union[_base_search_pb2.BM25, _Mapping]] = ..., near_vector: _Optional[_Union[_base_search_pb2.NearVector, _Mapping]] = ..., near_object: _Optional[_Union[_base_search_pb2.NearObject, _Mapping]] = ..., near_text: _Optional[_Union[_base_search_pb2.NearTextSearch, _Mapping]] = ..., near_image: _Optional[_Union[_base_search_pb2.NearImageSearch, _Mapping]] = ..., near_audio: _Optional[_Union[_base_search_pb2.NearAudioSearch, _Mapping]] = ..., near_video: _Optional[_Union[_base_search_pb2.NearVideoSearch, _Mapping]] = ..., near_depth: _Optional[_Union[_base_search_pb2.NearDepthSearch, _Mapping]] = ..., near_thermal: _Optional[_Union[_base_search_pb2.NearThermalSearch, _Mapping]] = ..., near_imu: _Optional[_Union[_base_search_pb2.NearIMUSearch, _Mapping]] = ..., generative: _Optional[_Union[_generative_pb2.GenerativeSearch, _Mapping]] = ..., rerank: _Optional[_Union[Rerank, _Mapping]] = ..., boost: _Optional[_Union[Boost, _Mapping]] = ..., uses_123_api: bool = ..., uses_125_api: bool = ..., uses_127_api: bool = ...) -> None: ... class GroupBy(_message.Message): __slots__ = ("path", "number_of_groups", "objects_per_group") @@ -312,3 +315,81 @@ class RefPropertiesResult(_message.Message): properties: _containers.RepeatedCompositeFieldContainer[PropertiesResult] prop_name: str def __init__(self, properties: _Optional[_Iterable[_Union[PropertiesResult, _Mapping]]] = ..., prop_name: _Optional[str] = ...) -> None: ... + +class Boost(_message.Message): + __slots__ = ("conditions", "weight", "depth") + class PropertyValueModifier(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): + __slots__ = () + PROPERTY_VALUE_MODIFIER_UNSPECIFIED: _ClassVar[Boost.PropertyValueModifier] + PROPERTY_VALUE_MODIFIER_LOG1P: _ClassVar[Boost.PropertyValueModifier] + PROPERTY_VALUE_MODIFIER_SQRT: _ClassVar[Boost.PropertyValueModifier] + PROPERTY_VALUE_MODIFIER_UNSPECIFIED: Boost.PropertyValueModifier + PROPERTY_VALUE_MODIFIER_LOG1P: Boost.PropertyValueModifier + PROPERTY_VALUE_MODIFIER_SQRT: Boost.PropertyValueModifier + class DecayCurve(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): + __slots__ = () + DECAY_CURVE_UNSPECIFIED: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_GAUSS: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_LINEAR: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_EXPONENTIAL: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_UNSPECIFIED: Boost.DecayCurve + DECAY_CURVE_GAUSS: Boost.DecayCurve + DECAY_CURVE_LINEAR: Boost.DecayCurve + DECAY_CURVE_EXPONENTIAL: Boost.DecayCurve + class PropertyValueFunction(_message.Message): + __slots__ = ("property", "modifier") + PROPERTY_FIELD_NUMBER: _ClassVar[int] + MODIFIER_FIELD_NUMBER: _ClassVar[int] + property: str + modifier: Boost.PropertyValueModifier + def __init__(self, property: _Optional[str] = ..., modifier: _Optional[_Union[Boost.PropertyValueModifier, str]] = ...) -> None: ... + class TimeDecayFunction(_message.Message): + __slots__ = ("property", "origin", "scale", "offset", "curve", "decay_value") + PROPERTY_FIELD_NUMBER: _ClassVar[int] + ORIGIN_FIELD_NUMBER: _ClassVar[int] + SCALE_FIELD_NUMBER: _ClassVar[int] + OFFSET_FIELD_NUMBER: _ClassVar[int] + CURVE_FIELD_NUMBER: _ClassVar[int] + DECAY_VALUE_FIELD_NUMBER: _ClassVar[int] + property: str + origin: str + scale: str + offset: str + curve: Boost.DecayCurve + decay_value: float + def __init__(self, property: _Optional[str] = ..., origin: _Optional[str] = ..., scale: _Optional[str] = ..., offset: _Optional[str] = ..., curve: _Optional[_Union[Boost.DecayCurve, str]] = ..., decay_value: _Optional[float] = ...) -> None: ... + class NumericDecayFunction(_message.Message): + __slots__ = ("property", "origin", "scale", "offset", "curve", "decay_value") + PROPERTY_FIELD_NUMBER: _ClassVar[int] + ORIGIN_FIELD_NUMBER: _ClassVar[int] + SCALE_FIELD_NUMBER: _ClassVar[int] + OFFSET_FIELD_NUMBER: _ClassVar[int] + CURVE_FIELD_NUMBER: _ClassVar[int] + DECAY_VALUE_FIELD_NUMBER: _ClassVar[int] + property: str + origin: float + scale: float + offset: float + curve: Boost.DecayCurve + decay_value: float + def __init__(self, property: _Optional[str] = ..., origin: _Optional[float] = ..., scale: _Optional[float] = ..., offset: _Optional[float] = ..., curve: _Optional[_Union[Boost.DecayCurve, str]] = ..., decay_value: _Optional[float] = ...) -> None: ... + class Condition(_message.Message): + __slots__ = ("filter", "time_decay", "property_value", "numeric_decay", "weight") + FILTER_FIELD_NUMBER: _ClassVar[int] + TIME_DECAY_FIELD_NUMBER: _ClassVar[int] + PROPERTY_VALUE_FIELD_NUMBER: _ClassVar[int] + NUMERIC_DECAY_FIELD_NUMBER: _ClassVar[int] + WEIGHT_FIELD_NUMBER: _ClassVar[int] + filter: _base_pb2.Filters + time_decay: Boost.TimeDecayFunction + property_value: Boost.PropertyValueFunction + numeric_decay: Boost.NumericDecayFunction + weight: float + def __init__(self, filter: _Optional[_Union[_base_pb2.Filters, _Mapping]] = ..., time_decay: _Optional[_Union[Boost.TimeDecayFunction, _Mapping]] = ..., property_value: _Optional[_Union[Boost.PropertyValueFunction, _Mapping]] = ..., numeric_decay: _Optional[_Union[Boost.NumericDecayFunction, _Mapping]] = ..., weight: _Optional[float] = ...) -> None: ... + CONDITIONS_FIELD_NUMBER: _ClassVar[int] + WEIGHT_FIELD_NUMBER: _ClassVar[int] + DEPTH_FIELD_NUMBER: _ClassVar[int] + conditions: _containers.RepeatedCompositeFieldContainer[Boost.Condition] + weight: float + depth: int + def __init__(self, conditions: _Optional[_Iterable[_Union[Boost.Condition, _Mapping]]] = ..., weight: _Optional[float] = ..., depth: _Optional[int] = ...) -> None: ... diff --git a/weaviate/proto/v1/v6300/v1/search_get_pb2.py b/weaviate/proto/v1/v6300/v1/search_get_pb2.py index 1b38e4fb6..8ec99714c 100644 --- a/weaviate/proto/v1/v6300/v1/search_get_pb2.py +++ b/weaviate/proto/v1/v6300/v1/search_get_pb2.py @@ -28,7 +28,7 @@ from weaviate.proto.v1.v6300.v1 import properties_pb2 as v1_dot_properties__pb2 -DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13v1/search_get.proto\x12\x0bweaviate.v1\x1a\rv1/base.proto\x1a\x14v1/base_search.proto\x1a\x13v1/generative.proto\x1a\x13v1/properties.proto\"\x9c\x0b\n\rSearchRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\x0e\n\x06tenant\x18\n \x01(\t\x12=\n\x11\x63onsistency_level\x18\x0b \x01(\x0e\x32\x1d.weaviate.v1.ConsistencyLevelH\x00\x88\x01\x01\x12\x37\n\nproperties\x18\x14 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequestH\x01\x88\x01\x01\x12\x33\n\x08metadata\x18\x15 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequestH\x02\x88\x01\x01\x12+\n\x08group_by\x18\x16 \x01(\x0b\x32\x14.weaviate.v1.GroupByH\x03\x88\x01\x01\x12\r\n\x05limit\x18\x1e \x01(\r\x12\x0e\n\x06offset\x18\x1f \x01(\r\x12\x0f\n\x07\x61utocut\x18 \x01(\r\x12\r\n\x05\x61\x66ter\x18! \x01(\t\x12$\n\x07sort_by\x18\" \x03(\x0b\x32\x13.weaviate.v1.SortBy\x12*\n\x07\x66ilters\x18( \x01(\x0b\x32\x14.weaviate.v1.FiltersH\x04\x88\x01\x01\x12/\n\rhybrid_search\x18) \x01(\x0b\x32\x13.weaviate.v1.HybridH\x05\x88\x01\x01\x12+\n\x0b\x62m25_search\x18* \x01(\x0b\x32\x11.weaviate.v1.BM25H\x06\x88\x01\x01\x12\x31\n\x0bnear_vector\x18+ \x01(\x0b\x32\x17.weaviate.v1.NearVectorH\x07\x88\x01\x01\x12\x31\n\x0bnear_object\x18, \x01(\x0b\x32\x17.weaviate.v1.NearObjectH\x08\x88\x01\x01\x12\x33\n\tnear_text\x18- \x01(\x0b\x32\x1b.weaviate.v1.NearTextSearchH\t\x88\x01\x01\x12\x35\n\nnear_image\x18. \x01(\x0b\x32\x1c.weaviate.v1.NearImageSearchH\n\x88\x01\x01\x12\x35\n\nnear_audio\x18/ \x01(\x0b\x32\x1c.weaviate.v1.NearAudioSearchH\x0b\x88\x01\x01\x12\x35\n\nnear_video\x18\x30 \x01(\x0b\x32\x1c.weaviate.v1.NearVideoSearchH\x0c\x88\x01\x01\x12\x35\n\nnear_depth\x18\x31 \x01(\x0b\x32\x1c.weaviate.v1.NearDepthSearchH\r\x88\x01\x01\x12\x39\n\x0cnear_thermal\x18\x32 \x01(\x0b\x32\x1e.weaviate.v1.NearThermalSearchH\x0e\x88\x01\x01\x12\x31\n\x08near_imu\x18\x33 \x01(\x0b\x32\x1a.weaviate.v1.NearIMUSearchH\x0f\x88\x01\x01\x12\x36\n\ngenerative\x18< \x01(\x0b\x32\x1d.weaviate.v1.GenerativeSearchH\x10\x88\x01\x01\x12(\n\x06rerank\x18= \x01(\x0b\x32\x13.weaviate.v1.RerankH\x11\x88\x01\x01\x12\x18\n\x0cuses_123_api\x18\x64 \x01(\x08\x42\x02\x18\x01\x12\x18\n\x0cuses_125_api\x18\x65 \x01(\x08\x42\x02\x18\x01\x12\x14\n\x0cuses_127_api\x18\x66 \x01(\x08\x42\x14\n\x12_consistency_levelB\r\n\x0b_propertiesB\x0b\n\t_metadataB\x0b\n\t_group_byB\n\n\x08_filtersB\x10\n\x0e_hybrid_searchB\x0e\n\x0c_bm25_searchB\x0e\n\x0c_near_vectorB\x0e\n\x0c_near_objectB\x0c\n\n_near_textB\r\n\x0b_near_imageB\r\n\x0b_near_audioB\r\n\x0b_near_videoB\r\n\x0b_near_depthB\x0f\n\r_near_thermalB\x0b\n\t_near_imuB\r\n\x0b_generativeB\t\n\x07_rerank\"L\n\x07GroupBy\x12\x0c\n\x04path\x18\x01 \x03(\t\x12\x18\n\x10number_of_groups\x18\x02 \x01(\x05\x12\x19\n\x11objects_per_group\x18\x03 \x01(\x05\")\n\x06SortBy\x12\x11\n\tascending\x18\x01 \x01(\x08\x12\x0c\n\x04path\x18\x02 \x03(\t\"\xf4\x01\n\x0fMetadataRequest\x12\x0c\n\x04uuid\x18\x01 \x01(\x08\x12\x0e\n\x06vector\x18\x02 \x01(\x08\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x04 \x01(\x08\x12\x10\n\x08\x64istance\x18\x05 \x01(\x08\x12\x11\n\tcertainty\x18\x06 \x01(\x08\x12\r\n\x05score\x18\x07 \x01(\x08\x12\x15\n\rexplain_score\x18\x08 \x01(\x08\x12\x15\n\ris_consistent\x18\t \x01(\x08\x12\x0f\n\x07vectors\x18\n \x03(\t\x12\x15\n\rquery_profile\x18\x0b \x01(\x08\"\xd1\x01\n\x11PropertiesRequest\x12\x1a\n\x12non_ref_properties\x18\x01 \x03(\t\x12\x39\n\x0eref_properties\x18\x02 \x03(\x0b\x32!.weaviate.v1.RefPropertiesRequest\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\x12$\n\x1creturn_all_nonref_properties\x18\x0b \x01(\x08\"\x8b\x01\n\x17ObjectPropertiesRequest\x12\x11\n\tprop_name\x18\x01 \x01(\t\x12\x1c\n\x14primitive_properties\x18\x02 \x03(\t\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\"\xb1\x01\n\x14RefPropertiesRequest\x12\x1a\n\x12reference_property\x18\x01 \x01(\t\x12\x32\n\nproperties\x18\x02 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequest\x12.\n\x08metadata\x18\x03 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequest\x12\x19\n\x11target_collection\x18\x04 \x01(\t\"8\n\x06Rerank\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x12\n\x05query\x18\x02 \x01(\tH\x00\x88\x01\x01\x42\x08\n\x06_query\"\xf7\x02\n\x0bSearchReply\x12\x0c\n\x04took\x18\x01 \x01(\x02\x12*\n\x07results\x18\x02 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12*\n\x19generative_grouped_result\x18\x03 \x01(\tB\x02\x18\x01H\x00\x88\x01\x01\x12\x34\n\x10group_by_results\x18\x04 \x03(\x0b\x32\x1a.weaviate.v1.GroupByResult\x12\x46\n\x1agenerative_grouped_results\x18\x05 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x01\x88\x01\x01\x12\x35\n\rquery_profile\x18\x06 \x01(\x0b\x32\x19.weaviate.v1.QueryProfileH\x02\x88\x01\x01\x42\x1c\n\x1a_generative_grouped_resultB\x1d\n\x1b_generative_grouped_resultsB\x10\n\x0e_query_profile\"\x9e\x03\n\x0cQueryProfile\x12\x36\n\x06shards\x18\x01 \x03(\x0b\x32&.weaviate.v1.QueryProfile.ShardProfile\x1a\x86\x01\n\rSearchProfile\x12\x45\n\x07\x64\x65tails\x18\x01 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.SearchProfile.DetailsEntry\x1a.\n\x0c\x44\x65tailsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\xcc\x01\n\x0cShardProfile\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04node\x18\x02 \x01(\t\x12\x46\n\x08searches\x18\x03 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.ShardProfile.SearchesEntry\x1aX\n\rSearchesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x36\n\x05value\x18\x02 \x01(\x0b\x32\'.weaviate.v1.QueryProfile.SearchProfile:\x02\x38\x01\"\x1c\n\x0bRerankReply\x12\r\n\x05score\x18\x01 \x01(\x01\"\xe9\x02\n\rGroupByResult\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\x0cmin_distance\x18\x02 \x01(\x02\x12\x14\n\x0cmax_distance\x18\x03 \x01(\x02\x12\x19\n\x11number_of_objects\x18\x04 \x01(\x03\x12*\n\x07objects\x18\x05 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12-\n\x06rerank\x18\x06 \x01(\x0b\x32\x18.weaviate.v1.RerankReplyH\x00\x88\x01\x01\x12\x39\n\ngenerative\x18\x07 \x01(\x0b\x32\x1c.weaviate.v1.GenerativeReplyB\x02\x18\x01H\x01\x88\x01\x01\x12=\n\x11generative_result\x18\x08 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x02\x88\x01\x01\x42\t\n\x07_rerankB\r\n\x0b_generativeB\x14\n\x12_generative_result\"\xb7\x01\n\x0cSearchResult\x12\x31\n\nproperties\x18\x01 \x01(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12-\n\x08metadata\x18\x02 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12\x36\n\ngenerative\x18\x03 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x00\x88\x01\x01\x42\r\n\x0b_generative\"\xf7\x04\n\x0eMetadataResult\x12\n\n\x02id\x18\x01 \x01(\t\x12\x12\n\x06vector\x18\x02 \x03(\x02\x42\x02\x18\x01\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x03\x12\"\n\x1a\x63reation_time_unix_present\x18\x04 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x05 \x01(\x03\x12%\n\x1dlast_update_time_unix_present\x18\x06 \x01(\x08\x12\x10\n\x08\x64istance\x18\x07 \x01(\x02\x12\x18\n\x10\x64istance_present\x18\x08 \x01(\x08\x12\x11\n\tcertainty\x18\t \x01(\x02\x12\x19\n\x11\x63\x65rtainty_present\x18\n \x01(\x08\x12\r\n\x05score\x18\x0b \x01(\x02\x12\x15\n\rscore_present\x18\x0c \x01(\x08\x12\x15\n\rexplain_score\x18\r \x01(\t\x12\x1d\n\x15\x65xplain_score_present\x18\x0e \x01(\x08\x12\x1a\n\ris_consistent\x18\x0f \x01(\x08H\x00\x88\x01\x01\x12\x16\n\ngenerative\x18\x10 \x01(\tB\x02\x18\x01\x12\x1e\n\x12generative_present\x18\x11 \x01(\x08\x42\x02\x18\x01\x12\x1d\n\x15is_consistent_present\x18\x12 \x01(\x08\x12\x14\n\x0cvector_bytes\x18\x13 \x01(\x0c\x12\x13\n\x0bid_as_bytes\x18\x14 \x01(\x0c\x12\x14\n\x0crerank_score\x18\x15 \x01(\x01\x12\x1c\n\x14rerank_score_present\x18\x16 \x01(\x08\x12%\n\x07vectors\x18\x17 \x03(\x0b\x32\x14.weaviate.v1.VectorsB\x10\n\x0e_is_consistent\"\x88\x02\n\x10PropertiesResult\x12\x33\n\tref_props\x18\x02 \x03(\x0b\x32 .weaviate.v1.RefPropertiesResult\x12\x19\n\x11target_collection\x18\x03 \x01(\t\x12-\n\x08metadata\x18\x04 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12.\n\rnon_ref_props\x18\x0b \x01(\x0b\x32\x17.weaviate.v1.Properties\x12\x1b\n\x13ref_props_requested\x18\x0c \x01(\x08J\x04\x08\x01\x10\x02J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\nJ\x04\x08\n\x10\x0b\"[\n\x13RefPropertiesResult\x12\x31\n\nproperties\x18\x01 \x03(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12\x11\n\tprop_name\x18\x02 \x01(\tBs\n#io.weaviate.client.grpc.protocol.v1B\x16WeaviateProtoSearchGetZ4github.com/weaviate/weaviate/grpc/generated;protocolb\x06proto3') +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x13v1/search_get.proto\x12\x0bweaviate.v1\x1a\rv1/base.proto\x1a\x14v1/base_search.proto\x1a\x13v1/generative.proto\x1a\x13v1/properties.proto\"\xce\x0b\n\rSearchRequest\x12\x12\n\ncollection\x18\x01 \x01(\t\x12\x0e\n\x06tenant\x18\n \x01(\t\x12=\n\x11\x63onsistency_level\x18\x0b \x01(\x0e\x32\x1d.weaviate.v1.ConsistencyLevelH\x00\x88\x01\x01\x12\x37\n\nproperties\x18\x14 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequestH\x01\x88\x01\x01\x12\x33\n\x08metadata\x18\x15 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequestH\x02\x88\x01\x01\x12+\n\x08group_by\x18\x16 \x01(\x0b\x32\x14.weaviate.v1.GroupByH\x03\x88\x01\x01\x12\r\n\x05limit\x18\x1e \x01(\r\x12\x0e\n\x06offset\x18\x1f \x01(\r\x12\x0f\n\x07\x61utocut\x18 \x01(\r\x12\r\n\x05\x61\x66ter\x18! \x01(\t\x12$\n\x07sort_by\x18\" \x03(\x0b\x32\x13.weaviate.v1.SortBy\x12*\n\x07\x66ilters\x18( \x01(\x0b\x32\x14.weaviate.v1.FiltersH\x04\x88\x01\x01\x12/\n\rhybrid_search\x18) \x01(\x0b\x32\x13.weaviate.v1.HybridH\x05\x88\x01\x01\x12+\n\x0b\x62m25_search\x18* \x01(\x0b\x32\x11.weaviate.v1.BM25H\x06\x88\x01\x01\x12\x31\n\x0bnear_vector\x18+ \x01(\x0b\x32\x17.weaviate.v1.NearVectorH\x07\x88\x01\x01\x12\x31\n\x0bnear_object\x18, \x01(\x0b\x32\x17.weaviate.v1.NearObjectH\x08\x88\x01\x01\x12\x33\n\tnear_text\x18- \x01(\x0b\x32\x1b.weaviate.v1.NearTextSearchH\t\x88\x01\x01\x12\x35\n\nnear_image\x18. \x01(\x0b\x32\x1c.weaviate.v1.NearImageSearchH\n\x88\x01\x01\x12\x35\n\nnear_audio\x18/ \x01(\x0b\x32\x1c.weaviate.v1.NearAudioSearchH\x0b\x88\x01\x01\x12\x35\n\nnear_video\x18\x30 \x01(\x0b\x32\x1c.weaviate.v1.NearVideoSearchH\x0c\x88\x01\x01\x12\x35\n\nnear_depth\x18\x31 \x01(\x0b\x32\x1c.weaviate.v1.NearDepthSearchH\r\x88\x01\x01\x12\x39\n\x0cnear_thermal\x18\x32 \x01(\x0b\x32\x1e.weaviate.v1.NearThermalSearchH\x0e\x88\x01\x01\x12\x31\n\x08near_imu\x18\x33 \x01(\x0b\x32\x1a.weaviate.v1.NearIMUSearchH\x0f\x88\x01\x01\x12\x36\n\ngenerative\x18< \x01(\x0b\x32\x1d.weaviate.v1.GenerativeSearchH\x10\x88\x01\x01\x12(\n\x06rerank\x18= \x01(\x0b\x32\x13.weaviate.v1.RerankH\x11\x88\x01\x01\x12&\n\x05\x62oost\x18> \x01(\x0b\x32\x12.weaviate.v1.BoostH\x12\x88\x01\x01\x12\x18\n\x0cuses_123_api\x18\x64 \x01(\x08\x42\x02\x18\x01\x12\x18\n\x0cuses_125_api\x18\x65 \x01(\x08\x42\x02\x18\x01\x12\x14\n\x0cuses_127_api\x18\x66 \x01(\x08\x42\x14\n\x12_consistency_levelB\r\n\x0b_propertiesB\x0b\n\t_metadataB\x0b\n\t_group_byB\n\n\x08_filtersB\x10\n\x0e_hybrid_searchB\x0e\n\x0c_bm25_searchB\x0e\n\x0c_near_vectorB\x0e\n\x0c_near_objectB\x0c\n\n_near_textB\r\n\x0b_near_imageB\r\n\x0b_near_audioB\r\n\x0b_near_videoB\r\n\x0b_near_depthB\x0f\n\r_near_thermalB\x0b\n\t_near_imuB\r\n\x0b_generativeB\t\n\x07_rerankB\x08\n\x06_boost\"L\n\x07GroupBy\x12\x0c\n\x04path\x18\x01 \x03(\t\x12\x18\n\x10number_of_groups\x18\x02 \x01(\x05\x12\x19\n\x11objects_per_group\x18\x03 \x01(\x05\")\n\x06SortBy\x12\x11\n\tascending\x18\x01 \x01(\x08\x12\x0c\n\x04path\x18\x02 \x03(\t\"\xf4\x01\n\x0fMetadataRequest\x12\x0c\n\x04uuid\x18\x01 \x01(\x08\x12\x0e\n\x06vector\x18\x02 \x01(\x08\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x04 \x01(\x08\x12\x10\n\x08\x64istance\x18\x05 \x01(\x08\x12\x11\n\tcertainty\x18\x06 \x01(\x08\x12\r\n\x05score\x18\x07 \x01(\x08\x12\x15\n\rexplain_score\x18\x08 \x01(\x08\x12\x15\n\ris_consistent\x18\t \x01(\x08\x12\x0f\n\x07vectors\x18\n \x03(\t\x12\x15\n\rquery_profile\x18\x0b \x01(\x08\"\xd1\x01\n\x11PropertiesRequest\x12\x1a\n\x12non_ref_properties\x18\x01 \x03(\t\x12\x39\n\x0eref_properties\x18\x02 \x03(\x0b\x32!.weaviate.v1.RefPropertiesRequest\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\x12$\n\x1creturn_all_nonref_properties\x18\x0b \x01(\x08\"\x8b\x01\n\x17ObjectPropertiesRequest\x12\x11\n\tprop_name\x18\x01 \x01(\t\x12\x1c\n\x14primitive_properties\x18\x02 \x03(\t\x12?\n\x11object_properties\x18\x03 \x03(\x0b\x32$.weaviate.v1.ObjectPropertiesRequest\"\xb1\x01\n\x14RefPropertiesRequest\x12\x1a\n\x12reference_property\x18\x01 \x01(\t\x12\x32\n\nproperties\x18\x02 \x01(\x0b\x32\x1e.weaviate.v1.PropertiesRequest\x12.\n\x08metadata\x18\x03 \x01(\x0b\x32\x1c.weaviate.v1.MetadataRequest\x12\x19\n\x11target_collection\x18\x04 \x01(\t\"8\n\x06Rerank\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x12\n\x05query\x18\x02 \x01(\tH\x00\x88\x01\x01\x42\x08\n\x06_query\"\xf7\x02\n\x0bSearchReply\x12\x0c\n\x04took\x18\x01 \x01(\x02\x12*\n\x07results\x18\x02 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12*\n\x19generative_grouped_result\x18\x03 \x01(\tB\x02\x18\x01H\x00\x88\x01\x01\x12\x34\n\x10group_by_results\x18\x04 \x03(\x0b\x32\x1a.weaviate.v1.GroupByResult\x12\x46\n\x1agenerative_grouped_results\x18\x05 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x01\x88\x01\x01\x12\x35\n\rquery_profile\x18\x06 \x01(\x0b\x32\x19.weaviate.v1.QueryProfileH\x02\x88\x01\x01\x42\x1c\n\x1a_generative_grouped_resultB\x1d\n\x1b_generative_grouped_resultsB\x10\n\x0e_query_profile\"\x9e\x03\n\x0cQueryProfile\x12\x36\n\x06shards\x18\x01 \x03(\x0b\x32&.weaviate.v1.QueryProfile.ShardProfile\x1a\x86\x01\n\rSearchProfile\x12\x45\n\x07\x64\x65tails\x18\x01 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.SearchProfile.DetailsEntry\x1a.\n\x0c\x44\x65tailsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\xcc\x01\n\x0cShardProfile\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04node\x18\x02 \x01(\t\x12\x46\n\x08searches\x18\x03 \x03(\x0b\x32\x34.weaviate.v1.QueryProfile.ShardProfile.SearchesEntry\x1aX\n\rSearchesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x36\n\x05value\x18\x02 \x01(\x0b\x32\'.weaviate.v1.QueryProfile.SearchProfile:\x02\x38\x01\"\x1c\n\x0bRerankReply\x12\r\n\x05score\x18\x01 \x01(\x01\"\xe9\x02\n\rGroupByResult\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x14\n\x0cmin_distance\x18\x02 \x01(\x02\x12\x14\n\x0cmax_distance\x18\x03 \x01(\x02\x12\x19\n\x11number_of_objects\x18\x04 \x01(\x03\x12*\n\x07objects\x18\x05 \x03(\x0b\x32\x19.weaviate.v1.SearchResult\x12-\n\x06rerank\x18\x06 \x01(\x0b\x32\x18.weaviate.v1.RerankReplyH\x00\x88\x01\x01\x12\x39\n\ngenerative\x18\x07 \x01(\x0b\x32\x1c.weaviate.v1.GenerativeReplyB\x02\x18\x01H\x01\x88\x01\x01\x12=\n\x11generative_result\x18\x08 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x02\x88\x01\x01\x42\t\n\x07_rerankB\r\n\x0b_generativeB\x14\n\x12_generative_result\"\xb7\x01\n\x0cSearchResult\x12\x31\n\nproperties\x18\x01 \x01(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12-\n\x08metadata\x18\x02 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12\x36\n\ngenerative\x18\x03 \x01(\x0b\x32\x1d.weaviate.v1.GenerativeResultH\x00\x88\x01\x01\x42\r\n\x0b_generative\"\xf7\x04\n\x0eMetadataResult\x12\n\n\x02id\x18\x01 \x01(\t\x12\x12\n\x06vector\x18\x02 \x03(\x02\x42\x02\x18\x01\x12\x1a\n\x12\x63reation_time_unix\x18\x03 \x01(\x03\x12\"\n\x1a\x63reation_time_unix_present\x18\x04 \x01(\x08\x12\x1d\n\x15last_update_time_unix\x18\x05 \x01(\x03\x12%\n\x1dlast_update_time_unix_present\x18\x06 \x01(\x08\x12\x10\n\x08\x64istance\x18\x07 \x01(\x02\x12\x18\n\x10\x64istance_present\x18\x08 \x01(\x08\x12\x11\n\tcertainty\x18\t \x01(\x02\x12\x19\n\x11\x63\x65rtainty_present\x18\n \x01(\x08\x12\r\n\x05score\x18\x0b \x01(\x02\x12\x15\n\rscore_present\x18\x0c \x01(\x08\x12\x15\n\rexplain_score\x18\r \x01(\t\x12\x1d\n\x15\x65xplain_score_present\x18\x0e \x01(\x08\x12\x1a\n\ris_consistent\x18\x0f \x01(\x08H\x00\x88\x01\x01\x12\x16\n\ngenerative\x18\x10 \x01(\tB\x02\x18\x01\x12\x1e\n\x12generative_present\x18\x11 \x01(\x08\x42\x02\x18\x01\x12\x1d\n\x15is_consistent_present\x18\x12 \x01(\x08\x12\x14\n\x0cvector_bytes\x18\x13 \x01(\x0c\x12\x13\n\x0bid_as_bytes\x18\x14 \x01(\x0c\x12\x14\n\x0crerank_score\x18\x15 \x01(\x01\x12\x1c\n\x14rerank_score_present\x18\x16 \x01(\x08\x12%\n\x07vectors\x18\x17 \x03(\x0b\x32\x14.weaviate.v1.VectorsB\x10\n\x0e_is_consistent\"\x88\x02\n\x10PropertiesResult\x12\x33\n\tref_props\x18\x02 \x03(\x0b\x32 .weaviate.v1.RefPropertiesResult\x12\x19\n\x11target_collection\x18\x03 \x01(\t\x12-\n\x08metadata\x18\x04 \x01(\x0b\x32\x1b.weaviate.v1.MetadataResult\x12.\n\rnon_ref_props\x18\x0b \x01(\x0b\x32\x17.weaviate.v1.Properties\x12\x1b\n\x13ref_props_requested\x18\x0c \x01(\x08J\x04\x08\x01\x10\x02J\x04\x08\x05\x10\x06J\x04\x08\x06\x10\x07J\x04\x08\x07\x10\x08J\x04\x08\x08\x10\tJ\x04\x08\t\x10\nJ\x04\x08\n\x10\x0b\"[\n\x13RefPropertiesResult\x12\x31\n\nproperties\x18\x01 \x03(\x0b\x32\x1d.weaviate.v1.PropertiesResult\x12\x11\n\tprop_name\x18\x02 \x01(\t\"\xb3\t\n\x05\x42oost\x12\x30\n\nconditions\x18\x01 \x03(\x0b\x32\x1c.weaviate.v1.Boost.Condition\x12\x13\n\x06weight\x18\x02 \x01(\x02H\x00\x88\x01\x01\x12\x12\n\x05\x64\x65pth\x18\x03 \x01(\rH\x01\x88\x01\x01\x1aw\n\x15PropertyValueFunction\x12\x10\n\x08property\x18\x01 \x01(\t\x12?\n\x08modifier\x18\x02 \x01(\x0e\x32(.weaviate.v1.Boost.PropertyValueModifierH\x00\x88\x01\x01\x42\x0b\n\t_modifier\x1a\xcb\x01\n\x11TimeDecayFunction\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x0e\n\x06origin\x18\x02 \x01(\t\x12\r\n\x05scale\x18\x03 \x01(\t\x12\x13\n\x06offset\x18\x04 \x01(\tH\x00\x88\x01\x01\x12\x31\n\x05\x63urve\x18\x05 \x01(\x0e\x32\x1d.weaviate.v1.Boost.DecayCurveH\x01\x88\x01\x01\x12\x18\n\x0b\x64\x65\x63\x61y_value\x18\x06 \x01(\x02H\x02\x88\x01\x01\x42\t\n\x07_offsetB\x08\n\x06_curveB\x0e\n\x0c_decay_value\x1a\xce\x01\n\x14NumericDecayFunction\x12\x10\n\x08property\x18\x01 \x01(\t\x12\x0e\n\x06origin\x18\x02 \x01(\x01\x12\r\n\x05scale\x18\x03 \x01(\x01\x12\x13\n\x06offset\x18\x04 \x01(\x01H\x00\x88\x01\x01\x12\x31\n\x05\x63urve\x18\x05 \x01(\x0e\x32\x1d.weaviate.v1.Boost.DecayCurveH\x01\x88\x01\x01\x12\x18\n\x0b\x64\x65\x63\x61y_value\x18\x06 \x01(\x02H\x02\x88\x01\x01\x42\t\n\x07_offsetB\x08\n\x06_curveB\x0e\n\x0c_decay_value\x1a\xa2\x02\n\tCondition\x12&\n\x06\x66ilter\x18\x01 \x01(\x0b\x32\x14.weaviate.v1.FiltersH\x00\x12:\n\ntime_decay\x18\x02 \x01(\x0b\x32$.weaviate.v1.Boost.TimeDecayFunctionH\x00\x12\x42\n\x0eproperty_value\x18\x03 \x01(\x0b\x32(.weaviate.v1.Boost.PropertyValueFunctionH\x00\x12@\n\rnumeric_decay\x18\x04 \x01(\x0b\x32\'.weaviate.v1.Boost.NumericDecayFunctionH\x00\x12\x13\n\x06weight\x18\x05 \x01(\x02H\x01\x88\x01\x01\x42\x0b\n\tconditionB\t\n\x07_weight\"\x85\x01\n\x15PropertyValueModifier\x12\'\n#PROPERTY_VALUE_MODIFIER_UNSPECIFIED\x10\x00\x12!\n\x1dPROPERTY_VALUE_MODIFIER_LOG1P\x10\x01\x12 \n\x1cPROPERTY_VALUE_MODIFIER_SQRT\x10\x02\"u\n\nDecayCurve\x12\x1b\n\x17\x44\x45\x43\x41Y_CURVE_UNSPECIFIED\x10\x00\x12\x15\n\x11\x44\x45\x43\x41Y_CURVE_GAUSS\x10\x01\x12\x16\n\x12\x44\x45\x43\x41Y_CURVE_LINEAR\x10\x02\x12\x1b\n\x17\x44\x45\x43\x41Y_CURVE_EXPONENTIAL\x10\x03\x42\t\n\x07_weightB\x08\n\x06_depthBs\n#io.weaviate.client.grpc.protocol.v1B\x16WeaviateProtoSearchGetZ4github.com/weaviate/weaviate/grpc/generated;protocolb\x06proto3') _globals = globals() _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals) @@ -55,43 +55,57 @@ _globals['_METADATARESULT'].fields_by_name['generative_present']._loaded_options = None _globals['_METADATARESULT'].fields_by_name['generative_present']._serialized_options = b'\030\001' _globals['_SEARCHREQUEST']._serialized_start=116 - _globals['_SEARCHREQUEST']._serialized_end=1552 - _globals['_GROUPBY']._serialized_start=1554 - _globals['_GROUPBY']._serialized_end=1630 - _globals['_SORTBY']._serialized_start=1632 - _globals['_SORTBY']._serialized_end=1673 - _globals['_METADATAREQUEST']._serialized_start=1676 - _globals['_METADATAREQUEST']._serialized_end=1920 - _globals['_PROPERTIESREQUEST']._serialized_start=1923 - _globals['_PROPERTIESREQUEST']._serialized_end=2132 - _globals['_OBJECTPROPERTIESREQUEST']._serialized_start=2135 - _globals['_OBJECTPROPERTIESREQUEST']._serialized_end=2274 - _globals['_REFPROPERTIESREQUEST']._serialized_start=2277 - _globals['_REFPROPERTIESREQUEST']._serialized_end=2454 - _globals['_RERANK']._serialized_start=2456 - _globals['_RERANK']._serialized_end=2512 - _globals['_SEARCHREPLY']._serialized_start=2515 - _globals['_SEARCHREPLY']._serialized_end=2890 - _globals['_QUERYPROFILE']._serialized_start=2893 - _globals['_QUERYPROFILE']._serialized_end=3307 - _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_start=2966 - _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_end=3100 - _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_start=3054 - _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_end=3100 - _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_start=3103 - _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_end=3307 - _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_start=3219 - _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_end=3307 - _globals['_RERANKREPLY']._serialized_start=3309 - _globals['_RERANKREPLY']._serialized_end=3337 - _globals['_GROUPBYRESULT']._serialized_start=3340 - _globals['_GROUPBYRESULT']._serialized_end=3701 - _globals['_SEARCHRESULT']._serialized_start=3704 - _globals['_SEARCHRESULT']._serialized_end=3887 - _globals['_METADATARESULT']._serialized_start=3890 - _globals['_METADATARESULT']._serialized_end=4521 - _globals['_PROPERTIESRESULT']._serialized_start=4524 - _globals['_PROPERTIESRESULT']._serialized_end=4788 - _globals['_REFPROPERTIESRESULT']._serialized_start=4790 - _globals['_REFPROPERTIESRESULT']._serialized_end=4881 + _globals['_SEARCHREQUEST']._serialized_end=1602 + _globals['_GROUPBY']._serialized_start=1604 + _globals['_GROUPBY']._serialized_end=1680 + _globals['_SORTBY']._serialized_start=1682 + _globals['_SORTBY']._serialized_end=1723 + _globals['_METADATAREQUEST']._serialized_start=1726 + _globals['_METADATAREQUEST']._serialized_end=1970 + _globals['_PROPERTIESREQUEST']._serialized_start=1973 + _globals['_PROPERTIESREQUEST']._serialized_end=2182 + _globals['_OBJECTPROPERTIESREQUEST']._serialized_start=2185 + _globals['_OBJECTPROPERTIESREQUEST']._serialized_end=2324 + _globals['_REFPROPERTIESREQUEST']._serialized_start=2327 + _globals['_REFPROPERTIESREQUEST']._serialized_end=2504 + _globals['_RERANK']._serialized_start=2506 + _globals['_RERANK']._serialized_end=2562 + _globals['_SEARCHREPLY']._serialized_start=2565 + _globals['_SEARCHREPLY']._serialized_end=2940 + _globals['_QUERYPROFILE']._serialized_start=2943 + _globals['_QUERYPROFILE']._serialized_end=3357 + _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_start=3016 + _globals['_QUERYPROFILE_SEARCHPROFILE']._serialized_end=3150 + _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_start=3104 + _globals['_QUERYPROFILE_SEARCHPROFILE_DETAILSENTRY']._serialized_end=3150 + _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_start=3153 + _globals['_QUERYPROFILE_SHARDPROFILE']._serialized_end=3357 + _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_start=3269 + _globals['_QUERYPROFILE_SHARDPROFILE_SEARCHESENTRY']._serialized_end=3357 + _globals['_RERANKREPLY']._serialized_start=3359 + _globals['_RERANKREPLY']._serialized_end=3387 + _globals['_GROUPBYRESULT']._serialized_start=3390 + _globals['_GROUPBYRESULT']._serialized_end=3751 + _globals['_SEARCHRESULT']._serialized_start=3754 + _globals['_SEARCHRESULT']._serialized_end=3937 + _globals['_METADATARESULT']._serialized_start=3940 + _globals['_METADATARESULT']._serialized_end=4571 + _globals['_PROPERTIESRESULT']._serialized_start=4574 + _globals['_PROPERTIESRESULT']._serialized_end=4838 + _globals['_REFPROPERTIESRESULT']._serialized_start=4840 + _globals['_REFPROPERTIESRESULT']._serialized_end=4931 + _globals['_BOOST']._serialized_start=4934 + _globals['_BOOST']._serialized_end=6137 + _globals['_BOOST_PROPERTYVALUEFUNCTION']._serialized_start=5034 + _globals['_BOOST_PROPERTYVALUEFUNCTION']._serialized_end=5153 + _globals['_BOOST_TIMEDECAYFUNCTION']._serialized_start=5156 + _globals['_BOOST_TIMEDECAYFUNCTION']._serialized_end=5359 + _globals['_BOOST_NUMERICDECAYFUNCTION']._serialized_start=5362 + _globals['_BOOST_NUMERICDECAYFUNCTION']._serialized_end=5568 + _globals['_BOOST_CONDITION']._serialized_start=5571 + _globals['_BOOST_CONDITION']._serialized_end=5861 + _globals['_BOOST_PROPERTYVALUEMODIFIER']._serialized_start=5864 + _globals['_BOOST_PROPERTYVALUEMODIFIER']._serialized_end=5997 + _globals['_BOOST_DECAYCURVE']._serialized_start=5999 + _globals['_BOOST_DECAYCURVE']._serialized_end=6116 # @@protoc_insertion_point(module_scope) diff --git a/weaviate/proto/v1/v6300/v1/search_get_pb2.pyi b/weaviate/proto/v1/v6300/v1/search_get_pb2.pyi index f631fb396..9f68ada4d 100644 --- a/weaviate/proto/v1/v6300/v1/search_get_pb2.pyi +++ b/weaviate/proto/v1/v6300/v1/search_get_pb2.pyi @@ -3,6 +3,7 @@ from weaviate.proto.v1.v6300.v1 import base_search_pb2 as _base_search_pb2 from weaviate.proto.v1.v6300.v1 import generative_pb2 as _generative_pb2 from weaviate.proto.v1.v6300.v1 import properties_pb2 as _properties_pb2 from google.protobuf.internal import containers as _containers +from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper from google.protobuf import descriptor as _descriptor from google.protobuf import message as _message from collections.abc import Iterable as _Iterable, Mapping as _Mapping @@ -11,7 +12,7 @@ from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union DESCRIPTOR: _descriptor.FileDescriptor class SearchRequest(_message.Message): - __slots__ = ("collection", "tenant", "consistency_level", "properties", "metadata", "group_by", "limit", "offset", "autocut", "after", "sort_by", "filters", "hybrid_search", "bm25_search", "near_vector", "near_object", "near_text", "near_image", "near_audio", "near_video", "near_depth", "near_thermal", "near_imu", "generative", "rerank", "uses_123_api", "uses_125_api", "uses_127_api") + __slots__ = ("collection", "tenant", "consistency_level", "properties", "metadata", "group_by", "limit", "offset", "autocut", "after", "sort_by", "filters", "hybrid_search", "bm25_search", "near_vector", "near_object", "near_text", "near_image", "near_audio", "near_video", "near_depth", "near_thermal", "near_imu", "generative", "rerank", "boost", "uses_123_api", "uses_125_api", "uses_127_api") COLLECTION_FIELD_NUMBER: _ClassVar[int] TENANT_FIELD_NUMBER: _ClassVar[int] CONSISTENCY_LEVEL_FIELD_NUMBER: _ClassVar[int] @@ -37,6 +38,7 @@ class SearchRequest(_message.Message): NEAR_IMU_FIELD_NUMBER: _ClassVar[int] GENERATIVE_FIELD_NUMBER: _ClassVar[int] RERANK_FIELD_NUMBER: _ClassVar[int] + BOOST_FIELD_NUMBER: _ClassVar[int] USES_123_API_FIELD_NUMBER: _ClassVar[int] USES_125_API_FIELD_NUMBER: _ClassVar[int] USES_127_API_FIELD_NUMBER: _ClassVar[int] @@ -65,10 +67,11 @@ class SearchRequest(_message.Message): near_imu: _base_search_pb2.NearIMUSearch generative: _generative_pb2.GenerativeSearch rerank: Rerank + boost: Boost uses_123_api: bool uses_125_api: bool uses_127_api: bool - def __init__(self, collection: _Optional[str] = ..., tenant: _Optional[str] = ..., consistency_level: _Optional[_Union[_base_pb2.ConsistencyLevel, str]] = ..., properties: _Optional[_Union[PropertiesRequest, _Mapping]] = ..., metadata: _Optional[_Union[MetadataRequest, _Mapping]] = ..., group_by: _Optional[_Union[GroupBy, _Mapping]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., autocut: _Optional[int] = ..., after: _Optional[str] = ..., sort_by: _Optional[_Iterable[_Union[SortBy, _Mapping]]] = ..., filters: _Optional[_Union[_base_pb2.Filters, _Mapping]] = ..., hybrid_search: _Optional[_Union[_base_search_pb2.Hybrid, _Mapping]] = ..., bm25_search: _Optional[_Union[_base_search_pb2.BM25, _Mapping]] = ..., near_vector: _Optional[_Union[_base_search_pb2.NearVector, _Mapping]] = ..., near_object: _Optional[_Union[_base_search_pb2.NearObject, _Mapping]] = ..., near_text: _Optional[_Union[_base_search_pb2.NearTextSearch, _Mapping]] = ..., near_image: _Optional[_Union[_base_search_pb2.NearImageSearch, _Mapping]] = ..., near_audio: _Optional[_Union[_base_search_pb2.NearAudioSearch, _Mapping]] = ..., near_video: _Optional[_Union[_base_search_pb2.NearVideoSearch, _Mapping]] = ..., near_depth: _Optional[_Union[_base_search_pb2.NearDepthSearch, _Mapping]] = ..., near_thermal: _Optional[_Union[_base_search_pb2.NearThermalSearch, _Mapping]] = ..., near_imu: _Optional[_Union[_base_search_pb2.NearIMUSearch, _Mapping]] = ..., generative: _Optional[_Union[_generative_pb2.GenerativeSearch, _Mapping]] = ..., rerank: _Optional[_Union[Rerank, _Mapping]] = ..., uses_123_api: bool = ..., uses_125_api: bool = ..., uses_127_api: bool = ...) -> None: ... + def __init__(self, collection: _Optional[str] = ..., tenant: _Optional[str] = ..., consistency_level: _Optional[_Union[_base_pb2.ConsistencyLevel, str]] = ..., properties: _Optional[_Union[PropertiesRequest, _Mapping]] = ..., metadata: _Optional[_Union[MetadataRequest, _Mapping]] = ..., group_by: _Optional[_Union[GroupBy, _Mapping]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., autocut: _Optional[int] = ..., after: _Optional[str] = ..., sort_by: _Optional[_Iterable[_Union[SortBy, _Mapping]]] = ..., filters: _Optional[_Union[_base_pb2.Filters, _Mapping]] = ..., hybrid_search: _Optional[_Union[_base_search_pb2.Hybrid, _Mapping]] = ..., bm25_search: _Optional[_Union[_base_search_pb2.BM25, _Mapping]] = ..., near_vector: _Optional[_Union[_base_search_pb2.NearVector, _Mapping]] = ..., near_object: _Optional[_Union[_base_search_pb2.NearObject, _Mapping]] = ..., near_text: _Optional[_Union[_base_search_pb2.NearTextSearch, _Mapping]] = ..., near_image: _Optional[_Union[_base_search_pb2.NearImageSearch, _Mapping]] = ..., near_audio: _Optional[_Union[_base_search_pb2.NearAudioSearch, _Mapping]] = ..., near_video: _Optional[_Union[_base_search_pb2.NearVideoSearch, _Mapping]] = ..., near_depth: _Optional[_Union[_base_search_pb2.NearDepthSearch, _Mapping]] = ..., near_thermal: _Optional[_Union[_base_search_pb2.NearThermalSearch, _Mapping]] = ..., near_imu: _Optional[_Union[_base_search_pb2.NearIMUSearch, _Mapping]] = ..., generative: _Optional[_Union[_generative_pb2.GenerativeSearch, _Mapping]] = ..., rerank: _Optional[_Union[Rerank, _Mapping]] = ..., boost: _Optional[_Union[Boost, _Mapping]] = ..., uses_123_api: bool = ..., uses_125_api: bool = ..., uses_127_api: bool = ...) -> None: ... class GroupBy(_message.Message): __slots__ = ("path", "number_of_groups", "objects_per_group") @@ -313,3 +316,81 @@ class RefPropertiesResult(_message.Message): properties: _containers.RepeatedCompositeFieldContainer[PropertiesResult] prop_name: str def __init__(self, properties: _Optional[_Iterable[_Union[PropertiesResult, _Mapping]]] = ..., prop_name: _Optional[str] = ...) -> None: ... + +class Boost(_message.Message): + __slots__ = ("conditions", "weight", "depth") + class PropertyValueModifier(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): + __slots__ = () + PROPERTY_VALUE_MODIFIER_UNSPECIFIED: _ClassVar[Boost.PropertyValueModifier] + PROPERTY_VALUE_MODIFIER_LOG1P: _ClassVar[Boost.PropertyValueModifier] + PROPERTY_VALUE_MODIFIER_SQRT: _ClassVar[Boost.PropertyValueModifier] + PROPERTY_VALUE_MODIFIER_UNSPECIFIED: Boost.PropertyValueModifier + PROPERTY_VALUE_MODIFIER_LOG1P: Boost.PropertyValueModifier + PROPERTY_VALUE_MODIFIER_SQRT: Boost.PropertyValueModifier + class DecayCurve(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): + __slots__ = () + DECAY_CURVE_UNSPECIFIED: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_GAUSS: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_LINEAR: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_EXPONENTIAL: _ClassVar[Boost.DecayCurve] + DECAY_CURVE_UNSPECIFIED: Boost.DecayCurve + DECAY_CURVE_GAUSS: Boost.DecayCurve + DECAY_CURVE_LINEAR: Boost.DecayCurve + DECAY_CURVE_EXPONENTIAL: Boost.DecayCurve + class PropertyValueFunction(_message.Message): + __slots__ = ("property", "modifier") + PROPERTY_FIELD_NUMBER: _ClassVar[int] + MODIFIER_FIELD_NUMBER: _ClassVar[int] + property: str + modifier: Boost.PropertyValueModifier + def __init__(self, property: _Optional[str] = ..., modifier: _Optional[_Union[Boost.PropertyValueModifier, str]] = ...) -> None: ... + class TimeDecayFunction(_message.Message): + __slots__ = ("property", "origin", "scale", "offset", "curve", "decay_value") + PROPERTY_FIELD_NUMBER: _ClassVar[int] + ORIGIN_FIELD_NUMBER: _ClassVar[int] + SCALE_FIELD_NUMBER: _ClassVar[int] + OFFSET_FIELD_NUMBER: _ClassVar[int] + CURVE_FIELD_NUMBER: _ClassVar[int] + DECAY_VALUE_FIELD_NUMBER: _ClassVar[int] + property: str + origin: str + scale: str + offset: str + curve: Boost.DecayCurve + decay_value: float + def __init__(self, property: _Optional[str] = ..., origin: _Optional[str] = ..., scale: _Optional[str] = ..., offset: _Optional[str] = ..., curve: _Optional[_Union[Boost.DecayCurve, str]] = ..., decay_value: _Optional[float] = ...) -> None: ... + class NumericDecayFunction(_message.Message): + __slots__ = ("property", "origin", "scale", "offset", "curve", "decay_value") + PROPERTY_FIELD_NUMBER: _ClassVar[int] + ORIGIN_FIELD_NUMBER: _ClassVar[int] + SCALE_FIELD_NUMBER: _ClassVar[int] + OFFSET_FIELD_NUMBER: _ClassVar[int] + CURVE_FIELD_NUMBER: _ClassVar[int] + DECAY_VALUE_FIELD_NUMBER: _ClassVar[int] + property: str + origin: float + scale: float + offset: float + curve: Boost.DecayCurve + decay_value: float + def __init__(self, property: _Optional[str] = ..., origin: _Optional[float] = ..., scale: _Optional[float] = ..., offset: _Optional[float] = ..., curve: _Optional[_Union[Boost.DecayCurve, str]] = ..., decay_value: _Optional[float] = ...) -> None: ... + class Condition(_message.Message): + __slots__ = ("filter", "time_decay", "property_value", "numeric_decay", "weight") + FILTER_FIELD_NUMBER: _ClassVar[int] + TIME_DECAY_FIELD_NUMBER: _ClassVar[int] + PROPERTY_VALUE_FIELD_NUMBER: _ClassVar[int] + NUMERIC_DECAY_FIELD_NUMBER: _ClassVar[int] + WEIGHT_FIELD_NUMBER: _ClassVar[int] + filter: _base_pb2.Filters + time_decay: Boost.TimeDecayFunction + property_value: Boost.PropertyValueFunction + numeric_decay: Boost.NumericDecayFunction + weight: float + def __init__(self, filter: _Optional[_Union[_base_pb2.Filters, _Mapping]] = ..., time_decay: _Optional[_Union[Boost.TimeDecayFunction, _Mapping]] = ..., property_value: _Optional[_Union[Boost.PropertyValueFunction, _Mapping]] = ..., numeric_decay: _Optional[_Union[Boost.NumericDecayFunction, _Mapping]] = ..., weight: _Optional[float] = ...) -> None: ... + CONDITIONS_FIELD_NUMBER: _ClassVar[int] + WEIGHT_FIELD_NUMBER: _ClassVar[int] + DEPTH_FIELD_NUMBER: _ClassVar[int] + conditions: _containers.RepeatedCompositeFieldContainer[Boost.Condition] + weight: float + depth: int + def __init__(self, conditions: _Optional[_Iterable[_Union[Boost.Condition, _Mapping]]] = ..., weight: _Optional[float] = ..., depth: _Optional[int] = ...) -> None: ... From 393d48725f546cf0653c5f5b4dac031f4b61e05a Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Fri, 22 May 2026 15:27:48 +1000 Subject: [PATCH 02/11] Add typing for filter --- weaviate/collections/classes/grpc.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index ec72f02d3..fdc127270 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -18,6 +18,7 @@ from pydantic import ConfigDict, Field from typing_extensions import ClassVar, TypeGuard, TypeVar +from weaviate.collections.classes.filters import FilterReturn from weaviate.collections.classes.types import _WeaviateInput from weaviate.exceptions import WeaviateInvalidInputError from weaviate.proto.v1 import base_search_pb2 @@ -297,7 +298,7 @@ class _PropertyValueFunction: @dataclass class _BoostCondition: - filter: Optional[Any] = None # noqa: A003 # FilterReturn + filter: Optional[FilterReturn] = None # noqa: A003 time_decay: Optional[_TimeDecayFunction] = None numeric_decay: Optional[_NumericDecayFunction] = None property_value: Optional[_PropertyValueFunction] = None @@ -359,7 +360,7 @@ def __init__(self) -> None: @staticmethod def filter( # noqa: A003 - filter: Any, # noqa: A002 + filter: FilterReturn, # noqa: A002 *, weight: Optional[float] = None, depth: Optional[int] = None, From 1a843d2486169397b48989a59ad67730b48d7f1e Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Fri, 22 May 2026 15:51:40 +1000 Subject: [PATCH 03/11] Review feedback --- integration/test_collection_boost.py | 2 ++ weaviate/collections/classes/grpc.py | 24 +++++++++--------------- 2 files changed, 11 insertions(+), 15 deletions(-) diff --git a/integration/test_collection_boost.py b/integration/test_collection_boost.py index ff3b3ee3a..8989ee901 100644 --- a/integration/test_collection_boost.py +++ b/integration/test_collection_boost.py @@ -115,6 +115,7 @@ def test_boost_numeric_decay(collection_factory: CollectionFactory) -> None: origin=50.0, scale=20.0, curve=Boost.Curve.LINEAR, + decay=0.5, weight=1.0, ), return_metadata=MetadataQuery(distance=True), @@ -135,6 +136,7 @@ def test_boost_time_decay(collection_factory: CollectionFactory) -> None: origin="2024-01-01T00:00:00Z", scale="365d", curve=Boost.Curve.EXPONENTIAL, + decay=0.3, weight=1.0, ), return_metadata=MetadataQuery(distance=True), diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index fdc127270..70e2318e7 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -382,7 +382,7 @@ def time_decay( scale: Union[str, timedelta], offset: Optional[Union[str, timedelta]] = None, curve: Optional[Union[_BoostCurve, str]] = None, - decay_value: Optional[float] = None, + decay: Optional[float] = None, weight: Optional[float] = None, depth: Optional[int] = None, ) -> _Boost: @@ -392,12 +392,12 @@ def time_decay( property: The date property name to compute distance from. origin: The origin point. Use "now" for current time or a datetime for a specific time. Defaults to "now". - scale: Distance from origin where score equals decay_value. Use timedelta + scale: Distance from origin where score equals decay. Use timedelta (e.g. timedelta(days=7)) or a string shorthand like "7d", "24h". offset: Documents within this distance from origin get full score (default "0"). Accepts the same types as scale. curve: Decay curve type: `Boost.Curve.EXPONENTIAL` (default), `Boost.Curve.GAUSSIAN`, or `Boost.Curve.LINEAR`. - decay_value: Score at scale distance from origin (default 0.5). + decay: Score at scale distance from origin (default 0.5). weight: Blending weight [0,1] controlling how much the rank affects final scores. depth: Number of results to rescore (default 100, max 10000). """ @@ -410,7 +410,7 @@ def time_decay( scale=_decay_value_to_str(scale), offset=_decay_value_to_str(offset) if offset is not None else None, curve=curve.value if isinstance(curve, _BoostCurve) else curve, - decay_value=decay_value, + decay_value=decay, ) ) ], @@ -426,7 +426,7 @@ def numeric_decay( scale: float, offset: Optional[float] = None, curve: Optional[Union[_BoostCurve, str]] = None, - decay_value: Optional[float] = None, + decay: Optional[float] = None, weight: Optional[float] = None, depth: Optional[int] = None, ) -> _Boost: @@ -435,10 +435,10 @@ def numeric_decay( Args: property: The numeric property name to compute distance from. origin: The origin point (numeric value). - scale: Distance from origin where score equals decay_value. + scale: Distance from origin where score equals decay. offset: Documents within this distance from origin get full score (default 0). curve: Decay curve type: `Boost.Curve.EXPONENTIAL` (default), `Boost.Curve.GAUSSIAN`, or `Boost.Curve.LINEAR`. - decay_value: Score at scale distance from origin (default 0.5). + decay: Score at scale distance from origin (default 0.5). weight: Blending weight [0,1] controlling how much the rank affects final scores. depth: Number of results to rescore (default 100, max 10000). """ @@ -451,7 +451,7 @@ def numeric_decay( scale=float(scale), offset=float(offset) if offset is not None else None, curve=curve.value if isinstance(curve, _BoostCurve) else curve, - decay_value=decay_value, + decay_value=decay, ) ) ], @@ -510,13 +510,7 @@ def blend( for r in ranks: for cond in r.conditions: if cond.weight is None and r.weight is not None: - cond = _BoostCondition( - filter=cond.filter, - time_decay=cond.time_decay, - numeric_decay=cond.numeric_decay, - property_value=cond.property_value, - weight=r.weight, - ) + cond.weight = r.weight conditions.append(cond) return _Boost(conditions=conditions, weight=weight, depth=depth) From 7727999cc6b647078851fa85cfc54f0093c71220 Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Tue, 26 May 2026 14:45:10 +1000 Subject: [PATCH 04/11] Improve docstrings --- weaviate/collections/classes/grpc.py | 21 ++++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index 70e2318e7..19d3067eb 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -430,12 +430,16 @@ def numeric_decay( weight: Optional[float] = None, depth: Optional[int] = None, ) -> _Boost: - """Apply numeric distance-based decay scoring from an origin value. + """Score decays with distance from a numeric origin — closer to the origin ranks higher. + + Use this when "closer to X is better" (e.g., prefer prices near $50, houses near 2000 sqft). + Requires you to define an origin and scale. For simple "higher is better" boosting without + an origin, use `Boost.property()` instead. Args: property: The numeric property name to compute distance from. - origin: The origin point (numeric value). - scale: Distance from origin where score equals decay. + origin: The target value — documents closest to this score highest. + scale: Distance from origin where score equals the decay value. offset: Documents within this distance from origin get full score (default 0). curve: Decay curve type: `Boost.Curve.EXPONENTIAL` (default), `Boost.Curve.GAUSSIAN`, or `Boost.Curve.LINEAR`. decay: Score at scale distance from origin (default 0.5). @@ -467,11 +471,18 @@ def property( # noqa: A003 weight: Optional[float] = None, depth: Optional[int] = None, ) -> _Boost: - """Rank by a numeric property's value directly. + """Boost by a numeric property's raw value — higher values rank higher. + + Use this for simple proportional boosting (e.g., popularity count, review score) + when you don't need to define an origin or scale. For distance-based decay from a + specific value, use `Boost.numeric_decay()` instead. + + Currently only supports numeric (int/float) properties. Args: - name: The property name to use as a ranking signal. + name: The numeric property name to use as a ranking signal. modifier: Score modifier: `Boost.Modifier.NONE` (default), `Boost.Modifier.LOG1P`, or `Boost.Modifier.SQRT`. + Use LOG1P or SQRT to dampen the effect of large value ranges. weight: Blending weight [0,1] controlling how much the rank affects final scores. depth: Number of results to rescore (default 100, max 10000). """ From 113fd7e98ac6038588197f85ae243792d6447a80 Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Tue, 2 Jun 2026 09:53:09 +1000 Subject: [PATCH 05/11] Enforce enum for Boost curves and modifiers --- integration/test_collection_boost.py | 15 +++++++++++++ weaviate/collections/classes/grpc.py | 23 ++++++++++---------- weaviate/collections/grpc/query.py | 32 ++++++++++++++-------------- 3 files changed, 42 insertions(+), 28 deletions(-) diff --git a/integration/test_collection_boost.py b/integration/test_collection_boost.py index 8989ee901..5be049f25 100644 --- a/integration/test_collection_boost.py +++ b/integration/test_collection_boost.py @@ -264,3 +264,18 @@ def test_boost_api_surface() -> None: assert len(b.conditions) == 2 assert b.weight == 0.8 assert b.depth == 200 + + +def test_boost_default_curve_is_unspecified() -> None: + """Omitting curve defaults to None (sent as UNSPECIFIED on the wire).""" + b = Boost.numeric_decay("price", origin=50.0, scale=20.0) + assert b.conditions[0].numeric_decay.curve is None + + b = Boost.time_decay("created", scale="7d") + assert b.conditions[0].time_decay.curve is None + + +def test_boost_default_modifier_is_unspecified() -> None: + """Omitting modifier defaults to None (sent as UNSPECIFIED on the wire).""" + b = Boost.property("count") + assert b.conditions[0].property_value.modifier is None diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index 19d3067eb..bfbaf2302 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -276,7 +276,7 @@ class _TimeDecayFunction: origin: str scale: str offset: Optional[str] = None - curve: Optional[str] = None + curve: Optional["_BoostCurve"] = None decay_value: Optional[float] = None @@ -286,14 +286,14 @@ class _NumericDecayFunction: origin: float scale: float offset: Optional[float] = None - curve: Optional[str] = None + curve: Optional["_BoostCurve"] = None decay_value: Optional[float] = None @dataclass class _PropertyValueFunction: property: str # noqa: A003 - modifier: Optional[str] = None + modifier: Optional["_BoostModifier"] = None @dataclass @@ -333,6 +333,7 @@ def _decay_value_to_str(val: Union[str, int, float, timedelta, datetime]) -> str class _BoostCurve(str, BaseEnum): """Decay curve type for distance-based rank scoring.""" + UNSPECIFIED = "unspecified" EXPONENTIAL = "exp" GAUSSIAN = "gauss" LINEAR = "linear" @@ -341,7 +342,7 @@ class _BoostCurve(str, BaseEnum): class _BoostModifier(str, BaseEnum): """Score modifier for property-value rank scoring.""" - NONE = "none" + UNSPECIFIED = "unspecified" LOG1P = "log1p" SQRT = "sqrt" @@ -381,7 +382,7 @@ def time_decay( origin: Optional[Union[str, datetime]] = None, scale: Union[str, timedelta], offset: Optional[Union[str, timedelta]] = None, - curve: Optional[Union[_BoostCurve, str]] = None, + curve: Optional[_BoostCurve] = None, decay: Optional[float] = None, weight: Optional[float] = None, depth: Optional[int] = None, @@ -409,7 +410,7 @@ def time_decay( origin=_decay_value_to_str(origin) if origin is not None else "", scale=_decay_value_to_str(scale), offset=_decay_value_to_str(offset) if offset is not None else None, - curve=curve.value if isinstance(curve, _BoostCurve) else curve, + curve=curve, decay_value=decay, ) ) @@ -425,7 +426,7 @@ def numeric_decay( origin: float, scale: float, offset: Optional[float] = None, - curve: Optional[Union[_BoostCurve, str]] = None, + curve: Optional[_BoostCurve] = None, decay: Optional[float] = None, weight: Optional[float] = None, depth: Optional[int] = None, @@ -454,7 +455,7 @@ def numeric_decay( origin=float(origin), scale=float(scale), offset=float(offset) if offset is not None else None, - curve=curve.value if isinstance(curve, _BoostCurve) else curve, + curve=curve, decay_value=decay, ) ) @@ -467,7 +468,7 @@ def numeric_decay( def property( # noqa: A003 name: str, *, - modifier: Optional[Union[_BoostModifier, str]] = None, + modifier: Optional[_BoostModifier] = None, weight: Optional[float] = None, depth: Optional[int] = None, ) -> _Boost: @@ -491,9 +492,7 @@ def property( # noqa: A003 _BoostCondition( property_value=_PropertyValueFunction( property=name, - modifier=modifier.value - if isinstance(modifier, _BoostModifier) - else modifier, + modifier=modifier, ) ) ], diff --git a/weaviate/collections/grpc/query.py b/weaviate/collections/grpc/query.py index c112d5574..d103e2db5 100644 --- a/weaviate/collections/grpc/query.py +++ b/weaviate/collections/grpc/query.py @@ -33,6 +33,8 @@ Rerank, TargetVectorJoinType, _Boost, + _BoostCurve, + _BoostModifier, _MetadataQuery, _QueryReference, _QueryReferenceMultiTarget, @@ -561,22 +563,18 @@ def _metadata_to_grpc(self, metadata: _MetadataQuery) -> search_get_pb2.Metadata _Boost_pb2 = search_get_pb2.Boost _CURVE_TO_PROTO = { - "exp": _Boost_pb2.DECAY_CURVE_EXPONENTIAL, - "gauss": _Boost_pb2.DECAY_CURVE_GAUSS, - "linear": _Boost_pb2.DECAY_CURVE_LINEAR, + _BoostCurve.UNSPECIFIED: _Boost_pb2.DECAY_CURVE_UNSPECIFIED, + _BoostCurve.EXPONENTIAL: _Boost_pb2.DECAY_CURVE_EXPONENTIAL, + _BoostCurve.GAUSSIAN: _Boost_pb2.DECAY_CURVE_GAUSS, + _BoostCurve.LINEAR: _Boost_pb2.DECAY_CURVE_LINEAR, } _MODIFIER_TO_PROTO = { - "none": _Boost_pb2.PROPERTY_VALUE_MODIFIER_UNSPECIFIED, - "log1p": _Boost_pb2.PROPERTY_VALUE_MODIFIER_LOG1P, - "sqrt": _Boost_pb2.PROPERTY_VALUE_MODIFIER_SQRT, + _BoostModifier.UNSPECIFIED: _Boost_pb2.PROPERTY_VALUE_MODIFIER_UNSPECIFIED, + _BoostModifier.LOG1P: _Boost_pb2.PROPERTY_VALUE_MODIFIER_LOG1P, + _BoostModifier.SQRT: _Boost_pb2.PROPERTY_VALUE_MODIFIER_SQRT, } - def __resolve_curve(self, curve: Optional[str]) -> "search_get_pb2.Boost.DecayCurve": - if curve is None: - return self._Boost_pb2.DECAY_CURVE_EXPONENTIAL - return self._CURVE_TO_PROTO.get(curve, self._Boost_pb2.DECAY_CURVE_EXPONENTIAL) - def __boost_to_grpc(self, boost: Optional[_Boost]) -> Optional[search_get_pb2.Boost]: if boost is None: return None @@ -593,7 +591,9 @@ def __boost_to_grpc(self, boost: Optional[_Boost]) -> Optional[search_get_pb2.Bo origin=cond.time_decay.origin, scale=cond.time_decay.scale, offset=cond.time_decay.offset, - curve=self.__resolve_curve(cond.time_decay.curve), + curve=self._CURVE_TO_PROTO[cond.time_decay.curve] + if cond.time_decay.curve is not None + else _B.DECAY_CURVE_UNSPECIFIED, decay_value=cond.time_decay.decay_value, ) ) @@ -604,7 +604,9 @@ def __boost_to_grpc(self, boost: Optional[_Boost]) -> Optional[search_get_pb2.Bo origin=cond.numeric_decay.origin, scale=cond.numeric_decay.scale, offset=cond.numeric_decay.offset, - curve=self.__resolve_curve(cond.numeric_decay.curve), + curve=self._CURVE_TO_PROTO[cond.numeric_decay.curve] + if cond.numeric_decay.curve is not None + else _B.DECAY_CURVE_UNSPECIFIED, decay_value=cond.numeric_decay.decay_value, ) ) @@ -612,9 +614,7 @@ def __boost_to_grpc(self, boost: Optional[_Boost]) -> Optional[search_get_pb2.Bo grpc_cond.property_value.CopyFrom( _B.PropertyValueFunction( property=cond.property_value.property, - modifier=self._MODIFIER_TO_PROTO.get( - cond.property_value.modifier, _B.PROPERTY_VALUE_MODIFIER_UNSPECIFIED - ) + modifier=self._MODIFIER_TO_PROTO[cond.property_value.modifier] if cond.property_value.modifier is not None else _B.PROPERTY_VALUE_MODIFIER_UNSPECIFIED, ) From 6985c2e1b7c6a90a160059c75d7e8d0f0b25eeb0 Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Tue, 2 Jun 2026 09:56:32 +1000 Subject: [PATCH 06/11] Rank -> boost naming --- weaviate/collections/classes/grpc.py | 30 ++++++++++++++-------------- 1 file changed, 15 insertions(+), 15 deletions(-) diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index bfbaf2302..b3a125b16 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -331,7 +331,7 @@ def _decay_value_to_str(val: Union[str, int, float, timedelta, datetime]) -> str class _BoostCurve(str, BaseEnum): - """Decay curve type for distance-based rank scoring.""" + """Decay curve type for distance-based boost scoring.""" UNSPECIFIED = "unspecified" EXPONENTIAL = "exp" @@ -340,7 +340,7 @@ class _BoostCurve(str, BaseEnum): class _BoostModifier(str, BaseEnum): - """Score modifier for property-value rank scoring.""" + """Score modifier for property-value boost scoring.""" UNSPECIFIED = "unspecified" LOG1P = "log1p" @@ -348,16 +348,16 @@ class _BoostModifier(str, BaseEnum): class Boost: - """Define soft-ranking conditions to boost or demote matching documents without excluding them. + """Define soft-scoring conditions to boost or demote matching documents without excluding them. - Use the static methods `boost()`, `decay()`, and `blend()` to create rank configurations. + Use the static methods `filter()`, `time_decay()`, `numeric_decay()`, `property()`, and `blend()` to create boost configurations. """ Curve = _BoostCurve Modifier = _BoostModifier def __init__(self) -> None: - raise TypeError("Boost cannot be instantiated. Use the static methods to create a rank.") + raise TypeError("Boost cannot be instantiated. Use the static methods to create a boost.") @staticmethod def filter( # noqa: A003 @@ -370,7 +370,7 @@ def filter( # noqa: A003 Args: filter: The filter condition (same as used in `filters=` parameter). - weight: Blending weight [0,1] controlling how much the rank affects final scores. + weight: Blending weight [0,1] controlling how much the boost affects final scores. depth: Number of results to rescore (default 100, max 10000). Higher values improve accuracy at the cost of performance. """ return _Boost(conditions=[_BoostCondition(filter=filter)], weight=weight, depth=depth) @@ -399,7 +399,7 @@ def time_decay( Accepts the same types as scale. curve: Decay curve type: `Boost.Curve.EXPONENTIAL` (default), `Boost.Curve.GAUSSIAN`, or `Boost.Curve.LINEAR`. decay: Score at scale distance from origin (default 0.5). - weight: Blending weight [0,1] controlling how much the rank affects final scores. + weight: Blending weight [0,1] controlling how much the boost affects final scores. depth: Number of results to rescore (default 100, max 10000). """ return _Boost( @@ -444,7 +444,7 @@ def numeric_decay( offset: Documents within this distance from origin get full score (default 0). curve: Decay curve type: `Boost.Curve.EXPONENTIAL` (default), `Boost.Curve.GAUSSIAN`, or `Boost.Curve.LINEAR`. decay: Score at scale distance from origin (default 0.5). - weight: Blending weight [0,1] controlling how much the rank affects final scores. + weight: Blending weight [0,1] controlling how much the boost affects final scores. depth: Number of results to rescore (default 100, max 10000). """ return _Boost( @@ -484,7 +484,7 @@ def property( # noqa: A003 name: The numeric property name to use as a ranking signal. modifier: Score modifier: `Boost.Modifier.NONE` (default), `Boost.Modifier.LOG1P`, or `Boost.Modifier.SQRT`. Use LOG1P or SQRT to dampen the effect of large value ranges. - weight: Blending weight [0,1] controlling how much the rank affects final scores. + weight: Blending weight [0,1] controlling how much the boost affects final scores. depth: Number of results to rescore (default 100, max 10000). """ return _Boost( @@ -502,22 +502,22 @@ def property( # noqa: A003 @staticmethod def blend( - *ranks: _Boost, + *boosts: _Boost, weight: Optional[float] = None, depth: Optional[int] = None, ) -> _Boost: - """Combine multiple rank conditions with individual weights. + """Combine multiple boost conditions with individual weights. - When blending, each sub-rank's weight becomes a per-condition weight, + When blending, each sub-boost's weight becomes a per-condition weight, and the `weight` parameter here controls the overall blending strength. Args: - *ranks: Rank objects created via `Boost.filter()`, `Boost.time_decay()`, `Boost.numeric_decay()`, or `Boost.property()`. - weight: Overall blending weight [0,1] for combining primary search and rank scores. + *boosts: Boost objects created via `Boost.filter()`, `Boost.time_decay()`, `Boost.numeric_decay()`, or `Boost.property()`. + weight: Overall blending weight [0,1] for combining primary search and boost scores. depth: Number of results to rescore (default 100, max 10000). Higher values improve accuracy at the cost of performance. """ conditions: List[_BoostCondition] = [] - for r in ranks: + for r in boosts: for cond in r.conditions: if cond.weight is None and r.weight is not None: cond.weight = r.weight From 4c00d798b3e31849ae80026916e0d664d2091196 Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Tue, 2 Jun 2026 09:58:07 +1000 Subject: [PATCH 07/11] Stop Boost.blend() mutating in place --- weaviate/collections/classes/grpc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index b3a125b16..22552ec44 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -1,4 +1,4 @@ -from dataclasses import dataclass +from dataclasses import dataclass, replace from datetime import datetime, timedelta from enum import Enum, auto from typing import ( @@ -520,7 +520,7 @@ def blend( for r in boosts: for cond in r.conditions: if cond.weight is None and r.weight is not None: - cond.weight = r.weight + cond = replace(cond, weight=r.weight) conditions.append(cond) return _Boost(conditions=conditions, weight=weight, depth=depth) From 470f036f5572087e9d71dae1c575304f58f073df Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Tue, 2 Jun 2026 09:59:32 +1000 Subject: [PATCH 08/11] Explicitly set 'now' for time decay --- weaviate/collections/classes/grpc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index 22552ec44..e0ecdb6ba 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -407,7 +407,7 @@ def time_decay( _BoostCondition( time_decay=_TimeDecayFunction( property=property, - origin=_decay_value_to_str(origin) if origin is not None else "", + origin=_decay_value_to_str(origin) if origin is not None else "now", scale=_decay_value_to_str(scale), offset=_decay_value_to_str(offset) if offset is not None else None, curve=curve, From 4489b692500c9eb663cb5525ddafa6c227a1daed Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Tue, 2 Jun 2026 10:17:08 +1000 Subject: [PATCH 09/11] Misc fixes --- integration/test_collection_boost.py | 10 ++++++++++ weaviate/collections/classes/grpc.py | 22 ++++++++++++++++------ 2 files changed, 26 insertions(+), 6 deletions(-) diff --git a/integration/test_collection_boost.py b/integration/test_collection_boost.py index 5be049f25..403e4db14 100644 --- a/integration/test_collection_boost.py +++ b/integration/test_collection_boost.py @@ -3,6 +3,7 @@ from integration.conftest import CollectionFactory from weaviate.classes.query import Boost, Filter, MetadataQuery from weaviate.collections.classes.config import Configure, DataType, Property +from weaviate.exceptions import WeaviateInvalidInputError from weaviate.collections.classes.data import DataObject @@ -266,6 +267,15 @@ def test_boost_api_surface() -> None: assert b.depth == 200 +def test_boost_blend_rejects_sub_boost_depth() -> None: + """blend() raises if any sub-boost has depth set.""" + with pytest.raises(WeaviateInvalidInputError): + Boost.blend( + Boost.property("count", depth=500), + depth=100, + ) + + def test_boost_default_curve_is_unspecified() -> None: """Omitting curve defaults to None (sent as UNSPECIFIED on the wire).""" b = Boost.numeric_decay("price", origin=50.0, scale=20.0) diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index e0ecdb6ba..1f57067bb 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -1,5 +1,5 @@ from dataclasses import dataclass, replace -from datetime import datetime, timedelta +from datetime import datetime, timedelta, timezone from enum import Enum, auto from typing import ( Any, @@ -312,7 +312,7 @@ class _Boost: depth: Optional[int] = None -def _decay_value_to_str(val: Union[str, int, float, timedelta, datetime]) -> str: +def _decay_value_to_str(val: Union[str, timedelta, datetime]) -> str: """Convert a decay parameter value to the string format expected by the server.""" if isinstance(val, timedelta): total_seconds = val.total_seconds() @@ -326,6 +326,8 @@ def _decay_value_to_str(val: Union[str, int, float, timedelta, datetime]) -> str return f"{int(total_seconds)}s" return f"{total_seconds}s" if isinstance(val, datetime): + if val.tzinfo is None: + val = val.replace(tzinfo=timezone.utc) return val.isoformat() return str(val) @@ -370,7 +372,7 @@ def filter( # noqa: A003 Args: filter: The filter condition (same as used in `filters=` parameter). - weight: Blending weight [0,1] controlling how much the boost affects final scores. + weight: Weight controlling how much the boost affects final scores. depth: Number of results to rescore (default 100, max 10000). Higher values improve accuracy at the cost of performance. """ return _Boost(conditions=[_BoostCondition(filter=filter)], weight=weight, depth=depth) @@ -399,7 +401,7 @@ def time_decay( Accepts the same types as scale. curve: Decay curve type: `Boost.Curve.EXPONENTIAL` (default), `Boost.Curve.GAUSSIAN`, or `Boost.Curve.LINEAR`. decay: Score at scale distance from origin (default 0.5). - weight: Blending weight [0,1] controlling how much the boost affects final scores. + weight: Weight controlling how much the boost affects final scores. depth: Number of results to rescore (default 100, max 10000). """ return _Boost( @@ -444,7 +446,7 @@ def numeric_decay( offset: Documents within this distance from origin get full score (default 0). curve: Decay curve type: `Boost.Curve.EXPONENTIAL` (default), `Boost.Curve.GAUSSIAN`, or `Boost.Curve.LINEAR`. decay: Score at scale distance from origin (default 0.5). - weight: Blending weight [0,1] controlling how much the boost affects final scores. + weight: Weight controlling how much the boost affects final scores. depth: Number of results to rescore (default 100, max 10000). """ return _Boost( @@ -484,7 +486,7 @@ def property( # noqa: A003 name: The numeric property name to use as a ranking signal. modifier: Score modifier: `Boost.Modifier.NONE` (default), `Boost.Modifier.LOG1P`, or `Boost.Modifier.SQRT`. Use LOG1P or SQRT to dampen the effect of large value ranges. - weight: Blending weight [0,1] controlling how much the boost affects final scores. + weight: Weight controlling how much the boost affects final scores. depth: Number of results to rescore (default 100, max 10000). """ return _Boost( @@ -515,7 +517,15 @@ def blend( *boosts: Boost objects created via `Boost.filter()`, `Boost.time_decay()`, `Boost.numeric_decay()`, or `Boost.property()`. weight: Overall blending weight [0,1] for combining primary search and boost scores. depth: Number of results to rescore (default 100, max 10000). Higher values improve accuracy at the cost of performance. + + Raises: + WeaviateInvalidInputError: If any sub-boost has `depth` set. Use the top-level `depth` parameter instead. """ + for r in boosts: + if r.depth is not None: + raise WeaviateInvalidInputError( + "Cannot set `depth` on sub-boosts passed to `blend()`. Use the top-level `depth` parameter instead." + ) conditions: List[_BoostCondition] = [] for r in boosts: for cond in r.conditions: From 0788043f5d924ea36e0af84dff97d8ca0bd5df71 Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Tue, 2 Jun 2026 10:18:26 +1000 Subject: [PATCH 10/11] Add BoostReturn --- weaviate/classes/query.py | 2 ++ weaviate/collections/classes/grpc.py | 5 ++++- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/weaviate/classes/query.py b/weaviate/classes/query.py index 2217fd90c..a5dd9282d 100644 --- a/weaviate/classes/query.py +++ b/weaviate/classes/query.py @@ -6,6 +6,7 @@ ) from weaviate.collections.classes.grpc import ( Boost, + BoostReturn, Diversity, GroupBy, HybridFusion, @@ -40,6 +41,7 @@ "QueryReference", "NearVector", "Boost", + "BoostReturn", "Rerank", "Sort", "TargetVectors", diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index 1f57067bb..8e2d44197 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -16,7 +16,7 @@ ) from pydantic import ConfigDict, Field -from typing_extensions import ClassVar, TypeGuard, TypeVar +from typing_extensions import ClassVar, TypeAlias, TypeGuard, TypeVar from weaviate.collections.classes.filters import FilterReturn from weaviate.collections.classes.types import _WeaviateInput @@ -312,6 +312,9 @@ class _Boost: depth: Optional[int] = None +BoostReturn: TypeAlias = _Boost + + def _decay_value_to_str(val: Union[str, timedelta, datetime]) -> str: """Convert a decay parameter value to the string format expected by the server.""" if isinstance(val, timedelta): From 9e311e1710f90c4180a6f3941cef3bab23694654 Mon Sep 17 00:00:00 2001 From: John Trengrove Date: Tue, 2 Jun 2026 10:25:03 +1000 Subject: [PATCH 11/11] Fix validation of blend --- weaviate/collections/classes/grpc.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/weaviate/collections/classes/grpc.py b/weaviate/collections/classes/grpc.py index 8e2d44197..1ed0dc10f 100644 --- a/weaviate/collections/classes/grpc.py +++ b/weaviate/collections/classes/grpc.py @@ -522,8 +522,10 @@ def blend( depth: Number of results to rescore (default 100, max 10000). Higher values improve accuracy at the cost of performance. Raises: - WeaviateInvalidInputError: If any sub-boost has `depth` set. Use the top-level `depth` parameter instead. + WeaviateInvalidInputError: If no boosts are provided or if any sub-boost has `depth` set. """ + if len(boosts) == 0: + raise WeaviateInvalidInputError("Boost.blend() requires at least one boost.") for r in boosts: if r.depth is not None: raise WeaviateInvalidInputError(