Completed first tranche of db_wiz

elm/db_wiz.py

@@ -140,6 +140,7 @@ def get_sql_for(self, query):
                    'Please only return the SQL query with no commentary or preface.'
                    .format(self.database_describe, query))
         out = super().chat(e_query, temperature=0)
+        print(out)
         return out
 
     def run_sql(self, sql):

examples/energy_wizard/run_db_wizard_app.py

@@ -27,6 +27,7 @@
 
 st.set_option('deprecation.showPyplotGlobalUse', False)
 
+
 if __name__ == '__main__':
     wizard = DataBaseWizard(model = model, connection_string = conn_string)
 
@@ -46,6 +47,17 @@
 
     if "messages" not in st.session_state:
         st.session_state.messages = []
+    else:
+        if st.button('Clear Chat'):
+            # Clearing Messages
+            st.session_state.messages = []
+            for message in st.session_state.messages:
+                with st.chat_message(message["role"]):
+                    st.markdown(message["content"])
+
+            # Clearing Wizard
+            wizard.clear()
+            wizard = DataBaseWizard(model = model, connection_string = conn_string)
 
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
@@ -75,3 +87,4 @@
 
         #st.session_state.messages.append({"role": "assistant",
         #                                  "content": full_response})
+

examples/energy_wizard/test_db_wizard_v2.ipynb

@@ -117,7 +117,7 @@
       "AND week_type = 'winter'\n",
       "AND sector = 'res'\n",
       "AND enduse = 'heating'\n",
-      "AND load_center <= 5\n",
+      "AND load_center BETWEEN 1 AND 5\n",
       "ORDER BY timestamp;\n"
      ]
     }