fix: handle missing __validators__ (#68)

#67 Signed-off-by: Gabe Goodhart <[email protected]>
jina-ai · Mar 4, 2024 · f0200e5 · f0200e5
1 parent 53cd4a5
commit f0200e5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vectordb/utils/create_doc_type.py b/vectordb/utils/create_doc_type.py
@@ -15,7 +15,8 @@ def create_output_doc_type(input_doc_type: Type['BaseDoc']):
     return create_model(
         input_doc_type.__name__ + 'WithMatchesAndScores',
         __base__=input_doc_type,
-        __validators__=input_doc_type.__validators__,
+        # NOTE: With pydantic>=2, __validators__ does not exist
+        __validators__=getattr(input_doc_type, "__validators__", None),
         matches=(DocList[input_doc_type], []),
         scores=(List[float], [])
     )