self-referencing object with discriminator produces undefined variables for non-pydantic v1 output type about datamodel-code-generator HOT 1 OPEN

{
    "$id": "schema",
    "definitions": {
        "Foo": {
            "oneOf": [
                {
                    "type": "string"
                },
                {
                    "type": "array",
                    "items": {
                        "$ref": "#/definitions/Foo"
                    }
                }
            ]
        }
    },
    "title": "State",
    "oneOf": [
        { "$ref": "#/definitions/Foo" }
    ]
}

# generated by datamodel-codegen:
#   filename:  schema.json
#   timestamp: 2024-02-23T17:00:22+00:00

from __future__ import annotations

from typing import List, Union

from pydantic import Field, RootModel


class Foo(RootModel[Union[str, List[Foo]]]):
    root: Union[str, List[Foo]]


class State(RootModel[Foo]):
    root: Foo = Field(..., title='State')


Foo.model_rebuild()

# generated by datamodel-codegen:
#   filename:  schema.json
#   timestamp: 2024-02-23T17:00:22+00:00

from __future__ import annotations

from typing import List, Union

from pydantic import Field, RootModel


class Foo(RootModel[Union[str, List["Foo"]]]):
    root: Union[str, List[Foo]]


class State(RootModel[Foo]):
    root: Foo = Field(..., title='State')


Foo.model_rebuild()