llama-index

1
try:
2
    import pydantic.v1 as pydantic
3
    from pydantic.v1 import (
4
        BaseConfig,
5
        BaseModel,
6
        Field,
7
        PrivateAttr,
8
        StrictFloat,
9
        StrictInt,
10
        StrictStr,
11
        create_model,
12
        root_validator,
13
        validator,
14
    )
15
    from pydantic.v1.error_wrappers import ValidationError
16
    from pydantic.v1.fields import FieldInfo
17
    from pydantic.v1.generics import GenericModel
18
except ImportError:
19
    import pydantic  # type: ignore
20
    from pydantic import (
21
        BaseConfig,
22
        BaseModel,
23
        Field,
24
        PrivateAttr,
25
        StrictFloat,
26
        StrictInt,
27
        StrictStr,
28
        create_model,
29
        root_validator,
30
        validator,
31
    )
32
    from pydantic.error_wrappers import ValidationError
33
    from pydantic.fields import FieldInfo
34
    from pydantic.generics import GenericModel
35

36
__all__ = [
37
    "pydantic",
38
    "BaseModel",
39
    "Field",
40
    "PrivateAttr",
41
    "root_validator",
42
    "validator",
43
    "create_model",
44
    "StrictFloat",
45
    "StrictInt",
46
    "StrictStr",
47
    "FieldInfo",
48
    "ValidationError",
49
    "GenericModel",
50
    "BaseConfig",
51
]
52