update demo to include db type

IDinsight · Aug 16, 2024 · 39eed11 · 39eed11
1 parent 3cf4250
commit 39eed11
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/demo.ipynb b/demo.ipynb
@@ -32,9 +32,10 @@
     "    AsyncSession,\n",
     "    create_async_engine,\n",
     ")\n",
+    "db_type=\"sqlite\"\n",
     "\n",
     "aengine = create_async_engine(\n",
-    "    url=\"sqlite+aiosqlite:///demo_databases/tn_covid_cases_11_may.sqlite\"\n",
+    "    url=f\"{db_type}+aiosqlite:///demo_databases/tn_covid_cases_11_may.sqlite\"\n",
     ")\n",
     "\n",
     "async_session = sessionmaker(\n",
@@ -82,6 +83,7 @@
     "        query,\n",
     "        session,\n",
     "        which_db,\n",
+    "        db_type,\n",
     "        llm,\n",
     "        guardrails_llm,\n",
     "        sys_message,\n",