Upload with huggingface_hub

app.ipynb

@@ -0,0 +1,106 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6d7e0fa0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6bc68605",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import gradio as gr\n",
+    "from chat import gradio as chat\n",
+    "from ner import gradio as ner\n",
+    "from math_demo import gradio as math_demo\n",
+    "from bash import gradio as bash\n",
+    "from pal import gradio as pal\n",
+    "from gatsby import gradio as gatsby\n",
+    "from qa import gradio as qa\n",
+    "from stats import gradio as stats\n",
+    "from selfask import gradio as selfask\n",
+    "from backtrack import gradio as backtrack"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a0a69443",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "CSS = \"\"\"\n",
+    "#clean div.form {border: 0px} \n",
+    "#response {border: 0px; background: #ffeec6} \n",
+    "#prompt {border: 0px;background: aliceblue} \n",
+    "#json {border: 0px} \n",
+    "#result {border: 0px; background: #c5e0e5} \n",
+    "#inner {margin: 10px; padding: 10px; font-size: 20px; } \n",
+    "#inner textarea {border: 0px}\n",
+    "div.gradio-container {color: black}\n",
+    "span.head {font-size: 60pt; font-family: cursive;}\n",
+    "body {\n",
+    "  --text-sm: 15px;\n",
+    "  --text-md: 18px;\n",
+    "  --text-lg: 20px;\n",
+    "  --input-text-size: 20px;\n",
+    "  --section-text-size: 20px;\n",
+    "}\n",
+    "\"\"\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7eabd4da",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ea4326bb",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "with gr.Blocks(css=CSS, theme=gr.themes.Monochrome()) as demo:\n",
+    "    gr.HTML(\"<center style='background:#B6B7BA'> <span class='head'>Mini</span><img src='https://user-images.githubusercontent.com/35882/227017900-0cacdfb7-37e2-47b1-9347-a233810d3544.png' width='20%' style='display:inline'><span class='head'>Chain</span></center><center> <br><a href='https://github.com/srush/minichain'>[library]</a> </center>\")\n",
+    "\n",
+    "    gr.TabbedInterface([math_demo, qa,  chat, gatsby,  ner, bash, pal, stats, selfask, backtrack],\n",
+    "                       [\"Math\", \"QA\", \"Chat\", \"Book\", \"NER\", \"Bash\", \"PAL\", \"Stats\", \"SelfAsk\", \"Backtrack\"],\n",
+    "                       css = CSS)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a7c58fbd",
+   "metadata": {
+    "lines_to_next_cell": 2
+   },
+   "outputs": [],
+   "source": [
+    "demo.launch()"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

app.py

@@ -8,6 +8,7 @@ from gatsby import gradio as gatsby
 from qa import gradio as qa
 from stats import gradio as stats
 from selfask import gradio as selfask
+from backtrack import gradio as backtrack
 
 CSS = """
 #clean div.form {border: 0px} 
@@ -21,8 +22,8 @@ div.gradio-container {color: black}
 span.head {font-size: 60pt; font-family: cursive;}
 body {
   --text-sm: 15px;
-  --text-md: 20px;
-  --text-lg: 22px;
+  --text-md: 18px;
+  --text-lg: 20px;
   --input-text-size: 20px;
   --section-text-size: 20px;
 }
@@ -31,10 +32,10 @@ body {
 
 
 with gr.Blocks(css=CSS, theme=gr.themes.Monochrome()) as demo:
-    gr.HTML("<center style='background:#B6B7BA'> <span class='head'>Mini</span><img src='https://user-images.githubusercontent.com/35882/227017900-0cacdfb7-37e2-47b1-9347-a233810d3544.png' width='20%' style='display:inline'><span class='head'>Chain</span> <br><a href='https://github.com/srush/minichain'>[code]</a>  <a href='https://user-images.githubusercontent.com/35882/218286642-67985b6f-d483-49be-825b-f62b72c469cd.png'>[docs]</a></center>")
+    gr.HTML("<center style='background:#B6B7BA'> <span class='head'>Mini</span><img src='https://user-images.githubusercontent.com/35882/227017900-0cacdfb7-37e2-47b1-9347-a233810d3544.png' width='20%' style='display:inline'><span class='head'>Chain</span></center><center> <br><a href='https://github.com/srush/minichain'>[library]</a> </center>")
 
-    gr.TabbedInterface([math_demo, qa,  chat, gatsby,  ner, bash, pal, stats, selfask],
-                       ["Math", "QA", "Chat", "Book", "NER", "Bash", "PAL", "Stats", "SelfAsk"],
+    gr.TabbedInterface([math_demo, qa,  chat, gatsby,  ner, bash, pal, stats, selfask, backtrack],
+                       ["Math", "QA", "Chat", "Book", "NER", "Bash", "PAL", "Stats", "SelfAsk", "Backtrack"],
                        css = CSS)
 
 demo.launch()

backtrack.ipynb

@@ -0,0 +1,126 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "20a65655",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "261f764b",
+   "metadata": {
+    "tags": [
+     "hide_inp"
+    ]
+   },
+   "outputs": [],
+   "source": [
+    "desc = \"\"\"\n",
+    "### Backtrack on Failure\n",
+    "\n",
+    "Chain that backtracks on failure. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/backtrack.ipynb)\n",
+    "\n",
+    "\"\"\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b6617fa6",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "from minichain import prompt, Mock, show, OpenAI\n",
+    "import minichain"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "36c42ea3",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "@prompt(Mock([\"dog\", \"blue\", \"cat\"]))\n",
+    "def prompt_generation(model):\n",
+    "    return model(\"\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "283167b5",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "@prompt(OpenAI(), template=\"Answer 'yes' is {{query}} is a color. Answer:\")\n",
+    "def prompt_validation(model, x):\n",
+    "    out = model(dict(query=x))\n",
+    "    if out.strip().lower().startswith(\"yes\"):\n",
+    "        return x\n",
+    "    return model.fail(1)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "76e99eee",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "def run():\n",
+    "    x = prompt_generation()\n",
+    "    return prompt_validation(x)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "45a2e713",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "gradio = show(run,\n",
+    "            examples = [],\n",
+    "            subprompts=[prompt_generation, prompt_validation],\n",
+    "            out_type=\"markdown\"\n",
+    "            )"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "24791bf0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "if __name__ == \"__main__\":\n",
+    "    gradio.launch()"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "tags,-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

backtrack.py

@@ -0,0 +1,38 @@
+# + tags=["hide_inp"]
+desc = """
+### Backtrack on Failure
+
+Chain that backtracks on failure. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/backtrack.ipynb)
+
+"""
+# -
+
+# $
+from minichain import prompt, Mock, show, OpenAI
+import minichain
+
+@prompt(Mock(["dog", "blue", "cat"]))
+def prompt_generation(model):
+    return model("")
+
+@prompt(OpenAI(), template="Answer 'yes' is {{query}} is a color. Answer:")
+def prompt_validation(model, x):
+    out = model(dict(query=x))
+    if out.strip().lower().startswith("yes"):
+        return x
+    return model.fail(1)
+    
+def run():
+    x = prompt_generation()
+    return prompt_validation(x)
+# $
+
+gradio = show(run,
+            examples = [],
+            subprompts=[prompt_generation, prompt_validation],
+            code=open("backtrack.py", "r").read().split("$")[1].strip().strip("#").strip(),
+            out_type="markdown"
+            )
+
+if __name__ == "__main__":
+    gradio.launch()

backtrack.py~

@@ -0,0 +1,34 @@
+# + tags=["hide_inp"]
+desc = """
+### Backtrack on Failure
+
+Chain that backtracks on failure. [[Code](https://github.com/srush/MiniChain/blob/main/examples/backtrack.py)]
+
+"""
+# -
+
+
+from minichain import prompt, Mock, show
+import minichain
+
+@prompt(Mock(["red", "blue"]))
+def prompt_function1(model, x):
+    return model(x)
+
+@prompt(Mock(["b"]), template_file="test.pmpt.tpl")
+def prompt_function2(model, x):
+    if x == "red":
+        return model.fail(1)
+    return model(dict(x=x))
+    
+def run(query):
+    x = prompt_function1(query)
+    return prompt_function2(prompt_function2(x))
+
+
+demo = show(run,
+            examples=["a"],
+            subprompts=[prompt_function1, prompt_function2, prompt_function2])
+
+if __name__ == "__main__":
+    demo.launch()

base.ipynb

@@ -0,0 +1,113 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "99422218",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "f8d55ae3",
+   "metadata": {},
+   "source": [
+    "Prompt from ...\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a2c96300",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "prompt = \"\"\"\n",
+    "Question: Who lived longer, Muhammad Ali or Alan Turing?\n",
+    "Are follow up questions needed here: Yes.\n",
+    "Follow up: How old was Muhammad Ali when he died?\n",
+    "Intermediate answer: Muhammad Ali was 74 years old when he died.\n",
+    "Follow up: How old was Alan Turing when he died?\n",
+    "Intermediate answer: Alan Turing was 41 years old when he died.\n",
+    "So the final answer is: Muhammad Ali\n",
+    "\n",
+    "Question: When was the founder of craigslist born?\n",
+    "Are follow up questions needed here: Yes.\n",
+    "Follow up: Who was the founder of craigslist?\n",
+    "Intermediate answer: Craigslist was founded by Craig Newmark.\n",
+    "Follow up: When was Craig Newmark born?\n",
+    "Intermediate answer: Craig Newmark was born on December 6, 1952.\n",
+    "So the final answer is: December 6, 1952\n",
+    "\n",
+    "Question: Who was the maternal grandfather of George Washington?\n",
+    "Are follow up questions needed here: Yes.\n",
+    "Follow up: Who was the mother of George Washington?\n",
+    "Intermediate answer: The mother of George Washington was Mary Ball Washington.\n",
+    "Follow up: Who was the father of Mary Ball Washington?\n",
+    "Intermediate answer: The father of Mary Ball Washington was Joseph Ball.\n",
+    "So the final answer is: Joseph Ball\n",
+    "\n",
+    "Question: Are both the directors of Jaws and Casino Royale from the same country?\n",
+    "Are follow up questions needed here: Yes.\n",
+    "Follow up: Who is the director of Jaws?\n",
+    "Intermediate answer: The director of Jaws is Steven Spielberg.\n",
+    "Follow up: Where is Steven Spielberg from?\n",
+    "Intermediate answer: The United States.\n",
+    "Follow up: Who is the director of Casino Royale?\n",
+    "Intermediate answer: The director of Casino Royale is Martin Campbell.\n",
+    "Follow up: Where is Martin Campbell from?\n",
+    "Intermediate answer: New Zealand.\n",
+    "So the final answer is: No\n",
+    "\n",
+    "Question: {{input}}\n",
+    "Are followup questions needed here: {% if agent_scratchpad %}Yes{%else%}No{% endif %}.\n",
+    "\"\"\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "2c142da2",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "import jinja2"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a26d9cd8",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "class SelfAsk:\n",
+    "    def render(self, input: str, agent_scratchpad: bool):\n",
+    "        return jinja.render(prompt, dict(input=input,\n",
+    "                                         agent_scatchpad=agent_scratchpad))\n",
+    "\n",
+    "    def parse(self, response: str):\n",
+    "        pass\n",
+    "\n",
+    "\n",
+    "    def stop(self):\n",
+    "        return []"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

bash.ipynb

@@ -2,15 +2,21 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
-   "id": "d1d5ec44",
+   "execution_count": null,
+   "id": "4ed306df",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1c9aac71",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:17.499299Z",
-     "iopub.status.busy": "2023-03-22T17:00:17.498973Z",
-     "iopub.status.idle": "2023-03-22T17:00:17.506839Z",
-     "shell.execute_reply": "2023-03-22T17:00:17.506087Z"
-    },
+    "lines_to_next_cell": 2,
     "tags": [
      "hide_inp"
     ]
@@ -21,32 +27,17 @@
     "desc = \"\"\"\n",
     "### Bash Command Suggestion\n",
     "\n",
-    "Chain that ask for a command-line question and then runs the bash command. [[Code](https://github.com/srush/MiniChain/blob/main/examples/bash.py)]\n",
+    "Chain that ask for a command-line question and then runs the bash command. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/bash.ipynb)\n",
     "\n",
     "(Adapted from LangChain [BashChain](https://langchain.readthedocs.io/en/latest/modules/chains/examples/llm_bash.html))\n",
     "\"\"\""
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "94100dff",
-   "metadata": {},
-   "source": [
-    "$"
-   ]
-  },
   {
    "cell_type": "code",
-   "execution_count": 2,
-   "id": "b2ae061e",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:17.509606Z",
-     "iopub.status.busy": "2023-03-22T17:00:17.509377Z",
-     "iopub.status.idle": "2023-03-22T17:00:18.808162Z",
-     "shell.execute_reply": "2023-03-22T17:00:18.807504Z"
-    }
-   },
+   "execution_count": null,
+   "id": "780fdf4d",
+   "metadata": {},
    "outputs": [],
    "source": [
     "from minichain import show, prompt, OpenAI, Bash"
@@ -54,15 +45,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "id": "9c440e40",
+   "execution_count": null,
+   "id": "6e8d844b",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:18.810785Z",
-     "iopub.status.busy": "2023-03-22T17:00:18.810465Z",
-     "iopub.status.idle": "2023-03-22T17:00:18.814003Z",
-     "shell.execute_reply": "2023-03-22T17:00:18.813511Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -75,15 +60,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
-   "id": "74115a3f",
+   "execution_count": null,
+   "id": "e529c0da",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:18.816250Z",
-     "iopub.status.busy": "2023-03-22T17:00:18.815868Z",
-     "iopub.status.idle": "2023-03-22T17:00:18.818697Z",
-     "shell.execute_reply": "2023-03-22T17:00:18.818263Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -95,16 +74,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
-   "id": "88210a7d",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:18.820762Z",
-     "iopub.status.busy": "2023-03-22T17:00:18.820577Z",
-     "iopub.status.idle": "2023-03-22T17:00:18.823279Z",
-     "shell.execute_reply": "2023-03-22T17:00:18.822838Z"
-    }
-   },
+   "execution_count": null,
+   "id": "9894b4b8",
+   "metadata": {},
    "outputs": [],
    "source": [
     "def bash(query):\n",
@@ -112,49 +84,21 @@
    ]
   },
   {
-   "cell_type": "markdown",
-   "id": "99d969c2",
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "721196f5",
    "metadata": {},
-   "source": [
-    "$"
-   ]
+   "outputs": [],
+   "source": []
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
-   "id": "fe4ffa7a",
+   "execution_count": null,
+   "id": "d18f754f",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:18.825360Z",
-     "iopub.status.busy": "2023-03-22T17:00:18.825177Z",
-     "iopub.status.idle": "2023-03-22T17:00:19.158113Z",
-     "shell.execute_reply": "2023-03-22T17:00:19.157509Z"
-    },
     "lines_to_next_cell": 2
    },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Running on local URL:  http://127.0.0.1:7861\n",
-      "\n",
-      "To create a public link, set `share=True` in `launch()`.\n"
-     ]
-    },
-    {
-     "data": {
-      "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7861/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
-      ],
-      "text/plain": [
-       "<IPython.core.display.HTML object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
+   "outputs": [],
    "source": [
     "gradio = show(bash,\n",
     "              subprompts=[cli_prompt, bash_run],\n",
@@ -163,7 +107,6 @@
     "                        \"Please write a bash script that prints 'Hello World' to the console.\"],\n",
     "              out_type=\"markdown\",\n",
     "              description=desc,\n",
-    "              code=open(\"bash.py\", \"r\").read().split(\"$\")[1].strip().strip(\"#\").strip(),\n",
     "              )\n",
     "if __name__ == \"__main__\":\n",
     "    gradio.launch()"
@@ -172,24 +115,9 @@
  ],
  "metadata": {
   "jupytext": {
-   "cell_metadata_filter": "tags,-all"
-  },
-  "kernelspec": {
-   "display_name": "minichain",
-   "language": "python",
-   "name": "minichain"
-  },
-  "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.10.6"
+   "cell_metadata_filter": "tags,-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
   }
  },
  "nbformat": 4,

bash.py

@@ -3,7 +3,7 @@
 desc = """
 ### Bash Command Suggestion
 
-Chain that ask for a command-line question and then runs the bash command. [[Code](https://github.com/srush/MiniChain/blob/main/examples/bash.py)]
+Chain that ask for a command-line question and then runs the bash command. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/bash.ipynb)
 
 (Adapted from LangChain [BashChain](https://langchain.readthedocs.io/en/latest/modules/chains/examples/llm_bash.html))
 """

chat.ipynb

@@ -0,0 +1,173 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "62e745e2",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "8b7656d3",
+   "metadata": {
+    "lines_to_next_cell": 0,
+    "tags": [
+     "hide_inp"
+    ]
+   },
+   "outputs": [],
+   "source": [
+    "desc = \"\"\"\n",
+    "### Chat\n",
+    "\n",
+    "A chat-like example for multi-turn chat with state. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/chat.ipynb)\n",
+    "\n",
+    "(Adapted from [LangChain](https://langchain.readthedocs.io/en/latest/modules/memory/examples/chatgpt_clone.html)'s version of this [blog post](https://www.engraved.blog/building-a-virtual-machine-inside/).)\n",
+    "\n",
+    "\"\"\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1c551f62",
+   "metadata": {
+    "lines_to_next_cell": 2
+   },
+   "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "4ad7d54a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from dataclasses import dataclass, replace\n",
+    "from typing import List, Tuple\n",
+    "from minichain import OpenAI, prompt, show"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "46da3feb",
+   "metadata": {},
+   "source": [
+    "Generic stateful Memory"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3f3fdeaa",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "MEMORY = 2"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7bb9b6f9",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "@dataclass\n",
+    "class State:\n",
+    "    memory: List[Tuple[str, str]]\n",
+    "    human_input: str = \"\"\n",
+    "\n",
+    "    def push(self, response: str) -> \"State\":\n",
+    "        memory = self.memory if len(self.memory) < MEMORY else self.memory[1:]\n",
+    "        return State(memory + [(self.human_input, response)])"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "06bfc54f",
+   "metadata": {},
+   "source": [
+    "Chat prompt with memory"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "abc6322d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "@prompt(OpenAI(), template_file=\"chat.pmpt.tpl\")\n",
+    "def chat_prompt(model, state: State) -> State:\n",
+    "    out = model(state)\n",
+    "    result = out.split(\"Assistant:\")[-1]\n",
+    "    return state.push(result)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3e2d2251",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "examples = [\n",
+    "    \"ls ~\",\n",
+    "    \"cd ~\",\n",
+    "    \"{Please make a file jokes.txt inside and put some jokes inside}\",\n",
+    "    \"\"\"echo -e \"x=lambda y:y*5+3;print('Result:' + str(x(6)))\" > run.py && python3 run.py\"\"\",\n",
+    "    \"\"\"echo -e \"print(list(filter(lambda x: all(x%d for d in range(2,x)),range(2,3**10)))[:10])\" > run.py && python3 run.py\"\"\",\n",
+    "    \"\"\"echo -e \"echo 'Hello from Docker\" > entrypoint.sh && echo -e \"FROM ubuntu:20.04\\nCOPY entrypoint.sh entrypoint.sh\\nENTRYPOINT [\\\"/bin/sh\\\",\\\"entrypoint.sh\\\"]\">Dockerfile && docker build . -t my_docker_image && docker run -t my_docker_image\"\"\",\n",
+    "    \"nvidia-smi\"\n",
+    "]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "da9746f9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "gradio = show(lambda command, state: chat_prompt(replace(state, human_input=command)),\n",
+    "              initial_state=State([]),\n",
+    "              subprompts=[chat_prompt],\n",
+    "              examples=examples,\n",
+    "              out_type=\"json\",\n",
+    "              description=desc,\n",
+    ")\n",
+    "if __name__ == \"__main__\":\n",
+    "    gradio.launch()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "db848300",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "tags,-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

chat.py

@@ -2,7 +2,7 @@
 desc = """
 ### Chat
 
-A chat-like example for multi-turn chat with state. [[Code](https://github.com/srush/MiniChain/blob/main/examples/chat.py)]
+A chat-like example for multi-turn chat with state. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/chat.ipynb)
 
 (Adapted from [LangChain](https://langchain.readthedocs.io/en/latest/modules/memory/examples/chatgpt_clone.html)'s version of this [blog post](https://www.engraved.blog/building-a-virtual-machine-inside/).)
 

color.ipynb

@@ -0,0 +1,96 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b56870fe",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "bef4e04e",
+   "metadata": {},
+   "source": [
+    "Answer a math problem with code.\n",
+    "Adapted from Dust [maths-generate-code](https://dust.tt/spolu/a/d12ac33169)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0bcf8e75",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from minichain import Backend, JinjaPrompt, Prompt, start_chain, SimplePrompt, show_log"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "715fd76e",
+   "metadata": {},
+   "source": [
+    "Prompt that asks LLM for code from math."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "72e3d463",
+   "metadata": {
+    "lines_to_next_cell": 2
+   },
+   "outputs": [],
+   "source": [
+    "class ColorPrompt(Prompt[str, bool]):\n",
+    "    def parse(inp: str) -> str:\n",
+    "        return f\"Answer 'Yes' if this is a color, {inp}. Answer:\"\n",
+    "\n",
+    "    def parse(out: str, inp) -> bool:\n",
+    "        # Encode the parsing logic\n",
+    "        return out.strip() == \"Yes\"  \n",
+    "ColorPrompt().show({\"inp\": \"dog\"}, \"No\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "77cbf319",
+   "metadata": {
+    "lines_to_next_cell": 2
+   },
+   "outputs": [],
+   "source": [
+    "with start_chain(\"color\") as backend:\n",
+    "    question = 'What is the sum of the powers of 3 (3^i) that are smaller than 100?'\n",
+    "    prompt = MathPrompt(backend.OpenAI()).chain(SimplePrompt(backend.Python()))\n",
+    "    result = prompt({\"question\": question})\n",
+    "    print(result)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "72cdff4c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "show_log(\"math.log\")"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

gatsby.ipynb

@@ -2,15 +2,21 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
-   "id": "5339272e",
+   "execution_count": null,
+   "id": "bf80e010",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a54429c0",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:42.285169Z",
-     "iopub.status.busy": "2023-03-22T17:00:42.284973Z",
-     "iopub.status.idle": "2023-03-22T17:00:42.291455Z",
-     "shell.execute_reply": "2023-03-22T17:00:42.290871Z"
-    },
+    "lines_to_next_cell": 2,
     "tags": [
      "hide_inp"
     ]
@@ -20,32 +26,17 @@
     "desc = \"\"\"\n",
     "### Book QA\n",
     "\n",
-    "Chain that does question answering with Hugging Face embeddings. [[Code](https://github.com/srush/MiniChain/blob/main/examples/gatsby.py)]\n",
+    "Chain that does question answering with Hugging Face embeddings. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/gatsby.ipynb)\n",
     "\n",
     "(Adapted from the [LlamaIndex example](https://github.com/jerryjliu/gpt_index/blob/main/examples/gatsby/TestGatsby.ipynb).)\n",
     "\"\"\""
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "f966010d",
-   "metadata": {},
-   "source": [
-    "$"
-   ]
-  },
   {
    "cell_type": "code",
-   "execution_count": 2,
-   "id": "5391c476",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:42.293927Z",
-     "iopub.status.busy": "2023-03-22T17:00:42.293738Z",
-     "iopub.status.idle": "2023-03-22T17:00:43.695402Z",
-     "shell.execute_reply": "2023-03-22T17:00:43.694722Z"
-    }
-   },
+   "execution_count": null,
+   "id": "71c343bb",
+   "metadata": {},
    "outputs": [],
    "source": [
     "import datasets\n",
@@ -55,7 +46,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "5b1b2a82",
+   "id": "564552bf",
    "metadata": {},
    "source": [
     "Load data with embeddings (computed beforehand)"
@@ -63,46 +54,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "id": "a54cf84e",
+   "execution_count": null,
+   "id": "75ba221e",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:43.698121Z",
-     "iopub.status.busy": "2023-03-22T17:00:43.697792Z",
-     "iopub.status.idle": "2023-03-22T17:00:43.730349Z",
-     "shell.execute_reply": "2023-03-22T17:00:43.729747Z"
-    },
     "lines_to_next_cell": 1
    },
-   "outputs": [
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "86d0a2ceb7ad4f99978e37c2719f2960",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "  0%|          | 0/1 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "text/plain": [
-       "Dataset({\n",
-       "    features: ['passages', 'embeddings'],\n",
-       "    num_rows: 52\n",
-       "})"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "gatsby = datasets.load_from_disk(\"gatsby\")\n",
     "gatsby.add_faiss_index(\"embeddings\")"
@@ -110,7 +67,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a9a08061",
+   "id": "ede57fd4",
    "metadata": {},
    "source": [
     "Fast KNN retieval prompt"
@@ -118,15 +75,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
-   "id": "3e69cd6b",
+   "execution_count": null,
+   "id": "71300bf7",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:43.735462Z",
-     "iopub.status.busy": "2023-03-22T17:00:43.735139Z",
-     "iopub.status.idle": "2023-03-22T17:00:43.738420Z",
-     "shell.execute_reply": "2023-03-22T17:00:43.737964Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -140,15 +91,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
-   "id": "14e22d0a",
+   "execution_count": null,
+   "id": "c311126e",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:43.740824Z",
-     "iopub.status.busy": "2023-03-22T17:00:43.740339Z",
-     "iopub.status.idle": "2023-03-22T17:00:43.743342Z",
-     "shell.execute_reply": "2023-03-22T17:00:43.742905Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -161,17 +106,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
-   "id": "bfca0bea",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:43.745377Z",
-     "iopub.status.busy": "2023-03-22T17:00:43.745056Z",
-     "iopub.status.idle": "2023-03-22T17:00:43.747768Z",
-     "shell.execute_reply": "2023-03-22T17:00:43.747352Z"
-    },
-    "lines_to_next_cell": 2
-   },
+   "execution_count": null,
+   "id": "18db52b6",
+   "metadata": {},
    "outputs": [],
    "source": [
     "def gatsby(query):\n",
@@ -180,50 +117,21 @@
    ]
   },
   {
-   "cell_type": "markdown",
-   "id": "159b0b85",
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "2ec09500",
    "metadata": {
     "lines_to_next_cell": 2
    },
-   "source": [
-    "$"
-   ]
+   "outputs": [],
+   "source": []
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
-   "id": "8e3f74d0",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:43.749935Z",
-     "iopub.status.busy": "2023-03-22T17:00:43.749750Z",
-     "iopub.status.idle": "2023-03-22T17:00:44.094814Z",
-     "shell.execute_reply": "2023-03-22T17:00:44.094179Z"
-    }
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Running on local URL:  http://127.0.0.1:7861\n",
-      "\n",
-      "To create a public link, set `share=True` in `launch()`.\n"
-     ]
-    },
-    {
-     "data": {
-      "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7861/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
-      ],
-      "text/plain": [
-       "<IPython.core.display.HTML object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
+   "execution_count": null,
+   "id": "2b5a5f98",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "gradio = show(gatsby,\n",
     "              subprompts=[get_neighbors, ask],\n",
@@ -231,432 +139,17 @@
     "                        \"What did the narrator do after getting back to Chicago?\"],\n",
     "              keys={\"HF_KEY\"},\n",
     "              description=desc,\n",
-    "              code=open(\"gatsby.py\", \"r\").read().split(\"$\")[1].strip().strip(\"#\").strip()\n",
     "              )\n",
     "if __name__ == \"__main__\":\n",
     "    gradio.launch()"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "fdbe3ca1",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "7e96a1dc",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:44.097451Z",
-     "iopub.status.busy": "2023-03-22T17:00:44.097193Z",
-     "iopub.status.idle": "2023-03-22T17:00:44.100135Z",
-     "shell.execute_reply": "2023-03-22T17:00:44.099538Z"
-    },
-    "tags": [
-     "hide_inp"
-    ]
-   },
-   "outputs": [],
-   "source": [
-    "# QAPrompt().show({\"question\": \"Who was Gatsby?\", \"docs\": [\"doc1\", \"doc2\", \"doc3\"]}, \"\")\n",
-    "# # -\n",
-    "\n",
-    "# show_log(\"gatsby.log\")"
-   ]
   }
  ],
  "metadata": {
   "jupytext": {
-   "cell_metadata_filter": "tags,-all"
-  },
-  "kernelspec": {
-   "display_name": "minichain",
-   "language": "python",
-   "name": "minichain"
-  },
-  "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.10.6"
-  },
-  "widgets": {
-   "application/vnd.jupyter.widget-state+json": {
-    "state": {
-     "044c0c7d8f264a629d9a834ad7a3d02b": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "FloatProgressModel",
-      "state": {
-       "_dom_classes": [],
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "FloatProgressModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/controls",
-       "_view_module_version": "2.0.0",
-       "_view_name": "ProgressView",
-       "bar_style": "success",
-       "description": "",
-       "description_allow_html": false,
-       "layout": "IPY_MODEL_71c01d11477848a0a65eb6d8fa5c290c",
-       "max": 1.0,
-       "min": 0.0,
-       "orientation": "horizontal",
-       "style": "IPY_MODEL_b2191e94a5124ecaa10fcbba69552cad",
-       "tabbable": null,
-       "tooltip": null,
-       "value": 1.0
-      }
-     },
-     "0d3429e6725e49a0a63d578b0b46a1b8": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/base",
-       "_model_module_version": "2.0.0",
-       "_model_name": "LayoutModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "LayoutView",
-       "align_content": null,
-       "align_items": null,
-       "align_self": null,
-       "border_bottom": null,
-       "border_left": null,
-       "border_right": null,
-       "border_top": null,
-       "bottom": null,
-       "display": null,
-       "flex": null,
-       "flex_flow": null,
-       "grid_area": null,
-       "grid_auto_columns": null,
-       "grid_auto_flow": null,
-       "grid_auto_rows": null,
-       "grid_column": null,
-       "grid_gap": null,
-       "grid_row": null,
-       "grid_template_areas": null,
-       "grid_template_columns": null,
-       "grid_template_rows": null,
-       "height": null,
-       "justify_content": null,
-       "justify_items": null,
-       "left": null,
-       "margin": null,
-       "max_height": null,
-       "max_width": null,
-       "min_height": null,
-       "min_width": null,
-       "object_fit": null,
-       "object_position": null,
-       "order": null,
-       "overflow": null,
-       "padding": null,
-       "right": null,
-       "top": null,
-       "visibility": null,
-       "width": null
-      }
-     },
-     "2d0cb296ceba4f28afac033d606f3f80": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/base",
-       "_model_module_version": "2.0.0",
-       "_model_name": "LayoutModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "LayoutView",
-       "align_content": null,
-       "align_items": null,
-       "align_self": null,
-       "border_bottom": null,
-       "border_left": null,
-       "border_right": null,
-       "border_top": null,
-       "bottom": null,
-       "display": null,
-       "flex": null,
-       "flex_flow": null,
-       "grid_area": null,
-       "grid_auto_columns": null,
-       "grid_auto_flow": null,
-       "grid_auto_rows": null,
-       "grid_column": null,
-       "grid_gap": null,
-       "grid_row": null,
-       "grid_template_areas": null,
-       "grid_template_columns": null,
-       "grid_template_rows": null,
-       "height": null,
-       "justify_content": null,
-       "justify_items": null,
-       "left": null,
-       "margin": null,
-       "max_height": null,
-       "max_width": null,
-       "min_height": null,
-       "min_width": null,
-       "object_fit": null,
-       "object_position": null,
-       "order": null,
-       "overflow": null,
-       "padding": null,
-       "right": null,
-       "top": null,
-       "visibility": null,
-       "width": null
-      }
-     },
-     "306271182a3b4f51bfd69ea6467b7394": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/base",
-       "_model_module_version": "2.0.0",
-       "_model_name": "LayoutModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "LayoutView",
-       "align_content": null,
-       "align_items": null,
-       "align_self": null,
-       "border_bottom": null,
-       "border_left": null,
-       "border_right": null,
-       "border_top": null,
-       "bottom": null,
-       "display": null,
-       "flex": null,
-       "flex_flow": null,
-       "grid_area": null,
-       "grid_auto_columns": null,
-       "grid_auto_flow": null,
-       "grid_auto_rows": null,
-       "grid_column": null,
-       "grid_gap": null,
-       "grid_row": null,
-       "grid_template_areas": null,
-       "grid_template_columns": null,
-       "grid_template_rows": null,
-       "height": null,
-       "justify_content": null,
-       "justify_items": null,
-       "left": null,
-       "margin": null,
-       "max_height": null,
-       "max_width": null,
-       "min_height": null,
-       "min_width": null,
-       "object_fit": null,
-       "object_position": null,
-       "order": null,
-       "overflow": null,
-       "padding": null,
-       "right": null,
-       "top": null,
-       "visibility": null,
-       "width": null
-      }
-     },
-     "71c01d11477848a0a65eb6d8fa5c290c": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/base",
-       "_model_module_version": "2.0.0",
-       "_model_name": "LayoutModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "LayoutView",
-       "align_content": null,
-       "align_items": null,
-       "align_self": null,
-       "border_bottom": null,
-       "border_left": null,
-       "border_right": null,
-       "border_top": null,
-       "bottom": null,
-       "display": null,
-       "flex": null,
-       "flex_flow": null,
-       "grid_area": null,
-       "grid_auto_columns": null,
-       "grid_auto_flow": null,
-       "grid_auto_rows": null,
-       "grid_column": null,
-       "grid_gap": null,
-       "grid_row": null,
-       "grid_template_areas": null,
-       "grid_template_columns": null,
-       "grid_template_rows": null,
-       "height": null,
-       "justify_content": null,
-       "justify_items": null,
-       "left": null,
-       "margin": null,
-       "max_height": null,
-       "max_width": null,
-       "min_height": null,
-       "min_width": null,
-       "object_fit": null,
-       "object_position": null,
-       "order": null,
-       "overflow": null,
-       "padding": null,
-       "right": null,
-       "top": null,
-       "visibility": null,
-       "width": null
-      }
-     },
-     "86d0a2ceb7ad4f99978e37c2719f2960": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HBoxModel",
-      "state": {
-       "_dom_classes": [],
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HBoxModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/controls",
-       "_view_module_version": "2.0.0",
-       "_view_name": "HBoxView",
-       "box_style": "",
-       "children": [
-        "IPY_MODEL_f4226b39d8114ae291c8c1f4150b97b3",
-        "IPY_MODEL_044c0c7d8f264a629d9a834ad7a3d02b",
-        "IPY_MODEL_cf381c0f6e0d41408fdff78193c8e003"
-       ],
-       "layout": "IPY_MODEL_0d3429e6725e49a0a63d578b0b46a1b8",
-       "tabbable": null,
-       "tooltip": null
-      }
-     },
-     "b2191e94a5124ecaa10fcbba69552cad": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "ProgressStyleModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "ProgressStyleModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "StyleView",
-       "bar_color": null,
-       "description_width": ""
-      }
-     },
-     "bb8de7546e2d4a3d8c4b53a3170a9aa3": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HTMLStyleModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HTMLStyleModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "StyleView",
-       "background": null,
-       "description_width": "",
-       "font_size": null,
-       "text_color": null
-      }
-     },
-     "cf381c0f6e0d41408fdff78193c8e003": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HTMLModel",
-      "state": {
-       "_dom_classes": [],
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HTMLModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/controls",
-       "_view_module_version": "2.0.0",
-       "_view_name": "HTMLView",
-       "description": "",
-       "description_allow_html": false,
-       "layout": "IPY_MODEL_2d0cb296ceba4f28afac033d606f3f80",
-       "placeholder": "​",
-       "style": "IPY_MODEL_e239bdcb8e0f48f9b0cec18f6280bd8a",
-       "tabbable": null,
-       "tooltip": null,
-       "value": " 1/1 [00:00&lt;00:00, 101.96it/s]"
-      }
-     },
-     "e239bdcb8e0f48f9b0cec18f6280bd8a": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HTMLStyleModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HTMLStyleModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "StyleView",
-       "background": null,
-       "description_width": "",
-       "font_size": null,
-       "text_color": null
-      }
-     },
-     "f4226b39d8114ae291c8c1f4150b97b3": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HTMLModel",
-      "state": {
-       "_dom_classes": [],
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HTMLModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/controls",
-       "_view_module_version": "2.0.0",
-       "_view_name": "HTMLView",
-       "description": "",
-       "description_allow_html": false,
-       "layout": "IPY_MODEL_306271182a3b4f51bfd69ea6467b7394",
-       "placeholder": "​",
-       "style": "IPY_MODEL_bb8de7546e2d4a3d8c4b53a3170a9aa3",
-       "tabbable": null,
-       "tooltip": null,
-       "value": "100%"
-      }
-     }
-    },
-    "version_major": 2,
-    "version_minor": 0
-   }
+   "cell_metadata_filter": "tags,-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
   }
  },
  "nbformat": 4,

