[Backport 3.0] fix: allow null auto_id primary column in DataFrame insert (#3673) (#3678)

Backport of #3673 to `3.0`.

Signed-off-by: VectorPeak <73048950+VectorPeak@users.noreply.github.com>
Signed-off-by: chatgpt-codex-connector[bot] <199175422+chatgpt-codex-connector[bot]@users.noreply.github.com>
Signed-off-by: pymilvus-bot <pymilvus-bot@users.noreply.github.com>
Co-authored-by: VectorPeak <garrufariw@gmail.com>
Co-authored-by: chatgpt-codex-connector[bot] <199175422+chatgpt-codex-connector[bot]@users.noreply.github.com>

Author: 3 people

pymilvus/orm/schema.py

@@ -1328,7 +1328,7 @@ def check_insert_schema(schema: CollectionSchema, data: Union[List[List], pd.Dat
             msg = f"Expect no data for auto_id primary field: {schema.primary_field.name}"
             raise DataNotMatchException(message=msg)
         columns = list(data.columns)
-        columns.remove(schema.primary_field)
+        columns.remove(schema.primary_field.name)
         data = data[columns]
 
     tmp_fields = list(

tests/unit/orm/test_schema.py

@@ -1756,6 +1756,18 @@ def test_check_insert_schema_auto_id_with_data(self):
         with pytest.raises(DataNotMatchException, match="auto_id"):
             check_insert_schema(schema, insert_data)
 
+    def test_check_insert_schema_auto_id_with_null_pk_column(self):
+        """Test auto_id DataFrame inserts tolerate an all-null pk column."""
+        schema = CollectionSchema(
+            [
+                FieldSchema("id", DataType.INT64, is_primary=True, auto_id=True),
+                FieldSchema("vec", DataType.FLOAT_VECTOR, dim=2),
+            ]
+        )
+        insert_data = pd.DataFrame({"id": [None, None], "vec": [[1.0, 2.0], [3.0, 4.0]]})
+
+        check_insert_schema(schema, insert_data)
+
 
 class TestCheckUpsertSchema:
     """Tests for check_upsert_schema function."""