Add muvera config

weaviate/collections/classes/config.py

@@ -33,6 +33,7 @@
 )
 from weaviate.collections.classes.config_vector_index import (
     VectorFilterStrategy,
+    _MuveraConfigCreate,
     _MultiVectorConfigCreate,
     _QuantizerConfigCreate,
     _VectorIndexConfigCreate,
@@ -1558,6 +1559,7 @@ class _SQConfig(_ConfigBase):
 
 @dataclass
 class _MultiVectorConfig(_ConfigBase):
+    muvera_config: Optional[_MuveraConfigCreate]
     aggregation: str
 
 
@@ -2034,12 +2036,28 @@ def __add_props(
 class _VectorIndexMultiVector:
     @staticmethod
     def multi_vector(
+        muvera_config: Optional[_MuveraConfigCreate] = None,
         aggregation: Optional[MultiVectorAggregation] = None,
     ) -> _MultiVectorConfigCreate:
         return _MultiVectorConfigCreate(
+            muveraConfig=muvera_config if muvera_config is not None else None,
             aggregation=aggregation.value if aggregation is not None else None,
         )
 
+    @staticmethod
+    def muvera_config(
+        enabled: Optional[bool] = None,
+        ksim: Optional[int] = None,
+        dprojections: Optional[int] = None,
+        repetitions: Optional[int] = None,
+    ) -> _MuveraConfigCreate:
+        return _MuveraConfigCreate(
+            enabled=enabled,
+            ksim=ksim,
+            dprojections=dprojections,
+            repetitions=repetitions,
+        )
+
 
 class _VectorIndexQuantizer:
     @staticmethod

weaviate/collections/classes/config_methods.py

@@ -155,6 +155,7 @@ def __get_multivector(config: Dict[str, Any]) -> Optional[_MultiVectorConfig]:
         if config.get("multivector") is None
         or not config.get("multivector", {"enabled": False}).get("enabled")
         else _MultiVectorConfig(
+            muvera_config=config["multivector"]["muveraConfig"],
             aggregation=config["multivector"]["aggregation"],
         )
     )
@@ -244,6 +245,7 @@ def __get_vector_config(
                 ),
                 vector_index_config=vector_index_config,
             )
+
         return named_vectors
     else:
         return None

weaviate/collections/classes/config_vector_index.py

@@ -42,7 +42,15 @@ class _MultiVectorConfigCreateBase(_ConfigCreateModel):
     enabled: bool = Field(default=True)
 
 
+class _MuveraConfigCreate(_ConfigCreateModel):
+    enabled: Optional[bool]
+    ksim: Optional[int]
+    dprojections: Optional[int]
+    repetitions: Optional[int]
+
+
 class _MultiVectorConfigCreate(_MultiVectorConfigCreateBase):
+    muveraConfig: Optional[_MuveraConfigCreate]
     aggregation: Optional[str]