o 貪g@sdZddlmZddlZddlZddlmZejddGdddZejGd d d Z Gd d d ej Z Gd dde Z dS)zAbstract interface for models.) annotationsN)SequenceT)frozenc@s6eZdZUdZdZded<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__rrrrb/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/google/generativeai/notebook/lib/model.pyrs   rc@s"eZdZUdZded<ded<dS) ModelResultsz0Results from calling AbstractModel.call_model().str model_inputz Sequence[str] text_resultsN)r r r r r rrrrr&s  rc@s eZdZej d d dd ZdS) AbstractModelNrr model_argsModelArguments | NonereturnrcCsdS)zExecutes the model.Nr)selfrrrrr call_model/szAbstractModel.call_modelNrrrrrr)r r r abcabstractmethodrrrrrr.src@seZdZdZ d d d d ZdS) EchoModelzTModel that returns the original input. This is primarily used for testing. NrrrrrrcCs,|r|jnd}|dur d}t||g|dS)N)rr)rr)rrrrrrrr<szEchoModel.call_modelrr)r r r r rrrrrr6sr) r __future__rr dataclassestypingr dataclassrrABCrrrrrrs