From b638a1a42a68db295708624dc76d508a1780c255 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kacper=20Kali=C5=84ski?=
 <47140412+KaQuMiQ@users.noreply.github.com>
Date: Wed, 17 Apr 2024 13:13:24 +0200
Subject: [PATCH] Fix type alias validation

---
 src/draive/parameters/validation.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/draive/parameters/validation.py b/src/draive/parameters/validation.py
index c21dd88..10abf71 100644
--- a/src/draive/parameters/validation.py
+++ b/src/draive/parameters/validation.py
@@ -8,7 +8,7 @@
 from collections import abc as collections_abc
 from collections.abc import Callable, Iterable
 from dataclasses import is_dataclass
-from typing import Any, ForwardRef, Protocol, TypeVar, get_args, get_origin
+from typing import Any, ForwardRef, Protocol, TypeAliasType, TypeVar, get_args, get_origin
 
 import typing_extensions
 
@@ -941,6 +941,12 @@ def validated_missing(value: Any) -> Any:
             else:
                 return _any_validator(verifier=verifier)
 
+        case type_alias if isinstance(type_alias, TypeAliasType):
+            return _class_instance_validator(
+                annotation=type_alias.__value__,
+                verifier=verifier,
+            )
+
         case class_type if inspect.isclass(class_type):
             return _class_instance_validator(
                 annotation=class_type,