Create script.py

script.py

@@ -0,0 +1,27 @@
+import os
+from langchain import OpenAI, SQLDatabase, SQLDatabaseChain
+
+api_key = os.getenv('OPENAI_API_KEY')
+
+db = SQLDatabase.from_uri("sqlite:///Chinook.db"
+    , include_tables=['Customer', 'Invoice']
+)
+
+llm = OpenAI(temperature=0.0, openai_api_key=api_key)
+
+db_chain = SQLDatabaseChain.from_llm(llm
+    , db
+    , verbose=True
+    , use_query_checker=True
+)
+
+queries = ("How many customers are there?"
+    , "Which state has the most customers?"
+    , "How many invoices are there?"
+    , "Which customer has the most invoices?"
+)
+
+for query in queries:
+    print(db_chain.run(query))
+
+# https://www.sqlitetutorial.net/sqlite-sample-database/