fixed a defect in the sandbox

Author: alessioricco

README.md

@@ -52,14 +52,22 @@ data = [('John Doe', 25, 50),
         ('Olivia Jackson', 29, 55)]
 df = pd.DataFrame(data, columns=['name', 'age', 'donation'])
 
-conv_df = PandasLLM(data=df, llm_api_key = os.environ.get("OPENAI_API_KEY"), verbose=True)
-result = conv_df.prompt("What is the average donation of people older than 30 who donated more than $50?")
+conv_df = PandasLLM(data=df, llm_api_key = os.environ.get("OPENAI_API_KEY"))
+result = conv_df.prompt("What is the average donation of people older than 40 who donated more than $50?")
+code = conv_df.code_block
+
+print(f"Executing the following expression of type {type(result)}:\n{code}\n\nResult is:\n {result}\n")
+# Executing the following expression of type <class 'numpy.float64'>:
+# result = df.loc[(df['age'] > 40) & (df['donation'] > 50), 'donation'].mean()
+
+# Result is:
+#  72.5
 
-print(f"Result ({type(result)}):\n {result}")
-# Result (<class 'numpy.float64'>):
-#  75.0
 ```
 
+There is also a chatbot available in the repository using the same dataset. 
+Look at example-chatbot.py 
+
 ## Contributing
 Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change. Please make sure to update tests as appropriate.
 

pandas_llm/__init__.py

@@ -333,7 +333,7 @@ def prompt(self, request: str):
         for cleaned_code in results:
 
             try:
-                result = self.execInSandbox(self, cleaned_code)
+                result = self._execInSandbox(self, cleaned_code)
             except Exception as e:
                 self._print(f"error {e}")
                 if not self.force_sandbox:

pandas_llm/example-chatbot.py

@@ -49,13 +49,8 @@
 
 def main():
 
-    # Set the OpenAI API key
-    openai_key = os.environ.get("OPENAI_API_KEY")
-    if openai_key is None:
-        print("No OpenAI API key provided. Exiting.")
-        return
-
-    conv_df = PandasLLM(data=df, llm_api_key = openai_key)
+    # Initialise library and set the OpenAI API key
+    conv_df = PandasLLM(data=df, llm_api_key = os.environ.get("OPENAI_API_KEY"))
     print()
     banner = """
     Welcome to the Donation Data CLI.
@@ -77,7 +72,9 @@ def main():
             break
 
         result = conv_df.prompt(prompt)
-        print(f"Result ({type(result)}):\n {result}")
+        code = conv_df.code_block
+        print(f"Executing the following expression of type {type(result)}:\n{code}\n\nResult is:\n {result}\n")
+        
 
 if __name__ == "__main__":
     main()

pandas_llm/example.py

@@ -21,9 +21,13 @@
         ('Olivia Jackson', 29, 55)]
 df = pd.DataFrame(data, columns=['name', 'age', 'donation'])
 
-conv_df = PandasLLM(data=df, llm_api_key = os.environ.get("OPENAI_API_KEY"), verbose=True)
-result = conv_df.prompt("What is the average donation of people older than 30 who donated more than $50?")
+conv_df = PandasLLM(data=df, llm_api_key = os.environ.get("OPENAI_API_KEY"))
+result = conv_df.prompt("What is the average donation of people older than 40 who donated more than $50?")
+code = conv_df.code_block
 
-print(f"Result ({type(result)}):\n {result}")
-# Result (<class 'numpy.float64'>):
-#  75.0
+print(f"Executing the following expression of type {type(result)}:\n{code}\n\nResult is:\n {result}\n")
+# Executing the following expression of type <class 'numpy.float64'>:
+# result = df.loc[(df['age'] > 40) & (df['donation'] > 50), 'donation'].mean()
+
+# Result is:
+#  72.5