o Zh@sUdZddlmZddlZddlmZddlmZddl m Z ddl m Z ddl m Z dd l m Z dd l mZdd l mZddlZdd lmZd e_ejZejZejZejZdaded<dddZd ddZGdddejZGdddeZejGdddej Z e!"e dS)!z,Colab Magics class. Installs %%llm magics. ) annotationsN) credentials)client)gspread_client) ipython_env)ipython_env_impl) magics_engine)post_process_utils) sheets_utils)magiczgenai-py-magiczipython_env.IPythonEnv | None _ipython_envreturnipython_env.IPythonEnvcCstdurtatS)z;Lazily constructs and returns a global IPythonEnv instance.N)r rZIPythonEnvImplrrZ/var/www/html/lang_env/lib/python3.10/site-packages/google/generativeai/notebook/magics.py_get_ipython_env4srcredscredentials.CredentialsNonecCstj|tddS)zSets up credentials. This is used for interacting Google APIs, such as Google Sheets. Args: creds: The credentials that will be used (e.g. to read from Google Sheets.) )renvN)r authorizer)rrrrr<src@s eZdZdZejdddZdS) AbstractMagicsz"Defines interface to Magics class. cell_line str | None cell_bodycCst)zPerform various LLM-related operations. Args: cell_line: String to pass to the MagicsEngine. cell_body: Contents of the cell body. )NotImplementedErrorselfrrrrrllmJszAbstractMagics.llmNrrrr)__name__ __module__ __qualname____doc__abcabstractmethodrrrrrrGsrc@s"eZdZdZddZd ddZd S) MagicsImplzActual class implementing the magics functionality. We use a separate class to ensure a single, global instance of the magics class. cCstjtd|_dS)N)r)rZ MagicsEnginer_engine)rrrr__init__\szMagicsImpl.__init__rrrcCs|pd}|pd}|j||S)Perform various LLM-related operations. Args: cell_line: String to pass to the MagicsEngine. cell_body: Contents of the cell body. Returns: Results from running MagicsEngine. )r'Z execute_cellrrrrr_s zMagicsImpl.llmNr)r r!r"r#r(rrrrrr&Usr&c@sReZdZdZdZedddZejdd d Z ejdd d Z ejdddZ dS)Magicsz|Class to register the magic with Colab. Objects of this class delegate all calls to a single, global instance. Nr rcCs|jdur t|_|jS)z.Retrieve global instance of the Magics object.N) _instancer&)clsrrr get_instanceys zMagics.get_instancerrrcCstj||dS)r))rr)r+r.rrrrrrs z Magics.llmcC |||SNrrrrrpalm z Magics.palmcCr/r0r1rrrrgeminir3z Magics.gemini)r rr) r r!r"r#r, classmethodr.r Zline_cell_magicrr2r4rrrrr+ns   r+)r r)rrr r)#r# __future__rr$Z google.authrZgoogle.generativeairZgenaiZgoogle.generativeai.notebookrrrrr r ZIPythonZ IPython.corer Z USER_AGENTZ SheetsInputsZ SheetsOutputsZpost_process_add_fnZpost_process_replace_fnr __annotations__rrABCrr&Z magics_classr+ get_ipythonZregister_magicsrrrrs4              '