dialoghelper 0.0.44__py3-none-any.whl

dialoghelper/__init__.py

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

dialoghelper/_modidx.py

@@ -0,0 +1,51 @@
+# Autogenerated by nbdev
+
+d = { 'settings': { 'branch': 'main',
+                'doc_baseurl': '/dialoghelper',
+                'doc_host': 'https://AnswerDotAI.github.io',
+                'git_url': 'https://github.com/AnswerDotAI/dialoghelper',
+                'lib_path': 'dialoghelper'},
+  'syms': { 'dialoghelper.capture': { 'dialoghelper.capture._capture_screen': ('capture.html#_capture_screen', 'dialoghelper/capture.py'),
+                                      'dialoghelper.capture.capture_screen': ('capture.html#capture_screen', 'dialoghelper/capture.py'),
+                                      'dialoghelper.capture.capture_tool': ('capture.html#capture_tool', 'dialoghelper/capture.py'),
+                                      'dialoghelper.capture.setup_share': ('capture.html#setup_share', 'dialoghelper/capture.py'),
+                                      'dialoghelper.capture.start_share': ('capture.html#start_share', 'dialoghelper/capture.py')},
+            'dialoghelper.core': { 'dialoghelper.core._add_msg_unsafe': ('core.html#_add_msg_unsafe', 'dialoghelper/core.py'),
+                                   'dialoghelper.core._find_frame_dict': ('core.html#_find_frame_dict', 'dialoghelper/core.py'),
+                                   '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.add_scr': ('core.html#add_scr', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.add_styles': ('core.html#add_styles', '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.event_get': ('core.html#event_get', '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.core.find_var': ('core.html#find_var', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.fire_event': ('core.html#fire_event', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.gist_file': ('core.html#gist_file', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.iife': ('core.html#iife', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.import_gist': ('core.html#import_gist', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.import_string': ('core.html#import_string', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.is_usable_tool': ('core.html#is_usable_tool', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.load_gist': ('core.html#load_gist', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.mk_toollist': ('core.html#mk_toollist', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.msg_idx': ('core.html#msg_idx', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.msg_insert_line': ('core.html#msg_insert_line', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.msg_replace_lines': ('core.html#msg_replace_lines', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.msg_str_replace': ('core.html#msg_str_replace', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.msg_strs_replace': ('core.html#msg_strs_replace', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.pop_data': ('core.html#pop_data', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.read_msg': ('core.html#read_msg', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.run_msg': ('core.html#run_msg', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.set_var': ('core.html#set_var', '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.url2note': ('core.html#url2note', 'dialoghelper/core.py')},
+            'dialoghelper.db_dc': {}}}

dialoghelper/capture.py

@@ -0,0 +1,45 @@
+"""Screen Capture."""
+
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/01_capture.ipynb.
+
+# %% auto 0
+__all__ = ['setup_share', 'start_share', 'capture_screen', 'capture_tool']
+
+# %% ../nbs/01_capture.ipynb
+from .core import *
+from fastcore.all import *
+from fasthtml.common import Div,Script
+from httpx import post as xpost
+from importlib import resources
+from lisette.core import *
+import PIL.Image
+
+import base64,json,time
+
+# %% ../nbs/01_capture.ipynb
+def setup_share():
+    "Setup screen sharing"
+    iife((resources.files('dialoghelper')/'screenshot.js').read_text())
+
+# %% ../nbs/01_capture.ipynb
+def start_share(): fire_event('shareScreen')
+
+# %% ../nbs/01_capture.ipynb
+def _capture_screen(timeout=15):
+    d = event_get('captureScreen', timeout)
+    if 'img_data' in d: return d.img_data
+    else: raise Exception(f'Capture failed: {d.error}')
+
+# %% ../nbs/01_capture.ipynb
+def capture_screen(timeout=15):
+    "Capture the screen as a PIL image."
+    res = _capture_screen()
+    data = base64.b64decode(res.split(',')[1])
+    return PIL.Image.open(BytesIO(data))
+
+# %% ../nbs/01_capture.ipynb
+def capture_tool(timeout:int=15):
+    "Capture the screen. Re-call this function to get the most recent screenshot, as needed. Use default timeout where possible"
+    try: d = _capture_screen(timeout)
+    except Exception as e: return f'Capture failed: {e}'
+    return ToolResponse([{'type': 'image_url', 'image_url': d}])

dialoghelper/core.py

@@ -0,0 +1,445 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/00_core.ipynb.
+
+# %% auto 0
+__all__ = ['md_cls_d', 'dh_settings', 'Placements', 'empty', 'add_styles', 'find_var', 'set_var', 'call_endp', 'find_dname',
+           'find_msg_id', 'curr_dialog', 'msg_idx', 'add_scr', 'iife', 'pop_data', 'fire_event', 'event_get',
+           'find_msgs', 'add_html', 'read_msg', 'add_msg', 'del_msg', 'update_msg', 'run_msg', 'url2note', 'ast_py',
+           'ast_grep', 'msg_insert_line', 'msg_str_replace', 'msg_strs_replace', 'msg_replace_lines', 'load_gist',
+           'gist_file', 'import_string', 'is_usable_tool', 'mk_toollist', 'import_gist', 'tool_info', 'fc_tool_info']
+
+# %% ../nbs/00_core.ipynb
+import json,importlib,linecache,re,inspect,uuid
+from typing import Dict
+from tempfile import TemporaryDirectory
+from ipykernel_helper import *
+from dataclasses import dataclass
+from fastcore.xml import to_xml
+
+from fastcore.utils import *
+from fastcore.meta import delegates
+from ghapi.all import *
+from fastlite import *
+from fastcore.xtras import asdict
+from inspect import currentframe,Parameter,signature
+from httpx import get as xget, post as xpost
+from IPython.display import display,Markdown
+from monsterui.all import franken_class_map,apply_classes
+from fasthtml.common import Safe,Script,Div
+
+# %% ../nbs/00_core.ipynb
+md_cls_d = {
+    **{f'h{i}': f'uk-h{i}' for i in range(1,5)},
+    'a': "uk-link",
+    'blockquote': "uk-blockquote",
+    'hr':'uk-divider-icon',
+    'table':'uk-table uk-table-sm uk-table-middle uk-table-divider border [&_tr]:divide-x w-[80%] mx-auto',
+    'ol': 'uk-list uk-list-decimal space-y-0', 
+    'ul': 'uk-list uk-list-bullet space-y-0',
+    'p': 'leading-tight',
+    'li': 'leading-tight',
+    'pre': '', 'pre code': '',
+    'code': 'tracking-tight'
+}
+
+def add_styles(s:str, cls_map:dict=None):
+    "Add solveit styles to `s`"
+    return Safe(apply_classes(s, class_map=cls_map or md_cls_d))
+
+# %% ../nbs/00_core.ipynb
+def _find_frame_dict(var:str):
+    "Find the dict (globals or locals) containing var"
+    frame = currentframe().f_back.f_back
+    while frame:
+        if var in frame.f_globals: return frame.f_globals
+        frame = frame.f_back
+    raise ValueError(f"Could not find {var} in any scope")
+
+def find_var(var:str):
+    "Search for var in all frames of the call stack"
+    return _find_frame_dict(var)[var]
+    
+def set_var(var:str, val):
+    "Set var to val after finding it in all frames of the call stack"
+    _find_frame_dict(var)[var] = val
+
+# %% ../nbs/00_core.ipynb
+dh_settings = {'port':5001}
+
+def call_endp(path, dname='', json=False, raiseex=False, **data):
+    if not dname: dname = find_dname()
+    data['dlg_name'] = dname
+    res = xpost(f'http://localhost:{dh_settings["port"]}/{path}', data=data)
+    if raiseex: res.raise_for_status()
+    try: return res.json() if json else res.text
+    except Exception as e: return str(e)
+
+# %% ../nbs/00_core.ipynb
+def find_dname():
+    "Get the message id by searching the call stack for __dialog_id."
+    return find_var('__dialog_name')
+
+# %% ../nbs/00_core.ipynb
+def find_msg_id():
+    "Get the message id by searching the call stack for __msg_id."
+    return find_var('__msg_id')
+
+# %% ../nbs/00_core.ipynb
+def curr_dialog(
+    with_messages:bool=False,  # Include messages as well?
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Get the current dialog info."
+    res = call_endp('curr_dialog_', dname, json=True, with_messages=with_messages)
+    if res: return {'name': res['name'], 'mode': res['mode']}
+
+# %% ../nbs/00_core.ipynb
+def msg_idx(
+    msgid=None,  # Message id to find (defaults to current message)
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Get absolute index of message in dialog."
+    if not msgid: msgid = find_msg_id()
+    return call_endp('msg_idx_', dname, json=True, msgid=msgid)['msgid']
+
+# %% ../nbs/00_core.ipynb
+def add_scr(scr, oob='beforeend:#js-script'):
+    "Swap a script element to the end of the js-script element"
+    if isinstance(scr,str): scr = Script(scr)
+    add_html(Div(scr, hx_swap_oob=oob))
+
+# %% ../nbs/00_core.ipynb
+def iife(code: str) -> str:
+    "Wrap javascript code string in an IIFE and execute it via `add_html`"
+    add_scr(f'''
+(async () => {{
+{code}
+}})();
+''')
+
+# %% ../nbs/00_core.ipynb
+def pop_data(idx, timeout=15):
+    url = 'http://localhost:5001/pop_data_blocking_'
+    return dict2obj(xpost(url, data={'data_id': idx, 'timeout': timeout}).json())
+
+# %% ../nbs/00_core.ipynb
+def fire_event(evt:str, data=None):
+    params = f"'{evt}'"
+    if data is not None: params += f", {json.dumps(data)}"
+    add_html(Script(f"htmx.trigger(document.body, {params});", id='js-event', hx_swap_oob='true'))
+
+# %% ../nbs/00_core.ipynb
+def event_get(evt:str, timeout=15, data=None):
+    "Call `fire_event` and then `pop_data` to get a response"
+    idx = uuid.uuid4()
+    if not data: data = {}
+    data['idx'] = str(idx)
+    fire_event(evt, data=data)
+    return pop_data(idx, timeout)
+
+# %% ../nbs/00_core.ipynb
+def find_msgs(
+    re_pattern:str='', # Optional regex to search for (re.DOTALL+re.MULTILINE is used)
+    msg_type:str=None, # optional limit by message type ('code', 'note', or 'prompt')
+    limit:int=None, # Optionally limit number of returned items
+    include_output:bool=True, # Include output in returned dict?
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Find `list[dict]` of messages in current specific dialog that contain the given information. To refer to a message found later, use its `id` field."
+    res = call_endp('find_msgs_', dname, json=True, re_pattern=re_pattern, msg_type=msg_type, limit=limit)['msgs']
+    if not include_output:
+        for o in res: o.pop('output', None)
+    return res
+
+# %% ../nbs/00_core.ipynb
+def add_html(
+    content:str, # The HTML to send to the client (generally should include hx-swap-oob)
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Send HTML to the browser to be swapped into the DOM"
+    call_endp('add_html_', dname, content=to_xml(content))
+
+# %% ../nbs/00_core.ipynb
+Placements = str_enum('Placements', 'add_after', 'add_before', 'at_start', 'at_end')
+
+# %% ../nbs/00_core.ipynb
+def read_msg(
+    n:int=-1,      # Message index (if relative, +ve is downwards)
+    relative:bool=True,  # Is `n` relative to current message (True) or absolute (False)?
+    msgid:str=None,  # Message id to find (defaults to current message)
+    view_range:list[int,int]=None, # Optional 1-indexed (start, end) line range for files, end=-1 for EOF
+    nums:bool=False, # Whether to show line numbers
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+    ):
+    """Get the message indexed in the current dialog.
+    - To get the exact message use `n=0` and `relative=True` together with `msgid`.
+    - To get a relative message use `n` (relative position index).
+    - To get the nth message use `n` with `relative=False`, e.g `n=0` first message, `n=-1` last message."""
+    if relative and not msgid: msgid = find_msg_id()
+    data = dict(n=n, relative=relative, msgid=msgid)
+    if view_range: data['view_range'] = view_range # None gets converted to '' so we avoid passing it to use the p.default
+    if nums: data['nums'] = nums
+    return call_endp('read_msg_', dname, json=True, **data)
+
+# %% ../nbs/00_core.ipynb
+def add_msg(
+    content:str, # Content of the message (i.e the message prompt, code, or note text)
+    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)
+    msg_type: str='note', # Message type, can be 'code', 'note', or 'prompt'
+    output:str='', # Prompt/code output; Code outputs must be .ipynb-compatible JSON array
+    time_run: str | None = '', # When was message executed
+    is_exported: int | None = 0, # Export message to a module?
+    skipped: int | None = 0, # Hide message from prompt?
+    i_collapsed: int | None = 0, # Collapse input?
+    o_collapsed: int | None = 0, # Collapse output?
+    heading_collapsed: int | None = 0, # Collapse heading section?
+    pinned: int | None = 0, # Pin to context?
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Add/update a message to the queue to show after code execution completes."
+    if placement not in ('at_start','at_end') and not msgid: msgid = find_msg_id()
+    res = call_endp(
+        'add_relative_', dname, content=content, placement=placement, msgid=msgid, msg_type=msg_type, output=output,
+        time_run=time_run, is_exported=is_exported, skipped=skipped, pinned=pinned,
+        i_collapsed=i_collapsed, o_collapsed=o_collapsed, heading_collapsed=heading_collapsed)
+    set_var('__msg_id', res)
+    return res
+
+# %% ../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(
+    content:str, # Content of the message (i.e the message prompt, code, or note text)
+    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()
+    res = call_endp(
+        'add_relative_', dname, content=content, placement=placement, msgid=msgid, run=run, **kwargs)
+    set_var('__msg_id', res)
+    return res
+
+# %% ../nbs/00_core.ipynb
+def _umsg(
+    content:str|None=None, # Content of the message (i.e the message prompt, code, or note text)
+    msg_type: str|None = None, # Message type, can be 'code', 'note', or 'prompt'
+    output:str|None = None, # Prompt/code output; Code outputs must be .ipynb-compatible JSON array
+    time_run: str | None = None, # When was message executed
+    is_exported: int | None = None, # Export message to a module?
+    skipped: int | None = None, # Hide message from prompt?
+    i_collapsed: int | None = None, # Collapse input?
+    o_collapsed: int | None = None, # Collapse output?
+    heading_collapsed: int | None = None, # Collapse heading section?
+    pinned: int | None = None # Pin to context?
+): ...
+
+# %% ../nbs/00_core.ipynb
+@delegates(_umsg)
+def update_msg(
+    msgid:str=None, # id of message to update (if None, uses current message)
+    msg:Optional[Dict]=None, # Dictionary of field keys/values to update
+    dname:str='', # Running dialog to get info for; defaults to current dialog
+    **kwargs):
+    """Update an existing message. Provide either `msg` OR field key/values to update.
+    - Use `content` param to update contents.
+    - Only include parameters to update--missing ones will be left unchanged."""
+    if not msgid and not msg: raise TypeError("update_msg needs either a dict message or `msgid=`")
+    res = call_endp('update_msg_', dname, msgid=msgid, **kwargs)
+    set_var('__msg_id', res)
+    return res
+
+# %% ../nbs/00_core.ipynb
+def run_msg(
+    msgid:str=None, # id of message to execute
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Adds a message to the run queue. Use read_msg to see the output once it runs."
+    return call_endp('add_runq_', dname, msgid=msgid, api=True)
+
+# %% ../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 msg_insert_line(
+    msgid:str,  # Message id to edit
+    insert_line: int, # The line number after which to insert the text (0 for beginning of file)
+    new_str: str, # The text to insert
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+    ):
+    "Insert text at a specific line number in a message"
+    return call_endp('msg_insert_line_', dname, json=True, msgid=msgid, insert_line=insert_line, new_str=new_str)
+
+# %% ../nbs/00_core.ipynb
+def msg_str_replace(
+    msgid:str,  # Message id to edit
+    old_str: str, # Text to find and replace
+    new_str: str, # Text to replace with
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Replace first occurrence of old_str with new_str in a message"
+    return call_endp('msg_str_replace_', dname, json=True, msgid=msgid, old_str=old_str, new_str=new_str)
+
+# %% ../nbs/00_core.ipynb
+def msg_strs_replace(
+    msgid:str,  # Message id to edit
+    old_strs:list[str], # List of strings to find and replace
+    new_strs:list[str], # List of replacement strings (must match length of old_strs)
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Replace multiple strings simultaneously in a message"
+    return call_endp('msg_strs_replace_', dname, json=True, msgid=msgid, old_strs=old_strs, new_strs=new_strs)
+
+# %% ../nbs/00_core.ipynb
+def msg_replace_lines(
+    msgid:str,  # Message id to edit
+    start_line:int, # Starting line number to replace (1-based indexing)
+    end_line:int, # Ending line number to replace (1-based indexing, inclusive)
+    new_content:str, # New content to replace the specified lines
+    dname:str='' # Running dialog to get info for; defaults to current dialog
+):
+    "Replace a range of lines with new content in a message"
+    return call_endp('msg_replace_lines_', dname, json=True, msgid=msgid, start_line=start_line, end_line=end_line, new_content=new_content)
+
+# %% ../nbs/00_core.ipynb
+def load_gist(gist_id:str):
+    "Retrieve a gist"
+    api = GhApi()
+    if '/' in gist_id: *_,user,gist_id = gist_id.split('/')
+    else: user = None
+    return api.gists.get(gist_id, user=user)
+
+# %% ../nbs/00_core.ipynb
+def gist_file(gist_id:str):
+    "Get the first file from a gist"
+    gist = load_gist(gist_id)
+    return first(gist.files.values())
+
+# %% ../nbs/00_core.ipynb
+def import_string(
+    code:str, # Code to import as a module
+    name:str  # Name of module to create
+):
+    with TemporaryDirectory() as tmpdir:
+        path = Path(tmpdir) / f"{name}.py"
+        path.write_text(code)
+        # linecache.cache storage allows inspect.getsource() after tmpdir lifetime ends
+        linecache.cache[str(path)] = (len(code), None, code.splitlines(keepends=True), str(path))
+        spec = importlib.util.spec_from_file_location(name, path)
+        module = importlib.util.module_from_spec(spec)
+        sys.modules[name] = module
+        spec.loader.exec_module(module)
+        return module
+
+# %% ../nbs/00_core.ipynb
+empty = Parameter.empty
+
+def is_usable_tool(func:callable):
+    "True if the function has a docstring and all parameters have types, meaning that it can be used as an LLM tool."    
+    if not func.__doc__ or not callable(func): return False
+    return all(p.annotation != empty for p in signature(func).parameters.values())
+
+# %% ../nbs/00_core.ipynb
+def mk_toollist(syms):
+    return "\n".join(f"- &`{sym.__name__}`: {sym.__doc__}" for sym in syms if is_usable_tool(sym))
+
+# %% ../nbs/00_core.ipynb
+def import_gist(
+    gist_id:str, # user/id or just id of gist to import as a module
+    mod_name:str=None, # module name to create (taken from gist filename if not passed)
+    add_global:bool=True, # add module to caller's globals?
+    import_wildcard:bool=False, # import all exported symbols to caller's globals
+    create_msg:bool=False # Add a message that lists usable tools
+):
+    "Import gist directly from string without saving to disk"
+    fil = gist_file(gist_id)
+    mod_name = mod_name or Path(fil['filename']).stem
+    module = import_string(fil['content'], mod_name)
+    glbs = currentframe().f_back.f_globals
+    if add_global: glbs[mod_name] = module
+    syms = getattr(module, '__all__', None)
+    if syms is None: syms = [o for o in dir(module) if not o.startswith('_')]
+    syms = [getattr(module, nm) for nm in syms]
+    if import_wildcard:
+        for sym in syms: glbs[sym.__name__] = sym
+    if create_msg:
+        pref = getattr(module, '__doc__', "Tools added to dialog:")
+        add_msg(f"{pref}\n\n{mk_toollist(syms)}")
+    return module
+
+# %% ../nbs/00_core.ipynb
+def tool_info():
+    cts='''Tools available from `dialoghelper`:
+
+- &`curr_dialog`: Get the current dialog info.
+- &`msg_idx`: Get absolute index of message in dialog.
+- &`add_html`: Send HTML to the browser to be swapped into the DOM using hx-swap-oob.
+- &`find_msg_id`: Get the current message id.
+- &`find_msgs`: Find messages in current specific dialog that contain the given information.
+  - (solveit can often get this id directly from its context, and will not need to use this if the required information is already available to it.)
+- &`read_msg`: Get the message indexed in the current dialog.
+  - To get the exact message use `n=0` and `relative=True` together with `msgid`.
+  - To get a relative message use `n` (relative position index).
+  - To get the nth message use `n` with `relative=False`, e.g `n=0` first message, `n=-1` last message.
+- &`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.
+- &`url2note`: Read URL as markdown, and add a note below current message with the result
+- &`msg_insert_line`: Insert text at a specific location in a message.
+- &`msg_str_replace`: Find and replace text in a message.
+- &`msg_strs_replace`: Find and replace multiple strings in a message.
+- &`msg_replace_lines`: Replace a range of lines in a message with new content.
+  - Always first use `read_msg( msgid=msgid, n=0, relative=True, nums=True)` to view the content with line numbers.'''
+    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/db_dc.py

@@ -0,0 +1,92 @@
+__all__ = ["_Meta", "Attachment", "Completion", "Completion_Mode", "Context", "Dialog", "Message", "Secret", "User"]
+from dataclasses import dataclass
+@dataclass
+class _Meta:
+    id: int | None = None
+    version: int | None = 0
+
+@dataclass
+class Attachment:
+    id: str | None = None
+    data: bytes | None = None
+    content_type: str | None = None
+    sid: int | None = None
+
+@dataclass
+class Completion:
+    req_id: str | None = None
+    did: int | None = None
+    mid: int | None = None
+    mtype: str | None = None
+    pfx: str | None = None
+    sfx: str | None = None
+    ctx: str | None = ''
+    mode: str | None = None
+    comp: str | None = None
+    logprobs: str | None = None
+    status: str | None = None
+
+@dataclass
+class Completion_Mode:
+    id: int | None = None
+    name: str | None = None
+    c_acceptichars: int | None = None
+    c_scon: int | None = None
+    c_triggerc: str | None = None
+    c_triggeronaccept: int | None = None
+    c_mftp: float | None = None
+    c_model: str | None = None
+    c_nlines: str | None = None
+    c_temp: float | None = None
+    p_acceptichars: int | None = None
+    p_scon: int | None = None
+    p_triggerc: str | None = None
+    p_triggeronaccept: int | None = None
+    p_mftp: float | None = None
+    p_model: str | None = None
+    p_nlines: str | None = None
+    p_temp: float | None = None
+
+@dataclass
+class Context:
+    name: str | None = None
+    context: str | None = None
+    token_count: int | None = 0
+    did: int | None = None
+
+@dataclass
+class Dialog:
+    id: int | None = None
+    name: str | None = None
+    mode: int | None = 2
+
+@dataclass
+class Message:
+    sid: str | None = None
+    mid: str | None = None
+    content: str | None = None
+    output: str | None = ''
+    input_tokens: int | None = '0'
+    output_tokens: int | None = '0'
+    msg_type: str | None = 'code'
+    time_run: str | None = ''
+    is_exported: int | None = '0'
+    skipped: int | None = '0'
+    did: int | None = None
+    i_collapsed: int | None = '0'
+    o_collapsed: int | None = '0'
+    header_collapsed: int | None = '0'
+    pinned: int | None = '0'
+    use_thinking: int | None = 0
+
+@dataclass
+class Secret:
+    name: str | None = None
+    secret: str | None = None
+
+@dataclass
+class User:
+    id: int | None = None
+    version: int | None = None
+    settings: str | None = None
+

dialoghelper/screenshot.js

@@ -0,0 +1,26 @@
+window.getScreenshot = async (maxWidth = 1280, maxHeight = 1024) => {
+    if (window?.vtrack) {
+        const cap = new ImageCapture(window.vtrack);
+        const img = await cap.grabFrame();
+        // rescale and convert to b64
+        const scaleX = maxWidth / img.width;
+        const scaleY = maxHeight / img.height;
+        const scale = Math.min(scaleX, scaleY, 1); // don't upscale
+        const newWidth = Math.floor(img.width * scale);
+        const newHeight = Math.floor(img.height * scale);
+        const c = document.createElement('canvas');
+        c.width = newWidth; c.height = newHeight;
+        c.getContext('2d').drawImage(img, 0, 0, newWidth, newHeight);
+        return c.toDataURL();
+    }
+}
+
+document.body.addEventListener('shareScreen', async (e) => {
+    vstream = await navigator.mediaDevices.getDisplayMedia();
+    window.vtrack = vstream.getVideoTracks()[0];
+});
+
+document.body.addEventListener('captureScreen', async (e) => {
+    pushData(e.detail.idx, {img_data: await getScreenshot()});
+});
+

dialoghelper-0.0.44.dist-info/METADATA

@@ -0,0 +1,113 @@
+Metadata-Version: 2.4
+Name: dialoghelper
+Version: 0.0.44
+Summary: Helper functions for solveit dialogs
+Home-page: https://github.com/AnswerDotAI/dialoghelper
+Author: Jeremy Howard
+Author-email: github@jhoward.fastmail.fm
+License: Apache Software License 2.0
+Keywords: nbdev jupyter notebook python
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Natural Language :: English
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: License :: OSI Approved :: Apache Software License
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: fastcore>=1.8.10
+Requires-Dist: fastlite
+Requires-Dist: ghapi
+Requires-Dist: ipykernel-helper
+Requires-Dist: claudette
+Requires-Dist: ast-grep-cli
+Requires-Dist: ast-grep-py
+Requires-Dist: MonsterUI
+Requires-Dist: lisette>=0.0.13
+Requires-Dist: pillow
+Provides-Extra: dev
+Requires-Dist: python-fasthtml; extra == "dev"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# dialoghelper
+
+
+<!-- WARNING: THIS FILE WAS AUTOGENERATED! DO NOT EDIT! -->
+
+This file will become your README and also the index of your
+documentation.
+
+## Developer Guide
+
+If you are new to using `nbdev` here are some useful pointers to get you
+started.
+
+### Install dialoghelper in Development mode
+
+``` sh
+# make sure dialoghelper package is installed in development mode
+$ pip install -e .
+
+# make changes under nbs/ directory
+# ...
+
+# compile to have changes apply to dialoghelper
+$ nbdev_prepare
+```
+
+## Usage
+
+### Installation
+
+Install latest from the GitHub
+[repository](https://github.com/AnswerDotAI/dialoghelper):
+
+``` sh
+$ pip install git+https://github.com/AnswerDotAI/dialoghelper.git
+```
+
+or from [conda](https://anaconda.org/AnswerDotAI/dialoghelper)
+
+``` sh
+$ conda install -c AnswerDotAI dialoghelper
+```
+
+or from [pypi](https://pypi.org/project/dialoghelper/)
+
+``` sh
+$ pip install dialoghelper
+```
+
+### Documentation
+
+Documentation can be found hosted on this GitHub
+[repository](https://github.com/AnswerDotAI/dialoghelper)’s
+[pages](https://AnswerDotAI.github.io/dialoghelper/). Additionally you
+can find package manager specific guidelines on
+[conda](https://anaconda.org/AnswerDotAI/dialoghelper) and
+[pypi](https://pypi.org/project/dialoghelper/) respectively.
+
+## How to use
+
+Fill me in please! Don’t forget code examples:
+
+``` python
+1+1
+```
+
+    2

dialoghelper-0.0.44.dist-info/RECORD

@@ -0,0 +1,12 @@
+dialoghelper/__init__.py,sha256=W3xZgCB22_feYaRCGZM4hqkrdBSs6VCfwsSqWGpO6bE,43
+dialoghelper/_modidx.py,sha256=2KIy8BtZ-ks0CGzg8xH6ptn-gPdFn7qZjB5ccB4evog,5442
+dialoghelper/capture.py,sha256=VxkO52HA33jcBm4W4lZS41d1Girg-ov_Mi5wTeBFV48,1429
+dialoghelper/core.py,sha256=O6tuHTyiVHmJVlNm0ghtqvFXgP1ElhnY5fEUhn0QXT4,20128
+dialoghelper/db_dc.py,sha256=mi2Q2am_SoAPGpNQg7KPFS5pR9WEapRXT8ypkNmMfw0,2330
+dialoghelper/screenshot.js,sha256=5x28oyCGdhcNGzb6EPoKpWzMCpCZOrgoePCN5SKkrwo,1023
+dialoghelper-0.0.44.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dialoghelper-0.0.44.dist-info/METADATA,sha256=_vr-B0ICukOtdDlWqpDmmhyX4eWvNLlvaobB_CWIrD8,2771
+dialoghelper-0.0.44.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dialoghelper-0.0.44.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
+dialoghelper-0.0.44.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
+dialoghelper-0.0.44.dist-info/RECORD,,

dialoghelper-0.0.44.dist-info/WHEEL

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

dialoghelper-0.0.44.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[nbdev]
+dialoghelper = dialoghelper._modidx:d

dialoghelper-0.0.44.dist-info/licenses/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

dialoghelper-0.0.44.dist-info/top_level.txt

@@ -0,0 +1 @@
+dialoghelper