docs: adding field descriptions to predefined audio document

punndcoder28 · punndcoder28 · commit 106b51cdbc48 · 2023-09-05T22:47:29.000+05:30
diff --git a/docarray/documents/audio.py b/docarray/documents/audio.py
@@ -2,6 +2,8 @@
 
 import numpy as np
 
+from pydantic import Field
+
 from docarray.base_doc import BaseDoc
 from docarray.typing import AnyEmbedding, AudioUrl
 from docarray.typing.bytes.audio_bytes import AudioBytes
@@ -94,11 +96,27 @@ class MultiModalDoc(BaseDoc):
     ```
     """
 
-    url: Optional[AudioUrl]
-    tensor: Optional[AudioTensor]
-    embedding: Optional[AnyEmbedding]
-    bytes_: Optional[AudioBytes]
-    frame_rate: Optional[int]
+    url: Optional[AudioUrl] = Field(
+        description='The url of the audio file.',
+        example='https://github.com/docarray/docarray/blob/main/tests/toydata/hello.mp3?raw=true',
+    )
+    tensor: Optional[AudioTensor] = Field(
+        description='''Tensor object of the audio which be specified to one of
+        `AudioNdArray`, `AudioTorchTensor`, `AudioTensorFlowTensor`''',
+    )
+    embedding: Optional[AnyEmbedding] = Field(
+        description='''Embedding field is used to store tensor objects of type
+        Tensorflow, PyTorch, NumPy and Jax''',
+        example='np.zeros((3, 32, 32))',
+    )
+    bytes_: Optional[AudioBytes] = Field(
+        description='''Bytes object of the image which is an instance of
+        `AudioBytes.''',
+    )
+    frame_rate: Optional[int] = Field(
+        description='''An integer representing the frame rate of the audio.''',
+        example=24,
+    )
 
     @classmethod
     def validate(
