dialoghelper 0.0.9__py3-none-any.whl → 0.0.10__py3-none-any.whl

dialoghelper/__init__.py

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

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/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
@@ -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/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.9.dist-info → dialoghelper-0.0.10.dist-info}/METADATA

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

dialoghelper-0.0.10.dist-info/RECORD

@@ -0,0 +1,10 @@
+dialoghelper/__init__.py,sha256=vGb5bWRnqikdeGjmhzb9R8htmRhc2CloD1a9bB7Ttvo,43
+dialoghelper/_modidx.py,sha256=fZ9-eTMaLdG-nvxxTe9_qjLwKIGS9_FU7KpFyCiF7DU,2529
+dialoghelper/core.py,sha256=nJyoSz4TlYY4lOLBV8TqSXc3PW61J82GA6eleDr6i-4,9442
+dialoghelper/db_dc.py,sha256=ieSie_K1HHXACEJVZ0rMkD_uo42GgmMqqKDYrodjhtM,2297
+dialoghelper-0.0.10.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dialoghelper-0.0.10.dist-info/METADATA,sha256=qSxn-0_wPe8LdbEDSQU6R2ArvrWHh2E8vle_Fw9hZJc,2558
+dialoghelper-0.0.10.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+dialoghelper-0.0.10.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
+dialoghelper-0.0.10.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
+dialoghelper-0.0.10.dist-info/RECORD,,

dialoghelper-0.0.9.dist-info/RECORD

@@ -1,9 +0,0 @@
-dialoghelper/__init__.py,sha256=8v08jLbD8mgo8kbkFtp9fIlkDQsiovYgy2xWvMrgXWY,42
-dialoghelper/_modidx.py,sha256=mYMJPTi5PlEZg_LHRYfaqtNWdvPb5i2MTvO5lNfpwtY,2245
-dialoghelper/core.py,sha256=aCcqPiyaEzERXcxH3R40ZZ-fx_BYBCg1_i_kf3Z2KeE,8034
-dialoghelper-0.0.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dialoghelper-0.0.9.dist-info/METADATA,sha256=fEOFgODIDpG47_BAgNQq6QUxj5dj6P9WPZC0wyArLp8,2557
-dialoghelper-0.0.9.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-dialoghelper-0.0.9.dist-info/entry_points.txt,sha256=wvDeX-XTS_XVjWiiPQe6yVfmyNwy9eCr36ewp9baFIg,46
-dialoghelper-0.0.9.dist-info/top_level.txt,sha256=VXLlkgltFs_q-XB9imt2G64I_-MPm1RnxlpvUWPuLKM,13
-dialoghelper-0.0.9.dist-info/RECORD,,