toolslm 0.0.2__py3-none-any.whl → 0.0.3__py3-none-any.whl

toolslm/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.2"
+__version__ = "0.0.3"

toolslm/_modidx.py

@@ -5,7 +5,13 @@ d = { 'settings': { 'branch': 'main',
                 'doc_host': 'https://AnswerDotAI.github.io',
                 'git_url': 'https://github.com/AnswerDotAI/toolslm',
                 'lib_path': 'toolslm'},
-  'syms': { 'toolslm.lmcode': { 'toolslm.lmcode._copy_loc': ('lmcode.html#_copy_loc', 'toolslm/lmcode.py'),
+  'syms': { 'toolslm.funccall': { 'toolslm.funccall._copy_loc': ('funccall.html#_copy_loc', 'toolslm/funccall.py'),
+                                  'toolslm.funccall._param': ('funccall.html#_param', 'toolslm/funccall.py'),
+                                  'toolslm.funccall._run': ('funccall.html#_run', 'toolslm/funccall.py'),
+                                  'toolslm.funccall._types': ('funccall.html#_types', 'toolslm/funccall.py'),
+                                  'toolslm.funccall.get_schema': ('funccall.html#get_schema', 'toolslm/funccall.py'),
+                                  'toolslm.funccall.python': ('funccall.html#python', 'toolslm/funccall.py')},
+            'toolslm.lmcode': { 'toolslm.lmcode._copy_loc': ('lmcode.html#_copy_loc', 'toolslm/lmcode.py'),
                                 'toolslm.lmcode._run': ('lmcode.html#_run', 'toolslm/lmcode.py'),
                                 'toolslm.lmcode.python': ('lmcode.html#python', 'toolslm/lmcode.py')},
             'toolslm.shell': { 'toolslm.shell.TerminalInteractiveShell.run_cell': ( 'shell.html#terminalinteractiveshell.run_cell',

toolslm/{lmcode.py → funccall.py}

@@ -1,13 +1,54 @@
-# AUTOGENERATED! DO NOT EDIT! File to edit: ../01_lmcode.ipynb.
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../01_funccall.ipynb.
 
 # %% auto 0
-__all__ = ['python']
+__all__ = ['empty', 'get_schema', 'python']
 
-# %% ../01_lmcode.ipynb 3
+# %% ../01_funccall.ipynb 2
+import inspect
+from fastcore.utils import *
+from fastcore.docments import docments
+
+# %% ../01_funccall.ipynb 3
+empty = inspect.Parameter.empty
+
+# %% ../01_funccall.ipynb 11
+def _types(t:type)->tuple[str,Optional[str]]:
+    "Tuple of json schema type name and (if appropriate) array item name."
+    if t is empty: raise TypeError('Missing type')
+    tmap = {int:"integer", float:"number", str:"string", bool:"boolean", list:"array", dict:"object"}
+    if getattr(t, '__origin__', None) in  (list,tuple): return "array", tmap.get(t.__args__[0], "object")
+    else: return tmap[t], None
+
+# %% ../01_funccall.ipynb 14
+def _param(name, info):
+    "json schema parameter given `name` and `info` from docments full dict."
+    paramt,itemt = _types(info.anno)
+    pschema = dict(type=paramt, description=info.docment or "")
+    if itemt: pschema["items"] = {"type": itemt}
+    if info.default is not empty: pschema["default"] = info.default
+    return pschema
+
+# %% ../01_funccall.ipynb 17
+def get_schema(f:callable)->dict:
+    "Convert function `f` into a JSON schema `dict` for tool use."
+    d = docments(f, full=True)
+    ret = d.pop('return')
+    d.pop('self', None) # Ignore `self` for methods
+    paramd = {
+        'type': "object",
+        'properties': {n:_param(n,o) for n,o in d.items() if n[0]!='_'},
+        'required': [n for n,o in d.items() if o.default is empty and n[0]!='_']
+    }
+    desc = f.__doc__
+    if ret.anno is not empty: desc += f'\n\nReturns:\n- type: {_types(ret.anno)[0]}'
+    if ret.docment: desc += f'\n- description: {ret.docment}'
+    return dict(name=f.__name__, description=desc or "", input_schema=paramd)
+
+# %% ../01_funccall.ipynb 22
 import ast, time, signal, traceback
 from fastcore.utils import *
 
-# %% ../01_lmcode.ipynb 5
+# %% ../01_funccall.ipynb 23
 def _copy_loc(new, orig):
     "Copy location information from original node to new node and all children."
     new = ast.copy_location(new, orig)
@@ -16,7 +57,7 @@ def _copy_loc(new, orig):
         elif isinstance(o, list): setattr(new, field, [_copy_loc(value, orig) for value in o])
     return new
 
-# %% ../01_lmcode.ipynb 7
+# %% ../01_funccall.ipynb 25
 def _run(code:str ):
     "Run `code`, returning final expression (similar to IPython)"
     tree = ast.parse(code)
@@ -39,7 +80,7 @@ def _run(code:str ):
     if _result is not None: return _result
     return stdout_buffer.getvalue().strip()
 
-# %% ../01_lmcode.ipynb 12
+# %% ../01_funccall.ipynb 30
 def python(code, # Code to execute
            timeout=5 # Maximum run time in seconds before a `TimeoutError` is raised
           ): # Result of last node, if it's an expression, or `None` otherwise

{toolslm-0.0.2.dist-info → toolslm-0.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: toolslm
-Version: 0.0.2
+Version: 0.0.3
 Summary: Tools to make language models a bit easier to use
 Home-page: https://github.com/AnswerDotAI/toolslm
 Author: Jeremy Howard

toolslm-0.0.3.dist-info/RECORD

@@ -0,0 +1,11 @@
+toolslm/__init__.py,sha256=4GZKi13lDTD25YBkGakhZyEQZWTER_OWQMNPoH_UM2c,22
+toolslm/_modidx.py,sha256=mf2R89-fA85sjtgETrs_981ZiKBXwlnfWSo7FYPQDBg,2531
+toolslm/funccall.py,sha256=VLJ16OPeyv4QwMY6pTOGig0uep2QbEvN4lXIk3OfqEw,3809
+toolslm/shell.py,sha256=GVqfL74NHw66zzZ7jvGVLjE55ZNJGBPvEb8kLz4aoYc,1576
+toolslm/xml.py,sha256=4DnFUYG09fB1ZTzcyeugKdkMU40MMMf3zZEHbDfSGIo,4635
+toolslm-0.0.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+toolslm-0.0.3.dist-info/METADATA,sha256=xdGav82GXXWNJUDvGg2Hr6WXrDmUHR3zXeJ45FUC4J8,3773
+toolslm-0.0.3.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+toolslm-0.0.3.dist-info/entry_points.txt,sha256=3os5YWuvVIFuweYaNtnwgv-xCe4Gb1KcZndFz8KWI0E,36
+toolslm-0.0.3.dist-info/top_level.txt,sha256=4hRTrFWayz_Kz5221XjvlpCwVFrW3WPi1P0fllkTq9s,8
+toolslm-0.0.3.dist-info/RECORD,,

toolslm-0.0.2.dist-info/RECORD

@@ -1,11 +0,0 @@
-toolslm/__init__.py,sha256=QvlVh4JTl3JL7jQAja76yKtT-IvF4631ASjWY1wS6AQ,22
-toolslm/_modidx.py,sha256=fmKi8RF2pdYu9iFlVY__PTCgOTaWgWYqUfrQBjimhxA,1860
-toolslm/lmcode.py,sha256=G28x34SsxQPGhSNGHggXh6dWaeGlxQH69nAu9tgdNF8,2104
-toolslm/shell.py,sha256=GVqfL74NHw66zzZ7jvGVLjE55ZNJGBPvEb8kLz4aoYc,1576
-toolslm/xml.py,sha256=4DnFUYG09fB1ZTzcyeugKdkMU40MMMf3zZEHbDfSGIo,4635
-toolslm-0.0.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-toolslm-0.0.2.dist-info/METADATA,sha256=KHlZaY6NAr7a0DCNTeq6rHxaJO5cze8jciBAT4Z7C-g,3773
-toolslm-0.0.2.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-toolslm-0.0.2.dist-info/entry_points.txt,sha256=3os5YWuvVIFuweYaNtnwgv-xCe4Gb1KcZndFz8KWI0E,36
-toolslm-0.0.2.dist-info/top_level.txt,sha256=4hRTrFWayz_Kz5221XjvlpCwVFrW3WPi1P0fllkTq9s,8
-toolslm-0.0.2.dist-info/RECORD,,