public

Author: jph00

lisette/_modidx.py

@@ -19,7 +19,6 @@
                               'lisette.core._has_search': ('core.html#_has_search', 'lisette/core.py'),
                               'lisette.core._is_img': ('core.html#_is_img', 'lisette/core.py'),
                               'lisette.core._lite_call_func': ('core.html#_lite_call_func', 'lisette/core.py'),
-                              'lisette.core._lite_mk_func': ('core.html#_lite_mk_func', 'lisette/core.py'),
                               'lisette.core._mk_content': ('core.html#_mk_content', 'lisette/core.py'),
                               'lisette.core._mk_img': ('core.html#_mk_img', 'lisette/core.py'),
                               'lisette.core._mk_prefill': ('core.html#_mk_prefill', 'lisette/core.py'),
@@ -30,6 +29,7 @@
                               'lisette.core.astream_result': ('core.html#astream_result', 'lisette/core.py'),
                               'lisette.core.cite_footnote': ('core.html#cite_footnote', 'lisette/core.py'),
                               'lisette.core.cite_footnotes': ('core.html#cite_footnotes', 'lisette/core.py'),
+                              'lisette.core.lite_mk_func': ('core.html#lite_mk_func', 'lisette/core.py'),
                               'lisette.core.litellm.ModelResponse._repr_markdown_': ( 'core.html#litellm.modelresponse._repr_markdown_',
                                                                                       'lisette/core.py'),
                               'lisette.core.mk_msg': ('core.html#mk_msg', 'lisette/core.py'),

lisette/core.py

@@ -3,8 +3,8 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/00_core.ipynb.
 
 # %% auto 0
-__all__ = ['effort', 'mk_msg', 'mk_msgs', 'stream_with_complete', 'cite_footnote', 'cite_footnotes', 'Chat', 'astream_result',
-           'AsyncChat', 'aformat_stream', 'adisplay_stream']
+__all__ = ['effort', 'mk_msg', 'mk_msgs', 'stream_with_complete', 'lite_mk_func', 'cite_footnote', 'cite_footnotes', 'Chat',
+           'astream_result', 'AsyncChat', 'aformat_stream', 'adisplay_stream']
 
 # %% ../nbs/00_core.ipynb
 import asyncio, base64, json, litellm, mimetypes
@@ -121,7 +121,7 @@ def stream_with_complete(gen, postproc=noop):
     return stream_chunk_builder(chunks)
 
 # %% ../nbs/00_core.ipynb
-def _lite_mk_func(f):
+def lite_mk_func(f):
     if isinstance(f, dict): return f
     return {'type':'function', 'function':get_schema(f, pname='parameters')}
 
@@ -172,7 +172,7 @@ def __init__(
         hist,tools = mk_msgs(hist),listify(tools)
         if ns is None and tools: ns = mk_ns(tools)
         elif ns is None: ns = globals()
-        self.tool_schemas = [_lite_mk_func(t) for t in tools] if tools else None
+        self.tool_schemas = [lite_mk_func(t) for t in tools] if tools else None
         store_attr()
 
     def _prep_msgs(self, msgs=None, prefill=None):

nbs/00_core.ipynb

@@ -950,7 +950,7 @@
    "outputs": [],
    "source": [
     "#| export\n",
-    "def _lite_mk_func(f):\n",
+    "def lite_mk_func(f):\n",
     "    if isinstance(f, dict): return f\n",
     "    return {'type':'function', 'function':get_schema(f, pname='parameters')}"
    ]
@@ -995,7 +995,7 @@
     }
    ],
    "source": [
-    "toolsc = _lite_mk_func(simple_add)\n",
+    "toolsc = lite_mk_func(simple_add)\n",
     "toolsc"
    ]
   },
@@ -1608,7 +1608,7 @@
     "        hist,tools = mk_msgs(hist),listify(tools)\n",
     "        if ns is None and tools: ns = mk_ns(tools)\n",
     "        elif ns is None: ns = globals()\n",
-    "        self.tool_schemas = [_lite_mk_func(t) for t in tools] if tools else None\n",
+    "        self.tool_schemas = [lite_mk_func(t) for t in tools] if tools else None\n",
     "        store_attr()\n",
     "    \n",
     "    def _prep_msgs(self, msgs=None, prefill=None):\n",
@@ -1834,6 +1834,16 @@
     "### Testing streaming"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f02c7ab1",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from time import sleep"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -1878,7 +1888,6 @@
     }
    ],
    "source": [
-    "from time import sleep\n",
     "chat2 = Chat(ms[1])\n",
     "stream_gen = chat2(\"Count to 5\", stream=True)\n",
     "for chunk in stream_gen:\n",