gatsby.py

@@ -2,7 +2,7 @@
 desc = """
 ### Book QA
 
-Chain that does question answering with Hugging Face embeddings. [[Code](https://github.com/srush/MiniChain/blob/main/examples/gatsby.py)]
+Chain that does question answering with Hugging Face embeddings. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/gatsby.ipynb)
 
 (Adapted from the [LlamaIndex example](https://github.com/jerryjliu/gpt_index/blob/main/examples/gatsby/TestGatsby.ipynb).)
 """
@@ -50,11 +50,3 @@ gradio = show(gatsby,
               )
 if __name__ == "__main__":
     gradio.launch()
-
-
-
-# + tags=["hide_inp"]
-# QAPrompt().show({"question": "Who was Gatsby?", "docs": ["doc1", "doc2", "doc3"]}, "")
-# # -
-
-# show_log("gatsby.log")

math_demo.ipynb

@@ -0,0 +1,130 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3284e5e6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3fbd7251",
+   "metadata": {
+    "lines_to_next_cell": 2,
+    "tags": [
+     "hide_inp"
+    ]
+   },
+   "outputs": [],
+   "source": [
+    "desc = \"\"\"\n",
+    "### Word Problem Solver\n",
+    "\n",
+    "Chain that solves a math word problem by first generating and then running Python code. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/math_demo.ipynb)\n",
+    "\n",
+    "(Adapted from Dust [maths-generate-code](https://dust.tt/spolu/a/d12ac33169))\n",
+    "\"\"\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "444f17d4",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from minichain import show, prompt, OpenAI, Python"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0a0b7432",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "@prompt(OpenAI(), template_file=\"math.pmpt.tpl\")\n",
+    "def math_prompt(model, question):\n",
+    "    \"Prompt to call GPT with a Jinja template\"\n",
+    "    return model(dict(question=question))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "78c1c973",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "@prompt(Python(), template=\"import math\\n{{code}}\")\n",
+    "def python(model, code):\n",
+    "    \"Prompt to call Python interpreter\"\n",
+    "    code = \"\\n\".join(code.strip().split(\"\\n\")[1:-1])\n",
+    "    return int(model(dict(code=code)))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "94491cd9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def math_demo(question):\n",
+    "    \"Chain them together\"\n",
+    "    return python(math_prompt(question))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d2e019aa",
+   "metadata": {
+    "lines_to_next_cell": 0,
+    "tags": [
+     "hide_inp"
+    ]
+   },
+   "outputs": [],
+   "source": [
+    "gradio = show(math_demo,\n",
+    "              examples=[\"What is the sum of the powers of 3 (3^i) that are smaller than 100?\",\n",
+    "                        \"What is the sum of the 10 first positive integers?\",],\n",
+    "                        # \"Carla is downloading a 200 GB file. She can download 2 GB/minute, but 40% of the way through the download, the download fails. Then Carla has to restart the download from the beginning. How load did it take her to download the file in minutes?\"],\n",
+    "              subprompts=[math_prompt, python],\n",
+    "              out_type=\"json\",\n",
+    "              description=desc,\n",
+    "              )\n",
+    "if __name__ == \"__main__\":\n",
+    "    gradio.launch()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "69b50b81",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "tags,-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

math_demo.py

@@ -2,7 +2,7 @@
 desc = """
 ### Word Problem Solver
 
-Chain that solves a math word problem by first generating and then running Python code. [[Code](https://github.com/srush/MiniChain/blob/main/examples/math.py)]
+Chain that solves a math word problem by first generating and then running Python code. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/math_demo.ipynb)
 
 (Adapted from Dust [maths-generate-code](https://dust.tt/spolu/a/d12ac33169))
 """
@@ -29,8 +29,8 @@ def math_demo(question):
     return python(math_prompt(question))
 
 # $
-    
-# + tags=["hide_inp"]    
+
+# + tags=["hide_inp"]
 gradio = show(math_demo,
               examples=["What is the sum of the powers of 3 (3^i) that are smaller than 100?",
                         "What is the sum of the 10 first positive integers?",],
@@ -43,4 +43,4 @@ gradio = show(math_demo,
 if __name__ == "__main__":
     gradio.launch()
 # -
-    
+

math_prompts.ipynb

@@ -0,0 +1,205 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "622575b8",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "83603f3b",
+   "metadata": {
+    "lines_to_next_cell": 2
+   },
+   "source": [
+    "Copyright 2022 PAL Authors. All rights reserved.\n",
+    "\n",
+    "Licensed under the Apache License, Version 2.0 (the \"License\");\n",
+    "you may not use this file except in compliance with the License.\n",
+    "You may obtain a copy of the License at\n",
+    "\n",
+    "    http://www.apache.org/licenses/LICENSE-2.0\n",
+    "\n",
+    "Unless required by applicable law or agreed to in writing, software\n",
+    "distributed under the License is distributed on an \"AS IS\" BASIS,\n",
+    "WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n",
+    "See the License for the specific language governing permissions and\n",
+    "limitations under the License."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "493da3c0",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "MATH_PROMPT = '''\n",
+    "Q: Olivia has $23. She bought five bagels for $3 each. How much money does she have left?\n",
+    "\n",
+    "# solution in Python:\n",
+    "\n",
+    "\n",
+    "def solution():\n",
+    "    \"\"\"Olivia has $23. She bought five bagels for $3 each. How much money does she have left?\"\"\"\n",
+    "    money_initial = 23\n",
+    "    bagels = 5\n",
+    "    bagel_cost = 3\n",
+    "    money_spent = bagels * bagel_cost\n",
+    "    money_left = money_initial - money_spent\n",
+    "    result = money_left\n",
+    "    return result\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "Q: Michael had 58 golf balls. On tuesday, he lost 23 golf balls. On wednesday, he lost 2 more. How many golf balls did he have at the end of wednesday?\n",
+    "\n",
+    "# solution in Python:\n",
+    "\n",
+    "\n",
+    "def solution():\n",
+    "    \"\"\"Michael had 58 golf balls. On tuesday, he lost 23 golf balls. On wednesday, he lost 2 more. How many golf balls did he have at the end of wednesday?\"\"\"\n",
+    "    golf_balls_initial = 58\n",
+    "    golf_balls_lost_tuesday = 23\n",
+    "    golf_balls_lost_wednesday = 2\n",
+    "    golf_balls_left = golf_balls_initial - golf_balls_lost_tuesday - golf_balls_lost_wednesday\n",
+    "    result = golf_balls_left\n",
+    "    return result\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "Q: There were nine computers in the server room. Five more computers were installed each day, from monday to thursday. How many computers are now in the server room?\n",
+    "\n",
+    "# solution in Python:\n",
+    "\n",
+    "\n",
+    "def solution():\n",
+    "    \"\"\"There were nine computers in the server room. Five more computers were installed each day, from monday to thursday. How many computers are now in the server room?\"\"\"\n",
+    "    computers_initial = 9\n",
+    "    computers_per_day = 5\n",
+    "    num_days = 4  # 4 days between monday and thursday\n",
+    "    computers_added = computers_per_day * num_days\n",
+    "    computers_total = computers_initial + computers_added\n",
+    "    result = computers_total\n",
+    "    return result\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "Q: Shawn has five toys. For Christmas, he got two toys each from his mom and dad. How many toys does he have now?\n",
+    "\n",
+    "# solution in Python:\n",
+    "\n",
+    "\n",
+    "def solution():\n",
+    "    \"\"\"Shawn has five toys. For Christmas, he got two toys each from his mom and dad. How many toys does he have now?\"\"\"\n",
+    "    toys_initial = 5\n",
+    "    mom_toys = 2\n",
+    "    dad_toys = 2\n",
+    "    total_received = mom_toys + dad_toys\n",
+    "    total_toys = toys_initial + total_received\n",
+    "    result = total_toys\n",
+    "    return result\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "Q: Jason had 20 lollipops. He gave Denny some lollipops. Now Jason has 12 lollipops. How many lollipops did Jason give to Denny?\n",
+    "\n",
+    "# solution in Python:\n",
+    "\n",
+    "\n",
+    "def solution():\n",
+    "    \"\"\"Jason had 20 lollipops. He gave Denny some lollipops. Now Jason has 12 lollipops. How many lollipops did Jason give to Denny?\"\"\"\n",
+    "    jason_lollipops_initial = 20\n",
+    "    jason_lollipops_after = 12\n",
+    "    denny_lollipops = jason_lollipops_initial - jason_lollipops_after\n",
+    "    result = denny_lollipops\n",
+    "    return result\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "Q: Leah had 32 chocolates and her sister had 42. If they ate 35, how many pieces do they have left in total?\n",
+    "\n",
+    "# solution in Python:\n",
+    "\n",
+    "\n",
+    "def solution():\n",
+    "    \"\"\"Leah had 32 chocolates and her sister had 42. If they ate 35, how many pieces do they have left in total?\"\"\"\n",
+    "    leah_chocolates = 32\n",
+    "    sister_chocolates = 42\n",
+    "    total_chocolates = leah_chocolates + sister_chocolates\n",
+    "    chocolates_eaten = 35\n",
+    "    chocolates_left = total_chocolates - chocolates_eaten\n",
+    "    result = chocolates_left\n",
+    "    return result\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "Q: If there are 3 cars in the parking lot and 2 more cars arrive, how many cars are in the parking lot?\n",
+    "\n",
+    "# solution in Python:\n",
+    "\n",
+    "\n",
+    "def solution():\n",
+    "    \"\"\"If there are 3 cars in the parking lot and 2 more cars arrive, how many cars are in the parking lot?\"\"\"\n",
+    "    cars_initial = 3\n",
+    "    cars_arrived = 2\n",
+    "    total_cars = cars_initial + cars_arrived\n",
+    "    result = total_cars\n",
+    "    return result\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "Q: There are 15 trees in the grove. Grove workers will plant trees in the grove today. After they are done, there will be 21 trees. How many trees did the grove workers plant today?\n",
+    "\n",
+    "# solution in Python:\n",
+    "\n",
+    "\n",
+    "def solution():\n",
+    "    \"\"\"There are 15 trees in the grove. Grove workers will plant trees in the grove today. After they are done, there will be 21 trees. How many trees did the grove workers plant today?\"\"\"\n",
+    "    trees_initial = 15\n",
+    "    trees_after = 21\n",
+    "    trees_added = trees_after - trees_initial\n",
+    "    result = trees_added\n",
+    "    return result\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "\n",
+    "Q: {question}\n",
+    "\n",
+    "# solution in Python:\n",
+    "'''.strip() + '\\n\\n\\n'"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

ner.ipynb

@@ -2,15 +2,21 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
-   "id": "6b40c8cc",
+   "execution_count": null,
+   "id": "609983df",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1f7ee63a",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:46.670809Z",
-     "iopub.status.busy": "2023-03-22T17:00:46.670088Z",
-     "iopub.status.idle": "2023-03-22T17:00:46.679313Z",
-     "shell.execute_reply": "2023-03-22T17:00:46.678767Z"
-    },
+    "lines_to_next_cell": 2,
     "tags": [
      "hide_inp"
     ]
@@ -21,31 +27,17 @@
     "desc = \"\"\"\n",
     "### Named Entity Recognition\n",
     "\n",
-    "Chain that does named entity recognition with arbitrary labels. [[Code](https://github.com/srush/MiniChain/blob/main/examples/ner.py)]\n",
+    "Chain that does named entity recognition with arbitrary labels. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/ner.ipynb)\n",
     "\n",
     "(Adapted from [promptify](https://github.com/promptslab/Promptify/blob/main/promptify/prompts/nlp/templates/ner.jinja)).\n",
     "\"\"\""
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "f6e33520",
-   "metadata": {},
-   "source": [
-    "$"
-   ]
-  },
   {
    "cell_type": "code",
-   "execution_count": 2,
-   "id": "de258da8",
+   "execution_count": null,
+   "id": "bedc40eb",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:46.681769Z",
-     "iopub.status.busy": "2023-03-22T17:00:46.681580Z",
-     "iopub.status.idle": "2023-03-22T17:00:47.985872Z",
-     "shell.execute_reply": "2023-03-22T17:00:47.985221Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -55,36 +47,23 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "id": "f517f9a5",
+   "execution_count": null,
+   "id": "316a1cab",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:47.988564Z",
-     "iopub.status.busy": "2023-03-22T17:00:47.988242Z",
-     "iopub.status.idle": "2023-03-22T17:00:47.991610Z",
-     "shell.execute_reply": "2023-03-22T17:00:47.991041Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
    "source": [
     "@prompt(OpenAI(), template_file = \"ner.pmpt.tpl\", parser=\"json\")\n",
-    "def ner_extract(model, **kwargs):\n",
+    "def ner_extract(model, kwargs):\n",
     "    return model(kwargs)"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
-   "id": "fe5ec878",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:47.993777Z",
-     "iopub.status.busy": "2023-03-22T17:00:47.993478Z",
-     "iopub.status.idle": "2023-03-22T17:00:47.996957Z",
-     "shell.execute_reply": "2023-03-22T17:00:47.996449Z"
-    }
-   },
+   "execution_count": null,
+   "id": "50d1c34a",
+   "metadata": {},
    "outputs": [],
    "source": [
     "@prompt(OpenAI())\n",
@@ -96,16 +75,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
-   "id": "cf059e37",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:47.999077Z",
-     "iopub.status.busy": "2023-03-22T17:00:47.998839Z",
-     "iopub.status.idle": "2023-03-22T17:00:48.002102Z",
-     "shell.execute_reply": "2023-03-22T17:00:48.001662Z"
-    }
-   },
+   "execution_count": null,
+   "id": "eaa2c029",
+   "metadata": {},
    "outputs": [],
    "source": [
     "def ner(text_input, labels, domain):\n",
@@ -114,137 +86,44 @@
    ]
   },
   {
-   "cell_type": "markdown",
-   "id": "fe198253",
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c4e8dfe8",
    "metadata": {},
-   "source": [
-    "$"
-   ]
+   "outputs": [],
+   "source": []
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
-   "id": "24ca918c",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:48.004490Z",
-     "iopub.status.busy": "2023-03-22T17:00:48.004142Z",
-     "iopub.status.idle": "2023-03-22T17:00:48.254708Z",
-     "shell.execute_reply": "2023-03-22T17:00:48.254069Z"
-    }
-   },
+   "execution_count": null,
+   "id": "546df8e9",
+   "metadata": {},
    "outputs": [],
    "source": [
     "gradio = show(ner,\n",
     "              examples=[[\"An NBA playoff pairing a year ago, the 76ers (39-20) meet the Miami Heat (32-29) for the first time this season on Monday night at home.\", \"Team, Date\", \"Sports\"]],\n",
     "              description=desc,\n",
     "              subprompts=[ner_extract, team_describe],\n",
-    "              code=open(\"ner.py\", \"r\").read().split(\"$\")[1].strip().strip(\"#\").strip(),\n",
     "              )"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
-   "id": "d5247097",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:48.257352Z",
-     "iopub.status.busy": "2023-03-22T17:00:48.257149Z",
-     "iopub.status.idle": "2023-03-22T17:00:48.319870Z",
-     "shell.execute_reply": "2023-03-22T17:00:48.319111Z"
-    },
-    "lines_to_next_cell": 2
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Running on local URL:  http://127.0.0.1:7861\n",
-      "\n",
-      "To create a public link, set `share=True` in `launch()`.\n"
-     ]
-    },
-    {
-     "data": {
-      "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7861/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
-      ],
-      "text/plain": [
-       "<IPython.core.display.HTML object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
-   "source": [
-    "if __name__ == \"__main__\":\n",
-    "    gradio.launch()"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "7cbb9856",
+   "execution_count": null,
+   "id": "3a5993c3",
    "metadata": {},
-   "source": [
-    "View prompt examples."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 8,
-   "id": "83d3ad7a",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:48.322776Z",
-     "iopub.status.busy": "2023-03-22T17:00:48.322504Z",
-     "iopub.status.idle": "2023-03-22T17:00:48.326248Z",
-     "shell.execute_reply": "2023-03-22T17:00:48.325611Z"
-    },
-    "tags": [
-     "hide_inp"
-    ]
-   },
    "outputs": [],
    "source": [
-    "# NERPrompt().show(\n",
-    "#     {\n",
-    "#         \"input\": \"I went to New York\",\n",
-    "#         \"domain\": \"Travel\",\n",
-    "#         \"labels\": [\"City\"]\n",
-    "#     },\n",
-    "#     '[{\"T\": \"City\", \"E\": \"New York\"}]',\n",
-    "# )\n",
-    "# # -\n",
-    "\n",
-    "# # View log.\n",
-    "\n",
-    "# minichain.show_log(\"ner.log\")"
+    "if __name__ == \"__main__\":\n",
+    "    gradio.launch()"
    ]
   }
  ],
  "metadata": {
   "jupytext": {
-   "cell_metadata_filter": "tags,-all"
-  },
-  "kernelspec": {
-   "display_name": "minichain",
-   "language": "python",
-   "name": "minichain"
-  },
-  "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.10.6"
+   "cell_metadata_filter": "tags,-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
   }
  },
  "nbformat": 4,

