rhUdZddlmZddlZddlmZddlmZddl m Z ddl m Z ddl m Z dd l m Z dd l mZdd l mZddlZdd lmZd e_ej(Zej*Zej,Zej.Zdaded<ddZddZGddej8ZGddeZej>Gddej@Z ejBjEe y)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_envcBttjatS)z;Lazily constructs and returns a global IPythonEnv instance.)r rIPythonEnvImplv/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/google/generativeai/notebook/magics.py_get_ipython_envr4s'668 rcBtj|ty)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.) )credsenvN)r authorizer)rs rrr<s5.>.@Arc6eZdZdZej ddZy)AbstractMagicsz"Defines interface to Magics class.ct)zPerform various LLM-related operations. Args: cell_line: String to pass to the MagicsEngine. cell_body: Contents of the cell body. )NotImplementedErrorself cell_line cell_bodys rllmzAbstractMagics.llmJs "##rNr str | Nonerr")__name__ __module__ __qualname____doc__abcabstractmethodr rrrrrGs,$$rrceZdZdZdZddZy) MagicsImplzActual class implementing the magics functionality. We use a separate class to ensure a single, global instance of the magics class. cJtjt|_y)N)r)r MagicsEnginer_engine)rs r__init__zMagicsImpl.__init__\s$116F6HI rcR|xsd}|xsd}|jj||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- execute_cellrs rr zMagicsImpl.llm_s-O O ||((I>>rNr!)r#r$r%r&r.r rrrr*r*Us J ?rr*ceZdZdZdZeddZejddZ ejddZ ejddZ y) Magicsz|Class to register the magic with Colab. Objects of this class delegate all calls to a single, global instance. NcP|jt|_|jS)z.Retrieve global instance of the Magics object.) _instancer*)clss r get_instancezMagics.get_instanceys! == &LCM}}rcLtjj||S)r0)rr)r4r8r rs rr z Magics.llms#""$((9 (RRrc&|j||SNr rs rpalmz Magics.palmxx 9--rc&|j||Sr;r<rs rgeminiz Magics.geminir>r)returnrr!) r#r$r%r&r6 classmethodr8r line_cell_magicr r=r@rrrr4r4nspI   S S .. ..rr4)rAzipython_env.IPythonEnv)rzcredentials.CredentialsrANone)#r& __future__rr' google.authrgoogle.generativeairgenaigoogle.generativeai.notebookrrrr r r IPython IPython.corer USER_AGENT SheetsInputs SheetsOutputspost_process_add_fnpost_process_replace_fnr __annotations__rrABCrr* magics_classr4 get_ipythonregister_magicsrrrrVs# #/7496;5 $(( ** )<<,DD/3 +2B $SWW $??2$.U\\$.$.N%%f-r