ipykernel-helper 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

ipykernel_helper/__init__.py

@@ -1,2 +1,2 @@
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 from .core import *

ipykernel_helper/_modidx.py

@@ -5,5 +5,14 @@ d = { 'settings': { 'branch': 'main',
                 'doc_host': 'https://AnswerDotAI.github.io',
                 'git_url': 'https://github.com/AnswerDotAI/ipykernel-helper',
                 'lib_path': 'ipykernel_helper'},
-  'syms': { 'ipykernel_helper.core': { 'ipykernel_helper.core.safe_repr': ('core.html#safe_repr', 'ipykernel_helper/core.py'),
+  'syms': { 'ipykernel_helper.core': { 'ipykernel_helper.core.AsyncKernelClient.interactive': ( 'core.html#asynckernelclient.interactive',
+                                                                                                'ipykernel_helper/core.py'),
+                                       'ipykernel_helper.core._get_schema': ('core.html#_get_schema', 'ipykernel_helper/core.py'),
+                                       'ipykernel_helper.core._rank': ('core.html#_rank', 'ipykernel_helper/core.py'),
+                                       'ipykernel_helper.core._signatures': ('core.html#_signatures', 'ipykernel_helper/core.py'),
+                                       'ipykernel_helper.core.get_schemas': ('core.html#get_schemas', 'ipykernel_helper/core.py'),
+                                       'ipykernel_helper.core.get_vars': ('core.html#get_vars', 'ipykernel_helper/core.py'),
+                                       'ipykernel_helper.core.ranked_complete': ('core.html#ranked_complete', 'ipykernel_helper/core.py'),
+                                       'ipykernel_helper.core.safe_repr': ('core.html#safe_repr', 'ipykernel_helper/core.py'),
+                                       'ipykernel_helper.core.sig_help': ('core.html#sig_help', 'ipykernel_helper/core.py'),
                                        'ipykernel_helper.core.user_items': ('core.html#user_items', 'ipykernel_helper/core.py')}}}

ipykernel_helper/core.py

@@ -3,12 +3,23 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/00_core.ipynb.
 
 # %% auto 0
-__all__ = ['safe_repr', 'user_items']
+__all__ = ['safe_repr', 'user_items', 'ranked_complete', 'sig_help', 'get_vars', 'get_schemas']
 
 # %% ../nbs/00_core.ipynb
-import typing
+from fastcore.meta import delegates
+from fastcore.utils import patch,dict2obj
+import typing,warnings
 from types import ModuleType, FunctionType, MethodType, BuiltinFunctionType
-from inspect import signature
+from inspect import signature, currentframe
+from functools import cmp_to_key,partial
+from toolslm.funccall import *
+
+from jupyter_client import AsyncKernelClient
+from IPython.core.completer import ProvisionalCompleterWarning
+from jedi import Interpreter, Script as jscript
+
+# %% ../nbs/00_core.ipynb
+warnings.filterwarnings('ignore', category=ProvisionalCompleterWarning)
 
 # %% ../nbs/00_core.ipynb
 def safe_repr(obj, max_len=200):
@@ -19,8 +30,9 @@ def safe_repr(obj, max_len=200):
     except Exception as e: return f"<repr error: {str(e)}>"
 
 # %% ../nbs/00_core.ipynb
-def user_items(ns, max_len=200, xtra_skip=()):
+def user_items(ns=None, max_len=200, xtra_skip=()):
     "Get user-defined vars & funcs from namespace."
+    if ns is None: ns = currentframe().f_back.f_globals
     ignore = {'In', 'Out', 'exit', 'quit', 'open', 'nbmeta', 'receive_nbmeta'}
     ignore.add(xtra_skip)
     rm_types = (
@@ -29,8 +41,77 @@ def user_items(ns, max_len=200, xtra_skip=()):
         getattr(typing, '_GenericAlias', ()),
         getattr(typing, '_SpecialForm', ())
     )
-    user_items = {k: v for k, v in ns.items() if not (k.startswith('_') or k in ignore)}
-    user_vars = {k:safe_repr(v, max_len=max_len) for k, v in user_items.items() if not isinstance(v, rm_types)}
+    user_items = {k:v for k, v in ns.items()
+                  if not k in ignore}
+    user_vars = {k:safe_repr(v, max_len=max_len)
+                 for k, v in user_items.items() if not k.startswith('_') and not isinstance(v, rm_types)}
     user_fns = {k:str(signature(v)) for k, v in user_items.items()
                 if isinstance(v, FunctionType) and v.__module__ == '__main__' and not k.startswith('__')}
     return user_vars,user_fns
+
+# %% ../nbs/00_core.ipynb
+@patch
+@delegates(AsyncKernelClient.execute_interactive)
+async def interactive(self:AsyncKernelClient, code:str, **kwargs):
+    "Like AsyncKernelClient.execute_interactive, but also returns outputs"
+    outputs = []
+    def _h(msg: dict): outputs.append(msg)
+    res = await self.execute_interactive(code, output_hook=_h, **kwargs)
+    return res,outputs
+
+# %% ../nbs/00_core.ipynb
+def _rank(c, s):
+    "Rank a completion `c` for text `s` with namespace `ns`."
+    parts = s.split('.')
+    is_public = not c.text.startswith('_')
+    if c.type=='param': r=1
+    elif c.mod=='__main__': r=2 # local
+    elif len(parts)>1 and parts[0]==c.mod: r=3 # module
+    elif c.mod=='builtins': r=4
+    else: r=5
+    return r if is_public else r+0.1
+
+# %% ../nbs/00_core.ipynb
+def ranked_complete(code, line_no=None, col_no=None):
+    ipy = get_ipython()
+    lines = code.splitlines(True)
+    if line_no: offset = sum(len(lines[i]) for i in range(line_no-1)) + col_no -1
+    else: offset = len(code)
+    cs = ipy.Completer.completions(code, offset)
+    def _c(a):
+        res = dict2obj({attr: getattr(a, attr) for attr in dir(a) if attr[0]!='_'})
+        res['mod']= getattr(ipy.user_ns.get(a.text, None), '__module__', None)
+        res['rank'] = _rank(res, s=code)
+        return res
+    return [_c(c) for c in cs if not c.text.startswith('__')]
+
+# %% ../nbs/00_core.ipynb
+def _signatures(ns, s, line, col):
+    ctx = Interpreter(s, [ns]).get_signatures(line, col)
+    if not ctx: ctx = jscript(s).get_signatures(line, col)
+    return ctx
+
+def sig_help(code, line_no=None, col_no=None):
+    ns = get_ipython().user_ns
+    ctx = _signatures(ns, code, line=line_no, col=col_no)
+    def _s(s): return {'label':s.description,'typ':s.type, 'mod':s.module_name, 'doc':s.docstring(),
+                       'idx':s.index, 'params':[{'name':p.name} for p in s.params]}
+    return [_s(opt) for opt in ctx]
+
+# %% ../nbs/00_core.ipynb
+def get_vars(vs:list):
+    "Get variables from namespace."
+    ns = get_ipython().user_ns
+    return {v:ns[v] for v in vs if v in ns}
+
+# %% ../nbs/00_core.ipynb
+def _get_schema(ns: dict, t):
+    "Check if tool `t` has errors."
+    if t not in ns: return f"`{t}` not found. Did you run it?"
+    try: return get_schema(ns[t])
+    except Exception as e: return f"`{t}`: {e}."
+
+def get_schemas(fs:list):
+    "Get schemas from namespace."
+    ns = get_ipython().user_ns
+    return {f:_get_schema(ns,f) for f in fs}

{ipykernel_helper-0.0.1.dist-info → ipykernel_helper-0.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ipykernel-helper
-Version: 0.0.1
+Version: 0.0.2
 Summary: Helpers for ipykernel and friends
 Home-page: https://github.com/AnswerDotAI/ipykernel-helper
 Author: Jeremy Howard
@@ -19,6 +19,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: fastcore
+Requires-Dist: toolslm>=0.2.0
 Provides-Extra: dev
 Dynamic: author
 Dynamic: author-email

ipykernel_helper-0.0.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+ipykernel_helper/__init__.py,sha256=oDLMnoGVIAzRy6kwJGpxSDGyOG-aI7DiiERW6jGauWk,42
+ipykernel_helper/_modidx.py,sha256=juRUbLKiU68mB88Ehn5cMLD2REiCSWOQd9BTdeDQz4Y,1727
+ipykernel_helper/core.py,sha256=UpePoI_QbKQM2BjJ3sQ-_54tdFaHHDfLHmSPmu4i7V4,4494
+ipykernel_helper-0.0.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ipykernel_helper-0.0.2.dist-info/METADATA,sha256=xuDKpEG0317k8OoE-_16n2gkthww1fotP4mPVVmZnLI,2557
+ipykernel_helper-0.0.2.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+ipykernel_helper-0.0.2.dist-info/entry_points.txt,sha256=HWiK9xz75QtZUaPaYrwpyH5B8MbW0Ea_vi11UmwBImM,54
+ipykernel_helper-0.0.2.dist-info/top_level.txt,sha256=_diD--64d9MauLE0pTxzZ58lkI8DvCrVc1hVAJsyc_Q,17
+ipykernel_helper-0.0.2.dist-info/RECORD,,

{ipykernel_helper-0.0.1.dist-info → ipykernel_helper-0.0.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

ipykernel_helper-0.0.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-ipykernel_helper/__init__.py,sha256=XoppFCWM_8XsGAbynKKKg7C3Mca0Kzf_ueJ_7U-l9SE,42
-ipykernel_helper/_modidx.py,sha256=RZGtgKFv14Ujyay4e9gpFdx3LVDpgXfYD8oiqMc_2FA,562
-ipykernel_helper/core.py,sha256=pG8zN9RjWNLLUOs8J130kEcHbBoMQJ-etDCZARiG6TM,1459
-ipykernel_helper-0.0.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ipykernel_helper-0.0.1.dist-info/METADATA,sha256=9P5TmkNHbAEUIDdP1LKHwOlY_vyP5n6CDwdYTsu-3Dw,2527
-ipykernel_helper-0.0.1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-ipykernel_helper-0.0.1.dist-info/entry_points.txt,sha256=HWiK9xz75QtZUaPaYrwpyH5B8MbW0Ea_vi11UmwBImM,54
-ipykernel_helper-0.0.1.dist-info/top_level.txt,sha256=_diD--64d9MauLE0pTxzZ58lkI8DvCrVc1hVAJsyc_Q,17
-ipykernel_helper-0.0.1.dist-info/RECORD,,