
    6i+
                    \   d dl mZ d dlmZmZmZ d dlZddlmZm	Z	 er	d dl
mZmZ n		 d dl
mZmZ eej                  j                  f	 	 	 	 	 	 	 d
d       Zeej                  j                  f	 	 	 	 	 	 	 dd       Zej                  j                  f	 	 	 	 	 	 	 dd	Zy# e$ r dZdZY yw xY w)    )annotations)TYPE_CHECKINGAnyoverloadN   )AsyncInstructor
Instructor)AsyncFireworks	Fireworksc                     y N clientmodekwargss      p/root/.openclaw/workspace/visionaryfx/venv/lib/python3.12/site-packages/instructor/providers/fireworks/client.pyfrom_fireworksr      s    
     c                     y r   r   r   s      r   r   r      s    
 r   c           
         t         j                  j                  t         j                  j                  h}||vr1ddlm}  |t        |      d|D cg c]  }t        |       c}      t         t        t        f      s$ddlm
}  |dt               j                         t         t              rCd fd}t        d t        j                  ||      t         j                  j                   |d	|S t         t              r[t#        d t        j                   j$                  j&                  j(                  |      t         j                  j                   |d	|S t+        d
      c c}w )Nr   )	ModeErrorr   )r   providervalid_modes)ClientErrorz@Client must be an instance of Fireworks or AsyncFireworks. Got: c                    K   d|v r-|d   du r& j                   j                  j                  | i |S  j                   j                  j                  | i | d {   S 7 w)NstreamT)chatcompletionsacreate)argsr   r   s     r   async_wrapperz%from_fireworks.<locals>.async_wrapper?   se     6!fX&6$&>6v{{..66GGG80088$I&IIIIs   AA%A#A%)creater   )r   r#   r   r   z*Client must be AsyncFireworks or Fireworks)r!   r   r   r   r   )
instructorModeFIREWORKS_TOOLSFIREWORKS_JSONcore.exceptionsr   str
isinstancer
   r   r   type__name__r   patchProvider	FIREWORKSr	   r   r   r#   AssertionError)r   r   r   r   r   mr   r"   s   `       r   r   r   "   s^    	''&&K
 ;0T )45AQ5
 	
 f~y9:2L))*,
 	

 &.)	J
  
##=tD((22	

 
 	
 &)$ 
##6;;+B+B+I+IPTU((22	

 
 	
 E
FFI 6s   E.)r   r   r   instructor.Moder   r   returnr	   )r   r
   r   r2   r   r   r3   r   )r   zFireworks | AsyncFireworksr   r2   r   r   r3   zInstructor | AsyncInstructor)
__future__r   typingr   r   r   r$   core.clientr   r	   fireworks.clientr
   r   ImportErrorr%   r'   r   r   r   r   <module>r9      s   " / /  6::> 
 'OO::
  	 
 
 'OO::
  	 
 'OO::4G&4G
4G 4G "	4G+  	s   B 	B+*B+