ner.py

@@ -3,7 +3,7 @@
 desc = """
 ### Named Entity Recognition
 
-Chain that does named entity recognition with arbitrary labels. [[Code](https://github.com/srush/MiniChain/blob/main/examples/ner.py)]
+Chain that does named entity recognition with arbitrary labels. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/ner.ipynb)
 
 (Adapted from [promptify](https://github.com/promptslab/Promptify/blob/main/promptify/prompts/nlp/templates/ner.jinja)).
 """
@@ -40,21 +40,3 @@ gradio = show(ner,
 
 if __name__ == "__main__":
     gradio.launch()
-
-
-# View prompt examples.
-
-# + tags=["hide_inp"]
-# NERPrompt().show(
-#     {
-#         "input": "I went to New York",
-#         "domain": "Travel",
-#         "labels": ["City"]
-#     },
-#     '[{"T": "City", "E": "New York"}]',
-# )
-# # -
-
-# # View log.
-
-# minichain.show_log("ner.log")

pal.ipynb

@@ -2,46 +2,38 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
-   "id": "c38ce0f6",
+   "execution_count": null,
+   "id": "c2508e7c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "8e61335c",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:50.962288Z",
-     "iopub.status.busy": "2023-03-22T17:00:50.961690Z",
-     "iopub.status.idle": "2023-03-22T17:00:50.980672Z",
-     "shell.execute_reply": "2023-03-22T17:00:50.978710Z"
-    }
+    "lines_to_next_cell": 2
    },
    "outputs": [],
    "source": [
     "desc = \"\"\"\n",
     "### Prompt-aided Language Models\n",
     "\n",
-    "Chain for answering complex problems by code generation and execution. [[Code](https://github.com/srush/MiniChain/blob/main/examples/pal.py)]\n",
+    "Chain for answering complex problems by code generation and execution. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/pal.ipynb)\n",
     "\n",
     "(Adapted from Prompt-aided Language Models [PAL](https://arxiv.org/pdf/2211.10435.pdf)).\n",
     "\"\"\""
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "feae8a53",
-   "metadata": {},
-   "source": [
-    "$"
-   ]
-  },
   {
    "cell_type": "code",
-   "execution_count": 2,
-   "id": "50b61935",
+   "execution_count": null,
+   "id": "ce0d9ca8",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:50.988390Z",
-     "iopub.status.busy": "2023-03-22T17:00:50.987754Z",
-     "iopub.status.idle": "2023-03-22T17:00:52.281141Z",
-     "shell.execute_reply": "2023-03-22T17:00:52.280472Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -51,15 +43,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "id": "a924b4d1",
+   "execution_count": null,
+   "id": "abce4228",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:52.283783Z",
-     "iopub.status.busy": "2023-03-22T17:00:52.283453Z",
-     "iopub.status.idle": "2023-03-22T17:00:52.286846Z",
-     "shell.execute_reply": "2023-03-22T17:00:52.286410Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -71,63 +57,34 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
-   "id": "0033b792",
+   "execution_count": null,
+   "id": "f695e1ac",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:52.288978Z",
-     "iopub.status.busy": "2023-03-22T17:00:52.288707Z",
-     "iopub.status.idle": "2023-03-22T17:00:52.291754Z",
-     "shell.execute_reply": "2023-03-22T17:00:52.291315Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
    "source": [
     "@prompt(Python())\n",
     "def python(model, inp):\n",
-    "    return int(model(inp + \"\\nprint(solution())\"))"
+    "    return float(model(inp + \"\\nprint(solution())\"))"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
-   "id": "60a5ddf7",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:52.294112Z",
-     "iopub.status.busy": "2023-03-22T17:00:52.293670Z",
-     "iopub.status.idle": "2023-03-22T17:00:52.296744Z",
-     "shell.execute_reply": "2023-03-22T17:00:52.296171Z"
-    },
-    "lines_to_next_cell": 1
-   },
+   "execution_count": null,
+   "id": "91dead77",
+   "metadata": {},
    "outputs": [],
    "source": [
     "def pal(question):\n",
     "    return python(pal_prompt(question))"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "c75fead5",
-   "metadata": {},
-   "source": [
-    "$"
-   ]
-  },
   {
    "cell_type": "code",
-   "execution_count": 6,
-   "id": "b0d40bd2",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:52.298984Z",
-     "iopub.status.busy": "2023-03-22T17:00:52.298796Z",
-     "iopub.status.idle": "2023-03-22T17:00:52.301777Z",
-     "shell.execute_reply": "2023-03-22T17:00:52.301305Z"
-    }
-   },
+   "execution_count": null,
+   "id": "dc1be82b",
+   "metadata": {},
    "outputs": [],
    "source": [
     "question = \"Melanie is a door-to-door saleswoman. She sold a third of her \" \\\n",
@@ -138,61 +95,25 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
-   "id": "3c4485a4",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:52.304005Z",
-     "iopub.status.busy": "2023-03-22T17:00:52.303713Z",
-     "iopub.status.idle": "2023-03-22T17:00:52.573239Z",
-     "shell.execute_reply": "2023-03-22T17:00:52.572557Z"
-    }
-   },
+   "execution_count": null,
+   "id": "782a0903",
+   "metadata": {},
    "outputs": [],
    "source": [
     "gradio = show(pal,\n",
     "              examples=[question],\n",
     "              subprompts=[pal_prompt, python],\n",
     "              description=desc,\n",
-    "              code=open(\"pal.py\", \"r\").read().split(\"$\")[1].strip().strip(\"#\").strip(),\n",
+    "              out_type=\"json\",\n",
     "              )"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
-   "id": "a5bebdcc",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:52.575840Z",
-     "iopub.status.busy": "2023-03-22T17:00:52.575635Z",
-     "iopub.status.idle": "2023-03-22T17:00:52.646986Z",
-     "shell.execute_reply": "2023-03-22T17:00:52.646403Z"
-    }
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Running on local URL:  http://127.0.0.1:7861\n",
-      "\n",
-      "To create a public link, set `share=True` in `launch()`.\n"
-     ]
-    },
-    {
-     "data": {
-      "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7861/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
-      ],
-      "text/plain": [
-       "<IPython.core.display.HTML object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
+   "execution_count": null,
+   "id": "337152d1",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "if __name__ == \"__main__\":\n",
     "    gradio.launch()"
@@ -201,24 +122,9 @@
  ],
  "metadata": {
   "jupytext": {
-   "cell_metadata_filter": "-all"
-  },
-  "kernelspec": {
-   "display_name": "minichain",
-   "language": "python",
-   "name": "minichain"
-  },
-  "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.10.6"
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
   }
  },
  "nbformat": 4,

pal.py

@@ -1,7 +1,7 @@
 desc = """
 ### Prompt-aided Language Models
 
-Chain for answering complex problems by code generation and execution. [[Code](https://github.com/srush/MiniChain/blob/main/examples/pal.py)]
+Chain for answering complex problems by code generation and execution. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/pal.ipynb)
 
 (Adapted from Prompt-aided Language Models [PAL](https://arxiv.org/pdf/2211.10435.pdf)).
 """

parallel.ipynb

@@ -0,0 +1,26 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6b039013",
+   "metadata": {
+    "lines_to_next_cell": 2
+   },
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

process.ipynb

