dialoghelper 0.0.31__tar.gz → 0.0.33__tar.gz

{dialoghelper-0.0.31/dialoghelper.egg-info → dialoghelper-0.0.33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.31
+Version: 0.0.33
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard
@@ -18,11 +18,13 @@ Classifier: License :: OSI Approved :: Apache Software License
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: fastcore>=1.8.5
+Requires-Dist: fastcore>=1.8.11
 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.33/dialoghelper/__init__.py

@@ -0,0 +1,2 @@
+__version__ = "0.0.33"
+from .core import *

{dialoghelper-0.0.31 → dialoghelper-0.0.33}/dialoghelper/_modidx.py

@@ -9,9 +9,12 @@ 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'),
+                                   'dialoghelper.core.fc_tool_info': ('core.html#fc_tool_info', 'dialoghelper/core.py'),
                                    'dialoghelper.core.find_dname': ('core.html#find_dname', 'dialoghelper/core.py'),
                                    'dialoghelper.core.find_msg_id': ('core.html#find_msg_id', 'dialoghelper/core.py'),
                                    'dialoghelper.core.find_msgs': ('core.html#find_msgs', 'dialoghelper/core.py'),

{dialoghelper-0.0.31 → dialoghelper-0.0.33}/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', 'url2note', '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',
+           'fc_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)
 
@@ -201,6 +202,25 @@ def url2note(
     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"
@@ -283,3 +303,18 @@ def tool_info():
 - &`update_msg`: Update an existing message.
 - &`url2note`: Read URL as markdown, and add a note below current message with the result'''
     add_msg(cts)
+
+# %% ../nbs/00_core.ipynb
+def fc_tool_info():
+    cts='''Tools available from `fastcore.tools`:
+
+- &`rg`: Run the `rg` command with the args in `argstr` (no need to backslash escape)
+- &`sed`: Run the `sed` command with the args in `argstr` (e.g for reading a section of a file)
+- &`view`: View directory or file contents with optional line range and numbers
+- &`create`: Creates a new file with the given content at the specified path
+- &`insert`: Insert new_str at specified line number
+- &`str_replace`: Replace first occurrence of old_str with new_str in file
+- &`strs_replace`: Replace for each str pair in old_strs,new_strs
+- &`replace_lines`: Replace lines in file using start and end line-numbers'''
+    add_msg(cts)
+    add_msg('from fastcore.tools import *', msg_type='code')

{dialoghelper-0.0.31 → dialoghelper-0.0.33/dialoghelper.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.31
+Version: 0.0.33
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard
@@ -18,11 +18,13 @@ Classifier: License :: OSI Approved :: Apache Software License
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: fastcore>=1.8.5
+Requires-Dist: fastcore>=1.8.11
 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.31 → dialoghelper-0.0.33}/dialoghelper.egg-info/requires.txt

@@ -1,8 +1,10 @@
-fastcore>=1.8.5
+fastcore>=1.8.11
 fastlite
 ghapi
 ipykernel-helper
 claudette
+ast-grep-cli
+ast-grep-py
 
 [dev]
 python-fasthtml

{dialoghelper-0.0.31 → dialoghelper-0.0.33}/settings.ini

@@ -1,11 +1,11 @@
 [DEFAULT]
 repo = dialoghelper
 lib_name = dialoghelper
-version = 0.0.31
+version = 0.0.33
 min_python = 3.9
 license = apache2
 black_formatting = False
-requirements = fastcore>=1.8.5 fastlite ghapi ipykernel-helper claudette
+requirements = fastcore>=1.8.11 fastlite ghapi ipykernel-helper claudette ast-grep-cli ast-grep-py
 dev_requirements = python-fasthtml
 doc_path = _docs
 lib_path = dialoghelper

dialoghelper-0.0.31/dialoghelper/__init__.py

@@ -1,2 +0,0 @@
-__version__ = "0.0.31"
-from .core import *