diff --git a/src/mistralai/utils/serializers.py b/src/mistralai/utils/serializers.py index 378a14c..65f2641 100644 --- a/src/mistralai/utils/serializers.py +++ b/src/mistralai/utils/serializers.py @@ -90,10 +90,10 @@ def serialize(i): def validate_int(b): - if b is None: - return None + if b is None or isinstance(b, int): + return b - if isinstance(b, (int, Unset)): + if b is Unset: return b if not isinstance(b, str): @@ -178,7 +178,7 @@ def is_nullable(field): if origin is Nullable or origin is OptionalNullable: return True - if not origin is Union or type(None) not in get_args(field): + if origin is not Union or type(None) not in get_args(field): return False for arg in get_args(field):