阺hdZddlmZddlZddlmZddlmZddlm Z Gddej Z Gd d Z dS) z=Maintains set of LLM models that can be instantiated by name.) annotationsN)Callable) text_model)modelceZdZdZdZdS) ModelNameechotextN)__name__ __module__ __qualname__ ECHO_MODEL TEXT_MODELs/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/google/generativeai/notebook/model_registry.pyrrsJJJJrrc.eZdZdZejZdZd dZdS) ModelRegistryz-Registry that instantiates and caches models.czi|_tjtjtjt ji|_dS)N) _model_cacherr model_lib EchoModelrr TextModel_model_constructors)selfs r__init__zModelRegistry.__init__#s1FH  )"5  *"6\    r model_namerreturnmodel_lib.AbstractModelch||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)rrs r get_modelzModelRegistry.get_model*s= T. . .,PD,DZ,P,R,RD j ) ,,rN)rrrr) r r r __doc__rr DEFAULT_MODELrr!rrrrrsE77(M    - - - - - -rr)r" __future__renumtypingrgoogle.generativeai.notebookr google.generativeai.notebook.librrEnumrrrrrr*sDC"""""" 333333??????  ----------r