Merge pull request #27 from mikonapoli/main

Adds `call_func_async`

Author: jph00

01_funccall.ipynb

@@ -1798,6 +1798,80 @@
     "call_func('subs', {'a': 1, 'b': 2}, ns=mk_ns(d))"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "591574b8-6b53-4908-8159-b87be42133f7",
+   "metadata": {},
+   "source": [
+    "### Async function calling"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "96a3a7d3-31ef-4cc6-b47c-35eaa8bbff8b",
+   "metadata": {},
+   "source": [
+    "Since tools defined by MCP servers are async function, it is probably a good idea to have an async version of `call_func`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e273507b-6e4b-40bb-ae23-6397e89a4d51",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'asums': <function __main__.asums(a, b)>}"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "async def asums(a, b): return a + b\n",
+    "ns = mk_ns(asums); ns"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7ac04e80-7bb9-4b52-8285-454684605d47",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| exports\n",
+    "async def call_func_async(fc_name, fc_inputs, ns):\n",
+    "    \"Awaits the function `fc_name` with the given `fc_inputs` using namespace `ns`.\"\n",
+    "    if not isinstance(ns, abc.Mapping): ns = mk_ns(*ns)\n",
+    "    func = ns[fc_name]\n",
+    "    return await func(**fc_inputs)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b83998ac-68e2-4dbe-b594-65fb4fdf59b8",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "3"
+      ]
+     },
+     "execution_count": null,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "await call_func_async('asums', {'a': 1, 'b': 2}, ns=[asums])"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "94ec4289",

toolslm/_modidx.py

@@ -26,6 +26,7 @@
                                   'toolslm.funccall._run': ('funccall.html#_run', 'toolslm/funccall.py'),
                                   'toolslm.funccall._types': ('funccall.html#_types', 'toolslm/funccall.py'),
                                   'toolslm.funccall.call_func': ('funccall.html#call_func', 'toolslm/funccall.py'),
+                                  'toolslm.funccall.call_func_async': ('funccall.html#call_func_async', 'toolslm/funccall.py'),
                                   'toolslm.funccall.get_schema': ('funccall.html#get_schema', 'toolslm/funccall.py'),
                                   'toolslm.funccall.mk_ns': ('funccall.html#mk_ns', 'toolslm/funccall.py'),
                                   'toolslm.funccall.python': ('funccall.html#python', 'toolslm/funccall.py')},

toolslm/funccall.py

@@ -1,7 +1,7 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../01_funccall.ipynb.
 
 # %% auto 0
-__all__ = ['empty', 'custom_types', 'get_schema', 'PathArg', 'python', 'mk_ns', 'call_func']
+__all__ = ['empty', 'custom_types', 'get_schema', 'PathArg', 'python', 'mk_ns', 'call_func', 'call_func_async']
 
 # %% ../01_funccall.ipynb 2
 import inspect
@@ -191,3 +191,10 @@ def call_func(fc_name, fc_inputs, ns):
     if not isinstance(ns, abc.Mapping): ns = mk_ns(*ns)
     func = ns[fc_name]
     return func(**fc_inputs)
+
+# %% ../01_funccall.ipynb 97
+async def call_func_async(fc_name, fc_inputs, ns):
+    "Awaits the function `fc_name` with the given `fc_inputs` using namespace `ns`."
+    if not isinstance(ns, abc.Mapping): ns = mk_ns(*ns)
+    func = ns[fc_name]
+    return await func(**fc_inputs)