[Backport 2.6] support warmup type params for fieldschema (#3255) (#3256)

Backport of #3255 to `2.6`.

Signed-off-by: chasingegg <chao.gao@zilliz.com>
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Gao <chao.gao@zilliz.com>

Author: pymilvus-bot

pymilvus/orm/schema.py

@@ -424,6 +424,9 @@ def add_field(self, field_name: str, datatype: DataType, **kwargs):
             if "mmap_enabled" in kwargs:
                 struct_schema._type_params["mmap_enabled"] = kwargs["mmap_enabled"]
 
+            if "warmup" in kwargs:
+                struct_schema._type_params["warmup"] = kwargs["warmup"]
+
             self._struct_fields.append(struct_schema)
             return self
 
@@ -489,6 +492,9 @@ def __init__(self, name: str, dtype: DataType, description: str = "", **kwargs)
         if "mmap_enabled" in kwargs:
             self._type_params["mmap_enabled"] = kwargs["mmap_enabled"]
 
+        if "warmup" in kwargs:
+            self._type_params["warmup"] = kwargs["warmup"]
+
         for key in ["analyzer_params", "multi_analyzer_params"]:
             if key in self._kwargs and isinstance(self._kwargs[key], dict):
                 self._kwargs[key] = orjson.dumps(self._kwargs[key]).decode(Config.EncodeProtocol)

tests/orm/test_schema.py

@@ -375,6 +375,16 @@ def test_mmap_enabled_param(self):
         field = FieldSchema("vec", DataType.FLOAT_VECTOR, dim=128, mmap_enabled=True)
         assert field._type_params.get("mmap_enabled") is True
 
+    def test_warmup_param(self):
+        """Test warmup type param."""
+        field = FieldSchema("vec", DataType.FLOAT_VECTOR, dim=128, warmup=True)
+        assert field._type_params.get("warmup") is True
+
+    def test_warmup_param_false(self):
+        """Test warmup type param with False value."""
+        field = FieldSchema("vec", DataType.FLOAT_VECTOR, dim=128, warmup=False)
+        assert field._type_params.get("warmup") is False
+
     def test_analyzer_params_dict(self):
         """Test analyzer_params as dict gets serialized."""
         field = FieldSchema(
@@ -658,6 +668,27 @@ def test_add_struct_field_with_mmap(self):
         assert len(schema.struct_fields) == 1
         assert schema.struct_fields[0]._type_params.get("mmap_enabled") is True
 
+    def test_add_struct_field_with_warmup(self):
+        """Test adding struct field with warmup."""
+        struct = StructFieldSchema()
+        struct.add_field("score", DataType.FLOAT)
+        schema = CollectionSchema(
+            [
+                FieldSchema("id", DataType.INT64, is_primary=True),
+                FieldSchema("vec", DataType.FLOAT_VECTOR, dim=128),
+            ]
+        )
+        schema.add_field(
+            "struct",
+            DataType.ARRAY,
+            element_type=DataType.STRUCT,
+            struct_schema=struct,
+            max_capacity=10,
+            warmup=True,
+        )
+        assert len(schema.struct_fields) == 1
+        assert schema.struct_fields[0]._type_params.get("warmup") is True
+
 
 class TestCollectionSchemaToDict:
     """Tests for CollectionSchema to_dict and construct_from_dict methods."""

tests/test_client_abstract.py

@@ -156,6 +156,28 @@ def test_field_schema_with_mmap_disabled(self):
 
         assert field.params["mmap_enabled"] is False
 
+    def test_field_schema_with_warmup_enabled(self):
+        """Test FieldSchema with warmup type parameter."""
+        warmup_param = MagicMock()
+        warmup_param.key = "warmup"
+        warmup_param.value = "true"
+
+        raw = self._create_mock_raw_field(type_params=[warmup_param])
+        field = FieldSchema(raw)
+
+        assert field.params["warmup"] == "true"
+
+    def test_field_schema_with_warmup_disabled(self):
+        """Test FieldSchema with warmup set to false."""
+        warmup_param = MagicMock()
+        warmup_param.key = "warmup"
+        warmup_param.value = "false"
+
+        raw = self._create_mock_raw_field(type_params=[warmup_param])
+        field = FieldSchema(raw)
+
+        assert field.params["warmup"] == "false"
+
     def test_field_schema_with_json_params(self):
         """Test FieldSchema with JSON type params."""
         params_param = MagicMock()