dialoghelper 0.0.30__py3-none-any.whl → 0.0.32__py3-none-any.whl

dialoghelper/__init__.py

@@ -1,2 +1,2 @@
-__version__ = "0.0.30"
+__version__ = "0.0.32"
 from .core import *

dialoghelper/_modidx.py

@@ -9,6 +9,8 @@ d = { 'settings': { 'branch': 'main',
                                    'dialoghelper.core._umsg': ('core.html#_umsg', 'dialoghelper/core.py'),
                                    'dialoghelper.core.add_html': ('core.html#add_html', 'dialoghelper/core.py'),
                                    'dialoghelper.core.add_msg': ('core.html#add_msg', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.ast_grep': ('core.html#ast_grep', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.ast_py': ('core.html#ast_py', 'dialoghelper/core.py'),
                                    'dialoghelper.core.call_endp': ('core.html#call_endp', 'dialoghelper/core.py'),
                                    'dialoghelper.core.curr_dialog': ('core.html#curr_dialog', 'dialoghelper/core.py'),
                                    'dialoghelper.core.del_msg': ('core.html#del_msg', 'dialoghelper/core.py'),
@@ -26,7 +28,8 @@ d = { 'settings': { 'branch': 'main',
                                    'dialoghelper.core.read_msg': ('core.html#read_msg', 'dialoghelper/core.py'),
                                    'dialoghelper.core.run_msg': ('core.html#run_msg', 'dialoghelper/core.py'),
                                    'dialoghelper.core.tool_info': ('core.html#tool_info', 'dialoghelper/core.py'),
-                                   'dialoghelper.core.update_msg': ('core.html#update_msg', 'dialoghelper/core.py')},
+                                   'dialoghelper.core.update_msg': ('core.html#update_msg', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.url2note': ('core.html#url2note', 'dialoghelper/core.py')},
             'dialoghelper.db_dc': {},
             'dialoghelper.experimental': { 'dialoghelper.experimental._pop_data': ( 'experimental.html#_pop_data',
                                                                                     'dialoghelper/experimental.py'),

dialoghelper/core.py

@@ -2,8 +2,9 @@
 
 # %% auto 0
 __all__ = ['Placements', 'empty', 'find_var', 'call_endp', 'find_dname', 'find_msg_id', 'curr_dialog', 'find_msgs', 'msg_idx',
-           'read_msg', 'add_html', 'del_msg', 'run_msg', 'add_msg', 'update_msg', 'load_gist', 'gist_file',
-           'import_string', 'is_usable_tool', 'mk_toollist', 'import_gist', 'tool_info', 'asdict']
+           'read_msg', 'add_html', 'run_msg', 'add_msg', 'del_msg', 'update_msg', 'url2note', 'ast_py', 'ast_grep',
+           'load_gist', 'gist_file', 'import_string', 'is_usable_tool', 'mk_toollist', 'import_gist', 'tool_info',
+           'asdict']
 
 # %% ../nbs/00_core.ipynb
 import json, importlib, linecache
@@ -106,14 +107,6 @@ def add_html(
     "Send HTML to the browser to be swapped into the DOM"
     call_endp('add_html_', dname, content=to_xml(content))
 
-# %% ../nbs/00_core.ipynb
-def del_msg(
-    msgid:str=None, # id of message to delete
-    dname:str='' # Running dialog to get info for; defaults to current dialog
-):
-    "Delete a message from the dialog."
-    call_endp('rm_msg_', dname, raiseex=True, msid=msgid)
-
 # %% ../nbs/00_core.ipynb
 def run_msg(
     msgid:str=None, # id of message to execute
@@ -148,6 +141,14 @@ def add_msg(
         time_run=time_run, is_exported=is_exported, skipped=skipped, pinned=pinned,
         i_collapsed=i_collapsed, o_collapsed=o_collapsed, heading_collapsed=heading_collapsed)
 
+# %% ../nbs/00_core.ipynb
+def del_msg(
+    msgid:str=None, # id of message to delete
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Delete a message from the dialog."
+    call_endp('rm_msg_', dname, raiseex=True, msid=msgid)
+
 # %% ../nbs/00_core.ipynb
 @delegates(add_msg)
 def _add_msg_unsafe(
@@ -155,12 +156,12 @@ def _add_msg_unsafe(
     placement:str='add_after', # Can be 'add_after', 'add_before', 'at_start', 'at_end'
     msgid:str=None, # id of message that placement is relative to (if None, uses current message)
     run:bool=False, # For prompts, send it to the AI; for code, execute it (*DANGEROUS -- be careful of what you run!)
+    dname:str='', # Running dialog to get info for; defaults to current dialog
     **kwargs
 ):
     """Add/update a message to the queue to show after code execution completes, and optionally run it. Be sure to pass a `sid` (stable id) not a `mid` (which is used only for sorting, and can change).
     *WARNING*--This can execute arbitrary code, so check carefully what you run!--*WARNING"""
     if placement not in ('at_start','at_end') and not msgid: msgid = find_msg_id()
-    dname = kwargs.pop('dname')
     return call_endp(
         'add_relative_', dname, content=content, placement=placement, msgid=msgid, run=run, **kwargs)
 
@@ -191,6 +192,35 @@ def update_msg(
     if not msgid and not msg: raise TypeError("update_msg needs either a dict message or `msgid=`")
     return call_endp('add_relative_', dname, placement='update', msgid=msgid, **kwargs)
 
+# %% ../nbs/00_core.ipynb
+def url2note(
+    url:str, # URL to read
+    extract_section:bool=True, # If url has an anchor, return only that section
+    selector:str=None # Select section(s) using BeautifulSoup.select (overrides extract_section)
+):
+    "Read URL as markdown, and add a note below current message with the result"
+    res = read_url(url, as_md=True, extract_section=extract_section, selector=selector)
+    return add_msg(res)
+
+# %% ../nbs/00_core.ipynb
+def ast_py(code:str):
+    "Get an SgRoot root node for python `code`"
+    from ast_grep_py import SgRoot
+    return SgRoot(code, "python").root()
+
+# %% ../nbs/00_core.ipynb
+def ast_grep(
+    pattern:str, # ast-grep pattern to search
+    path=".", # path to recursively search for files
+    lang="python" # language to search/scan
+):
+    "Use the `ast-grep` command to find `pattern` in `path`"
+    import json, subprocess
+    cmd = f"ast-grep --pattern '{pattern}' --lang {lang} --json=compact"
+    if path != ".": cmd = f"cd {path} && {cmd}"
+    res = subprocess.run(cmd, shell=True, capture_output=True, text=True)
+    return json.loads(res.stdout) if res.stdout else res.stderr
+
 # %% ../nbs/00_core.ipynb
 def load_gist(gist_id:str):
     "Retrieve a gist"
@@ -270,5 +300,6 @@ def tool_info():
 - &`read_msg`: Get the message indexed in the current dialog.
 - &`del_msg`: Delete a message from the dialog.
 - &`add_msg`: Add/update a message to the queue to show after code execution completes.
-- &`update_msg`: Update an existing message.'''
+- &`update_msg`: Update an existing message.
+- &`url2note`: Read URL as markdown, and add a note below current message with the result'''
     add_msg(cts)

{dialoghelper-0.0.30.dist-info → dialoghelper-0.0.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.30
+Version: 0.0.32
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard
@@ -23,6 +23,8 @@ Requires-Dist: fastlite
 Requires-Dist: ghapi
 Requires-Dist: ipykernel-helper
 Requires-Dist: claudette
+Requires-Dist: ast-grep-cli
+Requires-Dist: ast-grep-py
 Provides-Extra: dev
 Requires-Dist: python-fasthtml; extra == "dev"
 Dynamic: author

dialoghelper-0.0.32.dist-info/RECORD

@@ -0,0 +1,12 @@
+dialoghelper/__init__.py,sha256=xCO6wIOmUkstMqdlaHvUJ0lO8rN3Pp5NRbeq65mLndE,43
+dialoghelper/_modidx.py,sha256=uC3rfgmEe6SwBeLiFhlxx_F-v9TxiPDTpx-QZX9m_V0,4608
+dialoghelper/core.py,sha256=UTan-A1j8bkRTg9ojvE4P5LPvapV1a3QChF8PVbjObw,13650
+dialoghelper/db_dc.py,sha256=mi2Q2am_SoAPGpNQg7KPFS5pR9WEapRXT8ypkNmMfw0,2330
+dialoghelper/experimental.py,sha256=yTXzyY37MPW-9mqzbem-q6CCPXElN0GwAyOK-iJofMY,3910
+dialoghelper/screenshot.js,sha256=DDqlRhemwj8I0AloxMJBnBqDDIqu5knlT1lpG_0rphI,4851
+dialoghelper-0.0.32.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dialoghelper-0.0.32.dist-info/METADATA,sha256=R5fK2Xh4GLnxoIGlXFrpPtSX_67W8B0v6qtG-wD0t2Q,2692
+dialoghelper-0.0.32.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dialoghelper-0.0.32.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
+dialoghelper-0.0.32.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
+dialoghelper-0.0.32.dist-info/RECORD,,

dialoghelper-0.0.30.dist-info/RECORD

@@ -1,12 +0,0 @@
-dialoghelper/__init__.py,sha256=Y4J_58Un2NkktVq246vbfUB1fr9alL7LxlQNIsZE5ec,43
-dialoghelper/_modidx.py,sha256=Au4FWcKVoSRyKPwVeiqCCKTL8cZ2vGPnjgk4FQfMlyg,4273
-dialoghelper/core.py,sha256=c7wGjFYucWTCDfOYGex3QX728elUdqGR34FfxsspvsA,12319
-dialoghelper/db_dc.py,sha256=mi2Q2am_SoAPGpNQg7KPFS5pR9WEapRXT8ypkNmMfw0,2330
-dialoghelper/experimental.py,sha256=yTXzyY37MPW-9mqzbem-q6CCPXElN0GwAyOK-iJofMY,3910
-dialoghelper/screenshot.js,sha256=DDqlRhemwj8I0AloxMJBnBqDDIqu5knlT1lpG_0rphI,4851
-dialoghelper-0.0.30.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dialoghelper-0.0.30.dist-info/METADATA,sha256=PhB6_-rVkKRmbPunN35xi7qJpZ1HeS7BQkjOXCGQpcY,2637
-dialoghelper-0.0.30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dialoghelper-0.0.30.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
-dialoghelper-0.0.30.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
-dialoghelper-0.0.30.dist-info/RECORD,,