Update app.py

app.py

@@ -9,6 +9,7 @@ from llama_index.vector_stores import ChromaVectorStore
 from llama_index.storage.storage_context import StorageContext
 import chromadb
 from llama_index.readers.wikipedia import WikipediaReader
+from llama_index import SQLDatabase
 from llama_index.tools import FunctionTool
 from llama_index.vector_stores.types import (
     VectorStoreInfo,
@@ -156,6 +157,10 @@ async def init():
         "oppenheimer",
         engine
     )
+    
+    sql_database = SQLDatabase(
+        engine, 
+        include_tables=['barbie', 'oppenheimer'])
 
     sql_query_engine = NLSQLTableQueryEngine(
         sql_database=sql_database,