Notebook updated

superduper-io · Dec 2, 2023 · 97c0268 · 97c0268
1 parent 8d64772
commit 97c0268
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/chat_with_your_database.ipynb b/examples/chat_with_your_database.ipynb
@@ -122,8 +122,8 @@
     "    output, context = db.predict(\n",
     "        model_name='gpt-3.5-turbo',\n",
     "        input=search_term,\n",
-    "        context_raw=db.execute(RawSQL(f'DESCRIBE {table_name}')).as_pandas().to_csv()\n",
-    "        # context_raw=db.execute(RawSQL(f'SELECT * FROM {table_name} LIMIT 10')).as_pandas().to_csv() # Use in case of some other SQL databases like Postgres where `DESCRIBE` is not supported.\n",
+    "        context_select=db.execute(RawSQL(f'DESCRIBE {table_name}')).as_pandas().to_csv()\n",
+    "        # context_select=db.execute(RawSQL(f'SELECT * FROM {table_name} LIMIT 10')).as_pandas().to_csv() # Use in case of some other SQL databases like Postgres where `DESCRIBE` is not supported.\n",
     "    )\n",
     "    \n",
     "    try:\n",