o
    di                     @   s&   d dl mZ defddZdd ZdS )    )fieldsreturnc                 C   s   d}t | D ]Y}|d|j 7 }t|j}|dr/d}||dd d }|dd }nd	}|d
r>|dd }n|jj}|rM|d| d7 }n|d| d7 }||jd  d7 }q| j	durh| j	ndd d
| | _	| S )z=Class decorator for generate the documentation for dataclass.z
Attributes:
z   ztyping.OptionalT[   .Ftypingz (`z`, *optional*): z`): description
N z

)r   namestrtype
startswithfindsplit__name__metadata__doc__join)clsdoc	attributeattribute_typeoptionaltype_display r   _/lsinfo/ai/hellotax_ai/llm_service/venv_embed/lib/python3.10/site-packages/optimum/utils/doc.pygenerate_doc_dataclass   s$   


$r   c                    s    fdd}|S )Nc                    s.   | j pdd  }|jdi  | _ | S )Nr   r   )r   r   format)fnfunc_docdocstrdynamic_elementstextr   r   docstring_decorator6   s   z2add_dynamic_docstring.<locals>.docstring_decoratorr   )r%   r$   r#   r&   r   r"   r   add_dynamic_docstring1   s   r'   N)dataclassesr   r   r   r'   r   r   r   r   <module>   s   