
    5P@i	                    n    d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z  G d dej                        Zy	)
z!Model that uses the Text service.    )annotations)retry)text)
text_types)modelc                  H    e Zd ZdZ	 	 	 d	 	 	 	 	 	 	 	 	 ddZ	 d	 	 	 	 	 ddZy)		TextModelz*Concrete model that uses the Text service.Nc                Z    |||d<   |||d<   |||d<   t        j                  dd|i|S )Nr   temperaturecandidate_countprompt )r   generate_text)selfr   r   r   r   kwargss         q/home/www/lebenam.kofcorporation.com/venv/lib/python3.12/site-packages/google/generativeai/notebook/text_model.py_generate_textzTextModel._generate_text   sM     #F7O"$/F=!&(7F$%!!::6::    c                V   |t        j                         } t        j                  t        j                        | j
                        } |||j                  |j                  |j                        }t        j                  ||j                  D cg c]  }|d   	 c}      S c c}w )N)r   r   r   r   output)model_inputtext_results)	model_libModelArgumentsr   Retryif_transient_errorr   r   r   r   ModelResults
candidates)r   r   
model_argsretryable_fnresponsexs         r   
call_modelzTextModel.call_model+   s    
 "113J =u{{5#;#;<T=P=PQ"""..&66	
 %%#/7/B/BC!!H+C
 	
Cs   B&
)NNN)
r   strr   z
str | Noner   zfloat | Noner   z
int | Nonereturnztext_types.Completion)N)r   r$   r   zmodel_lib.ModelArguments | Noner%   zmodel_lib.ModelResults)__name__
__module____qualname____doc__r   r#   r   r   r   r	   r	      sk    4
 !$(&*;; ; "	;
 $; 
;& 7;

 4
 
 	
r   r	   N)r)   
__future__r   google.api_corer   google.generativeair   google.generativeai.typesr    google.generativeai.notebook.libr   r   AbstractModelr	   r   r   r   <module>r0      s+    ( " ! $ 0 ?(
	'' (
r   