rhdZddlmZddlZddlZddlmZejdGddZejGd d Z Gd d ejZ Gd de Z y)zAbstract interface for models.) annotationsN)SequenceT)frozenc<eZdZUdZdZded<dZded<dZded<y) 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 y/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/google/generativeai/notebook/lib/model.pyrrs'E: $K$"&OZ&rrc&eZdZUdZded<ded<y) ModelResultsz0Results from calling AbstractModel.call_model().str model_inputz Sequence[str] text_resultsN)r r r rrrrrrr&s:rrc@eZdZej d ddZy) AbstractModelNcy)zExecutes the model.Nr)selfr model_argss r call_modelzAbstractModel.call_model/srNrrrzModelArguments | Nonereturnr)r r r abcabstractmethodrrrrrr.s5DH"",A" ""rrc&eZdZdZ d ddZy) EchoModelzTModel that returns the original input. This is primarily used for testing. NcL|r |jnd}|d}t||g|zS)N)rr)r r)rrrr s rrzEchoModel.call_model<s99C*44  "O#%8  rrr)r r r rrrrrr$r$6s+ EI    ,A    rr$) r __future__rr! dataclassestypingr dataclassrrABCrr$rrrr,sw%" d# ' '$ '    "CGG"  r