fix: fix create pure python class iteratively

JoanFM · JoanFM · commit 2be2796bd90c · 2024-02-15T22:07:01.000+01:00
Signed-off-by: Joan Martinez &lt;joan.fontanals.martinez@jina.ai&gt;
diff --git a/docarray/utils/create_dynamic_doc_class.py b/docarray/utils/create_dynamic_doc_class.py
@@ -67,7 +67,8 @@ class MyDoc(BaseDoc):
         try:
             if safe_issubclass(field, DocList):
                 t: Any = field.doc_type
-                fields[field_name] = (List[t], field_info)
+                t_aux = create_pure_python_type_model(t)
+                fields[field_name] = (List[t_aux], field_info)
             else:
                 fields[field_name] = (field, field_info)
         except TypeError:
diff --git a/tests/units/util/test_create_dynamic_code_class.py b/tests/units/util/test_create_dynamic_code_class.py
@@ -278,10 +278,34 @@ class CustomDoc(BaseDoc):
         'Example here'
     ]
     assert (
-        new_custom_doc_model.schema().get('properties')['a']['another_extra']
-        == 'I am another extra'
+            new_custom_doc_model.schema().get('properties')['a']['another_extra']
+            == 'I am another extra'
     )
     assert (
-        new_custom_doc_model.schema().get('description')
-        == 'Here I have the description of the class'
+            new_custom_doc_model.schema().get('description')
+            == 'Here I have the description of the class'
     )
+
+
+def test_dynamic_class_creation_multiple_doclist_nested():
+    from docarray import BaseDoc, DocList
+
+    class MyTextDoc(BaseDoc):
+        text: str
+
+    class QuoteFile(BaseDoc):
+        texts: DocList[MyTextDoc]
+
+    class SearchResult(BaseDoc):
+        results: DocList[QuoteFile] = None
+
+    textlist = DocList[MyTextDoc]([MyTextDoc(text='hey')])
+    models_created_by_name = {}
+    SearchResult_aux = create_pure_python_type_model(SearchResult)
+    _ = create_base_doc_from_schema(SearchResult_aux.schema(), 'SearchResult',
+                                    models_created_by_name)
+    QuoteFile_reconstructed_in_gateway_from_Search_results = models_created_by_name['QuoteFile']
+
+    reconstructed_in_gateway_from_Search_results = QuoteFile_reconstructed_in_gateway_from_Search_results(
+        texts=textlist)
+    assert reconstructed_in_gateway_from_Search_results.texts[0].text == 'hey'