@@ -0,0 +1,62 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "870a6ac7",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "2f24257e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import sys"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a1d0bd19",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "print(\"!pip install -q git+https://github.com/srush/MiniChain\")\n",
+    "print(\"!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . \")\n",
+    "print()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "dc1c0742",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "for l in sys.stdin:\n",
+    "\n",
+    "    if l.strip() == \"# $\":\n",
+    "        continue\n",
+    "    if l.strip().startswith(\"code=\"):\n",
+    "        continue\n",
+    "    print(l, end=\"\")"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

process.py

@@ -0,0 +1,13 @@
+import sys
+
+print("!pip install -q git+https://github.com/srush/MiniChain")
+print("!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . ")
+print()
+
+for l in sys.stdin:
+
+    if l.strip() == "# $":
+        continue
+    if l.strip().startswith("code="):
+        continue
+    print(l, end="")

process.py~

@@ -0,0 +1,11 @@
+import sys
+
+print("!pip install -qqq git+https://github.com/srush/MiniChain@gradio")
+print()
+
+for l in sys.stdin:
+    print(l, end="")
+    if l.strip() == "# $":
+        continue
+    if l.strip().startswith("code="):
+        continue

qa.ipynb

@@ -2,15 +2,21 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
-   "id": "383d6fa9",
+   "execution_count": null,
+   "id": "4425fe57",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "4068c493",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:55.200350Z",
-     "iopub.status.busy": "2023-03-22T17:00:55.199620Z",
-     "iopub.status.idle": "2023-03-22T17:00:55.219766Z",
-     "shell.execute_reply": "2023-03-22T17:00:55.217755Z"
-    },
+    "lines_to_next_cell": 2,
     "tags": [
      "hide_inp"
     ]
@@ -20,32 +26,17 @@
     "desc = \"\"\"\n",
     "### Question Answering with Retrieval\n",
     "\n",
-    "Chain that answers questions with embeedding based retrieval. [[Code](https://github.com/srush/MiniChain/blob/main/examples/qa.py)]\n",
+    "Chain that answers questions with embeedding based retrieval. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/qa.ipynb)\n",
     "\n",
     "(Adapted from [OpenAI Notebook](https://github.com/openai/openai-cookbook/blob/main/examples/Question_answering_using_embeddings.ipynb).)\n",
     "\"\"\""
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "51bf7e4c",
-   "metadata": {},
-   "source": [
-    "$"
-   ]
-  },
   {
    "cell_type": "code",
-   "execution_count": 2,
-   "id": "ab7925a4",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:55.226549Z",
-     "iopub.status.busy": "2023-03-22T17:00:55.226361Z",
-     "iopub.status.idle": "2023-03-22T17:00:56.643976Z",
-     "shell.execute_reply": "2023-03-22T17:00:56.643271Z"
-    }
-   },
+   "execution_count": null,
+   "id": "67a325ce",
+   "metadata": {},
    "outputs": [],
    "source": [
     "import datasets\n",
@@ -56,7 +47,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "0725197d",
+   "id": "b24d6ab8",
    "metadata": {},
    "source": [
     "We use Hugging Face Datasets as the database by assigning\n",
@@ -65,45 +56,10 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "id": "d2c2312f",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:56.646848Z",
-     "iopub.status.busy": "2023-03-22T17:00:56.646502Z",
-     "iopub.status.idle": "2023-03-22T17:00:56.714869Z",
-     "shell.execute_reply": "2023-03-22T17:00:56.714309Z"
-    }
-   },
-   "outputs": [
-    {
-     "data": {
-      "application/vnd.jupyter.widget-view+json": {
-       "model_id": "7ec1305895ad44a29163c6c51f641600",
-       "version_major": 2,
-       "version_minor": 0
-      },
-      "text/plain": [
-       "  0%|          | 0/4 [00:00<?, ?it/s]"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    },
-    {
-     "data": {
-      "text/plain": [
-       "Dataset({\n",
-       "    features: ['title', 'heading', 'content', 'tokens', 'embeddings'],\n",
-       "    num_rows: 3964\n",
-       "})"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "execution_count": null,
+   "id": "6a9774bb",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "olympics = datasets.load_from_disk(\"olympics.data\")\n",
     "olympics.add_faiss_index(\"embeddings\")"
@@ -111,7 +67,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1b2badf2",
+   "id": "4fa3c179",
    "metadata": {},
    "source": [
     "Fast KNN retieval prompt"
@@ -119,15 +75,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
-   "id": "456c5610",
+   "execution_count": null,
+   "id": "c3e5273c",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:56.717227Z",
-     "iopub.status.busy": "2023-03-22T17:00:56.717039Z",
-     "iopub.status.idle": "2023-03-22T17:00:56.720303Z",
-     "shell.execute_reply": "2023-03-22T17:00:56.719854Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -141,15 +91,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
-   "id": "61ae797c",
+   "execution_count": null,
+   "id": "12df8217",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:56.722475Z",
-     "iopub.status.busy": "2023-03-22T17:00:56.722183Z",
-     "iopub.status.idle": "2023-03-22T17:00:56.725075Z",
-     "shell.execute_reply": "2023-03-22T17:00:56.724640Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -162,16 +106,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
-   "id": "0d7639e8",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:56.727258Z",
-     "iopub.status.busy": "2023-03-22T17:00:56.726952Z",
-     "iopub.status.idle": "2023-03-22T17:00:56.729629Z",
-     "shell.execute_reply": "2023-03-22T17:00:56.729172Z"
-    }
-   },
+   "execution_count": null,
+   "id": "df36d640",
+   "metadata": {},
    "outputs": [],
    "source": [
     "def qa(query):\n",
@@ -180,27 +117,18 @@
    ]
   },
   {
-   "cell_type": "markdown",
-   "id": "fb5a255a",
-   "metadata": {
-    "lines_to_next_cell": 2
-   },
-   "source": [
-    "$"
-   ]
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "08dd0eeb",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
-   "id": "399bbfeb",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:56.731957Z",
-     "iopub.status.busy": "2023-03-22T17:00:56.731562Z",
-     "iopub.status.idle": "2023-03-22T17:00:56.734263Z",
-     "shell.execute_reply": "2023-03-22T17:00:56.733828Z"
-    }
-   },
+   "execution_count": null,
+   "id": "76e13672",
+   "metadata": {},
    "outputs": [],
    "source": [
     "questions = [\"Who won the 2020 Summer Olympics men's high jump?\",\n",
@@ -212,467 +140,28 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
-   "id": "ba3a7294",
+   "execution_count": null,
+   "id": "add9d044",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:56.736360Z",
-     "iopub.status.busy": "2023-03-22T17:00:56.736081Z",
-     "iopub.status.idle": "2023-03-22T17:00:57.115392Z",
-     "shell.execute_reply": "2023-03-22T17:00:57.114736Z"
-    }
+    "lines_to_next_cell": 2
    },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Running on local URL:  http://127.0.0.1:7861\n",
-      "\n",
-      "To create a public link, set `share=True` in `launch()`.\n"
-     ]
-    },
-    {
-     "data": {
-      "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7861/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
-      ],
-      "text/plain": [
-       "<IPython.core.display.HTML object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
+   "outputs": [],
    "source": [
     "gradio = show(qa,\n",
     "              examples=questions,\n",
     "              subprompts=[get_neighbors, get_result],\n",
     "              description=desc,\n",
-    "              code=open(\"qa.py\", \"r\").read().split(\"$\")[1].strip().strip(\"#\").strip(),\n",
     "              )\n",
     "if __name__ == \"__main__\":\n",
     "    gradio.launch()"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "524d2c29",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "markdown",
-   "id": "500edc1f",
-   "metadata": {},
-   "source": [
-    "# + tags=[\"hide_inp\"]\n",
-    "QAPrompt().show(\n",
-    "    {\"question\": \"Who won the race?\", \"docs\": [\"doc1\", \"doc2\", \"doc3\"]}, \"Joe Bob\"\n",
-    ")\n",
-    "# -"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "599833ea",
-   "metadata": {},
-   "source": [
-    "show_log(\"qa.log\")"
-   ]
   }
  ],
  "metadata": {
   "jupytext": {
-   "cell_metadata_filter": "tags,-all"
-  },
-  "kernelspec": {
-   "display_name": "minichain",
-   "language": "python",
-   "name": "minichain"
-  },
-  "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.10.6"
-  },
-  "widgets": {
-   "application/vnd.jupyter.widget-state+json": {
-    "state": {
-     "0e04d7f51add4bc0a2d621f6fb5d59a8": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/base",
-       "_model_module_version": "2.0.0",
-       "_model_name": "LayoutModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "LayoutView",
-       "align_content": null,
-       "align_items": null,
-       "align_self": null,
-       "border_bottom": null,
-       "border_left": null,
-       "border_right": null,
-       "border_top": null,
-       "bottom": null,
-       "display": null,
-       "flex": null,
-       "flex_flow": null,
-       "grid_area": null,
-       "grid_auto_columns": null,
-       "grid_auto_flow": null,
-       "grid_auto_rows": null,
-       "grid_column": null,
-       "grid_gap": null,
-       "grid_row": null,
-       "grid_template_areas": null,
-       "grid_template_columns": null,
-       "grid_template_rows": null,
-       "height": null,
-       "justify_content": null,
-       "justify_items": null,
-       "left": null,
-       "margin": null,
-       "max_height": null,
-       "max_width": null,
-       "min_height": null,
-       "min_width": null,
-       "object_fit": null,
-       "object_position": null,
-       "order": null,
-       "overflow": null,
-       "padding": null,
-       "right": null,
-       "top": null,
-       "visibility": null,
-       "width": null
-      }
-     },
-     "292f111caba646bf9d542fb9b7cd91d7": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HTMLModel",
-      "state": {
-       "_dom_classes": [],
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HTMLModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/controls",
-       "_view_module_version": "2.0.0",
-       "_view_name": "HTMLView",
-       "description": "",
-       "description_allow_html": false,
-       "layout": "IPY_MODEL_90473b32017648bb994c5818b85de6a8",
-       "placeholder": "​",
-       "style": "IPY_MODEL_daaac50111eb4f46a537b26b0a84acd9",
-       "tabbable": null,
-       "tooltip": null,
-       "value": "100%"
-      }
-     },
-     "2cf8dd3e135442fdadeb416bc1398f9a": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/base",
-       "_model_module_version": "2.0.0",
-       "_model_name": "LayoutModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "LayoutView",
-       "align_content": null,
-       "align_items": null,
-       "align_self": null,
-       "border_bottom": null,
-       "border_left": null,
-       "border_right": null,
-       "border_top": null,
-       "bottom": null,
-       "display": null,
-       "flex": null,
-       "flex_flow": null,
-       "grid_area": null,
-       "grid_auto_columns": null,
-       "grid_auto_flow": null,
-       "grid_auto_rows": null,
-       "grid_column": null,
-       "grid_gap": null,
-       "grid_row": null,
-       "grid_template_areas": null,
-       "grid_template_columns": null,
-       "grid_template_rows": null,
-       "height": null,
-       "justify_content": null,
-       "justify_items": null,
-       "left": null,
-       "margin": null,
-       "max_height": null,
-       "max_width": null,
-       "min_height": null,
-       "min_width": null,
-       "object_fit": null,
-       "object_position": null,
-       "order": null,
-       "overflow": null,
-       "padding": null,
-       "right": null,
-       "top": null,
-       "visibility": null,
-       "width": null
-      }
-     },
-     "59f2ff1fb7644aff98e9909b2b3d4f15": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/base",
-       "_model_module_version": "2.0.0",
-       "_model_name": "LayoutModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "LayoutView",
-       "align_content": null,
-       "align_items": null,
-       "align_self": null,
-       "border_bottom": null,
-       "border_left": null,
-       "border_right": null,
-       "border_top": null,
-       "bottom": null,
-       "display": null,
-       "flex": null,
-       "flex_flow": null,
-       "grid_area": null,
-       "grid_auto_columns": null,
-       "grid_auto_flow": null,
-       "grid_auto_rows": null,
-       "grid_column": null,
-       "grid_gap": null,
-       "grid_row": null,
-       "grid_template_areas": null,
-       "grid_template_columns": null,
-       "grid_template_rows": null,
-       "height": null,
-       "justify_content": null,
-       "justify_items": null,
-       "left": null,
-       "margin": null,
-       "max_height": null,
-       "max_width": null,
-       "min_height": null,
-       "min_width": null,
-       "object_fit": null,
-       "object_position": null,
-       "order": null,
-       "overflow": null,
-       "padding": null,
-       "right": null,
-       "top": null,
-       "visibility": null,
-       "width": null
-      }
-     },
-     "5c1f4b9bdf7c4e8f98d2e8dce11af726": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "ProgressStyleModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "ProgressStyleModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "StyleView",
-       "bar_color": null,
-       "description_width": ""
-      }
-     },
-     "68b1e61a345f48e8a25e229af275def1": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "FloatProgressModel",
-      "state": {
-       "_dom_classes": [],
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "FloatProgressModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/controls",
-       "_view_module_version": "2.0.0",
-       "_view_name": "ProgressView",
-       "bar_style": "success",
-       "description": "",
-       "description_allow_html": false,
-       "layout": "IPY_MODEL_0e04d7f51add4bc0a2d621f6fb5d59a8",
-       "max": 4.0,
-       "min": 0.0,
-       "orientation": "horizontal",
-       "style": "IPY_MODEL_5c1f4b9bdf7c4e8f98d2e8dce11af726",
-       "tabbable": null,
-       "tooltip": null,
-       "value": 4.0
-      }
-     },
-     "7ec1305895ad44a29163c6c51f641600": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HBoxModel",
-      "state": {
-       "_dom_classes": [],
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HBoxModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/controls",
-       "_view_module_version": "2.0.0",
-       "_view_name": "HBoxView",
-       "box_style": "",
-       "children": [
-        "IPY_MODEL_292f111caba646bf9d542fb9b7cd91d7",
-        "IPY_MODEL_68b1e61a345f48e8a25e229af275def1",
-        "IPY_MODEL_fdaa96ec85664a7fb6343b81facc8b0a"
-       ],
-       "layout": "IPY_MODEL_59f2ff1fb7644aff98e9909b2b3d4f15",
-       "tabbable": null,
-       "tooltip": null
-      }
-     },
-     "90473b32017648bb994c5818b85de6a8": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/base",
-       "_model_module_version": "2.0.0",
-       "_model_name": "LayoutModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "LayoutView",
-       "align_content": null,
-       "align_items": null,
-       "align_self": null,
-       "border_bottom": null,
-       "border_left": null,
-       "border_right": null,
-       "border_top": null,
-       "bottom": null,
-       "display": null,
-       "flex": null,
-       "flex_flow": null,
-       "grid_area": null,
-       "grid_auto_columns": null,
-       "grid_auto_flow": null,
-       "grid_auto_rows": null,
-       "grid_column": null,
-       "grid_gap": null,
-       "grid_row": null,
-       "grid_template_areas": null,
-       "grid_template_columns": null,
-       "grid_template_rows": null,
-       "height": null,
-       "justify_content": null,
-       "justify_items": null,
-       "left": null,
-       "margin": null,
-       "max_height": null,
-       "max_width": null,
-       "min_height": null,
-       "min_width": null,
-       "object_fit": null,
-       "object_position": null,
-       "order": null,
-       "overflow": null,
-       "padding": null,
-       "right": null,
-       "top": null,
-       "visibility": null,
-       "width": null
-      }
-     },
-     "d3792312355a49ab93e82971aa1ca4c4": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HTMLStyleModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HTMLStyleModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "StyleView",
-       "background": null,
-       "description_width": "",
-       "font_size": null,
-       "text_color": null
-      }
-     },
-     "daaac50111eb4f46a537b26b0a84acd9": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HTMLStyleModel",
-      "state": {
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HTMLStyleModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/base",
-       "_view_module_version": "2.0.0",
-       "_view_name": "StyleView",
-       "background": null,
-       "description_width": "",
-       "font_size": null,
-       "text_color": null
-      }
-     },
-     "fdaa96ec85664a7fb6343b81facc8b0a": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "HTMLModel",
-      "state": {
-       "_dom_classes": [],
-       "_model_module": "@jupyter-widgets/controls",
-       "_model_module_version": "2.0.0",
-       "_model_name": "HTMLModel",
-       "_view_count": null,
-       "_view_module": "@jupyter-widgets/controls",
-       "_view_module_version": "2.0.0",
-       "_view_name": "HTMLView",
-       "description": "",
-       "description_allow_html": false,
-       "layout": "IPY_MODEL_2cf8dd3e135442fdadeb416bc1398f9a",
-       "placeholder": "​",
-       "style": "IPY_MODEL_d3792312355a49ab93e82971aa1ca4c4",
-       "tabbable": null,
-       "tooltip": null,
-       "value": " 4/4 [00:00&lt;00:00, 89.21it/s]"
-      }
-     }
-    },
-    "version_major": 2,
-    "version_minor": 0
-   }
+   "cell_metadata_filter": "tags,-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
   }
  },
  "nbformat": 4,

qa.py

@@ -2,7 +2,7 @@
 desc = """
 ### Question Answering with Retrieval
 
-Chain that answers questions with embeedding based retrieval. [[Code](https://github.com/srush/MiniChain/blob/main/examples/qa.py)]
+Chain that answers questions with embeedding based retrieval. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/qa.ipynb)
 
 (Adapted from [OpenAI Notebook](https://github.com/openai/openai-cookbook/blob/main/examples/Question_answering_using_embeddings.ipynb).)
 """
@@ -57,12 +57,3 @@ gradio = show(qa,
 if __name__ == "__main__":
     gradio.launch()
 
-
-
-# # + tags=["hide_inp"]
-# QAPrompt().show(
-#     {"question": "Who won the race?", "docs": ["doc1", "doc2", "doc3"]}, "Joe Bob"
-# )
-# # -
-
-# show_log("qa.log")

requirements.txt

@@ -1,4 +1,4 @@
 gradio==3.21.0
-git+https://github.com/srush/minichain@gradio
+git+https://github.com/srush/minichain
 manifest-ml
 faiss-cpu

selfask.ipynb

@@ -3,56 +3,39 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "12eeb7f2",
+   "id": "0ac58c93",
    "metadata": {
-    "lines_to_next_cell": 0
+    "lines_to_next_cell": 2
    },
    "outputs": [],
-   "source": []
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
-   "id": "0290a7f2",
+   "execution_count": null,
+   "id": "25776503",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:59.828303Z",
-     "iopub.status.busy": "2023-03-22T17:00:59.827586Z",
-     "iopub.status.idle": "2023-03-22T17:00:59.836907Z",
-     "shell.execute_reply": "2023-03-22T17:00:59.836357Z"
-    }
+    "lines_to_next_cell": 2
    },
    "outputs": [],
    "source": [
     "desc = \"\"\"\n",
     "### Self-Ask\n",
     "\n",
-    " Notebook implementation of the self-ask + Google tool use prompt.\n",
+    " Notebook implementation of the self-ask + Google tool use prompt. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/selfask.ipynb)\n",
     "\n",
     " (Adapted from [Self-Ask repo](https://github.com/ofirpress/self-ask))\n",
     "\"\"\""
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "29b0fe2a",
-   "metadata": {},
-   "source": [
-    "$"
-   ]
-  },
   {
    "cell_type": "code",
-   "execution_count": 2,
-   "id": "f3544dba",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:00:59.839279Z",
-     "iopub.status.busy": "2023-03-22T17:00:59.839084Z",
-     "iopub.status.idle": "2023-03-22T17:01:01.144851Z",
-     "shell.execute_reply": "2023-03-22T17:01:01.144194Z"
-    }
-   },
+   "execution_count": null,
+   "id": "aca4e9fd",
+   "metadata": {},
    "outputs": [],
    "source": [
     "from dataclasses import dataclass, replace\n",
@@ -62,16 +45,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "id": "a89961cf",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:01.147570Z",
-     "iopub.status.busy": "2023-03-22T17:01:01.147250Z",
-     "iopub.status.idle": "2023-03-22T17:01:01.151110Z",
-     "shell.execute_reply": "2023-03-22T17:01:01.150563Z"
-    }
-   },
+   "execution_count": null,
+   "id": "86bc5318",
+   "metadata": {},
    "outputs": [],
    "source": [
     "@dataclass\n",
@@ -84,15 +60,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
-   "id": "0ebdb069",
+   "execution_count": null,
+   "id": "671c9b48",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:01.153276Z",
-     "iopub.status.busy": "2023-03-22T17:01:01.152995Z",
-     "iopub.status.idle": "2023-03-22T17:01:01.156865Z",
-     "shell.execute_reply": "2023-03-22T17:01:01.156229Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -111,15 +81,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
-   "id": "a8fdbdcf",
+   "execution_count": null,
+   "id": "eae30812",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:01.159342Z",
-     "iopub.status.busy": "2023-03-22T17:01:01.159137Z",
-     "iopub.status.idle": "2023-03-22T17:01:01.162472Z",
-     "shell.execute_reply": "2023-03-22T17:01:01.162014Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -128,7 +92,7 @@
     "def google(model, state):\n",
     "    if state.next_query is None:\n",
     "        return state\n",
-    "    \n",
+    "\n",
     "    result = model(state.next_query)\n",
     "    return State(state.question,\n",
     "                 state.history + \"\\nIntermediate answer: \" + result + \"\\n\")"
@@ -136,17 +100,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
-   "id": "82883cd6",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:01.164441Z",
-     "iopub.status.busy": "2023-03-22T17:01:01.164266Z",
-     "iopub.status.idle": "2023-03-22T17:01:01.167251Z",
-     "shell.execute_reply": "2023-03-22T17:01:01.166820Z"
-    },
-    "lines_to_next_cell": 1
-   },
+   "execution_count": null,
+   "id": "7577534c",
+   "metadata": {},
    "outputs": [],
    "source": [
     "def selfask(question):\n",
@@ -157,55 +113,17 @@
     "    return state"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "4f9bf472",
-   "metadata": {},
-   "source": [
-    "$"
-   ]
-  },
   {
    "cell_type": "code",
-   "execution_count": 7,
-   "id": "20e37090",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:01.169337Z",
-     "iopub.status.busy": "2023-03-22T17:01:01.169152Z",
-     "iopub.status.idle": "2023-03-22T17:01:01.506567Z",
-     "shell.execute_reply": "2023-03-22T17:01:01.505941Z"
-    }
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Running on local URL:  http://127.0.0.1:7861\n",
-      "\n",
-      "To create a public link, set `share=True` in `launch()`.\n"
-     ]
-    },
-    {
-     "data": {
-      "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7861/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
-      ],
-      "text/plain": [
-       "<IPython.core.display.HTML object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
+   "execution_count": null,
+   "id": "6f2eabf1",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "gradio = show(selfask,\n",
     "              examples=[\"What is the zip code of the city where George Washington was born?\"],\n",
     "              subprompts=[self_ask, google] * 3,\n",
     "              description=desc,\n",
-    "              code=open(\"selfask.py\", \"r\").read().split(\"$\")[1].strip().strip(\"#\").strip(),\n",
     "              out_type=\"json\"\n",
     "              )\n",
     "if __name__ == \"__main__\":\n",
@@ -215,7 +133,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "fdf780fc",
+   "id": "182ccd80",
    "metadata": {},
    "outputs": [],
    "source": []
@@ -223,24 +141,9 @@
  ],
  "metadata": {
   "jupytext": {
-   "cell_metadata_filter": "-all"
-  },
-  "kernelspec": {
-   "display_name": "minichain",
-   "language": "python",
-   "name": "minichain"
-  },
-  "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.10.6"
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
   }
  },
  "nbformat": 4,

