fix: move UnionMode import outside TYPE_CHECKING for Pydantic runtime… (#2950)

* fix: move UnionMode import outside TYPE_CHECKING for Pydantic runtime access

The UnionMode type is used in GenerateConfig.union_mode field annotation.
Pydantic needs to access this type at runtime for model validation, but it
was only imported inside TYPE_CHECKING block, causing:

PydanticUserError: GenerateConfig is not fully defined; you should define
UnionMode, then call GenerateConfig.model_rebuild().

This follows the same pattern as other runtime-required imports in this file
(Path, DataModel, DataModelFieldBase, etc.) which have noqa: TC001 comments.

* test: add regression test for UnionMode runtime access

Add test that verifies UnionMode is available at runtime for Pydantic
model_rebuild(). This test runs in a subprocess to ensure a clean namespace
without pytest's import side effects, properly detecting when UnionMode
is incorrectly placed inside TYPE_CHECKING block.

---------

Co-authored-by: Koudai Aono <[email protected]>

Author: phil65

src/datamodel_code_generator/config.py

@@ -44,17 +44,14 @@
     DataModel,
     DataModelFieldBase,
 )
+from datamodel_code_generator.model.pydantic_v2 import UnionMode  # noqa: TC001 - used by Pydantic at runtime
 from datamodel_code_generator.model.scalar import DataTypeScalar
 from datamodel_code_generator.model.union import DataTypeUnion
 from datamodel_code_generator.parser import DefaultPutDict, LiteralType
 from datamodel_code_generator.types import DataTypeManager, StrictTypes  # noqa: TC001 - used by Pydantic at runtime
 from datamodel_code_generator.util import ConfigDict, is_pydantic_v2
 from datamodel_code_generator.validators import ModelValidators  # noqa: TC001 - used by Pydantic at runtime
 
-if TYPE_CHECKING:
-    from datamodel_code_generator.model.pydantic_v2 import UnionMode
-
-
 CallableSchema = Callable[[str], str]
 DumpResolveReferenceAction = Callable[[Iterable[str]], str]
 DefaultPutDictSchema = DefaultPutDict[str, str]

tests/data/expected/main/generate_config_union_mode.py

@@ -0,0 +1,10 @@
+# generated by datamodel-codegen:
+#   filename:  <stdin>
+
+from __future__ import annotations
+
+from pydantic import BaseModel, Field
+
+
+class Model(BaseModel):
+    value: str | int | None = Field(None, union_mode='left_to_right')

tests/main/test_main_general.py

@@ -26,6 +26,7 @@
 from datamodel_code_generator.arguments import _dataclass_arguments
 from datamodel_code_generator.config import GenerateConfig
 from datamodel_code_generator.format import CodeFormatter, PythonVersion
+from datamodel_code_generator.model.pydantic_v2 import UnionMode
 from datamodel_code_generator.parser.openapi import OpenAPIParser
 from tests.conftest import assert_output, create_assert_file_content, freeze_time
 from tests.main.conftest import (
@@ -2062,6 +2063,21 @@ def test_generate_with_config_object(output_file: Path) -> None:
     assert "user_name" in content
 
 
+@pytest.mark.skipif(pydantic.VERSION < "2.0.0", reason="GenerateConfig requires Pydantic v2")
+def test_generate_config_with_union_mode() -> None:
+    """Test GenerateConfig with union_mode field."""
+    config = GenerateConfig(
+        output_model_type=DataModelType.PydanticV2BaseModel,
+        union_mode=UnionMode.left_to_right,
+        disable_timestamp=True,
+    )
+    result = generate(
+        input_='{"type": "object", "properties": {"value": {"anyOf": [{"type": "string"}, {"type": "integer"}]}}}',
+        config=config,
+    )
+    assert_output(result, EXPECTED_MAIN_PATH / "generate_config_union_mode.py")
+
+
 @pytest.mark.skipif(pydantic.VERSION < "2.0.0", reason="GenerateConfig requires Pydantic v2")
 def test_generate_with_config_and_kwargs_raises_error(output_file: Path) -> None:
     """Test generate() raises error when both config and kwargs are provided."""