
    6iL                    "   d dl mZ d dlmZmZ d dlZddlmZmZ 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)    )annotations)AnyoverloadN   )AsyncInstructor
Instructor)CerebrasAsyncCerebrasc                     y N clientmodekwargss      o/root/.openclaw/workspace/visionaryfx/venv/lib/python3.12/site-packages/instructor/providers/cerebras/client.pyfrom_cerebrasr      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              r]| j                  j                  j                  }t!        d	| t        j"                  ||      t         j$                  j&                  |d|S | j                  j                  j                  }t)        d	| t        j"                  ||      t         j$                  j&                  |d|S c c}w )
Nr   )	ModeErrorr	   )r   providervalid_modes)ClientErrorz>Client must be an instance of Cerebras or AsyncCerebras. Got: )creater   )r   r   r   r   r   )
instructorModeCEREBRAS_TOOLSCEREBRAS_JSONcore.exceptionsr   str
isinstancer	   r
   r   type__name__chatcompletionsr   r   patchProviderCEREBRASr   )r   r   r   r   r   mr   r   s           r   r   r      sU    	&&%%K
 ;0T)45AQ5
 	
 fx782L))*,
 	

 &-(((// 
##6=((11	

 
 	
 [[$$++F vD9$$--	
  - 6s   E.)r   r	   r   instructor.Moder   r   returnr   )r   r
   r   r+   r   r   r,   r   )r   zCerebras | AsyncCerebrasr   r+   r   r   r,   zInstructor | AsyncInstructor)
__future__r   typingr   r   r   core.clientr   r   cerebras.cloud.sdkr	   r
   r   r   r   r   r   r   <module>r1      s    "    6 7 
 'OO::
  	 
 
 'OO::
  	 
 'OO::,$,
, , "	,r   