selfask.py

@@ -2,7 +2,7 @@
 desc = """
 ### Self-Ask
 
- Notebook implementation of the self-ask + Google tool use prompt.
+ Notebook implementation of the self-ask + Google tool use prompt. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/selfask.ipynb)
 
  (Adapted from [Self-Ask repo](https://github.com/ofirpress/self-ask))
 """
@@ -32,12 +32,12 @@ def self_ask(model, state):
         return replace(state, next_query=res)
     elif out.startswith("So the final answer is:"):
         return replace(state, final_answer=res)
-    
+
 @prompt(Google())
 def google(model, state):
     if state.next_query is None:
         return state
-    
+
     result = model(state.next_query)
     return State(state.question,
                  state.history + "\nIntermediate answer: " + result + "\n")

show.ipynb

@@ -0,0 +1,38 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cddd35ad",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "de75d893",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import json, sys\n",
+    "from eliottree import tasks_from_iterable, render_tasks\n",
+    "render_tasks(sys.stderr.write,\n",
+    "             tasks_from_iterable([json.loads(l) for l in open(\"bash.log\")]),\n",
+    "             colorize=True, human_readable=True)"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

stats.ipynb

@@ -3,55 +3,39 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "f34e932d",
+   "id": "9acde9e6",
    "metadata": {
-    "lines_to_next_cell": 0
+    "lines_to_next_cell": 2
    },
    "outputs": [],
-   "source": []
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
-   "id": "12c38ff0",
+   "execution_count": null,
+   "id": "8f7a9c7a",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:04.041687Z",
-     "iopub.status.busy": "2023-03-22T17:01:04.040967Z",
-     "iopub.status.idle": "2023-03-22T17:01:04.062830Z",
-     "shell.execute_reply": "2023-03-22T17:01:04.060675Z"
-    }
+    "lines_to_next_cell": 2
    },
    "outputs": [],
    "source": [
     "desc = \"\"\"\n",
     "### Typed Extraction\n",
     "\n",
-    "Information extraction that is automatically generated from a typed specification. [[Code](https://github.com/srush/MiniChain/blob/main/examples/stats.py)]\n",
+    "Information extraction that is automatically generated from a typed specification. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/pal.ipynb)\n",
     "\n",
     "(Novel to MiniChain)\n",
     "\"\"\""
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "7c3125d1",
-   "metadata": {},
-   "source": [
-    "$"
-   ]
-  },
   {
    "cell_type": "code",
-   "execution_count": 2,
-   "id": "fafef296",
+   "execution_count": null,
+   "id": "59806374",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:04.071923Z",
-     "iopub.status.busy": "2023-03-22T17:01:04.071227Z",
-     "iopub.status.idle": "2023-03-22T17:01:05.411226Z",
-     "shell.execute_reply": "2023-03-22T17:01:05.410575Z"
-    },
     "lines_to_next_cell": 1
    },
    "outputs": [],
@@ -64,7 +48,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "defad07f",
+   "id": "e1cb8aa5",
    "metadata": {},
    "source": [
     "Data specification"
@@ -72,16 +56,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "id": "b28d3bef",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:05.414380Z",
-     "iopub.status.busy": "2023-03-22T17:01:05.413863Z",
-     "iopub.status.idle": "2023-03-22T17:01:05.418873Z",
-     "shell.execute_reply": "2023-03-22T17:01:05.418413Z"
-    }
-   },
+   "execution_count": null,
+   "id": "79fc4394",
+   "metadata": {},
    "outputs": [],
    "source": [
     "class StatType(Enum):\n",
@@ -102,69 +79,25 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
-   "id": "7e79dbb5",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:05.420896Z",
-     "iopub.status.busy": "2023-03-22T17:01:05.420709Z",
-     "iopub.status.idle": "2023-03-22T17:01:05.423985Z",
-     "shell.execute_reply": "2023-03-22T17:01:05.423565Z"
-    },
-    "lines_to_next_cell": 1
-   },
+   "execution_count": null,
+   "id": "61aed166",
+   "metadata": {},
    "outputs": [],
    "source": [
     "@prompt(OpenAI(), template_file=\"stats.pmpt.tpl\", parser=\"json\")\n",
     "def stats(model, passage):\n",
     "    out = model(dict(passage=passage, typ=type_to_prompt(Player)))\n",
-    "    return [Player(**j) for j in out]  "
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "id": "ecc90cb1",
-   "metadata": {},
-   "source": [
-    "$"
+    "    return [Player(**j) for j in out]"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
-   "id": "73cfa15f",
+   "execution_count": null,
+   "id": "b6949159",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-03-22T17:01:05.426096Z",
-     "iopub.status.busy": "2023-03-22T17:01:05.425913Z",
-     "iopub.status.idle": "2023-03-22T17:01:05.738646Z",
-     "shell.execute_reply": "2023-03-22T17:01:05.737900Z"
-    },
     "lines_to_next_cell": 2
    },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Running on local URL:  http://127.0.0.1:7861\n",
-      "\n",
-      "To create a public link, set `share=True` in `launch()`.\n"
-     ]
-    },
-    {
-     "data": {
-      "text/html": [
-       "<div><iframe src=\"http://127.0.0.1:7861/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
-      ],
-      "text/plain": [
-       "<IPython.core.display.HTML object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
-    }
-   ],
+   "outputs": [],
    "source": [
     "article = open(\"sixers.txt\").read()\n",
     "gradio = show(lambda passage: stats(passage),\n",
@@ -172,7 +105,6 @@
     "              subprompts=[stats],\n",
     "              out_type=\"json\",\n",
     "              description=desc,\n",
-    "              code=open(\"stats.py\", \"r\").read().split(\"$\")[1].strip().strip(\"#\").strip(),\n",
     ")\n",
     "if __name__ == \"__main__\":\n",
     "    gradio.launch()"
@@ -180,7 +112,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "c2c3c29a",
+   "id": "764eec48",
    "metadata": {},
    "source": [
     "ExtractionPrompt().show({\"passage\": \"Harden had 10 rebounds.\"},\n",
@@ -189,7 +121,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d6453878",
+   "id": "fefa83c7",
    "metadata": {},
    "source": [
     "# View the run log."
@@ -197,7 +129,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "4439426d",
+   "id": "f84dac69",
    "metadata": {},
    "source": [
     "minichain.show_log(\"bash.log\")"
@@ -206,24 +138,9 @@
  ],
  "metadata": {
   "jupytext": {
-   "cell_metadata_filter": "-all"
-  },
-  "kernelspec": {
-   "display_name": "minichain",
-   "language": "python",
-   "name": "minichain"
-  },
-  "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.10.6"
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
   }
  },
  "nbformat": 4,

stats.py

@@ -2,7 +2,7 @@
 desc = """
 ### Typed Extraction
 
-Information extraction that is automatically generated from a typed specification. [[Code](https://github.com/srush/MiniChain/blob/main/examples/stats.py)]
+Information extraction that is automatically generated from a typed specification. [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/srush/MiniChain/blob/master/examples/pal.ipynb)
 
 (Novel to MiniChain)
 """
@@ -37,7 +37,7 @@ class Player:
 @prompt(OpenAI(), template_file="stats.pmpt.tpl", parser="json")
 def stats(model, passage):
     out = model(dict(passage=passage, typ=type_to_prompt(Player)))
-    return [Player(**j) for j in out]  
+    return [Player(**j) for j in out]
 
 # $
 

story.ipynb

@@ -0,0 +1,61 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "eff9255a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d4910f1d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from asyncio import gather"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3265015f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "@simple\n",
+    "def outline():\n",
+    "    return \"What is the outline?\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "fc23ad60",
+   "metadata": {
+    "lines_to_next_cell": 2
+   },
+   "outputs": [],
+   "source": [
+    "@composite\n",
+    "async def story(bot: Bot):\n",
+    "    calls = [bot(outline()), bot(characters())]\n",
+    "    outline, characters = gather([calls])"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

summary.ipynb

@@ -1,27 +1,31 @@
 {
  "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "806592e6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
   {
    "cell_type": "markdown",
-   "id": "9f2e2ab6",
+   "id": "67d25b4b",
    "metadata": {},
    "source": [
-    "Summarize a long document by chunking and summarizing parts.\n",
-    "Uses aynchronous calls to the API.\n",
-    "Adapted from LangChain [Map-Reduce summary](https://langchain.readthedocs.io/en/stable/_modules/langchain/chains/mapreduce.html)."
+    "Summarize a long document by chunking and summarizing parts.  Uses\n",
+    "aynchronous calls to the API.  Adapted from LangChain [Map-Reduce\n",
+    "summary](https://langchain.readthedocs.io/en/stable/_modules/langchain/chains/mapreduce.html)."
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
-   "id": "3bd927bd",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-02-26T01:42:15.831020Z",
-     "iopub.status.busy": "2023-02-26T01:42:15.830584Z",
-     "iopub.status.idle": "2023-02-26T01:42:15.896842Z",
-     "shell.execute_reply": "2023-02-26T01:42:15.896197Z"
-    }
-   },
+   "execution_count": null,
+   "id": "264b8a01",
+   "metadata": {},
    "outputs": [],
    "source": [
     "import trio"
@@ -29,16 +33,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
-   "id": "9cd3cb67",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-02-26T01:42:15.899476Z",
-     "iopub.status.busy": "2023-02-26T01:42:15.899291Z",
-     "iopub.status.idle": "2023-02-26T01:42:15.950081Z",
-     "shell.execute_reply": "2023-02-26T01:42:15.949433Z"
-    }
-   },
+   "execution_count": null,
+   "id": "dba80a0d",
+   "metadata": {},
    "outputs": [],
    "source": [
     "from minichain import TemplatePrompt, show_log, start_chain"
@@ -46,7 +43,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a8d9d96b",
+   "id": "aeb6c55b",
    "metadata": {
     "lines_to_next_cell": 2
    },
@@ -56,85 +53,39 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
-   "id": "bda5ebf7",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-02-26T01:42:15.952712Z",
-     "iopub.status.busy": "2023-02-26T01:42:15.952495Z",
-     "iopub.status.idle": "2023-02-26T01:42:15.955601Z",
-     "shell.execute_reply": "2023-02-26T01:42:15.955080Z"
-    },
-    "lines_to_next_cell": 1
-   },
+   "execution_count": null,
+   "id": "083f994e",
+   "metadata": {},
    "outputs": [],
    "source": [
-    "class SummaryPrompt(TemplatePrompt[str]):\n",
+    "class SummaryPrompt(TemplatePrompt):\n",
     "    template_file = \"summary.pmpt.tpl\""
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
-   "id": "f451f936",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-02-26T01:42:15.957639Z",
-     "iopub.status.busy": "2023-02-26T01:42:15.957462Z",
-     "iopub.status.idle": "2023-02-26T01:42:15.961518Z",
-     "shell.execute_reply": "2023-02-26T01:42:15.960983Z"
-    }
-   },
+   "execution_count": null,
+   "id": "a3c3e226",
+   "metadata": {},
    "outputs": [],
    "source": [
-    "def chunk(f):\n",
+    "def chunk(f, width=4000, overlap=800):\n",
     "    \"Split a documents into 4800 character overlapping chunks\"\n",
     "    text = open(f).read().replace(\"\\n\\n\", \"\\n\")\n",
     "    chunks = []\n",
-    "    W = 4000\n",
-    "    O = 800\n",
     "    for i in range(4):\n",
-    "        if i * W > len(text):\n",
+    "        if i * width > len(text):\n",
     "            break\n",
-    "        chunks.append({\"text\": text[i * W : (i + 1) * W + O]})\n",
+    "        chunks.append({\"text\": text[i * width : (i + 1) * width + overlap]})\n",
     "    return chunks"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
-   "id": "bebcd7dd",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-02-26T01:42:15.963729Z",
-     "iopub.status.busy": "2023-02-26T01:42:15.963411Z",
-     "iopub.status.idle": "2023-02-26T01:42:29.326051Z",
-     "shell.execute_reply": "2023-02-26T01:42:29.325114Z"
-    }
-   },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "\u001b[1mINFO    \u001b[0m \u001b[32m2023-02-25 20:42:16.679\u001b[0m: \u001b[36masync_openai.client\u001b[0m:\u001b[36m__init__\u001b[0m: \u001b[1mOpenAI Client initialized: https://api.openai.com/v1\u001b[0m\n"
-     ]
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "\u001b[1mINFO    \u001b[0m \u001b[32m2023-02-25 20:42:16.679\u001b[0m: \u001b[36masync_openai.client\u001b[0m:\u001b[36m__init__\u001b[0m: \u001b[1mOpenAI Client initialized: https://api.openai.com/v1\u001b[0m\n"
-     ]
-    },
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      " In response to Russia's aggression in Ukraine, President Biden has taken action to provide military, economic, and humanitarian assistance to Ukraine, impose economic sanctions on Russia, and close off American airspace to Russian flights. He has also passed the American Rescue Plan to provide economic relief to Americans and the Bipartisan Innovation Act to level the playing field with China and other competitors. This bill will create 10,000 new jobs and increase Intel's investment to $100 billion. President Biden is also fighting inflation by capping the cost of insulin at $35 a month and allowing Medicare to negotiate lower prices for prescription drugs.\n"
-     ]
-    }
-   ],
+   "execution_count": null,
+   "id": "73ac1c05",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "with start_chain(\"summary\") as backend:\n",
     "    prompt = SummaryPrompt(backend.OpenAI())\n",
@@ -149,569 +100,27 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
-   "id": "ae8efff6",
+   "execution_count": null,
+   "id": "fd616c8e",
    "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-02-26T01:42:29.329395Z",
-     "iopub.status.busy": "2023-02-26T01:42:29.329095Z",
-     "iopub.status.idle": "2023-02-26T01:42:29.387107Z",
-     "shell.execute_reply": "2023-02-26T01:42:29.386548Z"
-    },
     "tags": [
      "hide_inp"
     ]
    },
