
    
!iP
                    r    S r SSKJr  SSKJr  SSKJr  SSKJ	r	  SSK
Jr  Sr " S S	\R                  5      rg)
z!Model that uses the Text service.    )annotations)retryN)generation_types)modelzmodels/gemini-1.5-flashc                  X    \ rS rSrSr   S         SS jjr S	     S
S jjrSrg)	TextModel   z6Concrete model that uses the generate_content service.Nc                    0 nUb  X5S'   Ub  XES'   U=(       d    [         n[        R                  " US9n[        R                  R                  R
                  " S0 UD6nUR                  XS9$ )Ntemperaturecandidate_count)
model_name)generation_config )_DEFAULT_MODELgenaiGenerativeModeltypesr   GenerationConfiggenerate_content)	selfpromptr   r   r   
gen_configr   	gen_modelgcs	            a/home/james-whalen/.local/lib/python3.13/site-packages/google/generativeai/notebook/text_model.py_generate_textTextModel._generate_text   sq     
"(3}%&,;(),n
))Z@	[[))::HZH))&)GG    c                   Uc  [         R                  " 5       n[        R                  " [        R                  5      " U R
                  5      nU" UUR                  UR                  UR                  S9n/ nUR                   H>  nUR                  SR                  S UR                  R                   5       5      5        M@     [         R                  " UUS9$ )N)r   r   r   r    c              3  8   #    U  H  oR                   v   M     g 7fN)text).0ps     r   	<genexpr>'TextModel.call_model.<locals>.<genexpr>C   s     'H1s   )model_inputtext_results)	model_libModelArgumentsr   Retryif_transient_errorr   r   r   r   
candidatesappendjoincontentpartsModelResults)r   r(   
model_argsretryable_fnresponsetext_outputscs          r   
call_modelTextModel.call_model/   s    
 "113J {{5#;#;<T=P=PQ"""..&66	
 $$A'H		'H HI % %%#%
 	
r   r   )NNN)
r   strr   z
str | Noner   zfloat | Noner   z
int | Nonereturnz(generation_types.GenerateContentResponser"   )r(   r;   r4   zmodel_lib.ModelArguments | Noner<   zmodel_lib.ModelResults)__name__
__module____qualname____firstlineno____doc__r   r9   __static_attributes__r   r   r   r   r      sv    @
 !$(&*HH H "	H
 $H 
2H* 7;

 4
 
 	
 
r   r   )rA   
__future__r   google.api_corer   google.generativeaigenerativeair   google.generativeai.typesr    google.generativeai.notebook.libr   r*   r   AbstractModelr   r   r   r   <module>rJ      s0    ( " ! # 6 ?*.
	'' .
r   