notebooks/parallel-function-calling.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m A new release of pip is available: \u001b[0m\u001b[31;49m23.3.1\u001b[0m\u001b[39;49m -> \u001b[0m\u001b[32;49m23.3.2\u001b[0m\n",
      "\u001b[1m[\u001b[0m\u001b[34;49mnotice\u001b[0m\u001b[1;39;49m]\u001b[0m\u001b[39;49m To update, run: \u001b[0m\u001b[32;49mpip install --upgrade pip\u001b[0m\n",
      "Note: you may need to restart the kernel to use updated packages.\n"
     ]
    }
   ],
   "source": [
    "%pip install sqlmodel -q"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "from typing import Optional\n",
    "from sqlmodel import SQLModel, Field, create_engine, Session\n",
    "\n",
    "# SQL Model uses Pydantic Models under the hood\n",
    "\n",
    "class Character(SQLModel, table=True):\n",
    "    id: Optional[int] = Field(default=None, primary_key=True)\n",
    "    name: str\n",
    "    race: str\n",
    "    character_class: str\n",
    "    level: int\n",
    "    background: str\n",
    "    player_name: Optional[str] = None\n",
    "    experience_points: int = 0\n",
    "    strength: int\n",
    "    dexterity: int\n",
    "    constitution: int\n",
    "    intelligence: int\n",
    "    wisdom: int\n",
    "    charisma: int\n",
    "    hit_points: int\n",
    "    armor_class: int\n",
    "    alignment: str\n",
    "    skills: str  # Storing as comma-separated string\n",
    "    languages: str  # Storing as comma-separated string\n",
    "    equipment: str  # Storing as comma-separated string\n",
    "    spells: Optional[str] = None  # Storing as comma-separated string\n",
    "\n",
    "    def _repr_llm_(self):\n",
    "        return f\"<Character {self.id} {self.name}>\"\n",
    "    \n",
    "    def __repr__(self):\n",
    "        return f\"<Character {self.id} {self.name}>\"\n",
    "\n",
    "# SQLite Database URL\n",
    "DATABASE_URL = \"sqlite:///:memory:\"\n",
    "engine = create_engine(DATABASE_URL)\n",
    "\n",
    "# Create the database tables\n",
    "SQLModel.metadata.create_all(engine)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/svg+xml": [
       "<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"60\" height=\"60\" viewBox=\"0 0 1200 1200\">\n",
       "    <path d=\"M1102.8 896.4c1.2-1.2 1.2-2.398 1.2-2.398V318.004c0-1.2-1.2-1.2-1.2-2.399v-1.199c0-1.2-1.2-1.2-1.2-1.2v-1.198l-1.198-1.2s-1.2-1.199-2.399-1.199h-1.199l-486-282.01c-7.2-3.601-15.602-3.601-21.602 0l-483.6 279.6c-1.2 0-1.2 1.2-2.398 1.2h-1.2l-3.597 3.597s0 1.2-1.2 1.2l-1.199 1.198v1.2s0 1.199-1.199 1.199v571.196c0 1.198 1.2 2.397 1.2 3.6 0 1.2 1.198 2.4 2.398 3.602v1.2C98.399 898.8 99.602 900 100.8 900l2.399 2.398h2.398l483.6 279.6s1.2 0 1.2 1.2h17.988s1.198 0 1.198-1.2l483.6-279.6h2.399s1.199 0 1.199-1.199l1.2-1.2s1.198 0 1.198-1.198l1.2-1.2s0-1.199 1.199-1.199c.02 1.2.02-.004 1.219-.004zm-960-445.2 134.4 333.6c1.2 3.602 0 8.398-4.8 10.801l-123.6 51.602c-4.802 2.398-10.802-1.2-10.802-7.2l.004-387.61c0-2.398 3.602-3.597 4.8-1.198zm696 331.2H361.2c-6 0-9.602-7.2-7.2-12l238.8-414c3.602-4.8 10.802-4.8 13.2 0l238.8 414c3.598 6 0 12-6 12zm-531.6-43.199-154.8-382.8c-2.398-4.8 1.2-10.801 7.2-10.801L546 326.401c6 0 10.8 6 7.198 12l-232.8 402c-2.398 6-10.797 6-13.199-1.2zm526.8 100.8-228 280.8c-3.602 3.602-9.602 3.602-12 0l-228-280.8c-3.602-4.8 0-13.199 6-13.199h456c6 0 9.602 8.398 6 13.199zm45.602-99.602-232.8-402c-3.602-6 1.2-12 7.2-12l386.4 19.199c6 0 8.398 6 7.198 10.801l-154.8 382.8c-2.402 7.2-10.8 7.2-13.199 1.2zm-258-466.8v-175.2c0-6 7.2-9.602 12-7.2l350.4 202.8c2.398 1.2 1.2 4.802-1.2 4.802l-354-18c-3.6 0-7.198-3.602-7.198-7.2zm-51.602 7.2-354 18c-2.398 0-3.602-3.602-1.2-4.802l351.6-201.6c4.802-3.601 12 1.2 12 7.2v174c0 3.601-3.597 7.203-8.398 7.203zm-264 555.6 193.2 238.8c1.2 2.398-1.2 4.8-3.602 3.6l-315.6-182.4c-6-3.6-4.8-12 1.2-14.397l116.4-48c2.402-1.2 6.003 0 8.402 2.398zm596.4-2.4 116.4 48c6 2.4 7.2 10.802 1.2 14.399l-316.8 182.4c-2.4 1.199-4.802-1.2-3.602-3.602l193.2-238.8c3.601-2.398 7.203-3.597 9.601-2.398zm148.8 13.2-123.6-51.602c-3.602-1.2-6-6-4.8-10.801l134.4-333.6c1.198-2.399 4.8-1.2 4.8 1.199l.004 387.61c0 6-6 9.602-10.801 7.2z\"/>\n",
       "    <text x=\"600\" y=\"660\" font-family=\"Helvetica\" font-size=\"220\" font-weight=\"bold\" fill=\"black\" text-anchor=\"middle\" alignment-baseline=\"middle\">\n",
       "        13\n",
       "    </text>\n",
       "</svg>"
      ],
      "text/plain": [
       "<IPython.core.display.SVG object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "text/plain": [
       "13"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import random\n",
    "from IPython.display import SVG\n",
    "\n",
    "def d20(roll):\n",
    "    return SVG(f\"\"\"<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"60\" height=\"60\" viewBox=\"0 0 1200 1200\">\n",
    "    <path d=\"M1102.8 896.4c1.2-1.2 1.2-2.398 1.2-2.398V318.004c0-1.2-1.2-1.2-1.2-2.399v-1.199c0-1.2-1.2-1.2-1.2-1.2v-1.198l-1.198-1.2s-1.2-1.199-2.399-1.199h-1.199l-486-282.01c-7.2-3.601-15.602-3.601-21.602 0l-483.6 279.6c-1.2 0-1.2 1.2-2.398 1.2h-1.2l-3.597 3.597s0 1.2-1.2 1.2l-1.199 1.198v1.2s0 1.199-1.199 1.199v571.196c0 1.198 1.2 2.397 1.2 3.6 0 1.2 1.198 2.4 2.398 3.602v1.2C98.399 898.8 99.602 900 100.8 900l2.399 2.398h2.398l483.6 279.6s1.2 0 1.2 1.2h17.988s1.198 0 1.198-1.2l483.6-279.6h2.399s1.199 0 1.199-1.199l1.2-1.2s1.198 0 1.198-1.198l1.2-1.2s0-1.199 1.199-1.199c.02 1.2.02-.004 1.219-.004zm-960-445.2 134.4 333.6c1.2 3.602 0 8.398-4.8 10.801l-123.6 51.602c-4.802 2.398-10.802-1.2-10.802-7.2l.004-387.61c0-2.398 3.602-3.597 4.8-1.198zm696 331.2H361.2c-6 0-9.602-7.2-7.2-12l238.8-414c3.602-4.8 10.802-4.8 13.2 0l238.8 414c3.598 6 0 12-6 12zm-531.6-43.199-154.8-382.8c-2.398-4.8 1.2-10.801 7.2-10.801L546 326.401c6 0 10.8 6 7.198 12l-232.8 402c-2.398 6-10.797 6-13.199-1.2zm526.8 100.8-228 280.8c-3.602 3.602-9.602 3.602-12 0l-228-280.8c-3.602-4.8 0-13.199 6-13.199h456c6 0 9.602 8.398 6 13.199zm45.602-99.602-232.8-402c-3.602-6 1.2-12 7.2-12l386.4 19.199c6 0 8.398 6 7.198 10.801l-154.8 382.8c-2.402 7.2-10.8 7.2-13.199 1.2zm-258-466.8v-175.2c0-6 7.2-9.602 12-7.2l350.4 202.8c2.398 1.2 1.2 4.802-1.2 4.802l-354-18c-3.6 0-7.198-3.602-7.198-7.2zm-51.602 7.2-354 18c-2.398 0-3.602-3.602-1.2-4.802l351.6-201.6c4.802-3.601 12 1.2 12 7.2v174c0 3.601-3.597 7.203-8.398 7.203zm-264 555.6 193.2 238.8c1.2 2.398-1.2 4.8-3.602 3.6l-315.6-182.4c-6-3.6-4.8-12 1.2-14.397l116.4-48c2.402-1.2 6.003 0 8.402 2.398zm596.4-2.4 116.4 48c6 2.4 7.2 10.802 1.2 14.399l-316.8 182.4c-2.4 1.199-4.802-1.2-3.602-3.602l193.2-238.8c3.601-2.398 7.203-3.597 9.601-2.398zm148.8 13.2-123.6-51.602c-3.602-1.2-6-6-4.8-10.801l134.4-333.6c1.198-2.399 4.8-1.2 4.8 1.199l.004 387.61c0 6-6 9.602-10.801 7.2z\"/>\n",
    "    <text x=\"600\" y=\"660\" font-family=\"Helvetica\" font-size=\"220\" font-weight=\"bold\" fill=\"black\" text-anchor=\"middle\" alignment-baseline=\"middle\">\n",
    "        {roll}\n",
    "    </text>\n",
    "</svg>\"\"\")\n",
    "\n",
    "def roll_die(sides: int = 6):\n",
    "    \"\"\"Roll a die with the given number of sides.\"\"\"\n",
    "    roll = random.randint(1, sides)\n",
    "\n",
    "    if(sides == 20):\n",
    "        display(d20(roll))\n",
    "    \n",
    "    return roll\n",
    "\n",
    "# Function to add a new character\n",
    "def add_character(character: Character):\n",
    "    \"\"\"Adds a character to our characters database\"\"\"\n",
    "    with Session(engine) as session:\n",
    "        session.add(character)\n",
    "        session.commit\n",
    "\n",
    "    return character\n",
    "\n",
    "roll_die(20)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "from chatlab import FunctionRegistry\n",
    "\n",
    "fr = FunctionRegistry()\n",
    "fr.register_functions([roll_die, add_character])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "from openai import OpenAI\n",
    "client = OpenAI()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "from chatlab import tool_result\n",
    "\n",
    "async def chatloop(initial_messages):\n",
    "    \"\"\"Emit messages encountered as well as tool results, making sure to autorun tools and respond to the model.\"\"\"\n",
    "    buffer = initial_messages.copy()\n",
    "\n",
    "    resp = client.chat.completions.create(\n",
    "        model=\"gpt-3.5-turbo-1106\",\n",
    "        messages=initial_messages,\n",
    "\n",
    "        # Pass in the tools from the function registry. The model will choose\n",
    "        # whether it uses 0, 1, 2, or N many tools.\n",
    "        tools=fr.tools,\n",
    "        tool_choice=\"auto\"\n",
    "    )\n",
    "\n",
    "    message = resp.choices[0].message\n",
    "    buffer.append(message)\n",
    "\n",
    "    yield message\n",
    "\n",
    "    # call each of the tools\n",
    "    if message.tool_calls is not None:\n",
    "        for tool in message.tool_calls:\n",
    "            result = await fr.call(tool.function.name, tool.function.arguments)\n",
    "\n",
    "            # An assistant message with 'tool_calls' must be followed by tool messages responding to each 'tool_call_id'.\n",
    "            tool_call_response = tool_result(tool.id, content=str(result))\n",
    "            yield tool_call_response\n",
    "            buffer.append(tool_call_response)\n",
    "        \n",
    "        # Once all tools have been called, call the model again\n",
    "        async for m in chatloop(buffer):\n",
    "            yield m\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/svg+xml": [
       "<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"60\" height=\"60\" viewBox=\"0 0 1200 1200\">\n",
       "    <path d=\"M1102.8 896.4c1.2-1.2 1.2-2.398 1.2-2.398V318.004c0-1.2-1.2-1.2-1.2-2.399v-1.199c0-1.2-1.2-1.2-1.2-1.2v-1.198l-1.198-1.2s-1.2-1.199-2.399-1.199h-1.199l-486-282.01c-7.2-3.601-15.602-3.601-21.602 0l-483.6 279.6c-1.2 0-1.2 1.2-2.398 1.2h-1.2l-3.597 3.597s0 1.2-1.2 1.2l-1.199 1.198v1.2s0 1.199-1.199 1.199v571.196c0 1.198 1.2 2.397 1.2 3.6 0 1.2 1.198 2.4 2.398 3.602v1.2C98.399 898.8 99.602 900 100.8 900l2.399 2.398h2.398l483.6 279.6s1.2 0 1.2 1.2h17.988s1.198 0 1.198-1.2l483.6-279.6h2.399s1.199 0 1.199-1.199l1.2-1.2s1.198 0 1.198-1.198l1.2-1.2s0-1.199 1.199-1.199c.02 1.2.02-.004 1.219-.004zm-960-445.2 134.4 333.6c1.2 3.602 0 8.398-4.8 10.801l-123.6 51.602c-4.802 2.398-10.802-1.2-10.802-7.2l.004-387.61c0-2.398 3.602-3.597 4.8-1.198zm696 331.2H361.2c-6 0-9.602-7.2-7.2-12l238.8-414c3.602-4.8 10.802-4.8 13.2 0l238.8 414c3.598 6 0 12-6 12zm-531.6-43.199-154.8-382.8c-2.398-4.8 1.2-10.801 7.2-10.801L546 326.401c6 0 10.8 6 7.198 12l-232.8 402c-2.398 6-10.797 6-13.199-1.2zm526.8 100.8-228 280.8c-3.602 3.602-9.602 3.602-12 0l-228-280.8c-3.602-4.8 0-13.199 6-13.199h456c6 0 9.602 8.398 6 13.199zm45.602-99.602-232.8-402c-3.602-6 1.2-12 7.2-12l386.4 19.199c6 0 8.398 6 7.198 10.801l-154.8 382.8c-2.402 7.2-10.8 7.2-13.199 1.2zm-258-466.8v-175.2c0-6 7.2-9.602 12-7.2l350.4 202.8c2.398 1.2 1.2 4.802-1.2 4.802l-354-18c-3.6 0-7.198-3.602-7.198-7.2zm-51.602 7.2-354 18c-2.398 0-3.602-3.602-1.2-4.802l351.6-201.6c4.802-3.601 12 1.2 12 7.2v174c0 3.601-3.597 7.203-8.398 7.203zm-264 555.6 193.2 238.8c1.2 2.398-1.2 4.8-3.602 3.6l-315.6-182.4c-6-3.6-4.8-12 1.2-14.397l116.4-48c2.402-1.2 6.003 0 8.402 2.398zm596.4-2.4 116.4 48c6 2.4 7.2 10.802 1.2 14.399l-316.8 182.4c-2.4 1.199-4.802-1.2-3.602-3.602l193.2-238.8c3.601-2.398 7.203-3.597 9.601-2.398zm148.8 13.2-123.6-51.602c-3.602-1.2-6-6-4.8-10.801l134.4-333.6c1.198-2.399 4.8-1.2 4.8 1.199l.004 387.61c0 6-6 9.602-10.801 7.2z\"/>\n",
       "    <text x=\"600\" y=\"660\" font-family=\"Helvetica\" font-size=\"220\" font-weight=\"bold\" fill=\"black\" text-anchor=\"middle\" alignment-baseline=\"middle\">\n",
       "        6\n",
       "    </text>\n",
       "</svg>"
      ],
      "text/plain": [
       "<IPython.core.display.SVG object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "image/svg+xml": [
       "<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"60\" height=\"60\" viewBox=\"0 0 1200 1200\">\n",
       "    <path d=\"M1102.8 896.4c1.2-1.2 1.2-2.398 1.2-2.398V318.004c0-1.2-1.2-1.2-1.2-2.399v-1.199c0-1.2-1.2-1.2-1.2-1.2v-1.198l-1.198-1.2s-1.2-1.199-2.399-1.199h-1.199l-486-282.01c-7.2-3.601-15.602-3.601-21.602 0l-483.6 279.6c-1.2 0-1.2 1.2-2.398 1.2h-1.2l-3.597 3.597s0 1.2-1.2 1.2l-1.199 1.198v1.2s0 1.199-1.199 1.199v571.196c0 1.198 1.2 2.397 1.2 3.6 0 1.2 1.198 2.4 2.398 3.602v1.2C98.399 898.8 99.602 900 100.8 900l2.399 2.398h2.398l483.6 279.6s1.2 0 1.2 1.2h17.988s1.198 0 1.198-1.2l483.6-279.6h2.399s1.199 0 1.199-1.199l1.2-1.2s1.198 0 1.198-1.198l1.2-1.2s0-1.199 1.199-1.199c.02 1.2.02-.004 1.219-.004zm-960-445.2 134.4 333.6c1.2 3.602 0 8.398-4.8 10.801l-123.6 51.602c-4.802 2.398-10.802-1.2-10.802-7.2l.004-387.61c0-2.398 3.602-3.597 4.8-1.198zm696 331.2H361.2c-6 0-9.602-7.2-7.2-12l238.8-414c3.602-4.8 10.802-4.8 13.2 0l238.8 414c3.598 6 0 12-6 12zm-531.6-43.199-154.8-382.8c-2.398-4.8 1.2-10.801 7.2-10.801L546 326.401c6 0 10.8 6 7.198 12l-232.8 402c-2.398 6-10.797 6-13.199-1.2zm526.8 100.8-228 280.8c-3.602 3.602-9.602 3.602-12 0l-228-280.8c-3.602-4.8 0-13.199 6-13.199h456c6 0 9.602 8.398 6 13.199zm45.602-99.602-232.8-402c-3.602-6 1.2-12 7.2-12l386.4 19.199c6 0 8.398 6 7.198 10.801l-154.8 382.8c-2.402 7.2-10.8 7.2-13.199 1.2zm-258-466.8v-175.2c0-6 7.2-9.602 12-7.2l350.4 202.8c2.398 1.2 1.2 4.802-1.2 4.802l-354-18c-3.6 0-7.198-3.602-7.198-7.2zm-51.602 7.2-354 18c-2.398 0-3.602-3.602-1.2-4.802l351.6-201.6c4.802-3.601 12 1.2 12 7.2v174c0 3.601-3.597 7.203-8.398 7.203zm-264 555.6 193.2 238.8c1.2 2.398-1.2 4.8-3.602 3.6l-315.6-182.4c-6-3.6-4.8-12 1.2-14.397l116.4-48c2.402-1.2 6.003 0 8.402 2.398zm596.4-2.4 116.4 48c6 2.4 7.2 10.802 1.2 14.399l-316.8 182.4c-2.4 1.199-4.802-1.2-3.602-3.602l193.2-238.8c3.601-2.398 7.203-3.597 9.601-2.398zm148.8 13.2-123.6-51.602c-3.602-1.2-6-6-4.8-10.801l134.4-333.6c1.198-2.399 4.8-1.2 4.8 1.199l.004 387.61c0 6-6 9.602-10.801 7.2z\"/>\n",
       "    <text x=\"600\" y=\"660\" font-family=\"Helvetica\" font-size=\"220\" font-weight=\"bold\" fill=\"black\" text-anchor=\"middle\" alignment-baseline=\"middle\">\n",
       "        4\n",
       "    </text>\n",
       "</svg>"
      ],
      "text/plain": [
       "<IPython.core.display.SVG object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "image/svg+xml": [
       "<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"60\" height=\"60\" viewBox=\"0 0 1200 1200\">\n",
       "    <path d=\"M1102.8 896.4c1.2-1.2 1.2-2.398 1.2-2.398V318.004c0-1.2-1.2-1.2-1.2-2.399v-1.199c0-1.2-1.2-1.2-1.2-1.2v-1.198l-1.198-1.2s-1.2-1.199-2.399-1.199h-1.199l-486-282.01c-7.2-3.601-15.602-3.601-21.602 0l-483.6 279.6c-1.2 0-1.2 1.2-2.398 1.2h-1.2l-3.597 3.597s0 1.2-1.2 1.2l-1.199 1.198v1.2s0 1.199-1.199 1.199v571.196c0 1.198 1.2 2.397 1.2 3.6 0 1.2 1.198 2.4 2.398 3.602v1.2C98.399 898.8 99.602 900 100.8 900l2.399 2.398h2.398l483.6 279.6s1.2 0 1.2 1.2h17.988s1.198 0 1.198-1.2l483.6-279.6h2.399s1.199 0 1.199-1.199l1.2-1.2s1.198 0 1.198-1.198l1.2-1.2s0-1.199 1.199-1.199c.02 1.2.02-.004 1.219-.004zm-960-445.2 134.4 333.6c1.2 3.602 0 8.398-4.8 10.801l-123.6 51.602c-4.802 2.398-10.802-1.2-10.802-7.2l.004-387.61c0-2.398 3.602-3.597 4.8-1.198zm696 331.2H361.2c-6 0-9.602-7.2-7.2-12l238.8-414c3.602-4.8 10.802-4.8 13.2 0l238.8 414c3.598 6 0 12-6 12zm-531.6-43.199-154.8-382.8c-2.398-4.8 1.2-10.801 7.2-10.801L546 326.401c6 0 10.8 6 7.198 12l-232.8 402c-2.398 6-10.797 6-13.199-1.2zm526.8 100.8-228 280.8c-3.602 3.602-9.602 3.602-12 0l-228-280.8c-3.602-4.8 0-13.199 6-13.199h456c6 0 9.602 8.398 6 13.199zm45.602-99.602-232.8-402c-3.602-6 1.2-12 7.2-12l386.4 19.199c6 0 8.398 6 7.198 10.801l-154.8 382.8c-2.402 7.2-10.8 7.2-13.199 1.2zm-258-466.8v-175.2c0-6 7.2-9.602 12-7.2l350.4 202.8c2.398 1.2 1.2 4.802-1.2 4.802l-354-18c-3.6 0-7.198-3.602-7.198-7.2zm-51.602 7.2-354 18c-2.398 0-3.602-3.602-1.2-4.802l351.6-201.6c4.802-3.601 12 1.2 12 7.2v174c0 3.601-3.597 7.203-8.398 7.203zm-264 555.6 193.2 238.8c1.2 2.398-1.2 4.8-3.602 3.6l-315.6-182.4c-6-3.6-4.8-12 1.2-14.397l116.4-48c2.402-1.2 6.003 0 8.402 2.398zm596.4-2.4 116.4 48c6 2.4 7.2 10.802 1.2 14.399l-316.8 182.4c-2.4 1.199-4.802-1.2-3.602-3.602l193.2-238.8c3.601-2.398 7.203-3.597 9.601-2.398zm148.8 13.2-123.6-51.602c-3.602-1.2-6-6-4.8-10.801l134.4-333.6c1.198-2.399 4.8-1.2 4.8 1.199l.004 387.61c0 6-6 9.602-10.801 7.2z\"/>\n",
       "    <text x=\"600\" y=\"660\" font-family=\"Helvetica\" font-size=\"220\" font-weight=\"bold\" fill=\"black\" text-anchor=\"middle\" alignment-baseline=\"middle\">\n",
       "        8\n",
       "    </text>\n",
       "</svg>"
      ],
      "text/plain": [
       "<IPython.core.display.SVG object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "image/svg+xml": [
       "<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"60\" height=\"60\" viewBox=\"0 0 1200 1200\">\n",
       "    <path d=\"M1102.8 896.4c1.2-1.2 1.2-2.398 1.2-2.398V318.004c0-1.2-1.2-1.2-1.2-2.399v-1.199c0-1.2-1.2-1.2-1.2-1.2v-1.198l-1.198-1.2s-1.2-1.199-2.399-1.199h-1.199l-486-282.01c-7.2-3.601-15.602-3.601-21.602 0l-483.6 279.6c-1.2 0-1.2 1.2-2.398 1.2h-1.2l-3.597 3.597s0 1.2-1.2 1.2l-1.199 1.198v1.2s0 1.199-1.199 1.199v571.196c0 1.198 1.2 2.397 1.2 3.6 0 1.2 1.198 2.4 2.398 3.602v1.2C98.399 898.8 99.602 900 100.8 900l2.399 2.398h2.398l483.6 279.6s1.2 0 1.2 1.2h17.988s1.198 0 1.198-1.2l483.6-279.6h2.399s1.199 0 1.199-1.199l1.2-1.2s1.198 0 1.198-1.198l1.2-1.2s0-1.199 1.199-1.199c.02 1.2.02-.004 1.219-.004zm-960-445.2 134.4 333.6c1.2 3.602 0 8.398-4.8 10.801l-123.6 51.602c-4.802 2.398-10.802-1.2-10.802-7.2l.004-387.61c0-2.398 3.602-3.597 4.8-1.198zm696 331.2H361.2c-6 0-9.602-7.2-7.2-12l238.8-414c3.602-4.8 10.802-4.8 13.2 0l238.8 414c3.598 6 0 12-6 12zm-531.6-43.199-154.8-382.8c-2.398-4.8 1.2-10.801 7.2-10.801L546 326.401c6 0 10.8 6 7.198 12l-232.8 402c-2.398 6-10.797 6-13.199-1.2zm526.8 100.8-228 280.8c-3.602 3.602-9.602 3.602-12 0l-228-280.8c-3.602-4.8 0-13.199 6-13.199h456c6 0 9.602 8.398 6 13.199zm45.602-99.602-232.8-402c-3.602-6 1.2-12 7.2-12l386.4 19.199c6 0 8.398 6 7.198 10.801l-154.8 382.8c-2.402 7.2-10.8 7.2-13.199 1.2zm-258-466.8v-175.2c0-6 7.2-9.602 12-7.2l350.4 202.8c2.398 1.2 1.2 4.802-1.2 4.802l-354-18c-3.6 0-7.198-3.602-7.198-7.2zm-51.602 7.2-354 18c-2.398 0-3.602-3.602-1.2-4.802l351.6-201.6c4.802-3.601 12 1.2 12 7.2v174c0 3.601-3.597 7.203-8.398 7.203zm-264 555.6 193.2 238.8c1.2 2.398-1.2 4.8-3.602 3.6l-315.6-182.4c-6-3.6-4.8-12 1.2-14.397l116.4-48c2.402-1.2 6.003 0 8.402 2.398zm596.4-2.4 116.4 48c6 2.4 7.2 10.802 1.2 14.399l-316.8 182.4c-2.4 1.199-4.802-1.2-3.602-3.602l193.2-238.8c3.601-2.398 7.203-3.597 9.601-2.398zm148.8 13.2-123.6-51.602c-3.602-1.2-6-6-4.8-10.801l134.4-333.6c1.198-2.399 4.8-1.2 4.8 1.199l.004 387.61c0 6-6 9.602-10.801 7.2z\"/>\n",
       "    <text x=\"600\" y=\"660\" font-family=\"Helvetica\" font-size=\"220\" font-weight=\"bold\" fill=\"black\" text-anchor=\"middle\" alignment-baseline=\"middle\">\n",
       "        10\n",
       "    </text>\n",
       "</svg>"
      ],
      "text/plain": [
       "<IPython.core.display.SVG object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "image/svg+xml": [
       "<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"60\" height=\"60\" viewBox=\"0 0 1200 1200\">\n",
       "    <path d=\"M1102.8 896.4c1.2-1.2 1.2-2.398 1.2-2.398V318.004c0-1.2-1.2-1.2-1.2-2.399v-1.199c0-1.2-1.2-1.2-1.2-1.2v-1.198l-1.198-1.2s-1.2-1.199-2.399-1.199h-1.199l-486-282.01c-7.2-3.601-15.602-3.601-21.602 0l-483.6 279.6c-1.2 0-1.2 1.2-2.398 1.2h-1.2l-3.597 3.597s0 1.2-1.2 1.2l-1.199 1.198v1.2s0 1.199-1.199 1.199v571.196c0 1.198 1.2 2.397 1.2 3.6 0 1.2 1.198 2.4 2.398 3.602v1.2C98.399 898.8 99.602 900 100.8 900l2.399 2.398h2.398l483.6 279.6s1.2 0 1.2 1.2h17.988s1.198 0 1.198-1.2l483.6-279.6h2.399s1.199 0 1.199-1.199l1.2-1.2s1.198 0 1.198-1.198l1.2-1.2s0-1.199 1.199-1.199c.02 1.2.02-.004 1.219-.004zm-960-445.2 134.4 333.6c1.2 3.602 0 8.398-4.8 10.801l-123.6 51.602c-4.802 2.398-10.802-1.2-10.802-7.2l.004-387.61c0-2.398 3.602-3.597 4.8-1.198zm696 331.2H361.2c-6 0-9.602-7.2-7.2-12l238.8-414c3.602-4.8 10.802-4.8 13.2 0l238.8 414c3.598 6 0 12-6 12zm-531.6-43.199-154.8-382.8c-2.398-4.8 1.2-10.801 7.2-10.801L546 326.401c6 0 10.8 6 7.198 12l-232.8 402c-2.398 6-10.797 6-13.199-1.2zm526.8 100.8-228 280.8c-3.602 3.602-9.602 3.602-12 0l-228-280.8c-3.602-4.8 0-13.199 6-13.199h456c6 0 9.602 8.398 6 13.199zm45.602-99.602-232.8-402c-3.602-6 1.2-12 7.2-12l386.4 19.199c6 0 8.398 6 7.198 10.801l-154.8 382.8c-2.402 7.2-10.8 7.2-13.199 1.2zm-258-466.8v-175.2c0-6 7.2-9.602 12-7.2l350.4 202.8c2.398 1.2 1.2 4.802-1.2 4.802l-354-18c-3.6 0-7.198-3.602-7.198-7.2zm-51.602 7.2-354 18c-2.398 0-3.602-3.602-1.2-4.802l351.6-201.6c4.802-3.601 12 1.2 12 7.2v174c0 3.601-3.597 7.203-8.398 7.203zm-264 555.6 193.2 238.8c1.2 2.398-1.2 4.8-3.602 3.6l-315.6-182.4c-6-3.6-4.8-12 1.2-14.397l116.4-48c2.402-1.2 6.003 0 8.402 2.398zm596.4-2.4 116.4 48c6 2.4 7.2 10.802 1.2 14.399l-316.8 182.4c-2.4 1.199-4.802-1.2-3.602-3.602l193.2-238.8c3.601-2.398 7.203-3.597 9.601-2.398zm148.8 13.2-123.6-51.602c-3.602-1.2-6-6-4.8-10.801l134.4-333.6c1.198-2.399 4.8-1.2 4.8 1.199l.004 387.61c0 6-6 9.602-10.801 7.2z\"/>\n",
       "    <text x=\"600\" y=\"660\" font-family=\"Helvetica\" font-size=\"220\" font-weight=\"bold\" fill=\"black\" text-anchor=\"middle\" alignment-baseline=\"middle\">\n",
       "        10\n",
       "    </text>\n",
       "</svg>"
      ],
      "text/plain": [
       "<IPython.core.display.SVG object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "image/svg+xml": [
       "<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"60\" height=\"60\" viewBox=\"0 0 1200 1200\">\n",
       "    <path d=\"M1102.8 896.4c1.2-1.2 1.2-2.398 1.2-2.398V318.004c0-1.2-1.2-1.2-1.2-2.399v-1.199c0-1.2-1.2-1.2-1.2-1.2v-1.198l-1.198-1.2s-1.2-1.199-2.399-1.199h-1.199l-486-282.01c-7.2-3.601-15.602-3.601-21.602 0l-483.6 279.6c-1.2 0-1.2 1.2-2.398 1.2h-1.2l-3.597 3.597s0 1.2-1.2 1.2l-1.199 1.198v1.2s0 1.199-1.199 1.199v571.196c0 1.198 1.2 2.397 1.2 3.6 0 1.2 1.198 2.4 2.398 3.602v1.2C98.399 898.8 99.602 900 100.8 900l2.399 2.398h2.398l483.6 279.6s1.2 0 1.2 1.2h17.988s1.198 0 1.198-1.2l483.6-279.6h2.399s1.199 0 1.199-1.199l1.2-1.2s1.198 0 1.198-1.198l1.2-1.2s0-1.199 1.199-1.199c.02 1.2.02-.004 1.219-.004zm-960-445.2 134.4 333.6c1.2 3.602 0 8.398-4.8 10.801l-123.6 51.602c-4.802 2.398-10.802-1.2-10.802-7.2l.004-387.61c0-2.398 3.602-3.597 4.8-1.198zm696 331.2H361.2c-6 0-9.602-7.2-7.2-12l238.8-414c3.602-4.8 10.802-4.8 13.2 0l238.8 414c3.598 6 0 12-6 12zm-531.6-43.199-154.8-382.8c-2.398-4.8 1.2-10.801 7.2-10.801L546 326.401c6 0 10.8 6 7.198 12l-232.8 402c-2.398 6-10.797 6-13.199-1.2zm526.8 100.8-228 280.8c-3.602 3.602-9.602 3.602-12 0l-228-280.8c-3.602-4.8 0-13.199 6-13.199h456c6 0 9.602 8.398 6 13.199zm45.602-99.602-232.8-402c-3.602-6 1.2-12 7.2-12l386.4 19.199c6 0 8.398 6 7.198 10.801l-154.8 382.8c-2.402 7.2-10.8 7.2-13.199 1.2zm-258-466.8v-175.2c0-6 7.2-9.602 12-7.2l350.4 202.8c2.398 1.2 1.2 4.802-1.2 4.802l-354-18c-3.6 0-7.198-3.602-7.198-7.2zm-51.602 7.2-354 18c-2.398 0-3.602-3.602-1.2-4.802l351.6-201.6c4.802-3.601 12 1.2 12 7.2v174c0 3.601-3.597 7.203-8.398 7.203zm-264 555.6 193.2 238.8c1.2 2.398-1.2 4.8-3.602 3.6l-315.6-182.4c-6-3.6-4.8-12 1.2-14.397l116.4-48c2.402-1.2 6.003 0 8.402 2.398zm596.4-2.4 116.4 48c6 2.4 7.2 10.802 1.2 14.399l-316.8 182.4c-2.4 1.199-4.802-1.2-3.602-3.602l193.2-238.8c3.601-2.398 7.203-3.597 9.601-2.398zm148.8 13.2-123.6-51.602c-3.602-1.2-6-6-4.8-10.801l134.4-333.6c1.198-2.399 4.8-1.2 4.8 1.199l.004 387.61c0 6-6 9.602-10.801 7.2z\"/>\n",
       "    <text x=\"600\" y=\"660\" font-family=\"Helvetica\" font-size=\"220\" font-weight=\"bold\" fill=\"black\" text-anchor=\"middle\" alignment-baseline=\"middle\">\n",
       "        2\n",
       "    </text>\n",
       "</svg>"
      ],
      "text/plain": [
       "<IPython.core.display.SVG object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "image/svg+xml": [
       "<svg xmlns=\"http://www.w3.org/2000/svg\" width=\"60\" height=\"60\" viewBox=\"0 0 1200 1200\">\n",
       "    <path d=\"M1102.8 896.4c1.2-1.2 1.2-2.398 1.2-2.398V318.004c0-1.2-1.2-1.2-1.2-2.399v-1.199c0-1.2-1.2-1.2-1.2-1.2v-1.198l-1.198-1.2s-1.2-1.199-2.399-1.199h-1.199l-486-282.01c-7.2-3.601-15.602-3.601-21.602 0l-483.6 279.6c-1.2 0-1.2 1.2-2.398 1.2h-1.2l-3.597 3.597s0 1.2-1.2 1.2l-1.199 1.198v1.2s0 1.199-1.199 1.199v571.196c0 1.198 1.2 2.397 1.2 3.6 0 1.2 1.198 2.4 2.398 3.602v1.2C98.399 898.8 99.602 900 100.8 900l2.399 2.398h2.398l483.6 279.6s1.2 0 1.2 1.2h17.988s1.198 0 1.198-1.2l483.6-279.6h2.399s1.199 0 1.199-1.199l1.2-1.2s1.198 0 1.198-1.198l1.2-1.2s0-1.199 1.199-1.199c.02 1.2.02-.004 1.219-.004zm-960-445.2 134.4 333.6c1.2 3.602 0 8.398-4.8 10.801l-123.6 51.602c-4.802 2.398-10.802-1.2-10.802-7.2l.004-387.61c0-2.398 3.602-3.597 4.8-1.198zm696 331.2H361.2c-6 0-9.602-7.2-7.2-12l238.8-414c3.602-4.8 10.802-4.8 13.2 0l238.8 414c3.598 6 0 12-6 12zm-531.6-43.199-154.8-382.8c-2.398-4.8 1.2-10.801 7.2-10.801L546 326.401c6 0 10.8 6 7.198 12l-232.8 402c-2.398 6-10.797 6-13.199-1.2zm526.8 100.8-228 280.8c-3.602 3.602-9.602 3.602-12 0l-228-280.8c-3.602-4.8 0-13.199 6-13.199h456c6 0 9.602 8.398 6 13.199zm45.602-99.602-232.8-402c-3.602-6 1.2-12 7.2-12l386.4 19.199c6 0 8.398 6 7.198 10.801l-154.8 382.8c-2.402 7.2-10.8 7.2-13.199 1.2zm-258-466.8v-175.2c0-6 7.2-9.602 12-7.2l350.4 202.8c2.398 1.2 1.2 4.802-1.2 4.802l-354-18c-3.6 0-7.198-3.602-7.198-7.2zm-51.602 7.2-354 18c-2.398 0-3.602-3.602-1.2-4.802l351.6-201.6c4.802-3.601 12 1.2 12 7.2v174c0 3.601-3.597 7.203-8.398 7.203zm-264 555.6 193.2 238.8c1.2 2.398-1.2 4.8-3.602 3.6l-315.6-182.4c-6-3.6-4.8-12 1.2-14.397l116.4-48c2.402-1.2 6.003 0 8.402 2.398zm596.4-2.4 116.4 48c6 2.4 7.2 10.802 1.2 14.399l-316.8 182.4c-2.4 1.199-4.802-1.2-3.602-3.602l193.2-238.8c3.601-2.398 7.203-3.597 9.601-2.398zm148.8 13.2-123.6-51.602c-3.602-1.2-6-6-4.8-10.801l134.4-333.6c1.198-2.399 4.8-1.2 4.8 1.199l.004 387.61c0 6-6 9.602-10.801 7.2z\"/>\n",
       "    <text x=\"600\" y=\"660\" font-family=\"Helvetica\" font-size=\"220\" font-weight=\"bold\" fill=\"black\" text-anchor=\"middle\" alignment-baseline=\"middle\">\n",
       "        18\n",
       "    </text>\n",
       "</svg>"
      ],
      "text/plain": [
       "<IPython.core.display.SVG object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "text/markdown": [
       "> Here are the results for your character's stats:\n",
       "> \n",
       "> - Strength: 6\n",
       "> - Perception: 4\n",
       "> - Endurance: 8\n",
       "> - Charisma: 10\n",
       "> - Intelligence: 10\n",
       "> - Agility: 2\n",
       "> - Luck: 18"
      ],
      "text/plain": [
       "> Here are the results for your character's stats:\n",
       "> \n",
       "> - Strength: 6\n",
       "> - Perception: 4\n",
       "> - Endurance: 8\n",
       "> - Charisma: 10\n",
       "> - Intelligence: 10\n",
       "> - Agility: 2\n",
       "> - Luck: 18"
      ]
     },
     "metadata": {
      "text/markdown": {
       "chatlab": {
        "default": true
       }
      }
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "from pydantic import BaseModel\n",
    "from chatlab import system, user, Markdown\n",
    "\n",
    "async for message in chatloop([\n",
    "        system(\"Create your character for the Fallout RPG. The user is the DM.\"),\n",
    "        user(\"Roll for the following stats: Strength, Perception, Endurance, Charisma, Intelligence, Agility, and Luck.\")\n",
    "    ]):\n",
    "    # When message is a pydantic model, convert to a dict\n",
    "\n",
    "    if isinstance(message, BaseModel):\n",
    "        message = message.model_dump()\n",
    "\n",
    "    role = message['role']\n",
    "    content = message.get('content')\n",
    "\n",
    "    if(role == \"assistant\" and content is not None):\n",
    "        display(Markdown(\"> \" + content.replace(\"\\n\", \"\\n> \")))\n",
    "    if(role == \"tool\"):\n",
    "        pass "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "chatlab-3PJ-KiVK-py3.12",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.12.1"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}