-   "outputs": [
-    {
-     "data": {
-      "text/html": [
-       "\n",
-       "<!-- <link rel=\"stylesheet\" href=\"https://cdn.rawgit.com/Chalarangelo/mini.css/v3.0.1/dist/mini-default.min.css\"> -->\n",
-       " <main class=\"container\">\n",
-       "\n",
-       "<h3>SummaryPrompt</h3>\n",
-       "\n",
-       "<dl>\n",
-       "  <dt>Input:</dt>\n",
-       "  <dd>\n",
-       "<div class=\"highlight\"><pre><span></span><span class=\"p\">{</span><span class=\"s1\">&#39;text&#39;</span><span class=\"p\">:</span> <span class=\"s1\">&#39;One way to fight inflation is to drive down wages and make Americans poorer.&#39;</span><span class=\"p\">}</span>\n",
-       "</pre></div>\n",
-       "\n",
-       "\n",
-       "  </dd>\n",
-       "\n",
-       "  <dt> Full Prompt: </dt>\n",
-       "  <dd>\n",
-       "    <details>\n",
-       "      <summary>Prompt</summary>\n",
-       "      <p>Write a concise summary of the following:<br><br><br>\"<div style='color:red'>One way to fight inflation is to drive down wages and make Americans poorer.</div>\"<br><br><br>CONCISE SUMMARY:</p>\n",
-       "    </details>\n",
-       "  </dd>\n",
-       "\n",
-       "  <dt> Response: </dt>\n",
-       "  <dd>\n",
-       "    Make Americans poorer\n",
-       "  </dd>\n",
-       "\n",
-       "  <dt>Value:</dt>\n",
-       "  <dd>\n",
-       "<div class=\"highlight\"><pre><span></span><span class=\"n\">Make</span> <span class=\"n\">Americans</span> <span class=\"n\">poorer</span>\n",
-       "</pre></div>\n",
-       "\n",
-       "  </dd>\n",
-       "</main>\n"
-      ],
-      "text/plain": [
-       "HTML(html='\\n<!-- <link rel=\"stylesheet\" href=\"https://cdn.rawgit.com/Chalarangelo/mini.css/v3.0.1/dist/mini-default.min.css\"> -->\\n <main class=\"container\">\\n\\n<h3>SummaryPrompt</h3>\\n\\n<dl>\\n  <dt>Input:</dt>\\n  <dd>\\n<div class=\"highlight\"><pre><span></span><span class=\"p\">{</span><span class=\"s1\">&#39;text&#39;</span><span class=\"p\">:</span> <span class=\"s1\">&#39;One way to fight inflation is to drive down wages and make Americans poorer.&#39;</span><span class=\"p\">}</span>\\n</pre></div>\\n\\n\\n  </dd>\\n\\n  <dt> Full Prompt: </dt>\\n  <dd>\\n    <details>\\n      <summary>Prompt</summary>\\n      <p>Write a concise summary of the following:<br><br><br>\"<div style=\\'color:red\\'>One way to fight inflation is to drive down wages and make Americans poorer.</div>\"<br><br><br>CONCISE SUMMARY:</p>\\n    </details>\\n  </dd>\\n\\n  <dt> Response: </dt>\\n  <dd>\\n    Make Americans poorer\\n  </dd>\\n\\n  <dt>Value:</dt>\\n  <dd>\\n<div class=\"highlight\"><pre><span></span><span class=\"n\">Make</span> <span class=\"n\">Americans</span> <span class=\"n\">poorer</span>\\n</pre></div>\\n\\n  </dd>\\n</main>\\n')"
-      ]
-     },
-     "execution_count": 6,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "SummaryPrompt().show(\n",
-    "    {\n",
-    "        \"text\": \"One way to fight inflation is to drive down wages and make Americans poorer.\"\n",
-    "    },\n",
+    "    {\"text\": \"One way to fight is to drive down wages and make Americans poorer.\"},\n",
     "    \"Make Americans poorer\",\n",
     ")"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
-   "id": "59f0b406",
-   "metadata": {
-    "execution": {
-     "iopub.execute_input": "2023-02-26T01:42:29.389430Z",
-     "iopub.status.busy": "2023-02-26T01:42:29.389170Z",
-     "iopub.status.idle": "2023-02-26T01:42:29.435668Z",
-     "shell.execute_reply": "2023-02-26T01:42:29.434963Z"
-    }
-   },
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "\u001b[38;5;15m6174bdb9-8e7d-4859-9c91-d36235b2ad41\u001b[1m\u001b[0m\n",
-      "└── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m3.979s\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mInput Function\u001b[0m/2/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.001s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4minput\u001b[0m: \u001b[0m\n",
-      "    │   │   └── \u001b[38;5;4mtext\u001b[0m: r economy. The Ruble has lost 30% of its value. ⏎\n",
-      "    │   │       The Russian stock market has lost 40% of its value and trading remains suspended. Russia’s economy is reeling and Putin alone is to blame. ⏎\n",
-      "    │   │       Together with our allies we are providing support to the Ukrainians in their fight for freedom. Military assistance. Economic assistance. Humanitarian assistance. ⏎\n",
-      "    │   │       We are giving more than $1 Billion in direct assistance to Ukraine. ⏎\n",
-      "    │   │       And we will continue to aid the Ukrainian people as they defend their country and to help ease their suffering.  ⏎\n",
-      "    │   │       Let me be clear, our forces are not engaged and will not engage in conflict with Russian forces in Ukraine.  ⏎\n",
-      "    │   │       Our forces are not going to Europe to fight in Ukraine, but to defend our NATO Allies – in the event that Putin decides to keep moving west.  ⏎\n",
-      "    │   │       For that purpose we’ve mobilized American ground forces, air squadrons, and ship deployments to protect NATO countries including Poland, Romania, Latvia, Lithuania, and Estonia. ⏎\n",
-      "    │   │       As I have made crystal clear the United States and our Allies will defend every inch of territory of NATO countries with the full force of our collective power.  ⏎\n",
-      "    │   │       And we remain clear-eyed. The Ukrainians are fighting back with pure courage. But the next few days weeks, months, will be hard on them.  ⏎\n",
-      "    │   │       Putin has unleashed violence and chaos.  But while he may make gains on the battlefield – he will pay a continuing high price over the long run. ⏎\n",
-      "    │   │       And a proud Ukrainian people, who have known 30 years  of independence, have repeatedly shown that they will not tolerate anyone who tries to take their country backwards.  ⏎\n",
-      "    │   │       To all Americans, I will be honest with you, as I’ve always promised. A Russian dictator, invading a foreign country, has costs around the world. ⏎\n",
-      "    │   │       And I’m taking robust action to make sure the pain of our sanctions  is targeted at Russia’s economy. And I will use every tool at our disposal to protect American businesses and consumers. ⏎\n",
-      "    │   │       Tonight, I can announce that the United States has worked with 30 other countries to release 60 Million barrels of oil from reserves around the world.  ⏎\n",
-      "    │   │       America will lead that effort, releasing 30 Million barrels from our own Strategic Petroleum Reserve. And we stand ready to do more if necessary, unified with our allies.  ⏎\n",
-      "    │   │       These steps will help blunt gas prices here at home. And I know the news about what’s happening can seem alarming. ⏎\n",
-      "    │   │       But I want you to know that we are going to be okay. ⏎\n",
-      "    │   │       When the history of this era is written Putin’s war on Ukraine will have left Russia weaker and the rest of the world stronger. ⏎\n",
-      "    │   │       While it shouldn’t have taken something so terrible for people around the world to see what’s at stake now everyone sees it clearly. ⏎\n",
-      "    │   │       We see the unity among leaders of nations and a more unified Europe a more unified West. And we see unity among the people who are gathering in cities in large crowds around the world even in Russia to demonstrate their support for Ukraine.  ⏎\n",
-      "    │   │       In the battle between democracy and autocracy, democracies are rising to the moment, and the world is clearly choosing the side of peace and security. ⏎\n",
-      "    │   │       This is a real test. It’s going to take time. So let us continue to draw inspiration from the iron will of the Ukrainian people. ⏎\n",
-      "    │   │       To our fellow Ukrainian Americans who forge a deep bond that connects our two nations we stand with you. ⏎\n",
-      "    │   │       Putin may circle Kyiv with tanks, but he will never gain the hearts and souls of the Ukrainian people. ⏎\n",
-      "    │   │       He will never extinguish their love of freedom. He will never weaken the resolve of the free world. ⏎\n",
-      "    │   │       We meet tonight in an America that has lived through two of the hardest years this nation has ever faced. ⏎\n",
-      "    │   │       The pandemic has been punishing. ⏎\n",
-      "    │   │       And so many families are living paycheck to paycheck, struggling to keep up with the rising cost of food, gas, housing, and so much more. ⏎\n",
-      "    │   │       I understand. ⏎\n",
-      "    │   │       I remember when my Dad had to leave our home in Scranton, Pennsylvania to find work. I grew up in a family where if the price of food went up, you felt it. ⏎\n",
-      "    │   │       That’s why one of the first things I did as President was fight to pass the American Rescue Plan.  ⏎\n",
-      "    │   │       Because people were hurting. We needed to act, and we did. ⏎\n",
-      "    │   │       Few pieces of legislation have done more in a critical moment in our history to lift us out of crisis. ⏎\n",
-      "    │   │       It fueled our efforts to vaccinate the nation and combat COVID-19. It delivered immediate economic relief for tens of millions of Americans.  ⏎\n",
-      "    │   │       Helped put food on their table, keep a roof over their heads, and cut the cost of health insurance. ⏎\n",
-      "    │   │       And as my Dad used to say, it gave people a little breathing room. ⏎\n",
-      "    │   │       And unlike the $2 Trillion tax cut passed in the previous administration that benefitted the top 1% of Americans, the American Rescue Plan helped working people—and left no one behind. ⏎\n",
-      "    │   │       And it worked. It created jobs. Lots of jobs. ⏎\n",
-      "    │   │       In fact—our econ\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mInput Function\u001b[0m/2/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mPrompted\u001b[0m/3/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m3.978s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mprompt\u001b[0m: Write a concise summary of the following:⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   \"r economy. The Ruble has lost 30% of its value. ⏎\n",
-      "    │   │   The Russian stock market has lost 40% of its value and trading remains suspended. Russia’s economy is reeling and Putin alone is to blame. ⏎\n",
-      "    │   │   Together with our allies we are providing support to the Ukrainians in their fight for freedom. Military assistance. Economic assistance. Humanitarian assistance. ⏎\n",
-      "    │   │   We are giving more than $1 Billion in direct assistance to Ukraine. ⏎\n",
-      "    │   │   And we will continue to aid the Ukrainian people as they defend their country and to help ease their suffering.  ⏎\n",
-      "    │   │   Let me be clear, our forces are not engaged and will not engage in conflict with Russian forces in Ukraine.  ⏎\n",
-      "    │   │   Our forces are not going to Europe to fight in Ukraine, but to defend our NATO Allies – in the event that Putin decides to keep moving west.  ⏎\n",
-      "    │   │   For that purpose we’ve mobilized American ground forces, air squadrons, and ship deployments to protect NATO countries including Poland, Romania, Latvia, Lithuania, and Estonia. ⏎\n",
-      "    │   │   As I have made crystal clear the United States and our Allies will defend every inch of territory of NATO countries with the full force of our collective power.  ⏎\n",
-      "    │   │   And we remain clear-eyed. The Ukrainians are fighting back with pure courage. But the next few days weeks, months, will be hard on them.  ⏎\n",
-      "    │   │   Putin has unleashed violence and chaos.  But while he may make gains on the battlefield – he will pay a continuing high price over the long run. ⏎\n",
-      "    │   │   And a proud Ukrainian people, who have known 30 years  of independence, have repeatedly shown that they will not tolerate anyone who tries to take their country backwards.  ⏎\n",
-      "    │   │   To all Americans, I will be honest with you, as I’ve always promised. A Russian dictator, invading a foreign country, has costs around the world. ⏎\n",
-      "    │   │   And I’m taking robust action to make sure the pain of our sanctions  is targeted at Russia’s economy. And I will use every tool at our disposal to protect American businesses and consumers. ⏎\n",
-      "    │   │   Tonight, I can announce that the United States has worked with 30 other countries to release 60 Million barrels of oil from reserves around the world.  ⏎\n",
-      "    │   │   America will lead that effort, releasing 30 Million barrels from our own Strategic Petroleum Reserve. And we stand ready to do more if necessary, unified with our allies.  ⏎\n",
-      "    │   │   These steps will help blunt gas prices here at home. And I know the news about what’s happening can seem alarming. ⏎\n",
-      "    │   │   But I want you to know that we are going to be okay. ⏎\n",
-      "    │   │   When the history of this era is written Putin’s war on Ukraine will have left Russia weaker and the rest of the world stronger. ⏎\n",
-      "    │   │   While it shouldn’t have taken something so terrible for people around the world to see what’s at stake now everyone sees it clearly. ⏎\n",
-      "    │   │   We see the unity among leaders of nations and a more unified Europe a more unified West. And we see unity among the people who are gathering in cities in large crowds around the world even in Russia to demonstrate their support for Ukraine.  ⏎\n",
-      "    │   │   In the battle between democracy and autocracy, democracies are rising to the moment, and the world is clearly choosing the side of peace and security. ⏎\n",
-      "    │   │   This is a real test. It’s going to take time. So let us continue to draw inspiration from the iron will of the Ukrainian people. ⏎\n",
-      "    │   │   To our fellow Ukrainian Americans who forge a deep bond that connects our two nations we stand with you. ⏎\n",
-      "    │   │   Putin may circle Kyiv with tanks, but he will never gain the hearts and souls of the Ukrainian people. ⏎\n",
-      "    │   │   He will never extinguish their love of freedom. He will never weaken the resolve of the free world. ⏎\n",
-      "    │   │   We meet tonight in an America that has lived through two of the hardest years this nation has ever faced. ⏎\n",
-      "    │   │   The pandemic has been punishing. ⏎\n",
-      "    │   │   And so many families are living paycheck to paycheck, struggling to keep up with the rising cost of food, gas, housing, and so much more. ⏎\n",
-      "    │   │   I understand. ⏎\n",
-      "    │   │   I remember when my Dad had to leave our home in Scranton, Pennsylvania to find work. I grew up in a family where if the price of food went up, you felt it. ⏎\n",
-      "    │   │   That’s why one of the first things I did as President was fight to pass the American Rescue Plan.  ⏎\n",
-      "    │   │   Because people were hurting. We needed to act, and we did. ⏎\n",
-      "    │   │   Few pieces of legislation have done more in a critical moment in our history to lift us out of crisis. ⏎\n",
-      "    │   │   It fueled our efforts to vaccinate the nation and combat COVID-19. It delivered immediate economic relief for tens of millions of Americans.  ⏎\n",
-      "    │   │   Helped put food on their table, keep a roof over their heads, and cut the cost of health insurance. ⏎\n",
-      "    │   │   And as my Dad used to say, it gave people a little breathing room. ⏎\n",
-      "    │   │   And unlike the $2 Trillion tax cut passed in the previous administration that benefitted the top 1% of Americans, the American Rescue Plan helped working people—and left no one behind. ⏎\n",
-      "    │   │   And it worked. It created jobs. Lots of jobs. ⏎\n",
-      "    │   │   In fact—our econ\"⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   CONCISE SUMMARY:\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mPrompted\u001b[0m/3/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:20Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mResult\u001b[0m/4/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:20Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.000s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mresult\u001b[0m:  In response to Russian President Putin's invasion of Ukraine, the US and its allies are providing military, economic, and humanitarian assistance to the Ukrainians. The US is also leading an effort to release 60 million barrels of oil from reserves around the world, including 30 million from the US Strategic Petroleum Reserve, to help blunt gas prices. The US President has also passed the American Rescue Plan to provide economic relief to tens of millions of Americans and create jobs.\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mResult\u001b[0m/4/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:20Z\u001b[2m\u001b[0m\n",
-      "    └── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/5\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:20Z\u001b[2m\u001b[0m\n",
-      "\n",
-      "\u001b[38;5;15m2d2df820-8a1d-4db7-80ae-c54638c6a067\u001b[1m\u001b[0m\n",
-      "└── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m4.513s\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mInput Function\u001b[0m/2/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.001s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4minput\u001b[0m: \u001b[0m\n",
-      "    │   │   └── \u001b[38;5;4mtext\u001b[0m: st things I did as President was fight to pass the American Rescue Plan.  ⏎\n",
-      "    │   │       Because people were hurting. We needed to act, and we did. ⏎\n",
-      "    │   │       Few pieces of legislation have done more in a critical moment in our history to lift us out of crisis. ⏎\n",
-      "    │   │       It fueled our efforts to vaccinate the nation and combat COVID-19. It delivered immediate economic relief for tens of millions of Americans.  ⏎\n",
-      "    │   │       Helped put food on their table, keep a roof over their heads, and cut the cost of health insurance. ⏎\n",
-      "    │   │       And as my Dad used to say, it gave people a little breathing room. ⏎\n",
-      "    │   │       And unlike the $2 Trillion tax cut passed in the previous administration that benefitted the top 1% of Americans, the American Rescue Plan helped working people—and left no one behind. ⏎\n",
-      "    │   │       And it worked. It created jobs. Lots of jobs. ⏎\n",
-      "    │   │       In fact—our economy created over 6.5 Million new jobs just last year, more jobs created in one year  ⏎\n",
-      "    │   │       than ever before in the history of America. ⏎\n",
-      "    │   │       Our economy grew at a rate of 5.7% last year, the strongest growth in nearly 40 years, the first step in bringing fundamental change to an economy that hasn’t worked for the working people of this nation for too long.  ⏎\n",
-      "    │   │       For the past 40 years we were told that if we gave tax breaks to those at the very top, the benefits would trickle down to everyone else. ⏎\n",
-      "    │   │       But that trickle-down theory led to weaker economic growth, lower wages, bigger deficits, and the widest gap between those at the top and everyone else in nearly a century. ⏎\n",
-      "    │   │       Vice President Harris and I ran for office with a new economic vision for America. ⏎\n",
-      "    │   │       Invest in America. Educate Americans. Grow the workforce. Build the economy from the bottom up  ⏎\n",
-      "    │   │       and the middle out, not from the top down.  ⏎\n",
-      "    │   │       Because we know that when the middle class grows, the poor have a ladder up and the wealthy do very well. ⏎\n",
-      "    │   │       America used to have the best roads, bridges, and airports on Earth. ⏎\n",
-      "    │   │       Now our infrastructure is ranked 13th in the world. ⏎\n",
-      "    │   │       We won’t be able to compete for the jobs of the 21st Century if we don’t fix that. ⏎\n",
-      "    │   │       That’s why it was so important to pass the Bipartisan Infrastructure Law—the most sweeping investment to rebuild America in history. ⏎\n",
-      "    │   │       This was a bipartisan effort, and I want to thank the members of both parties who worked to make it happen. ⏎\n",
-      "    │   │       We’re done talking about infrastructure weeks. ⏎\n",
-      "    │   │       We’re going to have an infrastructure decade. ⏎\n",
-      "    │   │       It is going to transform America and put us on a path to win the economic competition of the 21st Century that we face with the rest of the world—particularly with China.  ⏎\n",
-      "    │   │       As I’ve told Xi Jinping, it is never a good bet to bet against the American people. ⏎\n",
-      "    │   │       We’ll create good jobs for millions of Americans, modernizing roads, airports, ports, and waterways all across America. ⏎\n",
-      "    │   │       And we’ll do it all to withstand the devastating effects of the climate crisis and promote environmental justice. ⏎\n",
-      "    │   │       We’ll build a national network of 500,000 electric vehicle charging stations, begin to replace poisonous lead pipes—so every child—and every American—has clean water to drink at home and at school, provide affordable high-speed internet for every American—urban, suburban, rural, and tribal communities. ⏎\n",
-      "    │   │       4,000 projects have already been announced. ⏎\n",
-      "    │   │       And tonight, I’m announcing that this year we will start fixing over 65,000 miles of highway and 1,500 bridges in disrepair. ⏎\n",
-      "    │   │       When we use taxpayer dollars to rebuild America – we are going to Buy American: buy American products to support American jobs. ⏎\n",
-      "    │   │       The federal government spends about $600 Billion a year to keep the country safe and secure. ⏎\n",
-      "    │   │       There’s been a law on the books for almost a century ⏎\n",
-      "    │   │       to make sure taxpayers’ dollars support American jobs and businesses. ⏎\n",
-      "    │   │       Every Administration says they’ll do it, but we are actually doing it. ⏎\n",
-      "    │   │       We will buy American to make sure everything from the deck of an aircraft carrier to the steel on highway guardrails are made in America. ⏎\n",
-      "    │   │       But to compete for the best jobs of the future, we also need to level the playing field with China and other competitors. ⏎\n",
-      "    │   │       That’s why it is so important to pass the Bipartisan Innovation Act sitting in Congress that will make record investments in emerging technologies and American manufacturing. ⏎\n",
-      "    │   │       Let me give you one example of why it’s so important to pass it. ⏎\n",
-      "    │   │       If you travel 20 miles east of Columbus, Ohio, you’ll find 1,000 empty acres of land. ⏎\n",
-      "    │   │       It won’t look like much, but if you stop and look closely, you’ll see a “Field of dreams,” the ground on which America’s future will be built. ⏎\n",
-      "    │   │       This is where Intel, the American company that helped build Silicon Valley, is going to build its $20 billion semiconductor “mega site”. ⏎\n",
-      "    │   │       Up to eight state-of-the-art factories in one place. 10,000 new good-paying jobs. ⏎\n",
-      "    │   │       Some of the most sophisticated manufacturing in the world to make computer chips the size of a fingertip that po\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mInput Function\u001b[0m/2/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mPrompted\u001b[0m/3/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m4.512s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mprompt\u001b[0m: Write a concise summary of the following:⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   \"st things I did as President was fight to pass the American Rescue Plan.  ⏎\n",
-      "    │   │   Because people were hurting. We needed to act, and we did. ⏎\n",
-      "    │   │   Few pieces of legislation have done more in a critical moment in our history to lift us out of crisis. ⏎\n",
-      "    │   │   It fueled our efforts to vaccinate the nation and combat COVID-19. It delivered immediate economic relief for tens of millions of Americans.  ⏎\n",
-      "    │   │   Helped put food on their table, keep a roof over their heads, and cut the cost of health insurance. ⏎\n",
-      "    │   │   And as my Dad used to say, it gave people a little breathing room. ⏎\n",
-      "    │   │   And unlike the $2 Trillion tax cut passed in the previous administration that benefitted the top 1% of Americans, the American Rescue Plan helped working people—and left no one behind. ⏎\n",
-      "    │   │   And it worked. It created jobs. Lots of jobs. ⏎\n",
-      "    │   │   In fact—our economy created over 6.5 Million new jobs just last year, more jobs created in one year  ⏎\n",
-      "    │   │   than ever before in the history of America. ⏎\n",
-      "    │   │   Our economy grew at a rate of 5.7% last year, the strongest growth in nearly 40 years, the first step in bringing fundamental change to an economy that hasn’t worked for the working people of this nation for too long.  ⏎\n",
-      "    │   │   For the past 40 years we were told that if we gave tax breaks to those at the very top, the benefits would trickle down to everyone else. ⏎\n",
-      "    │   │   But that trickle-down theory led to weaker economic growth, lower wages, bigger deficits, and the widest gap between those at the top and everyone else in nearly a century. ⏎\n",
-      "    │   │   Vice President Harris and I ran for office with a new economic vision for America. ⏎\n",
-      "    │   │   Invest in America. Educate Americans. Grow the workforce. Build the economy from the bottom up  ⏎\n",
-      "    │   │   and the middle out, not from the top down.  ⏎\n",
-      "    │   │   Because we know that when the middle class grows, the poor have a ladder up and the wealthy do very well. ⏎\n",
-      "    │   │   America used to have the best roads, bridges, and airports on Earth. ⏎\n",
-      "    │   │   Now our infrastructure is ranked 13th in the world. ⏎\n",
-      "    │   │   We won’t be able to compete for the jobs of the 21st Century if we don’t fix that. ⏎\n",
-      "    │   │   That’s why it was so important to pass the Bipartisan Infrastructure Law—the most sweeping investment to rebuild America in history. ⏎\n",
-      "    │   │   This was a bipartisan effort, and I want to thank the members of both parties who worked to make it happen. ⏎\n",
-      "    │   │   We’re done talking about infrastructure weeks. ⏎\n",
-      "    │   │   We’re going to have an infrastructure decade. ⏎\n",
-      "    │   │   It is going to transform America and put us on a path to win the economic competition of the 21st Century that we face with the rest of the world—particularly with China.  ⏎\n",
-      "    │   │   As I’ve told Xi Jinping, it is never a good bet to bet against the American people. ⏎\n",
-      "    │   │   We’ll create good jobs for millions of Americans, modernizing roads, airports, ports, and waterways all across America. ⏎\n",
-      "    │   │   And we’ll do it all to withstand the devastating effects of the climate crisis and promote environmental justice. ⏎\n",
-      "    │   │   We’ll build a national network of 500,000 electric vehicle charging stations, begin to replace poisonous lead pipes—so every child—and every American—has clean water to drink at home and at school, provide affordable high-speed internet for every American—urban, suburban, rural, and tribal communities. ⏎\n",
-      "    │   │   4,000 projects have already been announced. ⏎\n",
-      "    │   │   And tonight, I’m announcing that this year we will start fixing over 65,000 miles of highway and 1,500 bridges in disrepair. ⏎\n",
-      "    │   │   When we use taxpayer dollars to rebuild America – we are going to Buy American: buy American products to support American jobs. ⏎\n",
-      "    │   │   The federal government spends about $600 Billion a year to keep the country safe and secure. ⏎\n",
-      "    │   │   There’s been a law on the books for almost a century ⏎\n",
-      "    │   │   to make sure taxpayers’ dollars support American jobs and businesses. ⏎\n",
-      "    │   │   Every Administration says they’ll do it, but we are actually doing it. ⏎\n",
-      "    │   │   We will buy American to make sure everything from the deck of an aircraft carrier to the steel on highway guardrails are made in America. ⏎\n",
-      "    │   │   But to compete for the best jobs of the future, we also need to level the playing field with China and other competitors. ⏎\n",
-      "    │   │   That’s why it is so important to pass the Bipartisan Innovation Act sitting in Congress that will make record investments in emerging technologies and American manufacturing. ⏎\n",
-      "    │   │   Let me give you one example of why it’s so important to pass it. ⏎\n",
-      "    │   │   If you travel 20 miles east of Columbus, Ohio, you’ll find 1,000 empty acres of land. ⏎\n",
-      "    │   │   It won’t look like much, but if you stop and look closely, you’ll see a “Field of dreams,” the ground on which America’s future will be built. ⏎\n",
-      "    │   │   This is where Intel, the American company that helped build Silicon Valley, is going to build its $20 billion semiconductor “mega site”. ⏎\n",
-      "    │   │   Up to eight state-of-the-art factories in one place. 10,000 new good-paying jobs. ⏎\n",
-      "    │   │   Some of the most sophisticated manufacturing in the world to make computer chips the size of a fingertip that po\"⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   CONCISE SUMMARY:\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mPrompted\u001b[0m/3/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:21Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mResult\u001b[0m/4/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:21Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.000s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mresult\u001b[0m:  President Biden fought to pass the American Rescue Plan to provide economic relief to tens of millions of Americans and create jobs. He also passed the Bipartisan Infrastructure Law to rebuild America and the Bipartisan Innovation Act to level the playing field with China and other competitors. An example of this is Intel's $20 billion semiconductor \"mega site\" in Ohio, which will create 10,000 new jobs.\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mResult\u001b[0m/4/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:21Z\u001b[2m\u001b[0m\n",
-      "    └── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/5\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:21Z\u001b[2m\u001b[0m\n",
-      "\n",
-      "\u001b[38;5;15mdff81dcf-2073-4ba4-8856-d0f82c213311\u001b[1m\u001b[0m\n",
-      "└── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m6.248s\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mInput Function\u001b[0m/2/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.001s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4minput\u001b[0m: \u001b[0m\n",
-      "    │   │   └── \u001b[38;5;4mtext\u001b[0m: Madam Speaker, Madam Vice President, our First Lady and Second Gentleman. Members of Congress and the Cabinet. Justices of the Supreme Court. My fellow Americans.  ⏎\n",
-      "    │   │       Last year COVID-19 kept us apart. This year we are finally together again. ⏎\n",
-      "    │   │       Tonight, we meet as Democrats Republicans and Independents. But most importantly as Americans. ⏎\n",
-      "    │   │       With a duty to one another to the American people to the Constitution. ⏎\n",
-      "    │   │       And with an unwavering resolve that freedom will always triumph over tyranny. ⏎\n",
-      "    │   │       Six days ago, Russia’s Vladimir Putin sought to shake the foundations of the free world thinking he could make it bend to his menacing ways. But he badly miscalculated. ⏎\n",
-      "    │   │       He thought he could roll into Ukraine and the world would roll over. Instead he met a wall of strength he never imagined. ⏎\n",
-      "    │   │       He met the Ukrainian people. ⏎\n",
-      "    │   │       From President Zelenskyy to every Ukrainian, their fearlessness, their courage, their determination, inspires the world. ⏎\n",
-      "    │   │       Groups of citizens blocking tanks with their bodies. Everyone from students to retirees teachers turned soldiers defending their homeland. ⏎\n",
-      "    │   │       In this struggle as President Zelenskyy said in his speech to the European Parliament “Light will win over darkness.” The Ukrainian Ambassador to the United States is here tonight. ⏎\n",
-      "    │   │       Let each of us here tonight in this Chamber send an unmistakable signal to Ukraine and to the world. ⏎\n",
-      "    │   │       Please rise if you are able and show that, Yes, we the United States of America stand with the Ukrainian people. ⏎\n",
-      "    │   │       Throughout our history we’ve learned this lesson when dictators do not pay a price for their aggression they cause more chaos.   ⏎\n",
-      "    │   │       They keep moving.   ⏎\n",
-      "    │   │       And the costs and the threats to America and the world keep rising.   ⏎\n",
-      "    │   │       That’s why the NATO Alliance was created to secure peace and stability in Europe after World War 2. ⏎\n",
-      "    │   │       The United States is a member along with 29 other nations. ⏎\n",
-      "    │   │       It matters. American diplomacy matters. American resolve matters. ⏎\n",
-      "    │   │       Putin’s latest attack on Ukraine was premeditated and unprovoked. ⏎\n",
-      "    │   │       He rejected repeated efforts at diplomacy. ⏎\n",
-      "    │   │       He thought the West and NATO wouldn’t respond. And he thought he could divide us at home. Putin was wrong. We were ready.  Here is what we did.   ⏎\n",
-      "    │   │       We prepared extensively and carefully. ⏎\n",
-      "    │   │       We spent months building a coalition of other freedom-loving nations from Europe and the Americas to Asia and Africa to confront Putin. ⏎\n",
-      "    │   │       I spent countless hours unifying our European allies. We shared with the world in advance what we knew Putin was planning and precisely how he would try to falsely justify his aggression.  ⏎\n",
-      "    │   │       We countered Russia’s lies with truth.   ⏎\n",
-      "    │   │       And now that he has acted the free world is holding him accountable. ⏎\n",
-      "    │   │       Along with twenty-seven members of the European Union including France, Germany, Italy, as well as countries like the United Kingdom, Canada, Japan, Korea, Australia, New Zealand, and many others, even Switzerland. ⏎\n",
-      "    │   │       We are inflicting pain on Russia and supporting the people of Ukraine. Putin is now isolated from the world more than ever. ⏎\n",
-      "    │   │       Together with our allies –we are right now enforcing powerful economic sanctions. ⏎\n",
-      "    │   │       We are cutting off Russia’s largest banks from the international financial system.  ⏎\n",
-      "    │   │       Preventing Russia’s central bank from defending the Russian Ruble making Putin’s $630 Billion “war fund” worthless.   ⏎\n",
-      "    │   │       We are choking off Russia’s access to technology that will sap its economic strength and weaken its military for years to come.  ⏎\n",
-      "    │   │       Tonight I say to the Russian oligarchs and corrupt leaders who have bilked billions of dollars off this violent regime no more. ⏎\n",
-      "    │   │       The U.S. Department of Justice is assembling a dedicated task force to go after the crimes of Russian oligarchs.  ⏎\n",
-      "    │   │       We are joining with our European allies to find and seize your yachts your luxury apartments your private jets. We are coming for your ill-begotten gains. ⏎\n",
-      "    │   │       And tonight I am announcing that we will join our allies in closing off American air space to all Russian flights – further isolating Russia – and adding an additional squeeze –on their economy. The Ruble has lost 30% of its value. ⏎\n",
-      "    │   │       The Russian stock market has lost 40% of its value and trading remains suspended. Russia’s economy is reeling and Putin alone is to blame. ⏎\n",
-      "    │   │       Together with our allies we are providing support to the Ukrainians in their fight for freedom. Military assistance. Economic assistance. Humanitarian assistance. ⏎\n",
-      "    │   │       We are giving more than $1 Billion in direct assistance to Ukraine. ⏎\n",
-      "    │   │       And we will continue to aid the Ukrainian people as they defend their country and to help ease their suffering.  ⏎\n",
-      "    │   │       Let me be clear, our forces are not engaged and will not engage in conflict with Russian forces in Ukraine.  ⏎\n",
-      "    │   │       Our forces are not going to Europe to fight in Ukraine, but to defend our NATO Allies – in the event that Putin decides to keep moving west.  ⏎\n",
-      "    │   │       For that pu\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mInput Function\u001b[0m/2/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mPrompted\u001b[0m/3/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m6.247s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mprompt\u001b[0m: Write a concise summary of the following:⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   \"Madam Speaker, Madam Vice President, our First Lady and Second Gentleman. Members of Congress and the Cabinet. Justices of the Supreme Court. My fellow Americans.  ⏎\n",
-      "    │   │   Last year COVID-19 kept us apart. This year we are finally together again. ⏎\n",
-      "    │   │   Tonight, we meet as Democrats Republicans and Independents. But most importantly as Americans. ⏎\n",
-      "    │   │   With a duty to one another to the American people to the Constitution. ⏎\n",
-      "    │   │   And with an unwavering resolve that freedom will always triumph over tyranny. ⏎\n",
-      "    │   │   Six days ago, Russia’s Vladimir Putin sought to shake the foundations of the free world thinking he could make it bend to his menacing ways. But he badly miscalculated. ⏎\n",
-      "    │   │   He thought he could roll into Ukraine and the world would roll over. Instead he met a wall of strength he never imagined. ⏎\n",
-      "    │   │   He met the Ukrainian people. ⏎\n",
-      "    │   │   From President Zelenskyy to every Ukrainian, their fearlessness, their courage, their determination, inspires the world. ⏎\n",
-      "    │   │   Groups of citizens blocking tanks with their bodies. Everyone from students to retirees teachers turned soldiers defending their homeland. ⏎\n",
-      "    │   │   In this struggle as President Zelenskyy said in his speech to the European Parliament “Light will win over darkness.” The Ukrainian Ambassador to the United States is here tonight. ⏎\n",
-      "    │   │   Let each of us here tonight in this Chamber send an unmistakable signal to Ukraine and to the world. ⏎\n",
-      "    │   │   Please rise if you are able and show that, Yes, we the United States of America stand with the Ukrainian people. ⏎\n",
-      "    │   │   Throughout our history we’ve learned this lesson when dictators do not pay a price for their aggression they cause more chaos.   ⏎\n",
-      "    │   │   They keep moving.   ⏎\n",
-      "    │   │   And the costs and the threats to America and the world keep rising.   ⏎\n",
-      "    │   │   That’s why the NATO Alliance was created to secure peace and stability in Europe after World War 2. ⏎\n",
-      "    │   │   The United States is a member along with 29 other nations. ⏎\n",
-      "    │   │   It matters. American diplomacy matters. American resolve matters. ⏎\n",
-      "    │   │   Putin’s latest attack on Ukraine was premeditated and unprovoked. ⏎\n",
-      "    │   │   He rejected repeated efforts at diplomacy. ⏎\n",
-      "    │   │   He thought the West and NATO wouldn’t respond. And he thought he could divide us at home. Putin was wrong. We were ready.  Here is what we did.   ⏎\n",
-      "    │   │   We prepared extensively and carefully. ⏎\n",
-      "    │   │   We spent months building a coalition of other freedom-loving nations from Europe and the Americas to Asia and Africa to confront Putin. ⏎\n",
-      "    │   │   I spent countless hours unifying our European allies. We shared with the world in advance what we knew Putin was planning and precisely how he would try to falsely justify his aggression.  ⏎\n",
-      "    │   │   We countered Russia’s lies with truth.   ⏎\n",
-      "    │   │   And now that he has acted the free world is holding him accountable. ⏎\n",
-      "    │   │   Along with twenty-seven members of the European Union including France, Germany, Italy, as well as countries like the United Kingdom, Canada, Japan, Korea, Australia, New Zealand, and many others, even Switzerland. ⏎\n",
-      "    │   │   We are inflicting pain on Russia and supporting the people of Ukraine. Putin is now isolated from the world more than ever. ⏎\n",
-      "    │   │   Together with our allies –we are right now enforcing powerful economic sanctions. ⏎\n",
-      "    │   │   We are cutting off Russia’s largest banks from the international financial system.  ⏎\n",
-      "    │   │   Preventing Russia’s central bank from defending the Russian Ruble making Putin’s $630 Billion “war fund” worthless.   ⏎\n",
-      "    │   │   We are choking off Russia’s access to technology that will sap its economic strength and weaken its military for years to come.  ⏎\n",
-      "    │   │   Tonight I say to the Russian oligarchs and corrupt leaders who have bilked billions of dollars off this violent regime no more. ⏎\n",
-      "    │   │   The U.S. Department of Justice is assembling a dedicated task force to go after the crimes of Russian oligarchs.  ⏎\n",
-      "    │   │   We are joining with our European allies to find and seize your yachts your luxury apartments your private jets. We are coming for your ill-begotten gains. ⏎\n",
-      "    │   │   And tonight I am announcing that we will join our allies in closing off American air space to all Russian flights – further isolating Russia – and adding an additional squeeze –on their economy. The Ruble has lost 30% of its value. ⏎\n",
-      "    │   │   The Russian stock market has lost 40% of its value and trading remains suspended. Russia’s economy is reeling and Putin alone is to blame. ⏎\n",
-      "    │   │   Together with our allies we are providing support to the Ukrainians in their fight for freedom. Military assistance. Economic assistance. Humanitarian assistance. ⏎\n",
-      "    │   │   We are giving more than $1 Billion in direct assistance to Ukraine. ⏎\n",
-      "    │   │   And we will continue to aid the Ukrainian people as they defend their country and to help ease their suffering.  ⏎\n",
-      "    │   │   Let me be clear, our forces are not engaged and will not engage in conflict with Russian forces in Ukraine.  ⏎\n",
-      "    │   │   Our forces are not going to Europe to fight in Ukraine, but to defend our NATO Allies – in the event that Putin decides to keep moving west.  ⏎\n",
-      "    │   │   For that pu\"⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   CONCISE SUMMARY:\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mPrompted\u001b[0m/3/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:22Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mResult\u001b[0m/4/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:22Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.000s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mresult\u001b[0m:  In this speech, the speaker addresses the American people and the world, emphasizing the strength of the Ukrainian people in the face of Russia's aggression. The speaker also announces that the US and its allies are imposing economic sanctions on Russia, and providing military, economic, and humanitarian assistance to Ukraine. The US is also joining its allies in closing off American airspace to all Russian flights, and the Department of Justice is assembling a task force to go after the crimes of Russian oligarchs. The speaker also clarifies that US forces are not engaging in conflict with Russian forces in Ukraine, but are there to defend NATO allies.\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mResult\u001b[0m/4/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:22Z\u001b[2m\u001b[0m\n",
-      "    └── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/5\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:22Z\u001b[2m\u001b[0m\n",
-      "\n",
-      "\u001b[38;5;15m2039cff8-944d-4daf-91e1-a78747fce7c0\u001b[1m\u001b[0m\n",
-      "└── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m7.129s\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mInput Function\u001b[0m/2/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.003s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4minput\u001b[0m: \u001b[0m\n",
-      "    │   │   └── \u001b[38;5;4mtext\u001b[0m:  why it is so important to pass the Bipartisan Innovation Act sitting in Congress that will make record investments in emerging technologies and American manufacturing. ⏎\n",
-      "    │   │       Let me give you one example of why it’s so important to pass it. ⏎\n",
-      "    │   │       If you travel 20 miles east of Columbus, Ohio, you’ll find 1,000 empty acres of land. ⏎\n",
-      "    │   │       It won’t look like much, but if you stop and look closely, you’ll see a “Field of dreams,” the ground on which America’s future will be built. ⏎\n",
-      "    │   │       This is where Intel, the American company that helped build Silicon Valley, is going to build its $20 billion semiconductor “mega site”. ⏎\n",
-      "    │   │       Up to eight state-of-the-art factories in one place. 10,000 new good-paying jobs. ⏎\n",
-      "    │   │       Some of the most sophisticated manufacturing in the world to make computer chips the size of a fingertip that power the world and our everyday lives. ⏎\n",
-      "    │   │       Smartphones. The Internet. Technology we have yet to invent. ⏎\n",
-      "    │   │       But that’s just the beginning. ⏎\n",
-      "    │   │       Intel’s CEO, Pat Gelsinger, who is here tonight, told me they are ready to increase their investment from  ⏎\n",
-      "    │   │       $20 billion to $100 billion. ⏎\n",
-      "    │   │       That would be one of the biggest investments in manufacturing in American history. ⏎\n",
-      "    │   │       And all they’re waiting for is for you to pass this bill. ⏎\n",
-      "    │   │       So let’s not wait any longer. Send it to my desk. I’ll sign it.  ⏎\n",
-      "    │   │       And we will really take off. ⏎\n",
-      "    │   │       And Intel is not alone. ⏎\n",
-      "    │   │       There’s something happening in America. ⏎\n",
-      "    │   │       Just look around and you’ll see an amazing story. ⏎\n",
-      "    │   │       The rebirth of the pride that comes from stamping products “Made In America.” The revitalization of American manufacturing.   ⏎\n",
-      "    │   │       Companies are choosing to build new factories here, when just a few years ago, they would have built them overseas. ⏎\n",
-      "    │   │       That’s what is happening. Ford is investing $11 billion to build electric vehicles, creating 11,000 jobs across the country. ⏎\n",
-      "    │   │       GM is making the largest investment in its history—$7 billion to build electric vehicles, creating 4,000 jobs in Michigan. ⏎\n",
-      "    │   │       All told, we created 369,000 new manufacturing jobs in America just last year. ⏎\n",
-      "    │   │       Powered by people I’ve met like JoJo Burgess, from generations of union steelworkers from Pittsburgh, who’s here with us tonight. ⏎\n",
-      "    │   │       As Ohio Senator Sherrod Brown says, “It’s time to bury the label “Rust Belt.” ⏎\n",
-      "    │   │       It’s time. ⏎\n",
-      "    │   │       But with all the bright spots in our economy, record job growth and higher wages, too many families are struggling to keep up with the bills.  ⏎\n",
-      "    │   │       Inflation is robbing them of the gains they might otherwise feel. ⏎\n",
-      "    │   │       I get it. That’s why my top priority is getting prices under control. ⏎\n",
-      "    │   │       Look, our economy roared back faster than most predicted, but the pandemic meant that businesses had a hard time hiring enough workers to keep up production in their factories. ⏎\n",
-      "    │   │       The pandemic also disrupted global supply chains. ⏎\n",
-      "    │   │       When factories close, it takes longer to make goods and get them from the warehouse to the store, and prices go up. ⏎\n",
-      "    │   │       Look at cars. ⏎\n",
-      "    │   │       Last year, there weren’t enough semiconductors to make all the cars that people wanted to buy. ⏎\n",
-      "    │   │       And guess what, prices of automobiles went up. ⏎\n",
-      "    │   │       So—we have a choice. ⏎\n",
-      "    │   │       One way to fight inflation is to drive down wages and make Americans poorer.  ⏎\n",
-      "    │   │       I have a better plan to fight inflation. ⏎\n",
-      "    │   │       Lower your costs, not your wages. ⏎\n",
-      "    │   │       Make more cars and semiconductors in America. ⏎\n",
-      "    │   │       More infrastructure and innovation in America. ⏎\n",
-      "    │   │       More goods moving faster and cheaper in America. ⏎\n",
-      "    │   │       More jobs where you can earn a good living in America. ⏎\n",
-      "    │   │       And instead of relying on foreign supply chains, let’s make it in America. ⏎\n",
-      "    │   │       Economists call it “increasing the productive capacity of our economy.�� ⏎\n",
-      "    │   │       I call it building a better America. ⏎\n",
-      "    │   │       My plan to fight inflation will lower your costs and lower the deficit. ⏎\n",
-      "    │   │       17 Nobel laureates in economics say my plan will ease long-term inflationary pressures. Top business leaders and most Americans support my plan. And here’s the plan: ⏎\n",
-      "    │   │       First – cut the cost of prescription drugs. Just look at insulin. One in ten Americans has diabetes. In Virginia, I met a 13-year-old boy named Joshua Davis.  ⏎\n",
-      "    │   │       He and his Dad both have Type 1 diabetes, which means they need insulin every day. Insulin costs about $10 a vial to make.  ⏎\n",
-      "    │   │       But drug companies charge families like Joshua and his Dad up to 30 times more. I spoke with Joshua’s mom. ⏎\n",
-      "    │   │       Imagine what it’s like to look at your child who needs insulin and have no idea how you’re going to pay for it.  ⏎\n",
-      "    │   │       What it does to your dignity, your ability to look your child in the eye, to be the parent you expect to be. ⏎\n",
-      "    │   │       Joshua is here with us tonight. Yesterday was his birthday. Happy birthday, buddy.  ⏎\n",
-      "    │   │       For Joshua, and for the 200,000 other young people with Type 1 diabetes, let’s cap the cost of insulin at $35 a month so everyone can afford it.  ⏎\n",
-      "    │   │       Drug companies will still do very well. And while we’re at it let Medicare negotiate lower prices for prescription drugs, like the VA a\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mInput Function\u001b[0m/2/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mPrompted\u001b[0m/3/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:16Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m7.125s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mprompt\u001b[0m: Write a concise summary of the following:⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   \" why it is so important to pass the Bipartisan Innovation Act sitting in Congress that will make record investments in emerging technologies and American manufacturing. ⏎\n",
-      "    │   │   Let me give you one example of why it’s so important to pass it. ⏎\n",
-      "    │   │   If you travel 20 miles east of Columbus, Ohio, you’ll find 1,000 empty acres of land. ⏎\n",
-      "    │   │   It won’t look like much, but if you stop and look closely, you’ll see a “Field of dreams,” the ground on which America’s future will be built. ⏎\n",
-      "    │   │   This is where Intel, the American company that helped build Silicon Valley, is going to build its $20 billion semiconductor “mega site”. ⏎\n",
-      "    │   │   Up to eight state-of-the-art factories in one place. 10,000 new good-paying jobs. ⏎\n",
-      "    │   │   Some of the most sophisticated manufacturing in the world to make computer chips the size of a fingertip that power the world and our everyday lives. ⏎\n",
-      "    │   │   Smartphones. The Internet. Technology we have yet to invent. ⏎\n",
-      "    │   │   But that’s just the beginning. ⏎\n",
-      "    │   │   Intel’s CEO, Pat Gelsinger, who is here tonight, told me they are ready to increase their investment from  ⏎\n",
-      "    │   │   $20 billion to $100 billion. ⏎\n",
-      "    │   │   That would be one of the biggest investments in manufacturing in American history. ⏎\n",
-      "    │   │   And all they’re waiting for is for you to pass this bill. ⏎\n",
-      "    │   │   So let’s not wait any longer. Send it to my desk. I’ll sign it.  ⏎\n",
-      "    │   │   And we will really take off. ⏎\n",
-      "    │   │   And Intel is not alone. ⏎\n",
-      "    │   │   There’s something happening in America. ⏎\n",
-      "    │   │   Just look around and you’ll see an amazing story. ⏎\n",
-      "    │   │   The rebirth of the pride that comes from stamping products “Made In America.” The revitalization of American manufacturing.   ⏎\n",
-      "    │   │   Companies are choosing to build new factories here, when just a few years ago, they would have built them overseas. ⏎\n",
-      "    │   │   That’s what is happening. Ford is investing $11 billion to build electric vehicles, creating 11,000 jobs across the country. ⏎\n",
-      "    │   │   GM is making the largest investment in its history—$7 billion to build electric vehicles, creating 4,000 jobs in Michigan. ⏎\n",
-      "    │   │   All told, we created 369,000 new manufacturing jobs in America just last year. ⏎\n",
-      "    │   │   Powered by people I’ve met like JoJo Burgess, from generations of union steelworkers from Pittsburgh, who’s here with us tonight. ⏎\n",
-      "    │   │   As Ohio Senator Sherrod Brown says, “It’s time to bury the label “Rust Belt.” ⏎\n",
-      "    │   │   It’s time. ⏎\n",
-      "    │   │   But with all the bright spots in our economy, record job growth and higher wages, too many families are struggling to keep up with the bills.  ⏎\n",
-      "    │   │   Inflation is robbing them of the gains they might otherwise feel. ⏎\n",
-      "    │   │   I get it. That’s why my top priority is getting prices under control. ⏎\n",
-      "    │   │   Look, our economy roared back faster than most predicted, but the pandemic meant that businesses had a hard time hiring enough workers to keep up production in their factories. ⏎\n",
-      "    │   │   The pandemic also disrupted global supply chains. ⏎\n",
-      "    │   │   When factories close, it takes longer to make goods and get them from the warehouse to the store, and prices go up. ⏎\n",
-      "    │   │   Look at cars. ⏎\n",
-      "    │   │   Last year, there weren’t enough semiconductors to make all the cars that people wanted to buy. ⏎\n",
-      "    │   │   And guess what, prices of automobiles went up. ⏎\n",
-      "    │   │   So—we have a choice. ⏎\n",
-      "    │   │   One way to fight inflation is to drive down wages and make Americans poorer.  ⏎\n",
-      "    │   │   I have a better plan to fight inflation. ⏎\n",
-      "    │   │   Lower your costs, not your wages. ⏎\n",
-      "    │   │   Make more cars and semiconductors in America. ⏎\n",
-      "    │   │   More infrastructure and innovation in America. ⏎\n",
-      "    │   │   More goods moving faster and cheaper in America. ⏎\n",
-      "    │   │   More jobs where you can earn a good living in America. ⏎\n",
-      "    │   │   And instead of relying on foreign supply chains, let’s make it in America. ⏎\n",
-      "    │   │   Economists call it “increasing the productive capacity of our economy.” ⏎\n",
-      "    │   │   I call it building a better America. ⏎\n",
-      "    │   │   My plan to fight inflation will lower your costs and lower the deficit. ⏎\n",
-      "    │   │   17 Nobel laureates in economics say my plan will ease long-term inflationary pressures. Top business leaders and most Americans support my plan. And here’s the plan: ⏎\n",
-      "    │   │   First – cut the cost of prescription drugs. Just look at insulin. One in ten Americans has diabetes. In Virginia, I met a 13-year-old boy named Joshua Davis.  ⏎\n",
-      "    │   │   He and his Dad both have Type 1 diabetes, which means they need insulin every day. Insulin costs about $10 a vial to make.  ⏎\n",
-      "    │   │   But drug companies charge families like Joshua and his Dad up to 30 times more. I spoke with Joshua’s mom. ⏎\n",
-      "    │   │   Imagine what it’s like to look at your child who needs insulin and have no idea how you’re going to pay for it.  ⏎\n",
-      "    │   │   What it does to your dignity, your ability to look your child in the eye, to be the parent you expect to be. ⏎\n",
-      "    │   │   Joshua is here with us tonight. Yesterday was his birthday. Happy birthday, buddy.  ⏎\n",
-      "    │   │   For Joshua, and for the 200,000 other young people with Type 1 diabetes, let’s cap the cost of insulin at $35 a month so everyone can afford it.  ⏎\n",
-      "    │   │   Drug companies will still do very well. And while we’re at it let Medicare negotiate lower prices for prescription drugs, like the VA a\"⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   CONCISE SUMMARY:\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mPrompted\u001b[0m/3/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:23Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mResult\u001b[0m/4/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:23Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.000s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mresult\u001b[0m:  The Bipartisan Innovation Act is an important bill that is currently sitting in Congress. It will make record investments in emerging technologies and American manufacturing, such as Intel's $20 billion semiconductor \"mega site\" in Ohio. This bill will create 10,000 new jobs and increase Intel's investment to $100 billion. Other companies, such as Ford and GM, are also investing in American manufacturing, creating 369,000 new jobs in the last year. President Biden's plan to fight inflation is to lower costs, not wages, by increasing the productive capacity of the economy. This includes capping the cost of insulin at $35 a month and allowing Medicare to negotiate lower prices for prescription drugs.\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mResult\u001b[0m/4/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:23Z\u001b[2m\u001b[0m\n",
-      "    └── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/5\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:23Z\u001b[2m\u001b[0m\n",
-      "\n",
-      "\u001b[38;5;15ma2de8aab-500a-4df0-8764-3b3c091db0b9\u001b[1m\u001b[0m\n",
-      "└── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:23Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m5.830s\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mInput Function\u001b[0m/2/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:23Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.001s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4minput\u001b[0m: \u001b[0m\n",
-      "    │   │   └── \u001b[38;5;4mtext\u001b[0m:  In this speech, the speaker addresses the American people and the world, emphasizing the strength of the Ukrainian people in the face of Russia's aggression. The speaker also announces that the US and its allies are imposing economic sanctions on Russia, and providing military, economic, and humanitarian assistance to Ukraine. The US is also joining its allies in closing off American airspace to all Russian flights, and the Department of Justice is assembling a task force to go after the crimes of Russian oligarchs. The speaker also clarifies that US forces are not engaging in conflict with Russian forces in Ukraine, but are there to defend NATO allies.⏎\n",
-      "    │   │        In response to Russian President Putin's invasion of Ukraine, the US and its allies are providing military, economic, and humanitarian assistance to the Ukrainians. The US is also leading an effort to release 60 million barrels of oil from reserves around the world, including 30 million from the US Strategic Petroleum Reserve, to help blunt gas prices. The US President has also passed the American Rescue Plan to provide economic relief to tens of millions of Americans and create jobs.⏎\n",
-      "    │   │        President Biden fought to pass the American Rescue Plan to provide economic relief to tens of millions of Americans and create jobs. He also passed the Bipartisan Infrastructure Law to rebuild America and the Bipartisan Innovation Act to level the playing field with China and other competitors. An example of this is Intel's $20 billion semiconductor \"mega site\" in Ohio, which will create 10,000 new jobs.⏎\n",
-      "    │   │        The Bipartisan Innovation Act is an important bill that is currently sitting in Congress. It will make record investments in emerging technologies and American manufacturing, such as Intel's $20 billion semiconductor \"mega site\" in Ohio. This bill will create 10,000 new jobs and increase Intel's investment to $100 billion. Other companies, such as Ford and GM, are also investing in American manufacturing, creating 369,000 new jobs in the last year. President Biden's plan to fight inflation is to lower costs, not wages, by increasing the productive capacity of the economy. This includes capping the cost of insulin at $35 a month and allowing Medicare to negotiate lower prices for prescription drugs.\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mInput Function\u001b[0m/2/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:23Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mPrompted\u001b[0m/3/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:23Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m5.828s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mprompt\u001b[0m: Write a concise summary of the following:⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   \" In this speech, the speaker addresses the American people and the world, emphasizing the strength of the Ukrainian people in the face of Russia's aggression. The speaker also announces that the US and its allies are imposing economic sanctions on Russia, and providing military, economic, and humanitarian assistance to Ukraine. The US is also joining its allies in closing off American airspace to all Russian flights, and the Department of Justice is assembling a task force to go after the crimes of Russian oligarchs. The speaker also clarifies that US forces are not engaging in conflict with Russian forces in Ukraine, but are there to defend NATO allies.⏎\n",
-      "    │   │    In response to Russian President Putin's invasion of Ukraine, the US and its allies are providing military, economic, and humanitarian assistance to the Ukrainians. The US is also leading an effort to release 60 million barrels of oil from reserves around the world, including 30 million from the US Strategic Petroleum Reserve, to help blunt gas prices. The US President has also passed the American Rescue Plan to provide economic relief to tens of millions of Americans and create jobs.⏎\n",
-      "    │   │    President Biden fought to pass the American Rescue Plan to provide economic relief to tens of millions of Americans and create jobs. He also passed the Bipartisan Infrastructure Law to rebuild America and the Bipartisan Innovation Act to level the playing field with China and other competitors. An example of this is Intel's $20 billion semiconductor \"mega site\" in Ohio, which will create 10,000 new jobs.⏎\n",
-      "    │   │    The Bipartisan Innovation Act is an important bill that is currently sitting in Congress. It will make record investments in emerging technologies and American manufacturing, such as Intel's $20 billion semiconductor \"mega site\" in Ohio. This bill will create 10,000 new jobs and increase Intel's investment to $100 billion. Other companies, such as Ford and GM, are also investing in American manufacturing, creating 369,000 new jobs in the last year. President Biden's plan to fight inflation is to lower costs, not wages, by increasing the productive capacity of the economy. This includes capping the cost of insulin at $35 a month and allowing Medicare to negotiate lower prices for prescription drugs.\"⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   ⏎\n",
-      "    │   │   CONCISE SUMMARY:\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mPrompted\u001b[0m/3/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:29Z\u001b[2m\u001b[0m\n",
-      "    ├── \u001b[38;5;5mResult\u001b[0m/4/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:29Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m0.000s\u001b[2m\u001b[0m\n",
-      "    │   ├── \u001b[38;5;4mresult\u001b[0m:  In response to Russia's aggression in Ukraine, President Biden has taken action to provide military, economic, and humanitarian assistance to Ukraine, impose economic sanctions on Russia, and close off American airspace to Russian flights. He has also passed the American Rescue Plan to provide economic relief to Americans and the Bipartisan Innovation Act to level the playing field with China and other competitors. This bill will create 10,000 new jobs and increase Intel's investment to $100 billion. President Biden is also fighting inflation by capping the cost of insulin at $35 a month and allowing Medicare to negotiate lower prices for prescription drugs.\u001b[0m\n",
-      "    │   └── \u001b[38;5;5mResult\u001b[0m/4/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:29Z\u001b[2m\u001b[0m\n",
-      "    └── \u001b[38;5;5m<class '__main__.SummaryPrompt'>\u001b[0m/5\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:29Z\u001b[2m\u001b[0m\n",
-      "\n",
-      "\u001b[38;5;15mb77ccab5-cdad-4436-a3e1-230da294a528\u001b[1m\u001b[0m\n",
-      "└── \u001b[38;5;5msummary\u001b[0m/1\u001b[0m ⇒ \u001b[38;5;2mstarted\u001b[0m \u001b[38;5;15m2023-02-26 01:42:15Z\u001b[2m\u001b[0m ⧖ \u001b[38;5;4m13.358s\u001b[2m\u001b[0m\n",
-      "    └── \u001b[38;5;5msummary\u001b[0m/2\u001b[0m ⇒ \u001b[38;5;2msucceeded\u001b[0m \u001b[38;5;15m2023-02-26 01:42:29Z\u001b[2m\u001b[0m\n",
-      "\n"
-     ]
-    }
-   ],
+   "execution_count": null,
+   "id": "2489bd19",
+   "metadata": {},
+   "outputs": [],
    "source": [
     "show_log(\"summary.log\")"
    ]
