
    (HJj                        d dl mZ d dlmZmZ d dlmZmZ  ed       G d d             Ze G d d	             Z	e G d
 d             Z
 G d de      Zy)    )annotations)	dataclassfield)AnyProtocolT)frozenc                      e Zd ZU dZded<   dZded<   dZded<   d	Zded
<   dZded<   dZ	ded<   dZ
ded<   dZded<   dZded<   dZded<   y)TTSBatchOptionsg?floattemperatureg      ?top_p2   inttop_kg?repetition_penaltyi   
max_tokensautostr	lang_codeFboolstreamg       @streaming_interval   max_batch_sizeverboseN)__name__
__module____qualname__r   __annotations__r   r   r   r   r   r   r   r   r        b/Users/ahmed/devFolder/claude-voice/.venv/lib/python3.12/site-packages/mlx_audio/tts/continuous.pyr
   r
      sg    KE5E3O $$JIsFD ##NCGTr!   r
   c                      e Zd ZU ded<   ded<   dZded<   dZded<   dZd	ed
<   dZded<   dZd	ed<   dZ	ded<   dZ
ded<    ee      Zded<   y)TTSBatchItemr   sequence_idr   textNz
str | Nonevoiceinstructzfloat | Nonespeedgenderpitchr   	ref_audioref_textdefault_factorydict[str, Any]extra)r   r   r   r   r'   r(   r)   r*   r+   r,   r-   r   dictr1   r    r!   r"   r$   r$      sb    
IE:HjE<FJE<IsHj!$7E>7r!   r$   c                      e Zd ZU ded<   dZded<   dZded<   dZded	<   dZded
<   dZded<   dZ	ded<   dZ
ded<   dZded<    ee      Zded<   y)TTSBatchEventr   r%   Nr   audioz
int | Nonesample_rater   samplestoken_countFr   doneis_streaming_chunkis_final_chunkzBaseException | Noneerrorr.   r0   metadata)r   r   r   r   r5   r6   r7   r8   r9   r:   r;   r<   r   r2   r=   r    r!   r"   r4   r4   #   si    E3"K"GSKD$$$ ND "&E&$T:Hn:r!   r4   c                  H    e Zd Zedd       Zedd       Zd	dZd
dZddZy)TTSBatchSessionc                     y Nr    selfs    r"   idlezTTSBatchSession.idle2   s    r!   c                     y rA   r    rB   s    r"   available_slotszTTSBatchSession.available_slots5   s    &)r!   c                     y rA   r    )rC   itemss     r"   addzTTSBatchSession.add8   s    cr!   c                     y rA   r    )rC   r%   s     r"   cancelzTTSBatchSession.cancel:   s    r!   c                     y rA   r    rB   s    r"   stepzTTSBatchSession.step<   s    3r!   N)returnr   )rN   r   )rH   zlist[TTSBatchItem]rN   None)r%   r   rN   rO   )rN   zlist[TTSBatchEvent])	r   r   r   propertyrD   rF   rI   rK   rM   r    r!   r"   r?   r?   1   s'     ) )93.r!   r?   N)
__future__r   dataclassesr   r   typingr   r   r
   r$   r4   r?   r    r!   r"   <module>rT      sj    " (   $
 
 
 
8 
8 
8 
; 
; 
;/h /r!   