dialoghelper 0.0.8__tar.gz → 0.0.10__tar.gz

{dialoghelper-0.0.8/dialoghelper.egg-info → dialoghelper-0.0.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.8
+Version: 0.0.10
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard

dialoghelper-0.0.10/dialoghelper/__init__.py

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

{dialoghelper-0.0.8 → dialoghelper-0.0.10}/dialoghelper/_modidx.py

@@ -9,16 +9,19 @@ 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.export_dialog': ('core.html#export_dialog', 'dialoghelper/core.py'),
                                    'dialoghelper.core.find_dialog_id': ('core.html#find_dialog_id', '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.get_db': ('core.html#get_db', 'dialoghelper/core.py'),
                                    'dialoghelper.core.gist_file': ('core.html#gist_file', 'dialoghelper/core.py'),
+                                   'dialoghelper.core.import_dialog': ('core.html#import_dialog', '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.load_gist': ('core.html#load_gist', 'dialoghelper/core.py'),
                                    'dialoghelper.core.msg_idx': ('core.html#msg_idx', 'dialoghelper/core.py'),
                                    'dialoghelper.core.read_msg': ('core.html#read_msg', 'dialoghelper/core.py'),
                                    'dialoghelper.core.read_msg_ids': ('core.html#read_msg_ids', '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.db_dc': {}}}

{dialoghelper-0.0.8 → dialoghelper-0.0.10}/dialoghelper/core.py

@@ -2,13 +2,15 @@
 
 # %% auto 0
 __all__ = ['get_db', 'find_var', 'find_dialog_id', 'find_msgs', 'find_msg_id', 'read_msg_ids', 'msg_idx', 'read_msg', 'add_msg',
-           'update_msg', 'add_html', 'load_gist', 'gist_file', 'import_string', 'import_gist', 'asdict']
+           'update_msg', 'add_html', 'load_gist', 'gist_file', 'import_string', 'import_gist', 'export_dialog',
+           'import_dialog', 'asdict']
 
 # %% ../nbs/00_core.ipynb
 import inspect, json, importlib, linecache
 from typing import Dict
 from tempfile import TemporaryDirectory
 from ipykernel_helper import *
+from dataclasses import dataclass
 
 from fastcore.utils import *
 from fastcore.meta import delegates
@@ -142,8 +144,8 @@ def update_msg(
     Use `content` param to update contents. Be sure to pass a `sid` (stable id -- the pk) not a `mid`
     (which is used only for sorting, and can change).
     Only include parameters to update--missing ones will be left unchanged."""
-    sid = msg.pop('sid', sid)
     kw = (msg or {}) | kwargs
+    sid = kw.pop('sid', sid)
     if not sid: raise TypeError("update_msg needs either a dict message or `sid=...`")
     kw.pop('did', None)
     run_cmd('add_msg', placement='update', sid=sid, **kw)
@@ -197,3 +199,29 @@ def import_gist(
     module = import_string(fil['content'], mod_name)
     if add_global: inspect.currentframe().f_back.f_globals[mod_name] = module
     return module
+
+# %% ../nbs/00_core.ipynb
+__EXPORT_FIELDS = set('content output input_tokens output_tokens msg_type is_exported skipped pinned i_collapsed o_collapsed header_collapsed'.split())
+
+__REQUIRED_FIELDS = set('content output msg_type'.split())
+
+# %% ../nbs/00_core.ipynb
+def export_dialog(filename: str, did:int=None):
+    "Export dialog messages and optionally attachments to JSON"
+    if did is None: did = find_dialog_id()
+    db = get_db()
+    msgs = db.t.message('did=? and (pinned=0 or pinned is null)', [did], order_by='mid')
+    msg_data = [{k:getattr(msg,k) for k in __EXPORT_FIELDS if hasattr(msg, k)}
+                for msg in msgs]
+    result = {'messages': msg_data, 'dialog_name': db.t.dialog[did].name}
+    with open(filename, 'w') as f: json.dump(result, f, indent=2)
+
+# %% ../nbs/00_core.ipynb
+def import_dialog(fname, add_header=True):
+    "Import dialog messages from JSON file using `add_msg`"
+    data = json.loads(Path(fname).read_text())
+    for msg in data['messages'][::-1]:
+        opts = {k:msg[k] for k in __EXPORT_FIELDS - __REQUIRED_FIELDS if k in msg}
+        add_msg(msg.get('content',''), msg.get('msg_type','note'), msg.get('output',''), 'at_end', **opts)
+    if add_header: add_msg(f"# Imported Dialog `{fname}`", 'note', placement='at_end')
+    return f"Imported {len(data['messages'])} messages"

dialoghelper-0.0.10/dialoghelper/db_dc.py

@@ -0,0 +1,91 @@
+__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'
+
+@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-0.0.8 → dialoghelper-0.0.10/dialoghelper.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dialoghelper
-Version: 0.0.8
+Version: 0.0.10
 Summary: Helper functions for solveit dialogs
 Home-page: https://github.com/AnswerDotAI/dialoghelper
 Author: Jeremy Howard

{dialoghelper-0.0.8 → dialoghelper-0.0.10}/dialoghelper.egg-info/SOURCES.txt

@@ -7,6 +7,7 @@ setup.py
 dialoghelper/__init__.py
 dialoghelper/_modidx.py
 dialoghelper/core.py
+dialoghelper/db_dc.py
 dialoghelper.egg-info/PKG-INFO
 dialoghelper.egg-info/SOURCES.txt
 dialoghelper.egg-info/dependency_links.txt

{dialoghelper-0.0.8 → dialoghelper-0.0.10}/settings.ini

@@ -1,7 +1,7 @@
 [DEFAULT]
 repo = dialoghelper
 lib_name = dialoghelper
-version = 0.0.8
+version = 0.0.10
 min_python = 3.9
 license = apache2
 black_formatting = False

dialoghelper-0.0.8/dialoghelper/__init__.py

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