!i~SrSSKJr SSKrSSKJr SSKJr SSKJ r "SS\R5r "S S 5r g) z=Maintains set of LLM models that can be instantiated by name.) annotationsN)Callable) text_model)modelc\rSrSrSrSrSrg) ModelNameechotextN)__name__ __module__ __qualname____firstlineno__ ECHO_MODEL TEXT_MODEL__static_attributes__r e/home/james-whalen/.local/lib/python3.13/site-packages/google/generativeai/notebook/model_registry.pyrrs JJrrc@\rSrSrSr\R rSrSSjr Sr g) ModelRegistryz-Registry that instantiates and caches models.c0Ul[R[R[R [ R0Ulg)N) _model_cacherr model_lib EchoModelrr TextModel_model_constructors)selfs r__init__ModelRegistry.__init__#s7FH  )"5"5  *"6"6\  rcXR;a!URU"5URU'URU$)zGiven `model_name`, return the corresponding Model instance. Model instances are cached and reused for the same `model_name`. Args: model_name: The name of the model. Returns: The corresponding model instance for `model_name`. rr)r model_names r get_modelModelRegistry.get_model*s? .. .,0,D,DZ,P,RD  j )  ,,rr#N)r$rreturnzmodel_lib.AbstractModel) r rrr__doc__rr DEFAULT_MODELr r%rr rrrrs7((M  -rr)r( __future__renumtypingrgoogle.generativeai.notebookr google.generativeai.notebook.librrEnumrrr rrr0s4D" 3?  --r