o ;iX @s\dZddlmZmZmZddlmZddlmZddl m Z eGdddeZ d d Z d S) z{This module provides Jupyter/IPython extensions for Modal. Use in a notebook with `%load_ext modal.experimental.ipython`. )Magics line_magic magics_class)Cls) NotFoundError)Functionc@seZdZeddZdS) ModalMagicsc Cs|}|dstddS|dd}|d\}}}|s%tddSd|vr/d|}}n|dd\}}d d |d D}|D]~} | sHqC| } t| d krSqC| d } | } t| d krj| ddkrj| d} nt| dkr{td| ddSztj|| |d} | Wnt yt j|| |d} | Ynw| |j j | <|rtd| d|d|dqCtd| d|dqCdS)a1Lookup a deployed Modal Function or Class. **Example:** ```python notest %modal from my-app import my_function, MyClass as Foo # Now you can call my_function() and Foo from your notebook. my_function.remote() Foo().my_method.remote() ``` zfrom zcInvalid syntax. Use: %modal from [env/] import [, [as alias]]Nz import z)Invalid syntax. Missing 'import' keyword./cSsg|]}|qS)strip).0itemr r N/home/ubuntu/.local/lib/python3.10/site-packages/modal/experimental/ipython.py 3sz%ModalMagics.modal..,rasrzInvalid syntax in import item: z.. Expected format: [as alias])environment_namezLoaded z from environment z and app .z from app ) r startswithprint partitionsplitlenr from_namehydraterrshelluser_ns)selflineline_without_from env_app_partsep import_part environmentapp import_itemsrparts model_namealiasobjr r rmodalsL          zModalMagics.modalN)__name__ __module__ __qualname__rr.r r r rr sr cCs|tdS)N)register_magicsr )ipythonr r rload_ipython_extensionSsr4N) __doc__IPython.core.magicrrrclsr exceptionr functionsrr r4r r r rs    C