o €j6hã@s`dZddlmZddlZddlmZddlmZddlm Z Gdd„dej ƒZ Gd d „d ƒZ dS) z=Maintains set of LLM models that can be instantiated by name.é)Ú annotationsN)ÚCallable)Ú text_model)Úmodelc@seZdZdZdZdS)Ú ModelNameÚechoÚtextN)Ú__name__Ú __module__Ú __qualname__Ú ECHO_MODELÚ TEXT_MODEL©rrúz/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/google/generativeai/notebook/model_registry.pyrsrc@s(eZdZdZejZdd„Zd dd „Zd S) Ú ModelRegistryz-Registry that instantiates and caches models.cCs i|_tjtjtjtji|_dS)N) Ú _model_cacherr Ú model_libÚ EchoModelr rÚ TextModelÚ_model_constructors)ÚselfrrrÚ__init__#s þzModelRegistry.__init__Ú model_namerÚreturnúmodel_lib.AbstractModelcCs&||jvr|j|ƒ|j|<|j|S)aGiven `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)rrrrrÚ get_model*s  zModelRegistry.get_modelN)rrrr) r r r Ú__doc__rr Ú DEFAULT_MODELrrrrrrrs r)rÚ __future__rÚenumÚtypingrÚgoogle.generativeai.notebookrÚ google.generativeai.notebook.librrÚEnumrrrrrrÚs