@@ -719,24 +128,9 @@
  ],
  "metadata": {
   "jupytext": {
-   "cell_metadata_filter": "tags,-all"
-  },
-  "kernelspec": {
-   "display_name": "minichain",
-   "language": "python",
-   "name": "minichain"
-  },
-  "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.10.6"
+   "cell_metadata_filter": "tags,-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
   }
  },
  "nbformat": 4,

temp.ipynb

@@ -0,0 +1,186 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "2c01babb",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1cdd0343",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "import minichain\n",
+    "from dataclasses import fields, dataclass, is_dataclass\n",
+    "from typing import List\n",
+    "from enum import Enum"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e2cbc112",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "class ColorType(Enum):\n",
+    "    RED = 1\n",
+    "    GREEN = 2\n",
+    "    BLUE = 3"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "4b939b84",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "@dataclass\n",
+    "class Color:\n",
+    "    color: ColorType\n",
+    "    object: str\n",
+    "    explanation: str"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "17655157",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "markdown",
+   "id": "1cf9ab63",
+   "metadata": {},
+   "source": [
+    "class StatType(Enum):\n",
+    "    POINTS = 1\n",
+    "    REBOUNDS = 2\n",
+    "    ASSISTS = 3"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "43d8ada4",
+   "metadata": {},
+   "source": [
+    "@dataclass\n",
+    "class Stat:\n",
+    "    value: int\n",
+    "    stat: StatType"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "c5e29bce",
+   "metadata": {},
+   "source": [
+    "@dataclass\n",
+    "class Player:\n",
+    "    player: str\n",
+    "    stats: List[Stat]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "15f1c0ff",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "class T(minichain.TypedTemplatePrompt):\n",
+    "    template_file = \"stats.pmpt.tpl\"\n",
+    "    Out = Color"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "35f81370",
+   "metadata": {},
+   "source": [
+    "print(T().show({\"passage\": \"hello\"}, '[{\"player\": \"Harden\", \"stats\": {\"value\": 10, \"stat\": 2}}]'))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "07f012b6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "with minichain.start_chain(\"stats\") as backend:\n",
+    "    p = T(backend.OpenAI(max_tokens=512))\n",
+    "    print(p({\"passage\": open(\"sixers.txt\").read()}))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "b4f0d711",
+   "metadata": {
+    "lines_to_next_cell": 2
+   },
+   "source": [
+    "def enum(x):\n",
+    "    d = {e.name: e.value for e in x}\n",
+    "    # d[\"__enum__\"] = True\n",
+    "    return d"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "4aa96598",
+   "metadata": {},
+   "source": [
+    "def walk(x):\n",
+    "    print(x)\n",
+    "    if issubclass(x, Enum):\n",
+    "        return enum(x)\n",
+    "    if is_dataclass(x):\n",
+    "        return {y.name: walk(y.type) for y in fields(x)}\n",
+    "    return x.__name__\n",
+    "    # return [x for x in fields(B)]\n",
+    "            # print(x.name)\n",
+    "            # print(x.type)\n",
+    "            # if issubclass(x.type, Enum):\n",
+    "            # for e in x.type:\n",
+    "            # print(e.value)\n",
+    "            # print(e.name)\n",
+    "            # print(x)]"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "146e031e",
+   "metadata": {},
+   "source": [
+    "print(walk(B))"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

test.ipynb

@@ -0,0 +1,102 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e38ce387",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -q git+https://github.com/srush/MiniChain\n",
+    "!git clone git+https://github.com/srush/MiniChain; cp -fr MiniChain/examples/* . "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "819e17b6",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "from minichain import prompt, Mock, show\n",
+    "import minichain"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3cd78b45",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "@prompt(Mock([\"red\", \"blue\"]))\n",
+    "def prompt_function1(model, x):\n",
+    "    return model(x)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3be33c12",
+   "metadata": {
+    "lines_to_next_cell": 1
+   },
+   "outputs": [],
+   "source": [
+    "@prompt(Mock([\"b\"]), template_file=\"test.pmpt.tpl\")\n",
+    "def prompt_function2(model, x):\n",
+    "    if x == \"red\":\n",
+    "        return model.fail(1)\n",
+    "    return model(dict(x=x))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "9bb94056",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def run(query):\n",
+    "    x = prompt_function1(query)\n",
+    "    return prompt_function2(prompt_function2(x))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "1f0d5d5a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "demo = show(run,\n",
+    "            examples=[\"a\"],\n",
+    "            subprompts=[prompt_function1, prompt_function2, prompt_function2])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "be9be168",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "if __name__ == \"__main__\":\n",
+    "    demo.launch()"
+   ]
+  }
+ ],
+ "metadata": {
+  "jupytext": {
+   "cell_metadata_filter": "-all",
+   "main_language": "python",
+   "notebook_metadata_filter": "-all"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

test.py

@@ -1,12 +1,14 @@
 from minichain import prompt, Mock, show
 import minichain
 
-@prompt(Mock(["hello"]))
+@prompt(Mock(["red", "blue"]))
 def prompt_function1(model, x):
     return model(x)
 
 @prompt(Mock(["b"]), template_file="test.pmpt.tpl")
 def prompt_function2(model, x):
+    if x == "red":
+        return model.fail(1)
     return model(dict(x=x))
     
 def run(query):