From 39eed11904edb3bada9a3dc9258930637d13d26a Mon Sep 17 00:00:00 2001 From: poornimaramesh Date: Fri, 16 Aug 2024 23:49:53 +0300 Subject: [PATCH] update demo to include db type --- demo.ipynb | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/demo.ipynb b/demo.ipynb index ac25948..b980516 100644 --- 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",