阺hdZddlmZddlZddlZddlmZejdGddZejGd d Z Gd d ej Z Gd de Z dS)zAbstract interface for models.) annotationsN)SequenceT)frozenc>eZdZUdZdZded<dZded<dZded<dS) ModelArgumentsaCommon arguments for models. Attributes: model: The model string to use. If None a default model will be selected. temperature: The temperature. Must be greater-than-or-equal-to zero. candidate_count: Number of candidates to return. Nz str | Nonemodelz float | None temperaturez int | Nonecandidate_count)__name__ __module__ __qualname____doc__r__annotations__r r n/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/google/generativeai/notebook/lib/model.pyrrsQE $K$$$$"&O&&&&&&rrc(eZdZUdZded<ded<dS) ModelResultsz0Results from calling AbstractModel.call_model().str model_inputz Sequence[str] text_resultsN)r r r rrrrrrr&s0::rrc4eZdZej d d dZdS) AbstractModelNrr model_argsModelArguments | NonereturnrcdS)zExecutes the model.Nr)selfrrs r call_modelzAbstractModel.call_model/srNrrrrrr)r r r abcabstractmethodrrrrrr.s>DH"""""""rrceZdZdZ d d d ZdS) EchoModelzTModel that returns the original input. This is primarily used for testing. NrrrrrrcJ|r|jnd}|d}t||g|zS)N)rr)r r)rrrr s rrzEchoModel.call_model<sF9CL*44  "O#%8    rr r!)r r r rrrrrr%r%6s> EI        rr%) r __future__rr" dataclassestypingr dataclassrrABCrr%rrrr-s#%$"""""" d### ' ' ' ' ' ' '$# '         """""CG"""          r