From 096b82f2a1106c022f98675d466379d523b85fa6 Mon Sep 17 00:00:00 2001
From: blob42 <spike@w530>
Date: Thu, 2 Mar 2023 20:32:02 +0100
Subject: [PATCH] update notebook for utility

---
 docs/modules/utils/examples/docker.ipynb | 69 ++++++++++++++++--------
 1 file changed, 47 insertions(+), 22 deletions(-)

diff --git a/docs/modules/utils/examples/docker.ipynb b/docs/modules/utils/examples/docker.ipynb
index 50c57d71..168cb16b 100644
--- a/docs/modules/utils/examples/docker.ipynb
+++ b/docs/modules/utils/examples/docker.ipynb
@@ -1,14 +1,5 @@
 {
  "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "jukit_cell_id": "O4HPx3boF0"
-   },
-   "outputs": [],
-   "source": []
-  },
   {
    "cell_type": "markdown",
    "metadata": {
@@ -26,7 +17,7 @@
    },
    "outputs": [],
    "source": [
-    "from langchain.utilities.docker import DockerWrapper, DockerSocket"
+    "from langchain.utilities.docker import DockerWrapper"
    ]
   },
   {
@@ -37,13 +28,7 @@
    },
    "outputs": [],
    "source": [
-    "d = DockerWrapper()\n",
-    "query = \"\"\"\n",
-    "for i in $(seq 1 10)\n",
-    "do\n",
-    "    echo $i\n",
-    "done\n",
-    "\"\"\""
+    "d = DockerWrapper(image='shell')"
    ]
   },
   {
@@ -57,8 +42,6 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "[header] blocking IO\n",
-      "{'stdout': '1\\n2\\n3\\n4\\n5\\n6\\n7\\n8\\n9\\n10\\n'}\n",
       "1\n",
       "2\n",
       "3\n",
@@ -68,13 +51,55 @@
       "7\n",
       "8\n",
       "9\n",
-      "10\n",
-      "\n"
+      "10\n"
+     ]
+    }
+   ],
+   "source": [
+    "query = \"\"\"\n",
+    "for i in $(seq 1 10)\n",
+    "do\n",
+    "    echo $i\n",
+    "done\n",
+    "\"\"\"\n",
+    "print(d.exec_run(query))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "jukit_cell_id": "lGMqLz5sDo"
+   },
+   "outputs": [],
+   "source": [
+    "p = DockerWrapper(image='python')\n",
+    "\n",
+    "py_payload = \"\"\"\n",
+    "def hello_world():\n",
+    "    return 'hello world'\n",
+    "\n",
+    "hello_world()\n",
+    "\"\"\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {
+    "jukit_cell_id": "X04Wd6zbrk"
+   },
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "'hello world'\n"
      ]
     }
    ],
    "source": [
-    "print(d.exec_run(query, \"alpine\"))"
+    "print(p.exec_run(py_payload))"
    ]
   }
  ],