o
    ÔÙ¾iº  ã                   @  s\   d dl mZ d dlmZmZ d dlmZ erd dlmZ G dd„ deƒZ	G dd„ deƒZ
d	S )
é    )Úannotations)ÚABCÚabstractmethod)ÚTYPE_CHECKING)ÚTpModelWorkerc                   @  s$   e Zd Zedd„ ƒZedd„ ƒZdS )ÚBaseDraftWorkerc                   C  ó   d S ©N© r
   r
   r
   ú[/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/speculative/base_spec_worker.pyÚdraft   ó   zBaseDraftWorker.draftc                   C  r   r	   r
   r
   r
   r
   r   Údraft_extend   r   zBaseDraftWorker.draft_extendN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r
   r
   r
   r   r   
   s
    
r   c                   @  s<   e Zd Zeeddd„ƒƒZeeddd„ƒƒZedd	„ ƒZd
S )ÚBaseSpecWorkerÚreturnr   c                 C  r   r	   r
   ©Úselfr
   r
   r   Útarget_worker   ó   zBaseSpecWorker.target_workerr   c                 C  r   r	   r
   r   r
   r
   r   Údraft_worker   r   zBaseSpecWorker.draft_workerc                 C  r   r	   r
   r   r
   r
   r   Úclear_cache_pool   r   zBaseSpecWorker.clear_cache_poolN)r   r   )r   r   )r   r   r   Úpropertyr   r   r   r   r
   r
   r
   r   r      s    r   N)Ú
__future__r   Úabcr   r   Útypingr   Úsglang.srt.managers.tp_workerr   r   r   r
   r
   r
   r   Ú<module>   s    
