o ߭j@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 rIPythonEnvImplrr`/var/www/html/CropPilot/venv/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)r 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) 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.r1rrrrrrs z Magics.llmcC |||SNrrrrrpalm z Magics.palmcCr2r3r4rrrrgeminir6z Magics.gemini)r rr ) r!r"r#r$r/ classmethodr1r line_cell_magicrr5r7rrrrr.ns   r.)r r)rrr r)#r$ __future__rr% google.authrgoogle.generativeairgenaigoogle.generativeai.notebookrrrrr r IPython IPython.corer USER_AGENT SheetsInputs SheetsOutputspost_process_add_fnpost_process_replace_fnr __annotations__rrABCrr' magics_classr. get_ipythonregister_magicsrrrrs4              '