notebooks/knowledge-graph.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Note: you may need to restart the kernel to use updated packages.\n"
     ]
    }
   ],
   "source": [
    "%pip install graphviz -q"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/svg+xml": [
       "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
       "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
       " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
       "<!-- Generated by graphviz version 12.1.0 (20240811.2233)\n",
       " -->\n",
       "<!-- Pages: 1 -->\n",
       "<svg width=\"606pt\" height=\"487pt\"\n",
       " viewBox=\"0.00 0.00 606.33 486.50\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 482.5)\">\n",
       "<polygon fill=\"white\" stroke=\"none\" points=\"-4,4 -4,-482.5 602.33,-482.5 602.33,4 -4,4\"/>\n",
       "<!-- 1 -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>1</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightblue\" cx=\"300.18\" cy=\"-460.5\" rx=\"56.98\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"300.18\" y=\"-455.45\" font-family=\"Times,serif\" font-size=\"14.00\">Recruitment</text>\n",
       "</g>\n",
       "<!-- 2 -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>2</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightgreen\" cx=\"300.18\" cy=\"-372\" rx=\"54.93\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"300.18\" y=\"-366.95\" font-family=\"Times,serif\" font-size=\"14.00\">Onboarding</text>\n",
       "</g>\n",
       "<!-- 1&#45;&gt;2 -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>1&#45;&gt;2</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M300.18,-442.41C300.18,-430.76 300.18,-415.05 300.18,-401.52\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"303.68,-401.86 300.18,-391.86 296.68,-401.86 303.68,-401.86\"/>\n",
       "<text text-anchor=\"middle\" x=\"321.18\" y=\"-411.2\" font-family=\"Times,serif\" font-size=\"14.00\">leads to</text>\n",
       "</g>\n",
       "<!-- 3 -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>3</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightgrey\" cx=\"300.18\" cy=\"-283.5\" rx=\"105.6\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"300.18\" y=\"-278.45\" font-family=\"Times,serif\" font-size=\"14.00\">Training &amp; Development</text>\n",
       "</g>\n",
       "<!-- 2&#45;&gt;3 -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>2&#45;&gt;3</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M300.18,-353.91C300.18,-342.26 300.18,-326.55 300.18,-313.02\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"303.68,-313.36 300.18,-303.36 296.68,-313.36 303.68,-313.36\"/>\n",
       "<text text-anchor=\"middle\" x=\"332.8\" y=\"-322.7\" font-family=\"Times,serif\" font-size=\"14.00\">followed by</text>\n",
       "</g>\n",
       "<!-- 4 -->\n",
       "<g id=\"node4\" class=\"node\">\n",
       "<title>4</title>\n",
       "<ellipse fill=\"none\" stroke=\"yellow\" cx=\"300.18\" cy=\"-195\" rx=\"110.21\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"300.18\" y=\"-189.95\" font-family=\"Times,serif\" font-size=\"14.00\">Performance Management</text>\n",
       "</g>\n",
       "<!-- 3&#45;&gt;4 -->\n",
       "<g id=\"edge3\" class=\"edge\">\n",
       "<title>3&#45;&gt;4</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M300.18,-265.41C300.18,-253.76 300.18,-238.05 300.18,-224.52\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"303.68,-224.86 300.18,-214.86 296.68,-224.86 303.68,-224.86\"/>\n",
       "<text text-anchor=\"middle\" x=\"323.05\" y=\"-234.2\" font-family=\"Times,serif\" font-size=\"14.00\">supports</text>\n",
       "</g>\n",
       "<!-- 5 -->\n",
       "<g id=\"node5\" class=\"node\">\n",
       "<title>5</title>\n",
       "<ellipse fill=\"none\" stroke=\"orange\" cx=\"87.18\" cy=\"-106.5\" rx=\"87.18\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"87.18\" y=\"-101.45\" font-family=\"Times,serif\" font-size=\"14.00\">Succession Planning</text>\n",
       "</g>\n",
       "<!-- 4&#45;&gt;5 -->\n",
       "<g id=\"edge4\" class=\"edge\">\n",
       "<title>4&#45;&gt;5</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M260.61,-177.93C225.31,-163.6 173.51,-142.56 135.72,-127.21\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"137.44,-124.13 126.86,-123.61 134.81,-130.62 137.44,-124.13\"/>\n",
       "<text text-anchor=\"middle\" x=\"234.13\" y=\"-145.7\" font-family=\"Times,serif\" font-size=\"14.00\">informs</text>\n",
       "</g>\n",
       "<!-- 6 -->\n",
       "<g id=\"node6\" class=\"node\">\n",
       "<title>6</title>\n",
       "<ellipse fill=\"none\" stroke=\"pink\" cx=\"300.18\" cy=\"-106.5\" rx=\"108.16\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"300.18\" y=\"-101.45\" font-family=\"Times,serif\" font-size=\"14.00\">Compensation &amp; Benefits</text>\n",
       "</g>\n",
       "<!-- 4&#45;&gt;6 -->\n",
       "<g id=\"edge5\" class=\"edge\">\n",
       "<title>4&#45;&gt;6</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M300.18,-176.91C300.18,-165.26 300.18,-149.55 300.18,-136.02\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"303.68,-136.36 300.18,-126.36 296.68,-136.36 303.68,-136.36\"/>\n",
       "<text text-anchor=\"middle\" x=\"318.18\" y=\"-145.7\" font-family=\"Times,serif\" font-size=\"14.00\">affects</text>\n",
       "</g>\n",
       "<!-- 7 -->\n",
       "<g id=\"node7\" class=\"node\">\n",
       "<title>7</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightcoral\" cx=\"512.18\" cy=\"-106.5\" rx=\"86.15\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"512.18\" y=\"-101.45\" font-family=\"Times,serif\" font-size=\"14.00\">Employee Relations</text>\n",
       "</g>\n",
       "<!-- 4&#45;&gt;7 -->\n",
       "<g id=\"edge6\" class=\"edge\">\n",
       "<title>4&#45;&gt;7</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M339.56,-177.93C374.69,-163.6 426.25,-142.56 463.87,-127.21\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"464.74,-130.63 472.68,-123.62 462.1,-124.15 464.74,-130.63\"/>\n",
       "<text text-anchor=\"middle\" x=\"459.29\" y=\"-145.7\" font-family=\"Times,serif\" font-size=\"14.00\">enhanced by</text>\n",
       "</g>\n",
       "<!-- 8 -->\n",
       "<g id=\"node8\" class=\"node\">\n",
       "<title>8</title>\n",
       "<ellipse fill=\"none\" stroke=\"red\" cx=\"299.18\" cy=\"-18\" rx=\"56.47\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"299.18\" y=\"-12.95\" font-family=\"Times,serif\" font-size=\"14.00\">Offboarding</text>\n",
       "</g>\n",
       "<!-- 5&#45;&gt;8 -->\n",
       "<g id=\"edge7\" class=\"edge\">\n",
       "<title>5&#45;&gt;8</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M125.08,-90.04C161.55,-75.16 216.67,-52.67 255.09,-36.99\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"256.11,-40.35 264.05,-33.33 253.47,-33.87 256.11,-40.35\"/>\n",
       "<text text-anchor=\"middle\" x=\"235.79\" y=\"-57.2\" font-family=\"Times,serif\" font-size=\"14.00\">precedes</text>\n",
       "</g>\n",
       "<!-- 7&#45;&gt;8 -->\n",
       "<g id=\"edge8\" class=\"edge\">\n",
       "<title>7&#45;&gt;8</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M474.1,-90.04C437.46,-75.16 382.08,-52.67 343.48,-36.99\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"345.05,-33.85 334.47,-33.33 342.42,-40.34 345.05,-33.85\"/>\n",
       "<text text-anchor=\"middle\" x=\"454.38\" y=\"-57.2\" font-family=\"Times,serif\" font-size=\"14.00\">can lead to</text>\n",
       "</g>\n",
       "</g>\n",
       "</svg>\n"
      ],
      "text/plain": [
       "<graphviz.graphs.Digraph at 0x107d72390>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Warning: li is not a known color.\n",
      "Warning: lightbl is not a known color.\n",
      "Warning: li is not a known color.\n",
      "Warning: lightg is not a known color.\n",
      "Warning: li is not a known color.\n",
      "Warning: lightgre is not a known color.\n",
      "Warning: yell is not a known color.\n",
      "Warning: or is not a known color.\n",
      "Warning: p is not a known color.\n",
      "Warning: li is not a known color.\n",
      "Warning: lightcor is not a known color.\n",
      "Warning: bla is not a known color.\n",
      "Warning: bla is not a known color.\n",
      "Warning: b is not a known color.\n",
      "Warning: b is not a known color.\n",
      "Warning: blac is not a known color.\n",
      "Warning: blac is not a known color.\n",
      "Warning: bl is not a known color.\n",
      "Warning: bl is not a known color.\n"
     ]
    },
    {
     "data": {
      "text/markdown": [
       "I've created a visual representation of HR processes to help you understand the flow and relationships between different activities within HR. This includes:\n",
       "\n",
       "- **Recruitment**, leading to\n",
       "- **Onboarding**, followed by\n",
       "- **Training & Development**, which supports\n",
       "- **Performance Management**. This, in turn, informs\n",
       "- **Succession Planning**, affects\n",
       "- **Compensation & Benefits**, and is enhanced by\n",
       "- **Employee Relations**, which can lead to\n",
       "- **Offboarding**.\n",
       "\n",
       "This visual should give you a clear overview of how HR processes interlink and flow from one to the next."
      ],
      "text/plain": [
       "AssistantMessageView(display_id='26e8afcd-0796-4b0a-a3f0-55bd93c86f22', content=\"I've created a visual representation of HR processes to help you understand the flow and relationships between different activities within HR. This includes:\\n\\n- **Recruitment**, leading to\\n- **Onboarding**, followed by\\n- **Training & Development**, which supports\\n- **Performance Management**. This, in turn, informs\\n- **Succession Planning**, affects\\n- **Compensation & Benefits**, and is enhanced by\\n- **Employee Relations**, which can lead to\\n- **Offboarding**.\\n\\nThis visual should give you a clear overview of how HR processes interlink and flow from one to the next.\", finished=False, has_displayed=True)"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import warnings\n",
    "from graphviz import Digraph\n",
    "from pydantic import BaseModel, Field\n",
    "from typing import List\n",
    "from chatlab import Chat, system\n",
    "from chatlab.decorators import incremental_display\n",
    "\n",
    "class Node(BaseModel):\n",
    "    id: int\n",
    "    label: str\n",
    "    color: str\n",
    "\n",
    "class Edge(BaseModel):\n",
    "    source: int\n",
    "    target: int\n",
    "    label: str\n",
    "    color: str = Field(\"black\", description=\"A valid Graphviz color name or hex code.\")\n",
    "\n",
    "class KnowledgeGraph(BaseModel):\n",
    "    nodes: List[Node] = Field(..., default_factory=list)\n",
    "    edges: List[Edge] = Field(..., default_factory=list)\n",
    "\n",
    "\n",
    "def visualize_knowledge_graph(kg: KnowledgeGraph, comment: str = \"Knowledge Graph\"):\n",
    "    \"\"\"Visualizes a knowledge graph using graphviz.\"\"\"\n",
    "    if kg is None:\n",
    "        return\n",
    "\n",
    "    dot = Digraph(comment=comment)\n",
    "\n",
    "    for node in kg.nodes:\n",
    "        dot.node(str(node.id), node.label, color=node.color)\n",
    "\n",
    "    for edge in kg.edges:\n",
    "        dot.edge(str(edge.source), str(edge.target), label=edge.label, color=edge.color)\n",
    "\n",
    "    return dot\n",
    "\n",
    "\n",
    "@incremental_display(visualize_knowledge_graph)\n",
    "def store_knowledge_graph(kg: KnowledgeGraph, comment: str):\n",
    "    \"\"\"Creates a graphviz diagram for the user and stores it in their database.\"\"\"\n",
    "    return \"Stored graph, visual shown\"\n",
    "\n",
    "chat = Chat(\n",
    "    system(\"You are a helpful assistant that can create knowledge graphs and store them in a database. You only respond with the knowledge graph.\"),\n",
    "    model=\"gpt-4-turbo-preview\",\n",
    "    chat_functions=[store_knowledge_graph],\n",
    ")\n",
    "\n",
    "await chat(\"Visually teach me about HR processes\", temperature=0.7)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/markdown": [
       "It seems like there might be a misunderstanding. Could you please specify the company or organization you're asking about the founders of?"
      ],
      "text/plain": [
       "AssistantMessageView(display_id='3c5d41fe-655a-4bbe-96ab-ad5d1934593c', content=\"It seems like there might be a misunderstanding. Could you please specify the company or organization you're asking about the founders of?\", finished=False, has_displayed=True)"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "await chat(\"Who are the founders?\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<DisplayHandle display_id=230945832>"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "display(\"above here\", update=True, display_id=\"230945832\")\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/svg+xml": [
       "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
       "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
       " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
       "<!-- Generated by graphviz version 12.1.0 (20240811.2233)\n",
       " -->\n",
       "<!-- Pages: 1 -->\n",
       "<svg width=\"333pt\" height=\"221pt\"\n",
       " viewBox=\"0.00 0.00 333.21 221.00\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 217)\">\n",
       "<polygon fill=\"white\" stroke=\"none\" points=\"-4,4 -4,-217 329.21,-217 329.21,4 -4,4\"/>\n",
       "<!-- 1 -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>1</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightblue\" cx=\"212.75\" cy=\"-195\" rx=\"27.3\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"212.75\" y=\"-189.95\" font-family=\"Times,serif\" font-size=\"14.00\">Cats</text>\n",
       "</g>\n",
       "<!-- 2 -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>2</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightgreen\" cx=\"156.75\" cy=\"-106.5\" rx=\"37.53\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"156.75\" y=\"-101.45\" font-family=\"Times,serif\" font-size=\"14.00\">Felines</text>\n",
       "</g>\n",
       "<!-- 1&#45;&gt;2 -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>1&#45;&gt;2</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M202.47,-178.13C194.42,-165.69 183.04,-148.11 173.64,-133.6\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"176.72,-131.91 168.35,-125.42 170.84,-135.72 176.72,-131.91\"/>\n",
       "<text text-anchor=\"middle\" x=\"198.01\" y=\"-145.7\" font-family=\"Times,serif\" font-size=\"14.00\">are</text>\n",
       "</g>\n",
       "<!-- 5 -->\n",
       "<g id=\"node5\" class=\"node\">\n",
       "<title>5</title>\n",
       "<ellipse fill=\"none\" stroke=\"orange\" cx=\"268.75\" cy=\"-106.5\" rx=\"56.47\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"268.75\" y=\"-101.45\" font-family=\"Times,serif\" font-size=\"14.00\">Independent</text>\n",
       "</g>\n",
       "<!-- 1&#45;&gt;5 -->\n",
       "<g id=\"edge4\" class=\"edge\">\n",
       "<title>1&#45;&gt;5</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M223.02,-178.13C231.02,-165.77 242.31,-148.33 251.67,-133.88\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"254.45,-136.01 256.95,-125.72 248.58,-132.21 254.45,-136.01\"/>\n",
       "<text text-anchor=\"middle\" x=\"269.76\" y=\"-145.7\" font-family=\"Times,serif\" font-size=\"14.00\">are often</text>\n",
       "</g>\n",
       "<!-- 3 -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>3</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightgrey\" cx=\"46.75\" cy=\"-18\" rx=\"46.75\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"46.75\" y=\"-12.95\" font-family=\"Times,serif\" font-size=\"14.00\">Domestic</text>\n",
       "</g>\n",
       "<!-- 2&#45;&gt;3 -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>2&#45;&gt;3</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M138.09,-90.83C120.94,-77.34 95.27,-57.16 75.5,-41.61\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"77.83,-38.99 67.8,-35.56 73.5,-44.49 77.83,-38.99\"/>\n",
       "<text text-anchor=\"middle\" x=\"129.22\" y=\"-57.2\" font-family=\"Times,serif\" font-size=\"14.00\">can be</text>\n",
       "</g>\n",
       "<!-- 4 -->\n",
       "<g id=\"node4\" class=\"node\">\n",
       "<title>4</title>\n",
       "<ellipse fill=\"none\" stroke=\"yellow\" cx=\"156.75\" cy=\"-18\" rx=\"45.72\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"156.75\" y=\"-12.95\" font-family=\"Times,serif\" font-size=\"14.00\">Predators</text>\n",
       "</g>\n",
       "<!-- 2&#45;&gt;4 -->\n",
       "<g id=\"edge3\" class=\"edge\">\n",
       "<title>2&#45;&gt;4</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M156.75,-88.41C156.75,-76.76 156.75,-61.05 156.75,-47.52\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"160.25,-47.86 156.75,-37.86 153.25,-47.86 160.25,-47.86\"/>\n",
       "<text text-anchor=\"middle\" x=\"165\" y=\"-57.2\" font-family=\"Times,serif\" font-size=\"14.00\">are</text>\n",
       "</g>\n",
       "<!-- 6 -->\n",
       "<g id=\"node6\" class=\"node\">\n",
       "<title>6</title>\n",
       "<ellipse fill=\"none\" stroke=\"pink\" cx=\"271.75\" cy=\"-18\" rx=\"51.35\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"271.75\" y=\"-12.95\" font-family=\"Times,serif\" font-size=\"14.00\">Carnivores</text>\n",
       "</g>\n",
       "<!-- 2&#45;&gt;6 -->\n",
       "<g id=\"edge5\" class=\"edge\">\n",
       "<title>2&#45;&gt;6</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M176.25,-90.83C194.26,-77.28 221.26,-56.97 241.96,-41.4\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"243.88,-44.34 249.76,-35.53 239.67,-38.75 243.88,-44.34\"/>\n",
       "<text text-anchor=\"middle\" x=\"232.8\" y=\"-57.2\" font-family=\"Times,serif\" font-size=\"14.00\">are</text>\n",
       "</g>\n",
       "</g>\n",
       "</svg>\n"
      ],
      "text/plain": [
       "<graphviz.graphs.Digraph at 0x107d828d0>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Warning: light is not a known color.\n",
      "Warning: l is not a known color.\n",
      "Warning: lightg is not a known color.\n",
      "Warning: li is not a known color.\n",
      "Warning: lightg is not a known color.\n",
      "Warning: y is not a known color.\n",
      "Warning: or is not a known color.\n",
      "Warning: bl is not a known color.\n",
      "Warning: bla is not a known color.\n",
      "Warning: b is not a known color.\n",
      "Warning: bl is not a known color.\n"
     ]
    },
    {
     "data": {
      "image/svg+xml": [
       "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
       "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
       " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
       "<!-- Generated by graphviz version 12.1.0 (20240811.2233)\n",
       " -->\n",
       "<!-- Pages: 1 -->\n",
       "<svg width=\"365pt\" height=\"221pt\"\n",
       " viewBox=\"0.00 0.00 365.10 221.00\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 217)\">\n",
       "<polygon fill=\"white\" stroke=\"none\" points=\"-4,4 -4,-217 361.1,-217 361.1,4 -4,4\"/>\n",
       "<!-- 1 -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>1</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightblue\" cx=\"219.75\" cy=\"-195\" rx=\"30.37\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"219.75\" y=\"-189.95\" font-family=\"Times,serif\" font-size=\"14.00\">Dogs</text>\n",
       "</g>\n",
       "<!-- 2 -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>2</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightgreen\" cx=\"173.75\" cy=\"-106.5\" rx=\"40.6\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"173.75\" y=\"-101.45\" font-family=\"Times,serif\" font-size=\"14.00\">Canines</text>\n",
       "</g>\n",
       "<!-- 1&#45;&gt;2 -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>1&#45;&gt;2</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M210.88,-177.32C204.43,-165.19 195.56,-148.52 188.1,-134.5\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"191.24,-132.94 183.45,-125.76 185.06,-136.23 191.24,-132.94\"/>\n",
       "<text text-anchor=\"middle\" x=\"209.12\" y=\"-145.7\" font-family=\"Times,serif\" font-size=\"14.00\">are</text>\n",
       "</g>\n",
       "<!-- 5 -->\n",
       "<g id=\"node5\" class=\"node\">\n",
       "<title>5</title>\n",
       "<ellipse fill=\"none\" stroke=\"orange\" cx=\"264.75\" cy=\"-106.5\" rx=\"32.41\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"264.75\" y=\"-101.45\" font-family=\"Times,serif\" font-size=\"14.00\">Loyal</text>\n",
       "</g>\n",
       "<!-- 1&#45;&gt;5 -->\n",
       "<g id=\"edge4\" class=\"edge\">\n",
       "<title>1&#45;&gt;5</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M228.42,-177.32C234.77,-165.12 243.52,-148.3 250.84,-134.23\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"253.89,-135.96 255.4,-125.47 247.68,-132.73 253.89,-135.96\"/>\n",
       "<text text-anchor=\"middle\" x=\"282.28\" y=\"-145.7\" font-family=\"Times,serif\" font-size=\"14.00\">are known as</text>\n",
       "</g>\n",
       "<!-- 3 -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>3</title>\n",
       "<ellipse fill=\"none\" stroke=\"lightgrey\" cx=\"46.75\" cy=\"-18\" rx=\"46.75\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"46.75\" y=\"-12.95\" font-family=\"Times,serif\" font-size=\"14.00\">Domestic</text>\n",
       "</g>\n",
       "<!-- 2&#45;&gt;3 -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>2&#45;&gt;3</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M152.2,-90.83C131.89,-76.99 101.21,-56.1 78.18,-40.41\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"80.31,-37.62 70.07,-34.89 76.37,-43.41 80.31,-37.62\"/>\n",
       "<text text-anchor=\"middle\" x=\"139.25\" y=\"-57.2\" font-family=\"Times,serif\" font-size=\"14.00\">can be</text>\n",
       "</g>\n",
       "<!-- 4 -->\n",
       "<g id=\"node4\" class=\"node\">\n",
       "<title>4</title>\n",
       "<ellipse fill=\"none\" stroke=\"yellow\" cx=\"173.75\" cy=\"-18\" rx=\"62.61\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"173.75\" y=\"-12.95\" font-family=\"Times,serif\" font-size=\"14.00\">Pack Animals</text>\n",
       "</g>\n",
       "<!-- 2&#45;&gt;4 -->\n",
       "<g id=\"edge3\" class=\"edge\">\n",
       "<title>2&#45;&gt;4</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M173.75,-88.41C173.75,-76.76 173.75,-61.05 173.75,-47.52\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"177.25,-47.86 173.75,-37.86 170.25,-47.86 177.25,-47.86\"/>\n",
       "<text text-anchor=\"middle\" x=\"182\" y=\"-57.2\" font-family=\"Times,serif\" font-size=\"14.00\">are</text>\n",
       "</g>\n",
       "<!-- 6 -->\n",
       "<g id=\"node6\" class=\"node\">\n",
       "<title>6</title>\n",
       "<ellipse fill=\"none\" stroke=\"pink\" cx=\"305.75\" cy=\"-18\" rx=\"51.35\" ry=\"18\"/>\n",
       "<text text-anchor=\"middle\" x=\"305.75\" y=\"-12.95\" font-family=\"Times,serif\" font-size=\"14.00\">Carnivores</text>\n",
       "</g>\n",
       "<!-- 2&#45;&gt;6 -->\n",
       "<g id=\"edge5\" class=\"edge\">\n",
       "<title>2&#45;&gt;6</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M195.84,-91.02C216.87,-77.24 248.78,-56.33 272.81,-40.58\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"274.53,-43.64 280.98,-35.23 270.69,-37.78 274.53,-43.64\"/>\n",
       "<text text-anchor=\"middle\" x=\"259.82\" y=\"-57.2\" font-family=\"Times,serif\" font-size=\"14.00\">are</text>\n",
       "</g>\n",
       "</g>\n",
       "</svg>\n"
      ],
      "text/plain": [
       "<graphviz.graphs.Digraph at 0x107aad9d0>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Warning: light is not a known color.\n",
      "Warning: l is not a known color.\n",
      "Warning: lightg is not a known color.\n",
      "Warning: li is not a known color.\n",
      "Warning: lightg is not a known color.\n",
      "Warning: yell is not a known color.\n",
      "Warning: o is not a known color.\n",
      "Warning: orang is not a known color.\n",
      "Warning: pi is not a known color.\n",
      "Warning: blac is not a known color.\n",
      "Warning: bl is not a known color.\n"
     ]
    },
    {
     "data": {
      "text/markdown": [
       "I've created two knowledge graphs for you:\n",
       "\n",
       "1. **About Cats**: This graph explores the characteristics of cats, highlighting their nature as felines, domestic animals, predators, independent beings, and carnivores.\n",
       "\n",
       "2. **About Dogs**: This graph delves into the world of dogs, showcasing their identity as canines, domestic animals, pack animals, known for their loyalty, and as carnivores.\n",
       "\n",
       "Both visuals should give you a clear overview of the unique attributes and classifications of cats and dogs."
      ],
      "text/plain": [
       "AssistantMessageView(display_id='ebbcdc3d-8843-4fca-aa65-f4972d36f638', content=\"I've created two knowledge graphs for you:\\n\\n1. **About Cats**: This graph explores the characteristics of cats, highlighting their nature as felines, domestic animals, predators, independent beings, and carnivores.\\n\\n2. **About Dogs**: This graph delves into the world of dogs, showcasing their identity as canines, domestic animals, pack animals, known for their loyalty, and as carnivores.\\n\\nBoth visuals should give you a clear overview of the unique attributes and classifications of cats and dogs.\", finished=False, has_displayed=True)"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "with warnings.catch_warnings():\n",
    "    warnings.simplefilter(\"ignore\")\n",
    "    await chat(\"Ok now make two knowledge graphs. One about cats and one about dogs.\")"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "chatlab-3kMKfU-i-py3.11",
   "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.11.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}