Update utils.py

utils.py

@@ -362,6 +362,18 @@ class SQLDatabaseChainPatched(SQLDatabaseChain):
             ).strip()
             # self.llm_chain = self.revert_to_small_model(chain=self.llm_chain)
             self.intermediate_steps['result'] = final_result
+
+            # provide explanation
+            input_text += f"{final_result}\nExplanation:"
+            llm_inputs["input"] = input_text
+            self.llm_chain, n_tokens4 = self.prepare_llm(llm_inputs, chain=self.llm_chain)
+            # self.intermediate_steps['n_tokens_list'].append(n_tokens3)
+            explanation = self.llm_chain.predict(
+                callbacks=_run_manager.get_child(),
+                **llm_inputs,
+            ).strip()
+            # self.llm_chain = self.revert_to_small_model(chain=self.llm_chain)
+            self.intermediate_steps['query_explanation'] = explanation
             
             #if 'result' in self.intermediate_steps:
             #    self.intermediate_steps['translated_result'] = translate(self.llms['4k'], orig_question, self.intermediate_steps['result'])
@@ -379,6 +391,18 @@ class SQLDatabaseChainPatched(SQLDatabaseChain):
                 ).strip()
                 # self.llm_chain = self.revert_to_small_model(chain=self.llm_chain)
                 self.intermediate_steps['result'] = final_result
+    
+                # provide explanation
+                input_text += f"{final_result}\nExplanation:"
+                llm_inputs["input"] = input_text
+                self.llm_chain, n_tokens4 = self.prepare_llm(llm_inputs, chain=self.llm_chain)
+                # self.intermediate_steps['n_tokens_list'].append(n_tokens3)
+                explanation = self.llm_chain.predict(
+                    callbacks=_run_manager.get_child(),
+                    **llm_inputs,
+                ).strip()
+                # self.llm_chain = self.revert_to_small_model(chain=self.llm_chain)
+                self.intermediate_steps['query_explanation'] = explanation
                 #if 'result' in self.intermediate_steps:
                 #    self.intermediate_steps['translated_result'] = translate(self.llms['4k'], orig_question, self.intermediate_steps['result'])