type annotations

Author: Solveit

nbs/00_core.ipynb

@@ -2009,6 +2009,17 @@
     "from fastcore.meta import delegates"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "162a5ba9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| export\n",
+    "from typing import Callable"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -2019,17 +2030,15 @@
     "#| export\n",
     "@delegates(completion)\n",
     "def structured(\n",
-    "    m,          # The LiteLLM model string\n",
-    "    msgs:list,  # A list of messages \n",
-    "    tool,       # The tool to be used for creating the structured output. Can also be a class, dataclass or Pydantic model        \n",
+    "    m:str,          # The LiteLLM model string\n",
+    "    msgs:list,      # A list of messages \n",
+    "    tool:Callable,  # The tool to be used for creating the structured output. Can also be a class, dataclass or Pydantic model        \n",
     "    **kwargs):\n",
     "    \"Return the value of the tool call (generally used for structured outputs)\"\n",
+    "\n",
     "    t = lite_mk_func(tool)\n",
-    "    \n",
     "    r = completion(m, msgs, tools=[t], tool_choice=t, **kwargs)\n",
-    "\n",
     "    args = json.loads(r.choices[0].message.tool_calls[0].function.arguments)\n",
-    "\n",
     "    return tool(**args)"
    ]
   },
@@ -2065,7 +2074,7 @@
     {
      "data": {
       "text/plain": [
-       "President(first='Thomas', last='Jefferson', spouse='Martha Jefferson Randolph', years_in_office='1801-1809', birthplace='Shadwell, Virginia', birth_year=1743)"
+       "President(first='Thomas', last='Jefferson', spouse='Martha Wayles Skelton', years_in_office='1801-1809', birthplace='Shadwell, Virginia, British America', birth_year=1743)"
       ]
      },
      "execution_count": null,