U h7@sUdZddlmZddlZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z ddl Z dd lmZe jZe jZe jZe jZdad ed <ddddZdddddZGdddejZGdddeZejGdddejZe edS)z-Colab Magics class. Installs %%palm magics. ) annotationsN) credentials)gspread_client) ipython_env)ipython_env_impl) magics_engine)post_process_utils) sheets_utils)magiczipython_env.IPythonEnv | None _ipython_envzipython_env.IPythonEnvreturncCstdkrtatS)z;Lazily constructs and returns a global IPythonEnv instance.N)r rZIPythonEnvImplrrG/tmp/pip-unpacked-wheel-doshhd5e/google/generativeai/notebook/magics.py_get_ipython_env.srzcredentials.CredentialsNone)credsr cCstj|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)rrrrr6src@s&eZdZdZejdddddZdS)AbstractMagicsz"Defines interface to Magics class. str | None cell_line cell_bodycCs tdS)zPerform various LLM-related operations. Args: cell_line: String to pass to the MagicsEngine. cell_body: Contents of the cell body. N)NotImplementedErrorselfrrrrrpalmDszAbstractMagics.palmN)__name__ __module__ __qualname____doc__abcabstractmethodrrrrrrAsrc@s(eZdZdZddZdddddZdS) 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__VszMagicsImpl.__init__rrcCs|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_cellrrrrrYs zMagicsImpl.palmN)rrr r!r&rrrrrr$Osr$c@s<eZdZdZdZeddddZejdddd d Z dS) MagicszvClass to register the magic with Colab. Objects of this class delegate all calls to a single, global instance. Nrr cCs|jdkrt|_|jS)z.Retrieve global instance of the Magics object.N) _instancer$)clsrrr get_instancess zMagics.get_instancerrcCstj||dS)r'r)r)r,rrrrrrzs z Magics.palm) rrr r!r* classmethodr,r Zline_cell_magicrrrrrr)hs r))r! __future__rr"Z google.authrZgoogle.generativeai.notebookrrrrrr ZIPythonZ IPython.corer Z SheetsInputsZ SheetsOutputsZpost_process_add_fnZpost_process_replace_fnr __annotations__rrABCrr$Z magics_classr)Z get_ipythonZregister_magicsrrrrs.