Refactor

1 files changed, 52 insertions, 1 deletions

diff --git a/notebooks/testing.ipynb b/notebooks/testing.ipynb
index 18827fc..bcaa613 100644
--- a/notebooks/testing.ipynb
+++ b/notebooks/testing.ipynb
@@ -98,10 +98,61 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 5,
    "id": "05f7068c-b4c6-47b2-ac62-79c021838500",
    "metadata": {},
    "outputs": [],
+   "source": [
+    "from rag.generator import get_generator"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "id": "31dd3e7a-1d6c-4cc6-8ffe-f83478f95875",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "x = get_generator(\"ollama\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "id": "749c58c4-5250-464c-b4f4-b5b4a00be3b3",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "y = get_generator(\"ollama\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "id": "c59ece14-11e0-416c-afad-bc2ff6a526ec",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "True"
+      ]
+     },
+     "execution_count": 8,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "x is y"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0e664d3d-a787-45e5-8b71-8e5e0f348443",
+   "metadata": {},
+   "outputs": [],
    "source": []
   }
  ],