U h@sdZddlmZddlZddlZddlmZejddGdddZejGd d d Z Gd d d ej Z Gd dde Z dS)zAbstract interface for models.) annotationsN)SequenceT)frozenc@s6eZdZUdZdZded<dZded<dZded<dS) ModelArgumentszCommon 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__rrrrJ/tmp/pip-unpacked-wheel-doshhd5e/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&eZdZejdddddddZdS) AbstractModelNrModelArguments | Nonerr model_argsreturncCsdS)zExecutes the model.Nr)selfrrrrr call_model0szAbstractModel.call_model)N)r r r abcabstractmethodrrrrrr.src@s$eZdZdZd ddddddZdS) EchoModelzPModel that returns the original input. This is primarily used for testing. NrrrrcCs,|r |jnd}|dkrd}t||g|dS)N)rr)rr)rrrrrrrr=szEchoModel.call_model)N)r r r r rrrrrr7sr) r __future__rrZ dataclassestypingrZ dataclassrrABCrrrrrrs