docarray
diff --git a/Collapse file
‎docarray/utils/create_dynamic_doc_class.py‎
Copy file name to clipboardExpand all lines: docarray/utils/create_dynamic_doc_class.py
+8Lines changed: 8 additions & 0 deletions b/Collapse file
‎docarray/utils/create_dynamic_doc_class.py‎
Copy file name to clipboardExpand all lines: docarray/utils/create_dynamic_doc_class.py
+8Lines changed: 8 additions & 0 deletions
diff --git a/Collapse file
‎tests/units/util/test_create_dynamic_code_class.py‎
Copy file name to clipboardExpand all lines: tests/units/util/test_create_dynamic_code_class.py
+6Lines changed: 6 additions & 0 deletions b/Collapse file
‎tests/units/util/test_create_dynamic_code_class.py‎
Copy file name to clipboardExpand all lines: tests/units/util/test_create_dynamic_code_class.py
+6Lines changed: 6 additions & 0 deletions
@@ -6,6 +6,8 @@
 from docarray.utils._internal._typing import safe_issubclass
 
 
+RESERVED_KEYS = ['type', 'anyOf', '$ref', 'additionalProperties', 'allOf', 'items', 'definitions', 'properties', 'default']
+
 def create_pure_python_type_model(model: Any) -> BaseDoc:
     """
     Take a Pydantic model and cast DocList fields into List fields.
@@ -239,5 +241,11 @@ class MyDoc(BaseDoc):
         )
 
     model = create_model(base_doc_name, __base__=BaseDoc, **fields)
+    model.__config__.title = schema.get('title', model.__config__.title)
+
+    for k in RESERVED_KEYS:
+        if k in schema:
+            schema.pop(k)
+    model.__config__.schema_extra = schema
     cached_models[base_doc_name] = model
     return model
@@ -243,6 +243,8 @@ class ResultTestDoc(BaseDoc):
 
 def test_create_with_field_info():
     class CustomDoc(BaseDoc):
+        """Here I have the description of the class"""
+
         a: str = Field(examples=['Example here'], another_extra='I am another extra')
 
     CustomDocCopy = create_pure_python_type_model(CustomDoc)
@@ -256,3 +258,7 @@ class CustomDoc(BaseDoc):
         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'
+    )