o
    H!i                     @   sr   d dl mZ d dlmZ d dlmZmZmZ ddlm	Z	 G dd dedd	Z
e	fd
eeef dede
fddZdS )    )PathLike)Path)Tuple	TypedDictUnion   )MODEL_FILE_PREFIXc                   @   s.   e Zd ZU eeef ed< eed< eed< dS )
ModelPathspaddleonnxomN)__name__
__module____qualname__r   r   __annotations__ r   r   o/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddlex/inference/utils/model_paths.pyr	      s   
 r	   F)total	model_dirmodel_file_prefixreturnc                 C   s   t | } i }d }| | d  r| | d }n| | d  r)| | d }|r?| | d  r?|| | d f|d< | | d  rQ| | d |d< | | d  rc| | d |d< |S )	Nz.jsonz.pdmodelz
.pdiparamsr
   z.onnxr   z.omr   )r   exists)r   r   Zmodel_pathsZpd_model_pathr   r   r   get_model_paths   s    r   N)osr   pathlibr   typingr   r   r   	constantsr   r	   strr   r   r   r   r   <module>   s   
